199 lines
4.1 KiB
C++
199 lines
4.1 KiB
C++
//========= Copyright Valve Corporation, All rights reserved. ============//
|
|
//
|
|
// Purpose:
|
|
//
|
|
//=============================================================================//
|
|
|
|
#ifndef SAVERESTORE_UTLMAP_H
|
|
#define SAVERESTORE_UTLMAP_H
|
|
|
|
#include "utlmap.h"
|
|
#include "saverestore_utlrbtree.h"
|
|
|
|
#if defined( _WIN32 )
|
|
#pragma once
|
|
#endif
|
|
|
|
template <class UTLMAP, int KEY_TYPE, int FIELD_TYPE>
|
|
class CUtlMapDataOps : public CDefSaveRestoreOps
|
|
{
|
|
public:
|
|
CUtlMapDataOps()
|
|
{
|
|
UTLCLASS_SAVERESTORE_VALIDATE_TYPE( KEY_TYPE );
|
|
UTLCLASS_SAVERESTORE_VALIDATE_TYPE( FIELD_TYPE );
|
|
}
|
|
|
|
virtual void Save( const SaveRestoreFieldInfo_t &fieldInfo, ISave *pSave )
|
|
{
|
|
datamap_t *pKeyDatamap = CTypedescDeducer<KEY_TYPE>::Deduce( (UTLMAP *)NULL );
|
|
datamap_t *pFieldDatamap = CTypedescDeducer<FIELD_TYPE>::Deduce( (UTLMAP *)NULL );
|
|
typedescription_t dataDesc[] =
|
|
{
|
|
{
|
|
(fieldtype_t)KEY_TYPE,
|
|
"K",
|
|
{ 0, 0 },
|
|
1,
|
|
FTYPEDESC_SAVE,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
pKeyDatamap,
|
|
sizeof(KEY_TYPE),
|
|
},
|
|
|
|
{
|
|
(fieldtype_t)FIELD_TYPE,
|
|
"T",
|
|
{ offsetof(typename UTLMAP::Node_t, elem), 0 },
|
|
1,
|
|
FTYPEDESC_SAVE,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
pFieldDatamap,
|
|
sizeof(FIELD_TYPE),
|
|
}
|
|
};
|
|
|
|
datamap_t dataMap =
|
|
{
|
|
dataDesc,
|
|
2,
|
|
"um",
|
|
NULL,
|
|
false,
|
|
false,
|
|
0,
|
|
#ifdef _DEBUG
|
|
true
|
|
#endif
|
|
};
|
|
|
|
typename UTLMAP::CTree *pUtlRBTree = ((UTLMAP *)fieldInfo.pField)->AccessTree();
|
|
|
|
pSave->StartBlock();
|
|
|
|
int nElems = pUtlRBTree->Count();
|
|
pSave->WriteInt( &nElems, 1 );
|
|
|
|
typename UTLMAP::CTree::IndexType_t i = pUtlRBTree->FirstInorder();
|
|
while ( i != pUtlRBTree->InvalidIndex() )
|
|
{
|
|
typename UTLMAP::CTree::ElemType_t &elem = pUtlRBTree->Element( i );
|
|
|
|
pSave->WriteAll( &elem, &dataMap );
|
|
|
|
i = pUtlRBTree->NextInorder( i );
|
|
}
|
|
pSave->EndBlock();
|
|
}
|
|
|
|
virtual void Restore( const SaveRestoreFieldInfo_t &fieldInfo, IRestore *pRestore )
|
|
{
|
|
datamap_t *pKeyDatamap = CTypedescDeducer<KEY_TYPE>::Deduce( (UTLMAP *)NULL );
|
|
datamap_t *pFieldDatamap = CTypedescDeducer<FIELD_TYPE>::Deduce( (UTLMAP *)NULL );
|
|
typedescription_t dataDesc[] =
|
|
{
|
|
{
|
|
(fieldtype_t)KEY_TYPE,
|
|
"K",
|
|
{ 0, 0 },
|
|
1,
|
|
FTYPEDESC_SAVE,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
pKeyDatamap,
|
|
sizeof(KEY_TYPE),
|
|
},
|
|
|
|
{
|
|
(fieldtype_t)FIELD_TYPE,
|
|
"T",
|
|
{ offsetof(typename UTLMAP::Node_t, elem), 0 },
|
|
1,
|
|
FTYPEDESC_SAVE,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
pFieldDatamap,
|
|
sizeof(FIELD_TYPE),
|
|
}
|
|
};
|
|
|
|
datamap_t dataMap =
|
|
{
|
|
dataDesc,
|
|
2,
|
|
"um",
|
|
NULL,
|
|
false,
|
|
false,
|
|
0,
|
|
#ifdef _DEBUG
|
|
true
|
|
#endif
|
|
};
|
|
|
|
UTLMAP *pUtlMap = ((UTLMAP *)fieldInfo.pField);
|
|
|
|
pRestore->StartBlock();
|
|
|
|
int nElems = pRestore->ReadInt();
|
|
typename UTLMAP::CTree::ElemType_t temp;
|
|
|
|
while ( nElems-- )
|
|
{
|
|
pRestore->ReadAll( &temp, &dataMap );
|
|
pUtlMap->Insert( temp.key, temp.elem );
|
|
}
|
|
|
|
pRestore->EndBlock();
|
|
}
|
|
|
|
virtual void MakeEmpty( const SaveRestoreFieldInfo_t &fieldInfo )
|
|
{
|
|
UTLMAP *pUtlMap = (UTLMAP *)fieldInfo.pField;
|
|
pUtlMap->RemoveAll();
|
|
}
|
|
|
|
virtual bool IsEmpty( const SaveRestoreFieldInfo_t &fieldInfo )
|
|
{
|
|
UTLMAP *pUtlMap = (UTLMAP *)fieldInfo.pField;
|
|
return ( pUtlMap->Count() == 0 );
|
|
}
|
|
|
|
};
|
|
|
|
//-------------------------------------
|
|
|
|
template <int KEYTYPE, int FIELD_TYPE>
|
|
class CUtlMapDataopsInstantiator
|
|
{
|
|
public:
|
|
template <class UTLMAP>
|
|
static ISaveRestoreOps *GetDataOps(UTLMAP *)
|
|
{
|
|
static CUtlMapDataOps<UTLMAP, KEYTYPE, FIELD_TYPE> ops;
|
|
return &ops;
|
|
}
|
|
};
|
|
|
|
//-------------------------------------
|
|
|
|
#define SaveUtlMap( pSave, pUtlMap, fieldtype) \
|
|
CUtlMapDataopsInstantiator<fieldtype>::GetDataOps( pUtlMap )->Save( pUtlMap, pSave );
|
|
|
|
#define RestoreUtlMap( pRestore, pUtlMap, fieldtype) \
|
|
CUtlMapDataopsInstantiator<fieldtype>::GetDataOps( pUtlMap )->Restore( pUtlMap, pRestore );
|
|
|
|
//-------------------------------------
|
|
|
|
#define DEFINE_UTLMAP(name,keyType,fieldtype) \
|
|
{ FIELD_CUSTOM, #name, { offsetof(classNameTypedef,name), 0 }, 1, FTYPEDESC_SAVE, NULL, CUtlMapDataopsInstantiator<keyType, fieldtype>::GetDataOps(&(((classNameTypedef *)0)->name)), NULL }
|
|
|
|
|
|
#endif // SAVERESTORE_UTLMAP_H
|