ObjectArx 事务封装

枫叶棋语

2. #pragma once
   
3. #include <acadstrc.h>
   
4. #include <AcApDocWindow.h>
   
5. #include <acarray.h>
   
6. #include <acdb.h>
   
7. #include <acdocman.h>
   
8. #include <acestext.h>
   
9. #include <AcString.h>
   
10. #include <acutads.h>
    
11. #include <adesk.h>
    
12. #include <dbdict.h>
    
13. #include <dbid.h>
    
14. #include <dbmain.h>
    
15. #include <dbObject.h>
    
16. #include <dbsymtb.h>
    
17. #include <dbtrans.h>
    
18. #include <vector>
    
19. #include <Windows.h>
    
20. class DbTrans {
    
21. private:
    
22.   Acad::ErrorStatus m_es = Acad::eOk;
    
23. public:
    
24. 
    
25.   AcApDocument* pDoc;
    
26.   AcDbDatabase* pDb;
    
27.   AcDbTransactionManager* pTm;
    
28.   AcTransaction* pTrans;
    
29.   bool IsOk() const { return m_es == Acad::eOk; }
    
30. 
    
31. 
    
32.   DbTrans()
    
33.   {
    
34.     pDoc = curDoc();
    
35.     acDocManager->lockDocument(pDoc);
    
36.     pDb = pDoc->database();
    
37.     pTm = pDb->transactionManager();
    
38.     pTrans = pTm->startTransaction();
    
39.   }
    
40.   ~DbTrans()
    
41.   {
    
42.     Commit();
    
43.     acDocManager->unlockDocument(pDoc);
    
44.   }
    
45. 
    
46.   void Commit()
    
47.   {
    
48. 
    
49.     if (IsOk())
    
50.     {
    
51.       pTm->endTransaction();
    
52. 
    
53.       /*acutPrintf(L"\n提交事务\n");*/
    
54.     }
    
55.     else
    
56.     {
    
57.       pTm->abortTransaction();
    
58.       acutPrintf(L"\n异常操作,退出事务\n");
    
59.     }
    
60. 
    
61.   }
    
62. 
    
63.   template <class T = AcDbEntity>
    
64.   T* getObject(AcDbObjectId objid, AcDb::OpenMode mode = AcDb::OpenMode::kForRead, bool openErased = false)
    
65.   {
    
66.     T* pObj = nullptr;
    
67.     m_es = pTrans->getObject<T>(pObj, objid, mode, openErased);
    
68.     if (IsOk()) {
    
69.       return pObj;
    
70.     }
    
71.     else {
    
72.       return nullptr;
    
73.     }
    
74.   }
    
75. 
    
76.   template<typename T = AcDbEntity>
    
77.   std::vector <T*> getObjects(AcArray<AcDbObjectId> ids, AcDb::OpenMode mode = AcDb::OpenMode::kForRead, bool openErased = false) {
    
78.     std::vector<T*> ents;
    
79.     for (auto objid : ids) {
    
80.       T* pObj = NULL;
    
81.       m_es = pTrans->getObject<T>(pObj, objid, mode, openErased);
    
82.       if (IsOk()) {
    
83.         ents.push_back(pObj);
    
84.       }
    
85.     }
    
86.     return ents;
    
87.   }
    
88. 
    
89.   template<typename TableType>
    
90.   AcDbObjectId getDictId(TableType* table, AcString name) {
    
91.     AcDbObjectId tableId = AcDbObjectId::kNull;
    
92.     auto es = table->getAt(name, tableId);
    
93.     if (es != Acad::eOk)
    
94.       acutPrintf(L"不存在%ls的记录\n", name);
    
95.     return tableId;
    
96.   }
    
97. 
    
98.   AcDbBlockTableRecord* currentSpace()
    
99.   {
    
100.     return this->getObject<AcDbBlockTableRecord>(pDb->currentSpaceId(), AcDb::kForWrite);
     
101.   }
     
102. 
     
103.   AcDbBlockTableRecord* paperSpace()
     
104.   {
     
105.     return this->getObject<AcDbBlockTableRecord>(pDb->paperSpaceVportId(), AcDb::kForWrite);
     
106.   }
     
107.   AcDbBlockTable* getBlockTable() {
     
108.     return this->getObject<AcDbBlockTable>(pDb->blockTableId(), AcDb::kForWrite);
     
109.   }
     
110.   AcDbRegAppTable* getRegAppTable() {
     
111.     return this->getObject<AcDbRegAppTable>(pDb->regAppTableId(), AcDb::kForWrite);
     
112.   }
     
113.   AcDbDimStyleTable* getDimStyleTable() {
     
114.     return this->getObject<AcDbDimStyleTable>(pDb->dimStyleTableId(), AcDb::kForWrite);
     
115.   }
     
116.   AcDbLayerTable* getLayerTable() {
     
117.     return this->getObject<AcDbLayerTable>(pDb->layerTableId(), AcDb::kForWrite);
     
118.   }
     
119.   AcDbLinetypeTable* getLineTypeTable() {
     
120.     return this->getObject<AcDbLinetypeTable>(pDb->linetypeTableId(), AcDb::kForWrite);
     
121.   }
     
122.   AcDbTextStyleTable* getTextStyleTable() {
     
123.     return this->getObject<AcDbTextStyleTable>(pDb->textStyleTableId(), AcDb::kForWrite);
     
124.   }
     
125.   AcDbUCSTable* getUcsTable() {
     
126.     return this->getObject<AcDbUCSTable>(pDb->UCSTableId(), AcDb::kForWrite);
     
127.   }
     
128.   AcDbViewTable* getViewTable() {
     
129.     return this->getObject<AcDbViewTable>(pDb->viewTableId(), AcDb::kForWrite);
     
130.   }
     
131.   AcDbViewportTable* getViewportTable() {
     
132.     return this->getObject<AcDbViewportTable>(pDb->viewportTableId(), AcDb::kForWrite);
     
133.   }
     
134.   AcDbDictionary* getMLeaderStyleDictionary() {
     
135.     return this->getObject<AcDbDictionary>(pDb->mleaderStyleDictionaryId(), AcDb::kForWrite);
     
136.   }
     
137.   AcDbDictionary* getMlineStyleDictionary() {
     
138.     return this->getObject<AcDbDictionary>(pDb->mLStyleDictionaryId(), AcDb::kForWrite);
     
139.   }
     
140.   AcDbDictionary* getMaterialDictionary() {
     
141.     return this->getObject<AcDbDictionary>(pDb->materialDictionaryId(), AcDb::kForWrite);
     
142.   }
     
143.   AcDbDictionary* getPlotSettingsDictionary() {
     
144.     return this->getObject<AcDbDictionary>(pDb->plotSettingsDictionaryId(), AcDb::kForWrite);
     
145.   }
     
146.   AcDbDictionary* getPlotStyleNameDictionary() {
     
147.     return this->getObject<AcDbDictionary>(pDb->plotStyleNameDictionaryId(), AcDb::kForWrite);
     
148.   }
     
149.   AcDbDictionary* getLayoutDictionary() {
     
150.     return this->getObject<AcDbDictionary>(pDb->layoutDictionaryId(), AcDb::kForWrite);
     
151.   }
     
152. 
     
153. 
     
154.   AcDbObjectId addEntity(AcDbBlockTableRecord* btr, AcDbEntity* ent)
     
155.   {
     
156.     AcDbObjectId objId;
     
157.     if (!btr) {
     
158.       acutPrintf(L"\n错误：块表记录指针为空");
     
159.       return objId;
     
160.     }
     
161. 
     
162.     Acad::ErrorStatus es = btr->appendAcDbEntity(objId, ent);
     
163.     if (es != Acad::eOk) {
     
164.       acutPrintf(L"\n错误：无法将实体添加到块表记录 (错误码: %s)", acadErrorStatusText(es));
     
165.       return objId;
     
166.     }
     
167. 
     
168.     es = pTm->addNewlyCreatedDBRObject(ent, true);
     
169.     if (es != Acad::eOk) {
     
170.       acutPrintf(L"\n错误：无法将实体添加到事务管理器 (错误码: %s)", acadErrorStatusText(es));
     
171.       return objId;
     
172.     }
     
173.     return objId;
     
174.   }
     
175. 
     
176.   template<typename T>
     
177.   AcArray<AcDbObjectId> addEntities(AcDbBlockTableRecord* btr, AcArray<T*> ents)
     
178.   {
     
179.     AcArray<AcDbObjectId> objIds;
     
180.     for (AcDbEntity* ent : ents) {
     
181.       auto  objId = addEntity(btr, ent);
     
182.       if (objId == AcDbObjectId::kNull) continue;
     
183.       objIds.append(objId);
     
184.     }
     
185.     return objIds;
     
186.   }
     
187. 
     
188.   bool upopen(AcDbObject* pobj) {
     
189.     // 确保对象在事务中
     
190.     bool isWritable = pobj->isWriteEnabled();
     
191.     if (!isWritable) {
     
192.       pobj->upgradeOpen();
     
193.       return pobj->isWriteEnabled();
     
194.     }
     
195.     return isWritable;
     
196.   }
     
197. 
     
198.   void erase(AcDbObject* pObj) {
     
199.     if (!pObj->isErased() && !pObj->isNewObject()) {
     
200.       if (upopen(pObj)) {
     
201.         pObj->erase();
     
202.       }
     
203.     }
     
204.   }
     
205. 
     
206.   void flushScreen()
     
207.   {
     
208.     pTm->queueForGraphicsFlush();
     
209.   }
     
210. };
     
211. 
     
212. 
     
213. 
     
214. inline static void setFocusDoc() {
     
215.   auto windowHwnd = acDocWindowManager->activeDocumentWindow()->windowHandle();
     
216.   SetFocus(windowHwnd);
     
217. }
     

#pragma once<br /> #include <acadstrc.h><br /> #include <AcApDocWindow.h><br /> #include <acarray.h><br /> #include <acdb.h><br /> #include <acdocman.h><br /> #include <acestext.h><br /> #include <AcString.h><br /> #include <acutads.h><br /> #include <adesk.h><br /> #include <dbdict.h><br /> #include <dbid.h><br /> #include <dbmain.h><br /> #include <dbObject.h><br /> #include <dbsymtb.h><br /> #include <dbtrans.h><br /> #include <vector><br /> #include <Windows.h><br /> class DbTrans {<br /> private:<br />   Acad::ErrorStatus m_es = Acad::eOk;<br /> public:<br /> <br />   AcApDocument* pDoc;<br />   AcDbDatabase* pDb;<br />   AcDbTransactionManager* pTm;<br />   AcTransaction* pTrans;<br />   bool IsOk() const { return m_es == Acad::eOk; }<br /> <br /> <br />   DbTrans()<br />   {<br />     pDoc = curDoc();<br />     acDocManager->lockDocument(pDoc);<br />     pDb = pDoc->database();<br />     pTm = pDb->transactionManager();<br />     pTrans = pTm->startTransaction();<br />   }<br />   ~DbTrans()<br />   {<br />     Commit();<br />     acDocManager->unlockDocument(pDoc);<br />   }<br /> <br />   void Commit()<br />   {<br /> <br />     if (IsOk())<br />     {<br />       pTm->endTransaction();<br /> <br />       /*acutPrintf(L"\n提交事务\n");*/<br />     }<br />     else<br />     {<br />       pTm->abortTransaction();<br />       acutPrintf(L"\n异常操作,退出事务\n");<br />     }<br /> <br />   }<br /> <br />   template <class T = AcDbEntity><br />   T* getObject(AcDbObjectId objid, AcDb::OpenMode mode = AcDb::OpenMode::kForRead, bool openErased = false)<br />   {<br />     T* pObj = nullptr;<br />     m_es = pTrans->getObject<T>(pObj, objid, mode, openErased);<br />     if (IsOk()) {<br />       return pObj;<br />     }<br />     else {<br />       return nullptr;<br />     }<br />   }<br /> <br />   template<typename T = AcDbEntity><br />   std::vector <T*> getObjects(AcArray<AcDbObjectId> ids, AcDb::OpenMode mode = AcDb::OpenMode::kForRead, bool openErased = false) {<br />     std::vector<T*> ents;<br />     for (auto objid : ids) {<br />       T* pObj = NULL;<br />       m_es = pTrans->getObject<T>(pObj, objid, mode, openErased);<br />       if (IsOk()) {<br />         ents.push_back(pObj);<br />       }<br />     }<br />     return ents;<br />   }<br /> <br />   template<typename TableType><br />   AcDbObjectId getDictId(TableType* table, AcString name) {<br />     AcDbObjectId tableId = AcDbObjectId::kNull;<br />     auto es = table->getAt(name, tableId);<br />     if (es != Acad::eOk)<br />       acutPrintf(L"不存在%ls的记录\n", name);<br />     return tableId;<br />   }<br /> <br />   AcDbBlockTableRecord* currentSpace()<br />   {<br />     return this->getObject<AcDbBlockTableRecord>(pDb->currentSpaceId(), AcDb::kForWrite);<br />   }<br /> <br />   AcDbBlockTableRecord* paperSpace()<br />   {<br />     return this->getObject<AcDbBlockTableRecord>(pDb->paperSpaceVportId(), AcDb::kForWrite);<br />   }<br />   AcDbBlockTable* getBlockTable() {<br />     return this->getObject<AcDbBlockTable>(pDb->blockTableId(), AcDb::kForWrite);<br />   }<br />   AcDbRegAppTable* getRegAppTable() {<br />     return this->getObject<AcDbRegAppTable>(pDb->regAppTableId(), AcDb::kForWrite);<br />   }<br />   AcDbDimStyleTable* getDimStyleTable() {<br />     return this->getObject<AcDbDimStyleTable>(pDb->dimStyleTableId(), AcDb::kForWrite);<br />   }<br />   AcDbLayerTable* getLayerTable() {<br />     return this->getObject<AcDbLayerTable>(pDb->layerTableId(), AcDb::kForWrite);<br />   }<br />   AcDbLinetypeTable* getLineTypeTable() {<br />     return this->getObject<AcDbLinetypeTable>(pDb->linetypeTableId(), AcDb::kForWrite);<br />   }<br />   AcDbTextStyleTable* getTextStyleTable() {<br />     return this->getObject<AcDbTextStyleTable>(pDb->textStyleTableId(), AcDb::kForWrite);<br />   }<br />   AcDbUCSTable* getUcsTable() {<br />     return this->getObject<AcDbUCSTable>(pDb->UCSTableId(), AcDb::kForWrite);<br />   }<br />   AcDbViewTable* getViewTable() {<br />     return this->getObject<AcDbViewTable>(pDb->viewTableId(), AcDb::kForWrite);<br />   }<br />   AcDbViewportTable* getViewportTable() {<br />     return this->getObject<AcDbViewportTable>(pDb->viewportTableId(), AcDb::kForWrite);<br />   }<br />   AcDbDictionary* getMLeaderStyleDictionary() {<br />     return this->getObject<AcDbDictionary>(pDb->mleaderStyleDictionaryId(), AcDb::kForWrite);<br />   }<br />   AcDbDictionary* getMlineStyleDictionary() {<br />     return this->getObject<AcDbDictionary>(pDb->mLStyleDictionaryId(), AcDb::kForWrite);<br />   }<br />   AcDbDictionary* getMaterialDictionary() {<br />     return this->getObject<AcDbDictionary>(pDb->materialDictionaryId(), AcDb::kForWrite);<br />   }<br />   AcDbDictionary* getPlotSettingsDictionary() {<br />     return this->getObject<AcDbDictionary>(pDb->plotSettingsDictionaryId(), AcDb::kForWrite);<br />   }<br />   AcDbDictionary* getPlotStyleNameDictionary() {<br />     return this->getObject<AcDbDictionary>(pDb->plotStyleNameDictionaryId(), AcDb::kForWrite);<br />   }<br />   AcDbDictionary* getLayoutDictionary() {<br />     return this->getObject<AcDbDictionary>(pDb->layoutDictionaryId(), AcDb::kForWrite);<br />   }<br /> <br /> <br />   AcDbObjectId addEntity(AcDbBlockTableRecord* btr, AcDbEntity* ent)<br />   {<br />     AcDbObjectId objId;<br />     if (!btr) {<br />       acutPrintf(L"\n错误：块表记录指针为空");<br />       return objId;<br />     }<br /> <br />     Acad::ErrorStatus es = btr->appendAcDbEntity(objId, ent);<br />     if (es != Acad::eOk) {<br />       acutPrintf(L"\n错误：无法将实体添加到块表记录 (错误码: %s)", acadErrorStatusText(es));<br />       return objId;<br />     }<br /> <br />     es = pTm->addNewlyCreatedDBRObject(ent, true);<br />     if (es != Acad::eOk) {<br />       acutPrintf(L"\n错误：无法将实体添加到事务管理器 (错误码: %s)", acadErrorStatusText(es));<br />       return objId;<br />     }<br />     return objId;<br />   }<br /> <br />   template<typename T><br />   AcArray<AcDbObjectId> addEntities(AcDbBlockTableRecord* btr, AcArray<T*> ents)<br />   {<br />     AcArray<AcDbObjectId> objIds;<br />     for (AcDbEntity* ent : ents) {<br />       auto  objId = addEntity(btr, ent);<br />       if (objId == AcDbObjectId::kNull) continue;<br />       objIds.append(objId);<br />     }<br />     return objIds;<br />   }<br /> <br />   bool upopen(AcDbObject* pobj) {<br />     // 确保对象在事务中<br />     bool isWritable = pobj->isWriteEnabled();<br />     if (!isWritable) {<br />       pobj->upgradeOpen();<br />       return pobj->isWriteEnabled();<br />     }<br />     return isWritable;<br />   }<br /> <br />   void erase(AcDbObject* pObj) {<br />     if (!pObj->isErased() && !pObj->isNewObject()) {<br />       if (upopen(pObj)) {<br />         pObj->erase();<br />       }<br />     }<br />   }<br /> <br />   void flushScreen()<br />   {<br />     pTm->queueForGraphicsFlush();<br />   }<br /> };<br /> <br /> <br /> <br /> inline static void setFocusDoc() {<br />   auto windowHwnd = acDocWindowManager->activeDocumentWindow()->windowHandle();<br />   SetFocus(windowHwnd);<br /> }<br />

使用示例：不需要对line 进行close 或者delete

1. void addLine() {
   
2. 
   
3.   DbTrans tr;
   
4.   auto btr = tr.currentSpace();
   
5.   AcDbLine* line = new AcDbLine({ 0,0,0 }, { 50,50,0 });
   
6.   tr.addEntity(btr, line);
   
7. }
   

复制代码