Add SPAdj parameter to account for call frame setup SP adjustment.
[oota-llvm.git] / include / llvm / Target / MRegisterInfo.h
1 //===- Target/MRegisterInfo.h - Target Register Information -----*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file was developed by the LLVM research group and is distributed under
6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file describes an abstract interface used to get information about a
11 // target machines register file.  This information is used for a variety of
12 // purposed, especially register allocation.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #ifndef LLVM_TARGET_MREGISTERINFO_H
17 #define LLVM_TARGET_MREGISTERINFO_H
18
19 #include "llvm/CodeGen/MachineBasicBlock.h"
20 #include "llvm/CodeGen/ValueTypes.h"
21 #include <cassert>
22 #include <functional>
23
24 namespace llvm {
25
26 class BitVector;
27 class CalleeSavedInfo;
28 class MachineFunction;
29 class MachineInstr;
30 class MachineLocation;
31 class MachineMove;
32 class RegScavenger;
33 class TargetRegisterClass;
34 class Type;
35
36 /// TargetRegisterDesc - This record contains all of the information known about
37 /// a particular register.  The AliasSet field (if not null) contains a pointer
38 /// to a Zero terminated array of registers that this register aliases.  This is
39 /// needed for architectures like X86 which have AL alias AX alias EAX.
40 /// Registers that this does not apply to simply should set this to null.
41 /// The SubRegs field is a zero terminated array of registers that are
42 /// sub-registers of the specific register, e.g. AL, AH are sub-registers of AX.
43 /// The SuperRegs field is a zero terminated array of registers that are
44 /// super-registers of the specific register, e.g. RAX, EAX, are super-registers
45 /// of AX.
46 ///
47 struct TargetRegisterDesc {
48   const char     *Name;         // Assembly language name for the register
49   const unsigned *AliasSet;     // Register Alias Set, described above
50   const unsigned *SubRegs;      // Sub-register set, described above
51   const unsigned *SuperRegs;    // Super-register set, described above
52 };
53
54 class TargetRegisterClass {
55 public:
56   typedef const unsigned* iterator;
57   typedef const unsigned* const_iterator;
58
59   typedef const MVT::ValueType* vt_iterator;
60   typedef const TargetRegisterClass* const * sc_iterator;
61 private:
62   unsigned ID;
63   bool  isSubClass;
64   const vt_iterator VTs;
65   const sc_iterator SubClasses;
66   const sc_iterator SuperClasses;
67   const unsigned RegSize, Alignment;    // Size & Alignment of register in bytes
68   const iterator RegsBegin, RegsEnd;
69 public:
70   TargetRegisterClass(unsigned id,
71                       const MVT::ValueType *vts,
72                       const TargetRegisterClass * const *subcs,
73                       const TargetRegisterClass * const *supcs,
74                       unsigned RS, unsigned Al, iterator RB, iterator RE)
75     : ID(id), VTs(vts), SubClasses(subcs), SuperClasses(supcs),
76     RegSize(RS), Alignment(Al), RegsBegin(RB), RegsEnd(RE) {}
77   virtual ~TargetRegisterClass() {}     // Allow subclasses
78   
79   /// getID() - Return the register class ID number.
80   ///
81   unsigned getID() const { return ID; }
82   
83   /// begin/end - Return all of the registers in this class.
84   ///
85   iterator       begin() const { return RegsBegin; }
86   iterator         end() const { return RegsEnd; }
87
88   /// getNumRegs - Return the number of registers in this class.
89   ///
90   unsigned getNumRegs() const { return RegsEnd-RegsBegin; }
91
92   /// getRegister - Return the specified register in the class.
93   ///
94   unsigned getRegister(unsigned i) const {
95     assert(i < getNumRegs() && "Register number out of range!");
96     return RegsBegin[i];
97   }
98
99   /// contains - Return true if the specified register is included in this
100   /// register class.
101   bool contains(unsigned Reg) const {
102     for (iterator I = begin(), E = end(); I != E; ++I)
103       if (*I == Reg) return true;
104     return false;
105   }
106
107   /// hasType - return true if this TargetRegisterClass has the ValueType vt.
108   ///
109   bool hasType(MVT::ValueType vt) const {
110     for(int i = 0; VTs[i] != MVT::Other; ++i)
111       if (VTs[i] == vt)
112         return true;
113     return false;
114   }
115   
116   /// vt_begin / vt_end - Loop over all of the value types that can be
117   /// represented by values in this register class.
118   vt_iterator vt_begin() const {
119     return VTs;
120   }
121
122   vt_iterator vt_end() const {
123     vt_iterator I = VTs;
124     while (*I != MVT::Other) ++I;
125     return I;
126   }
127
128   /// hasSubRegClass - return true if the specified TargetRegisterClass is a
129   /// sub-register class of this TargetRegisterClass.
130   bool hasSubRegClass(const TargetRegisterClass *cs) const {
131     for (int i = 0; SubClasses[i] != NULL; ++i) 
132       if (SubClasses[i] == cs)
133         return true;
134     return false;
135   }
136
137   /// subclasses_begin / subclasses_end - Loop over all of the sub-classes of
138   /// this register class.
139   sc_iterator subclasses_begin() const {
140     return SubClasses;
141   }
142   
143   sc_iterator subclasses_end() const {
144     sc_iterator I = SubClasses;
145     while (*I != NULL) ++I;
146     return I;
147   }
148   
149   /// hasSuperRegClass - return true if the specified TargetRegisterClass is a
150   /// super-register class of this TargetRegisterClass.
151   bool hasSuperRegClass(const TargetRegisterClass *cs) const {
152     for (int i = 0; SuperClasses[i] != NULL; ++i) 
153       if (SuperClasses[i] == cs)
154         return true;
155     return false;
156   }
157
158   /// superclasses_begin / superclasses_end - Loop over all of the super-classes
159   /// of this register class.
160   sc_iterator superclasses_begin() const {
161     return SuperClasses;
162   }
163   
164   sc_iterator superclasses_end() const {
165     sc_iterator I = SuperClasses;
166     while (*I != NULL) ++I;
167     return I;
168   }
169   
170   /// allocation_order_begin/end - These methods define a range of registers
171   /// which specify the registers in this class that are valid to register
172   /// allocate, and the preferred order to allocate them in.  For example,
173   /// callee saved registers should be at the end of the list, because it is
174   /// cheaper to allocate caller saved registers.
175   ///
176   /// These methods take a MachineFunction argument, which can be used to tune
177   /// the allocatable registers based on the characteristics of the function.
178   /// One simple example is that the frame pointer register can be used if
179   /// frame-pointer-elimination is performed.
180   ///
181   /// By default, these methods return all registers in the class.
182   ///
183   virtual iterator allocation_order_begin(const MachineFunction &MF) const {
184     return begin();
185   }
186   virtual iterator allocation_order_end(const MachineFunction &MF)   const {
187     return end();
188   }
189
190
191
192   /// getSize - Return the size of the register in bytes, which is also the size
193   /// of a stack slot allocated to hold a spilled copy of this register.
194   unsigned getSize() const { return RegSize; }
195
196   /// getAlignment - Return the minimum required alignment for a register of
197   /// this class.
198   unsigned getAlignment() const { return Alignment; }
199 };
200
201
202 /// MRegisterInfo base class - We assume that the target defines a static array
203 /// of TargetRegisterDesc objects that represent all of the machine registers
204 /// that the target has.  As such, we simply have to track a pointer to this
205 /// array so that we can turn register number into a register descriptor.
206 ///
207 class MRegisterInfo {
208 public:
209   typedef const TargetRegisterClass * const * regclass_iterator;
210 private:
211   const TargetRegisterDesc *Desc;             // Pointer to the descriptor array
212   unsigned NumRegs;                           // Number of entries in the array
213
214   regclass_iterator RegClassBegin, RegClassEnd;   // List of regclasses
215
216   int CallFrameSetupOpcode, CallFrameDestroyOpcode;
217 protected:
218   MRegisterInfo(const TargetRegisterDesc *D, unsigned NR,
219                 regclass_iterator RegClassBegin, regclass_iterator RegClassEnd,
220                 int CallFrameSetupOpcode = -1, int CallFrameDestroyOpcode = -1);
221   virtual ~MRegisterInfo();
222 public:
223
224   enum {                        // Define some target independent constants
225     /// NoRegister - This physical register is not a real target register.  It
226     /// is useful as a sentinal.
227     NoRegister = 0,
228
229     /// FirstVirtualRegister - This is the first register number that is
230     /// considered to be a 'virtual' register, which is part of the SSA
231     /// namespace.  This must be the same for all targets, which means that each
232     /// target is limited to 1024 registers.
233     FirstVirtualRegister = 1024
234   };
235
236   /// isPhysicalRegister - Return true if the specified register number is in
237   /// the physical register namespace.
238   static bool isPhysicalRegister(unsigned Reg) {
239     assert(Reg && "this is not a register!");
240     return Reg < FirstVirtualRegister;
241   }
242
243   /// isVirtualRegister - Return true if the specified register number is in
244   /// the virtual register namespace.
245   static bool isVirtualRegister(unsigned Reg) {
246     assert(Reg && "this is not a register!");
247     return Reg >= FirstVirtualRegister;
248   }
249
250   /// getAllocatableSet - Returns a bitset indexed by register number
251   /// indicating if a register is allocatable or not. If a register class is
252   /// specified, returns the subset for the class.
253   BitVector getAllocatableSet(MachineFunction &MF,
254                               const TargetRegisterClass *RC = NULL) const;
255
256   const TargetRegisterDesc &operator[](unsigned RegNo) const {
257     assert(RegNo < NumRegs &&
258            "Attempting to access record for invalid register number!");
259     return Desc[RegNo];
260   }
261
262   /// Provide a get method, equivalent to [], but more useful if we have a
263   /// pointer to this object.
264   ///
265   const TargetRegisterDesc &get(unsigned RegNo) const {
266     return operator[](RegNo);
267   }
268
269   /// getAliasSet - Return the set of registers aliased by the specified
270   /// register, or a null list of there are none.  The list returned is zero
271   /// terminated.
272   ///
273   const unsigned *getAliasSet(unsigned RegNo) const {
274     return get(RegNo).AliasSet;
275   }
276
277   /// getSubRegisters - Return the set of registers that are sub-registers of
278   /// the specified register, or a null list of there are none. The list
279   /// returned is zero terminated.
280   ///
281   const unsigned *getSubRegisters(unsigned RegNo) const {
282     return get(RegNo).SubRegs;
283   }
284
285   /// getSuperRegisters - Return the set of registers that are super-registers
286   /// of the specified register, or a null list of there are none. The list
287   /// returned is zero terminated.
288   ///
289   const unsigned *getSuperRegisters(unsigned RegNo) const {
290     return get(RegNo).SuperRegs;
291   }
292
293   /// getName - Return the symbolic target specific name for the specified
294   /// physical register.
295   const char *getName(unsigned RegNo) const {
296     return get(RegNo).Name;
297   }
298
299   /// getNumRegs - Return the number of registers this target has
300   /// (useful for sizing arrays holding per register information)
301   unsigned getNumRegs() const {
302     return NumRegs;
303   }
304
305   /// areAliases - Returns true if the two registers alias each other,
306   /// false otherwise
307   bool areAliases(unsigned regA, unsigned regB) const {
308     for (const unsigned *Alias = getAliasSet(regA); *Alias; ++Alias)
309       if (*Alias == regB) return true;
310     return false;
311   }
312
313   /// regsOverlap - Returns true if the two registers are equal or alias
314   /// each other. The registers may be virtual register.
315   bool regsOverlap(unsigned regA, unsigned regB) const {
316     if (regA == regB)
317       return true;
318
319     if (isVirtualRegister(regA) || isVirtualRegister(regB))
320       return false;
321     return areAliases(regA, regB);
322   }
323
324   /// isSubRegister - Returns true if regB is a sub-register of regA.
325   ///
326   bool isSubRegister(unsigned regA, unsigned regB) const {
327     for (const unsigned *SR = getSubRegisters(regA); *SR; ++SR)
328       if (*SR == regB) return true;
329     return false;
330   }
331
332   /// isSuperRegister - Returns true if regB is a super-register of regA.
333   ///
334   bool isSuperRegister(unsigned regA, unsigned regB) const {
335     for (const unsigned *SR = getSuperRegisters(regA); *SR; ++SR)
336       if (*SR == regB) return true;
337     return false;
338   }
339
340   /// getCalleeSavedRegs - Return a null-terminated list of all of the
341   /// callee saved registers on this target. The register should be in the
342   /// order of desired callee-save stack frame offset. The first register is
343   /// closed to the incoming stack pointer if stack grows down, and vice versa.
344   virtual const unsigned* getCalleeSavedRegs() const = 0;
345
346   /// getCalleeSavedRegClasses - Return a null-terminated list of the preferred
347   /// register classes to spill each callee saved register with.  The order and
348   /// length of this list match the getCalleeSaveRegs() list.
349   virtual const TargetRegisterClass* const *getCalleeSavedRegClasses() const =0;
350
351   /// getReservedRegs - Returns a bitset indexed by physical register number
352   /// indicating if a register is a special register that has particular uses and
353   /// should be considered unavailable at all times, e.g. SP, RA. This is used by
354   /// register scavenger to determine what registers are free.
355   virtual BitVector getReservedRegs(const MachineFunction &MF) const = 0;
356
357   /// getSubReg - Returns the physical register number of sub-register "Index"
358   /// for physical register RegNo.
359   virtual unsigned getSubReg(unsigned RegNo, unsigned Index) const = 0;
360
361   //===--------------------------------------------------------------------===//
362   // Register Class Information
363   //
364
365   /// Register class iterators
366   ///
367   regclass_iterator regclass_begin() const { return RegClassBegin; }
368   regclass_iterator regclass_end() const { return RegClassEnd; }
369
370   unsigned getNumRegClasses() const {
371     return regclass_end()-regclass_begin();
372   }
373   
374   /// getRegClass - Returns the register class associated with the enumeration
375   /// value.  See class TargetOperandInfo.
376   const TargetRegisterClass *getRegClass(unsigned i) const {
377     assert(i <= getNumRegClasses() && "Register Class ID out of range");
378     return i ? RegClassBegin[i - 1] : NULL;
379   }
380
381   //===--------------------------------------------------------------------===//
382   // Interfaces used by the register allocator and stack frame
383   // manipulation passes to move data around between registers,
384   // immediates and memory.  FIXME: Move these to TargetInstrInfo.h.
385   //
386
387   /// spillCalleeSavedRegisters - Issues instruction(s) to spill all callee saved
388   /// registers and returns true if it isn't possible / profitable to do so by
389   /// issuing a series of store instructions via storeRegToStackSlot(). Returns
390   /// false otherwise.
391   virtual bool spillCalleeSavedRegisters(MachineBasicBlock &MBB,
392                                          MachineBasicBlock::iterator MI,
393                                 const std::vector<CalleeSavedInfo> &CSI) const {
394     return false;
395   }
396
397   /// restoreCalleeSavedRegisters - Issues instruction(s) to restore all callee
398   /// saved registers and returns true if it isn't possible / profitable to do
399   /// so by issuing a series of load instructions via loadRegToStackSlot().
400   /// Returns false otherwise.
401   virtual bool restoreCalleeSavedRegisters(MachineBasicBlock &MBB,
402                                            MachineBasicBlock::iterator MI,
403                                 const std::vector<CalleeSavedInfo> &CSI) const {
404     return false;
405   }
406
407   virtual void storeRegToStackSlot(MachineBasicBlock &MBB,
408                                    MachineBasicBlock::iterator MI,
409                                    unsigned SrcReg, int FrameIndex,
410                                    const TargetRegisterClass *RC) const = 0;
411
412   virtual void loadRegFromStackSlot(MachineBasicBlock &MBB,
413                                     MachineBasicBlock::iterator MI,
414                                     unsigned DestReg, int FrameIndex,
415                                     const TargetRegisterClass *RC) const = 0;
416
417   virtual void copyRegToReg(MachineBasicBlock &MBB,
418                             MachineBasicBlock::iterator MI,
419                             unsigned DestReg, unsigned SrcReg,
420                             const TargetRegisterClass *RC) const = 0;
421
422   /// reMaterialize - Re-issue the specified 'original' instruction at the
423   /// specific location targeting a new destination register.
424   virtual void reMaterialize(MachineBasicBlock &MBB,
425                              MachineBasicBlock::iterator MI,
426                              unsigned DestReg,
427                              const MachineInstr *Orig) const = 0;
428
429   /// foldMemoryOperand - Attempt to fold a load or store of the
430   /// specified stack slot into the specified machine instruction for
431   /// the specified operand.  If this is possible, a new instruction
432   /// is returned with the specified operand folded, otherwise NULL is
433   /// returned. The client is responsible for removing the old
434   /// instruction and adding the new one in the instruction stream
435   virtual MachineInstr* foldMemoryOperand(MachineInstr* MI,
436                                           unsigned OpNum,
437                                           int FrameIndex) const {
438     return 0;
439   }
440
441   /// targetHandlesStackFrameRounding - Returns true if the target is responsible
442   /// for rounding up the stack frame (probably at emitPrologue time).
443   virtual bool targetHandlesStackFrameRounding() const {
444     return false;
445   }
446
447   /// requiresRegisterScavenging - returns true if the target requires (and
448   /// can make use of) the register scavenger.
449   virtual bool requiresRegisterScavenging(const MachineFunction &MF) const {
450     return false;
451   }
452   
453   /// hasFP - Return true if the specified function should have a dedicated frame
454   /// pointer register. For most targets this is true only if the function has
455   /// variable sized allocas or if frame pointer elimination is disabled.
456   virtual bool hasFP(const MachineFunction &MF) const = 0;
457
458   // hasReservedCallFrame - Under normal circumstances, when a frame pointer is
459   // not required, we reserve argument space for call sites in the function
460   // immediately on entry to the current function. This eliminates the need for
461   // add/sub sp brackets around call sites. Returns true if the call frame is
462   // included as part of the stack frame.
463   virtual bool hasReservedCallFrame(MachineFunction &MF) const {
464     return !hasFP(MF);
465   }
466
467   /// getCallFrameSetup/DestroyOpcode - These methods return the opcode of the
468   /// frame setup/destroy instructions if they exist (-1 otherwise).  Some
469   /// targets use pseudo instructions in order to abstract away the difference
470   /// between operating with a frame pointer and operating without, through the
471   /// use of these two instructions.
472   ///
473   int getCallFrameSetupOpcode() const { return CallFrameSetupOpcode; }
474   int getCallFrameDestroyOpcode() const { return CallFrameDestroyOpcode; }
475
476
477   /// eliminateCallFramePseudoInstr - This method is called during prolog/epilog
478   /// code insertion to eliminate call frame setup and destroy pseudo
479   /// instructions (but only if the Target is using them).  It is responsible
480   /// for eliminating these instructions, replacing them with concrete
481   /// instructions.  This method need only be implemented if using call frame
482   /// setup/destroy pseudo instructions.
483   ///
484   virtual void
485   eliminateCallFramePseudoInstr(MachineFunction &MF,
486                                 MachineBasicBlock &MBB,
487                                 MachineBasicBlock::iterator MI) const {
488     assert(getCallFrameSetupOpcode()== -1 && getCallFrameDestroyOpcode()== -1 &&
489            "eliminateCallFramePseudoInstr must be implemented if using"
490            " call frame setup/destroy pseudo instructions!");
491     assert(0 && "Call Frame Pseudo Instructions do not exist on this target!");
492   }
493
494   /// processFunctionBeforeCalleeSavedScan - This method is called immediately
495   /// before PrologEpilogInserter scans the physical registers used to determine
496   /// what callee saved registers should be spilled. This method is optional.
497   virtual void processFunctionBeforeCalleeSavedScan(MachineFunction &MF,
498                                                 RegScavenger *RS = NULL) const {
499
500   }
501
502   /// processFunctionBeforeFrameFinalized - This method is called immediately
503   /// before the specified functions frame layout (MF.getFrameInfo()) is
504   /// finalized.  Once the frame is finalized, MO_FrameIndex operands are
505   /// replaced with direct constants.  This method is optional.
506   ///
507   virtual void processFunctionBeforeFrameFinalized(MachineFunction &MF) const {
508   }
509
510   /// eliminateFrameIndex - This method must be overriden to eliminate abstract
511   /// frame indices from instructions which may use them.  The instruction
512   /// referenced by the iterator contains an MO_FrameIndex operand which must be
513   /// eliminated by this method.  This method may modify or replace the
514   /// specified instruction, as long as it keeps the iterator pointing the the
515   /// finished product. SPAdj is the SP adjustment due to call frame setup
516   /// instruction. The return value is the number of instructions added to
517   /// (negative if removed from) the basic block.
518   ///
519   virtual void eliminateFrameIndex(MachineBasicBlock::iterator MI,
520                                    int SPAdj, RegScavenger *RS=NULL) const = 0;
521
522   /// emitProlog/emitEpilog - These methods insert prolog and epilog code into
523   /// the function. The return value is the number of instructions
524   /// added to (negative if removed from) the basic block (entry for prologue).
525   ///
526   virtual void emitPrologue(MachineFunction &MF) const = 0;
527   virtual void emitEpilogue(MachineFunction &MF,
528                             MachineBasicBlock &MBB) const = 0;
529                             
530   //===--------------------------------------------------------------------===//
531   /// Debug information queries.
532   
533   /// getDwarfRegNum - Map a target register to an equivalent dwarf register
534   /// number.  Returns -1 if there is no equivalent value.
535   virtual int getDwarfRegNum(unsigned RegNum) const = 0;
536
537   /// getFrameRegister - This method should return the register used as a base
538   /// for values allocated in the current stack frame.
539   virtual unsigned getFrameRegister(MachineFunction &MF) const = 0;
540   
541   /// getRARegister - This method should return the register where the return
542   /// address can be found.
543   virtual unsigned getRARegister() const = 0;
544   
545   /// getLocation - This method should return the actual location of a frame
546   /// variable given the frame index.  The location is returned in ML.
547   /// Subclasses should override this method for special handling of frame
548   /// variables and call MRegisterInfo::getLocation for the default action.
549   virtual void getLocation(MachineFunction &MF, unsigned Index,
550                            MachineLocation &ML) const;
551                            
552   /// getInitialFrameState - Returns a list of machine moves that are assumed
553   /// on entry to all functions.  Note that LabelID is ignored (assumed to be
554   /// the beginning of the function.)
555   virtual void getInitialFrameState(std::vector<MachineMove> &Moves) const;
556 };
557
558 // This is useful when building IndexedMaps keyed on virtual registers
559 struct VirtReg2IndexFunctor : std::unary_function<unsigned, unsigned> {
560   unsigned operator()(unsigned Reg) const {
561     return Reg - MRegisterInfo::FirstVirtualRegister;
562   }
563 };
564
565 } // End llvm namespace
566
567 #endif