diff options
| author | Joe Ludwig <[email protected]> | 2013-06-26 15:22:04 -0700 |
|---|---|---|
| committer | Joe Ludwig <[email protected]> | 2013-06-26 15:22:04 -0700 |
| commit | 39ed87570bdb2f86969d4be821c94b722dc71179 (patch) | |
| tree | abc53757f75f40c80278e87650ea92808274aa59 /sp/src/game/shared/saverestore_utlmap.h | |
| download | source-sdk-2013-39ed87570bdb2f86969d4be821c94b722dc71179.tar.xz source-sdk-2013-39ed87570bdb2f86969d4be821c94b722dc71179.zip | |
First version of the SOurce SDK 2013
Diffstat (limited to 'sp/src/game/shared/saverestore_utlmap.h')
| -rw-r--r-- | sp/src/game/shared/saverestore_utlmap.h | 198 |
1 files changed, 198 insertions, 0 deletions
diff --git a/sp/src/game/shared/saverestore_utlmap.h b/sp/src/game/shared/saverestore_utlmap.h new file mode 100644 index 00000000..ab78e7fd --- /dev/null +++ b/sp/src/game/shared/saverestore_utlmap.h @@ -0,0 +1,198 @@ +//========= 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
|