lib/Target/PTX/PTXAsmPrinter.h

   1 //===-- PTXAsmPrinter.h - Print machine code to a PTX file ------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // PTX Assembly printer class.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef PTXASMPRINTER_H
  15 #define PTXASMPRINTER_H
  16
  17 #include "PTX.h"
  18 #include "PTXTargetMachine.h"
  19 #include "llvm/ADT/StringMap.h"
  20 #include "llvm/CodeGen/AsmPrinter.h"
  21 #include "llvm/Support/Compiler.h"
  22
  23 namespace llvm {
  24
  25 class MCOperand;
  26
  27 class LLVM_LIBRARY_VISIBILITY PTXAsmPrinter : public AsmPrinter {
  28 public:
  29   explicit PTXAsmPrinter(TargetMachine &TM, MCStreamer &Streamer)
  30     : AsmPrinter(TM, Streamer) {}
  31
  32   const char *getPassName() const { return "PTX Assembly Printer"; }
  33
  34   bool doFinalization(Module &M);
  35
  36   virtual void EmitStartOfAsmFile(Module &M);
  37   virtual void EmitFunctionBodyStart();
  38   virtual void EmitFunctionBodyEnd();
  39   virtual void EmitFunctionEntryLabel();
  40   virtual void EmitInstruction(const MachineInstr *MI);
  41
  42   unsigned GetOrCreateSourceID(StringRef FileName,
  43                                StringRef DirName);
  44
  45   MCOperand GetSymbolRef(const MachineOperand &MO, const MCSymbol *Symbol);
  46   MCOperand lowerOperand(const MachineOperand &MO);
  47
  48 private:
  49   void EmitVariableDeclaration(const GlobalVariable *gv);
  50   void EmitFunctionDeclaration(const Function* func);
  51
  52   StringMap<unsigned> SourceIdMap;
  53 }; // class PTXAsmPrinter
  54 } // namespace llvm
  55
  56 #endif
  57