serialize_map.h

// Copyright 2009-2022 NTESS. Under the terms
// of Contract DE-NA0003525 with NTESS, the U.S.
// Government retains certain rights in this software.
//
// Copyright (c) 2009-2022, NTESS
// All rights reserved.
//
// This file is part of the SST software package. For license
// information, see the LICENSE file in the top level directory of the
// distribution.

#ifndef SST_CORE_SERIALIZATION_SERIALIZE_MAP_H
#define SST_CORE_SERIALIZATION_SERIALIZE_MAP_H

#include "sst/core/serialization/serializer.h"

#include <map>
#include <unordered_map>

namespace SST {
namespace Core {
namespace Serialization {

template <class Key, class Value>
class serialize<std::map<Key, Value>>
{
    typedef std::map<Key, Value> Map;

public:
    void operator()(Map& m, serializer& ser)
    {
        typedef typename std::map<Key, Value>::iterator iterator;
        switch ( ser.mode() ) {

        case serializer::SIZER:
        {
            size_t size = m.size();
            ser.size(size);
            iterator it, end = m.end();
            for ( it = m.begin(); it != end; ++it ) {
                // keys are const values - annoyingly
                serialize<Key>()(const_cast<Key&>(it->first), ser);
                serialize<Value>()(it->second, ser);
            }
            break;
        }
        case serializer::PACK:
        {
            size_t size = m.size();
            ser.pack(size);
            iterator it, end = m.end();
            for ( it = m.begin(); it != end; ++it ) {
                serialize<Key>()(const_cast<Key&>(it->first), ser);
                serialize<Value>()(it->second, ser);
            }
            break;
        }
        case serializer::UNPACK:
        {
            size_t size;
            ser.unpack(size);
            for ( size_t i = 0; i < size; ++i ) {
                Key   k = {};
                Value v = {};
                serialize<Key>()(k, ser);
                serialize<Value>()(v, ser);
                m[k] = v;
            }
            break;
        }
        }
    }
};

template <class Key, class Value>
class serialize<std::unordered_map<Key, Value>>
{
    typedef std::unordered_map<Key, Value> Map;

public:
    void operator()(Map& m, serializer& ser)
    {
        typedef typename std::unordered_map<Key, Value>::iterator iterator;
        switch ( ser.mode() ) {

        case serializer::SIZER:
        {
            size_t size = m.size();
            ser.size(size);
            iterator it, end = m.end();
            for ( it = m.begin(); it != end; ++it ) {
                // keys are const values - annoyingly
                serialize<Key>()(const_cast<Key&>(it->first), ser);
                serialize<Value>()(it->second, ser);
            }
            break;
        }
        case serializer::PACK:
        {
            size_t size = m.size();
            ser.pack(size);
            iterator it, end = m.end();
            for ( it = m.begin(); it != end; ++it ) {
                serialize<Key>()(const_cast<Key&>(it->first), ser);
                serialize<Value>()(it->second, ser);
            }
            break;
        }
        case serializer::UNPACK:
        {
            size_t size;
            ser.unpack(size);
            for ( size_t i = 0; i < size; ++i ) {
                Key   k = {};
                Value v = {};
                serialize<Key>()(k, ser);
                serialize<Value>()(v, ser);
                m[k] = v;
            }
            break;
        }
        }
    }
};

} // namespace Serialization
} // namespace Core
} // namespace SST

#endif // SST_CORE_SERIALIZATION_SERIALIZE_MAP_H