[MIR] Add compilation unit reference context

1 files changed, 61 insertions, 0 deletions

diff --git a/backend/include/rcc/CodeGen/CompilationUnit.h b/backend/include/rcc/CodeGen/CompilationUnit.h
new file mode 100644
index 0000000..d798438
--- /dev/null
+++ b/backend/include/rcc/CodeGen/CompilationUnit.h
@@ -0,0 +1,61 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef RCC_CODEGEN_COMPILATIONUNIT_H
+#define RCC_CODEGEN_COMPILATIONUNIT_H
+
+#include "rcc/CodeGen/StableID.h"
+#include "rcc/Support/Error.h"
+
+#include <cstddef>
+#include <functional>
+#include <memory>
+#include <string>
+#include <string_view>
+#include <unordered_map>
+#include <vector>
+
+namespace rcc {
+
+class MachineFunction;
+
+class CompilationUnit final {
+public:
+  CompilationUnit();
+  CompilationUnit(const CompilationUnit &) = delete;
+  CompilationUnit &operator=(const CompilationUnit &) = delete;
+  CompilationUnit(CompilationUnit &&) noexcept;
+  CompilationUnit &operator=(CompilationUnit &&) noexcept;
+  ~CompilationUnit();
+
+  [[nodiscard]] Expected<MachineFunction *> createMachineFunction(std::string Name);
+
+  [[nodiscard]] std::size_t functionCount() const noexcept {
+    return Functions.size();
+  }
+
+  [[nodiscard]] MachineFunction *functionAt(std::size_t Index) noexcept;
+  [[nodiscard]] const MachineFunction *functionAt(std::size_t Index) const noexcept;
+  [[nodiscard]] MachineFunction *function(std::string_view Name) noexcept;
+  [[nodiscard]] const MachineFunction *function(std::string_view Name) const noexcept;
+  [[nodiscard]] MachineFunction *function(MachineFunctionStableID ID) noexcept;
+  [[nodiscard]] const MachineFunction *function(MachineFunctionStableID ID) const noexcept;
+
+  [[nodiscard]] Expected<void> verify() const;
+
+private:
+  struct StringHash final {
+    using is_transparent = void; // NOLINT(readability-identifier-naming)
+
+    [[nodiscard]] std::size_t operator()(std::string_view Value) const noexcept {
+      return std::hash<std::string_view>{}(Value);
+    }
+  };
+
+  std::vector<std::unique_ptr<MachineFunction>> Functions;
+  std::unordered_map<std::string, std::size_t, StringHash, std::equal_to<>> FunctionNameToIndex;
+  StableIDGenerator<MachineFunctionStableID> FunctionIDGenerator;
+};
+
+} // namespace rcc
+
+#endif // RCC_CODEGEN_COMPILATIONUNIT_H