This patch adds a new NVPTX back-end to LLVM which supports code generation for NVIDI...
[oota-llvm.git] / lib / Target / NVPTX / NVPTXRegisterInfo.h
1 //===- NVPTXRegisterInfo.h - NVPTX Register Information Impl ----*- 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 contains the NVPTX implementation of the TargetRegisterInfo class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #ifndef NVPTXREGISTERINFO_H
15 #define NVPTXREGISTERINFO_H
16
17 #include "ManagedStringPool.h"
18 #include "llvm/Target/TargetRegisterInfo.h"
19
20
21 #define GET_REGINFO_HEADER
22 #include "NVPTXGenRegisterInfo.inc"
23 #include "llvm/Target/TargetRegisterInfo.h"
24 #include <sstream>
25
26 namespace llvm {
27
28 // Forward Declarations.
29 class TargetInstrInfo;
30 class NVPTXSubtarget;
31
32 class NVPTXRegisterInfo : public NVPTXGenRegisterInfo {
33 private:
34   const TargetInstrInfo &TII;
35   const NVPTXSubtarget &ST;
36   bool Is64Bit;
37   // Hold Strings that can be free'd all together with NVPTXRegisterInfo
38   ManagedStringPool     ManagedStrPool;
39
40 public:
41   NVPTXRegisterInfo(const TargetInstrInfo &tii,
42                     const NVPTXSubtarget &st);
43
44
45   //------------------------------------------------------
46   // Pure virtual functions from TargetRegisterInfo
47   //------------------------------------------------------
48
49   // NVPTX callee saved registers
50   virtual const uint16_t*
51   getCalleeSavedRegs(const MachineFunction *MF = 0) const;
52
53   // NVPTX callee saved register classes
54   virtual const TargetRegisterClass* const *
55   getCalleeSavedRegClasses(const MachineFunction *MF) const;
56
57   virtual BitVector getReservedRegs(const MachineFunction &MF) const;
58
59   virtual void eliminateFrameIndex(MachineBasicBlock::iterator MI,
60                                    int SPAdj,
61                                    RegScavenger *RS=NULL) const;
62
63   void eliminateCallFramePseudoInstr(MachineFunction &MF,
64                                      MachineBasicBlock &MBB,
65                                      MachineBasicBlock::iterator I) const;
66
67   virtual int getDwarfRegNum(unsigned RegNum, bool isEH) const;
68   virtual unsigned getFrameRegister(const MachineFunction &MF) const;
69   virtual unsigned getRARegister() const;
70
71   ManagedStringPool *getStrPool() const {
72     return const_cast<ManagedStringPool *>(&ManagedStrPool);
73   }
74
75   const char *getName(unsigned RegNo) const {
76     std::stringstream O;
77     O << "reg" << RegNo;
78     return getStrPool()->getManagedString(O.str().c_str())->c_str();
79   }
80
81 };
82
83
84 std::string getNVPTXRegClassName (const TargetRegisterClass *RC);
85 std::string getNVPTXRegClassStr (const TargetRegisterClass *RC);
86 bool isNVPTXVectorRegClass (const TargetRegisterClass *RC);
87 std::string getNVPTXElemClassName (const TargetRegisterClass *RC);
88 int getNVPTXVectorSize (const TargetRegisterClass *RC);
89 const TargetRegisterClass *getNVPTXElemClass(const TargetRegisterClass *RC);
90
91 } // end namespace llvm
92
93
94 #endif