1HEADmaster

1 files changed, 107 insertions, 0 deletions

diff --git a/public/gcsdk/refcount.h b/public/gcsdk/refcount.h
new file mode 100644
index 0000000..ceaa1aa
--- /dev/null
+++ b/public/gcsdk/refcount.h
@@ -0,0 +1,107 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// $NoKeywords: $
+//=============================================================================
+
+#ifndef GCREFCOUNT_H
+#define GCREFCOUNT_H
+
+#include "tier0/memdbgon.h"
+
+namespace GCSDK
+{
+
+// Base class for ref counted classes.  Derive from this to be refcounted.  Note:
+// you can no longer be deleted directly or declared on the stack.  Make your
+// derived class' destructor private to ensure you can't be deleted directly or declared
+// on the stack.
+
+// utility class
+template< class T >
+class CAutoPtr
+{
+	T *m_pT;
+public:
+	CAutoPtr()
+	{
+		m_pT = NULL;
+	}
+
+	~CAutoPtr()
+	{
+		delete m_pT;
+	}
+
+	T *reset( T *p )
+	{
+		delete m_pT;
+		m_pT = p;
+		return m_pT;
+	}
+
+	T *TakeOwnership()
+	{
+		T *p = m_pT;
+		m_pT = NULL;
+		return p;
+	}
+
+	T *release( )
+	{
+		T *pT = m_pT;
+		m_pT = NULL;
+		return pT;
+	}
+
+	T *operator->()
+	{
+		return m_pT;
+	}
+
+	operator T*()
+	{
+		return m_pT;
+	}
+
+protected:
+	T *operator=(T*p)
+	{
+		AssertMsg( NULL == m_pT, "If this assert fires, you're leaking.\n" );
+		m_pT = p;
+		return m_pT;
+	}
+};
+
+class CRefCount 
+{
+public:
+	CRefCount() { m_cRef = 1; }		// we are born with a ref count of 1
+
+	// increment ref count
+	int AddRef() { return ThreadInterlockedIncrement( &m_cRef ); }
+
+	// delete ourselves when ref count reaches 0
+	int Release() 
+	{ 
+		Assert( m_cRef > 0 ); 
+		int cRef = ThreadInterlockedDecrement( &m_cRef ); 
+		if ( 0 == cRef )
+			DestroyThis();
+		return cRef;
+	}
+protected:
+	// Classes that derive from this should make their destructors private and virtual!
+	virtual ~CRefCount() { Assert( 0 == m_cRef ); }
+
+	virtual void DestroyThis() { delete this; }		// derived classes may override this if they want to be part of a mem pool
+
+	volatile int32 m_cRef;					// ref count of this object
+};
+
+#define SAFE_RELEASE( x )		if ( NULL != ( x ) ) { ( x )->Release(); x = NULL; }
+
+} // namespace GCSDK
+
+#include "tier0/memdbgoff.h"
+
+#endif // GCREFCOUNT_H