Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

398 lines
13 KiB

5 years ago
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: A cache of a bunch of CSharedObjects
//
//=============================================================================
#include "stdafx.h"
#include <time.h>
// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"
namespace GCSDK
{
#ifdef GC
static GCConVar add_object_clean_do_has_element( "add_object_clean_do_has_element", "0", 0, "Enables AddObjectClean() checking that the cache doesn't already have this pointer" );
#endif
//----------------------------------------------------------------------------
// Purpose: Constructor
//----------------------------------------------------------------------------
CSharedObjectTypeCache::CSharedObjectTypeCache( int nTypeID )
: m_nTypeID( nTypeID )
{
}
//----------------------------------------------------------------------------
// Purpose: Destructor
//----------------------------------------------------------------------------
CSharedObjectTypeCache::~CSharedObjectTypeCache()
{
for ( int i = 0; i < m_vecObjects.Count(); i++ )
{
// NULL the entry so that this SO isn't found during
// cleanup assertion checking.
CSharedObject *pObj = m_vecObjects[ i ];
m_vecObjects[ i ] = NULL;
#ifdef GC
if ( pObj->BShouldDeleteByCache() )
{
#if ENABLE_SO_CONSTRUCT_DESTRUCT_PARANOIA
--pObj->m_nRefCount;
AssertMsg1( pObj->m_nRefCount == 0, "Destroying shared object %s that's still in use!", pObj->GetDebugString().String() );
#endif // ENABLE_SO_CONSTRUCT_DESTRUCT_PARANOIA
delete pObj;
}
#else
delete pObj;
#endif
}
m_vecObjects.Purge();
}
//----------------------------------------------------------------------------
// Purpose: Common shared add-to-cache code shared between AddObject() and
// AddObjectClean().
//----------------------------------------------------------------------------
void CSharedObjectTypeCache::AddObjectInternal( CSharedObject *pObject )
{
Assert( pObject );
m_vecObjects.AddToTail( pObject );
#ifdef GC
#if ENABLE_SO_CONSTRUCT_DESTRUCT_PARANOIA
AssertMsg1( pObject->m_nRefCount >= 0, "AddObjectInternal(): Invalid ref count for shared object %s", pObject->GetDebugString().String() );
++pObject->m_nRefCount;
#endif // ENABLE_SO_CONSTRUCT_DESTRUCT_PARANOIA
#endif
}
//----------------------------------------------------------------------------
// Purpose: Adds a shared object of the appropriate type to this type cache.
//----------------------------------------------------------------------------
bool CSharedObjectTypeCache::AddObject( CSharedObject *pObject )
{
Assert( pObject );
Assert( m_nTypeID == pObject->GetTypeID() );
if( m_vecObjects.HasElement( pObject ) )
return false;
AddObjectInternal( pObject );
return true;
}
//----------------------------------------------------------------------------
// Purpose: Adds an object without dirtying. This is done when the object
// is just being loaded from SQL or memcached, so it's safe not to do the
// has element check.
//----------------------------------------------------------------------------
bool CSharedObjectTypeCache::AddObjectClean( CSharedObject *pObject )
{
Assert( m_nTypeID == pObject->GetTypeID() );
#ifdef GC
if ( add_object_clean_do_has_element.GetBool() )
{
Assert( !m_vecObjects.HasElement( pObject ) );
if( m_vecObjects.HasElement( pObject ) )
return false;
}
#endif
AddObjectInternal( pObject );
return true;
}
//----------------------------------------------------------------------------
// Purpose: Destroys the object matching the one passed in. This could be the
// same one or simply one with matching index fields.
//----------------------------------------------------------------------------
CSharedObject *CSharedObjectTypeCache::RemoveObject( const CSharedObject & soIndex )
{
Assert( m_nTypeID == soIndex.GetTypeID() ); // This is probably harmless, but it's most likely a bug
int nIndex = FindSharedObjectIndex( soIndex );
if( m_vecObjects.IsValidIndex( nIndex ) )
{
return RemoveObjectByIndex( nIndex );
}
else
{
return NULL;
}
}
CSharedObject *CSharedObjectTypeCache::RemoveObjectByIndex( uint32 nObj )
{
CSharedObject *pObj = m_vecObjects[nObj];
#ifdef GC
#if ENABLE_SO_CONSTRUCT_DESTRUCT_PARANOIA
AssertMsg1( pObj->m_nRefCount > 0, "Invalid ref count for shared object %s", pObj->GetDebugString().String() );
--pObj->m_nRefCount;
#endif // ENABLE_SO_CONSTRUCT_DESTRUCT_PARANOIA
#endif // GC
m_vecObjects.Remove( nObj );
return pObj;
}
//----------------------------------------------------------------------------
// Purpose: Empties the object lists and deletes all elements
//----------------------------------------------------------------------------
void CSharedObjectTypeCache::DestroyAllObjects()
{
for ( int i = 0; i < m_vecObjects.Count(); i++ )
{
#ifdef GC
if ( m_vecObjects[i]->BShouldDeleteByCache() )
{
#if ENABLE_SO_CONSTRUCT_DESTRUCT_PARANOIA
--m_vecObjects[i]->m_nRefCount;
AssertMsg1( m_vecObjects[i]->m_nRefCount == 0, "Destroying shared object %s that's still in use!", m_vecObjects[i]->GetDebugString().String() );
#endif // ENABLE_SO_CONSTRUCT_DESTRUCT_PARANOIA
delete m_vecObjects[i];
}
#else
delete m_vecObjects[i];
#endif
}
m_vecObjects.Purge();
}
//----------------------------------------------------------------------------
// Purpose: Empties the object lists but doesn't delete any of the objects
//----------------------------------------------------------------------------
void CSharedObjectTypeCache::RemoveAllObjectsWithoutDeleting()
{
m_vecObjects.RemoveAll();
}
//----------------------------------------------------------------------------
// Purpose: Makes sure there's room in the object vector for the suggested
// number of items
//----------------------------------------------------------------------------
void CSharedObjectTypeCache::EnsureCapacity( uint32 nItems )
{
m_vecObjects.EnsureCapacity( nItems );
}
//----------------------------------------------------------------------------
// Purpose: Searches the object list for an object that matches the provided
// object on its index fields.
//----------------------------------------------------------------------------
CSharedObject *CSharedObjectTypeCache::FindSharedObject( const CSharedObject & soIndex )
{
int nIndex = FindSharedObjectIndex( soIndex );
if( m_vecObjects.IsValidIndex( nIndex ) )
return m_vecObjects[nIndex];
else
return NULL;
}
//----------------------------------------------------------------------------
// Purpose: Searches the object list for an object that matches the provided
// object on its index fields.
//----------------------------------------------------------------------------
int CSharedObjectTypeCache::FindSharedObjectIndex( const CSharedObject & soIndex ) const
{
FOR_EACH_VEC( m_vecObjects, nObj )
{
if( m_vecObjects[nObj]->BIsKeyEqual( soIndex ) )
return nObj;
}
return -1;
}
//----------------------------------------------------------------------------
// Purpose: Dumps all the objects in the type cache
//----------------------------------------------------------------------------
void CSharedObjectTypeCache::Dump() const
{
EmitInfo( SPEW_CONSOLE, SPEW_ALWAYS, LOG_ALWAYS, "\tTypeCache for %d (%d objects):\n", GetTypeID(), m_vecObjects.Count() );
FOR_EACH_VEC( m_vecObjects, nObj )
{
m_vecObjects[nObj]->Dump();
}
}
//----------------------------------------------------------------------------
// Purpose: Claims all the memory for the cache and its objects
//----------------------------------------------------------------------------
#ifdef DBGFLAG_VALIDATE
void CSharedObjectTypeCache::Validate( CValidator &validator, const char *pchName )
{
VALIDATE_SCOPE();
ValidateObj( m_vecObjects );
FOR_EACH_VEC( m_vecObjects, nIndex )
{
m_vecObjects[nIndex]->Validate( validator, "m_vecObjects[n]" );
}
}
#endif
//----------------------------------------------------------------------------
// Purpose: Constructor
//----------------------------------------------------------------------------
CSharedObjectCache::CSharedObjectCache( )
: m_mapObjects( DefLessFunc(int) )
, m_ulVersion( 0 )
{
}
//----------------------------------------------------------------------------
// Purpose: Destructor
//----------------------------------------------------------------------------
CSharedObjectCache::~CSharedObjectCache()
{
FOR_EACH_MAP( m_mapObjects, nTypeIndex )
{
delete m_mapObjects[nTypeIndex];
}
m_mapObjects.Purge();
}
//----------------------------------------------------------------------------
// Purpose: Returns the type cache for the specified type ID, returning NULL
// if the cache didn't previously exist.
//----------------------------------------------------------------------------
CSharedObjectTypeCache *CSharedObjectCache::FindBaseTypeCache( int nClassID ) const
{
int nIndex = m_mapObjects.Find( nClassID );
CSharedObjectTypeCache *pTypeCache = NULL;
if( m_mapObjects.IsValidIndex( nIndex ) )
{
pTypeCache = m_mapObjects[nIndex];
}
return pTypeCache;
}
//----------------------------------------------------------------------------
// Purpose: Returns the type cache for the specified type ID, creating a new
// cache and returning it if one didn't previously exist. Never intended
// to return NULL.
//----------------------------------------------------------------------------
CSharedObjectTypeCache *CSharedObjectCache::CreateBaseTypeCache( int nClassID )
{
//see if we already have an existing one
CSharedObjectTypeCache *pCache = FindBaseTypeCache( nClassID );
if( pCache )
return pCache;
//nope, need to create one
CSharedObjectTypeCache* pTypeCache = AllocateTypeCache( nClassID );
m_mapObjects.Insert( nClassID, pTypeCache );
#if 0
// Kyle says: this is the newer way of managing caches on Dota but we haven't
// brought any of it over yet
m_CacheObjects.AddToTail( pTypeCache );
//sort this cache for faster access
std::sort( m_CacheObjects.begin(), m_CacheObjects.end(), SortCacheByTypeID );
#endif
return pTypeCache;
}
//----------------------------------------------------------------------------
// Purpose: Adds a shared object to the cache.
//----------------------------------------------------------------------------
bool CSharedObjectCache::AddObject( CSharedObject *pSharedObject )
{
CSharedObjectTypeCache *pTypeCache = CreateBaseTypeCache( pSharedObject->GetTypeID() );
if ( !pTypeCache->AddObject( pSharedObject ) )
return false;
MarkDirty();
return true;
}
//----------------------------------------------------------------------------
// Purpose: Removes the object matching the one passed in from this cache,
// without destroying the actual object.
//----------------------------------------------------------------------------
CSharedObject *CSharedObjectCache::RemoveObject( const CSharedObject & soIndex )
{
CSharedObjectTypeCache *pTypeCache = FindBaseTypeCache( soIndex.GetTypeID() );
if( !pTypeCache )
return NULL;
MarkDirty();
return pTypeCache->RemoveObject( soIndex );
}
//----------------------------------------------------------------------------
// Purpose: Empties the object lists but doesn't delete any of the objects
//----------------------------------------------------------------------------
void CSharedObjectCache::RemoveAllObjectsWithoutDeleting()
{
FOR_EACH_MAP_FAST( m_mapObjects, nType )
{
m_mapObjects[nType]->RemoveAllObjectsWithoutDeleting();
}
MarkDirty();
}
//----------------------------------------------------------------------------
// Purpose: Searches the object list for an object that matches the provided
// object on its index fields.
//----------------------------------------------------------------------------
CSharedObject *CSharedObjectCache::FindSharedObject( const CSharedObject & soIndex )
{
CSharedObjectTypeCache *pTypeCache = FindBaseTypeCache( soIndex.GetTypeID() );
if( pTypeCache )
return pTypeCache->FindSharedObject( soIndex );
else
return NULL;
}
//----------------------------------------------------------------------------
// Purpose: Dumps all the objects in the type cache
//----------------------------------------------------------------------------
void CSharedObjectCache::Dump() const
{
EmitInfo( SPEW_CONSOLE, SPEW_ALWAYS, LOG_ALWAYS, "SharedObjectCache for %s (%d types):\n", GetOwner().Render(), m_mapObjects.Count() );
FOR_EACH_MAP( m_mapObjects, nTypeIndex )
{
m_mapObjects[nTypeIndex]->Dump();
}
}
//----------------------------------------------------------------------------
// Purpose: Claims all the memory for the cache
//----------------------------------------------------------------------------
#ifdef DBGFLAG_VALIDATE
void CSharedObjectCache::Validate( CValidator &validator, const char *pchName )
{
VALIDATE_SCOPE();
ValidateObj( m_mapObjects );
FOR_EACH_MAP( m_mapObjects, nTypeIndex )
{
m_mapObjects[nTypeIndex]->Validate( validator, "m_mapObjects[n]" );
}
}
#endif
} // namespace GCSDK