DInsight

Code: Select all

void TestBuildChildToParentMap()
{
	....
	IModel_DGPtr iModel_DG;
	model.QueryInterface(__uuidof(IModel_DG), (void**)&iModel_DG);
	IMapUnkn_DGPtr map;
	iModel_DG->BuildChildToParentMap(&map);

	// Get flat array of all independently of child/parent 
	IEntityArray_DGPtr entities;
	iModel_DG->GetEntityArray(VARIANT_TRUE, &entities);

	int n = 0;
	entities->GetCount(&n);

	for (int i = 0; i < n; i++)
	{
		IEntity_DGPtr iEntity;
		entities->GetAt(i, &iEntity);

		IObject_DG* iObject = nullptr;
		iEntity->QueryInterface(__uuidof(IObject_DG), (void**)&iObject);

		__int64 hash = 0;
		iObject->GetHash(&hash);

		int nChildren = 0;
		iEntity->GetChildCount(&nChildren);

		for (int k = 0; k < nChildren; k++)
		{
			IEntity_DGPtr iEntityChild;
			iEntity->GetChildAt(k, &iEntityChild);

			IUnknown* iResult = nullptr;
			map->GetAt(iEntityChild, &iResult);

			IEntity_DGPtr iEntityResult;
			iResult->QueryInterface(__uuidof(IEntity_DG), (void**)&iEntityResult);

			Entity* pEntInv = nullptr;
			iEntityResult->QueryInterface(IID_IInverse, (void**)&pEntInv);

			IObject_DG* iObjectChild = nullptr;
			iEntityResult->QueryInterface(__uuidof(IObject_DG), (void**)&iObjectChild);

			VARIANT_BOOL same = VARIANT_FALSE;
			iObjectChild->IsSameInstance(iObject, &same);
			assert(same == VARIANT_TRUE);

			// Hash must be the same also (testing hash):
			__int64 hashChild = 0;
			iObjectChild->GetHash(&hashChild);
			assert(hashChild == hash);

			RELEASE(iObjectChild); RELEASE(iEntityResult);	RELEASE(iEntityChild);
		}

		RELEASE(iObject); RELEASE(iEntity);
	}
	RELEASE(entities);	RELEASE(map);
}