121 lines
3.4 KiB
C++
121 lines
3.4 KiB
C++
|
//========= Copyright Valve Corporation, All rights reserved. ============//
|
||
|
|
//
|
||
|
|
// Purpose:
|
||
|
|
//
|
||
|
|
//=============================================================================
|
||
|
|
|
||
|
|
#include "dmserializers.h"
|
||
|
|
#include "dmebaseimporter.h"
|
||
|
|
#include "datamodel/idatamodel.h"
|
||
|
|
#include "datamodel/dmelement.h"
|
||
|
|
#include "tier1/KeyValues.h"
|
||
|
|
#include "tier1/utlbuffer.h"
|
||
|
|
#include "tier1/utlmap.h"
|
||
|
|
#include <limits.h>
|
||
|
|
|
||
|
|
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
// Format converter
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
class CImportSFMV5 : public CSFMBaseImporter
|
||
|
|
{
|
||
|
|
typedef CSFMBaseImporter BaseClass;
|
||
|
|
public:
|
||
|
|
CImportSFMV5( char const *formatName, char const *nextFormatName );
|
||
|
|
|
||
|
|
private:
|
||
|
|
virtual bool DoFixup( CDmElement *pSourceRoot );
|
||
|
|
|
||
|
|
|
||
|
|
void FixupElement( CDmElement *pElement );
|
||
|
|
// Fixes up all elements
|
||
|
|
void BuildList( CDmElement *pElement, CUtlRBTree< CDmElement *, int >& list );
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
// Singleton instance
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
static CImportSFMV5 s_ImportSFMV5( "sfm_v5", "sfm_v6" );
|
||
|
|
|
||
|
|
void InstallSFMV5Importer( IDataModel *pFactory )
|
||
|
|
{
|
||
|
|
pFactory->AddLegacyUpdater( &s_ImportSFMV5 );
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
// Constructor
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
CImportSFMV5::CImportSFMV5( char const *formatName, char const *nextFormatName ) :
|
||
|
|
BaseClass( formatName, nextFormatName )
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
// Fixes up all elements
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
void CImportSFMV5::FixupElement( CDmElement *pElement )
|
||
|
|
{
|
||
|
|
if ( !pElement )
|
||
|
|
return;
|
||
|
|
|
||
|
|
const char *pType = pElement->GetTypeString();
|
||
|
|
|
||
|
|
if ( !V_stricmp( pType, "DmeSpotLight" ) )
|
||
|
|
{
|
||
|
|
pElement->SetType( "DmeProjectedLight" );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
// Fixes up all elements
|
||
|
|
//-----------------------------------------------------------------------------
|
||
|
|
void CImportSFMV5::BuildList( CDmElement *pElement, CUtlRBTree< CDmElement *, int >& list )
|
||
|
|
{
|
||
|
|
if ( !pElement )
|
||
|
|
return;
|
||
|
|
|
||
|
|
if ( list.Find( pElement ) != list.InvalidIndex() )
|
||
|
|
return;
|
||
|
|
|
||
|
|
list.Insert( pElement );
|
||
|
|
|
||
|
|
// Descend to bottom of tree, then do fixup coming back up the tree
|
||
|
|
for ( CDmAttribute *pAttribute = pElement->FirstAttribute(); pAttribute; pAttribute = pAttribute->NextAttribute() )
|
||
|
|
{
|
||
|
|
if ( pAttribute->GetType() == AT_ELEMENT )
|
||
|
|
{
|
||
|
|
CDmElement *pElementAt = pAttribute->GetValueElement<CDmElement>( );
|
||
|
|
BuildList( pElementAt, list );
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( pAttribute->GetType() == AT_ELEMENT_ARRAY )
|
||
|
|
{
|
||
|
|
CDmrElementArray<> array( pAttribute );
|
||
|
|
int nCount = array.Count();
|
||
|
|
for ( int i = 0; i < nCount; ++i )
|
||
|
|
{
|
||
|
|
CDmElement *pChild = array[ i ];
|
||
|
|
BuildList( pChild, list );
|
||
|
|
}
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bool CImportSFMV5::DoFixup( CDmElement *pSourceRoot )
|
||
|
|
{
|
||
|
|
CUtlRBTree< CDmElement *, int > fixlist( 0, 0, DefLessFunc( CDmElement * ) );
|
||
|
|
BuildList( pSourceRoot, fixlist );
|
||
|
|
for ( int i = fixlist.FirstInorder(); i != fixlist.InvalidIndex() ; i = fixlist.NextInorder( i ) )
|
||
|
|
{
|
||
|
|
// Search and replace in the entire tree!
|
||
|
|
FixupElement( fixlist[ i ] );
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|