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
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#ifndef dom_canvas_DmdStdContainers_h
#define dom_canvas_DmdStdContainers_h
#include "mozilla/MemoryReporting.h"
#include "mozilla/layers/BuildConstants.h"
#include <unordered_map>
#include <unordered_set>
namespace mozilla::webgl {
// -
template <class T>
class dmd_allocator {
public:
using value_type = T;
private:
size_t mMallocSize = 0;
public:
dmd_allocator() = default;
// -
template <class U>
friend class dmd_allocator;
template <class U>
explicit dmd_allocator(const dmd_allocator<U>& rhs) {
if constexpr (kIsDmd) {
mMallocSize = rhs.mMallocSize;
}
}
// -
value_type* allocate(const size_t n) {
const auto p = std::allocator<value_type>{}.allocate(n);
if constexpr (kIsDmd) {
mMallocSize += moz_malloc_size_of(p);
}
return p;
}
void deallocate(value_type* const p, const size_t n) {
if constexpr (kIsDmd) {
mMallocSize -= moz_malloc_size_of(p);
}
std::allocator<value_type>{}.deallocate(p, n);
}
// -
size_t SizeOfExcludingThis(mozilla::MallocSizeOf) const {
return mMallocSize;
}
};
// -
template <class Key, class T, class Hash = std::hash<Key>,
class KeyEqual = std::equal_to<Key>,
class Allocator = dmd_allocator<std::pair<const Key, T>>,
class _StdT = std::unordered_map<Key, T, Hash, KeyEqual, Allocator>>
class dmd_unordered_map : public _StdT {
public:
using StdT = _StdT;
size_t SizeOfExcludingThis(mozilla::MallocSizeOf mso) const {
const auto& a = StdT::get_allocator();
return a.SizeOfExcludingThis(mso);
}
};
// -
template <class Key, class Hash = std::hash<Key>,
class KeyEqual = std::equal_to<Key>,
class Allocator = dmd_allocator<Key>,
class _StdT = std::unordered_set<Key, Hash, KeyEqual, Allocator>>
class dmd_unordered_set : public _StdT {
public:
using StdT = _StdT;
size_t SizeOfExcludingThis(mozilla::MallocSizeOf mso) const {
const auto& a = StdT::get_allocator();
return a.SizeOfExcludingThis(mso);
}
};
// -
} // namespace mozilla::webgl
#endif // dom_canvas_DmdStdContainers_h
|
