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