1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#ifndef CR_MAP_TYPE_IMPL_H
#define CR_MAP_TYPE_IMPL_H
#include "creg_cond.h"
#include <unordered_map>
#define SPRING_HASH_MAP std::unordered_map
#include <map>
#ifdef USING_CREG
#include <string>
#include <boost/shared_ptr.hpp>
namespace creg
{
// implement map insert() function with an interface common to all map types
template<typename T>
struct MapInserter
{
typedef typename T::iterator iterator;
typedef typename T::value_type value_type;
iterator insert(T& m, value_type& p) {
// std::map<>::insert returns std::pair<iterator, bool>.
return m.insert(p).first;
}
};
template<typename TKey, typename TValue>
struct MapInserter<std::multimap<TKey, TValue> >
{
typedef typename std::multimap<TKey, TValue>::iterator iterator;
typedef typename std::multimap<TKey, TValue>::value_type value_type;
iterator insert(std::multimap<TKey, TValue>& m, value_type& p) {
// std::multimap<>::insert returns iterator.
return m.insert(p);
}
};
// map/multimap, this template assumes that the map key type is copyable
template<typename T>
struct MapType : public IType
{
boost::shared_ptr<IType> keyType, mappedType;
typedef typename T::iterator iterator;
MapType(boost::shared_ptr<IType> keyType, boost::shared_ptr<IType> mappedType) :
keyType(keyType), mappedType(mappedType) {}
~MapType() {
}
void Serialize(ISerializer *s, void *instance)
{
T& ct = *(T*)instance;
if (s->IsWriting()) {
int size = ct.size();
s->SerializeInt(&size, sizeof(int));
for (iterator i = ct.begin(); i != ct.end(); ++i) {
keyType->Serialize(s, (void*) &i->first);
mappedType->Serialize(s, &i->second);
}
} else {
ct.clear();
int size;
s->SerializeInt(&size, sizeof(int));
for (int a = 0; a < size; a++) {
typename T::value_type pt;
// only allow copying of the key type
keyType->Serialize(s, (void*) &pt.first);
iterator i = MapInserter<T>().insert(ct, pt);
mappedType->Serialize(s, &i->second);
}
}
}
std::string GetName() const { return "map<" + keyType->GetName() + ", " + mappedType->GetName() + ">"; }
size_t GetSize() const { return sizeof(T); }
};
// Map type
template<typename TKey, typename TValue>
struct DeduceType<std::map<TKey, TValue> > {
static boost::shared_ptr<IType> Get() {
return boost::shared_ptr<IType>(new MapType<std::map<TKey, TValue> >(DeduceType<TKey>::Get(), DeduceType<TValue>::Get()));
}
};
// Multimap
template<typename TKey, typename TValue>
struct DeduceType<std::multimap<TKey, TValue> > {
static boost::shared_ptr<IType> Get() {
return boost::shared_ptr<IType>(new MapType<std::multimap<TKey, TValue> >(DeduceType<TKey>::Get(), DeduceType<TValue>::Get()));
}
};
// Hash map
template<typename TKey, typename TValue>
struct DeduceType<SPRING_HASH_MAP<TKey, TValue> > {
static boost::shared_ptr<IType> Get() {
return boost::shared_ptr<IType>(new MapType<SPRING_HASH_MAP<TKey, TValue> >(DeduceType<TKey>::Get(), DeduceType<TValue>::Get()));
}
};
template<typename T>
struct PairType : public IType
{
PairType(boost::shared_ptr<IType> first, boost::shared_ptr<IType> second):
firstType(first), secondType(second) {}
~PairType() {
}
boost::shared_ptr<IType> firstType, secondType;
void Serialize(ISerializer *s, void *instance)
{
T& p = *(T*)instance;
firstType->Serialize(s, &p.first);
secondType->Serialize(s, &p.second);
}
std::string GetName() const { return "pair<" + firstType->GetName() + "," + secondType->GetName() + ">"; }
size_t GetSize() const { return sizeof(T); }
};
// std::pair
template<typename TFirst, typename TSecond>
struct DeduceType<std::pair<TFirst, TSecond> >
{
static boost::shared_ptr<IType> Get() {
return boost::shared_ptr<IType>(new PairType<std::pair<TFirst, TSecond> >(DeduceType<TFirst>::Get(), DeduceType<TSecond>::Get()));
}
};
}
#endif // USING_CREG
#endif // CR_MAP_TYPE_IMPL_H
|
