blob: 79d58692e9f6da09245bfcddea1d7e96434d22aa [file] [log] [blame]
Rafael Espindola7bc59bc2006-05-14 22:18:28 +00001//===-- ARMISelDAGToDAG.cpp - A dag to dag inst selector for ARM ----------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by Chris Lattner and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file defines an instruction selector for the ARM target.
11//
12//===----------------------------------------------------------------------===//
13
14#include "ARM.h"
Evan Chenga8e29892007-01-19 07:51:42 +000015#include "ARMISelLowering.h"
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000016#include "ARMTargetMachine.h"
Evan Chenga8e29892007-01-19 07:51:42 +000017#include "ARMAddressingModes.h"
Rafael Espindola84b19be2006-07-16 01:02:57 +000018#include "llvm/CallingConv.h"
Evan Chenga8e29892007-01-19 07:51:42 +000019#include "llvm/Constants.h"
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000020#include "llvm/DerivedTypes.h"
21#include "llvm/Function.h"
22#include "llvm/Intrinsics.h"
23#include "llvm/CodeGen/MachineFrameInfo.h"
24#include "llvm/CodeGen/MachineFunction.h"
25#include "llvm/CodeGen/MachineInstrBuilder.h"
26#include "llvm/CodeGen/SelectionDAG.h"
27#include "llvm/CodeGen/SelectionDAGISel.h"
28#include "llvm/CodeGen/SSARegMap.h"
29#include "llvm/Target/TargetLowering.h"
Chris Lattner72939122007-05-03 00:32:00 +000030#include "llvm/Target/TargetOptions.h"
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000031#include "llvm/Support/Debug.h"
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000032using namespace llvm;
33
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000034//===--------------------------------------------------------------------===//
35/// ARMDAGToDAGISel - ARM specific code to select ARM machine
36/// instructions for SelectionDAG operations.
37///
38namespace {
39class ARMDAGToDAGISel : public SelectionDAGISel {
40 ARMTargetLowering Lowering;
41
Evan Chenga8e29892007-01-19 07:51:42 +000042 /// Subtarget - Keep a pointer to the ARMSubtarget around so that we can
43 /// make the right decision when generating code for different targets.
44 const ARMSubtarget *Subtarget;
45
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000046public:
Evan Chenga8e29892007-01-19 07:51:42 +000047 ARMDAGToDAGISel(ARMTargetMachine &TM)
48 : SelectionDAGISel(Lowering), Lowering(TM),
49 Subtarget(&TM.getSubtarget<ARMSubtarget>()) {
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000050 }
51
Evan Chenga8e29892007-01-19 07:51:42 +000052 virtual const char *getPassName() const {
53 return "ARM Instruction Selection";
54 }
55
Evan Cheng9ade2182006-08-26 05:34:46 +000056 SDNode *Select(SDOperand Op);
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000057 virtual void InstructionSelectBasicBlock(SelectionDAG &DAG);
Evan Chenga8e29892007-01-19 07:51:42 +000058 bool SelectAddrMode2(SDOperand Op, SDOperand N, SDOperand &Base,
59 SDOperand &Offset, SDOperand &Opc);
60 bool SelectAddrMode2Offset(SDOperand Op, SDOperand N,
61 SDOperand &Offset, SDOperand &Opc);
62 bool SelectAddrMode3(SDOperand Op, SDOperand N, SDOperand &Base,
63 SDOperand &Offset, SDOperand &Opc);
64 bool SelectAddrMode3Offset(SDOperand Op, SDOperand N,
65 SDOperand &Offset, SDOperand &Opc);
66 bool SelectAddrMode5(SDOperand Op, SDOperand N, SDOperand &Base,
Evan Cheng0d538262006-11-08 20:34:28 +000067 SDOperand &Offset);
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000068
Evan Chenga8e29892007-01-19 07:51:42 +000069 bool SelectAddrModePC(SDOperand Op, SDOperand N, SDOperand &Offset,
70 SDOperand &Label);
71
72 bool SelectThumbAddrModeRR(SDOperand Op, SDOperand N, SDOperand &Base,
73 SDOperand &Offset);
Evan Cheng79d43262007-01-24 02:21:22 +000074 bool SelectThumbAddrModeRI5(SDOperand Op, SDOperand N, unsigned Scale,
Evan Chengcea117d2007-01-30 02:35:32 +000075 SDOperand &Base, SDOperand &OffImm,
76 SDOperand &Offset);
Evan Chengc38f2bc2007-01-23 22:59:13 +000077 bool SelectThumbAddrModeS1(SDOperand Op, SDOperand N, SDOperand &Base,
Evan Chengcea117d2007-01-30 02:35:32 +000078 SDOperand &OffImm, SDOperand &Offset);
Evan Chengc38f2bc2007-01-23 22:59:13 +000079 bool SelectThumbAddrModeS2(SDOperand Op, SDOperand N, SDOperand &Base,
Evan Chengcea117d2007-01-30 02:35:32 +000080 SDOperand &OffImm, SDOperand &Offset);
Evan Chengc38f2bc2007-01-23 22:59:13 +000081 bool SelectThumbAddrModeS4(SDOperand Op, SDOperand N, SDOperand &Base,
Evan Chengcea117d2007-01-30 02:35:32 +000082 SDOperand &OffImm, SDOperand &Offset);
Evan Chenga8e29892007-01-19 07:51:42 +000083 bool SelectThumbAddrModeSP(SDOperand Op, SDOperand N, SDOperand &Base,
Evan Cheng79d43262007-01-24 02:21:22 +000084 SDOperand &OffImm);
Evan Chenga8e29892007-01-19 07:51:42 +000085
86 bool SelectShifterOperandReg(SDOperand Op, SDOperand N, SDOperand &A,
87 SDOperand &B, SDOperand &C);
88
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000089 // Include the pieces autogenerated from the target description.
90#include "ARMGenDAGISel.inc"
91};
Evan Chenga8e29892007-01-19 07:51:42 +000092}
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000093
94void ARMDAGToDAGISel::InstructionSelectBasicBlock(SelectionDAG &DAG) {
95 DEBUG(BB->dump());
96
97 DAG.setRoot(SelectRoot(DAG.getRoot()));
Rafael Espindola7bc59bc2006-05-14 22:18:28 +000098 DAG.RemoveDeadNodes();
99
100 ScheduleAndEmitDAG(DAG);
101}
102
Evan Cheng0d538262006-11-08 20:34:28 +0000103bool ARMDAGToDAGISel::SelectAddrMode2(SDOperand Op, SDOperand N,
Evan Chenga8e29892007-01-19 07:51:42 +0000104 SDOperand &Base, SDOperand &Offset,
105 SDOperand &Opc) {
Evan Chenga13fd102007-03-13 21:05:54 +0000106 if (N.getOpcode() == ISD::MUL) {
107 if (ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
108 // X * [3,5,9] -> X + X * [2,4,8] etc.
109 int RHSC = (int)RHS->getValue();
110 if (RHSC & 1) {
111 RHSC = RHSC & ~1;
112 ARM_AM::AddrOpc AddSub = ARM_AM::add;
113 if (RHSC < 0) {
114 AddSub = ARM_AM::sub;
115 RHSC = - RHSC;
116 }
117 if (isPowerOf2_32(RHSC)) {
118 unsigned ShAmt = Log2_32(RHSC);
119 Base = Offset = N.getOperand(0);
120 Opc = CurDAG->getTargetConstant(ARM_AM::getAM2Opc(AddSub, ShAmt,
121 ARM_AM::lsl),
122 MVT::i32);
123 return true;
124 }
125 }
126 }
127 }
128
Evan Chenga8e29892007-01-19 07:51:42 +0000129 if (N.getOpcode() != ISD::ADD && N.getOpcode() != ISD::SUB) {
130 Base = N;
131 if (N.getOpcode() == ISD::FrameIndex) {
132 int FI = cast<FrameIndexSDNode>(N)->getIndex();
133 Base = CurDAG->getTargetFrameIndex(FI, TLI.getPointerTy());
134 } else if (N.getOpcode() == ARMISD::Wrapper) {
135 Base = N.getOperand(0);
136 }
137 Offset = CurDAG->getRegister(0, MVT::i32);
138 Opc = CurDAG->getTargetConstant(ARM_AM::getAM2Opc(ARM_AM::add, 0,
139 ARM_AM::no_shift),
140 MVT::i32);
Rafael Espindola6e8c6492006-11-08 17:07:32 +0000141 return true;
142 }
Evan Chenga8e29892007-01-19 07:51:42 +0000143
144 // Match simple R +/- imm12 operands.
145 if (N.getOpcode() == ISD::ADD)
146 if (ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
147 int RHSC = (int)RHS->getValue();
Evan Chenge966d642007-01-24 02:45:25 +0000148 if ((RHSC >= 0 && RHSC < 0x1000) ||
149 (RHSC < 0 && RHSC > -0x1000)) { // 12 bits.
Evan Chenga8e29892007-01-19 07:51:42 +0000150 Base = N.getOperand(0);
Evan Chenge966d642007-01-24 02:45:25 +0000151 if (Base.getOpcode() == ISD::FrameIndex) {
152 int FI = cast<FrameIndexSDNode>(Base)->getIndex();
153 Base = CurDAG->getTargetFrameIndex(FI, TLI.getPointerTy());
154 }
Evan Chenga8e29892007-01-19 07:51:42 +0000155 Offset = CurDAG->getRegister(0, MVT::i32);
Evan Chenge966d642007-01-24 02:45:25 +0000156
157 ARM_AM::AddrOpc AddSub = ARM_AM::add;
158 if (RHSC < 0) {
159 AddSub = ARM_AM::sub;
160 RHSC = - RHSC;
161 }
162 Opc = CurDAG->getTargetConstant(ARM_AM::getAM2Opc(AddSub, RHSC,
Evan Chenga8e29892007-01-19 07:51:42 +0000163 ARM_AM::no_shift),
164 MVT::i32);
165 return true;
Rafael Espindola6e8c6492006-11-08 17:07:32 +0000166 }
Evan Chenga8e29892007-01-19 07:51:42 +0000167 }
168
169 // Otherwise this is R +/- [possibly shifted] R
170 ARM_AM::AddrOpc AddSub = N.getOpcode() == ISD::ADD ? ARM_AM::add:ARM_AM::sub;
171 ARM_AM::ShiftOpc ShOpcVal = ARM_AM::getShiftOpcForNode(N.getOperand(1));
172 unsigned ShAmt = 0;
173
174 Base = N.getOperand(0);
175 Offset = N.getOperand(1);
176
177 if (ShOpcVal != ARM_AM::no_shift) {
178 // Check to see if the RHS of the shift is a constant, if not, we can't fold
179 // it.
180 if (ConstantSDNode *Sh =
181 dyn_cast<ConstantSDNode>(N.getOperand(1).getOperand(1))) {
182 ShAmt = Sh->getValue();
183 Offset = N.getOperand(1).getOperand(0);
184 } else {
185 ShOpcVal = ARM_AM::no_shift;
Rafael Espindola6e8c6492006-11-08 17:07:32 +0000186 }
187 }
Evan Chenga8e29892007-01-19 07:51:42 +0000188
189 // Try matching (R shl C) + (R).
190 if (N.getOpcode() == ISD::ADD && ShOpcVal == ARM_AM::no_shift) {
191 ShOpcVal = ARM_AM::getShiftOpcForNode(N.getOperand(0));
192 if (ShOpcVal != ARM_AM::no_shift) {
193 // Check to see if the RHS of the shift is a constant, if not, we can't
194 // fold it.
195 if (ConstantSDNode *Sh =
196 dyn_cast<ConstantSDNode>(N.getOperand(0).getOperand(1))) {
197 ShAmt = Sh->getValue();
198 Offset = N.getOperand(0).getOperand(0);
199 Base = N.getOperand(1);
200 } else {
201 ShOpcVal = ARM_AM::no_shift;
202 }
203 }
204 }
205
206 Opc = CurDAG->getTargetConstant(ARM_AM::getAM2Opc(AddSub, ShAmt, ShOpcVal),
207 MVT::i32);
Rafael Espindola6e8c6492006-11-08 17:07:32 +0000208 return true;
209}
210
Evan Chenga8e29892007-01-19 07:51:42 +0000211bool ARMDAGToDAGISel::SelectAddrMode2Offset(SDOperand Op, SDOperand N,
212 SDOperand &Offset, SDOperand &Opc) {
213 unsigned Opcode = Op.getOpcode();
214 ISD::MemIndexedMode AM = (Opcode == ISD::LOAD)
215 ? cast<LoadSDNode>(Op)->getAddressingMode()
216 : cast<StoreSDNode>(Op)->getAddressingMode();
217 ARM_AM::AddrOpc AddSub = (AM == ISD::PRE_INC || AM == ISD::POST_INC)
218 ? ARM_AM::add : ARM_AM::sub;
219 if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(N)) {
220 int Val = (int)C->getValue();
221 if (Val >= 0 && Val < 0x1000) { // 12 bits.
222 Offset = CurDAG->getRegister(0, MVT::i32);
223 Opc = CurDAG->getTargetConstant(ARM_AM::getAM2Opc(AddSub, Val,
224 ARM_AM::no_shift),
225 MVT::i32);
226 return true;
227 }
228 }
229
230 Offset = N;
231 ARM_AM::ShiftOpc ShOpcVal = ARM_AM::getShiftOpcForNode(N);
232 unsigned ShAmt = 0;
233 if (ShOpcVal != ARM_AM::no_shift) {
234 // Check to see if the RHS of the shift is a constant, if not, we can't fold
235 // it.
236 if (ConstantSDNode *Sh = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
237 ShAmt = Sh->getValue();
238 Offset = N.getOperand(0);
239 } else {
240 ShOpcVal = ARM_AM::no_shift;
241 }
242 }
243
244 Opc = CurDAG->getTargetConstant(ARM_AM::getAM2Opc(AddSub, ShAmt, ShOpcVal),
245 MVT::i32);
Rafael Espindola32bd5f42006-10-17 18:04:53 +0000246 return true;
247}
248
Evan Chenga8e29892007-01-19 07:51:42 +0000249
250bool ARMDAGToDAGISel::SelectAddrMode3(SDOperand Op, SDOperand N,
251 SDOperand &Base, SDOperand &Offset,
252 SDOperand &Opc) {
253 if (N.getOpcode() == ISD::SUB) {
254 // X - C is canonicalize to X + -C, no need to handle it here.
255 Base = N.getOperand(0);
256 Offset = N.getOperand(1);
257 Opc = CurDAG->getTargetConstant(ARM_AM::getAM3Opc(ARM_AM::sub, 0),MVT::i32);
258 return true;
259 }
260
261 if (N.getOpcode() != ISD::ADD) {
262 Base = N;
263 if (N.getOpcode() == ISD::FrameIndex) {
264 int FI = cast<FrameIndexSDNode>(N)->getIndex();
265 Base = CurDAG->getTargetFrameIndex(FI, TLI.getPointerTy());
266 }
267 Offset = CurDAG->getRegister(0, MVT::i32);
268 Opc = CurDAG->getTargetConstant(ARM_AM::getAM3Opc(ARM_AM::add, 0),MVT::i32);
269 return true;
270 }
271
272 // If the RHS is +/- imm8, fold into addr mode.
273 if (ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
274 int RHSC = (int)RHS->getValue();
Evan Chenge966d642007-01-24 02:45:25 +0000275 if ((RHSC >= 0 && RHSC < 256) ||
276 (RHSC < 0 && RHSC > -256)) { // note -256 itself isn't allowed.
Evan Chenga8e29892007-01-19 07:51:42 +0000277 Base = N.getOperand(0);
Evan Chenge966d642007-01-24 02:45:25 +0000278 if (Base.getOpcode() == ISD::FrameIndex) {
279 int FI = cast<FrameIndexSDNode>(Base)->getIndex();
280 Base = CurDAG->getTargetFrameIndex(FI, TLI.getPointerTy());
281 }
Evan Chenga8e29892007-01-19 07:51:42 +0000282 Offset = CurDAG->getRegister(0, MVT::i32);
Evan Chenge966d642007-01-24 02:45:25 +0000283
284 ARM_AM::AddrOpc AddSub = ARM_AM::add;
285 if (RHSC < 0) {
286 AddSub = ARM_AM::sub;
287 RHSC = - RHSC;
288 }
289 Opc = CurDAG->getTargetConstant(ARM_AM::getAM3Opc(AddSub, RHSC),MVT::i32);
Evan Chenga8e29892007-01-19 07:51:42 +0000290 return true;
291 }
292 }
293
294 Base = N.getOperand(0);
295 Offset = N.getOperand(1);
296 Opc = CurDAG->getTargetConstant(ARM_AM::getAM3Opc(ARM_AM::add, 0), MVT::i32);
297 return true;
298}
299
300bool ARMDAGToDAGISel::SelectAddrMode3Offset(SDOperand Op, SDOperand N,
301 SDOperand &Offset, SDOperand &Opc) {
302 unsigned Opcode = Op.getOpcode();
303 ISD::MemIndexedMode AM = (Opcode == ISD::LOAD)
304 ? cast<LoadSDNode>(Op)->getAddressingMode()
305 : cast<StoreSDNode>(Op)->getAddressingMode();
306 ARM_AM::AddrOpc AddSub = (AM == ISD::PRE_INC || AM == ISD::POST_INC)
307 ? ARM_AM::add : ARM_AM::sub;
308 if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(N)) {
309 int Val = (int)C->getValue();
310 if (Val >= 0 && Val < 256) {
311 Offset = CurDAG->getRegister(0, MVT::i32);
312 Opc = CurDAG->getTargetConstant(ARM_AM::getAM3Opc(AddSub, Val), MVT::i32);
313 return true;
314 }
315 }
316
317 Offset = N;
318 Opc = CurDAG->getTargetConstant(ARM_AM::getAM3Opc(AddSub, 0), MVT::i32);
319 return true;
320}
321
322
323bool ARMDAGToDAGISel::SelectAddrMode5(SDOperand Op, SDOperand N,
324 SDOperand &Base, SDOperand &Offset) {
325 if (N.getOpcode() != ISD::ADD) {
326 Base = N;
327 if (N.getOpcode() == ISD::FrameIndex) {
328 int FI = cast<FrameIndexSDNode>(N)->getIndex();
329 Base = CurDAG->getTargetFrameIndex(FI, TLI.getPointerTy());
330 } else if (N.getOpcode() == ARMISD::Wrapper) {
331 Base = N.getOperand(0);
332 }
333 Offset = CurDAG->getTargetConstant(ARM_AM::getAM5Opc(ARM_AM::add, 0),
334 MVT::i32);
335 return true;
336 }
337
338 // If the RHS is +/- imm8, fold into addr mode.
339 if (ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
340 int RHSC = (int)RHS->getValue();
341 if ((RHSC & 3) == 0) { // The constant is implicitly multiplied by 4.
342 RHSC >>= 2;
Evan Chenge966d642007-01-24 02:45:25 +0000343 if ((RHSC >= 0 && RHSC < 256) ||
344 (RHSC < 0 && RHSC > -256)) { // note -256 itself isn't allowed.
Evan Chenga8e29892007-01-19 07:51:42 +0000345 Base = N.getOperand(0);
Evan Chenge966d642007-01-24 02:45:25 +0000346 if (Base.getOpcode() == ISD::FrameIndex) {
347 int FI = cast<FrameIndexSDNode>(Base)->getIndex();
348 Base = CurDAG->getTargetFrameIndex(FI, TLI.getPointerTy());
349 }
350
351 ARM_AM::AddrOpc AddSub = ARM_AM::add;
352 if (RHSC < 0) {
353 AddSub = ARM_AM::sub;
354 RHSC = - RHSC;
355 }
356 Offset = CurDAG->getTargetConstant(ARM_AM::getAM5Opc(AddSub, RHSC),
Evan Chenga8e29892007-01-19 07:51:42 +0000357 MVT::i32);
358 return true;
359 }
360 }
361 }
362
363 Base = N;
364 Offset = CurDAG->getTargetConstant(ARM_AM::getAM5Opc(ARM_AM::add, 0),
365 MVT::i32);
366 return true;
367}
368
369bool ARMDAGToDAGISel::SelectAddrModePC(SDOperand Op, SDOperand N,
370 SDOperand &Offset, SDOperand &Label) {
371 if (N.getOpcode() == ARMISD::PIC_ADD && N.hasOneUse()) {
372 Offset = N.getOperand(0);
373 SDOperand N1 = N.getOperand(1);
374 Label = CurDAG->getTargetConstant(cast<ConstantSDNode>(N1)->getValue(),
375 MVT::i32);
376 return true;
377 }
378 return false;
379}
380
381bool ARMDAGToDAGISel::SelectThumbAddrModeRR(SDOperand Op, SDOperand N,
382 SDOperand &Base, SDOperand &Offset){
Evan Chengc38f2bc2007-01-23 22:59:13 +0000383 if (N.getOpcode() != ISD::ADD) {
384 Base = N;
385 // We must materialize a zero in a reg! Returning an constant here won't
386 // work since its node is -1 so it won't get added to the selection queue.
387 // Explicitly issue a tMOVri8 node!
Evan Cheng9f6636f2007-03-19 07:48:02 +0000388 Offset = SDOperand(CurDAG->getTargetNode(ARM::tMOVi8, MVT::i32,
Evan Chengc38f2bc2007-01-23 22:59:13 +0000389 CurDAG->getTargetConstant(0, MVT::i32)), 0);
390 return true;
391 }
392
Evan Chenga8e29892007-01-19 07:51:42 +0000393 Base = N.getOperand(0);
394 Offset = N.getOperand(1);
395 return true;
396}
397
Evan Cheng79d43262007-01-24 02:21:22 +0000398bool
399ARMDAGToDAGISel::SelectThumbAddrModeRI5(SDOperand Op, SDOperand N,
400 unsigned Scale, SDOperand &Base,
Evan Chengcea117d2007-01-30 02:35:32 +0000401 SDOperand &OffImm, SDOperand &Offset) {
Evan Cheng79d43262007-01-24 02:21:22 +0000402 if (Scale == 4) {
403 SDOperand TmpBase, TmpOffImm;
404 if (SelectThumbAddrModeSP(Op, N, TmpBase, TmpOffImm))
405 return false; // We want to select tLDRspi / tSTRspi instead.
Evan Cheng012f2d92007-01-24 08:53:17 +0000406 if (N.getOpcode() == ARMISD::Wrapper &&
407 N.getOperand(0).getOpcode() == ISD::TargetConstantPool)
408 return false; // We want to select tLDRpci instead.
Evan Cheng79d43262007-01-24 02:21:22 +0000409 }
410
Evan Chenga8e29892007-01-19 07:51:42 +0000411 if (N.getOpcode() != ISD::ADD) {
412 Base = (N.getOpcode() == ARMISD::Wrapper) ? N.getOperand(0) : N;
Evan Chengc38f2bc2007-01-23 22:59:13 +0000413 Offset = CurDAG->getRegister(0, MVT::i32);
414 OffImm = CurDAG->getTargetConstant(0, MVT::i32);
Evan Chenga8e29892007-01-19 07:51:42 +0000415 return true;
416 }
417
Evan Chengad0e4652007-02-06 00:22:06 +0000418 // Thumb does not have [sp, r] address mode.
419 RegisterSDNode *LHSR = dyn_cast<RegisterSDNode>(N.getOperand(0));
420 RegisterSDNode *RHSR = dyn_cast<RegisterSDNode>(N.getOperand(1));
421 if ((LHSR && LHSR->getReg() == ARM::SP) ||
422 (RHSR && RHSR->getReg() == ARM::SP)) {
423 Base = N;
424 Offset = CurDAG->getRegister(0, MVT::i32);
425 OffImm = CurDAG->getTargetConstant(0, MVT::i32);
426 return true;
427 }
428
Evan Chenga8e29892007-01-19 07:51:42 +0000429 // If the RHS is + imm5 * scale, fold into addr mode.
430 if (ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
431 int RHSC = (int)RHS->getValue();
432 if ((RHSC & (Scale-1)) == 0) { // The constant is implicitly multiplied.
433 RHSC /= Scale;
434 if (RHSC >= 0 && RHSC < 32) {
435 Base = N.getOperand(0);
Evan Chengc38f2bc2007-01-23 22:59:13 +0000436 Offset = CurDAG->getRegister(0, MVT::i32);
437 OffImm = CurDAG->getTargetConstant(RHSC, MVT::i32);
Evan Chenga8e29892007-01-19 07:51:42 +0000438 return true;
439 }
440 }
441 }
442
Evan Chengc38f2bc2007-01-23 22:59:13 +0000443 Base = N.getOperand(0);
444 Offset = N.getOperand(1);
445 OffImm = CurDAG->getTargetConstant(0, MVT::i32);
446 return true;
Evan Chenga8e29892007-01-19 07:51:42 +0000447}
448
Evan Chengc38f2bc2007-01-23 22:59:13 +0000449bool ARMDAGToDAGISel::SelectThumbAddrModeS1(SDOperand Op, SDOperand N,
Evan Chengcea117d2007-01-30 02:35:32 +0000450 SDOperand &Base, SDOperand &OffImm,
451 SDOperand &Offset) {
452 return SelectThumbAddrModeRI5(Op, N, 1, Base, OffImm, Offset);
Evan Chenga8e29892007-01-19 07:51:42 +0000453}
454
Evan Chengc38f2bc2007-01-23 22:59:13 +0000455bool ARMDAGToDAGISel::SelectThumbAddrModeS2(SDOperand Op, SDOperand N,
Evan Chengcea117d2007-01-30 02:35:32 +0000456 SDOperand &Base, SDOperand &OffImm,
457 SDOperand &Offset) {
458 return SelectThumbAddrModeRI5(Op, N, 2, Base, OffImm, Offset);
Evan Chenga8e29892007-01-19 07:51:42 +0000459}
460
Evan Chengc38f2bc2007-01-23 22:59:13 +0000461bool ARMDAGToDAGISel::SelectThumbAddrModeS4(SDOperand Op, SDOperand N,
Evan Chengcea117d2007-01-30 02:35:32 +0000462 SDOperand &Base, SDOperand &OffImm,
463 SDOperand &Offset) {
464 return SelectThumbAddrModeRI5(Op, N, 4, Base, OffImm, Offset);
Evan Chenga8e29892007-01-19 07:51:42 +0000465}
466
467bool ARMDAGToDAGISel::SelectThumbAddrModeSP(SDOperand Op, SDOperand N,
Evan Cheng79d43262007-01-24 02:21:22 +0000468 SDOperand &Base, SDOperand &OffImm) {
Evan Chenga8e29892007-01-19 07:51:42 +0000469 if (N.getOpcode() == ISD::FrameIndex) {
470 int FI = cast<FrameIndexSDNode>(N)->getIndex();
471 Base = CurDAG->getTargetFrameIndex(FI, TLI.getPointerTy());
Evan Cheng79d43262007-01-24 02:21:22 +0000472 OffImm = CurDAG->getTargetConstant(0, MVT::i32);
Evan Chenga8e29892007-01-19 07:51:42 +0000473 return true;
474 }
Evan Cheng79d43262007-01-24 02:21:22 +0000475
Evan Chengad0e4652007-02-06 00:22:06 +0000476 if (N.getOpcode() != ISD::ADD)
477 return false;
478
479 RegisterSDNode *LHSR = dyn_cast<RegisterSDNode>(N.getOperand(0));
Evan Cheng8c1a73a2007-02-06 09:11:20 +0000480 if (N.getOperand(0).getOpcode() == ISD::FrameIndex ||
481 (LHSR && LHSR->getReg() == ARM::SP)) {
Evan Cheng79d43262007-01-24 02:21:22 +0000482 // If the RHS is + imm8 * scale, fold into addr mode.
483 if (ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
484 int RHSC = (int)RHS->getValue();
485 if ((RHSC & 3) == 0) { // The constant is implicitly multiplied.
486 RHSC >>= 2;
487 if (RHSC >= 0 && RHSC < 256) {
Evan Chengad0e4652007-02-06 00:22:06 +0000488 Base = N.getOperand(0);
Evan Cheng8c1a73a2007-02-06 09:11:20 +0000489 if (Base.getOpcode() == ISD::FrameIndex) {
490 int FI = cast<FrameIndexSDNode>(Base)->getIndex();
491 Base = CurDAG->getTargetFrameIndex(FI, TLI.getPointerTy());
492 }
Evan Cheng79d43262007-01-24 02:21:22 +0000493 OffImm = CurDAG->getTargetConstant(RHSC, MVT::i32);
494 return true;
495 }
496 }
497 }
498 }
Evan Chenga8e29892007-01-19 07:51:42 +0000499
500 return false;
501}
502
503bool ARMDAGToDAGISel::SelectShifterOperandReg(SDOperand Op,
504 SDOperand N,
505 SDOperand &BaseReg,
506 SDOperand &ShReg,
507 SDOperand &Opc) {
508 ARM_AM::ShiftOpc ShOpcVal = ARM_AM::getShiftOpcForNode(N);
509
510 // Don't match base register only case. That is matched to a separate
511 // lower complexity pattern with explicit register operand.
512 if (ShOpcVal == ARM_AM::no_shift) return false;
513
514 BaseReg = N.getOperand(0);
515 unsigned ShImmVal = 0;
516 if (ConstantSDNode *RHS = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
517 ShReg = CurDAG->getRegister(0, MVT::i32);
518 ShImmVal = RHS->getValue() & 31;
519 } else {
520 ShReg = N.getOperand(1);
521 }
522 Opc = CurDAG->getTargetConstant(ARM_AM::getSORegOpc(ShOpcVal, ShImmVal),
523 MVT::i32);
524 return true;
525}
526
527
Evan Cheng9ade2182006-08-26 05:34:46 +0000528SDNode *ARMDAGToDAGISel::Select(SDOperand Op) {
Rafael Espindola337c4ad62006-06-12 12:28:08 +0000529 SDNode *N = Op.Val;
Evan Chenga8e29892007-01-19 07:51:42 +0000530 unsigned Opcode = N->getOpcode();
531
532 if (Opcode >= ISD::BUILTIN_OP_END && Opcode < ARMISD::FIRST_NUMBER)
533 return NULL; // Already selected.
Rafael Espindola337c4ad62006-06-12 12:28:08 +0000534
535 switch (N->getOpcode()) {
Evan Chenga8e29892007-01-19 07:51:42 +0000536 default: break;
537 case ISD::Constant: {
538 unsigned Val = cast<ConstantSDNode>(N)->getValue();
539 bool UseCP = true;
540 if (Subtarget->isThumb())
541 UseCP = (Val > 255 && // MOV
542 ~Val > 255 && // MOV + MVN
543 !ARM_AM::isThumbImmShiftedVal(Val)); // MOV + LSL
544 else
545 UseCP = (ARM_AM::getSOImmVal(Val) == -1 && // MOV
546 ARM_AM::getSOImmVal(~Val) == -1 && // MVN
547 !ARM_AM::isSOImmTwoPartVal(Val)); // two instrs.
548 if (UseCP) {
549 SDOperand CPIdx =
550 CurDAG->getTargetConstantPool(ConstantInt::get(Type::Int32Ty, Val),
551 TLI.getPointerTy());
Evan Cheng012f2d92007-01-24 08:53:17 +0000552
553 SDNode *ResNode;
554 if (Subtarget->isThumb())
Evan Chengfa775d02007-03-19 07:20:03 +0000555 ResNode = CurDAG->getTargetNode(ARM::tLDRcp, MVT::i32, MVT::Other,
Evan Cheng012f2d92007-01-24 08:53:17 +0000556 CPIdx, CurDAG->getEntryNode());
557 else {
558 SDOperand Ops[] = {
559 CPIdx,
560 CurDAG->getRegister(0, MVT::i32),
561 CurDAG->getTargetConstant(0, MVT::i32),
562 CurDAG->getEntryNode()
563 };
Evan Chengfa775d02007-03-19 07:20:03 +0000564 ResNode=CurDAG->getTargetNode(ARM::LDRcp, MVT::i32, MVT::Other, Ops, 4);
Evan Cheng012f2d92007-01-24 08:53:17 +0000565 }
Evan Chenga8e29892007-01-19 07:51:42 +0000566 ReplaceUses(Op, SDOperand(ResNode, 0));
567 return NULL;
568 }
569
570 // Other cases are autogenerated.
Rafael Espindola337c4ad62006-06-12 12:28:08 +0000571 break;
Evan Chenga8e29892007-01-19 07:51:42 +0000572 }
Rafael Espindolaf819a492006-11-09 13:58:55 +0000573 case ISD::FrameIndex: {
Evan Chenga8e29892007-01-19 07:51:42 +0000574 // Selects to ADDri FI, 0 which in turn will become ADDri SP, imm.
Rafael Espindolaf819a492006-11-09 13:58:55 +0000575 int FI = cast<FrameIndexSDNode>(N)->getIndex();
Evan Chenga8e29892007-01-19 07:51:42 +0000576 unsigned Opc = Subtarget->isThumb() ? ARM::tADDrSPi : ARM::ADDri;
577 SDOperand TFI = CurDAG->getTargetFrameIndex(FI, TLI.getPointerTy());
578 return CurDAG->SelectNodeTo(N, Opc, MVT::i32, TFI,
579 CurDAG->getTargetConstant(0, MVT::i32));
580 }
Evan Chengad0e4652007-02-06 00:22:06 +0000581 case ISD::ADD: {
582 // Select add sp, c to tADDhirr.
583 SDOperand N0 = Op.getOperand(0);
584 SDOperand N1 = Op.getOperand(1);
585 RegisterSDNode *LHSR = dyn_cast<RegisterSDNode>(Op.getOperand(0));
586 RegisterSDNode *RHSR = dyn_cast<RegisterSDNode>(Op.getOperand(1));
587 if (LHSR && LHSR->getReg() == ARM::SP) {
588 std::swap(N0, N1);
589 std::swap(LHSR, RHSR);
590 }
591 if (RHSR && RHSR->getReg() == ARM::SP) {
592 AddToISelQueue(N0);
593 AddToISelQueue(N1);
594 return CurDAG->SelectNodeTo(N, ARM::tADDhirr, Op.getValueType(), N0, N1);
595 }
596 break;
597 }
Evan Chenga8e29892007-01-19 07:51:42 +0000598 case ISD::MUL:
Evan Cheng79d43262007-01-24 02:21:22 +0000599 if (Subtarget->isThumb())
600 break;
Evan Chenga8e29892007-01-19 07:51:42 +0000601 if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
602 unsigned RHSV = C->getValue();
603 if (!RHSV) break;
604 if (isPowerOf2_32(RHSV-1)) { // 2^n+1?
605 SDOperand V = Op.getOperand(0);
606 AddToISelQueue(V);
607 unsigned ShImm = ARM_AM::getSORegOpc(ARM_AM::lsl, Log2_32(RHSV-1));
608 SDOperand Ops[] = { V, V, CurDAG->getRegister(0, MVT::i32),
609 CurDAG->getTargetConstant(ShImm, MVT::i32)
610 };
611 return CurDAG->SelectNodeTo(N, ARM::ADDrs, MVT::i32, Ops, 4);
612 }
613 if (isPowerOf2_32(RHSV+1)) { // 2^n-1?
614 SDOperand V = Op.getOperand(0);
615 AddToISelQueue(V);
616 unsigned ShImm = ARM_AM::getSORegOpc(ARM_AM::lsl, Log2_32(RHSV+1));
617 SDOperand Ops[] = { V, V, CurDAG->getRegister(0, MVT::i32),
618 CurDAG->getTargetConstant(ShImm, MVT::i32)
619 };
620 return CurDAG->SelectNodeTo(N, ARM::RSBrs, MVT::i32, Ops, 4);
621 }
622 }
623 break;
624 case ARMISD::FMRRD:
625 AddToISelQueue(Op.getOperand(0));
626 return CurDAG->getTargetNode(ARM::FMRRD, MVT::i32, MVT::i32,
627 Op.getOperand(0));
628 case ARMISD::MULHILOU:
629 AddToISelQueue(Op.getOperand(0));
630 AddToISelQueue(Op.getOperand(1));
631 return CurDAG->getTargetNode(ARM::UMULL, MVT::i32, MVT::i32,
632 Op.getOperand(0), Op.getOperand(1));
633 case ARMISD::MULHILOS:
634 AddToISelQueue(Op.getOperand(0));
635 AddToISelQueue(Op.getOperand(1));
636 return CurDAG->getTargetNode(ARM::SMULL, MVT::i32, MVT::i32,
637 Op.getOperand(0), Op.getOperand(1));
638 case ISD::LOAD: {
639 LoadSDNode *LD = cast<LoadSDNode>(Op);
640 ISD::MemIndexedMode AM = LD->getAddressingMode();
641 MVT::ValueType LoadedVT = LD->getLoadedVT();
642 if (AM != ISD::UNINDEXED) {
643 SDOperand Offset, AMOpc;
644 bool isPre = (AM == ISD::PRE_INC) || (AM == ISD::PRE_DEC);
645 unsigned Opcode = 0;
646 bool Match = false;
647 if (LoadedVT == MVT::i32 &&
648 SelectAddrMode2Offset(Op, LD->getOffset(), Offset, AMOpc)) {
649 Opcode = isPre ? ARM::LDR_PRE : ARM::LDR_POST;
650 Match = true;
651 } else if (LoadedVT == MVT::i16 &&
652 SelectAddrMode3Offset(Op, LD->getOffset(), Offset, AMOpc)) {
653 Match = true;
654 Opcode = (LD->getExtensionType() == ISD::SEXTLOAD)
655 ? (isPre ? ARM::LDRSH_PRE : ARM::LDRSH_POST)
656 : (isPre ? ARM::LDRH_PRE : ARM::LDRH_POST);
657 } else if (LoadedVT == MVT::i8 || LoadedVT == MVT::i1) {
658 if (LD->getExtensionType() == ISD::SEXTLOAD) {
659 if (SelectAddrMode3Offset(Op, LD->getOffset(), Offset, AMOpc)) {
660 Match = true;
661 Opcode = isPre ? ARM::LDRSB_PRE : ARM::LDRSB_POST;
662 }
663 } else {
664 if (SelectAddrMode2Offset(Op, LD->getOffset(), Offset, AMOpc)) {
665 Match = true;
666 Opcode = isPre ? ARM::LDRB_PRE : ARM::LDRB_POST;
667 }
668 }
669 }
Rafael Espindolaf819a492006-11-09 13:58:55 +0000670
Evan Chenga8e29892007-01-19 07:51:42 +0000671 if (Match) {
672 SDOperand Chain = LD->getChain();
673 SDOperand Base = LD->getBasePtr();
674 AddToISelQueue(Chain);
675 AddToISelQueue(Base);
676 AddToISelQueue(Offset);
677 SDOperand Ops[] = { Base, Offset, AMOpc, Chain };
678 return CurDAG->getTargetNode(Opcode, MVT::i32, MVT::i32,
679 MVT::Other, Ops, 4);
680 }
681 }
682 // Other cases are autogenerated.
Rafael Espindolaf819a492006-11-09 13:58:55 +0000683 break;
Rafael Espindola337c4ad62006-06-12 12:28:08 +0000684 }
Rafael Espindolaf819a492006-11-09 13:58:55 +0000685 }
Rafael Espindola7bc59bc2006-05-14 22:18:28 +0000686
Evan Chenga8e29892007-01-19 07:51:42 +0000687 return SelectCode(Op);
688}
Rafael Espindola7bc59bc2006-05-14 22:18:28 +0000689
690/// createARMISelDag - This pass converts a legalized DAG into a
691/// ARM-specific DAG, ready for instruction scheduling.
692///
Evan Chenga8e29892007-01-19 07:51:42 +0000693FunctionPass *llvm::createARMISelDag(ARMTargetMachine &TM) {
Rafael Espindola7bc59bc2006-05-14 22:18:28 +0000694 return new ARMDAGToDAGISel(TM);
695}