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
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
/* 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"
#ifdef USING_CREG
#ifdef _MSC_VER
#define SPRING_HASH_MAP stdext::hash_map
#include <hash_map>
#elif __GNUG__
/* Test for GCC >= 4.3.2 */
#if __GNUC__ > 4 || \
(__GNUC__ == 4 && (__GNUC_MINOR__ > 3 || \
(__GNUC_MINOR__ == 3 && \
__GNUC_PATCHLEVEL__ >= 2)))
#include <tr1/unordered_map>
#define SPRING_HASH_MAP std::tr1::unordered_map
#else
#define SPRING_HASH_MAP __gnu_cxx::hash_map
#include <ext/hash_map>
#endif
#else
#error Unsupported compiler
#endif
#include <string>
#include <map>
#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 {
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() { return "map<" + keyType->GetName() + ", " + mappedType->GetName(); }
};
// Map type
template<typename TKey, typename TValue>
struct DeduceType<std::map<TKey, TValue> > {
boost::shared_ptr<IType> Get() {
DeduceType<TValue> valuetype;
DeduceType<TKey> keytype;
return boost::shared_ptr<IType>(new MapType<std::map<TKey, TValue> >(keytype.Get(), valuetype.Get()));
}
};
// Multimap
template<typename TKey, typename TValue>
struct DeduceType<std::multimap<TKey, TValue> > {
boost::shared_ptr<IType> Get() {
DeduceType<TValue> valuetype;
DeduceType<TKey> keytype;
return boost::shared_ptr<IType>(new MapType<std::multimap<TKey, TValue> >(keytype.Get(), valuetype.Get()));
}
};
// Hash map
template<typename TKey, typename TValue>
struct DeduceType<SPRING_HASH_MAP<TKey, TValue> > {
boost::shared_ptr<IType> Get() {
DeduceType<TValue> valuetype;
DeduceType<TKey> keytype;
return boost::shared_ptr<IType>(new MapType<SPRING_HASH_MAP<TKey, TValue> >(keytype.Get(), valuetype.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() { return "pair<" + firstType->GetName() + "," + secondType->GetName() + ">"; }
};
// std::pair
template<typename TFirst, typename TSecond>
struct DeduceType<std::pair<TFirst, TSecond> >
{
boost::shared_ptr<IType> Get() {
DeduceType<TFirst> first;
DeduceType<TSecond> second;
return boost::shared_ptr<IType>(new PairType<std::pair<TFirst, TSecond> >(first.Get(), second.Get()));
}
};
};
#endif // USING_CREG
#endif // CR_MAP_TYPE_IMPL_H
|
