From e5fdceb9b869f010a2a75638fd38aaf1867f3008 Mon Sep 17 00:00:00 2001
From: Richard-Qin-X <richard.qin.001@gmail.com>
Date: Sun, 9 Aug 2026 04:40:17 +0800
Subject: [CodeGen] Complete register allocation infrastructure

---
 backend/include/rcc/CodeGen/Spiller.h | 41 +++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 backend/include/rcc/CodeGen/Spiller.h

(limited to 'backend/include/rcc/CodeGen/Spiller.h')

diff --git a/backend/include/rcc/CodeGen/Spiller.h b/backend/include/rcc/CodeGen/Spiller.h
new file mode 100644
index 0000000..3d1fc66
--- /dev/null
+++ b/backend/include/rcc/CodeGen/Spiller.h
@@ -0,0 +1,41 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef RCC_CODEGEN_SPILLER_H
+#define RCC_CODEGEN_SPILLER_H
+
+#include "rcc/CodeGen/LiveRangeEdit.h"
+#include "rcc/CodeGen/TargetRegisterAllocationInfo.h"
+
+#include <optional>
+#include <vector>
+
+namespace rcc {
+
+class Spiller final {
+public:
+  Spiller(MachineFunction &Function, const TargetDescription &Target,
+          const TargetRegisterAllocationInfo &TargetInfo, VirtRegMap &Mapping,
+          bool EnableRematerialization = true)
+      : Function(&Function), Target(&Target), TargetInfo(&TargetInfo), Mapping(&Mapping),
+        Edit(Function, Target, Mapping), EnableRematerialization(EnableRematerialization) {}
+
+  [[nodiscard]] Expected<std::vector<Register>> spill(Register VirtualReg);
+
+private:
+  [[nodiscard]] Expected<std::vector<Register>> spillAtInstructions(Register VirtualReg);
+  [[nodiscard]] Expected<void> spillLiveIn(Register VirtualReg, RegisterClassID Class,
+                                           FrameIndex StackSlot);
+  [[nodiscard]] Expected<std::optional<std::vector<Register>>>
+  tryRematerialize(Register VirtualReg);
+
+  MachineFunction *Function;
+  const TargetDescription *Target;
+  const TargetRegisterAllocationInfo *TargetInfo;
+  VirtRegMap *Mapping;
+  LiveRangeEdit Edit;
+  bool EnableRematerialization;
+};
+
+} // namespace rcc
+
+#endif // RCC_CODEGEN_SPILLER_H
-- 
cgit v1.2.3