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
|
#include "CodegenInput.hpp"
#include <Macros.hpp>
#include <Utils.hpp>
#include <robin_hood.h>
#include <variant>
struct SomeDecl {
std::variant<DeclStruct, DeclFunction, DeclEnum> v;
};
class CodegenInput::Private {
public:
// We want address stability for everything
robin_hood::unordered_node_map<std::string, SomeDecl, StringHash, StringEqual> decls;
robin_hood::unordered_node_map<std::string, DeclNamespace, StringHash, StringEqual> namespaces;
};
CodegenInput::CodegenInput()
: m{ new Private() } //
{
}
CodegenInput::~CodegenInput() {
delete m;
}
#define STORE_DECL_OF_TYPE(DeclType, fullname, decl) \
auto [iter, success] = m->decls.try_emplace(std::move(fullname), SomeDecl{ .v = std::move(decl) }); \
auto& key = iter->first; \
auto& val = iter->second; \
auto& declRef = std::get<DeclType>(val.v); \
declRef.fullname = key; \
return &declRef
DeclEnum* CodegenInput::AddEnum(std::string fullname, DeclEnum decl) {
#if CODEGEN_DEBUG_PRINT
printf("Committed enum '%s'\n", decl.name.c_str());
for (auto& elm : decl.elements) {
printf(" - element %s = %" PRId64 "\n", elm.name.c_str(), elm.value);
}
#endif
STORE_DECL_OF_TYPE(DeclEnum, fullname, decl);
}
DeclStruct* CodegenInput::AddStruct(std::string fullname, DeclStruct decl) {
#if CODEGEN_DEBUG_PRINT
printf("Committed struct '%s'\n", decl.name.c_str());
printf(" Base classes:\n");
for (auto& base : decl.baseClasses) {
printf(" - %.*s\n", PRINTF_STRING_VIEW(base->name));
}
#endif
STORE_DECL_OF_TYPE(DeclStruct, fullname, decl);
}
#define FIND_DECL_OF_TYPE(DeclType) \
auto iter = m->decls.find(name); \
if (iter != m->decls.end()) { \
auto& some = iter->second.v; \
if (auto decl = std::get_if<DeclType>(&some)) { \
return decl; \
} \
} \
return nullptr
const DeclEnum* CodegenInput::FindEnum(std::string_view name) const {
FIND_DECL_OF_TYPE(DeclEnum);
}
const DeclStruct* CodegenInput::FindStruct(std::string_view name) const {
FIND_DECL_OF_TYPE(DeclStruct);
}
DeclNamespace* CodegenInput::AddNamespace(DeclNamespace ns) {
auto path = Utils::MakeFullName(""sv, &ns);
auto [iter, success] = m->namespaces.try_emplace(std::move(path), std::move(ns));
auto& nsRef = iter->second;
if (success) {
nsRef.fullname = iter->first;
}
return &nsRef;
}
const DeclNamespace* CodegenInput::FindNamespace(std::string_view fullname) const {
auto iter = m->namespaces.find(fullname);
if (iter != m->namespaces.end()) {
return &iter->second;
} else {
return nullptr;
}
}
DeclNamespace* CodegenInput::FindNamespace(std::string_view name) {
return const_cast<DeclNamespace*>(const_cast<const CodegenInput*>(this)->FindNamespace(name));
}
|
