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

---
 backend/include/rcc/CodeGen/MachineFunction.h | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

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

diff --git a/backend/include/rcc/CodeGen/MachineFunction.h b/backend/include/rcc/CodeGen/MachineFunction.h
index 38d88f9..b687573 100644
--- a/backend/include/rcc/CodeGen/MachineFunction.h
+++ b/backend/include/rcc/CodeGen/MachineFunction.h
@@ -3,12 +3,14 @@
 #ifndef RCC_CODEGEN_MACHINEFUNCTION_H
 #define RCC_CODEGEN_MACHINEFUNCTION_H
 
+#include "rcc/CodeGen/MachineCompilationStatus.h"
 #include "rcc/CodeGen/MachineConstantPool.h"
 #include "rcc/CodeGen/MachineFrameInfo.h"
 #include "rcc/CodeGen/MachineFunctionProperties.h"
 #include "rcc/CodeGen/MachineJumpTableInfo.h"
 #include "rcc/CodeGen/MachineRegisterInfo.h"
 #include "rcc/CodeGen/StableID.h"
+#include "rcc/CodeGen/TargetFunctionState.h"
 #include "rcc/Support/Error.h"
 
 #include <cstddef>
@@ -87,6 +89,29 @@ public:
     return Properties;
   }
 
+  [[nodiscard]] const TargetFunctionState &targetState() const noexcept {
+    return TargetState;
+  }
+
+  [[nodiscard]] Expected<void> setTargetState(TargetFunctionState State);
+
+  template <TargetFunctionStateValue ValueT> [[nodiscard]] ValueT *getTargetInfo() noexcept {
+    return TargetState.getIf<ValueT>();
+  }
+
+  template <TargetFunctionStateValue ValueT>
+  [[nodiscard]] const ValueT *getTargetInfo() const noexcept {
+    return TargetState.getIf<ValueT>();
+  }
+
+  [[nodiscard]] MachineCompilationStatus &compilationStatus() noexcept {
+    return CompilationStatus;
+  }
+
+  [[nodiscard]] const MachineCompilationStatus &compilationStatus() const noexcept {
+    return CompilationStatus;
+  }
+
   [[nodiscard]] Expected<MachineBasicBlock *> createBasicBlock();
   [[nodiscard]] Expected<MachineInstr *> createInstruction(Opcode InstructionOpcode);
   [[nodiscard]] Expected<const MachineMemOperand *>
@@ -141,6 +166,8 @@ private:
   MachineConstantPool ConstantPool;
   MachineJumpTableInfo JumpTables;
   MachineFunctionProperties Properties;
+  TargetFunctionState TargetState;
+  MachineCompilationStatus CompilationStatus;
   MachineBasicBlock *FirstBlock = nullptr;
   MachineBasicBlock *LastBlock = nullptr;
   std::vector<MachineBasicBlock *> AllBlocks;
-- 
cgit v1.2.3