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.
 
 
 
 
 
 

95 lines
2.5 KiB

//========= 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