225 lines
6.6 KiB
C++
225 lines
6.6 KiB
C++
//========= Copyright Valve Corporation, All rights reserved. ============//
|
|
//
|
|
// Purpose:
|
|
//
|
|
// $NoKeywords: $
|
|
//=============================================================================//
|
|
|
|
#include "cbase.h"
|
|
|
|
#include "stringregistry.h"
|
|
#include "ai_namespaces.h"
|
|
|
|
// memdbgon must be the last include file in a .cpp file!!!
|
|
#include "tier0/memdbgon.h"
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// CAI_GlobalNamespace
|
|
//
|
|
|
|
CAI_GlobalNamespace::CAI_GlobalNamespace()
|
|
: m_pSymbols( new CStringRegistry ),
|
|
m_NextGlobalBase( GLOBAL_IDS_BASE )
|
|
{
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
CAI_GlobalNamespace::~CAI_GlobalNamespace()
|
|
{
|
|
delete m_pSymbols;
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
void CAI_GlobalNamespace::Clear()
|
|
{
|
|
m_pSymbols->ClearStrings();
|
|
m_NextGlobalBase = GLOBAL_IDS_BASE;
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
void CAI_GlobalNamespace::AddSymbol( const char *pszSymbol, int symbolID )
|
|
{
|
|
MEM_ALLOC_CREDIT();
|
|
AssertMsg( symbolID != -1, "Invalid symbol id passed to CAI_GlobalNamespace::AddSymbol()" );
|
|
if (symbolID == -1 )
|
|
return;
|
|
|
|
AssertMsg( AI_IdIsGlobal( symbolID ), ("Local symbol ID passed to CAI_GlobalNamespace::AddSymbol()") );
|
|
AssertMsg( !IdToSymbol( symbolID ) , ("Duplicate symbol ID passed to CAI_GlobalNamespace::AddSymbol()") );
|
|
AssertMsg( SymbolToId( pszSymbol ) == -1, ("Duplicate symbol passed to CAI_GlobalNamespace::AddSymbol()") );
|
|
|
|
m_pSymbols->AddString( pszSymbol, symbolID );
|
|
|
|
if ( m_NextGlobalBase < symbolID + 1 )
|
|
m_NextGlobalBase = symbolID + 1;
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
const char *CAI_GlobalNamespace::IdToSymbol( int symbolID ) const
|
|
{
|
|
AssertMsg( AI_IdIsGlobal( symbolID ), ("Local symbol ID passed to CAI_GlobalNamespace::IdToSymbol()") );
|
|
if ( symbolID == -1 )
|
|
return "<<null>>";
|
|
return (const_cast<CStringRegistry *>(m_pSymbols))->GetStringText( symbolID );
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
int CAI_GlobalNamespace::SymbolToId( const char *pszSymbol ) const
|
|
{
|
|
return (const_cast<CStringRegistry *>(m_pSymbols))->GetStringID( pszSymbol );
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
int CAI_GlobalNamespace::NextGlobalBase() const
|
|
{
|
|
return m_NextGlobalBase;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// CAI_LocalIdSpace
|
|
//
|
|
// Purpose: Maps per class IDs to global IDs, so that various classes can use
|
|
// the same integer in local space to represent different globally
|
|
// unique integers. Used for schedules, tasks, conditons and squads
|
|
//
|
|
|
|
CAI_LocalIdSpace::CAI_LocalIdSpace( bool fIsRoot )
|
|
: m_pGlobalNamespace( NULL ),
|
|
m_pParentIDSpace( NULL ),
|
|
m_globalBase( (fIsRoot) ? 0 : -1 ),
|
|
m_localBase( (fIsRoot) ? 0 : MAX_STRING_INDEX ),
|
|
m_localTop( -1 ),
|
|
m_globalTop( -1 )
|
|
{
|
|
};
|
|
|
|
//-------------------------------------
|
|
|
|
bool CAI_LocalIdSpace::Init( CAI_GlobalNamespace *pGlobalNamespace, CAI_LocalIdSpace *pParentIDSpace )
|
|
{
|
|
if ( m_globalTop != -1 )
|
|
{
|
|
m_localBase = (!pParentIDSpace) ? 0 : MAX_STRING_INDEX;
|
|
m_localTop = -1;
|
|
m_globalTop = -1;
|
|
}
|
|
|
|
m_pParentIDSpace = pParentIDSpace;
|
|
m_pGlobalNamespace = pGlobalNamespace;
|
|
m_globalBase = m_pGlobalNamespace->NextGlobalBase();
|
|
|
|
return true;
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
bool CAI_LocalIdSpace::AddSymbol( const char *pszSymbol, int localId, const char *pszDebugSymbolType, const char *pszDebugOwner )
|
|
{
|
|
AssertMsg( AI_IdIsLocal( localId ), ("Global symbol ID passed to CAI_LocalIdSpace::AddSymbol()") );
|
|
|
|
if ( !m_pGlobalNamespace )
|
|
{
|
|
DevMsg( "ERROR: Adding symbol to uninitialized table %s\n", pszDebugOwner );
|
|
return false;
|
|
}
|
|
|
|
if ( !IsLocalBaseSet() && !SetLocalBase( localId ) )
|
|
{
|
|
DevMsg( "ERROR: Bad %s LOCALID for %s\n", pszDebugSymbolType, pszDebugOwner );
|
|
return false;
|
|
}
|
|
else if (localId < GetLocalBase() )
|
|
{
|
|
DevMsg("ERROR: %s First added %s must be first LOCALID!\n", pszDebugSymbolType, pszDebugOwner);
|
|
return false;
|
|
}
|
|
|
|
AssertMsg3( LocalToGlobal( localId ) == -1 || !m_pGlobalNamespace->IdToSymbol( LocalToGlobal( localId ) ) , "Duplicate symbol ID passed to CAI_LocalIdSpace::AddSymbol(): %s (%d), had %s", pszSymbol, localId, m_pGlobalNamespace->IdToSymbol( LocalToGlobal( localId ) ) );
|
|
AssertMsg( m_pGlobalNamespace->SymbolToId( pszSymbol ) == -1, "Duplicate symbol passed to CAI_LocalIdSpace::AddSymbol()" );
|
|
|
|
if ( m_localTop != -1 )
|
|
{
|
|
if ( localId > m_localTop )
|
|
{
|
|
m_localTop = localId;
|
|
m_globalTop = ( m_localTop - m_localBase ) + m_globalBase;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
m_localTop = m_localBase;
|
|
m_globalTop = m_globalBase;
|
|
}
|
|
|
|
m_pGlobalNamespace->AddSymbol( pszSymbol, LocalToGlobal( localId ) );
|
|
return true;
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
bool CAI_LocalIdSpace::SetLocalBase( int newBase )
|
|
{
|
|
AssertMsg( AI_IdIsLocal( newBase ), ("Global symbol ID passed to CAI_LocalIdSpace::SetLocalBase()") );
|
|
|
|
if (m_localBase == MAX_STRING_INDEX)
|
|
{
|
|
// UNDONE: Make sure this is the largest Index in the list so far??
|
|
m_localBase = newBase;
|
|
if ( !m_pParentIDSpace || !m_pParentIDSpace->IsLocalBaseSet() || m_localBase > m_pParentIDSpace->m_localBase )// @Note (toml 08-20-02): this doesn't actually catch namespace collisions, only woefully bad situations where a derived class' offset is less than the base
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
int CAI_LocalIdSpace::GlobalToLocal( int globalID ) const
|
|
{
|
|
if ( globalID == -1 )
|
|
return -1;
|
|
|
|
AssertMsg( AI_IdIsGlobal( globalID ), ("Local symbol ID passed to CAI_LocalIdSpace::GlobalToLocal()") );
|
|
|
|
const CAI_LocalIdSpace *pCurrentMap = this;
|
|
|
|
do
|
|
{
|
|
if ( pCurrentMap->IsLocalBaseSet() && globalID >= pCurrentMap->GetGlobalBase() && globalID <= pCurrentMap->GetGlobalTop() )
|
|
return ( globalID - pCurrentMap->GetGlobalBase() + pCurrentMap->GetLocalBase() );
|
|
pCurrentMap = pCurrentMap->m_pParentIDSpace;
|
|
} while ( pCurrentMap != NULL );
|
|
|
|
// AssertMsg( 0, ("Invalid ID passed to CAI_LocalIdSpace::GlobalToLocal()") );
|
|
return -1;
|
|
}
|
|
|
|
//-------------------------------------
|
|
|
|
int CAI_LocalIdSpace::LocalToGlobal( int localID ) const
|
|
{
|
|
if ( localID == -1 )
|
|
return -1;
|
|
|
|
AssertMsg( AI_IdIsLocal( localID ), ("Global symbol ID passed to CAI_LocalIdSpace::LocalToGlobal()") );
|
|
|
|
const CAI_LocalIdSpace *pCurrentMap = this;
|
|
|
|
do
|
|
{
|
|
if ( pCurrentMap->IsLocalBaseSet() && localID >= pCurrentMap->GetLocalBase() && localID <= pCurrentMap->GetLocalTop() )
|
|
return ( localID + pCurrentMap->GetGlobalBase() - pCurrentMap->GetLocalBase() );
|
|
pCurrentMap = pCurrentMap->m_pParentIDSpace;
|
|
} while ( pCurrentMap != NULL );
|
|
|
|
// AssertMsg( 0, ("Invalid ID passed to CAI_LocalIdSpace::LocalToGlobal()") );
|
|
return -1;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|