Add a InitSections method to the streamer interface.
[oota-llvm.git] / include / llvm / MC / MCStreamer.h
1 //===- MCStreamer.h - High-level Streaming Machine Code Output --*- 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 // This file declares the MCStreamer class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #ifndef LLVM_MC_MCSTREAMER_H
15 #define LLVM_MC_MCSTREAMER_H
16
17 #include "llvm/System/DataTypes.h"
18 #include "llvm/MC/MCDirectives.h"
19
20 namespace llvm {
21   class MCAsmInfo;
22   class MCCodeEmitter;
23   class MCContext;
24   class MCExpr;
25   class MCInst;
26   class MCInstPrinter;
27   class MCSection;
28   class MCSymbol;
29   class StringRef;
30   class TargetAsmBackend;
31   class Twine;
32   class raw_ostream;
33   class formatted_raw_ostream;
34
35   /// MCStreamer - Streaming machine code generation interface.  This interface
36   /// is intended to provide a programatic interface that is very similar to the
37   /// level that an assembler .s file provides.  It has callbacks to emit bytes,
38   /// handle directives, etc.  The implementation of this interface retains
39   /// state to know what the current section is etc.
40   ///
41   /// There are multiple implementations of this interface: one for writing out
42   /// a .s file, and implementations that write out .o files of various formats.
43   ///
44   class MCStreamer {
45     MCContext &Context;
46
47     MCStreamer(const MCStreamer&); // DO NOT IMPLEMENT
48     MCStreamer &operator=(const MCStreamer&); // DO NOT IMPLEMENT
49
50   protected:
51     MCStreamer(MCContext &Ctx);
52
53     /// CurSection - This is the current section code is being emitted to, it is
54     /// kept up to date by SwitchSection.
55     const MCSection *CurSection;
56
57     /// PrevSection - This is the previous section code is being emitted to, it
58     /// is kept up to date by SwitchSection.
59     const MCSection *PrevSection;
60
61   public:
62     virtual ~MCStreamer();
63
64     MCContext &getContext() const { return Context; }
65
66     /// @name Assembly File Formatting.
67     /// @{
68
69     /// isVerboseAsm - Return true if this streamer supports verbose assembly
70     /// and if it is enabled.
71     virtual bool isVerboseAsm() const { return false; }
72
73     /// hasRawTextSupport - Return true if this asm streamer supports emitting
74     /// unformatted text to the .s file with EmitRawText.
75     virtual bool hasRawTextSupport() const { return false; }
76
77     /// AddComment - Add a comment that can be emitted to the generated .s
78     /// file if applicable as a QoI issue to make the output of the compiler
79     /// more readable.  This only affects the MCAsmStreamer, and only when
80     /// verbose assembly output is enabled.
81     ///
82     /// If the comment includes embedded \n's, they will each get the comment
83     /// prefix as appropriate.  The added comment should not end with a \n.
84     virtual void AddComment(const Twine &T) {}
85
86     /// GetCommentOS - Return a raw_ostream that comments can be written to.
87     /// Unlike AddComment, you are required to terminate comments with \n if you
88     /// use this method.
89     virtual raw_ostream &GetCommentOS();
90
91     /// AddBlankLine - Emit a blank line to a .s file to pretty it up.
92     virtual void AddBlankLine() {}
93
94     /// @}
95
96     /// @name Symbol & Section Management
97     /// @{
98
99     /// getCurrentSection - Return the current section that the streamer is
100     /// emitting code to.
101     const MCSection *getCurrentSection() const { return CurSection; }
102
103     /// getPreviousSection - Return the previous section that the streamer is
104     /// emitting code to.
105     const MCSection *getPreviousSection() const { return PrevSection; }
106
107     /// SwitchSection - Set the current section where code is being emitted to
108     /// @p Section.  This is required to update CurSection.
109     ///
110     /// This corresponds to assembler directives like .section, .text, etc.
111     virtual void SwitchSection(const MCSection *Section) = 0;
112
113     /// InitSections - Create the default sections and set the initial one.
114     virtual void InitSections() = 0;
115
116     /// EmitLabel - Emit a label for @p Symbol into the current section.
117     ///
118     /// This corresponds to an assembler statement such as:
119     ///   foo:
120     ///
121     /// @param Symbol - The symbol to emit. A given symbol should only be
122     /// emitted as a label once, and symbols emitted as a label should never be
123     /// used in an assignment.
124     virtual void EmitLabel(MCSymbol *Symbol) = 0;
125
126     /// EmitAssemblerFlag - Note in the output the specified @p Flag
127     virtual void EmitAssemblerFlag(MCAssemblerFlag Flag) = 0;
128
129     /// EmitAssignment - Emit an assignment of @p Value to @p Symbol.
130     ///
131     /// This corresponds to an assembler statement such as:
132     ///  symbol = value
133     ///
134     /// The assignment generates no code, but has the side effect of binding the
135     /// value in the current context. For the assembly streamer, this prints the
136     /// binding into the .s file.
137     ///
138     /// @param Symbol - The symbol being assigned to.
139     /// @param Value - The value for the symbol.
140     virtual void EmitAssignment(MCSymbol *Symbol, const MCExpr *Value) = 0;
141
142     /// EmitSymbolAttribute - Add the given @p Attribute to @p Symbol.
143     virtual void EmitSymbolAttribute(MCSymbol *Symbol,
144                                      MCSymbolAttr Attribute) = 0;
145
146     /// EmitSymbolDesc - Set the @p DescValue for the @p Symbol.
147     ///
148     /// @param Symbol - The symbol to have its n_desc field set.
149     /// @param DescValue - The value to set into the n_desc field.
150     virtual void EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) = 0;
151
152     /// BeginCOFFSymbolDef - Start emitting COFF symbol definition
153     ///
154     /// @param Symbol - The symbol to have its External & Type fields set.
155     virtual void BeginCOFFSymbolDef(const MCSymbol *Symbol) = 0;
156
157     /// EmitCOFFSymbolStorageClass - Emit the storage class of the symbol.
158     ///
159     /// @param StorageClass - The storage class the symbol should have.
160     virtual void EmitCOFFSymbolStorageClass(int StorageClass) = 0;
161
162     /// EmitCOFFSymbolType - Emit the type of the symbol.
163     ///
164     /// @param Type - A COFF type identifier (see COFF::SymbolType in X86COFF.h)
165     virtual void EmitCOFFSymbolType(int Type) = 0;
166
167     /// EndCOFFSymbolDef - Marks the end of the symbol definition.
168     virtual void EndCOFFSymbolDef() = 0;
169
170     /// EmitELFSize - Emit an ELF .size directive.
171     ///
172     /// This corresponds to an assembler statement such as:
173     ///  .size symbol, expression
174     ///
175     virtual void EmitELFSize(MCSymbol *Symbol, const MCExpr *Value) = 0;
176
177     /// EmitCommonSymbol - Emit a common symbol.
178     ///
179     /// @param Symbol - The common symbol to emit.
180     /// @param Size - The size of the common symbol.
181     /// @param ByteAlignment - The alignment of the symbol if
182     /// non-zero. This must be a power of 2.
183     virtual void EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size,
184                                   unsigned ByteAlignment) = 0;
185
186     /// EmitLocalCommonSymbol - Emit a local common (.lcomm) symbol.
187     ///
188     /// @param Symbol - The common symbol to emit.
189     /// @param Size - The size of the common symbol.
190     virtual void EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size) = 0;
191
192     /// EmitZerofill - Emit the zerofill section and an optional symbol.
193     ///
194     /// @param Section - The zerofill section to create and or to put the symbol
195     /// @param Symbol - The zerofill symbol to emit, if non-NULL.
196     /// @param Size - The size of the zerofill symbol.
197     /// @param ByteAlignment - The alignment of the zerofill symbol if
198     /// non-zero. This must be a power of 2 on some targets.
199     virtual void EmitZerofill(const MCSection *Section, MCSymbol *Symbol = 0,
200                               unsigned Size = 0,unsigned ByteAlignment = 0) = 0;
201
202     /// EmitTBSSSymbol - Emit a thread local bss (.tbss) symbol.
203     ///
204     /// @param Section - The thread local common section.
205     /// @param Symbol - The thread local common symbol to emit.
206     /// @param Size - The size of the symbol.
207     /// @param ByteAlignment - The alignment of the thread local common symbol
208     /// if non-zero.  This must be a power of 2 on some targets.
209     virtual void EmitTBSSSymbol(const MCSection *Section, MCSymbol *Symbol,
210                                 uint64_t Size, unsigned ByteAlignment = 0) = 0;
211     /// @}
212     /// @name Generating Data
213     /// @{
214
215     /// EmitBytes - Emit the bytes in \arg Data into the output.
216     ///
217     /// This is used to implement assembler directives such as .byte, .ascii,
218     /// etc.
219     virtual void EmitBytes(StringRef Data, unsigned AddrSpace) = 0;
220
221     /// EmitValue - Emit the expression @p Value into the output as a native
222     /// integer of the given @p Size bytes.
223     ///
224     /// This is used to implement assembler directives such as .word, .quad,
225     /// etc.
226     ///
227     /// @param Value - The value to emit.
228     /// @param Size - The size of the integer (in bytes) to emit. This must
229     /// match a native machine width.
230     virtual void EmitValue(const MCExpr *Value, unsigned Size,
231                            unsigned AddrSpace = 0) = 0;
232
233     /// EmitIntValue - Special case of EmitValue that avoids the client having
234     /// to pass in a MCExpr for constant integers.
235     virtual void EmitIntValue(uint64_t Value, unsigned Size,
236                               unsigned AddrSpace = 0);
237
238     /// EmitSymbolValue - Special case of EmitValue that avoids the client
239     /// having to pass in a MCExpr for MCSymbols.
240     virtual void EmitSymbolValue(const MCSymbol *Sym, unsigned Size,
241                                  unsigned AddrSpace);
242
243     /// EmitGPRel32Value - Emit the expression @p Value into the output as a
244     /// gprel32 (32-bit GP relative) value.
245     ///
246     /// This is used to implement assembler directives such as .gprel32 on
247     /// targets that support them.
248     virtual void EmitGPRel32Value(const MCExpr *Value) = 0;
249
250     /// EmitFill - Emit NumBytes bytes worth of the value specified by
251     /// FillValue.  This implements directives such as '.space'.
252     virtual void EmitFill(uint64_t NumBytes, uint8_t FillValue,
253                           unsigned AddrSpace);
254
255     /// EmitZeros - Emit NumBytes worth of zeros.  This is a convenience
256     /// function that just wraps EmitFill.
257     void EmitZeros(uint64_t NumBytes, unsigned AddrSpace) {
258       EmitFill(NumBytes, 0, AddrSpace);
259     }
260
261
262     /// EmitValueToAlignment - Emit some number of copies of @p Value until
263     /// the byte alignment @p ByteAlignment is reached.
264     ///
265     /// If the number of bytes need to emit for the alignment is not a multiple
266     /// of @p ValueSize, then the contents of the emitted fill bytes is
267     /// undefined.
268     ///
269     /// This used to implement the .align assembler directive.
270     ///
271     /// @param ByteAlignment - The alignment to reach. This must be a power of
272     /// two on some targets.
273     /// @param Value - The value to use when filling bytes.
274     /// @param ValueSize - The size of the integer (in bytes) to emit for
275     /// @p Value. This must match a native machine width.
276     /// @param MaxBytesToEmit - The maximum numbers of bytes to emit, or 0. If
277     /// the alignment cannot be reached in this many bytes, no bytes are
278     /// emitted.
279     virtual void EmitValueToAlignment(unsigned ByteAlignment, int64_t Value = 0,
280                                       unsigned ValueSize = 1,
281                                       unsigned MaxBytesToEmit = 0) = 0;
282
283     /// EmitCodeAlignment - Emit nops until the byte alignment @p ByteAlignment
284     /// is reached.
285     ///
286     /// This used to align code where the alignment bytes may be executed.  This
287     /// can emit different bytes for different sizes to optimize execution.
288     ///
289     /// @param ByteAlignment - The alignment to reach. This must be a power of
290     /// two on some targets.
291     /// @param MaxBytesToEmit - The maximum numbers of bytes to emit, or 0. If
292     /// the alignment cannot be reached in this many bytes, no bytes are
293     /// emitted.
294     virtual void EmitCodeAlignment(unsigned ByteAlignment,
295                                    unsigned MaxBytesToEmit = 0) = 0;
296
297     /// EmitValueToOffset - Emit some number of copies of @p Value until the
298     /// byte offset @p Offset is reached.
299     ///
300     /// This is used to implement assembler directives such as .org.
301     ///
302     /// @param Offset - The offset to reach. This may be an expression, but the
303     /// expression must be associated with the current section.
304     /// @param Value - The value to use when filling bytes.
305     virtual void EmitValueToOffset(const MCExpr *Offset,
306                                    unsigned char Value = 0) = 0;
307
308     /// @}
309
310     /// EmitFileDirective - Switch to a new logical file.  This is used to
311     /// implement the '.file "foo.c"' assembler directive.
312     virtual void EmitFileDirective(StringRef Filename) = 0;
313
314     /// EmitDwarfFileDirective - Associate a filename with a specified logical
315     /// file number.  This implements the DWARF2 '.file 4 "foo.c"' assembler
316     /// directive.
317     virtual void EmitDwarfFileDirective(unsigned FileNo,StringRef Filename) = 0;
318
319     /// EmitInstruction - Emit the given @p Instruction into the current
320     /// section.
321     virtual void EmitInstruction(const MCInst &Inst) = 0;
322
323     /// EmitRawText - If this file is backed by a assembly streamer, this dumps
324     /// the specified string in the output .s file.  This capability is
325     /// indicated by the hasRawTextSupport() predicate.  By default this aborts.
326     virtual void EmitRawText(StringRef String);
327     void EmitRawText(const Twine &String);
328
329     /// Finish - Finish emission of machine code.
330     virtual void Finish() = 0;
331   };
332
333   /// createNullStreamer - Create a dummy machine code streamer, which does
334   /// nothing. This is useful for timing the assembler front end.
335   MCStreamer *createNullStreamer(MCContext &Ctx);
336
337   /// createAsmStreamer - Create a machine code streamer which will print out
338   /// assembly for the native target, suitable for compiling with a native
339   /// assembler.
340   ///
341   /// \param InstPrint - If given, the instruction printer to use. If not given
342   /// the MCInst representation will be printed.  This method takes ownership of
343   /// InstPrint.
344   ///
345   /// \param CE - If given, a code emitter to use to show the instruction
346   /// encoding inline with the assembly. This method takes ownership of \arg CE.
347   ///
348   /// \param ShowInst - Whether to show the MCInst representation inline with
349   /// the assembly.
350   MCStreamer *createAsmStreamer(MCContext &Ctx, formatted_raw_ostream &OS,
351                                 bool isLittleEndian, bool isVerboseAsm,
352                                 MCInstPrinter *InstPrint = 0,
353                                 MCCodeEmitter *CE = 0,
354                                 bool ShowInst = false);
355
356   /// createMachOStreamer - Create a machine code streamer which will generate
357   /// Mach-O format object files.
358   ///
359   /// Takes ownership of \arg TAB and \arg CE.
360   MCStreamer *createMachOStreamer(MCContext &Ctx, TargetAsmBackend &TAB,
361                                   raw_ostream &OS, MCCodeEmitter *CE,
362                                   bool RelaxAll = false);
363
364   /// createWinCOFFStreamer - Create a machine code streamer which will
365   /// generate Microsoft COFF format object files.
366   ///
367   /// Takes ownership of \arg TAB and \arg CE.
368   MCStreamer *createWinCOFFStreamer(MCContext &Ctx,
369                                     TargetAsmBackend &TAB,
370                                     MCCodeEmitter &CE, raw_ostream &OS,
371                                     bool RelaxAll = false);
372
373   /// createELFStreamer - Create a machine code streamer which will generate
374   /// ELF format object files.
375   MCStreamer *createELFStreamer(MCContext &Ctx, TargetAsmBackend &TAB,
376                                 raw_ostream &OS, MCCodeEmitter *CE,
377                                 bool RelaxAll = false);
378
379   /// createLoggingStreamer - Create a machine code streamer which just logs the
380   /// API calls and then dispatches to another streamer.
381   ///
382   /// The new streamer takes ownership of the \arg Child.
383   MCStreamer *createLoggingStreamer(MCStreamer *Child, raw_ostream &OS);
384
385 } // end namespace llvm
386
387 #endif