lib/Target/SystemZ/SystemZAsmPrinter.h

   1 //===-- SystemZAsmPrinter.h - SystemZ LLVM assembly printer ----*- C++ -*--===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef LLVM_LIB_TARGET_SYSTEMZ_SYSTEMZASMPRINTER_H
  10 #define LLVM_LIB_TARGET_SYSTEMZ_SYSTEMZASMPRINTER_H
  11
  12 #include "SystemZTargetMachine.h"
  13 #include "SystemZMCInstLower.h"
  14 #include "llvm/CodeGen/AsmPrinter.h"
  15 #include "llvm/CodeGen/StackMaps.h"
  16 #include "llvm/Support/Compiler.h"
  17
  18 namespace llvm {
  19 class MCStreamer;
  20 class MachineBasicBlock;
  21 class MachineInstr;
  22 class Module;
  23 class raw_ostream;
  24
  25 class LLVM_LIBRARY_VISIBILITY SystemZAsmPrinter : public AsmPrinter {
  26 private:
  27   StackMaps SM;
  28
  29 public:
  30   SystemZAsmPrinter(TargetMachine &TM, std::unique_ptr<MCStreamer> Streamer)
  31       : AsmPrinter(TM, std::move(Streamer)), SM(*this) {}
  32
  33   // Override AsmPrinter.
  34   StringRef getPassName() const override { return "SystemZ Assembly Printer"; }
  35   void EmitInstruction(const MachineInstr *MI) override;
  36   void EmitMachineConstantPoolValue(MachineConstantPoolValue *MCPV) override;
  37   void EmitEndOfAsmFile(Module &M) override;
  38   bool PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
  39                        const char *ExtraCode, raw_ostream &OS) override;
  40   bool PrintAsmMemoryOperand(const MachineInstr *MI, unsigned OpNo,
  41                              const char *ExtraCode, raw_ostream &OS) override;
  42
  43   bool doInitialization(Module &M) override {
  44     SM.reset();
  45     return AsmPrinter::doInitialization(M);
  46   }
  47
  48 private:
  49   void LowerSTACKMAP(const MachineInstr &MI);
  50   void LowerPATCHPOINT(const MachineInstr &MI, SystemZMCInstLower &Lower);
  51 };
  52 } // end namespace llvm
  53
  54 #endif