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