//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
//=============================================================================//

#ifndef SAVERESTORE_UTLMAP_H
#define SAVERESTORE_UTLMAP_H

#include "saverestore_utlrbtree.h"
#include "utlmap.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