138 lines
2.7 KiB
C++
138 lines
2.7 KiB
C++
//========= Copyright Valve Corporation, All rights reserved. ============//
|
|
//
|
|
// Purpose:
|
|
//
|
|
// $NoKeywords: $
|
|
//=============================================================================//
|
|
#include "cbase.h"
|
|
#include "iclassmap.h"
|
|
#include "utldict.h"
|
|
|
|
// memdbgon must be the last include file in a .cpp file!!!
|
|
#include "tier0/memdbgon.h"
|
|
|
|
class classentry_t
|
|
{
|
|
public:
|
|
classentry_t()
|
|
{
|
|
mapname[ 0 ] = 0;
|
|
factory = 0;
|
|
size = -1;
|
|
}
|
|
|
|
char const *GetMapName() const
|
|
{
|
|
return mapname;
|
|
}
|
|
|
|
void SetMapName( char const *newname )
|
|
{
|
|
Q_strncpy( mapname, newname, sizeof( mapname ) );
|
|
}
|
|
|
|
DISPATCHFUNCTION factory;
|
|
int size;
|
|
private:
|
|
char mapname[ 40 ];
|
|
};
|
|
|
|
class CClassMap : public IClassMap
|
|
{
|
|
public:
|
|
virtual void Add( const char *mapname, const char *classname, int size, DISPATCHFUNCTION factory /*= 0*/ );
|
|
virtual const char *Lookup( const char *classname );
|
|
virtual C_BaseEntity *CreateEntity( const char *mapname );
|
|
virtual int GetClassSize( const char *classname );
|
|
|
|
private:
|
|
CUtlDict< classentry_t, unsigned short > m_ClassDict;
|
|
};
|
|
|
|
IClassMap& GetClassMap( void )
|
|
{
|
|
static CClassMap g_Classmap;
|
|
return g_Classmap;
|
|
}
|
|
|
|
void CClassMap::Add( const char *mapname, const char *classname, int size, DISPATCHFUNCTION factory = 0 )
|
|
{
|
|
const char *map = Lookup( classname );
|
|
if ( map && !Q_strcasecmp( mapname, map ) )
|
|
return;
|
|
|
|
if ( map )
|
|
{
|
|
int index = m_ClassDict.Find( classname );
|
|
Assert( index != m_ClassDict.InvalidIndex() );
|
|
m_ClassDict.RemoveAt( index );
|
|
}
|
|
|
|
classentry_t element;
|
|
element.SetMapName( mapname );
|
|
element.factory = factory;
|
|
element.size = size;
|
|
m_ClassDict.Insert( classname, element );
|
|
}
|
|
|
|
const char *CClassMap::Lookup( const char *classname )
|
|
{
|
|
unsigned short index;
|
|
static classentry_t lookup;
|
|
|
|
index = m_ClassDict.Find( classname );
|
|
if ( index == m_ClassDict.InvalidIndex() )
|
|
return NULL;
|
|
|
|
lookup = m_ClassDict.Element( index );
|
|
return lookup.GetMapName();
|
|
}
|
|
|
|
C_BaseEntity *CClassMap::CreateEntity( const char *mapname )
|
|
{
|
|
int c = m_ClassDict.Count();
|
|
int i;
|
|
|
|
for ( i = 0; i < c; i++ )
|
|
{
|
|
classentry_t *lookup = &m_ClassDict[ i ];
|
|
if ( !lookup )
|
|
continue;
|
|
|
|
if ( Q_stricmp( lookup->GetMapName(), mapname ) )
|
|
continue;
|
|
|
|
if ( !lookup->factory )
|
|
{
|
|
#if defined( _DEBUG )
|
|
Msg( "No factory for %s/%s\n", lookup->GetMapName(), m_ClassDict.GetElementName( i ) );
|
|
#endif
|
|
continue;
|
|
}
|
|
|
|
return ( *lookup->factory )();
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
int CClassMap::GetClassSize( const char *classname )
|
|
{
|
|
int c = m_ClassDict.Count();
|
|
int i;
|
|
|
|
for ( i = 0; i < c; i++ )
|
|
{
|
|
classentry_t *lookup = &m_ClassDict[ i ];
|
|
if ( !lookup )
|
|
continue;
|
|
|
|
if ( Q_strcmp( lookup->GetMapName(), classname ) )
|
|
continue;
|
|
|
|
return lookup->size;
|
|
}
|
|
|
|
return -1;
|
|
}
|