92 lines
2.9 KiB
C++
92 lines
2.9 KiB
C++
//========= 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
|