From cfdbcc469f9691bd9dce3b0b86982a63d86ca6b7 Mon Sep 17 00:00:00 2001
From: Richard-Qin-X <richard.qin.001@gmail.com>
Date: Sat, 1 Aug 2026 14:50:04 +0800
Subject: [CodeGen] Add target state and compilation status

---
 .../include/rcc/CodeGen/MachineFunctionCloner.h    | 45 ++++++++++++++++++++++
 1 file changed, 45 insertions(+)

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

diff --git a/backend/include/rcc/CodeGen/MachineFunctionCloner.h b/backend/include/rcc/CodeGen/MachineFunctionCloner.h
index 3abdf78..658614f 100644
--- a/backend/include/rcc/CodeGen/MachineFunctionCloner.h
+++ b/backend/include/rcc/CodeGen/MachineFunctionCloner.h
@@ -16,7 +16,9 @@
 namespace rcc {
 
 class InstructionSet;
+class MachineBasicBlock;
 class MachineFunction;
+class MachineInstr;
 
 struct MachineFunctionCloneOptions final {
   MachineFunctionStableID NewStableID;
@@ -40,6 +42,49 @@ struct MachineFunctionCloneResult final {
   std::unordered_map<JumpTableIndex, JumpTableIndex> JumpTableIndices;
 };
 
+class MachineFunctionCloneContext final {
+public:
+  [[nodiscard]] const MachineFunction &sourceFunction() const noexcept {
+    return Source;
+  }
+
+  [[nodiscard]] MachineFunction &destinationFunction() const noexcept {
+    return Destination;
+  }
+
+  [[nodiscard]] Expected<MachineBasicBlock *> mapBlock(const MachineBasicBlock &SourceBlock) const;
+  [[nodiscard]] Expected<MachineInstr *>
+  mapInstruction(const MachineInstr &SourceInstruction) const;
+  [[nodiscard]] Expected<MachineBlockStableID>
+  mapBlockStableID(MachineBlockStableID SourceID) const;
+  [[nodiscard]] Expected<MachineInstrStableID>
+  mapInstructionStableID(MachineInstrStableID SourceID) const;
+  [[nodiscard]] Expected<Register> mapRegister(Register SourceRegister) const;
+  [[nodiscard]] Expected<FrameIndex> mapFrameIndex(FrameIndex SourceIndex) const;
+  [[nodiscard]] Expected<ConstantPoolIndex>
+  mapConstantPoolIndex(ConstantPoolIndex SourceIndex) const;
+  [[nodiscard]] Expected<JumpTableIndex> mapJumpTableIndex(JumpTableIndex SourceIndex) const;
+
+private:
+  MachineFunctionCloneContext(
+      const MachineFunction &Source, MachineFunction &Destination,
+      const MachineFunctionCloneResult &Result,
+      const std::unordered_map<const MachineBasicBlock *, MachineBasicBlock *> &Blocks,
+      const std::unordered_map<const MachineInstr *, MachineInstr *> &Instructions) noexcept
+      : Source(Source), Destination(Destination), Result(Result), Blocks(Blocks),
+        Instructions(Instructions) {}
+
+  const MachineFunction &Source;
+  MachineFunction &Destination;
+  const MachineFunctionCloneResult &Result;
+  const std::unordered_map<const MachineBasicBlock *, MachineBasicBlock *> &Blocks;
+  const std::unordered_map<const MachineInstr *, MachineInstr *> &Instructions;
+
+  friend Expected<MachineFunctionCloneResult>
+  cloneMachineFunction(const MachineFunction &Source, const InstructionSet &Instructions,
+                       MachineFunctionCloneOptions Options);
+};
+
 [[nodiscard]] Expected<MachineFunctionCloneResult>
 cloneMachineFunction(const MachineFunction &Source, const InstructionSet &Instructions,
                      MachineFunctionCloneOptions Options);
-- 
cgit v1.2.3