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/lib/CodeGen/RegAllocBase.h | 51 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 backend/lib/CodeGen/RegAllocBase.h

(limited to 'backend/lib/CodeGen/RegAllocBase.h')

diff --git a/backend/lib/CodeGen/RegAllocBase.h b/backend/lib/CodeGen/RegAllocBase.h
new file mode 100644
index 0000000..5af5ed0
--- /dev/null
+++ b/backend/lib/CodeGen/RegAllocBase.h
@@ -0,0 +1,51 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+// The private header has no stable include-root-relative guard name.
+// NOLINTNEXTLINE(llvm-header-guard)
+#ifndef RCC_LIB_CODEGEN_REGALLOCBASE_H
+#define RCC_LIB_CODEGEN_REGALLOCBASE_H
+
+#include "RegAllocInternal.h"
+#include "rcc/CodeGen/MachineLiveness.h"
+#include "rcc/CodeGen/RegAllocCommon.h"
+#include "rcc/CodeGen/TargetRegisterAllocationInfo.h"
+#include "rcc/CodeGen/VirtRegMap.h"
+
+#include <vector>
+
+namespace rcc {
+
+struct AllocationRoundResult final {
+  std::vector<Register> Spilled;
+};
+
+class RegAllocBase {
+public:
+  virtual ~RegAllocBase() = default;
+
+  [[nodiscard]] Expected<regalloc_detail::AllocationRunResult> run();
+
+protected:
+  RegAllocBase(MachineFunction &Function, const TargetDescription &Target,
+               const TargetRegisterAllocationInfo &TargetInfo, MachineAnalysisManager &Analyses,
+               RegisterAllocationOptions Options) noexcept
+      : Function(Function), Target(Target), TargetInfo(TargetInfo), Analyses(Analyses),
+        Options(Options) {}
+
+  [[nodiscard]] virtual Expected<AllocationRoundResult>
+  allocateRound(const LiveIntervalsAnalysis::Result &Intervals, VirtRegMap &Mapping) = 0;
+
+  MachineFunction &Function;
+  const TargetDescription &Target;
+  const TargetRegisterAllocationInfo &TargetInfo;
+  MachineAnalysisManager &Analyses;
+  RegisterAllocationOptions Options;
+};
+
+[[nodiscard]] std::vector<const LiveInterval *>
+orderedAllocationIntervals(const LiveIntervalsAnalysis::Result &Intervals,
+                           const MachineFunction &Function);
+
+} // namespace rcc
+
+#endif // RCC_LIB_CODEGEN_REGALLOCBASE_H
-- 
cgit v1.2.3