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
|
#ifndef OBJECTGRAPH_H
#define OBJECTGRAPH_H
#include <vector>
#include <unordered_map>
class ObjectNode {
public:
ObjectNode()
: children(), objectId(nullptr), classId(nullptr), visited(false)
{
}
std::vector<ObjectNode*> children;
void *objectId;
void *classId;
bool visited;
void print(size_t gcCounter, FILE *out) {
// To make things more compact, if the node was already visited, don't
// traverse its children. It's enough to note that it is there.
if (visited) {
fprintf(out, "e %zx %zx %zx %zx\n", gcCounter, 0,
reinterpret_cast<uintptr_t>(objectId),
reinterpret_cast<uintptr_t>(classId));
return;
}
visited = true;
fprintf(out, "e %zx %zx %zx %zx\n", gcCounter, children.size(),
reinterpret_cast<uintptr_t>(objectId),
reinterpret_cast<uintptr_t>(classId));
for (ObjectNode* child: children) {
child->print(gcCounter, out);
}
}
};
class ObjectGraph {
public:
ObjectGraph() {}
void addRoot(void* rootObjectID, void *rootClassID) {
index(nullptr, nullptr, rootObjectID, rootClassID);
}
void index(void *keyObjectID, void *keyClassID, void *valObjectID, void *valClassID) {
auto keyIt = m_graph.find(keyObjectID);
if (keyIt == m_graph.end()) {
ObjectNode node;
node.objectId = keyObjectID;
node.classId = keyClassID;
keyIt = m_graph.insert(std::make_pair(keyObjectID, node)).first;
}
auto valIt = m_graph.find(valObjectID);
if (valIt == m_graph.end()) {
ObjectNode node;
node.objectId = valObjectID;
node.classId = valClassID;
valIt = m_graph.insert(std::make_pair(valObjectID, node)).first;
}
keyIt->second.children.push_back(&(valIt->second));
}
void print(size_t gcCounter, FILE* out) {
auto it = m_graph.find(nullptr);
if (it == m_graph.end()) {
return;
}
ObjectNode *node = &(it->second);
node->print(gcCounter, out);
clear();
}
void clear() {
m_graph.clear();
}
/* m_graph is in fact a flat collection of ObjectNodes. The graph structure
is provided by ObjectNode::children which point to ObjectNode instances
living inside m_graph. All instances of ObjectNode are residing
in m_graph, regardless of their position in the actual graph.
NULL key is somewhat special - the ObjectNode corresponding to it is a fake
ObjectNode whose children are GC roots. */
static std::unordered_map<void*, ObjectNode> m_graph;
};
#endif // OBJECTGRAPH_H
|