Don't leak a MachineInstruction from Thumb1InstrInfo::restoreCalleeSavedRegisters.
[oota-llvm.git] / lib / Target / ARM / Thumb1InstrInfo.cpp
1 //===- Thumb1InstrInfo.cpp - Thumb-1 Instruction 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 contains the Thumb-1 implementation of the TargetInstrInfo class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "Thumb1InstrInfo.h"
15 #include "ARM.h"
16 #include "ARMGenInstrInfo.inc"
17 #include "ARMMachineFunctionInfo.h"
18 #include "llvm/CodeGen/MachineFrameInfo.h"
19 #include "llvm/CodeGen/MachineInstrBuilder.h"
20 #include "llvm/CodeGen/MachineMemOperand.h"
21 #include "llvm/CodeGen/PseudoSourceValue.h"
22 #include "llvm/ADT/SmallVector.h"
23 #include "Thumb1InstrInfo.h"
24
25 using namespace llvm;
26
27 Thumb1InstrInfo::Thumb1InstrInfo(const ARMSubtarget &STI)
28   : ARMBaseInstrInfo(STI), RI(*this, STI) {
29 }
30
31 unsigned Thumb1InstrInfo::getUnindexedOpcode(unsigned Opc) const {
32   return 0;
33 }
34
35 bool Thumb1InstrInfo::copyRegToReg(MachineBasicBlock &MBB,
36                                    MachineBasicBlock::iterator I,
37                                    unsigned DestReg, unsigned SrcReg,
38                                    const TargetRegisterClass *DestRC,
39                                    const TargetRegisterClass *SrcRC) const {
40   DebugLoc DL = DebugLoc::getUnknownLoc();
41   if (I != MBB.end()) DL = I->getDebugLoc();
42
43   if (DestRC == ARM::GPRRegisterClass) {
44     if (SrcRC == ARM::GPRRegisterClass) {
45       BuildMI(MBB, I, DL, get(ARM::tMOVgpr2gpr), DestReg).addReg(SrcReg);
46       return true;
47     } else if (SrcRC == ARM::tGPRRegisterClass) {
48       BuildMI(MBB, I, DL, get(ARM::tMOVtgpr2gpr), DestReg).addReg(SrcReg);
49       return true;
50     }
51   } else if (DestRC == ARM::tGPRRegisterClass) {
52     if (SrcRC == ARM::GPRRegisterClass) {
53       BuildMI(MBB, I, DL, get(ARM::tMOVgpr2tgpr), DestReg).addReg(SrcReg);
54       return true;
55     } else if (SrcRC == ARM::tGPRRegisterClass) {
56       BuildMI(MBB, I, DL, get(ARM::tMOVr), DestReg).addReg(SrcReg);
57       return true;
58     }
59   }
60
61   return false;
62 }
63
64 bool Thumb1InstrInfo::
65 canFoldMemoryOperand(const MachineInstr *MI,
66                      const SmallVectorImpl<unsigned> &Ops) const {
67   if (Ops.size() != 1) return false;
68
69   unsigned OpNum = Ops[0];
70   unsigned Opc = MI->getOpcode();
71   switch (Opc) {
72   default: break;
73   case ARM::tMOVr:
74   case ARM::tMOVtgpr2gpr:
75   case ARM::tMOVgpr2tgpr:
76   case ARM::tMOVgpr2gpr: {
77     if (OpNum == 0) { // move -> store
78       unsigned SrcReg = MI->getOperand(1).getReg();
79       if (TargetRegisterInfo::isPhysicalRegister(SrcReg) &&
80           !isARMLowRegister(SrcReg))
81         // tSpill cannot take a high register operand.
82         return false;
83     } else {          // move -> load
84       unsigned DstReg = MI->getOperand(0).getReg();
85       if (TargetRegisterInfo::isPhysicalRegister(DstReg) &&
86           !isARMLowRegister(DstReg))
87         // tRestore cannot target a high register operand.
88         return false;
89     }
90     return true;
91   }
92   }
93
94   return false;
95 }
96
97 void Thumb1InstrInfo::
98 storeRegToStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator I,
99                     unsigned SrcReg, bool isKill, int FI,
100                     const TargetRegisterClass *RC) const {
101   DebugLoc DL = DebugLoc::getUnknownLoc();
102   if (I != MBB.end()) DL = I->getDebugLoc();
103
104   assert((RC == ARM::tGPRRegisterClass ||
105           (TargetRegisterInfo::isPhysicalRegister(SrcReg) &&
106            isARMLowRegister(SrcReg))) && "Unknown regclass!");
107
108   if (RC == ARM::tGPRRegisterClass ||
109       (TargetRegisterInfo::isPhysicalRegister(SrcReg) &&
110        isARMLowRegister(SrcReg))) {
111     MachineFunction &MF = *MBB.getParent();
112     MachineFrameInfo &MFI = *MF.getFrameInfo();
113     MachineMemOperand *MMO =
114       MF.getMachineMemOperand(PseudoSourceValue::getFixedStack(FI),
115                               MachineMemOperand::MOStore, 0,
116                               MFI.getObjectSize(FI),
117                               MFI.getObjectAlignment(FI));
118     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::tSpill))
119                    .addReg(SrcReg, getKillRegState(isKill))
120                    .addFrameIndex(FI).addImm(0).addMemOperand(MMO));
121   }
122 }
123
124 void Thumb1InstrInfo::
125 loadRegFromStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator I,
126                      unsigned DestReg, int FI,
127                      const TargetRegisterClass *RC) const {
128   DebugLoc DL = DebugLoc::getUnknownLoc();
129   if (I != MBB.end()) DL = I->getDebugLoc();
130
131   assert((RC == ARM::tGPRRegisterClass ||
132           (TargetRegisterInfo::isPhysicalRegister(DestReg) &&
133            isARMLowRegister(DestReg))) && "Unknown regclass!");
134
135   if (RC == ARM::tGPRRegisterClass ||
136       (TargetRegisterInfo::isPhysicalRegister(DestReg) &&
137        isARMLowRegister(DestReg))) {
138     MachineFunction &MF = *MBB.getParent();
139     MachineFrameInfo &MFI = *MF.getFrameInfo();
140     MachineMemOperand *MMO =
141       MF.getMachineMemOperand(PseudoSourceValue::getFixedStack(FI),
142                               MachineMemOperand::MOLoad, 0,
143                               MFI.getObjectSize(FI),
144                               MFI.getObjectAlignment(FI));
145     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::tRestore), DestReg)
146                    .addFrameIndex(FI).addImm(0).addMemOperand(MMO));
147   }
148 }
149
150 bool Thumb1InstrInfo::
151 spillCalleeSavedRegisters(MachineBasicBlock &MBB,
152                           MachineBasicBlock::iterator MI,
153                           const std::vector<CalleeSavedInfo> &CSI) const {
154   if (CSI.empty())
155     return false;
156
157   DebugLoc DL = DebugLoc::getUnknownLoc();
158   if (MI != MBB.end()) DL = MI->getDebugLoc();
159
160   MachineInstrBuilder MIB = BuildMI(MBB, MI, DL, get(ARM::tPUSH));
161   AddDefaultPred(MIB);
162   for (unsigned i = CSI.size(); i != 0; --i) {
163     unsigned Reg = CSI[i-1].getReg();
164     // Add the callee-saved register as live-in. It's killed at the spill.
165     MBB.addLiveIn(Reg);
166     MIB.addReg(Reg, RegState::Kill);
167   }
168   return true;
169 }
170
171 bool Thumb1InstrInfo::
172 restoreCalleeSavedRegisters(MachineBasicBlock &MBB,
173                             MachineBasicBlock::iterator MI,
174                             const std::vector<CalleeSavedInfo> &CSI) const {
175   MachineFunction &MF = *MBB.getParent();
176   ARMFunctionInfo *AFI = MF.getInfo<ARMFunctionInfo>();
177   if (CSI.empty())
178     return false;
179
180   bool isVarArg = AFI->getVarArgsRegSaveSize() > 0;
181   DebugLoc DL = MI->getDebugLoc();
182   MachineInstrBuilder MIB = BuildMI(MF, DL, get(ARM::tPOP));
183   AddDefaultPred(MIB);
184
185   bool NumRegs = false;
186   for (unsigned i = CSI.size(); i != 0; --i) {
187     unsigned Reg = CSI[i-1].getReg();
188     if (Reg == ARM::LR) {
189       // Special epilogue for vararg functions. See emitEpilogue
190       if (isVarArg)
191         continue;
192       Reg = ARM::PC;
193       (*MIB).setDesc(get(ARM::tPOP_RET));
194       MI = MBB.erase(MI);
195     }
196     MIB.addReg(Reg, getDefRegState(true));
197     NumRegs = true;
198   }
199
200   // It's illegal to emit pop instruction without operands.
201   if (NumRegs)
202     MBB.insert(MI, &*MIB);
203   else
204     MF.DeleteMachineInstr(MIB);
205
206   return true;
207 }
208
209 MachineInstr *Thumb1InstrInfo::
210 foldMemoryOperandImpl(MachineFunction &MF, MachineInstr *MI,
211                       const SmallVectorImpl<unsigned> &Ops, int FI) const {
212   if (Ops.size() != 1) return NULL;
213
214   unsigned OpNum = Ops[0];
215   unsigned Opc = MI->getOpcode();
216   MachineInstr *NewMI = NULL;
217   switch (Opc) {
218   default: break;
219   case ARM::tMOVr:
220   case ARM::tMOVtgpr2gpr:
221   case ARM::tMOVgpr2tgpr:
222   case ARM::tMOVgpr2gpr: {
223     if (OpNum == 0) { // move -> store
224       unsigned SrcReg = MI->getOperand(1).getReg();
225       bool isKill = MI->getOperand(1).isKill();
226       if (TargetRegisterInfo::isPhysicalRegister(SrcReg) &&
227           !isARMLowRegister(SrcReg))
228         // tSpill cannot take a high register operand.
229         break;
230       NewMI = AddDefaultPred(BuildMI(MF, MI->getDebugLoc(), get(ARM::tSpill))
231                              .addReg(SrcReg, getKillRegState(isKill))
232                              .addFrameIndex(FI).addImm(0));
233     } else {          // move -> load
234       unsigned DstReg = MI->getOperand(0).getReg();
235       if (TargetRegisterInfo::isPhysicalRegister(DstReg) &&
236           !isARMLowRegister(DstReg))
237         // tRestore cannot target a high register operand.
238         break;
239       bool isDead = MI->getOperand(0).isDead();
240       NewMI = AddDefaultPred(BuildMI(MF, MI->getDebugLoc(), get(ARM::tRestore))
241                              .addReg(DstReg,
242                                      RegState::Define | getDeadRegState(isDead))
243                              .addFrameIndex(FI).addImm(0));
244     }
245     break;
246   }
247   }
248
249   return NewMI;
250 }