Add support for std::unordered_{map,set} to memusage.h

This commit is contained in:
Pieter Wuille 2017-04-21 05:26:23 -07:00
parent 27faa6cccd
commit 344a2c4122

View File

@ -12,6 +12,8 @@
#include <map> #include <map>
#include <set> #include <set>
#include <vector> #include <vector>
#include <unordered_map>
#include <unordered_set>
#include <boost/foreach.hpp> #include <boost/foreach.hpp>
#include <boost/unordered_set.hpp> #include <boost/unordered_set.hpp>
@ -149,7 +151,7 @@ static inline size_t DynamicUsage(const std::shared_ptr<X>& p)
// Boost data structures // Boost data structures
template<typename X> template<typename X>
struct boost_unordered_node : private X struct unordered_node : private X
{ {
private: private:
void* ptr; void* ptr;
@ -158,13 +160,25 @@ private:
template<typename X, typename Y> template<typename X, typename Y>
static inline size_t DynamicUsage(const boost::unordered_set<X, Y>& s) static inline size_t DynamicUsage(const boost::unordered_set<X, Y>& s)
{ {
return MallocUsage(sizeof(boost_unordered_node<X>)) * s.size() + MallocUsage(sizeof(void*) * s.bucket_count()); return MallocUsage(sizeof(unordered_node<X>)) * s.size() + MallocUsage(sizeof(void*) * s.bucket_count());
} }
template<typename X, typename Y, typename Z> template<typename X, typename Y, typename Z>
static inline size_t DynamicUsage(const boost::unordered_map<X, Y, Z>& m) static inline size_t DynamicUsage(const boost::unordered_map<X, Y, Z>& m)
{ {
return MallocUsage(sizeof(boost_unordered_node<std::pair<const X, Y> >)) * m.size() + MallocUsage(sizeof(void*) * m.bucket_count()); return MallocUsage(sizeof(unordered_node<std::pair<const X, Y> >)) * m.size() + MallocUsage(sizeof(void*) * m.bucket_count());
}
template<typename X, typename Y>
static inline size_t DynamicUsage(const std::unordered_set<X, Y>& s)
{
return MallocUsage(sizeof(unordered_node<X>)) * s.size() + MallocUsage(sizeof(void*) * s.bucket_count());
}
template<typename X, typename Y, typename Z>
static inline size_t DynamicUsage(const std::unordered_map<X, Y, Z>& m)
{
return MallocUsage(sizeof(unordered_node<std::pair<const X, Y> >)) * m.size() + MallocUsage(sizeof(void*) * m.bucket_count());
} }
} }