XCore target: Refactor the loading of constants into a register
[oota-llvm.git] / lib / Target / XCore / XCoreRegisterInfo.cpp
1 //===-- XCoreRegisterInfo.cpp - XCore Register Information ----------------===//
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 XCore implementation of the MRegisterInfo class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "XCoreRegisterInfo.h"
15 #include "XCore.h"
16 #include "XCoreInstrInfo.h"
17 #include "XCoreMachineFunctionInfo.h"
18 #include "llvm/ADT/BitVector.h"
19 #include "llvm/ADT/STLExtras.h"
20 #include "llvm/CodeGen/MachineFrameInfo.h"
21 #include "llvm/CodeGen/MachineFunction.h"
22 #include "llvm/CodeGen/MachineInstrBuilder.h"
23 #include "llvm/CodeGen/MachineModuleInfo.h"
24 #include "llvm/CodeGen/MachineRegisterInfo.h"
25 #include "llvm/CodeGen/RegisterScavenging.h"
26 #include "llvm/IR/Function.h"
27 #include "llvm/IR/Type.h"
28 #include "llvm/Support/Debug.h"
29 #include "llvm/Support/ErrorHandling.h"
30 #include "llvm/Support/MathExtras.h"
31 #include "llvm/Support/raw_ostream.h"
32 #include "llvm/Target/TargetFrameLowering.h"
33 #include "llvm/Target/TargetMachine.h"
34 #include "llvm/Target/TargetOptions.h"
35
36 #define GET_REGINFO_TARGET_DESC
37 #include "XCoreGenRegisterInfo.inc"
38
39 using namespace llvm;
40
41 XCoreRegisterInfo::XCoreRegisterInfo()
42   : XCoreGenRegisterInfo(XCore::LR) {
43 }
44
45 // helper functions
46 static inline bool isImmUs(unsigned val) {
47   return val <= 11;
48 }
49
50 static inline bool isImmU6(unsigned val) {
51   return val < (1 << 6);
52 }
53
54 static inline bool isImmU16(unsigned val) {
55   return val < (1 << 16);
56 }
57
58
59 static void InsertFPImmInst(MachineBasicBlock::iterator II,
60                             const XCoreInstrInfo &TII,
61                             unsigned Reg, unsigned FrameReg, int Offset ) {
62   MachineInstr &MI = *II;
63   MachineBasicBlock &MBB = *MI.getParent();
64   DebugLoc dl = MI.getDebugLoc();
65
66   switch (MI.getOpcode()) {
67   case XCore::LDWFI:
68     BuildMI(MBB, II, dl, TII.get(XCore::LDW_2rus), Reg)
69           .addReg(FrameReg)
70           .addImm(Offset);
71     break;
72   case XCore::STWFI:
73     BuildMI(MBB, II, dl, TII.get(XCore::STW_2rus))
74           .addReg(Reg, getKillRegState(MI.getOperand(0).isKill()))
75           .addReg(FrameReg)
76           .addImm(Offset);
77     break;
78   case XCore::LDAWFI:
79     BuildMI(MBB, II, dl, TII.get(XCore::LDAWF_l2rus), Reg)
80           .addReg(FrameReg)
81           .addImm(Offset);
82     break;
83   default:
84     llvm_unreachable("Unexpected Opcode");
85   }
86 }
87
88 static void InsertFPConstInst(MachineBasicBlock::iterator II,
89                               const XCoreInstrInfo &TII,
90                               unsigned Reg, unsigned FrameReg,
91                               int Offset, RegScavenger *RS ) {
92   assert(RS && "requiresRegisterScavenging failed");
93   MachineInstr &MI = *II;
94   MachineBasicBlock &MBB = *MI.getParent();
95   DebugLoc dl = MI.getDebugLoc();
96
97   unsigned ScratchOffset = RS->scavengeRegister(&XCore::GRRegsRegClass, II, 0);
98   RS->setUsed(ScratchOffset);
99   TII.loadImmediate(MBB, II, ScratchOffset, Offset);
100
101   switch (MI.getOpcode()) {
102   case XCore::LDWFI:
103     BuildMI(MBB, II, dl, TII.get(XCore::LDW_3r), Reg)
104           .addReg(FrameReg)
105           .addReg(ScratchOffset, RegState::Kill);
106     break;
107   case XCore::STWFI:
108     BuildMI(MBB, II, dl, TII.get(XCore::STW_l3r))
109           .addReg(Reg, getKillRegState(MI.getOperand(0).isKill()))
110           .addReg(FrameReg)
111           .addReg(ScratchOffset, RegState::Kill);
112     break;
113   case XCore::LDAWFI:
114     BuildMI(MBB, II, dl, TII.get(XCore::LDAWF_l3r), Reg)
115           .addReg(FrameReg)
116           .addReg(ScratchOffset, RegState::Kill);
117     break;
118   default:
119     llvm_unreachable("Unexpected Opcode");
120   }
121 }
122
123 static void InsertSPImmInst(MachineBasicBlock::iterator II,
124                             const XCoreInstrInfo &TII,
125                             unsigned Reg, int Offset) {
126   MachineInstr &MI = *II;
127   MachineBasicBlock &MBB = *MI.getParent();
128   DebugLoc dl = MI.getDebugLoc();
129   bool isU6 = isImmU6(Offset);
130   switch (MI.getOpcode()) {
131   int NewOpcode;
132   case XCore::LDWFI:
133     NewOpcode = (isU6) ? XCore::LDWSP_ru6 : XCore::LDWSP_lru6;
134     BuildMI(MBB, II, dl, TII.get(NewOpcode), Reg)
135           .addImm(Offset);
136     break;
137   case XCore::STWFI:
138     NewOpcode = (isU6) ? XCore::STWSP_ru6 : XCore::STWSP_lru6;
139     BuildMI(MBB, II, dl, TII.get(NewOpcode))
140           .addReg(Reg, getKillRegState(MI.getOperand(0).isKill()))
141           .addImm(Offset);
142     break;
143   case XCore::LDAWFI:
144     NewOpcode = (isU6) ? XCore::LDAWSP_ru6 : XCore::LDAWSP_lru6;
145     BuildMI(MBB, II, dl, TII.get(NewOpcode), Reg)
146           .addImm(Offset);
147     break;
148   default:
149     llvm_unreachable("Unexpected Opcode");
150   }
151 }
152
153 static void InsertSPConstInst(MachineBasicBlock::iterator II,
154                                 const XCoreInstrInfo &TII,
155                                 unsigned Reg, int Offset, RegScavenger *RS ) {
156   assert(RS && "requiresRegisterScavenging failed");
157   MachineInstr &MI = *II;
158   MachineBasicBlock &MBB = *MI.getParent();
159   DebugLoc dl = MI.getDebugLoc();
160   unsigned OpCode = MI.getOpcode();
161
162   unsigned ScratchBase;
163   if (OpCode==XCore::STWFI) {
164     ScratchBase = RS->scavengeRegister(&XCore::GRRegsRegClass, II, 0);
165     RS->setUsed(ScratchBase);
166   } else
167     ScratchBase = Reg;
168   BuildMI(MBB, II, dl, TII.get(XCore::LDAWSP_ru6), ScratchBase).addImm(0);
169   unsigned ScratchOffset = RS->scavengeRegister(&XCore::GRRegsRegClass, II, 0);
170   RS->setUsed(ScratchOffset);
171   TII.loadImmediate(MBB, II, ScratchOffset, Offset);
172
173   switch (OpCode) {
174   case XCore::LDWFI:
175     BuildMI(MBB, II, dl, TII.get(XCore::LDW_3r), Reg)
176           .addReg(ScratchBase, RegState::Kill)
177           .addReg(ScratchOffset, RegState::Kill);
178     break;
179   case XCore::STWFI:
180     BuildMI(MBB, II, dl, TII.get(XCore::STW_l3r))
181           .addReg(Reg, getKillRegState(MI.getOperand(0).isKill()))
182           .addReg(ScratchBase, RegState::Kill)
183           .addReg(ScratchOffset, RegState::Kill);
184     break;
185   case XCore::LDAWFI:
186     BuildMI(MBB, II, dl, TII.get(XCore::LDAWF_l3r), Reg)
187           .addReg(ScratchBase, RegState::Kill)
188           .addReg(ScratchOffset, RegState::Kill);
189     break;
190   default:
191     llvm_unreachable("Unexpected Opcode");
192   }
193 }
194
195 bool XCoreRegisterInfo::needsFrameMoves(const MachineFunction &MF) {
196   return MF.getMMI().hasDebugInfo() ||
197     MF.getFunction()->needsUnwindTableEntry();
198 }
199
200 const uint16_t* XCoreRegisterInfo::getCalleeSavedRegs(const MachineFunction *MF)
201                                                                          const {
202   static const uint16_t CalleeSavedRegs[] = {
203     XCore::R4, XCore::R5, XCore::R6, XCore::R7,
204     XCore::R8, XCore::R9, XCore::R10, XCore::LR,
205     0
206   };
207   return CalleeSavedRegs;
208 }
209
210 BitVector XCoreRegisterInfo::getReservedRegs(const MachineFunction &MF) const {
211   BitVector Reserved(getNumRegs());
212   const TargetFrameLowering *TFI = MF.getTarget().getFrameLowering();
213
214   Reserved.set(XCore::CP);
215   Reserved.set(XCore::DP);
216   Reserved.set(XCore::SP);
217   Reserved.set(XCore::LR);
218   if (TFI->hasFP(MF)) {
219     Reserved.set(XCore::R10);
220   }
221   return Reserved;
222 }
223
224 bool
225 XCoreRegisterInfo::requiresRegisterScavenging(const MachineFunction &MF) const {
226   return true;
227 }
228
229 bool
230 XCoreRegisterInfo::trackLivenessAfterRegAlloc(const MachineFunction &MF) const {
231   return true;
232 }
233
234 bool
235 XCoreRegisterInfo::useFPForScavengingIndex(const MachineFunction &MF) const {
236   return false;
237 }
238
239 void
240 XCoreRegisterInfo::eliminateFrameIndex(MachineBasicBlock::iterator II,
241                                        int SPAdj, unsigned FIOperandNum,
242                                        RegScavenger *RS) const {
243   assert(SPAdj == 0 && "Unexpected");
244   MachineInstr &MI = *II;
245   MachineOperand &FrameOp = MI.getOperand(FIOperandNum);
246   int FrameIndex = FrameOp.getIndex();
247
248   MachineFunction &MF = *MI.getParent()->getParent();
249   const XCoreInstrInfo &TII =
250           *static_cast<const XCoreInstrInfo*>(MF.getTarget().getInstrInfo());
251
252   const TargetFrameLowering *TFI = MF.getTarget().getFrameLowering();
253   int Offset = MF.getFrameInfo()->getObjectOffset(FrameIndex);
254   int StackSize = MF.getFrameInfo()->getStackSize();
255
256   #ifndef NDEBUG
257   DEBUG(errs() << "\nFunction         : " 
258         << MF.getName() << "\n");
259   DEBUG(errs() << "<--------->\n");
260   DEBUG(MI.print(errs()));
261   DEBUG(errs() << "FrameIndex         : " << FrameIndex << "\n");
262   DEBUG(errs() << "FrameOffset        : " << Offset << "\n");
263   DEBUG(errs() << "StackSize          : " << StackSize << "\n");
264   #endif
265
266   Offset += StackSize;
267
268   unsigned FrameReg = getFrameRegister(MF);
269
270   // Special handling of DBG_VALUE instructions.
271   if (MI.isDebugValue()) {
272     MI.getOperand(FIOperandNum).ChangeToRegister(FrameReg, false /*isDef*/);
273     MI.getOperand(FIOperandNum + 1).ChangeToImmediate(Offset);
274     return;
275   }
276
277   // fold constant into offset.
278   Offset += MI.getOperand(FIOperandNum + 1).getImm();
279   MI.getOperand(FIOperandNum + 1).ChangeToImmediate(0);
280   
281   assert(Offset%4 == 0 && "Misaligned stack offset");
282   DEBUG(errs() << "Offset             : " << Offset << "\n" << "<--------->\n");
283   Offset/=4;
284   
285   unsigned Reg = MI.getOperand(0).getReg();
286   assert(XCore::GRRegsRegClass.contains(Reg) && "Unexpected register operand");
287
288   if (TFI->hasFP(MF)) {
289     if (isImmUs(Offset))
290       InsertFPImmInst(II, TII, Reg, FrameReg, Offset);
291     else
292       InsertFPConstInst(II, TII, Reg, FrameReg, Offset, RS);
293   } else {
294     if (isImmU16(Offset))
295       InsertSPImmInst(II, TII, Reg, Offset);
296     else
297       InsertSPConstInst(II, TII, Reg, Offset, RS);
298   }
299   // Erase old instruction.
300   MachineBasicBlock &MBB = *MI.getParent();
301   MBB.erase(II);
302 }
303
304
305 unsigned XCoreRegisterInfo::getFrameRegister(const MachineFunction &MF) const {
306   const TargetFrameLowering *TFI = MF.getTarget().getFrameLowering();
307
308   return TFI->hasFP(MF) ? XCore::R10 : XCore::SP;
309 }