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
|
#ifndef DB_HPP
#define DB_HPP
// db.hpp - convenience wrappers over std::map<K, V>
//
// Copyright © 2013 Ben Longbons <b.r.longbons@gmail.com>
//
// This file is part of The Mana World (Athena server)
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
# include "sanity.hpp"
# include <map>
template<class K, class V>
class Map
{
typedef std::map<K, V> Impl;
Impl impl;
public:
Map() = default;
Map(std::initializer_list<std::pair<const K, V>> il)
: impl(il)
{}
typedef typename Impl::iterator iterator;
typedef typename Impl::const_iterator const_iterator;
iterator begin() { return impl.begin(); }
iterator end() { return impl.end(); }
const_iterator begin() const { return impl.begin(); }
const_iterator end() const { return impl.end(); }
V *search(const K& k)
{
iterator it = impl.find(k);
if (it == impl.end())
return nullptr;
return &it->second;
}
const V *search(const K& k) const
{
const_iterator it = impl.find(k);
if (it == impl.end())
return nullptr;
return &it->second;
}
void insert(K k, V v)
{
// As far as I can tell, this is the simplest way to
// implement move-only insert-with-replacement.
iterator it = impl.lower_bound(k);
// invariant: if it is valid, it->first >= k
if (it != impl.end() && it->first == k)
it->second = std::move(v);
else
it = impl.insert(std::pair<K, V>(std::move(k), std::move(v))).first;
return (void)&it->second;
}
V *init(K k)
{
return &impl[k];
}
void erase(const K& k)
{
impl.erase(k);
}
void clear()
{
impl.clear();
}
bool empty()
{
return impl.empty();
}
};
template<class K, class V>
class DMap
{
typedef Map<K, V> Impl;
Impl impl;
public:
typedef typename Impl::iterator iterator;
typedef typename Impl::const_iterator const_iterator;
iterator begin() { return impl.begin(); }
iterator end() { return impl.end(); }
const_iterator begin() const { return impl.begin(); }
const_iterator end() const { return impl.end(); }
V get(K k)
{
V *vp = impl.search(k);
return vp ? *vp : V();
}
void put(K k, V v)
{
if (v == V())
impl.erase(k);
else
impl.insert(k, v);
}
void clear()
{
impl.clear();
}
};
#endif // DB_HPP
|
