add some more assertions. Remove code to handle dllimport on darwin.
[oota-llvm.git] / lib / Target / X86 / AsmPrinter / X86ATTAsmPrinter.h
1 //===-- X86ATTAsmPrinter.h - Convert X86 LLVM code to AT&T assembly -------===//
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 // AT&T assembly code printer class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #ifndef X86ATTASMPRINTER_H
15 #define X86ATTASMPRINTER_H
16
17 #include "../X86.h"
18 #include "../X86MachineFunctionInfo.h"
19 #include "../X86TargetMachine.h"
20 #include "llvm/ADT/StringSet.h"
21 #include "llvm/CodeGen/AsmPrinter.h"
22 #include "llvm/CodeGen/DwarfWriter.h"
23 #include "llvm/CodeGen/MachineModuleInfo.h"
24 #include "llvm/CodeGen/ValueTypes.h"
25 #include "llvm/Support/Compiler.h"
26
27 namespace llvm {
28
29 class MachineJumpTableInfo;
30 class MCContext;
31 class MCInst;
32 class MCStreamer;
33
34 class VISIBILITY_HIDDEN X86ATTAsmPrinter : public AsmPrinter {
35   const X86Subtarget *Subtarget;
36   
37   MCContext *Context;
38   MCStreamer *Streamer;
39  public:
40   explicit X86ATTAsmPrinter(raw_ostream &O, X86TargetMachine &TM,
41                             const TargetAsmInfo *T, bool V)
42     : AsmPrinter(O, TM, T, V) {
43     Subtarget = &TM.getSubtarget<X86Subtarget>();
44     Context = 0;
45     Streamer = 0;
46   }
47
48   virtual const char *getPassName() const {
49     return "X86 AT&T-Style Assembly Printer";
50   }
51
52   void getAnalysisUsage(AnalysisUsage &AU) const {
53     AU.setPreservesAll();
54     if (Subtarget->isTargetDarwin() ||
55         Subtarget->isTargetELF() ||
56         Subtarget->isTargetCygMing()) {
57       AU.addRequired<MachineModuleInfo>();
58     }
59     AU.addRequired<DwarfWriter>();
60     AsmPrinter::getAnalysisUsage(AU);
61   }
62
63   bool doInitialization(Module &M);
64   bool doFinalization(Module &M);
65
66   /// printInstruction - This method is automatically generated by tablegen
67   /// from the instruction set description.  This method returns true if the
68   /// machine instruction was sufficiently described to print it, otherwise it
69   /// returns false.
70   bool printInstruction(const MachineInstr *MI);
71   
72   
73   // New MCInst printing stuff.
74   bool printInstruction(const MCInst *MI);
75
76   void printOperand(const MCInst *MI, unsigned OpNo,
77                     const char *Modifier = 0);
78   void printMemReference(const MCInst *MI, unsigned Op);
79   void printLeaMemReference(const MCInst *MI, unsigned Op);
80   void printSSECC(const MCInst *MI, unsigned Op);
81   void printPICLabel(const MCInst *MI, unsigned Op);
82   void print_pcrel_imm(const MCInst *MI, unsigned OpNo);
83   
84   void printi8mem(const MCInst *MI, unsigned OpNo) {
85     printMemReference(MI, OpNo);
86   }
87   void printi16mem(const MCInst *MI, unsigned OpNo) {
88     printMemReference(MI, OpNo);
89   }
90   void printi32mem(const MCInst *MI, unsigned OpNo) {
91     printMemReference(MI, OpNo);
92   }
93   void printi64mem(const MCInst *MI, unsigned OpNo) {
94     printMemReference(MI, OpNo);
95   }
96   void printi128mem(const MCInst *MI, unsigned OpNo) {
97     printMemReference(MI, OpNo);
98   }
99   void printf32mem(const MCInst *MI, unsigned OpNo) {
100     printMemReference(MI, OpNo);
101   }
102   void printf64mem(const MCInst *MI, unsigned OpNo) {
103     printMemReference(MI, OpNo);
104   }
105   void printf80mem(const MCInst *MI, unsigned OpNo) {
106     printMemReference(MI, OpNo);
107   }
108   void printf128mem(const MCInst *MI, unsigned OpNo) {
109     printMemReference(MI, OpNo);
110   }
111   void printlea32mem(const MCInst *MI, unsigned OpNo) {
112     printLeaMemReference(MI, OpNo);
113   }
114   void printlea64mem(const MCInst *MI, unsigned OpNo) {
115     printLeaMemReference(MI, OpNo);
116   }
117   void printlea64_32mem(const MCInst *MI, unsigned OpNo) {
118     printLeaMemReference(MI, OpNo);
119   }
120   
121   
122
123   // These methods are used by the tablegen'erated instruction printer.
124   void printOperand(const MachineInstr *MI, unsigned OpNo,
125                     const char *Modifier = 0);
126   void print_pcrel_imm(const MachineInstr *MI, unsigned OpNo);
127   void printi8mem(const MachineInstr *MI, unsigned OpNo) {
128     printMemReference(MI, OpNo);
129   }
130   void printi16mem(const MachineInstr *MI, unsigned OpNo) {
131     printMemReference(MI, OpNo);
132   }
133   void printi32mem(const MachineInstr *MI, unsigned OpNo) {
134     printMemReference(MI, OpNo);
135   }
136   void printi64mem(const MachineInstr *MI, unsigned OpNo) {
137     printMemReference(MI, OpNo);
138   }
139   void printi128mem(const MachineInstr *MI, unsigned OpNo) {
140     printMemReference(MI, OpNo);
141   }
142   void printi256mem(const MachineInstr *MI, unsigned OpNo) {
143     printMemReference(MI, OpNo);
144   }
145   void printf32mem(const MachineInstr *MI, unsigned OpNo) {
146     printMemReference(MI, OpNo);
147   }
148   void printf64mem(const MachineInstr *MI, unsigned OpNo) {
149     printMemReference(MI, OpNo);
150   }
151   void printf80mem(const MachineInstr *MI, unsigned OpNo) {
152     printMemReference(MI, OpNo);
153   }
154   void printf128mem(const MachineInstr *MI, unsigned OpNo) {
155     printMemReference(MI, OpNo);
156   }
157   void printf256mem(const MachineInstr *MI, unsigned OpNo) {
158     printMemReference(MI, OpNo);
159   }
160   void printlea32mem(const MachineInstr *MI, unsigned OpNo) {
161     printLeaMemReference(MI, OpNo);
162   }
163   void printlea64mem(const MachineInstr *MI, unsigned OpNo) {
164     printLeaMemReference(MI, OpNo);
165   }
166   void printlea64_32mem(const MachineInstr *MI, unsigned OpNo) {
167     printLeaMemReference(MI, OpNo, "subreg64");
168   }
169
170   bool printAsmMRegister(const MachineOperand &MO, char Mode);
171   bool PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
172                        unsigned AsmVariant, const char *ExtraCode);
173   bool PrintAsmMemoryOperand(const MachineInstr *MI, unsigned OpNo,
174                              unsigned AsmVariant, const char *ExtraCode);
175
176   void printMachineInstruction(const MachineInstr *MI);
177   void printSSECC(const MachineInstr *MI, unsigned Op);
178   void printMemReference(const MachineInstr *MI, unsigned Op,
179                          const char *Modifier=NULL);
180   void printLeaMemReference(const MachineInstr *MI, unsigned Op,
181                             const char *Modifier=NULL);
182   void printPICJumpTableSetLabel(unsigned uid,
183                                  const MachineBasicBlock *MBB) const;
184   void printPICJumpTableSetLabel(unsigned uid, unsigned uid2,
185                                  const MachineBasicBlock *MBB) const {
186     AsmPrinter::printPICJumpTableSetLabel(uid, uid2, MBB);
187   }
188   void printPICJumpTableEntry(const MachineJumpTableInfo *MJTI,
189                               const MachineBasicBlock *MBB,
190                               unsigned uid) const;
191
192   void printPICLabel(const MachineInstr *MI, unsigned Op);
193   void printModuleLevelGV(const GlobalVariable* GVar);
194
195   void PrintPICBaseSymbol() const;
196   
197   bool runOnMachineFunction(MachineFunction &F);
198
199   void emitFunctionHeader(const MachineFunction &MF);
200
201   // Necessary for Darwin to print out the apprioriate types of linker stubs
202   StringSet<> FnStubs, GVStubs, HiddenGVStubs;
203
204   // Necessary for dllexport support
205   StringSet<> DLLExportedFns, DLLExportedGVs;
206
207   // We have to propagate some information about MachineFunction to
208   // AsmPrinter. It's ok, when we're printing the function, since we have
209   // access to MachineFunction and can get the appropriate MachineFunctionInfo.
210   // Unfortunately, this is not possible when we're printing reference to
211   // Function (e.g. calling it and so on). Even more, there is no way to get the
212   // corresponding MachineFunctions: it can even be not created at all. That's
213   // why we should use additional structure, when we're collecting all necessary
214   // information.
215   //
216   // This structure is using e.g. for name decoration for stdcall & fastcall'ed
217   // function, since we have to use arguments' size for decoration.
218   typedef std::map<const Function*, X86MachineFunctionInfo> FMFInfoMap;
219   FMFInfoMap FunctionInfoMap;
220
221   void decorateName(std::string& Name, const GlobalValue* GV);
222 };
223
224 } // end namespace llvm
225
226 #endif