471 lines
11 KiB
C++
471 lines
11 KiB
C++
#include "stdafx.h"
|
|
#include "StdArx.h"
|
|
|
|
#include "AecDbFloorTreeView.h"
|
|
#include "KTSetting.h"
|
|
#include "acgitransient.h"
|
|
|
|
#ifdef EARTHWORM_EXPORT
|
|
#define DLLIMPEXP __declspec(dllexport)
|
|
#else
|
|
#define DLLIMPEXP __declspec(dllimport)
|
|
#endif
|
|
|
|
double DLLIMPEXP PixelToWcs(int nPixel);
|
|
|
|
const LPCTSTR g_sBuiding = _T("构筑物");
|
|
const LPCTSTR g_sTextStyle = _T("Standard");
|
|
|
|
Adesk::UInt32 AecDbFloorTreeView::kCurrentVersionNumber = 1;
|
|
|
|
ACRX_DXF_DEFINE_MEMBERS (AecDbFloorTreeView, AecDbEntityBase,
|
|
AcDb::kDHL_CURRENT, AcDb::kMReleaseCurrent,
|
|
AcDbProxyEntity::kNoOperation | AcDbProxyEntity::kDisableProxyWarning,
|
|
AEC_FLOORTREEVIEW,
|
|
"AecDbFloorTreeView"
|
|
"|Product Desc: shanghua earthworm app obj"
|
|
"|Company: shanghua inc"
|
|
"|WEB Address: www.shanghuasoft.com")
|
|
|
|
AecDbFloorTreeView::AecDbFloorTreeView(void)
|
|
{
|
|
m_nFlag = 100;
|
|
m_sDbVerGUID = _T("");
|
|
}
|
|
|
|
AecDbFloorTreeView::~AecDbFloorTreeView(void)
|
|
{
|
|
|
|
}
|
|
|
|
void AecDbFloorTreeView::SetDbVersionGUID(LPCTSTR pszID)
|
|
{
|
|
assertWriteEnabled();
|
|
if (pszID != NULL)
|
|
m_sDbVerGUID = pszID;
|
|
}
|
|
|
|
LPCTSTR AecDbFloorTreeView::GetDbVersionGUID() const
|
|
{
|
|
assertReadEnabled();
|
|
return (LPCTSTR)m_sDbVerGUID;
|
|
}
|
|
|
|
Acad::ErrorStatus AecDbFloorTreeView::dwgInFields(AcDbDwgFiler* pFiler)
|
|
{
|
|
assertWriteEnabled();
|
|
|
|
//----- Read parent class information first.
|
|
Acad::ErrorStatus es = AecDbEntityBase::dwgInFields (pFiler) ;
|
|
if ( es != Acad::eOk )
|
|
return (es) ;
|
|
|
|
//----- Object version number needs to be read first
|
|
Adesk::UInt32 version =0 ;
|
|
if ( (es =pFiler->readUInt32 (&version)) != Acad::eOk )
|
|
return (es) ;
|
|
if ( version > AecDbFloorTreeView::kCurrentVersionNumber )
|
|
return (Acad::eMakeMeProxy) ;
|
|
|
|
pFiler->readUInt32(&m_nFlag);
|
|
|
|
TCHAR* pszStr = NULL;
|
|
es = pFiler->readString(&pszStr);
|
|
if (pszStr != NULL)
|
|
{
|
|
m_sDbVerGUID = pszStr;
|
|
}
|
|
|
|
return (pFiler->filerStatus ()) ;
|
|
}
|
|
|
|
Acad::ErrorStatus AecDbFloorTreeView::dwgOutFields(AcDbDwgFiler* pFiler) const
|
|
{
|
|
assertReadEnabled () ;
|
|
//----- Save parent class information first.
|
|
Acad::ErrorStatus es = AecDbEntityBase::dwgOutFields (pFiler) ;
|
|
if ( es != Acad::eOk )
|
|
return (es) ;
|
|
|
|
//----- Object version number needs to be saved first
|
|
if ( (es =pFiler->writeUInt32 (AecDbFloorTreeView::kCurrentVersionNumber)) != Acad::eOk )
|
|
return (es) ;
|
|
|
|
pFiler->writeUInt32(m_nFlag);
|
|
pFiler->writeString((LPCTSTR)m_sDbVerGUID);
|
|
|
|
return (pFiler->filerStatus ()) ;
|
|
}
|
|
|
|
Acad::ErrorStatus AecDbFloorTreeView::Draw2d(AcGiDraw &dc) const
|
|
{
|
|
assertReadEnabled();
|
|
|
|
Clean2DBuffer();
|
|
|
|
double dRad = PixelToWcs(10);
|
|
AcGePoint3d ptCen(0,0,0);
|
|
CPoint pt(0,0);
|
|
acedDwgPoint ptWcs;
|
|
acedCoordFromPixelToWorld(pt,ptWcs);
|
|
ptCen = asPnt3d(ptWcs);
|
|
|
|
AcDbObjectId idSt = KTSetting::InitTextStyle(_T("Pipe_Dim"));
|
|
|
|
AcDbText* pText = new AcDbText;
|
|
pText->setTextStyle(idSt);
|
|
pText->setTextString(g_sBuiding);//数据
|
|
pText->setHeight(dRad);
|
|
pText->setWidthFactor(0.7);
|
|
pText->setRotation(0.0);
|
|
pText->setHorizontalMode(AcDb::kTextLeft);
|
|
pText->setPosition( ptCen);
|
|
pText->setLayer(acdbCurDwg()->clayer());
|
|
pText->setColorIndex(1);
|
|
|
|
dc.DrawEntity(pText);
|
|
Add2DBuffer(pText);
|
|
|
|
ShowFloorChild(dc,pText);
|
|
|
|
return Acad::eOk;
|
|
}
|
|
|
|
Acad::ErrorStatus AecDbFloorTreeView::DrawNodeStatus(AcGiDraw &dc,const AcGePoint3d& ptPos,bool bExpand) const
|
|
{
|
|
assertReadEnabled();
|
|
|
|
double dCmd = PixelToWcs(4);
|
|
|
|
AcGePoint3d ptGe[5];
|
|
ptGe[0] = ptPos;
|
|
ptGe[0] -= AcGeVector3d::kXAxis * 0.8 * dCmd;
|
|
ptGe[0] -= AcGeVector3d::kYAxis * 0.8 * dCmd;
|
|
|
|
ptGe[1] = ptGe[0];
|
|
ptGe[1] += AcGeVector3d::kXAxis * dCmd;
|
|
|
|
ptGe[2] = ptGe[1];
|
|
ptGe[2] += AcGeVector3d::kYAxis * dCmd;
|
|
|
|
ptGe[3] = ptGe[0] + AcGeVector3d::kYAxis * dCmd;
|
|
ptGe[4] = ptGe[0];
|
|
|
|
AcDbPolyline* pL = new AcDbPolyline;
|
|
for (int i = 0;i < 5;i++)
|
|
{
|
|
pL->addVertexAt(i,ptGe[i].convert2d(AcGePlane::kXYPlane));
|
|
}
|
|
pL->setColorIndex(1);
|
|
|
|
dc.DrawEntity(pL);
|
|
Add2DBuffer(pL);
|
|
|
|
return Acad::eOk;
|
|
}
|
|
|
|
Acad::ErrorStatus AecDbFloorTreeView::ShowFloorChild(AcGiDraw &dc,AcDbText* pParent) const
|
|
{
|
|
assertReadEnabled();
|
|
|
|
if (pParent == NULL)
|
|
return Acad::eOk;
|
|
|
|
AcDbDatabase *pDb = acdbCurDwg();
|
|
if (pDb != NULL)
|
|
{
|
|
int nFloor = -1;
|
|
AcDbIntArray arFloors;
|
|
|
|
Acad::ErrorStatus es;
|
|
AcDbBlockTable* pTable = NULL;
|
|
if ( Acad::eOk == (es = pDb->getBlockTable(pTable,AcDb::kForRead) ) )
|
|
{
|
|
AcDbBlockTableRecord* pRecord = NULL;
|
|
if (Acad::eOk == (es = pTable->getAt(ACDB_MODEL_SPACE,pRecord,AcDb::kForRead)))
|
|
{
|
|
AcDbEntity* pEnt = NULL;
|
|
AcDbBlockTableRecordIterator* pIt = NULL;
|
|
pRecord->newIterator(pIt);
|
|
|
|
if (pIt != NULL)
|
|
{
|
|
for (pIt->start();!pIt->done();pIt->step())
|
|
{
|
|
if ( Acad::eOk == (es = pIt->getEntity(pEnt,AcDb::kForRead) ) )
|
|
{
|
|
//墙，梁，板，柱子
|
|
AecDbXWall* pWall = AecDbXWall::cast(pEnt);
|
|
if (pWall != NULL)
|
|
{
|
|
nFloor = -1;
|
|
if ( pWall->GetSpecialtyData(AcDb::kForRead)->GetAt(KEY_FLOOR,nFloor) )
|
|
{
|
|
if (arFloors.find(nFloor) == -1)
|
|
{
|
|
arFloors.append(nFloor);
|
|
}
|
|
}
|
|
}
|
|
|
|
AecDbColumn* pCol = AecDbColumn::cast(pEnt);
|
|
if (pCol != NULL)
|
|
{
|
|
nFloor = -1;
|
|
if ( pCol->GetSpecialtyData(AcDb::kForRead)->GetAt(KEY_FLOOR,nFloor) )
|
|
{
|
|
if (arFloors.find(nFloor) == -1)
|
|
{
|
|
arFloors.append(nFloor);
|
|
}
|
|
}
|
|
}
|
|
|
|
AecDbBeam* pBeam = AecDbBeam::cast(pEnt);
|
|
if (pBeam != NULL)
|
|
{
|
|
nFloor = -1;
|
|
if ( pBeam->GetSpecialtyData(AcDb::kForRead)->GetAt(KEY_FLOOR,nFloor) )
|
|
{
|
|
if (arFloors.find(nFloor) == -1)
|
|
{
|
|
arFloors.append(nFloor);
|
|
}
|
|
}
|
|
}
|
|
|
|
/*AecDbXSlab* pSlab = AecDbXSlab::cast(pEnt);
|
|
if (pSlab != NULL)
|
|
{
|
|
nFloor = -1;
|
|
if ( pSlab->GetSpecialtyData()->GetAt(KEY_FLOOR,nFloor) )
|
|
{
|
|
if (arFloors.find(nFloor) == -1)
|
|
{
|
|
arFloors.append(nFloor);
|
|
}
|
|
}
|
|
}*/
|
|
|
|
AecDbXSlab* pBasin = AecDbXSlab::cast(pEnt);
|
|
if (pBasin != NULL)
|
|
{
|
|
nFloor = -1;
|
|
if ( pBasin->GetSpecialtyData(AcDb::kForRead)->GetAt(KEY_FLOOR,nFloor) )
|
|
{
|
|
if (arFloors.find(nFloor) == -1)
|
|
{
|
|
arFloors.append(nFloor);
|
|
}
|
|
}
|
|
}
|
|
pEnt->close();
|
|
}
|
|
}
|
|
delete pIt;
|
|
}
|
|
pRecord->close();
|
|
}
|
|
pTable->close();
|
|
}
|
|
|
|
AcGePoint3d ptPos = pParent->position();//
|
|
double dH = pParent->height();
|
|
dH *= 1.5;
|
|
|
|
AcGePoint3d ptLnTop = ptPos;
|
|
ptLnTop.x -= 0.2 * dH;
|
|
DrawNodeStatus(dc,ptLnTop,true);
|
|
|
|
AcGePoint3d ptText = ptPos;
|
|
ptText.x += dH;
|
|
ptText.y -= dH;
|
|
|
|
AcGePoint3d ptLnV = ptLnTop;
|
|
|
|
CString sText;
|
|
for (int i = 0;i < arFloors.length();i++)
|
|
{
|
|
nFloor = arFloors.at(i);
|
|
sText.Format(_T("第 %d 层"),i + 1);
|
|
|
|
AcDbText* pText = new AcDbText;
|
|
pText->setTextStyle(pParent->textStyle());
|
|
pText->setTextString(sText);//数据
|
|
pText->setHeight(pParent->height());
|
|
pText->setWidthFactor(0.7);
|
|
pText->setRotation(0.0);
|
|
pText->setHorizontalMode(AcDb::kTextLeft);
|
|
pText->setPosition( ptText);
|
|
pText->setLayer(acdbCurDwg()->clayer());
|
|
pText->setColorIndex(1);
|
|
|
|
dc.DrawEntity(pText);
|
|
Add2DBuffer(pText);
|
|
|
|
ptLnV.y = ptText.y;//竖线
|
|
DrawNodeStatus(dc,ptLnV,true);
|
|
|
|
AcDbLine* pLine = new AcDbLine;
|
|
pLine->setStartPoint(ptLnV);
|
|
pLine->setEndPoint(ptText);
|
|
dc.DrawEntity(pLine);
|
|
Add2DBuffer(pLine);
|
|
|
|
ptText.y -= dH;
|
|
}
|
|
|
|
if (!arFloors.isEmpty())
|
|
{
|
|
AcDbLine* pLine = new AcDbLine;
|
|
pLine->setStartPoint(ptLnTop);
|
|
pLine->setEndPoint(ptLnV);
|
|
dc.DrawEntity(pLine);
|
|
Add2DBuffer(pLine);
|
|
}
|
|
}
|
|
return Acad::eOk;
|
|
}
|
|
|
|
Acad::ErrorStatus AecDbFloorTreeView::ShowChildItems(AcGiDraw &dc,AcDbText* pParent) const
|
|
{
|
|
return Acad::eOk;
|
|
}
|
|
|
|
Acad::ErrorStatus AecDbFloorTreeView::subGetGripPoints (AcGePoint3dArray& gripPoints,AcDbIntArray& osnapModes,AcDbIntArray& geomIds) const
|
|
{
|
|
return Acad::eOk;
|
|
}
|
|
|
|
Acad::ErrorStatus AecDbFloorTreeView::subMoveGripPointsAt(const AcDbIntArray& indices,const AcGeVector3d& offset)
|
|
{
|
|
return Acad::eOk;
|
|
}
|
|
|
|
extern AcArray<AcDbEntity*> g_arFloorViewPtrs;
|
|
extern AcArray<int> g_viewportNumbers;
|
|
|
|
void ClearFloorTreeView()
|
|
{
|
|
AcGiTransientManager* pTransientManager = acgiGetTransientManager();
|
|
|
|
int numOfMarkers = g_arFloorViewPtrs.length();
|
|
if (numOfMarkers > 0)
|
|
{
|
|
for(int index = 0; index < numOfMarkers; index++)
|
|
{
|
|
AcDbEntity *pMarker = g_arFloorViewPtrs.at(index);
|
|
pTransientManager->eraseTransient(pMarker,g_viewportNumbers);
|
|
delete pMarker;
|
|
}
|
|
g_arFloorViewPtrs.removeAll();
|
|
}
|
|
}
|
|
|
|
bool FloorTreeViewIsExist()
|
|
{
|
|
AcDbDatabase *pDb = acdbCurDwg();
|
|
if (pDb != NULL)
|
|
{
|
|
TCHAR* pszGUID = NULL;
|
|
pDb->getVersionGuid(pszGUID);
|
|
|
|
if (pszGUID == NULL)
|
|
return false;
|
|
|
|
for(int index = 0; index < g_viewportNumbers.length(); index++)
|
|
{
|
|
AcDbEntity *pMarker = g_arFloorViewPtrs.at(index);
|
|
AecDbFloorTreeView* pFloorView = AecDbFloorTreeView::cast(pMarker);
|
|
if (pFloorView != NULL)
|
|
{
|
|
CString sID = pFloorView->GetDbVersionGUID();
|
|
if (sID.CompareNoCase(pszGUID) == 0)
|
|
{
|
|
acutDelString(pszGUID);
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
acutDelString(pszGUID);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void AddEntToTransient(AcDbEntity* pEntTo)
|
|
{
|
|
AcDbDatabase *pDb = acdbCurDwg();
|
|
if (pDb != NULL)
|
|
{
|
|
AcGiTransientManager* pTransientManager = acgiGetTransientManager();
|
|
AcDbTransactionManager* pTM = pDb->transactionManager();
|
|
if (pTM != NULL)
|
|
{
|
|
AcTransaction *pTransaction = pTM->startTransaction();
|
|
|
|
struct resbuf res;
|
|
acedGetVar(L"CVPORT", &res);
|
|
if (g_viewportNumbers.find(res.resval.rint) == -1)
|
|
g_viewportNumbers.append(res.resval.rint);
|
|
|
|
if (pTransientManager != NULL)
|
|
{
|
|
g_arFloorViewPtrs.append(pEntTo);
|
|
|
|
TCHAR* pszID = NULL;
|
|
pDb->getVersionGuid(pszID);
|
|
|
|
if ( pTransientManager->addTransient(pEntTo,kAcGiDirectShortTerm,0,g_viewportNumbers) )
|
|
{
|
|
acutPrintf(_T("\naddTransient success 2!"));
|
|
}
|
|
}
|
|
pTM->endTransaction();
|
|
pTM->queueForGraphicsFlush();
|
|
}
|
|
}
|
|
}
|
|
|
|
void ShowFloorTreeView()
|
|
{
|
|
AcDbDatabase *pDb = acdbCurDwg();
|
|
if (pDb != NULL)
|
|
{
|
|
AcGiTransientManager* pTransientManager = acgiGetTransientManager();
|
|
AcDbTransactionManager* pTM = pDb->transactionManager();
|
|
if (pTM != NULL)
|
|
{
|
|
if (FloorTreeViewIsExist())
|
|
{
|
|
pTM->queueForGraphicsFlush();
|
|
return;
|
|
}
|
|
|
|
AcTransaction *pTransaction = pTM->startTransaction();
|
|
|
|
struct resbuf res;
|
|
acedGetVar(L"CVPORT", &res);
|
|
if (g_viewportNumbers.find(res.resval.rint) == -1)
|
|
g_viewportNumbers.append(res.resval.rint);
|
|
|
|
if (pTransientManager != NULL)
|
|
{
|
|
AecDbFloorTreeView* pFloorView = new AecDbFloorTreeView;
|
|
g_arFloorViewPtrs.append(pFloorView);
|
|
|
|
TCHAR* pszID = NULL;
|
|
pDb->getVersionGuid(pszID);
|
|
|
|
CString sID = pszID;
|
|
pFloorView->SetDbVersionGUID(sID);
|
|
|
|
if ( pTransientManager->addTransient(pFloorView,kAcGiDirectShortTerm,0,g_viewportNumbers) )
|
|
{
|
|
acutPrintf(_T("\naddTransient success 2!"));
|
|
}
|
|
}
|
|
pTM->endTransaction();
|
|
pTM->queueForGraphicsFlush();
|
|
}
|
|
}
|
|
} |