Add support for unique_ptr and shared_ptr to memusage

(cherry picked from commit bitcoin/bitcoin@1b9e6d3c1a)
This commit is contained in:
Pieter Wuille 2016-05-30 16:50:14 +02:00 committed by Jack Grigg
parent 48ead0bea1
commit e863fdaebb
1 changed files with 24 additions and 0 deletions

View File

@ -69,6 +69,15 @@ private:
X x;
};
struct stl_shared_counter
{
/* Various platforms use different sized counters here.
* Conservatively assume that they won't be larger than size_t. */
void* class_type;
size_t use_count;
size_t weak_count;
};
template<typename X>
static inline size_t DynamicUsage(const std::vector<X>& v)
{
@ -93,6 +102,21 @@ static inline size_t DynamicUsage(const std::map<X, Y, C>& m)
return MallocUsage(sizeof(stl_tree_node<std::pair<const X, Y> >)) * m.size();
}
template<typename X>
static inline size_t DynamicUsage(const std::unique_ptr<X>& p)
{
return p ? MallocUsage(sizeof(X)) : 0;
}
template<typename X>
static inline size_t DynamicUsage(const std::shared_ptr<X>& p)
{
// A shared_ptr can either use a single continuous memory block for both
// the counter and the storage (when using std::make_shared), or separate.
// We can't observe the difference, however, so assume the worst.
return p ? MallocUsage(sizeof(X)) + MallocUsage(sizeof(stl_shared_counter)) : 0;
}
// Boost data structures
template<typename X>