serialize_map.h

Go to the documentation of this file.

#ifndef SERIALIZE_MAP_H
#define SERIALIZE_MAP_H

#include <map>
#include <sprockit/unordered.h>
#include <sprockit/serializer.h>

namespace sprockit {

namespace pvt {

template <class Map, class Key, class Value>
void
serialize_map(Map& m, serializer& ser)
{
  typedef typename Map::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 (int i=0; i < size; ++i){
      Key k;
      Value v;
      serialize<Key>()(k, ser);
      serialize<Value>()(v, ser);
      m[k] = v;
    }
    break;
  }
  }
}

} //end ns private

template <class Key, class Value>
class serialize<std::map<Key,Value> > {
  typedef std::map<Key,Value> Map;
 public:
  void operator()(Map& m, serializer& ser){
    pvt::serialize_map<Map,Key,Value>(m,ser);
  }
};

#if !SPKT_ENABLE_ORDERED_MAP
template <class Key, class Value>
class serialize<spkt_unordered_map<Key,Value> > {
  typedef spkt_unordered_map<Key,Value> Map;
 public:
  void operator()(Map& m, serializer& ser){
    pvt::serialize_map<Map,Key,Value>(m,ser);
  }
};
#endif

}

#endif // SERIALIZE_MAP_H