Team Fortress 2 Source Code as on 22/4/2020
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.
 
 
 
 
 
 

85 lines
1.7 KiB

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
//=============================================================================
#ifndef THREADSAFEREFCOUNTEDOBJECT_H
#define THREADSAFEREFCOUNTEDOBJECT_H
#ifdef _WIN32
#pragma once
#endif
// This class can be used for fast access to an object from multiple threads,
// and the main thread can wait until the threads are done using the object before it frees the object.
template< class T >
class CThreadSafeRefCountedObject
{
public:
CThreadSafeRefCountedObject( T initVal )
{
m_RefCount = 0;
m_pObject = initVal;
m_RefCount = 0;
}
void Init( T pObj )
{
Assert( ThreadInMainThread() );
Assert( !m_pObject );
m_RefCount = 0;
m_pObject = pObj;
m_RefCount = 1;
}
// Threads that access the object need to use AddRef/Release to access it.
T AddRef()
{
if ( ++m_RefCount > 1 )
{
return m_pObject;
}
else
{
// If the refcount was 0 when we called this, then the whitelist is about to be freed.
--m_RefCount;
return NULL;
}
}
void ReleaseRef( T pObj )
{
if ( --m_RefCount >= 1 )
{
Assert( m_pObject == pObj );
}
}
// The main thread can use this to access the object, since only it can Init() and Free() the object.
T GetInMainThread()
{
Assert( ThreadInMainThread() );
return m_pObject;
}
// The main thread calls this after it has released its last reference to the object.
void ResetWhenNoRemainingReferences( T newValue )
{
Assert( ThreadInMainThread() );
// Wait until we can free it.
while ( m_RefCount > 0 )
{
CThread::Sleep( 20 );
}
m_pObject = newValue;
}
private:
CInterlockedIntT<long> m_RefCount;
T m_pObject;
};
#endif // THREADSAFEREFCOUNTEDOBJECT_H