1 files changed, 95 insertions, 0 deletions
diff --git a/game/shared/saverestore_bitstring.h b/game/shared/saverestore_bitstring.h
new file mode 100644
index 0000000..e7a6c80
--- /dev/null
+++ b/game/shared/saverestore_bitstring.h
@@ -0,0 +1,95 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose:
+//
+// $NoKeywords: $
+//=============================================================================//
+
+#ifndef SAVERESTORE_BITSTRING_H
+#define SAVERESTORE_BITSTRING_H
+
+#include "isaverestore.h"
+
+#if defined( _WIN32 )
+#pragma once
+#endif
+
+//-------------------------------------
+
+template <class BITSTRING>
+class CVarBitVecSaveRestoreOps : public CDefSaveRestoreOps
+{
+public:
+	CVarBitVecSaveRestoreOps()
+	{
+	}
+
+	// save data type interface
+	virtual void Save( const SaveRestoreFieldInfo_t &fieldInfo, ISave *pSave )
+	{
+		BITSTRING *pBitString = (BITSTRING *)fieldInfo.pField;
+		int numBits = pBitString->GetNumBits();
+		pSave->WriteInt( &numBits );
+		pSave->WriteInt( pBitString->Base(), pBitString->GetNumDWords() );
+	}
+	
+	virtual void Restore( const SaveRestoreFieldInfo_t &fieldInfo, IRestore *pRestore )
+	{
+		BITSTRING *pBitString = (BITSTRING *)fieldInfo.pField;
+		int numBits = pRestore->ReadInt();
+		if ( !pBitString->IsFixedSize() )
+			pBitString->Resize( numBits );
+		else
+		{
+			Assert( pBitString->GetNumBits() >= numBits );
+			pBitString->ClearAll();
+		}
+		int numIntsInStream = CalcNumIntsForBits( numBits );
+		int readSize = MIN( pBitString->GetNumDWords(), numIntsInStream );
+		pRestore->ReadInt( pBitString->Base(), numIntsInStream );
+
+		numIntsInStream -= readSize;
+		while ( numIntsInStream-- > 0 )
+		{
+			int ignored;
+			pRestore->ReadInt( &ignored, 1 );
+		}
+	}
+	
+	virtual void MakeEmpty( const SaveRestoreFieldInfo_t &fieldInfo )
+	{
+		BITSTRING *pBitString = (BITSTRING *)fieldInfo.pField;
+		pBitString->ClearAll();
+	}
+
+	virtual bool IsEmpty( const SaveRestoreFieldInfo_t &fieldInfo )
+	{
+		BITSTRING *pBitString = (BITSTRING *)fieldInfo.pField;
+		return pBitString->IsAllClear();
+	}
+};
+
+//-------------------------------------
+
+template <class BITSTRING>
+ISaveRestoreOps *GetBitstringDataOps(BITSTRING *)
+{
+	static CVarBitVecSaveRestoreOps<BITSTRING> ops;
+	return &ops;
+}
+
+//-------------------------------------
+
+#define SaveBitString( pSave, pBitString, fieldtype) \
+	CDataopsInstantiator<fieldtype>::GetDataOps( pBitString )->Save( pBitString, pSave );
+
+#define RestoreBitString( pRestore, pBitString, fieldtype) \
+	CDataopsInstantiator<fieldtype>::GetDataOps( pBitString )->Restore( pBitString, pRestore );
+
+//-------------------------------------
+
+#define DEFINE_BITSTRING(name) \
+	{ FIELD_CUSTOM, #name, { offsetof(classNameTypedef,name), 0 }, 1, FTYPEDESC_SAVE, NULL, GetBitstringDataOps(&(((classNameTypedef *)0)->name)), NULL }
+
+#endif // SAVERESTORE_BITSTRING_H
+