[CodeGen] Add target machine registry

4 files changed, 125 insertions, 15 deletions

diff --git a/backend/include/rcc/Target/RISCV/RISCVPipeline.h b/backend/include/rcc/Target/RISCV/RISCVPipeline.h
index 3e8ea3d..f556ded 100644
--- a/backend/include/rcc/Target/RISCV/RISCVPipeline.h
+++ b/backend/include/rcc/Target/RISCV/RISCVPipeline.h
@@ -3,10 +3,10 @@
 #ifndef RCC_TARGET_RISCV_RISCVPIPELINE_H
 #define RCC_TARGET_RISCV_RISCVPIPELINE_H
 
-#include "rcc/CodeGen/CommandFlags.h"
 #include "rcc/CodeGen/MachinePassManager.h"
 #include "rcc/Support/Error.h"
 #include "rcc/Target/RISCV/RISCVSubtarget.h"
+#include "rcc/Target/TargetMachine.h"
 
 #include <memory>
 #include <string>
@@ -21,21 +21,13 @@ class TargetMachineOutlinerInfo;
 
 namespace riscv {
 
-struct RISCVPipelineOptions final {
-  OptimizationLevel Optimization = OptimizationLevel::O2;
-  bool UseTargetCopyInstructions = false;
-  GlobalISelAbortMode GlobalISelAbort = GlobalISelAbortMode::Enable;
-  bool DebugMachineScheduler = false;
-  bool DebugPrologueEpilogue = false;
-};
-
 [[nodiscard]] Expected<void> addCanonicalPipeline(MachineFunctionPassManager &Pipeline,
                                                   const MachineFunction &Function,
                                                   const RISCVSubtarget &Subtarget,
-                                                  RISCVPipelineOptions Options = {});
+                                                  TargetPipelineOptions Options = {});
 [[nodiscard]] Expected<std::unique_ptr<MachineFunctionPass>>
 createCanonicalPipelinePass(std::string_view Name, const RISCVSubtarget &Subtarget,
-                            RISCVPipelineOptions Options = {});
+                            TargetPipelineOptions Options = {});
 [[nodiscard]] Expected<std::vector<std::string>>
 canonicalPipelinePassNames(const MachineFunction &Function, const RISCVSubtarget &Subtarget);
 [[nodiscard]] Expected<std::unique_ptr<TargetMachineOutlinerInfo>>
diff --git a/backend/include/rcc/Target/RISCV/RISCVSubtarget.h b/backend/include/rcc/Target/RISCV/RISCVSubtarget.h
index 52e4b03..a5ba01a 100644
--- a/backend/include/rcc/Target/RISCV/RISCVSubtarget.h
+++ b/backend/include/rcc/Target/RISCV/RISCVSubtarget.h
@@ -142,10 +142,10 @@ private:
   CodeModel Code;
   RISCVABI ABI;
   std::unique_ptr<TargetFrameLowering> FrameLoweringInfo;
-  mutable std::unique_ptr<TargetCallLowering> CallLoweringInfo;
-  mutable std::unique_ptr<InstructionSelector> InstSelector;
-  mutable std::unique_ptr<LegalizerInfo> Legalizer;
-  mutable std::unique_ptr<RISCVRegisterBankInfo> RegBankInfo;
+  std::unique_ptr<TargetCallLowering> CallLoweringInfo;
+  std::unique_ptr<InstructionSelector> InstSelector;
+  std::unique_ptr<LegalizerInfo> Legalizer;
+  std::unique_ptr<RISCVRegisterBankInfo> RegBankInfo;
   std::unique_ptr<TargetSchedulingHooks> SchedulingHooks;
   std::unique_ptr<TargetSchedulingModel> SchedulingModel;
 };
diff --git a/backend/include/rcc/Target/TargetMachine.h b/backend/include/rcc/Target/TargetMachine.h
new file mode 100644
index 0000000..e45b8ca
--- /dev/null
+++ b/backend/include/rcc/Target/TargetMachine.h
@@ -0,0 +1,69 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef RCC_TARGET_TARGETMACHINE_H
+#define RCC_TARGET_TARGETMACHINE_H
+
+#include "rcc/CodeGen/MachineOutliner.h"
+#include "rcc/CodeGen/MachinePassManager.h"
+#include "rcc/CodeGen/TargetDescription.h"
+#include "rcc/Support/Error.h"
+#include "rcc/Target/TargetOptions.h"
+
+#include <cstdint>
+#include <memory>
+#include <string>
+#include <string_view>
+#include <vector>
+
+namespace rcc {
+
+class CompilationUnit;
+class MachineFunction;
+class Target;
+
+class TargetMachine {
+public:
+  TargetMachine(const TargetMachine &) = delete;
+  TargetMachine &operator=(const TargetMachine &) = delete;
+  TargetMachine(TargetMachine &&) = delete;
+  TargetMachine &operator=(TargetMachine &&) = delete;
+  virtual ~TargetMachine();
+
+  [[nodiscard]] const Target &target() const noexcept {
+    return RegisteredTarget;
+  }
+
+  [[nodiscard]] const TargetDescription &targetDescription() const noexcept {
+    return Description;
+  }
+
+  [[nodiscard]] const TargetMachineOptions &options() const noexcept {
+    return Options;
+  }
+
+  [[nodiscard]] virtual Expected<void>
+  addCodeGenPasses(MachineFunctionPassManager &Pipeline, const MachineFunction &Function,
+                   TargetPipelineOptions PipelineOptions) const = 0;
+  [[nodiscard]] virtual Expected<std::unique_ptr<MachineFunctionPass>>
+  createMachinePass(std::string_view Name, TargetPipelineOptions PipelineOptions) const = 0;
+  [[nodiscard]] virtual Expected<std::vector<std::string>>
+  codeGenPassNames(const MachineFunction &Function) const = 0;
+  [[nodiscard]] virtual Expected<std::unique_ptr<TargetMachineOutlinerInfo>>
+  createOutlinerInfo() const = 0;
+  [[nodiscard]] virtual Expected<std::string> emitAssembly(const CompilationUnit &Unit) const = 0;
+  [[nodiscard]] virtual Expected<std::vector<std::uint8_t>>
+  emitObject(const CompilationUnit &Unit) const = 0;
+
+protected:
+  TargetMachine(const Target &RegisteredTarget, TargetMachineOptions Options,
+                TargetDescription Description);
+
+private:
+  const Target &RegisteredTarget;
+  TargetMachineOptions Options;
+  TargetDescription Description;
+};
+
+} // namespace rcc
+
+#endif // RCC_TARGET_TARGETMACHINE_H
diff --git a/backend/include/rcc/Target/TargetOptions.h b/backend/include/rcc/Target/TargetOptions.h
new file mode 100644
index 0000000..1db183c
--- /dev/null
+++ b/backend/include/rcc/Target/TargetOptions.h
@@ -0,0 +1,49 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef RCC_TARGET_TARGETOPTIONS_H
+#define RCC_TARGET_TARGETOPTIONS_H
+
+#include "rcc/CodeGen/CommandFlags.h"
+
+#include <string>
+#include <vector>
+
+namespace rcc {
+
+struct TargetOption final {
+  std::string Name;
+  std::string Value;
+
+  friend bool operator==(const TargetOption &, const TargetOption &) = default;
+};
+
+struct TargetMachineOptions final {
+  std::string Triple;
+  std::string CPU;
+  std::string TuneCPU;
+  std::vector<std::string> Features;
+  std::string ABI;
+  std::string DataLayout;
+  RelocationModel Relocation = RelocationModel::Static;
+  CodeModel Code = CodeModel::Small;
+  OptimizationLevel Optimization = OptimizationLevel::O2;
+  std::vector<TargetOption> Options;
+  std::string ABILoweringVersion;
+  std::string AssemblerFlavor;
+  std::string AssemblerVersion;
+  std::vector<TargetOption> AssemblerOptions;
+
+  friend bool operator==(const TargetMachineOptions &, const TargetMachineOptions &) = default;
+};
+
+struct TargetPipelineOptions final {
+  OptimizationLevel Optimization = OptimizationLevel::O2;
+  bool UseTargetCopyInstructions = false;
+  GlobalISelAbortMode GlobalISelAbort = GlobalISelAbortMode::Enable;
+  bool DebugMachineScheduler = false;
+  bool DebugPrologueEpilogue = false;
+};
+
+} // namespace rcc
+
+#endif // RCC_TARGET_TARGETOPTIONS_H