Add support for stack map generation in the X86 backend.
[oota-llvm.git] / lib / Target / X86 / X86AsmPrinter.h
1 //===-- X86AsmPrinter.h - X86 implementation of AsmPrinter ------*- 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 #ifndef X86ASMPRINTER_H
11 #define X86ASMPRINTER_H
12
13 #include "X86.h"
14 #include "X86MachineFunctionInfo.h"
15 #include "X86TargetMachine.h"
16 #include "llvm/CodeGen/AsmPrinter.h"
17 #include "llvm/CodeGen/MachineModuleInfo.h"
18 #include "llvm/CodeGen/ValueTypes.h"
19 #include "llvm/CodeGen/StackMaps.h"
20 #include "llvm/Support/Compiler.h"
21
22 namespace llvm {
23
24 class MCStreamer;
25
26 class LLVM_LIBRARY_VISIBILITY X86AsmPrinter : public AsmPrinter {
27   const X86Subtarget *Subtarget;
28   StackMaps SM;
29
30   // Parses operands of PATCHPOINT and STACKMAP to produce stack map Location
31   // structures. Returns a result location and an iterator to the operand
32   // immediately following the operands consumed.
33   //
34   // This method is implemented in X86MCInstLower.cpp.
35   static std::pair<StackMaps::Location, MachineInstr::const_mop_iterator>
36     stackmapOperandParser(MachineInstr::const_mop_iterator MOI,
37                           MachineInstr::const_mop_iterator MOE);
38
39  public:
40   explicit X86AsmPrinter(TargetMachine &TM, MCStreamer &Streamer)
41     : AsmPrinter(TM, Streamer), SM(*this, stackmapOperandParser) {
42     Subtarget = &TM.getSubtarget<X86Subtarget>();
43   }
44
45   virtual const char *getPassName() const LLVM_OVERRIDE {
46     return "X86 Assembly / Object Emitter";
47   }
48
49   const X86Subtarget &getSubtarget() const { return *Subtarget; }
50
51   virtual void EmitStartOfAsmFile(Module &M) LLVM_OVERRIDE;
52
53   virtual void EmitEndOfAsmFile(Module &M) LLVM_OVERRIDE;
54
55   virtual void EmitInstruction(const MachineInstr *MI) LLVM_OVERRIDE;
56
57   void printSymbolOperand(const MachineOperand &MO, raw_ostream &O);
58
59   // These methods are used by the tablegen'erated instruction printer.
60   void printOperand(const MachineInstr *MI, unsigned OpNo, raw_ostream &O,
61                     const char *Modifier = 0, unsigned AsmVariant = 0);
62   void printPCRelImm(const MachineInstr *MI, unsigned OpNo, raw_ostream &O);
63
64   bool printAsmMRegister(const MachineOperand &MO, char Mode, raw_ostream &O);
65   virtual bool PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
66                                unsigned AsmVariant, const char *ExtraCode,
67                                raw_ostream &OS) LLVM_OVERRIDE;
68   virtual bool PrintAsmMemoryOperand(const MachineInstr *MI, unsigned OpNo,
69                                      unsigned AsmVariant, const char *ExtraCode,
70                                      raw_ostream &OS) LLVM_OVERRIDE;
71
72   void printMemReference(const MachineInstr *MI, unsigned Op, raw_ostream &O,
73                          const char *Modifier=NULL);
74   void printLeaMemReference(const MachineInstr *MI, unsigned Op, raw_ostream &O,
75                             const char *Modifier=NULL);
76
77   void printIntelMemReference(const MachineInstr *MI, unsigned Op,
78                               raw_ostream &O, const char *Modifier=NULL,
79                               unsigned AsmVariant = 1);
80
81   virtual bool runOnMachineFunction(MachineFunction &F) LLVM_OVERRIDE;
82 };
83
84 } // end namespace llvm
85
86 #endif