[mips] Improve support for the .set at/noat assembler directives.
[oota-llvm.git] / lib / Target / Mips / MipsTargetStreamer.h
1 //===-- MipsTargetStreamer.h - Mips Target Streamer ------------*- 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 LLVM_LIB_TARGET_MIPS_MIPSTARGETSTREAMER_H
11 #define LLVM_LIB_TARGET_MIPS_MIPSTARGETSTREAMER_H
12
13 #include "MCTargetDesc/MipsABIFlagsSection.h"
14 #include "MCTargetDesc/MipsABIInfo.h"
15 #include "llvm/MC/MCELFStreamer.h"
16 #include "llvm/MC/MCRegisterInfo.h"
17 #include "llvm/MC/MCStreamer.h"
18
19 namespace llvm {
20
21 struct MipsABIFlagsSection;
22
23 class MipsTargetStreamer : public MCTargetStreamer {
24 public:
25   MipsTargetStreamer(MCStreamer &S);
26   virtual void emitDirectiveSetMicroMips();
27   virtual void emitDirectiveSetNoMicroMips();
28   virtual void emitDirectiveSetMips16();
29   virtual void emitDirectiveSetNoMips16();
30
31   virtual void emitDirectiveSetReorder();
32   virtual void emitDirectiveSetNoReorder();
33   virtual void emitDirectiveSetMacro();
34   virtual void emitDirectiveSetNoMacro();
35   virtual void emitDirectiveSetMsa();
36   virtual void emitDirectiveSetNoMsa();
37   virtual void emitDirectiveSetAt();
38   virtual void emitDirectiveSetAtWithArg(unsigned RegNo);
39   virtual void emitDirectiveSetNoAt();
40   virtual void emitDirectiveEnd(StringRef Name);
41
42   virtual void emitDirectiveEnt(const MCSymbol &Symbol);
43   virtual void emitDirectiveAbiCalls();
44   virtual void emitDirectiveNaN2008();
45   virtual void emitDirectiveNaNLegacy();
46   virtual void emitDirectiveOptionPic0();
47   virtual void emitDirectiveOptionPic2();
48   virtual void emitFrame(unsigned StackReg, unsigned StackSize,
49                          unsigned ReturnReg);
50   virtual void emitMask(unsigned CPUBitmask, int CPUTopSavedRegOff);
51   virtual void emitFMask(unsigned FPUBitmask, int FPUTopSavedRegOff);
52
53   virtual void emitDirectiveSetArch(StringRef Arch);
54   virtual void emitDirectiveSetMips0();
55   virtual void emitDirectiveSetMips1();
56   virtual void emitDirectiveSetMips2();
57   virtual void emitDirectiveSetMips3();
58   virtual void emitDirectiveSetMips4();
59   virtual void emitDirectiveSetMips5();
60   virtual void emitDirectiveSetMips32();
61   virtual void emitDirectiveSetMips32R2();
62   virtual void emitDirectiveSetMips32R6();
63   virtual void emitDirectiveSetMips64();
64   virtual void emitDirectiveSetMips64R2();
65   virtual void emitDirectiveSetMips64R6();
66   virtual void emitDirectiveSetDsp();
67   virtual void emitDirectiveSetNoDsp();
68   virtual void emitDirectiveSetPop();
69   virtual void emitDirectiveSetPush();
70
71   // PIC support
72   virtual void emitDirectiveCpLoad(unsigned RegNo);
73   virtual void emitDirectiveCpsetup(unsigned RegNo, int RegOrOffset,
74                                     const MCSymbol &Sym, bool IsReg);
75
76   /// Emit a '.module fp=value' directive using the given values.
77   /// Updates the .MIPS.abiflags section
78   virtual void emitDirectiveModuleFP(MipsABIFlagsSection::FpABIKind Value,
79                                      bool Is32BitABI) {
80     ABIFlagsSection.setFpABI(Value, Is32BitABI);
81   }
82
83   /// Emit a '.module fp=value' directive using the current values of the
84   /// .MIPS.abiflags section.
85   void emitDirectiveModuleFP() {
86     emitDirectiveModuleFP(ABIFlagsSection.getFpABI(),
87                           ABIFlagsSection.Is32BitABI);
88   }
89
90   virtual void emitDirectiveModuleOddSPReg(bool Enabled, bool IsO32ABI);
91   virtual void emitDirectiveSetFp(MipsABIFlagsSection::FpABIKind Value){};
92   virtual void emitMipsAbiFlags(){};
93   void forbidModuleDirective() { ModuleDirectiveAllowed = false; }
94   bool isModuleDirectiveAllowed() { return ModuleDirectiveAllowed; }
95
96   // This method enables template classes to set internal abi flags
97   // structure values.
98   template <class PredicateLibrary>
99   void updateABIInfo(const PredicateLibrary &P) {
100     ABI = &P.getABI();
101     ABIFlagsSection.setAllFromPredicates(P);
102   }
103
104   MipsABIFlagsSection &getABIFlagsSection() { return ABIFlagsSection; }
105   const MipsABIInfo &getABI() const {
106     assert(ABI && "ABI hasn't been set!");
107     return *ABI;
108   }
109
110 protected:
111   const MipsABIInfo *ABI;
112   MipsABIFlagsSection ABIFlagsSection;
113
114   bool GPRInfoSet;
115   unsigned GPRBitMask;
116   int GPROffset;
117
118   bool FPRInfoSet;
119   unsigned FPRBitMask;
120   int FPROffset;
121
122   bool FrameInfoSet;
123   int FrameOffset;
124   unsigned FrameReg;
125   unsigned ReturnReg;
126
127 private:
128   bool ModuleDirectiveAllowed;
129 };
130
131 // This part is for ascii assembly output
132 class MipsTargetAsmStreamer : public MipsTargetStreamer {
133   formatted_raw_ostream &OS;
134
135 public:
136   MipsTargetAsmStreamer(MCStreamer &S, formatted_raw_ostream &OS);
137   void emitDirectiveSetMicroMips() override;
138   void emitDirectiveSetNoMicroMips() override;
139   void emitDirectiveSetMips16() override;
140   void emitDirectiveSetNoMips16() override;
141
142   void emitDirectiveSetReorder() override;
143   void emitDirectiveSetNoReorder() override;
144   void emitDirectiveSetMacro() override;
145   void emitDirectiveSetNoMacro() override;
146   void emitDirectiveSetMsa() override;
147   void emitDirectiveSetNoMsa() override;
148   void emitDirectiveSetAt() override;
149   void emitDirectiveSetAtWithArg(unsigned RegNo) override;
150   void emitDirectiveSetNoAt() override;
151   void emitDirectiveEnd(StringRef Name) override;
152
153   void emitDirectiveEnt(const MCSymbol &Symbol) override;
154   void emitDirectiveAbiCalls() override;
155   void emitDirectiveNaN2008() override;
156   void emitDirectiveNaNLegacy() override;
157   void emitDirectiveOptionPic0() override;
158   void emitDirectiveOptionPic2() override;
159   void emitFrame(unsigned StackReg, unsigned StackSize,
160                  unsigned ReturnReg) override;
161   void emitMask(unsigned CPUBitmask, int CPUTopSavedRegOff) override;
162   void emitFMask(unsigned FPUBitmask, int FPUTopSavedRegOff) override;
163
164   void emitDirectiveSetArch(StringRef Arch) override;
165   void emitDirectiveSetMips0() override;
166   void emitDirectiveSetMips1() override;
167   void emitDirectiveSetMips2() override;
168   void emitDirectiveSetMips3() override;
169   void emitDirectiveSetMips4() override;
170   void emitDirectiveSetMips5() override;
171   void emitDirectiveSetMips32() override;
172   void emitDirectiveSetMips32R2() override;
173   void emitDirectiveSetMips32R6() override;
174   void emitDirectiveSetMips64() override;
175   void emitDirectiveSetMips64R2() override;
176   void emitDirectiveSetMips64R6() override;
177   void emitDirectiveSetDsp() override;
178   void emitDirectiveSetNoDsp() override;
179   void emitDirectiveSetPop() override;
180   void emitDirectiveSetPush() override;
181
182   // PIC support
183   void emitDirectiveCpLoad(unsigned RegNo) override;
184   void emitDirectiveCpsetup(unsigned RegNo, int RegOrOffset,
185                             const MCSymbol &Sym, bool IsReg) override;
186
187   // ABI Flags
188   void emitDirectiveModuleFP(MipsABIFlagsSection::FpABIKind Value,
189                              bool Is32BitABI) override;
190   void emitDirectiveModuleOddSPReg(bool Enabled, bool IsO32ABI) override;
191   void emitDirectiveSetFp(MipsABIFlagsSection::FpABIKind Value) override;
192   void emitMipsAbiFlags() override;
193 };
194
195 // This part is for ELF object output
196 class MipsTargetELFStreamer : public MipsTargetStreamer {
197   bool MicroMipsEnabled;
198   const MCSubtargetInfo &STI;
199   bool Pic;
200
201 public:
202   bool isMicroMipsEnabled() const { return MicroMipsEnabled; }
203   MCELFStreamer &getStreamer();
204   MipsTargetELFStreamer(MCStreamer &S, const MCSubtargetInfo &STI);
205
206   void emitLabel(MCSymbol *Symbol) override;
207   void emitAssignment(MCSymbol *Symbol, const MCExpr *Value) override;
208   void finish() override;
209
210   void emitDirectiveSetMicroMips() override;
211   void emitDirectiveSetNoMicroMips() override;
212   void emitDirectiveSetMips16() override;
213
214   void emitDirectiveSetNoReorder() override;
215   void emitDirectiveEnd(StringRef Name) override;
216
217   void emitDirectiveEnt(const MCSymbol &Symbol) override;
218   void emitDirectiveAbiCalls() override;
219   void emitDirectiveNaN2008() override;
220   void emitDirectiveNaNLegacy() override;
221   void emitDirectiveOptionPic0() override;
222   void emitDirectiveOptionPic2() override;
223   void emitFrame(unsigned StackReg, unsigned StackSize,
224                  unsigned ReturnReg) override;
225   void emitMask(unsigned CPUBitmask, int CPUTopSavedRegOff) override;
226   void emitFMask(unsigned FPUBitmask, int FPUTopSavedRegOff) override;
227
228   // PIC support
229   void emitDirectiveCpLoad(unsigned RegNo) override;
230   void emitDirectiveCpsetup(unsigned RegNo, int RegOrOffset,
231                             const MCSymbol &Sym, bool IsReg) override;
232
233   // ABI Flags
234   void emitDirectiveModuleOddSPReg(bool Enabled, bool IsO32ABI) override;
235   void emitMipsAbiFlags() override;
236 };
237 }
238 #endif