1 files changed, 164 insertions, 0 deletions
diff --git a/backend/include/rcc/CodeGen/MachineInstr.h b/backend/include/rcc/CodeGen/MachineInstr.h
new file mode 100644
index 0000000..376273f
--- /dev/null
+++ b/backend/include/rcc/CodeGen/MachineInstr.h
@@ -0,0 +1,164 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef RCC_CODEGEN_MACHINEINSTR_H
+#define RCC_CODEGEN_MACHINEINSTR_H
+
+#include "rcc/CodeGen/MachineIRTypes.h"
+#include "rcc/CodeGen/MachineOperand.h"
+#include "rcc/CodeGen/StableID.h"
+#include "rcc/Support/Error.h"
+
+#include <cstddef>
+#include <functional>
+#include <optional>
+#include <span>
+#include <utility>
+#include <vector>
+
+namespace rcc {
+
+class MachineBasicBlock;
+class MachineFunction;
+class MachineMemOperand;
+class MachineRegisterInfo;
+
+class MachineInstr final {
+public:
+  MachineInstr(const MachineInstr &) = delete;
+  MachineInstr &operator=(const MachineInstr &) = delete;
+  MachineInstr(MachineInstr &&) = delete;
+  MachineInstr &operator=(MachineInstr &&) = delete;
+  ~MachineInstr();
+
+  [[nodiscard]] Opcode opcode() const noexcept {
+    return InstructionOpcode;
+  }
+
+  [[nodiscard]] Expected<void> setOpcode(Opcode NewOpcode);
+
+  [[nodiscard]] MachineInstrStableID stableID() const noexcept {
+    return StableID;
+  }
+
+  [[nodiscard]] MachineFunction &function() noexcept {
+    return *OwnerFunction;
+  }
+
+  [[nodiscard]] const MachineFunction &function() const noexcept {
+    return *OwnerFunction;
+  }
+
+  [[nodiscard]] MachineBasicBlock *parent() noexcept {
+    return ParentBlock;
+  }
+
+  [[nodiscard]] const MachineBasicBlock *parent() const noexcept {
+    return ParentBlock;
+  }
+
+  [[nodiscard]] MachineInstr *previous() noexcept {
+    return PreviousInstruction;
+  }
+
+  [[nodiscard]] const MachineInstr *previous() const noexcept {
+    return PreviousInstruction;
+  }
+
+  [[nodiscard]] MachineInstr *next() noexcept {
+    return NextInstruction;
+  }
+
+  [[nodiscard]] const MachineInstr *next() const noexcept {
+    return NextInstruction;
+  }
+
+  [[nodiscard]] bool isLinked() const noexcept {
+    return ParentBlock != nullptr;
+  }
+
+  [[nodiscard]] std::size_t operandCount() const noexcept {
+    return Operands.size();
+  }
+
+  [[nodiscard]] MachineOperand *operandAt(std::size_t Index) noexcept;
+  [[nodiscard]] const MachineOperand *operandAt(std::size_t Index) const noexcept;
+  [[nodiscard]] std::span<const MachineOperand> operands() const noexcept {
+    return Operands;
+  }
+
+  void reserveOperands(std::size_t Capacity);
+  [[nodiscard]] Expected<void> addOperand(MachineOperand Operand);
+  [[nodiscard]] Expected<void> insertOperand(std::size_t Index, MachineOperand Operand);
+  [[nodiscard]] Expected<void> eraseOperand(std::size_t Index);
+
+  [[nodiscard]] Expected<void> tieOperands(std::size_t DefIndex, std::size_t UseIndex);
+  [[nodiscard]] Expected<void> untieOperand(std::size_t Index);
+
+  [[nodiscard]] Expected<void> addMemOperand(const MachineMemOperand &MemOperand);
+  [[nodiscard]] Expected<void> eraseMemOperand(std::size_t Index);
+  [[nodiscard]] std::span<const MachineMemOperand *const> memOperands() const noexcept {
+    return MemOperands;
+  }
+
+  [[nodiscard]] MachineInstrFlags flags() const noexcept {
+    return Flags;
+  }
+
+  void setFlags(MachineInstrFlags NewFlags) noexcept {
+    Flags = NewFlags;
+  }
+
+  [[nodiscard]] const std::optional<SourceLocation> &sourceLocation() const noexcept {
+    return Location;
+  }
+
+  void setSourceLocation(SourceLocation NewLocation) {
+    Location = std::move(NewLocation);
+  }
+
+  void clearSourceLocation() noexcept {
+    Location.reset();
+  }
+
+  [[nodiscard]] std::optional<OriginID> origin() const noexcept {
+    return Origin;
+  }
+
+  void setOrigin(OriginID NewOrigin) noexcept {
+    Origin = NewOrigin;
+  }
+
+  void clearOrigin() noexcept {
+    Origin.reset();
+  }
+
+private:
+  MachineInstr(MachineFunction &OwnerFunction, Opcode InstructionOpcode,
+               MachineInstrStableID StableID);
+
+  [[nodiscard]] MachineRegisterInfo *linkedRegisterInfo() noexcept;
+  [[nodiscard]] Expected<void> attachToBlock(MachineBasicBlock &Block);
+  void detachFromBlock() noexcept;
+  void adjustTiesForInsertion(std::size_t Index) noexcept;
+  void adjustTiesForErasure(std::size_t Index) noexcept;
+
+  MachineFunction *OwnerFunction;
+  Opcode InstructionOpcode;
+  MachineInstrStableID StableID;
+  MachineBasicBlock *ParentBlock = nullptr;
+  MachineInstr *PreviousInstruction = nullptr;
+  MachineInstr *NextInstruction = nullptr;
+  std::vector<MachineOperand> Operands;
+  std::vector<const MachineMemOperand *> MemOperands;
+  MachineInstrFlags Flags;
+  std::optional<SourceLocation> Location;
+  std::optional<OriginID> Origin;
+
+  friend class MachineArena;
+  friend class MachineBasicBlock;
+  friend class MachineFunction;
+};
+
+} // namespace rcc
+
+#endif // RCC_CODEGEN_MACHINEINSTR_H