blob: 1de6487eb23bb8b06cdb85f17fa97b7b9eb1e09b (
plain)
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
// SPDX-License-Identifier: GPL-3.0-or-later
#include "rcc/CodeGen/CompilationUnit.h"
#include "rcc/CodeGen/MachineFunction.h"
#include <cstddef>
#include <memory>
#include <string>
#include <string_view>
#include <unordered_set>
#include <utility>
namespace rcc {
CompilationUnit::CompilationUnit() = default;
CompilationUnit::CompilationUnit(CompilationUnit &&) noexcept = default;
CompilationUnit &CompilationUnit::operator=(CompilationUnit &&) noexcept = default;
CompilationUnit::~CompilationUnit() = default;
Expected<MachineFunction *> CompilationUnit::createMachineFunction(std::string Name) {
if (Name.empty())
return makeUnexpected(ErrorCode::InvalidArgument, "machine function name cannot be empty");
if (FunctionNameToIndex.contains(Name)) {
return makeUnexpected(ErrorCode::InvalidArgument,
"machine function name is already present in the compilation unit");
}
auto StableID = FunctionIDGenerator.allocate();
if (!StableID)
return std::unexpected(StableID.error());
auto Function = MachineFunction::create(std::move(Name), *StableID);
if (!Function)
return std::unexpected(Function.error());
MachineFunction *Result = Function->get();
const std::size_t Index = Functions.size();
FunctionNameToIndex.emplace(std::string(Result->name()), Index);
Functions.push_back(std::move(*Function));
return Result;
}
MachineFunction *CompilationUnit::functionAt(std::size_t Index) noexcept {
if (Index >= Functions.size())
return nullptr;
return Functions[Index].get();
}
const MachineFunction *CompilationUnit::functionAt(std::size_t Index) const noexcept {
if (Index >= Functions.size())
return nullptr;
return Functions[Index].get();
}
MachineFunction *CompilationUnit::function(std::string_view Name) noexcept {
const auto Position = FunctionNameToIndex.find(Name);
if (Position == FunctionNameToIndex.end())
return nullptr;
return Functions[Position->second].get();
}
const MachineFunction *CompilationUnit::function(std::string_view Name) const noexcept {
const auto Position = FunctionNameToIndex.find(Name);
if (Position == FunctionNameToIndex.end())
return nullptr;
return Functions[Position->second].get();
}
MachineFunction *CompilationUnit::function(MachineFunctionStableID ID) noexcept {
for (const std::unique_ptr<MachineFunction> &Function : Functions) {
if (Function->stableID() == ID)
return Function.get();
}
return nullptr;
}
const MachineFunction *CompilationUnit::function(MachineFunctionStableID ID) const noexcept {
for (const std::unique_ptr<MachineFunction> &Function : Functions) {
if (Function->stableID() == ID)
return Function.get();
}
return nullptr;
}
Expected<void> CompilationUnit::verify() const {
if (FunctionNameToIndex.size() != Functions.size()) {
return makeUnexpected(ErrorCode::VerificationError,
"compilation unit function name index has an inconsistent size");
}
std::unordered_set<MachineFunctionStableID> StableIDs;
for (std::size_t Index = 0; Index < Functions.size(); ++Index) {
const std::unique_ptr<MachineFunction> &Function = Functions[Index];
if (Function == nullptr) {
return makeUnexpected(ErrorCode::VerificationError,
"compilation unit contains a null machine function");
}
if (Function->name().empty() || !Function->stableID().isValid()) {
return makeUnexpected(ErrorCode::VerificationError,
"compilation unit contains an invalid machine function identity");
}
if (!StableIDs.insert(Function->stableID()).second) {
return makeUnexpected(ErrorCode::VerificationError,
"compilation unit contains duplicate machine function StableIDs");
}
const auto Position = FunctionNameToIndex.find(Function->name());
if (Position == FunctionNameToIndex.end() || Position->second != Index) {
return makeUnexpected(ErrorCode::VerificationError,
"compilation unit function name index is inconsistent");
}
}
return {};
}
} // namespace rcc
|
