Add MCRI::getNumSubRegIndices() and start checking SubRegIndex ranges.
[oota-llvm.git] / include / llvm / MC / MCRegisterInfo.h
1 //=== MC/MCRegisterInfo.h - Target Register Description ---------*- 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_MC_MCREGISTERINFO_H
17 #define LLVM_MC_MCREGISTERINFO_H
18
19 #include "llvm/ADT/DenseMap.h"
20 #include "llvm/Support/ErrorHandling.h"
21 #include <cassert>
22
23 namespace llvm {
24
25 /// MCRegisterClass - Base class of TargetRegisterClass.
26 class MCRegisterClass {
27 public:
28   typedef const uint16_t* iterator;
29   typedef const uint16_t* const_iterator;
30
31   const char *Name;
32   const iterator RegsBegin;
33   const uint8_t *const RegSet;
34   const uint16_t RegsSize;
35   const uint16_t RegSetSize;
36   const uint16_t ID;
37   const uint16_t RegSize, Alignment; // Size & Alignment of register in bytes
38   const int8_t CopyCost;
39   const bool Allocatable;
40
41   /// getID() - Return the register class ID number.
42   ///
43   unsigned getID() const { return ID; }
44
45   /// getName() - Return the register class name for debugging.
46   ///
47   const char *getName() const { return Name; }
48
49   /// begin/end - Return all of the registers in this class.
50   ///
51   iterator       begin() const { return RegsBegin; }
52   iterator         end() const { return RegsBegin + RegsSize; }
53
54   /// getNumRegs - Return the number of registers in this class.
55   ///
56   unsigned getNumRegs() const { return RegsSize; }
57
58   /// getRegister - Return the specified register in the class.
59   ///
60   unsigned getRegister(unsigned i) const {
61     assert(i < getNumRegs() && "Register number out of range!");
62     return RegsBegin[i];
63   }
64
65   /// contains - Return true if the specified register is included in this
66   /// register class.  This does not include virtual registers.
67   bool contains(unsigned Reg) const {
68     unsigned InByte = Reg % 8;
69     unsigned Byte = Reg / 8;
70     if (Byte >= RegSetSize)
71       return false;
72     return (RegSet[Byte] & (1 << InByte)) != 0;
73   }
74
75   /// contains - Return true if both registers are in this class.
76   bool contains(unsigned Reg1, unsigned Reg2) const {
77     return contains(Reg1) && contains(Reg2);
78   }
79
80   /// getSize - Return the size of the register in bytes, which is also the size
81   /// of a stack slot allocated to hold a spilled copy of this register.
82   unsigned getSize() const { return RegSize; }
83
84   /// getAlignment - Return the minimum required alignment for a register of
85   /// this class.
86   unsigned getAlignment() const { return Alignment; }
87
88   /// getCopyCost - Return the cost of copying a value between two registers in
89   /// this class. A negative number means the register class is very expensive
90   /// to copy e.g. status flag register classes.
91   int getCopyCost() const { return CopyCost; }
92
93   /// isAllocatable - Return true if this register class may be used to create
94   /// virtual registers.
95   bool isAllocatable() const { return Allocatable; }
96 };
97
98 /// MCRegisterDesc - This record contains all of the information known about
99 /// a particular register.  The Overlaps field contains a pointer to a zero
100 /// terminated array of registers that this register aliases, starting with
101 /// itself. This is needed for architectures like X86 which have AL alias AX
102 /// alias EAX. The SubRegs field is a zero terminated array of registers that
103 /// are sub-registers of the specific register, e.g. AL, AH are sub-registers of
104 /// AX. The SuperRegs field is a zero terminated array of registers that are
105 /// super-registers of the specific register, e.g. RAX, EAX, are super-registers
106 /// of AX.
107 ///
108 struct MCRegisterDesc {
109   uint32_t Name;      // Printable name for the reg (for debugging)
110   uint32_t Overlaps;  // Overlapping registers, described above
111   uint32_t SubRegs;   // Sub-register set, described above
112   uint32_t SuperRegs; // Super-register set, described above
113
114   // Offset into MCRI::SubRegIndices of a list of sub-register indices for each
115   // sub-register in SubRegs.
116   uint32_t SubRegIndices;
117
118   // RegUnits - Points to the list of register units. The low 4 bits holds the
119   // Scale, the high bits hold an offset into DiffLists. See MCRegUnitIterator.
120   uint32_t RegUnits;
121 };
122
123 /// MCRegisterInfo base class - We assume that the target defines a static
124 /// array of MCRegisterDesc objects that represent all of the machine
125 /// registers that the target has.  As such, we simply have to track a pointer
126 /// to this array so that we can turn register number into a register
127 /// descriptor.
128 ///
129 /// Note this class is designed to be a base class of TargetRegisterInfo, which
130 /// is the interface used by codegen. However, specific targets *should never*
131 /// specialize this class. MCRegisterInfo should only contain getters to access
132 /// TableGen generated physical register data. It must not be extended with
133 /// virtual methods.
134 ///
135 class MCRegisterInfo {
136 public:
137   typedef const MCRegisterClass *regclass_iterator;
138
139   /// DwarfLLVMRegPair - Emitted by tablegen so Dwarf<->LLVM reg mappings can be
140   /// performed with a binary search.
141   struct DwarfLLVMRegPair {
142     unsigned FromReg;
143     unsigned ToReg;
144
145     bool operator<(DwarfLLVMRegPair RHS) const { return FromReg < RHS.FromReg; }
146   };
147 private:
148   const MCRegisterDesc *Desc;                 // Pointer to the descriptor array
149   unsigned NumRegs;                           // Number of entries in the array
150   unsigned RAReg;                             // Return address register
151   const MCRegisterClass *Classes;             // Pointer to the regclass array
152   unsigned NumClasses;                        // Number of entries in the array
153   unsigned NumRegUnits;                       // Number of regunits.
154   const uint16_t (*RegUnitRoots)[2];          // Pointer to regunit root table.
155   const uint16_t *DiffLists;                  // Pointer to the difflists array
156   const char *RegStrings;                     // Pointer to the string table.
157   const uint16_t *SubRegIndices;              // Pointer to the subreg lookup
158                                               // array.
159   unsigned NumSubRegIndices;                  // Number of subreg indices.
160   const uint16_t *RegEncodingTable;           // Pointer to array of register
161                                               // encodings.
162
163   unsigned L2DwarfRegsSize;
164   unsigned EHL2DwarfRegsSize;
165   unsigned Dwarf2LRegsSize;
166   unsigned EHDwarf2LRegsSize;
167   const DwarfLLVMRegPair *L2DwarfRegs;        // LLVM to Dwarf regs mapping
168   const DwarfLLVMRegPair *EHL2DwarfRegs;      // LLVM to Dwarf regs mapping EH
169   const DwarfLLVMRegPair *Dwarf2LRegs;        // Dwarf to LLVM regs mapping
170   const DwarfLLVMRegPair *EHDwarf2LRegs;      // Dwarf to LLVM regs mapping EH
171   DenseMap<unsigned, int> L2SEHRegs;          // LLVM to SEH regs mapping
172
173 public:
174   /// DiffListIterator - Base iterator class that can traverse the
175   /// differentially encoded register and regunit lists in DiffLists.
176   /// Don't use this class directly, use one of the specialized sub-classes
177   /// defined below.
178   class DiffListIterator {
179     uint16_t Val;
180     const uint16_t *List;
181
182   protected:
183     /// Create an invalid iterator. Call init() to point to something useful.
184     DiffListIterator() : Val(0), List(0) {}
185
186     /// init - Point the iterator to InitVal, decoding subsequent values from
187     /// DiffList. The iterator will initially point to InitVal, sub-classes are
188     /// responsible for skipping the seed value if it is not part of the list.
189     void init(uint16_t InitVal, const uint16_t *DiffList) {
190       Val = InitVal;
191       List = DiffList;
192     }
193
194     /// advance - Move to the next list position, return the applied
195     /// differential. This function does not detect the end of the list, that
196     /// is the caller's responsibility (by checking for a 0 return value).
197     unsigned advance() {
198       assert(isValid() && "Cannot move off the end of the list.");
199       uint16_t D = *List++;
200       Val += D;
201       return D;
202     }
203
204   public:
205
206     /// isValid - returns true if this iterator is not yet at the end.
207     bool isValid() const { return List; }
208
209     /// Dereference the iterator to get the value at the current position.
210     unsigned operator*() const { return Val; }
211
212     /// Pre-increment to move to the next position.
213     void operator++() {
214       // The end of the list is encoded as a 0 differential.
215       if (!advance())
216         List = 0;
217     }
218   };
219
220   // These iterators are allowed to sub-class DiffListIterator and access
221   // internal list pointers.
222   friend class MCSubRegIterator;
223   friend class MCSuperRegIterator;
224   friend class MCRegAliasIterator;
225   friend class MCRegUnitIterator;
226   friend class MCRegUnitRootIterator;
227
228   /// InitMCRegisterInfo - Initialize MCRegisterInfo, called by TableGen
229   /// auto-generated routines. *DO NOT USE*.
230   void InitMCRegisterInfo(const MCRegisterDesc *D, unsigned NR, unsigned RA,
231                           const MCRegisterClass *C, unsigned NC,
232                           const uint16_t (*RURoots)[2],
233                           unsigned NRU,
234                           const uint16_t *DL,
235                           const char *Strings,
236                           const uint16_t *SubIndices,
237                           unsigned NumIndices,
238                           const uint16_t *RET) {
239     Desc = D;
240     NumRegs = NR;
241     RAReg = RA;
242     Classes = C;
243     DiffLists = DL;
244     RegStrings = Strings;
245     NumClasses = NC;
246     RegUnitRoots = RURoots;
247     NumRegUnits = NRU;
248     SubRegIndices = SubIndices;
249     NumSubRegIndices = NumIndices;
250     RegEncodingTable = RET;
251   }
252
253   /// mapLLVMRegsToDwarfRegs - Used to initialize LLVM register to Dwarf
254   /// register number mapping. Called by TableGen auto-generated routines.
255   /// *DO NOT USE*.
256   void mapLLVMRegsToDwarfRegs(const DwarfLLVMRegPair *Map, unsigned Size,
257                               bool isEH) {
258     if (isEH) {
259       EHL2DwarfRegs = Map;
260       EHL2DwarfRegsSize = Size;
261     } else {
262       L2DwarfRegs = Map;
263       L2DwarfRegsSize = Size;
264     }
265   }
266
267   /// mapDwarfRegsToLLVMRegs - Used to initialize Dwarf register to LLVM
268   /// register number mapping. Called by TableGen auto-generated routines.
269   /// *DO NOT USE*.
270   void mapDwarfRegsToLLVMRegs(const DwarfLLVMRegPair *Map, unsigned Size,
271                               bool isEH) {
272     if (isEH) {
273       EHDwarf2LRegs = Map;
274       EHDwarf2LRegsSize = Size;
275     } else {
276       Dwarf2LRegs = Map;
277       Dwarf2LRegsSize = Size;
278     }
279   }
280
281   /// mapLLVMRegToSEHReg - Used to initialize LLVM register to SEH register
282   /// number mapping. By default the SEH register number is just the same
283   /// as the LLVM register number.
284   /// FIXME: TableGen these numbers. Currently this requires target specific
285   /// initialization code.
286   void mapLLVMRegToSEHReg(unsigned LLVMReg, int SEHReg) {
287     L2SEHRegs[LLVMReg] = SEHReg;
288   }
289
290   /// getRARegister - This method should return the register where the return
291   /// address can be found.
292   unsigned getRARegister() const {
293     return RAReg;
294   }
295
296   const MCRegisterDesc &operator[](unsigned RegNo) const {
297     assert(RegNo < NumRegs &&
298            "Attempting to access record for invalid register number!");
299     return Desc[RegNo];
300   }
301
302   /// Provide a get method, equivalent to [], but more useful if we have a
303   /// pointer to this object.
304   ///
305   const MCRegisterDesc &get(unsigned RegNo) const {
306     return operator[](RegNo);
307   }
308
309   /// getSubReg - Returns the physical register number of sub-register "Index"
310   /// for physical register RegNo. Return zero if the sub-register does not
311   /// exist.
312   unsigned getSubReg(unsigned Reg, unsigned Idx) const;
313
314   /// getMatchingSuperReg - Return a super-register of the specified register
315   /// Reg so its sub-register of index SubIdx is Reg.
316   unsigned getMatchingSuperReg(unsigned Reg, unsigned SubIdx,
317                                const MCRegisterClass *RC) const;
318
319   /// getSubRegIndex - For a given register pair, return the sub-register index
320   /// if the second register is a sub-register of the first. Return zero
321   /// otherwise.
322   unsigned getSubRegIndex(unsigned RegNo, unsigned SubRegNo) const;
323
324   /// getName - Return the human-readable symbolic target-specific name for the
325   /// specified physical register.
326   const char *getName(unsigned RegNo) const {
327     return RegStrings + get(RegNo).Name;
328   }
329
330   /// getNumRegs - Return the number of registers this target has (useful for
331   /// sizing arrays holding per register information)
332   unsigned getNumRegs() const {
333     return NumRegs;
334   }
335
336   /// getNumSubRegIndices - Return the number of sub-register indices
337   /// understood by the target. Index 0 is reserved for the no-op sub-register,
338   /// while 1 to getNumSubRegIndices() - 1 represent real sub-registers.
339   unsigned getNumSubRegIndices() const {
340     return NumSubRegIndices;
341   }
342
343   /// getNumRegUnits - Return the number of (native) register units in the
344   /// target. Register units are numbered from 0 to getNumRegUnits() - 1. They
345   /// can be accessed through MCRegUnitIterator defined below.
346   unsigned getNumRegUnits() const {
347     return NumRegUnits;
348   }
349
350   /// getDwarfRegNum - Map a target register to an equivalent dwarf register
351   /// number.  Returns -1 if there is no equivalent value.  The second
352   /// parameter allows targets to use different numberings for EH info and
353   /// debugging info.
354   int getDwarfRegNum(unsigned RegNum, bool isEH) const;
355
356   /// getLLVMRegNum - Map a dwarf register back to a target register.
357   ///
358   int getLLVMRegNum(unsigned RegNum, bool isEH) const;
359
360   /// getSEHRegNum - Map a target register to an equivalent SEH register
361   /// number.  Returns LLVM register number if there is no equivalent value.
362   int getSEHRegNum(unsigned RegNum) const;
363
364   regclass_iterator regclass_begin() const { return Classes; }
365   regclass_iterator regclass_end() const { return Classes+NumClasses; }
366
367   unsigned getNumRegClasses() const {
368     return (unsigned)(regclass_end()-regclass_begin());
369   }
370
371   /// getRegClass - Returns the register class associated with the enumeration
372   /// value.  See class MCOperandInfo.
373   const MCRegisterClass getRegClass(unsigned i) const {
374     assert(i < getNumRegClasses() && "Register Class ID out of range");
375     return Classes[i];
376   }
377
378    /// getEncodingValue - Returns the encoding for RegNo
379   uint16_t getEncodingValue(unsigned RegNo) const {
380     assert(RegNo < NumRegs &&
381            "Attempting to get encoding for invalid register number!");
382     return RegEncodingTable[RegNo];
383   }
384
385 };
386
387 //===----------------------------------------------------------------------===//
388 //                          Register List Iterators
389 //===----------------------------------------------------------------------===//
390
391 // MCRegisterInfo provides lists of super-registers, sub-registers, and
392 // aliasing registers. Use these iterator classes to traverse the lists.
393
394 /// MCSubRegIterator enumerates all sub-registers of Reg.
395 class MCSubRegIterator : public MCRegisterInfo::DiffListIterator {
396 public:
397   MCSubRegIterator(unsigned Reg, const MCRegisterInfo *MCRI) {
398     init(Reg, MCRI->DiffLists + MCRI->get(Reg).SubRegs);
399     ++*this;
400   }
401 };
402
403 /// MCSuperRegIterator enumerates all super-registers of Reg.
404 class MCSuperRegIterator : public MCRegisterInfo::DiffListIterator {
405 public:
406   MCSuperRegIterator(unsigned Reg, const MCRegisterInfo *MCRI) {
407     init(Reg, MCRI->DiffLists + MCRI->get(Reg).SuperRegs);
408     ++*this;
409   }
410 };
411
412 /// MCRegAliasIterator enumerates all registers aliasing Reg.
413 /// If IncludeSelf is set, Reg itself is included in the list.
414 class MCRegAliasIterator : public MCRegisterInfo::DiffListIterator {
415 public:
416   MCRegAliasIterator(unsigned Reg, const MCRegisterInfo *MCRI,
417                      bool IncludeSelf) {
418     init(Reg, MCRI->DiffLists + MCRI->get(Reg).Overlaps);
419     // Initially, the iterator points to Reg itself.
420     if (!IncludeSelf)
421       ++*this;
422   }
423 };
424
425 //===----------------------------------------------------------------------===//
426 //                               Register Units
427 //===----------------------------------------------------------------------===//
428
429 // Register units are used to compute register aliasing. Every register has at
430 // least one register unit, but it can have more. Two registers overlap if and
431 // only if they have a common register unit.
432 //
433 // A target with a complicated sub-register structure will typically have many
434 // fewer register units than actual registers. MCRI::getNumRegUnits() returns
435 // the number of register units in the target.
436
437 // MCRegUnitIterator enumerates a list of register units for Reg. The list is
438 // in ascending numerical order.
439 class MCRegUnitIterator : public MCRegisterInfo::DiffListIterator {
440 public:
441   /// MCRegUnitIterator - Create an iterator that traverses the register units
442   /// in Reg.
443   MCRegUnitIterator(unsigned Reg, const MCRegisterInfo *MCRI) {
444     // Decode the RegUnits MCRegisterDesc field.
445     unsigned RU = MCRI->get(Reg).RegUnits;
446     unsigned Scale = RU & 15;
447     unsigned Offset = RU >> 4;
448
449     // Initialize the iterator to Reg * Scale, and the List pointer to
450     // DiffLists + Offset.
451     init(Reg * Scale, MCRI->DiffLists + Offset);
452
453     // That may not be a valid unit, we need to advance by one to get the real
454     // unit number. The first differential can be 0 which would normally
455     // terminate the list, but since we know every register has at least one
456     // unit, we can allow a 0 differential here.
457     advance();
458   }
459 };
460
461 // Each register unit has one or two root registers. The complete set of
462 // registers containing a register unit is the union of the roots and their
463 // super-registers. All registers aliasing Unit can be visited like this:
464 //
465 //   for (MCRegUnitRootIterator RI(Unit, MCRI); RI.isValid(); ++RI) {
466 //     unsigned Root = *RI;
467 //     visit(Root);
468 //     for (MCSuperRegIterator SI(Root, MCRI); SI.isValid(); ++SI)
469 //       visit(*SI);
470 //    }
471
472 /// MCRegUnitRootIterator enumerates the root registers of a register unit.
473 class MCRegUnitRootIterator {
474   uint16_t Reg0;
475   uint16_t Reg1;
476 public:
477   MCRegUnitRootIterator(unsigned RegUnit, const MCRegisterInfo *MCRI) {
478     assert(RegUnit < MCRI->getNumRegUnits() && "Invalid register unit");
479     Reg0 = MCRI->RegUnitRoots[RegUnit][0];
480     Reg1 = MCRI->RegUnitRoots[RegUnit][1];
481   }
482
483   /// Dereference to get the current root register.
484   unsigned operator*() const {
485     return Reg0;
486   }
487
488   /// isValid - Check if the iterator is at the end of the list.
489   bool isValid() const {
490     return Reg0;
491   }
492
493   /// Preincrement to move to the next root register.
494   void operator++() {
495     assert(isValid() && "Cannot move off the end of the list.");
496     Reg0 = Reg1;
497     Reg1 = 0;
498   }
499 };
500
501 } // End llvm namespace
502
503 #endif