be16f80abc
Any projects which want to use Hush code from now on will need to be licensed as GPLv3 or we will send the lawyers: https://www.softwarefreedom.org/ Notably, Komodo (KMD) is licensed as GPLv2 and is no longer compatible to receive code changes, without causing legal issues. MIT projects, such as Zcash, also cannot pull in changes from the Hush Full Node without permission from The Hush Developers, which may in some circumstances grant an MIT license on a case-by-case basis.
95 lines
3.2 KiB
C++
95 lines
3.2 KiB
C++
// Copyright (c) 2012-2014 The Bitcoin Core developers
|
|
// Distributed under the GPLv3 software license, see the accompanying
|
|
// file COPYING or https://www.gnu.org/licenses/gpl-3.0.en.html
|
|
|
|
#ifndef BITCOIN_LIMITEDMAP_H
|
|
#define BITCOIN_LIMITEDMAP_H
|
|
|
|
#include <assert.h>
|
|
#include <map>
|
|
|
|
/** STL-like map container that only keeps the N elements with the highest value. */
|
|
template <typename K, typename V>
|
|
class limitedmap
|
|
{
|
|
public:
|
|
typedef K key_type;
|
|
typedef V mapped_type;
|
|
typedef std::pair<const key_type, mapped_type> value_type;
|
|
typedef typename std::map<K, V>::const_iterator const_iterator;
|
|
typedef typename std::map<K, V>::size_type size_type;
|
|
|
|
protected:
|
|
std::map<K, V> map;
|
|
typedef typename std::map<K, V>::iterator iterator;
|
|
std::multimap<V, iterator> rmap;
|
|
typedef typename std::multimap<V, iterator>::iterator rmap_iterator;
|
|
size_type nMaxSize;
|
|
|
|
public:
|
|
limitedmap(size_type nMaxSizeIn = 0) { nMaxSize = nMaxSizeIn; }
|
|
const_iterator begin() const { return map.begin(); }
|
|
const_iterator end() const { return map.end(); }
|
|
size_type size() const { return map.size(); }
|
|
bool empty() const { return map.empty(); }
|
|
const_iterator find(const key_type& k) const { return map.find(k); }
|
|
size_type count(const key_type& k) const { return map.count(k); }
|
|
void insert(const value_type& x)
|
|
{
|
|
std::pair<iterator, bool> ret = map.insert(x);
|
|
if (ret.second) {
|
|
if (nMaxSize && map.size() == nMaxSize) {
|
|
map.erase(rmap.begin()->second);
|
|
rmap.erase(rmap.begin());
|
|
}
|
|
rmap.insert(make_pair(x.second, ret.first));
|
|
}
|
|
return;
|
|
}
|
|
void erase(const key_type& k)
|
|
{
|
|
iterator itTarget = map.find(k);
|
|
if (itTarget == map.end())
|
|
return;
|
|
std::pair<rmap_iterator, rmap_iterator> itPair = rmap.equal_range(itTarget->second);
|
|
for (rmap_iterator it = itPair.first; it != itPair.second; ++it)
|
|
if (it->second == itTarget) {
|
|
rmap.erase(it);
|
|
map.erase(itTarget);
|
|
return;
|
|
}
|
|
// Shouldn't ever get here
|
|
assert(0);
|
|
}
|
|
void update(const_iterator itIn, const mapped_type& v)
|
|
{
|
|
// TODO: When we switch to C++11, use map.erase(itIn, itIn) to get the non-const iterator.
|
|
iterator itTarget = map.find(itIn->first);
|
|
if (itTarget == map.end())
|
|
return;
|
|
std::pair<rmap_iterator, rmap_iterator> itPair = rmap.equal_range(itTarget->second);
|
|
for (rmap_iterator it = itPair.first; it != itPair.second; ++it)
|
|
if (it->second == itTarget) {
|
|
rmap.erase(it);
|
|
itTarget->second = v;
|
|
rmap.insert(make_pair(v, itTarget));
|
|
return;
|
|
}
|
|
// Shouldn't ever get here
|
|
assert(0);
|
|
}
|
|
size_type max_size() const { return nMaxSize; }
|
|
size_type max_size(size_type s)
|
|
{
|
|
if (s)
|
|
while (map.size() > s) {
|
|
map.erase(rmap.begin()->second);
|
|
rmap.erase(rmap.begin());
|
|
}
|
|
nMaxSize = s;
|
|
return nMaxSize;
|
|
}
|
|
};
|
|
|
|
#endif // BITCOIN_LIMITEDMAP_H
|