blob: e7a62a2691b858deefdfb28d121844a5a3cc959c (
plain) (
blame)
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
|
#pragma once
#include <cstddef>
#include <unordered_set>
class ObjectVisitor;
class Object
{
public:
virtual ~Object() {}
virtual void VisitReferences(ObjectVisitor* visitor, void* state) = 0;
};
class ObjectVisitor
{
public:
virtual ~ObjectVisitor() {}
// Object** to allows the reference to be moved
virtual void VisitReference(Object* from, Object** to, void* state) = 0;
};
class GarbageCollector : public ObjectVisitor
{
public:
virtual void* Allocate(size_t size) = 0;
// Object** to allows the root to be moved
virtual void SetRoot(Object** root) = 0;
virtual void Shutdown() = 0;
};
class DestroyVisitor : public ObjectVisitor
{
public:
virtual void Destroy(Object* o)
{
auto insert_result = m_Visited.insert(o);
if (insert_result.second)
{
o->VisitReferences(this, nullptr);
o->~Object();
}
}
void VisitReference(Object*, Object** to, void*) override
{
if (*to)
{
Destroy(*to);
}
}
std::unordered_set<Object*> m_Visited;
};
std::unique_ptr<GarbageCollector> CreateGarbageCollector(int argc,
char* argv[]);
|
