void CMaxMaterialCollection::OnPreFileLinkEvent()
{
//stash all FileLinked materials int a tab. When the event is complete,
//we need to update refcounts and notify any clients or removals
//assert(mCacheLinkedMtls.size() == 0);
if(mCacheLinkedMtls.size() != 0)
return; //this can happen on PRE_NEW which spawns a PRE_BIND, no need to collect again
IVizLinkTable* linktbl =
static_cast<IVizLinkTable*>(GetCOREInterface(FILELINKMGR_INTERFACE_ID));
if(!linktbl)
return;
//Iterate through the linked materials. For any new material, create a COM wrapper.
//Creating the wrapper will also fire and event to the palette system.
int numlinks = linktbl->NumLinkedFiles();
IVizLinkTable::Iterator it;
for(int i =0; i<numlinks; i++)
{
if(!linktbl->GetLinkID(i, it))
continue;
LinkTableRecord *pLTR = linktbl->RecordAt(it);
if((pLTR == NULL) || (pLTR->ClassID() != DWGTABLERECORD_CLASSID))
continue;
ReferenceMaker* pMtls = pLTR->GetReference(DTR_MTLTABLE_REF);
if((pMtls == NULL) || (pMtls->ClassID() != VZMATERIALTABLE_CLASSID))
continue;
int numMtls = pMtls->NumRefs();
//enlarge our tab
// mCacheLinkedMtls.Resize(mCacheLinkedMtls.Count() + numMtls);
for(int i=0; i< numMtls; i++)
{
//get the max material to be wrapped
ReferenceTarget *rtarg = pMtls->GetReference(i);
if((rtarg == NULL) || (rtarg->SuperClassID() != MATERIAL_CLASS_ID))
continue;
Mtl* pMtl = (Mtl*)rtarg;
assert(pMtl);
//find the wrapper, which we should have already
ifxClxnType::iterator it = mMtlMap.find((DWORD_PTR)pMtl);
if(it != mMtlMap.end())
mCacheLinkedMtls.insert(*it);
else
assert(0);
//mCacheLinkedMtls.Append(1, &pMtl);
}//next mtl
}//next link
}
const TCHAR *RefWatch::GetName() {
SClass_ID sID = watchMe->SuperClassID();
if ((sID == MATERIAL_CLASS_ID) || (sID == TEXMAP_CLASS_ID))
return ((MtlBase *)watchMe)->GetName();
else if (sID == BASENODE_CLASS_ID)
return ((INode *)watchMe)->GetName();
else if (sID == ATMOSPHERIC_CLASS_ID)
return ((Atmospheric *)watchMe)->GetName();
else if ((sID == OSM_CLASS_ID) || (sID == WSM_CLASS_ID) ||
(sID == GEOMOBJECT_CLASS_ID) || (sID == WSM_OBJECT_CLASS_ID))
return ((BaseObject *)watchMe)->GetObjectName();
return _T("(Unknown)");
}
void CMaxMaterialCollection::OnFileLinkMtlsChanged()
{
//look for any new materials (i.e unwrapped) and add them
//existing ones should be updated
IVizLinkTable* linktbl =
static_cast<IVizLinkTable*>(GetCOREInterface(FILELINKMGR_INTERFACE_ID));
if(!linktbl)
return;
//Iterate through the linked materials. For any new material, create a COM wrapper.
//Creating the wrapper will also fire and event to the palette system.
int numlinks = linktbl->NumLinkedFiles();
IVizLinkTable::Iterator it;
for(int j =0; j<numlinks; j++)
{
if(!linktbl->GetLinkID(j, it))
continue;
LinkTableRecord *pLTR = linktbl->RecordAt(it);
if((pLTR == NULL) || (pLTR->ClassID() != DWGTABLERECORD_CLASSID))
continue;
ReferenceMaker* pMtls = pLTR->GetReference(DTR_MTLTABLE_REF);
if((pMtls == NULL) || (pMtls->ClassID() != VZMATERIALTABLE_CLASSID))
continue;
int numMtls = pMtls->NumRefs();
for(int i=0; i< numMtls; i++)
{
//get the max material to be wrapped
ReferenceTarget *rtarg = pMtls->GetReference(i);
if((rtarg == NULL) || (rtarg->SuperClassID() != MATERIAL_CLASS_ID))
continue;
Mtl* pMtl = (Mtl*)rtarg;
assert(pMtl);
DWORD_PTR key = (DWORD_PTR) pMtl;
CMaxMaterial* pWrapper = findItemKey(key);
if (pWrapper != NULL)
{
pWrapper->CheckReference(false, true);
#ifdef TP_SUSPEND_FOR_FILELINK
//processed so remove it from the cache
mCacheLinkedMtls.erase(key);
#endif
continue;
}
//Create the wrapper
CComObject<CMaxMaterial> *pWrapperObject;
HRESULT hr = CComObject<CMaxMaterial>::CreateInstance(&pWrapperObject);
assert(SUCCEEDED(hr));
if(!SUCCEEDED(hr))
{
s_PreventRecursion = false;
return;
}
//Initalize the new wrapper
pWrapperObject->setCollection(this);
//TODO validate the downcast
pWrapperObject->setMtl((Mtl *)pMtl);
}//next mtl
}//next link
#ifdef TP_SUSPEND_FOR_FILELINK
//Note that if a material is removed by filelink, we should really remove entirely
//anything left in mCacheLinkedMtls at this point is a candidate for destructo
//or at least an update
ifxClxnType::iterator iter=mCacheLinkedMtls.begin();
while(iter!=mCacheLinkedMtls.end())
{
MtlWrapper *pWrapper = (*iter).second;
assert(pWrapper);
OnMtlDeleted(pWrapper, (*iter).first);
// pMtlWrapper->CheckReference(false, true);
iter++;
}
mCacheLinkedMtls.clear();
#endif
}