#include "stdafx.h"
#include "DwgCompareDbMgr.h"
#include <dbidmap.h>

CDwgCompareDbInfo::CDwgCompareDbInfo(void)
: m_sName(_T(""))
, m_sPath(_T(""))
, m_pDb(NULL)
, m_iEntCount(0)
{

}

CDwgCompareDbInfo::~CDwgCompareDbInfo(void)
{
	if (m_pDb != NULL)
	{
		delete m_pDb;
		m_pDb = NULL;
	}
}


BOOL CDwgCompareDbInfo::LoadFile( const CString sName, const CString sPath )
{
	if (m_pDb != NULL)
	{
		delete m_pDb;
	}
	m_pDb = new AcDbDatabase(false, true);

	XString sFile(256), sFile1(256);
	sFile1 << sPath;
	if(!FindFile(sFile1, sFile))
	{
		return FALSE;
	}
#if ADS > 20
	Acad::ErrorStatus es = m_pDb->readDwgFile(sFile1, AcDbDatabase::OpenMode::kForReadAndAllShare, false);
#else
	Acad::ErrorStatus es = m_pDb->readDwgFile(sFile1, _SH_DENYNO, false);
#endif
	// modify by cq 2017-01-05 支持CAD2017
	if (es != Acad::eOk)
	{
		return FALSE;
	}
	//设置插入基点为原点
	es = m_pDb->setInsbase(AcGePoint3d::kOrigin);
	//es = m_pDb->updateExt(false);

	m_sPath = sPath;
	m_sName = sName;
	//获取所有图层名
	InitLayerInfo();

	return TRUE;
}

BOOL CDwgCompareDbInfo::LoadFile( const CString sName, const CString sPath, XPoly2D poly )
{
	if (!LoadFile(sName, sPath))
	{
		return FALSE;
	}
	//遍历模型空间,删除不在entPoly内的实体
	AcDbBlockTable * pBlockTab = NULL;
	Acad::ErrorStatus es = m_pDb->getBlockTable(pBlockTab, AcDb::kForRead);
	if (es==Acad::eOk && pBlockTab!=NULL)
	{
		AcDbBlockTableRecord * pRec = NULL;
		es = pBlockTab->getAt(ACDB_MODEL_SPACE, pRec, AcDb::kForRead);
		pBlockTab->close();

		AcDbBlockTableRecordIterator *pIterator = NULL;
		if (es==Acad::eOk && pRec!=NULL)
		{
			pRec->newIterator(pIterator);
			pRec->close();
		}
		for(pIterator->start(); !pIterator->done(); pIterator->step())
		{
			AcDbEntity *pEnt = NULL;
			es = pIterator->getEntity(pEnt, AcDb::kForWrite);
			if (es!=Acad::eOk || pEnt==NULL)
			{
				continue;
			}
			AcDbExtents bExtent;
			pEnt->getGeomExtents(bExtent);
			XPoint ptMax = bExtent.maxPoint();
			XPoint ptMin = bExtent.minPoint();
			ptMax.z = 0;
			ptMin.z = 0;
			//只保留和poly有交集的实体
			if((ptMax&poly)==PR_INSIDE || (ptMin&poly)==PR_INSIDE)
			{
				pEnt->close();
				continue;
			}
			pEnt->erase();
			pEnt->close();
		}
		delete pIterator;
	}
	//m_pDb->updateExt(false);

	return TRUE;
}
BOOL CDwgCompareDbInfo::LoadDb( CString& csFilePath, AcDbDatabase*& pDb, const AcDbObjectIdArray& ObjIds, AcGePoint3d& ptBase )
{
	if (csFilePath.IsEmpty() || pDb == NULL || ObjIds.length()==0)
	{
		return FALSE;
	}
	if (m_pDb != NULL)
	{
		delete m_pDb;
	}
	m_pDb = new AcDbDatabase(false, true);
	Acad::ErrorStatus es = pDb->wblock(m_pDb, ObjIds, ptBase);
	if (es != Acad::eOk)
	{
		CString csError;
		csError.Format(_T("\n文件加载失败:%s"), csFilePath);
		acutPrintf(csError);
		return FALSE;
	}

	m_sPath = csFilePath;
	m_sName = m_sPath.Right(m_sPath.GetLength() - m_sPath.ReverseFind(_T('\\')) -1);
	//获取所有图层名
	InitLayerInfo();
	//m_pDb->updateExt(false);
	return TRUE;
}

void CDwgCompareDbInfo::InitLayerInfo()
{
	if (m_pDb==NULL)
	{
		return;
	}
	m_vtLayerAll.clear();

	AcDbLayerTable * pLayerTab = NULL;
	Acad::ErrorStatus es = m_pDb->getLayerTable(pLayerTab, AcDb::kForRead);
	if (es!=Acad::eOk || pLayerTab==NULL)
	{
		return;
	}
	AcDbLayerTableIterator * pIte = NULL;
	es = pLayerTab->newIterator(pIte);
	pLayerTab->close();
	if (es!=Acad::eOk || pIte==NULL)
	{
		return;
	}
	for (pIte->start(); !pIte->done(); pIte->step())
	{
		AcDbLayerTableRecord * pLayerRec = NULL;
		es = pIte->getRecord(pLayerRec, AcDb::kForRead);
		if (es!=Acad::eOk || pLayerRec==NULL)
		{
			continue;
		}
		TCHAR * szLayer;
		pLayerRec->getName(szLayer);
		m_vtLayerAll.push_back(szLayer);
		delete szLayer;
		pLayerRec->close();
	}
	delete pIte;
}

void CDwgCompareDbInfo::GetAllLayer( std::vector<CString>& vtLayer ) const
{
	vtLayer = m_vtLayerAll;
}
CString CDwgCompareDbInfo::GetFileName() const
{
	return m_sName;
}

CString CDwgCompareDbInfo::GetFilePath() const
{
	return m_sPath;
}

AcDbDatabase * CDwgCompareDbInfo::GetDb() const
{
	return m_pDb;
}

int CDwgCompareDbInfo::InitEntCount()
{
	m_iEntCount = 0;
	if (m_pDb == NULL)
	{
		return m_iEntCount;
	}
	//遍历模型空间,统计实体个数
	AcDbBlockTable * pBlockTab = NULL;
	Acad::ErrorStatus es = m_pDb->getBlockTable(pBlockTab, AcDb::kForRead);
	if (es==Acad::eOk && pBlockTab!=NULL)
	{
		AcDbBlockTableRecord * pRec = NULL;
		es = pBlockTab->getAt(ACDB_MODEL_SPACE, pRec, AcDb::kForRead);
		pBlockTab->close();
		
		AcDbBlockTableRecordIterator *pIterator = NULL;
		if (es==Acad::eOk && pRec!=NULL)
		{
			pRec->newIterator(pIterator);
			pRec->close();
		}
		
		std::map<AcDbObjectId, int> mapEntCountInBlock;	//块内实体个数
		
		for(pIterator->start(); !pIterator->done(); pIterator->step())
		{
			m_iEntCount++;
			//块参照统计块内实体数
			AcDbEntity *pEnt = NULL;
			es = pIterator->getEntity(pEnt, AcDb::kForRead);
			if (es!=Acad::eOk || pEnt==NULL)
			{
				continue;
			}
			if (pEnt->isKindOf(AcDbBlockReference::desc()))
			{
				AcDbBlockReference * pRef = NULL;
				pRef = AcDbBlockReference::cast(pEnt);
				if (pRef==NULL)
				{
					continue;
				}
				AcDbObjectId idBlock = pRef->blockTableRecord();
				m_iEntCount += GetEntCountInBlock(idBlock, mapEntCountInBlock);
			}
			pEnt->close();
		}
		
		delete pIterator;
	}
	return m_iEntCount;
}

int CDwgCompareDbInfo::GetEntCount() const
{
	return m_iEntCount;
}

int CDwgCompareDbInfo::GetEntCountInBlock( AcDbObjectId& idBlock, std::map<AcDbObjectId, int>& mapBlock )
{
	int iCount = 0;
	if (idBlock.isNull())
	{
		return 0;
	}
	//已经计算过的块
	if (mapBlock[idBlock] > 0)
	{
		return mapBlock[idBlock];
	}
	//未计算过的，遍历
	AcDbBlockTableRecord * pRec = NULL;
	Acad::ErrorStatus es = acdbOpenObject(pRec, idBlock, AcDb::kForRead);
	if (es!=Acad::eOk || pRec==NULL)
	{
		return 0;
	}
	AcDbBlockTableRecordIterator* pIter = NULL;
	es = pRec->newIterator(pIter);
	pRec->close();
	if (es!=Acad::eOk || pIter==NULL)
	{
		return 0;
	}
	for (pIter->start(); !pIter->done(); pIter->step())
	{
		iCount++;
	}
	delete pIter;
	mapBlock[idBlock] = iCount;
	
	return iCount;
}
//////////////////////////////////////////////////////////////////////////
CDwgCompareDbMgr::CDwgCompareDbMgr(void)
: m_bAllLayer(TRUE)
, m_bCmpClosedLayer(TRUE)
, m_bCmpFrozenLayer(TRUE)
, m_bCmpLockedlayer(TRUE)
{
}

CDwgCompareDbMgr::~CDwgCompareDbMgr(void)
{
	if (m_pDb != NULL)
	{
		delete m_pDb;
		m_pDb = NULL;
	}
}


void CDwgCompareDbMgr::SetCompareLayer( std::vector<CString>& vtLayer )
{
	m_vtLayerCmp.clear();
	m_vtLayerCmp = vtLayer;
}

void CDwgCompareDbMgr::Init()
{
	m_sName.Empty();
	m_sPath.Empty();
	m_vtLayerAll.clear();
	m_vtLayerCmp.clear();
	if (m_pDb != NULL)
	{
		delete m_pDb;
		m_pDb = NULL;
	}
}

void CDwgCompareDbMgr::GetCmpLayer( std::vector<CString>& vtLayer ) const
{
	vtLayer.clear();
	//无特殊条件都比对
	if (m_bCmpClosedLayer && m_bCmpFrozenLayer && m_bCmpLockedlayer)
	{
		vtLayer = m_vtLayerCmp;
		return;
	}
	//有不比对的条件,vtLayer中删除相应的图层
	AcDbLayerTable * pLayerTab = NULL;
	Acad::ErrorStatus es = m_pDb->getLayerTable(pLayerTab, AcDb::kForRead);
	if (es!=Acad::eOk || pLayerTab==NULL)
	{
		return;
	}

	for (int i=0; i<(int)m_vtLayerCmp.size(); i++)
	{
		AcDbLayerTableRecord * pLayerRec = NULL;
		es = pLayerTab->getAt(m_vtLayerCmp[i], pLayerRec, AcDb::kForRead);
		if (es!=Acad::eOk || pLayerRec==NULL)
		{
			continue;
		}
		BOOL bAppend = TRUE;
		if ( (!m_bCmpClosedLayer && pLayerRec->isOff())
			|| (!m_bCmpFrozenLayer && pLayerRec->isFrozen())
			|| (!m_bCmpLockedlayer && pLayerRec->isLocked())
			)
		{
			bAppend = FALSE;
		}
		if (bAppend)
		{
			vtLayer.push_back(m_vtLayerCmp[i]);
		}
		pLayerRec->close();
	}
	pLayerTab->close();
}