[原创]DBTransaction类

#define AC2008
using System;
using System.IO;
using System.Collections;
using System.Collections.Generic;
using System.Collections.Specialized;
using Autodesk..ApplicationServices;
using Autodesk.AutoCAD.DatabaseServices;
using Autodesk.AutoCAD.EditorInput;
using Autodesk.AutoCAD.Geometry;
namespace TlsCad
{
    public class DBTransaction : IDisposable
    {
        private Database m_Database;
        private Transaction m_Transaction;
        private bool m_Commit = false;
        private BlockTableRecord m_BlockTableRecord;
        private BlockTable m_BlockTable;
        private TextStyleTable m_TextStyleTable;
        private LinetypeTable m_LinetypeTable;
        private RegAppTable m_RegAppTable;
        private LayerTable m_LayerTable;
        private UcsTable m_UcsTable;
        private ViewTable m_ViewTable;
        private ViewportTable m_ViewportTable;
        private DrawOrderTable m_DrawOrderTable;
        private DBDictionary m_GroupDictionary;
        //事务相关
        #region Tran
        //事务初始化,并只读方式打开块表
        private void Initialize()
        {
            try
            {
                m_Transaction = m_Database.TransactionManager.StartTransaction();
                m_BlockTable = (BlockTable)m_Transaction.GetObject(m_Database.BlockTableId, OpenMode.ForRead, false);
            }
            catch (Autodesk.AutoCAD.Runtime.Exception ex)
            {
                CadHelper.InfoMessageBox(ex.Message);
            }
        }
        //创建当前活动文档的事务(默认提交)
        public DBTransaction()
        {
            m_Database = HostApplicationServices.WorkingDatabase;
            Initialize();
        }
        //创建当前活动文档的事务
        //commit = true 提交事务
        //commit = false 不提交
        public DBTransaction(bool commit)
        {
            m_Commit = !commit;
            m_Database = HostApplicationServices.WorkingDatabase;
            Initialize();
        }
        //创建指定数据库的事务,一般用于临时数据库(默认提交)
        public DBTransaction(Database database)
        {
            m_Database = database;
            Initialize();
        }
        //创建指定数据库的事务
        //commit = true 提交事务
        //commit = false 不提交
        public DBTransaction(Database database, bool commit)
        {
            m_Commit = !commit;
            m_Database = database;
            Initialize();
        }
        //创建临时数据库的事务,并读入指定的文档(默认提交)
        public DBTransaction(string fileName)
        {
            try
            {
                m_Database = new Database(false, true);
                m_Database.ReadDwgFile(fileName, FileShare.Read, true, null);
                Initialize();
            }
            catch
            { }
        }
        //创建临时数据库的事务,并读入指定的文档
        //commit = true 提交事务
        //commit = false 不提交
        public DBTransaction(string DwgFileName, bool commit)
        {
            m_Commit = !commit;
            try
            {
                m_Database = new Database(false, true);
                m_Database.ReadDwgFile(DwgFileName, FileShare.Read, true, null);
                Initialize();
            }
            catch
            { }
        }
        //
        void IDisposable.Dispose()
        {
            Commit();
            m_Transaction.Dispose();
        }
        //按事务初始化参数决定事物是否提交
        public void Commit()
        {
            if (!m_Commit)
            {
                m_Transaction.Commit();
                m_Commit = true;
            }
        }
        //撤销事务
        public void Abort()
        {
            m_Transaction.Abort();
        }
        #endregion
        //在符号表中获取对应键值的记录Id
        //弥补索引器的Bug
        //即会获取已清除并存在符号表的记录
        //但2010版该Bug已消除
        public ObjectId GetIdFromSymbolTable(SymbolTable st, string key)
        {
            if (st.Has(key))
            {
                ObjectId idres = st[key];
                if (!idres.IsErased)
                    return idres;
#if AC2008
                foreach (ObjectId id in st)
                {
                    if (!id.IsErased)
                    {
                        SymbolTableRecord str = (SymbolTableRecord)m_Transaction.GetObject(id, OpenMode.ForRead);
                        if (str.Name == key)
                            return id;
                    }
                }
#endif
            }
            return ObjectId.Null;
        }
        //属性
        #region Properties
        public Database Database
        {
            get { return m_Database; }
        }
        public Transaction Transaction
        {
            get { return m_Transaction; }
        }
        public BlockTableRecord CurrentBlockTableRecord
        {
            get { return m_BlockTableRecord; }
        }
        public BlockTable BlockTable
        {
            get { return m_BlockTable; }
        }        public LayerTable LayerTable
        {
            get { return m_LayerTable; }
        }
        public TextStyleTable TextStyleTable
        {
            get { return m_TextStyleTable; }
        }
        public RegAppTable RegAppTable
        {
            get { return m_RegAppTable; }
        }
        public LinetypeTable LinetypeTable
        {
            get { return m_LinetypeTable; }
        }
        public UcsTable UcsTable
        {
            get { return m_UcsTable; }
        }
        public ViewTable ViewTable
        {
            get { return m_ViewTable; }
        }
        public ViewportTable ViewportTable
        {
            get { return m_ViewportTable; }
        }
        public DrawOrderTable DrawOrderTable
        {
            get { return m_DrawOrderTable; }
        }
        public DBDictionary GroupDictionary
        {
            get { return m_GroupDictionary; }
        }
        #endregion
        //块表记录
        #region BlockTableRecord
        
        public BlockTableRecord OpenBlockTableRecord(ObjectId id, OpenMode openmode)
        {
            m_BlockTableRecord = (BlockTableRecord)m_Transaction.GetObject(
                id,
                openmode,
                false);
            return m_BlockTableRecord;
        }
        public BlockTableRecord OpenBlockTableRecord(ObjectId id)
        {
            return OpenBlockTableRecord(id, OpenMode.ForWrite);
        }
        public BlockTableRecord OpenBlockTableRecord(string name, OpenMode openmode)
        {
            ObjectId id = GetIdFromSymbolTable(m_BlockTable, name);
            if (id == ObjectId.Null)
                return null;
            return OpenBlockTableRecord(id, openmode);
        }
        public BlockTableRecord OpenBlockTableRecord(string name)
        {
            return OpenBlockTableRecord(m_BlockTable[name], OpenMode.ForWrite);
        }
        public void OpenCurrentSpace(OpenMode openmode)
        {
            OpenBlockTableRecord(m_Database.CurrentSpaceId, openmode);
        }
        public void OpenCurrentSpace()
        {
            OpenCurrentSpace(OpenMode.ForWrite);
        }
        public void OpenPaperSpace(OpenMode openmode)
        {
            OpenBlockTableRecord(BlockTableRecord.PaperSpace, openmode);
        }
        public void OpenPaperSpace()
        {
            OpenPaperSpace(OpenMode.ForWrite);
        }
        public void OpenModelSpace(OpenMode openmode)
        {
            OpenBlockTableRecord(BlockTableRecord.ModelSpace, openmode);
        }        public void OpenModelSpace()
        {
            OpenModelSpace(OpenMode.ForWrite);
        }
        public DBObject GetObject(ObjectId id, OpenMode openmode)
        {
            return m_Transaction.GetObject(id, openmode, false);
        }
        #endregion
        //层表
        #region LayerTable
        public LayerTable OpenLayerTable(OpenMode openmode)
        {
            m_LayerTable = (LayerTable)m_Transaction.GetObject(
                m_Database.LayerTableId,
                openmode,
                false);
            return m_LayerTable;
        }
        public LayerTable OpenLayerTable()
        {
            return OpenLayerTable(OpenMode.ForWrite);
        }
        public ObjectId AddLayer(string name, Autodesk.AutoCAD.Colors.Color color, ObjectId linetypeid, LineWeight lineweight)
        {
            ObjectId id = GetIdFromSymbolTable(m_LayerTable, name);
            if (id == ObjectId.Null)
            {
                LayerTableRecord layer = new LayerTableRecord();
                layer.Name = name;
                layer.Color = color;
                layer.LinetypeObjectId = linetypeid;
                layer.LineWeight = lineweight;
                m_LayerTable.UpgradeOpen();
                id = m_LayerTable.Add(layer);
                m_Transaction.AddNewlyCreatedDBObject(layer, true);
            }
            return id;
        }
        #endregion
        //文字式样表
        #region TextStyleTable
        public TextStyleTable OpenTextStyleTable(OpenMode openmode)
        {
            m_TextStyleTable = (TextStyleTable)m_Transaction.GetObject(
                m_Database.TextStyleTableId,
                openmode,
                false);
            return m_TextStyleTable;
        }
        public TextStyleTable OpenTextStyleTable()
        {
            return OpenTextStyleTable(OpenMode.ForWrite);
        }
        public ObjectId AddTextStyle(string name, string smallfont, string bigfont, double height, double xscale)
        {
            ObjectId id = GetIdFromSymbolTable(m_TextStyleTable, name);
            if (id == ObjectId.Null)
            {
                TextStyleTableRecord tstr = new TextStyleTableRecord();
                tstr.Name = name;
                tstr.FileName = smallfont;
                tstr.BigFontFileName = bigfont;
                tstr.TextSize = height;
                tstr.XScale = xscale;
                m_TextStyleTable.UpgradeOpen();
                id = m_TextStyleTable.Add(tstr);
                m_Transaction.AddNewlyCreatedDBObject(tstr, true);
            }
            return id;
        }        #endregion
        //注册应用程序
        #region RegAppTable
        public RegAppTable OpenRegAppTable(OpenMode openmode)
        {
            m_RegAppTable = (RegAppTable)m_Transaction.GetObject(
                m_Database.RegAppTableId,
                openmode,
                false);
            return m_RegAppTable;
        }
        public RegAppTable OpenRegAppTable()
        {
            return OpenRegAppTable(OpenMode.ForWrite);
        }
        public ObjectId RegApp(string name)
        {
            ObjectId id = GetIdFromSymbolTable(m_RegAppTable, name);
            if (id == ObjectId.Null)
            {
                RegAppTableRecord regapp = new RegAppTableRecord();
                regapp.Name = name;
                id = m_RegAppTable.Add(regapp);
                m_Transaction.AddNewlyCreatedDBObject(regapp, true);
            }
            return id;
        }
        #endregion
        //线型表
        #region LinetypeTable
        public LinetypeTable OpenLinetypeTable(OpenMode openmode)
        {
            m_LinetypeTable = (LinetypeTable)m_Transaction.GetObject(
                m_Database.LinetypeTableId,
                openmode,
                false);
            return m_LinetypeTable;
        }
        public LinetypeTable OpenLinetypeTable()
        {
            return OpenLinetypeTable(OpenMode.ForWrite);
        }
        public SelectionSet SelectByLineWeight(LineWeight lineWeight)
        {
            List filter = new List();
            filter.Add(new TypedValue(370, lineWeight));
            OpenLayerTable(OpenMode.ForRead);
            List lays = new List();
            foreach (ObjectId id in LayerTable)
            {
                LayerTableRecord ltr = (LayerTableRecord)GetObject(id, OpenMode.ForRead);
                if (ltr.LineWeight == lineWeight)
                {
                    lays.Add(ltr.Name);
                }
            }
            if (lays.Count > 0)
            {
                string s = string.Join(",", lays.ToArray());
                filter.Insert(0, new TypedValue(-4, ""));
                filter.Add(new TypedValue(-4, "or>"));
            }
            PromptSelectionResult res = CadHelper.Editor.SelectAll(new SelectionFilter(filter.ToArray()));
            return res.Value;
        }
        #endregion
        //用户坐标系
        #region UcsTable
        public UcsTable OpenUcsTable(OpenMode openmode)
        {
            m_UcsTable = (UcsTable)m_Transaction.GetObject(
                m_Database.UcsTableId,
                openmode,
                false);
            return m_UcsTable;
        }
        public UcsTable OpenUcsTable()
        {
            return OpenUcsTable(OpenMode.ForWrite);
        }
        #endregion
        //视图
        #region ViewTable
        public ViewTable OpenViewTable(OpenMode openmode)
        {
            m_ViewTable = (ViewTable)m_Transaction.GetObject(
                m_Database.ViewTableId,
                openmode,
                false);
            return m_ViewTable;
        }
        public ViewTable OpenViewTable()
        {
            return OpenViewTable(OpenMode.ForWrite);
        }
        #endregion
        //视口
        #region ViewportTable
        public ViewportTable OpenViewportTable(OpenMode openmode)
        {
            m_ViewportTable = (ViewportTable)m_Transaction.GetObject(
                m_Database.ViewportTableId,
                openmode,
                false);
            return m_ViewportTable;
        }
        public ViewportTable OpenViewportTable()
        {
            return OpenViewportTable(OpenMode.ForWrite);
        }
        #endregion
        //调整实体显示
        #region DrawOrderTable
        public DrawOrderTable OpenDrawOrderTable(OpenMode openmode)
        {
            m_DrawOrderTable = (DrawOrderTable)m_Transaction.GetObject(
                m_BlockTableRecord.DrawOrderTableId,
                openmode,
                false);
            return m_DrawOrderTable;
        }
        public DrawOrderTable OpenDrawOrderTable()
        {
            return OpenDrawOrderTable(OpenMode.ForWrite);
        }
        #endregion
        //编组字典
        #region GroupDictionary
        public DBDictionary OpenGroupDictionary(OpenMode openmode)
        {
            m_GroupDictionary = (DBDictionary)m_Transaction.GetObject(
                m_Database.GroupDictionaryId,
                openmode,
                false);
            return m_GroupDictionary;
        } 
        public DBDictionary OpenGroupDictionary()
        {
            return OpenGroupDictionary(OpenMode.ForWrite);
        }
        public ObjectId AddGroup(string name, ObjectIdCollection ids)
        {
            if (m_GroupDictionary.Contains(name))
            {
                return ObjectId.Null;
            }
            else
            {
                m_GroupDictionary.UpgradeOpen();
                Group g = new Group();
                g.Append(ids);
                m_GroupDictionary.SetAt(name, g);
                m_Transaction.AddNewlyCreatedDBObject(g, true);
                return g.ObjectId;
            }
        }
        public List GetGroups(Entity ent)
        {
            List gs = new List();
            foreach (DBDictionaryEntry obj in m_GroupDictionary)
            {
                Group g = 
                    (Group)m_Transaction.GetObject(
                    obj.Value, 
                    OpenMode.ForRead,
                    false);
                if (g.Has(ent))
                    gs.Add(g);
            }
            return gs;
        }
        #endregion
        //在当前块表记录中加入实体
        #region Add Entity
        public ObjectId AddEntity(Entity entity)
        {
            ObjectId id = m_BlockTableRecord.AppendEntity(entity);
            m_Transaction.AddNewlyCreatedDBObject(entity, true);
            return id;
        }
        public ObjectId[] AddEntity(DBObjectCollection objs)
        {
            ObjectId[] ids = new ObjectId[objs.Count];
            for (int i = 0; i  objs)
        {
            ObjectId[] ids = new ObjectId[objs.Count];
            for (int i = 0; i  objs)
        {
            ObjectId[] ids = new ObjectId[objs.Count];
            for (int i = 0; i  AppendAttribToBlock(ObjectId blkrefid, List atts)
        {
            BlockReference blkref = (BlockReference)m_Transaction.GetObject(blkrefid, OpenMode.ForWrite);
            return AppendAttribToBlock(blkref, atts);
        }
        public Dictionary AppendAttribToBlock(BlockReference blkref, List atts)
        {
            BlockTableRecord blkdef = (BlockTableRecord)m_Transaction.GetObject(blkref.BlockTableRecord, OpenMode.ForRead);
            int i = 0;
            if (blkdef.HasAttributeDefinitions)
            {
                Dictionary attribs = new Dictionary();
                foreach (ObjectId id in blkdef)
                {
                    DBObject ent = GetObject(id, OpenMode.ForRead);
                    if (ent is AttributeDefinition)
                    {
                        AttributeDefinition attdef = (AttributeDefinition)ent;
                        AttributeReference attref = new AttributeReference();
                        attref.SetAttributeFromBlock(attdef, blkref.BlockTransform);
                        if (i  AppendAttribToBlock(ObjectId blkrefid, List atts)
        {
            BlockReference blkref = (BlockReference)m_Transaction.GetObject(blkrefid, OpenMode.ForWrite);
            return AppendAttribToBlock(blkref, atts);
        }
        public Dictionary AppendAttribToBlock(BlockReference blkref, List atts)
        {
            Dictionary attribs = new Dictionary();
            for (int i = 0; i  atts)
        {
            BlockReference blkref = new BlockReference(Point3d.Origin, blkdefid);
            ObjectId id = AddEntity(blkref);
            BlockRefJig jig = new BlockRefJig(blkref, AppendAttribToBlock(blkref, atts));
            jig.SetPromptCounter(0);
            PromptResult res = CadHelper.Editor.Drag(jig);
            if (res.Status == PromptStatus.OK)
            {
                jig.SetPromptCounter(1);
                res = CadHelper.Editor.Drag(jig);
                if (res.Status == PromptStatus.OK)
                {
                    return id;
                }
            }
            blkref.Erase();
            return ObjectId.Null;
        }        public ObjectId InsertBlock(string name, List attribs)
        {
            return InsertBlock(m_BlockTable[name], attribs);
        }
        public bool GetBlockDefFromFile(string filename, string blkdefname, bool over)
        {
            try
            {
                bool has = m_BlockTable.Has(blkdefname);
                if ((has && over) || !has)
                {
                    using (DBTransaction t = new DBTransaction(filename))
                    {
                        ObjectId id = t.BlockTable[blkdefname];
                        BlockTableRecord btr =
                            (BlockTableRecord)t.GetObject(id, OpenMode.ForRead);
                        if (!btr.IsAnonymous && !btr.IsLayout)
                        {
                            ObjectIdCollection ids = new ObjectIdCollection();
                            ids.Add(t.BlockTable[blkdefname]);
                            t.Database.Wblock(
                                m_Database,
                                ids,
                                new Point3d(),
                                DuplicateRecordCloning.Replace
                                );
                        }
                    }
                }
                return true;
            }
            catch
            { }
            return false;
        }
        public bool GetBlockDefFromFile(string filename, bool over)
        {
            try
            {
                FileInfo fi = new FileInfo(filename);
                string blkdefname = fi.Name;
                if (blkdefname.Contains("."))
                {
                    blkdefname = blkdefname.Substring(0, blkdefname.LastIndexOf('.'));
                }
                bool has = m_BlockTable.Has(blkdefname);
                if ((has && over) || !has)
                {
                    Database db = new Database();
                    db.ReadDwgFile(filename, FileShare.Read, true, null);
                    m_Database.Insert(BlockTableRecord.ModelSpace, blkdefname, db, false);
                }
                return true;
            }
            catch
            { }
            return false;
        }
        #endregion
        //字典
        #region XRecord
        public DBDictionary GetSubDict(DBDictionary dict, bool createsubdict, params string[] dictnames)
        {
            if (createsubdict)
            {
                dict.UpgradeOpen();
                dict.TreatElementsAsHard = true;
            }
            for (int i = 0; i  m_Attribs;
        public BlockRefJig(BlockReference blkref, Dictionary attribs)
            : base(blkref)
        {
            m_Position = new Point3d();
            m_Angle = 0;
            m_Ucs = CadHelper.Editor.CurrentUserCoordinateSystem;
            m_Attribs = attribs;
            Update();
        }
        protected override SamplerStatus Sampler(JigPrompts prompts)
        {
            switch (m_PromptCounter)
            {
                case 0:
                    {
                        JigPromptPointOptions jigOpts = new JigPromptPointOptions("\n请输入基点:");
                        jigOpts.UserInputControls =
                            UserInputControls.Accept3dCoordinates |
                            UserInputControls.NoZeroResponseAccepted |
                            UserInputControls.NoNegativeResponseAccepted;
                        PromptPointResult res = prompts.AcquirePoint(jigOpts);
                        Point3d pnt = res.Value;
                        if (pnt != m_Position)
                        {
                            m_Position = pnt;
                            m_BasePoint = m_Position;
                        }
                        else
                            return SamplerStatus.NoChange;
                        if (res.Status == PromptStatus.Cancel)
                            return SamplerStatus.Cancel;
                        else
                            return SamplerStatus.OK;
                    }
                case 1:
                    {
                        JigPromptAngleOptions jigOpts = new JigPromptAngleOptions("\n请输入旋转角度:");
                        jigOpts.UserInputControls =
                            UserInputControls.Accept3dCoordinates |
                            UserInputControls.NoNegativeResponseAccepted |
                            UserInputControls.GovernedByUCSDetect |
                            UserInputControls.UseBasePointElevation;
                        jigOpts.Cursor = CursorType.RubberBand;
                        jigOpts.UseBasePoint = true;
                        jigOpts.BasePoint = m_BasePoint;
                        PromptDoubleResult res = prompts.AcquireAngle(jigOpts);
                        double angleTemp = res.Value;
                        if (angleTemp != m_Angle)
                            m_Angle = angleTemp;
                        else
                            return SamplerStatus.NoChange;
                        if (res.Status == PromptStatus.Cancel)
                            return SamplerStatus.Cancel;
                        else
                            return SamplerStatus.OK;
                    }
                default:
                    return SamplerStatus.NoChange;
            }
        }
        protected override bool Update()
        {
            try
            {
                /*Ucs下Jig的流程:
                 * 1.先将图元在Wcs下摆正,即//xy平面
                 * 2.将获取点坐标转换到Ucs下
                 * 3.将图元在Ucs下摆正
                 * 4.矩阵变换到Wcs
                 */
                BlockReference blkref = (BlockReference)Entity;
                blkref.Normal = Vector3d.ZAxis;
                blkref.Position = CadHelper.Wcs2Ucs(m_Position);
                blkref.Rotation = m_Angle;
                blkref.TransformBy(m_Ucs);
                //将属性引用按块引用的Ocs矩阵变换
                if (m_Attribs != null)
                {
                    m_Mat = blkref.BlockTransform;
                    foreach (KeyValuePair att in m_Attribs)
                    {
                        AttributeReference attref = att.Value;
                        string s = attref.TextString;
                        attref.SetAttributeFromBlock(att.Key, m_Mat);
                        attref.TextString = s;
                    }
                }
                
            }
            catch
            {
                return false;
            }
            return true;
        }
        public void SetPromptCounter(int i)
        {
            if (i == 0 || i == 1)
            {
                m_PromptCounter = i;
            }
        }
    }
}