[CodeGen] Complete register allocation infrastructure

1 files changed, 32 insertions, 0 deletions

diff --git a/backend/include/rcc/CodeGen/LiveRangeEdit.h b/backend/include/rcc/CodeGen/LiveRangeEdit.h
new file mode 100644
index 0000000..66696df
--- /dev/null
+++ b/backend/include/rcc/CodeGen/LiveRangeEdit.h
@@ -0,0 +1,32 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef RCC_CODEGEN_LIVERANGEEDIT_H
+#define RCC_CODEGEN_LIVERANGEEDIT_H
+
+#include "rcc/CodeGen/VirtRegMap.h"
+
+#include <optional>
+
+namespace rcc {
+
+class TargetDescription;
+
+class LiveRangeEdit final {
+public:
+  LiveRangeEdit(MachineFunction &Function, const TargetDescription &Target, VirtRegMap &Mapping)
+      : Function(&Function), Target(&Target), Mapping(&Mapping) {}
+
+  [[nodiscard]] Expected<Register> createFrom(Register Parent, bool Spillable,
+                                              std::optional<Register> Origin = std::nullopt);
+  [[nodiscard]] Expected<FrameIndex> getOrCreateSpillSlot(Register VirtualReg);
+  [[nodiscard]] Expected<void> eraseVirtualRegister(Register VirtualReg);
+
+private:
+  MachineFunction *Function;
+  const TargetDescription *Target;
+  VirtRegMap *Mapping;
+};
+
+} // namespace rcc
+
+#endif // RCC_CODEGEN_LIVERANGEEDIT_H