[CodeGen] Separate assembly emission ownership

1 files changed, 40 insertions, 5 deletions

diff --git a/backend/include/rcc/CodeGen/AsmPrinter.h b/backend/include/rcc/CodeGen/AsmPrinter.h
index 82bad7d..4145093 100644
--- a/backend/include/rcc/CodeGen/AsmPrinter.h
+++ b/backend/include/rcc/CodeGen/AsmPrinter.h
@@ -3,14 +3,24 @@
 #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 {
@@ -25,31 +35,56 @@ struct AsmPrinterOptions final {
   bool VerifyInput = true;
 };
 
-class AsmPrinter final {
+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), Context(&Context), Streamer(&Streamer),
-        Options(Options) {}
+      : 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 UnitEmissionStarted = false;
-  bool UnitEmissionFinished = false;
+  bool EmissionStarted = false;
+  bool EmissionFinished = false;
 };
 
 } // namespace rcc