1 files changed, 92 insertions, 0 deletions
diff --git a/backend/include/rcc/CodeGen/AsmPrinter.h b/backend/include/rcc/CodeGen/AsmPrinter.h
new file mode 100644
index 0000000..4145093
--- /dev/null
+++ b/backend/include/rcc/CodeGen/AsmPrinter.h
@@ -0,0 +1,92 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef RCC_CODEGEN_ASMPRINTER_H
+#define RCC_CODEGEN_ASMPRINTER_H
+
+#include "rcc/IR/DataLayout.h"
+#include "rcc/Support/Error.h"
+
+#include <cstddef>
+#include <cstdint>
+#include <utility>
+
+namespace rcc {
+
+class CompilationUnit;
+class GlobalInitializer;
+class MachineFunction;
+class MachineJumpTableEntry;
+class MachineMCContext;
+struct GenericConstantData;
+class TargetConstant;
+class TargetDescription;
+class TargetLoweringObjectFile;
+class TargetMCInstLowering;
+
+namespace mc {
+class MCStreamer;
+} // namespace mc
+
+struct AsmPrinterOptions final {
+  bool EmitFunctionDirectives = true;
+  bool EmitCFI = true;
+  bool EmitBlockLabels = true;
+  bool EmitInstructionSymbols = true;
+  bool VerifyInput = true;
+};
+
+class AsmPrinter {
+public:
+  AsmPrinter(const TargetDescription &Target, const TargetMCInstLowering &Lowering,
+             const TargetLoweringObjectFile &ObjectFile, DataLayout Layout,
+             MachineMCContext &Context, mc::MCStreamer &Streamer,
+             AsmPrinterOptions Options = {}) noexcept
+      : Target(&Target), Lowering(&Lowering), ObjectFile(&ObjectFile), Layout(std::move(Layout)),
+        Context(&Context), Streamer(&Streamer), Options(Options) {}
+  virtual ~AsmPrinter() = default;
+
+  [[nodiscard]] Expected<void> emitCompilationUnit(const CompilationUnit &Unit);
+  [[nodiscard]] Expected<void> emitFunction(const MachineFunction &Function);
+
+protected:
+  [[nodiscard]] virtual Expected<void> emitStartOfAsmFile();
+  [[nodiscard]] virtual Expected<void> emitEndOfAsmFile();
+  [[nodiscard]] virtual Expected<void> emitTargetConstant(const TargetConstant &Constant);
+  [[nodiscard]] virtual Expected<void> emitTargetJumpTable(const MachineFunction &Function,
+                                                           const MachineJumpTableEntry &Entry);
+
+  [[nodiscard]] MachineMCContext &machineContext() const noexcept {
+    return *Context;
+  }
+
+  [[nodiscard]] mc::MCStreamer &streamer() const noexcept {
+    return *Streamer;
+  }
+
+private:
+  [[nodiscard]] Expected<void> verifyFunction(const MachineFunction &Function) const;
+  [[nodiscard]] Expected<void> emitGlobalObjects(const CompilationUnit &Unit);
+  [[nodiscard]] Expected<void> emitFunctionImpl(const MachineFunction &Function);
+  [[nodiscard]] Expected<void> emitFunctionBody(const MachineFunction &Function);
+  [[nodiscard]] Expected<void> emitConstantPool(const MachineFunction &Function);
+  [[nodiscard]] Expected<void> emitJumpTables(const MachineFunction &Function);
+  [[nodiscard]] Expected<void> emitGenericConstant(const GenericConstantData &Constant,
+                                                   std::uint64_t AllocationSize,
+                                                   std::size_t Depth = 0);
+  [[nodiscard]] Expected<void> emitGlobalInitializer(const GlobalInitializer &Initializer,
+                                                     std::size_t Depth = 0);
+
+  const TargetDescription *Target;
+  const TargetMCInstLowering *Lowering;
+  const TargetLoweringObjectFile *ObjectFile;
+  DataLayout Layout;
+  MachineMCContext *Context;
+  mc::MCStreamer *Streamer;
+  AsmPrinterOptions Options;
+  bool EmissionStarted = false;
+  bool EmissionFinished = false;
+};
+
+} // namespace rcc
+
+#endif // RCC_CODEGEN_ASMPRINTER_H