-Revert parts of 84326 and 84411. Distinquishing between fixed and non-fixed
[oota-llvm.git] / lib / Target / ARM / ARMBaseInstrInfo.cpp
1 //===- ARMBaseInstrInfo.cpp - ARM 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 Base ARM implementation of the TargetInstrInfo class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "ARMBaseInstrInfo.h"
15 #include "ARM.h"
16 #include "ARMAddressingModes.h"
17 #include "ARMGenInstrInfo.inc"
18 #include "ARMMachineFunctionInfo.h"
19 #include "llvm/ADT/STLExtras.h"
20 #include "llvm/CodeGen/LiveVariables.h"
21 #include "llvm/CodeGen/MachineFrameInfo.h"
22 #include "llvm/CodeGen/MachineInstrBuilder.h"
23 #include "llvm/CodeGen/MachineJumpTableInfo.h"
24 #include "llvm/CodeGen/MachineMemOperand.h"
25 #include "llvm/CodeGen/PseudoSourceValue.h"
26 #include "llvm/MC/MCAsmInfo.h"
27 #include "llvm/Support/CommandLine.h"
28 #include "llvm/Support/ErrorHandling.h"
29 using namespace llvm;
30
31 static cl::opt<bool>
32 EnableARM3Addr("enable-arm-3-addr-conv", cl::Hidden,
33                cl::desc("Enable ARM 2-addr to 3-addr conv"));
34
35 ARMBaseInstrInfo::ARMBaseInstrInfo()
36   : TargetInstrInfoImpl(ARMInsts, array_lengthof(ARMInsts)) {
37 }
38
39 MachineInstr *
40 ARMBaseInstrInfo::convertToThreeAddress(MachineFunction::iterator &MFI,
41                                         MachineBasicBlock::iterator &MBBI,
42                                         LiveVariables *LV) const {
43   // FIXME: Thumb2 support.
44
45   if (!EnableARM3Addr)
46     return NULL;
47
48   MachineInstr *MI = MBBI;
49   MachineFunction &MF = *MI->getParent()->getParent();
50   unsigned TSFlags = MI->getDesc().TSFlags;
51   bool isPre = false;
52   switch ((TSFlags & ARMII::IndexModeMask) >> ARMII::IndexModeShift) {
53   default: return NULL;
54   case ARMII::IndexModePre:
55     isPre = true;
56     break;
57   case ARMII::IndexModePost:
58     break;
59   }
60
61   // Try splitting an indexed load/store to an un-indexed one plus an add/sub
62   // operation.
63   unsigned MemOpc = getUnindexedOpcode(MI->getOpcode());
64   if (MemOpc == 0)
65     return NULL;
66
67   MachineInstr *UpdateMI = NULL;
68   MachineInstr *MemMI = NULL;
69   unsigned AddrMode = (TSFlags & ARMII::AddrModeMask);
70   const TargetInstrDesc &TID = MI->getDesc();
71   unsigned NumOps = TID.getNumOperands();
72   bool isLoad = !TID.mayStore();
73   const MachineOperand &WB = isLoad ? MI->getOperand(1) : MI->getOperand(0);
74   const MachineOperand &Base = MI->getOperand(2);
75   const MachineOperand &Offset = MI->getOperand(NumOps-3);
76   unsigned WBReg = WB.getReg();
77   unsigned BaseReg = Base.getReg();
78   unsigned OffReg = Offset.getReg();
79   unsigned OffImm = MI->getOperand(NumOps-2).getImm();
80   ARMCC::CondCodes Pred = (ARMCC::CondCodes)MI->getOperand(NumOps-1).getImm();
81   switch (AddrMode) {
82   default:
83     assert(false && "Unknown indexed op!");
84     return NULL;
85   case ARMII::AddrMode2: {
86     bool isSub = ARM_AM::getAM2Op(OffImm) == ARM_AM::sub;
87     unsigned Amt = ARM_AM::getAM2Offset(OffImm);
88     if (OffReg == 0) {
89       if (ARM_AM::getSOImmVal(Amt) == -1)
90         // Can't encode it in a so_imm operand. This transformation will
91         // add more than 1 instruction. Abandon!
92         return NULL;
93       UpdateMI = BuildMI(MF, MI->getDebugLoc(),
94                          get(isSub ? ARM::SUBri : ARM::ADDri), WBReg)
95         .addReg(BaseReg).addImm(Amt)
96         .addImm(Pred).addReg(0).addReg(0);
97     } else if (Amt != 0) {
98       ARM_AM::ShiftOpc ShOpc = ARM_AM::getAM2ShiftOpc(OffImm);
99       unsigned SOOpc = ARM_AM::getSORegOpc(ShOpc, Amt);
100       UpdateMI = BuildMI(MF, MI->getDebugLoc(),
101                          get(isSub ? ARM::SUBrs : ARM::ADDrs), WBReg)
102         .addReg(BaseReg).addReg(OffReg).addReg(0).addImm(SOOpc)
103         .addImm(Pred).addReg(0).addReg(0);
104     } else
105       UpdateMI = BuildMI(MF, MI->getDebugLoc(),
106                          get(isSub ? ARM::SUBrr : ARM::ADDrr), WBReg)
107         .addReg(BaseReg).addReg(OffReg)
108         .addImm(Pred).addReg(0).addReg(0);
109     break;
110   }
111   case ARMII::AddrMode3 : {
112     bool isSub = ARM_AM::getAM3Op(OffImm) == ARM_AM::sub;
113     unsigned Amt = ARM_AM::getAM3Offset(OffImm);
114     if (OffReg == 0)
115       // Immediate is 8-bits. It's guaranteed to fit in a so_imm operand.
116       UpdateMI = BuildMI(MF, MI->getDebugLoc(),
117                          get(isSub ? ARM::SUBri : ARM::ADDri), WBReg)
118         .addReg(BaseReg).addImm(Amt)
119         .addImm(Pred).addReg(0).addReg(0);
120     else
121       UpdateMI = BuildMI(MF, MI->getDebugLoc(),
122                          get(isSub ? ARM::SUBrr : ARM::ADDrr), WBReg)
123         .addReg(BaseReg).addReg(OffReg)
124         .addImm(Pred).addReg(0).addReg(0);
125     break;
126   }
127   }
128
129   std::vector<MachineInstr*> NewMIs;
130   if (isPre) {
131     if (isLoad)
132       MemMI = BuildMI(MF, MI->getDebugLoc(),
133                       get(MemOpc), MI->getOperand(0).getReg())
134         .addReg(WBReg).addReg(0).addImm(0).addImm(Pred);
135     else
136       MemMI = BuildMI(MF, MI->getDebugLoc(),
137                       get(MemOpc)).addReg(MI->getOperand(1).getReg())
138         .addReg(WBReg).addReg(0).addImm(0).addImm(Pred);
139     NewMIs.push_back(MemMI);
140     NewMIs.push_back(UpdateMI);
141   } else {
142     if (isLoad)
143       MemMI = BuildMI(MF, MI->getDebugLoc(),
144                       get(MemOpc), MI->getOperand(0).getReg())
145         .addReg(BaseReg).addReg(0).addImm(0).addImm(Pred);
146     else
147       MemMI = BuildMI(MF, MI->getDebugLoc(),
148                       get(MemOpc)).addReg(MI->getOperand(1).getReg())
149         .addReg(BaseReg).addReg(0).addImm(0).addImm(Pred);
150     if (WB.isDead())
151       UpdateMI->getOperand(0).setIsDead();
152     NewMIs.push_back(UpdateMI);
153     NewMIs.push_back(MemMI);
154   }
155
156   // Transfer LiveVariables states, kill / dead info.
157   if (LV) {
158     for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
159       MachineOperand &MO = MI->getOperand(i);
160       if (MO.isReg() && MO.getReg() &&
161           TargetRegisterInfo::isVirtualRegister(MO.getReg())) {
162         unsigned Reg = MO.getReg();
163
164         LiveVariables::VarInfo &VI = LV->getVarInfo(Reg);
165         if (MO.isDef()) {
166           MachineInstr *NewMI = (Reg == WBReg) ? UpdateMI : MemMI;
167           if (MO.isDead())
168             LV->addVirtualRegisterDead(Reg, NewMI);
169         }
170         if (MO.isUse() && MO.isKill()) {
171           for (unsigned j = 0; j < 2; ++j) {
172             // Look at the two new MI's in reverse order.
173             MachineInstr *NewMI = NewMIs[j];
174             if (!NewMI->readsRegister(Reg))
175               continue;
176             LV->addVirtualRegisterKilled(Reg, NewMI);
177             if (VI.removeKill(MI))
178               VI.Kills.push_back(NewMI);
179             break;
180           }
181         }
182       }
183     }
184   }
185
186   MFI->insert(MBBI, NewMIs[1]);
187   MFI->insert(MBBI, NewMIs[0]);
188   return NewMIs[0];
189 }
190
191 // Branch analysis.
192 bool
193 ARMBaseInstrInfo::AnalyzeBranch(MachineBasicBlock &MBB,MachineBasicBlock *&TBB,
194                                 MachineBasicBlock *&FBB,
195                                 SmallVectorImpl<MachineOperand> &Cond,
196                                 bool AllowModify) const {
197   // If the block has no terminators, it just falls into the block after it.
198   MachineBasicBlock::iterator I = MBB.end();
199   if (I == MBB.begin() || !isUnpredicatedTerminator(--I))
200     return false;
201
202   // Get the last instruction in the block.
203   MachineInstr *LastInst = I;
204
205   // If there is only one terminator instruction, process it.
206   unsigned LastOpc = LastInst->getOpcode();
207   if (I == MBB.begin() || !isUnpredicatedTerminator(--I)) {
208     if (isUncondBranchOpcode(LastOpc)) {
209       TBB = LastInst->getOperand(0).getMBB();
210       return false;
211     }
212     if (isCondBranchOpcode(LastOpc)) {
213       // Block ends with fall-through condbranch.
214       TBB = LastInst->getOperand(0).getMBB();
215       Cond.push_back(LastInst->getOperand(1));
216       Cond.push_back(LastInst->getOperand(2));
217       return false;
218     }
219     return true;  // Can't handle indirect branch.
220   }
221
222   // Get the instruction before it if it is a terminator.
223   MachineInstr *SecondLastInst = I;
224
225   // If there are three terminators, we don't know what sort of block this is.
226   if (SecondLastInst && I != MBB.begin() && isUnpredicatedTerminator(--I))
227     return true;
228
229   // If the block ends with a B and a Bcc, handle it.
230   unsigned SecondLastOpc = SecondLastInst->getOpcode();
231   if (isCondBranchOpcode(SecondLastOpc) && isUncondBranchOpcode(LastOpc)) {
232     TBB =  SecondLastInst->getOperand(0).getMBB();
233     Cond.push_back(SecondLastInst->getOperand(1));
234     Cond.push_back(SecondLastInst->getOperand(2));
235     FBB = LastInst->getOperand(0).getMBB();
236     return false;
237   }
238
239   // If the block ends with two unconditional branches, handle it.  The second
240   // one is not executed, so remove it.
241   if (isUncondBranchOpcode(SecondLastOpc) && isUncondBranchOpcode(LastOpc)) {
242     TBB = SecondLastInst->getOperand(0).getMBB();
243     I = LastInst;
244     if (AllowModify)
245       I->eraseFromParent();
246     return false;
247   }
248
249   // ...likewise if it ends with a branch table followed by an unconditional
250   // branch. The branch folder can create these, and we must get rid of them for
251   // correctness of Thumb constant islands.
252   if (isJumpTableBranchOpcode(SecondLastOpc) &&
253       isUncondBranchOpcode(LastOpc)) {
254     I = LastInst;
255     if (AllowModify)
256       I->eraseFromParent();
257     return true;
258   }
259
260   // Otherwise, can't handle this.
261   return true;
262 }
263
264
265 unsigned ARMBaseInstrInfo::RemoveBranch(MachineBasicBlock &MBB) const {
266   MachineBasicBlock::iterator I = MBB.end();
267   if (I == MBB.begin()) return 0;
268   --I;
269   if (!isUncondBranchOpcode(I->getOpcode()) &&
270       !isCondBranchOpcode(I->getOpcode()))
271     return 0;
272
273   // Remove the branch.
274   I->eraseFromParent();
275
276   I = MBB.end();
277
278   if (I == MBB.begin()) return 1;
279   --I;
280   if (!isCondBranchOpcode(I->getOpcode()))
281     return 1;
282
283   // Remove the branch.
284   I->eraseFromParent();
285   return 2;
286 }
287
288 unsigned
289 ARMBaseInstrInfo::InsertBranch(MachineBasicBlock &MBB, MachineBasicBlock *TBB,
290                                MachineBasicBlock *FBB,
291                              const SmallVectorImpl<MachineOperand> &Cond) const {
292   // FIXME this should probably have a DebugLoc argument
293   DebugLoc dl = DebugLoc::getUnknownLoc();
294
295   ARMFunctionInfo *AFI = MBB.getParent()->getInfo<ARMFunctionInfo>();
296   int BOpc   = !AFI->isThumbFunction()
297     ? ARM::B : (AFI->isThumb2Function() ? ARM::t2B : ARM::tB);
298   int BccOpc = !AFI->isThumbFunction()
299     ? ARM::Bcc : (AFI->isThumb2Function() ? ARM::t2Bcc : ARM::tBcc);
300
301   // Shouldn't be a fall through.
302   assert(TBB && "InsertBranch must not be told to insert a fallthrough");
303   assert((Cond.size() == 2 || Cond.size() == 0) &&
304          "ARM branch conditions have two components!");
305
306   if (FBB == 0) {
307     if (Cond.empty()) // Unconditional branch?
308       BuildMI(&MBB, dl, get(BOpc)).addMBB(TBB);
309     else
310       BuildMI(&MBB, dl, get(BccOpc)).addMBB(TBB)
311         .addImm(Cond[0].getImm()).addReg(Cond[1].getReg());
312     return 1;
313   }
314
315   // Two-way conditional branch.
316   BuildMI(&MBB, dl, get(BccOpc)).addMBB(TBB)
317     .addImm(Cond[0].getImm()).addReg(Cond[1].getReg());
318   BuildMI(&MBB, dl, get(BOpc)).addMBB(FBB);
319   return 2;
320 }
321
322 bool ARMBaseInstrInfo::
323 ReverseBranchCondition(SmallVectorImpl<MachineOperand> &Cond) const {
324   ARMCC::CondCodes CC = (ARMCC::CondCodes)(int)Cond[0].getImm();
325   Cond[0].setImm(ARMCC::getOppositeCondition(CC));
326   return false;
327 }
328
329 bool ARMBaseInstrInfo::
330 PredicateInstruction(MachineInstr *MI,
331                      const SmallVectorImpl<MachineOperand> &Pred) const {
332   unsigned Opc = MI->getOpcode();
333   if (isUncondBranchOpcode(Opc)) {
334     MI->setDesc(get(getMatchingCondBranchOpcode(Opc)));
335     MI->addOperand(MachineOperand::CreateImm(Pred[0].getImm()));
336     MI->addOperand(MachineOperand::CreateReg(Pred[1].getReg(), false));
337     return true;
338   }
339
340   int PIdx = MI->findFirstPredOperandIdx();
341   if (PIdx != -1) {
342     MachineOperand &PMO = MI->getOperand(PIdx);
343     PMO.setImm(Pred[0].getImm());
344     MI->getOperand(PIdx+1).setReg(Pred[1].getReg());
345     return true;
346   }
347   return false;
348 }
349
350 bool ARMBaseInstrInfo::
351 SubsumesPredicate(const SmallVectorImpl<MachineOperand> &Pred1,
352                   const SmallVectorImpl<MachineOperand> &Pred2) const {
353   if (Pred1.size() > 2 || Pred2.size() > 2)
354     return false;
355
356   ARMCC::CondCodes CC1 = (ARMCC::CondCodes)Pred1[0].getImm();
357   ARMCC::CondCodes CC2 = (ARMCC::CondCodes)Pred2[0].getImm();
358   if (CC1 == CC2)
359     return true;
360
361   switch (CC1) {
362   default:
363     return false;
364   case ARMCC::AL:
365     return true;
366   case ARMCC::HS:
367     return CC2 == ARMCC::HI;
368   case ARMCC::LS:
369     return CC2 == ARMCC::LO || CC2 == ARMCC::EQ;
370   case ARMCC::GE:
371     return CC2 == ARMCC::GT;
372   case ARMCC::LE:
373     return CC2 == ARMCC::LT;
374   }
375 }
376
377 bool ARMBaseInstrInfo::DefinesPredicate(MachineInstr *MI,
378                                     std::vector<MachineOperand> &Pred) const {
379   // FIXME: This confuses implicit_def with optional CPSR def.
380   const TargetInstrDesc &TID = MI->getDesc();
381   if (!TID.getImplicitDefs() && !TID.hasOptionalDef())
382     return false;
383
384   bool Found = false;
385   for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
386     const MachineOperand &MO = MI->getOperand(i);
387     if (MO.isReg() && MO.getReg() == ARM::CPSR) {
388       Pred.push_back(MO);
389       Found = true;
390     }
391   }
392
393   return Found;
394 }
395
396
397 /// FIXME: Works around a gcc miscompilation with -fstrict-aliasing
398 static unsigned getNumJTEntries(const std::vector<MachineJumpTableEntry> &JT,
399                                 unsigned JTI) DISABLE_INLINE;
400 static unsigned getNumJTEntries(const std::vector<MachineJumpTableEntry> &JT,
401                                 unsigned JTI) {
402   return JT[JTI].MBBs.size();
403 }
404
405 /// GetInstSize - Return the size of the specified MachineInstr.
406 ///
407 unsigned ARMBaseInstrInfo::GetInstSizeInBytes(const MachineInstr *MI) const {
408   const MachineBasicBlock &MBB = *MI->getParent();
409   const MachineFunction *MF = MBB.getParent();
410   const MCAsmInfo *MAI = MF->getTarget().getMCAsmInfo();
411
412   // Basic size info comes from the TSFlags field.
413   const TargetInstrDesc &TID = MI->getDesc();
414   unsigned TSFlags = TID.TSFlags;
415
416   unsigned Opc = MI->getOpcode();
417   switch ((TSFlags & ARMII::SizeMask) >> ARMII::SizeShift) {
418   default: {
419     // If this machine instr is an inline asm, measure it.
420     if (MI->getOpcode() == ARM::INLINEASM)
421       return getInlineAsmLength(MI->getOperand(0).getSymbolName(), *MAI);
422     if (MI->isLabel())
423       return 0;
424     switch (Opc) {
425     default:
426       llvm_unreachable("Unknown or unset size field for instr!");
427     case TargetInstrInfo::IMPLICIT_DEF:
428     case TargetInstrInfo::KILL:
429     case TargetInstrInfo::DBG_LABEL:
430     case TargetInstrInfo::EH_LABEL:
431       return 0;
432     }
433     break;
434   }
435   case ARMII::Size8Bytes: return 8;          // ARM instruction x 2.
436   case ARMII::Size4Bytes: return 4;          // ARM / Thumb2 instruction.
437   case ARMII::Size2Bytes: return 2;          // Thumb1 instruction.
438   case ARMII::SizeSpecial: {
439     switch (Opc) {
440     case ARM::CONSTPOOL_ENTRY:
441       // If this machine instr is a constant pool entry, its size is recorded as
442       // operand #2.
443       return MI->getOperand(2).getImm();
444     case ARM::Int_eh_sjlj_setjmp:
445       return 24;
446     case ARM::t2Int_eh_sjlj_setjmp:
447       return 20;
448     case ARM::BR_JTr:
449     case ARM::BR_JTm:
450     case ARM::BR_JTadd:
451     case ARM::tBR_JTr:
452     case ARM::t2BR_JT:
453     case ARM::t2TBB:
454     case ARM::t2TBH: {
455       // These are jumptable branches, i.e. a branch followed by an inlined
456       // jumptable. The size is 4 + 4 * number of entries. For TBB, each
457       // entry is one byte; TBH two byte each.
458       unsigned EntrySize = (Opc == ARM::t2TBB)
459         ? 1 : ((Opc == ARM::t2TBH) ? 2 : 4);
460       unsigned NumOps = TID.getNumOperands();
461       MachineOperand JTOP =
462         MI->getOperand(NumOps - (TID.isPredicable() ? 3 : 2));
463       unsigned JTI = JTOP.getIndex();
464       const MachineJumpTableInfo *MJTI = MF->getJumpTableInfo();
465       const std::vector<MachineJumpTableEntry> &JT = MJTI->getJumpTables();
466       assert(JTI < JT.size());
467       // Thumb instructions are 2 byte aligned, but JT entries are 4 byte
468       // 4 aligned. The assembler / linker may add 2 byte padding just before
469       // the JT entries.  The size does not include this padding; the
470       // constant islands pass does separate bookkeeping for it.
471       // FIXME: If we know the size of the function is less than (1 << 16) *2
472       // bytes, we can use 16-bit entries instead. Then there won't be an
473       // alignment issue.
474       unsigned InstSize = (Opc == ARM::tBR_JTr || Opc == ARM::t2BR_JT) ? 2 : 4;
475       unsigned NumEntries = getNumJTEntries(JT, JTI);
476       if (Opc == ARM::t2TBB && (NumEntries & 1))
477         // Make sure the instruction that follows TBB is 2-byte aligned.
478         // FIXME: Constant island pass should insert an "ALIGN" instruction
479         // instead.
480         ++NumEntries;
481       return NumEntries * EntrySize + InstSize;
482     }
483     default:
484       // Otherwise, pseudo-instruction sizes are zero.
485       return 0;
486     }
487   }
488   }
489   return 0; // Not reached
490 }
491
492 /// Return true if the instruction is a register to register move and
493 /// leave the source and dest operands in the passed parameters.
494 ///
495 bool
496 ARMBaseInstrInfo::isMoveInstr(const MachineInstr &MI,
497                               unsigned &SrcReg, unsigned &DstReg,
498                               unsigned& SrcSubIdx, unsigned& DstSubIdx) const {
499   SrcSubIdx = DstSubIdx = 0; // No sub-registers.
500
501   switch (MI.getOpcode()) {
502   default: break;
503   case ARM::FCPYS:
504   case ARM::FCPYD:
505   case ARM::VMOVD:
506   case  ARM::VMOVQ: {
507     SrcReg = MI.getOperand(1).getReg();
508     DstReg = MI.getOperand(0).getReg();
509     return true;
510   }
511   case ARM::MOVr:
512   case ARM::tMOVr:
513   case ARM::tMOVgpr2tgpr:
514   case ARM::tMOVtgpr2gpr:
515   case ARM::tMOVgpr2gpr:
516   case ARM::t2MOVr: {
517     assert(MI.getDesc().getNumOperands() >= 2 &&
518            MI.getOperand(0).isReg() &&
519            MI.getOperand(1).isReg() &&
520            "Invalid ARM MOV instruction");
521     SrcReg = MI.getOperand(1).getReg();
522     DstReg = MI.getOperand(0).getReg();
523     return true;
524   }
525   }
526
527   return false;
528 }
529
530 unsigned
531 ARMBaseInstrInfo::isLoadFromStackSlot(const MachineInstr *MI,
532                                       int &FrameIndex) const {
533   switch (MI->getOpcode()) {
534   default: break;
535   case ARM::LDR:
536   case ARM::t2LDRs:  // FIXME: don't use t2LDRs to access frame.
537     if (MI->getOperand(1).isFI() &&
538         MI->getOperand(2).isReg() &&
539         MI->getOperand(3).isImm() &&
540         MI->getOperand(2).getReg() == 0 &&
541         MI->getOperand(3).getImm() == 0) {
542       FrameIndex = MI->getOperand(1).getIndex();
543       return MI->getOperand(0).getReg();
544     }
545     break;
546   case ARM::t2LDRi12:
547   case ARM::tRestore:
548     if (MI->getOperand(1).isFI() &&
549         MI->getOperand(2).isImm() &&
550         MI->getOperand(2).getImm() == 0) {
551       FrameIndex = MI->getOperand(1).getIndex();
552       return MI->getOperand(0).getReg();
553     }
554     break;
555   case ARM::FLDD:
556   case ARM::FLDS:
557     if (MI->getOperand(1).isFI() &&
558         MI->getOperand(2).isImm() &&
559         MI->getOperand(2).getImm() == 0) {
560       FrameIndex = MI->getOperand(1).getIndex();
561       return MI->getOperand(0).getReg();
562     }
563     break;
564   }
565
566   return 0;
567 }
568
569 unsigned
570 ARMBaseInstrInfo::isStoreToStackSlot(const MachineInstr *MI,
571                                      int &FrameIndex) const {
572   switch (MI->getOpcode()) {
573   default: break;
574   case ARM::STR:
575   case ARM::t2STRs: // FIXME: don't use t2STRs to access frame.
576     if (MI->getOperand(1).isFI() &&
577         MI->getOperand(2).isReg() &&
578         MI->getOperand(3).isImm() &&
579         MI->getOperand(2).getReg() == 0 &&
580         MI->getOperand(3).getImm() == 0) {
581       FrameIndex = MI->getOperand(1).getIndex();
582       return MI->getOperand(0).getReg();
583     }
584     break;
585   case ARM::t2STRi12:
586   case ARM::tSpill:
587     if (MI->getOperand(1).isFI() &&
588         MI->getOperand(2).isImm() &&
589         MI->getOperand(2).getImm() == 0) {
590       FrameIndex = MI->getOperand(1).getIndex();
591       return MI->getOperand(0).getReg();
592     }
593     break;
594   case ARM::FSTD:
595   case ARM::FSTS:
596     if (MI->getOperand(1).isFI() &&
597         MI->getOperand(2).isImm() &&
598         MI->getOperand(2).getImm() == 0) {
599       FrameIndex = MI->getOperand(1).getIndex();
600       return MI->getOperand(0).getReg();
601     }
602     break;
603   }
604
605   return 0;
606 }
607
608 bool
609 ARMBaseInstrInfo::copyRegToReg(MachineBasicBlock &MBB,
610                                MachineBasicBlock::iterator I,
611                                unsigned DestReg, unsigned SrcReg,
612                                const TargetRegisterClass *DestRC,
613                                const TargetRegisterClass *SrcRC) const {
614   DebugLoc DL = DebugLoc::getUnknownLoc();
615   if (I != MBB.end()) DL = I->getDebugLoc();
616
617   if (DestRC != SrcRC) {
618     // Allow DPR / DPR_VFP2 / DPR_8 cross-class copies
619     // Allow QPR / QPR_VFP2 cross-class copies
620     if (DestRC == ARM::DPRRegisterClass) {
621       if (SrcRC == ARM::DPR_VFP2RegisterClass ||
622           SrcRC == ARM::DPR_8RegisterClass) {
623       } else
624         return false;
625     } else if (DestRC == ARM::DPR_VFP2RegisterClass) {
626       if (SrcRC == ARM::DPRRegisterClass ||
627           SrcRC == ARM::DPR_8RegisterClass) {
628       } else
629         return false;
630     } else if (DestRC == ARM::DPR_8RegisterClass) {
631       if (SrcRC == ARM::DPRRegisterClass ||
632           SrcRC == ARM::DPR_VFP2RegisterClass) {
633       } else
634         return false;
635     } else if ((DestRC == ARM::QPRRegisterClass &&
636                 SrcRC == ARM::QPR_VFP2RegisterClass) ||
637                (DestRC == ARM::QPR_VFP2RegisterClass &&
638                 SrcRC == ARM::QPRRegisterClass)) {
639     } else
640       return false;
641   }
642
643   if (DestRC == ARM::GPRRegisterClass) {
644     AddDefaultCC(AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::MOVr),
645                                         DestReg).addReg(SrcReg)));
646   } else if (DestRC == ARM::SPRRegisterClass) {
647     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::FCPYS), DestReg)
648                    .addReg(SrcReg));
649   } else if ((DestRC == ARM::DPRRegisterClass) ||
650              (DestRC == ARM::DPR_VFP2RegisterClass) ||
651              (DestRC == ARM::DPR_8RegisterClass)) {
652     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::FCPYD), DestReg)
653                    .addReg(SrcReg));
654   } else if (DestRC == ARM::QPRRegisterClass ||
655              DestRC == ARM::QPR_VFP2RegisterClass) {
656     BuildMI(MBB, I, DL, get(ARM::VMOVQ), DestReg).addReg(SrcReg);
657   } else {
658     return false;
659   }
660
661   return true;
662 }
663
664 void ARMBaseInstrInfo::
665 storeRegToStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator I,
666                     unsigned SrcReg, bool isKill, int FI,
667                     const TargetRegisterClass *RC) const {
668   DebugLoc DL = DebugLoc::getUnknownLoc();
669   if (I != MBB.end()) DL = I->getDebugLoc();
670   MachineFunction &MF = *MBB.getParent();
671   MachineFrameInfo &MFI = *MF.getFrameInfo();
672
673   MachineMemOperand *MMO =
674     MF.getMachineMemOperand(PseudoSourceValue::getFixedStack(FI),
675                             MachineMemOperand::MOStore, 0,
676                             MFI.getObjectSize(FI),
677                             MFI.getObjectAlignment(FI));
678
679   if (RC == ARM::GPRRegisterClass) {
680     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::STR))
681                    .addReg(SrcReg, getKillRegState(isKill))
682                    .addFrameIndex(FI).addReg(0).addImm(0).addMemOperand(MMO));
683   } else if (RC == ARM::DPRRegisterClass ||
684              RC == ARM::DPR_VFP2RegisterClass ||
685              RC == ARM::DPR_8RegisterClass) {
686     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::FSTD))
687                    .addReg(SrcReg, getKillRegState(isKill))
688                    .addFrameIndex(FI).addImm(0).addMemOperand(MMO));
689   } else if (RC == ARM::SPRRegisterClass) {
690     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::FSTS))
691                    .addReg(SrcReg, getKillRegState(isKill))
692                    .addFrameIndex(FI).addImm(0).addMemOperand(MMO));
693   } else {
694     assert((RC == ARM::QPRRegisterClass ||
695             RC == ARM::QPR_VFP2RegisterClass) && "Unknown regclass!");
696     // FIXME: Neon instructions should support predicates
697     BuildMI(MBB, I, DL, get(ARM::VSTRQ)).addReg(SrcReg, getKillRegState(isKill))
698       .addFrameIndex(FI).addImm(0).addMemOperand(MMO);
699   }
700 }
701
702 void ARMBaseInstrInfo::
703 loadRegFromStackSlot(MachineBasicBlock &MBB, MachineBasicBlock::iterator I,
704                      unsigned DestReg, int FI,
705                      const TargetRegisterClass *RC) const {
706   DebugLoc DL = DebugLoc::getUnknownLoc();
707   if (I != MBB.end()) DL = I->getDebugLoc();
708   MachineFunction &MF = *MBB.getParent();
709   MachineFrameInfo &MFI = *MF.getFrameInfo();
710
711   MachineMemOperand *MMO =
712     MF.getMachineMemOperand(PseudoSourceValue::getFixedStack(FI),
713                             MachineMemOperand::MOLoad, 0,
714                             MFI.getObjectSize(FI),
715                             MFI.getObjectAlignment(FI));
716
717   if (RC == ARM::GPRRegisterClass) {
718     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::LDR), DestReg)
719                    .addFrameIndex(FI).addReg(0).addImm(0).addMemOperand(MMO));
720   } else if (RC == ARM::DPRRegisterClass ||
721              RC == ARM::DPR_VFP2RegisterClass ||
722              RC == ARM::DPR_8RegisterClass) {
723     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::FLDD), DestReg)
724                    .addFrameIndex(FI).addImm(0).addMemOperand(MMO));
725   } else if (RC == ARM::SPRRegisterClass) {
726     AddDefaultPred(BuildMI(MBB, I, DL, get(ARM::FLDS), DestReg)
727                    .addFrameIndex(FI).addImm(0).addMemOperand(MMO));
728   } else {
729     assert((RC == ARM::QPRRegisterClass ||
730             RC == ARM::QPR_VFP2RegisterClass) && "Unknown regclass!");
731     // FIXME: Neon instructions should support predicates
732     BuildMI(MBB, I, DL, get(ARM::VLDRQ), DestReg).addFrameIndex(FI).addImm(0).addMemOperand(MMO);
733   }
734 }
735
736 MachineInstr *ARMBaseInstrInfo::
737 foldMemoryOperandImpl(MachineFunction &MF, MachineInstr *MI,
738                       const SmallVectorImpl<unsigned> &Ops, int FI) const {
739   if (Ops.size() != 1) return NULL;
740
741   unsigned OpNum = Ops[0];
742   unsigned Opc = MI->getOpcode();
743   MachineInstr *NewMI = NULL;
744   if (Opc == ARM::MOVr || Opc == ARM::t2MOVr) {
745     // If it is updating CPSR, then it cannot be folded.
746     if (MI->getOperand(4).getReg() == ARM::CPSR && !MI->getOperand(4).isDead())
747       return NULL;
748     unsigned Pred = MI->getOperand(2).getImm();
749     unsigned PredReg = MI->getOperand(3).getReg();
750     if (OpNum == 0) { // move -> store
751       unsigned SrcReg = MI->getOperand(1).getReg();
752       bool isKill = MI->getOperand(1).isKill();
753       bool isUndef = MI->getOperand(1).isUndef();
754       if (Opc == ARM::MOVr)
755         NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::STR))
756           .addReg(SrcReg, getKillRegState(isKill) | getUndefRegState(isUndef))
757           .addFrameIndex(FI).addReg(0).addImm(0).addImm(Pred).addReg(PredReg);
758       else // ARM::t2MOVr
759         NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::t2STRi12))
760           .addReg(SrcReg, getKillRegState(isKill) | getUndefRegState(isUndef))
761           .addFrameIndex(FI).addImm(0).addImm(Pred).addReg(PredReg);
762     } else {          // move -> load
763       unsigned DstReg = MI->getOperand(0).getReg();
764       bool isDead = MI->getOperand(0).isDead();
765       bool isUndef = MI->getOperand(0).isUndef();
766       if (Opc == ARM::MOVr)
767         NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::LDR))
768           .addReg(DstReg,
769                   RegState::Define |
770                   getDeadRegState(isDead) |
771                   getUndefRegState(isUndef))
772           .addFrameIndex(FI).addReg(0).addImm(0).addImm(Pred).addReg(PredReg);
773       else // ARM::t2MOVr
774         NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::t2LDRi12))
775           .addReg(DstReg,
776                   RegState::Define |
777                   getDeadRegState(isDead) |
778                   getUndefRegState(isUndef))
779           .addFrameIndex(FI).addImm(0).addImm(Pred).addReg(PredReg);
780     }
781   } else if (Opc == ARM::tMOVgpr2gpr ||
782              Opc == ARM::tMOVtgpr2gpr ||
783              Opc == ARM::tMOVgpr2tgpr) {
784     if (OpNum == 0) { // move -> store
785       unsigned SrcReg = MI->getOperand(1).getReg();
786       bool isKill = MI->getOperand(1).isKill();
787       bool isUndef = MI->getOperand(1).isUndef();
788       NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::t2STRi12))
789         .addReg(SrcReg, getKillRegState(isKill) | getUndefRegState(isUndef))
790         .addFrameIndex(FI).addImm(0).addImm(ARMCC::AL).addReg(0);
791     } else {          // move -> load
792       unsigned DstReg = MI->getOperand(0).getReg();
793       bool isDead = MI->getOperand(0).isDead();
794       bool isUndef = MI->getOperand(0).isUndef();
795       NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::t2LDRi12))
796         .addReg(DstReg,
797                 RegState::Define |
798                 getDeadRegState(isDead) |
799                 getUndefRegState(isUndef))
800         .addFrameIndex(FI).addImm(0).addImm(ARMCC::AL).addReg(0);
801     }
802   } else if (Opc == ARM::FCPYS) {
803     unsigned Pred = MI->getOperand(2).getImm();
804     unsigned PredReg = MI->getOperand(3).getReg();
805     if (OpNum == 0) { // move -> store
806       unsigned SrcReg = MI->getOperand(1).getReg();
807       bool isKill = MI->getOperand(1).isKill();
808       bool isUndef = MI->getOperand(1).isUndef();
809       NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::FSTS))
810         .addReg(SrcReg, getKillRegState(isKill) | getUndefRegState(isUndef))
811         .addFrameIndex(FI)
812         .addImm(0).addImm(Pred).addReg(PredReg);
813     } else {          // move -> load
814       unsigned DstReg = MI->getOperand(0).getReg();
815       bool isDead = MI->getOperand(0).isDead();
816       bool isUndef = MI->getOperand(0).isUndef();
817       NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::FLDS))
818         .addReg(DstReg,
819                 RegState::Define |
820                 getDeadRegState(isDead) |
821                 getUndefRegState(isUndef))
822         .addFrameIndex(FI).addImm(0).addImm(Pred).addReg(PredReg);
823     }
824   }
825   else if (Opc == ARM::FCPYD) {
826     unsigned Pred = MI->getOperand(2).getImm();
827     unsigned PredReg = MI->getOperand(3).getReg();
828     if (OpNum == 0) { // move -> store
829       unsigned SrcReg = MI->getOperand(1).getReg();
830       bool isKill = MI->getOperand(1).isKill();
831       bool isUndef = MI->getOperand(1).isUndef();
832       NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::FSTD))
833         .addReg(SrcReg, getKillRegState(isKill) | getUndefRegState(isUndef))
834         .addFrameIndex(FI).addImm(0).addImm(Pred).addReg(PredReg);
835     } else {          // move -> load
836       unsigned DstReg = MI->getOperand(0).getReg();
837       bool isDead = MI->getOperand(0).isDead();
838       bool isUndef = MI->getOperand(0).isUndef();
839       NewMI = BuildMI(MF, MI->getDebugLoc(), get(ARM::FLDD))
840         .addReg(DstReg,
841                 RegState::Define |
842                 getDeadRegState(isDead) |
843                 getUndefRegState(isUndef))
844         .addFrameIndex(FI).addImm(0).addImm(Pred).addReg(PredReg);
845     }
846   }
847
848   return NewMI;
849 }
850
851 MachineInstr*
852 ARMBaseInstrInfo::foldMemoryOperandImpl(MachineFunction &MF,
853                                         MachineInstr* MI,
854                                         const SmallVectorImpl<unsigned> &Ops,
855                                         MachineInstr* LoadMI) const {
856   // FIXME
857   return 0;
858 }
859
860 bool
861 ARMBaseInstrInfo::canFoldMemoryOperand(const MachineInstr *MI,
862                                    const SmallVectorImpl<unsigned> &Ops) const {
863   if (Ops.size() != 1) return false;
864
865   unsigned Opc = MI->getOpcode();
866   if (Opc == ARM::MOVr || Opc == ARM::t2MOVr) {
867     // If it is updating CPSR, then it cannot be folded.
868     return MI->getOperand(4).getReg() != ARM::CPSR ||
869       MI->getOperand(4).isDead();
870   } else if (Opc == ARM::tMOVgpr2gpr ||
871              Opc == ARM::tMOVtgpr2gpr ||
872              Opc == ARM::tMOVgpr2tgpr) {
873     return true;
874   } else if (Opc == ARM::FCPYS || Opc == ARM::FCPYD) {
875     return true;
876   } else if (Opc == ARM::VMOVD || Opc == ARM::VMOVQ) {
877     return false; // FIXME
878   }
879
880   return false;
881 }
882
883 /// getInstrPredicate - If instruction is predicated, returns its predicate
884 /// condition, otherwise returns AL. It also returns the condition code
885 /// register by reference.
886 ARMCC::CondCodes
887 llvm::getInstrPredicate(const MachineInstr *MI, unsigned &PredReg) {
888   int PIdx = MI->findFirstPredOperandIdx();
889   if (PIdx == -1) {
890     PredReg = 0;
891     return ARMCC::AL;
892   }
893
894   PredReg = MI->getOperand(PIdx+1).getReg();
895   return (ARMCC::CondCodes)MI->getOperand(PIdx).getImm();
896 }
897
898
899 int llvm::getMatchingCondBranchOpcode(int Opc) {
900   if (Opc == ARM::B)
901     return ARM::Bcc;
902   else if (Opc == ARM::tB)
903     return ARM::tBcc;
904   else if (Opc == ARM::t2B)
905       return ARM::t2Bcc;
906
907   llvm_unreachable("Unknown unconditional branch opcode!");
908   return 0;
909 }
910
911
912 void llvm::emitARMRegPlusImmediate(MachineBasicBlock &MBB,
913                                MachineBasicBlock::iterator &MBBI, DebugLoc dl,
914                                unsigned DestReg, unsigned BaseReg, int NumBytes,
915                                ARMCC::CondCodes Pred, unsigned PredReg,
916                                const ARMBaseInstrInfo &TII) {
917   bool isSub = NumBytes < 0;
918   if (isSub) NumBytes = -NumBytes;
919
920   while (NumBytes) {
921     unsigned RotAmt = ARM_AM::getSOImmValRotate(NumBytes);
922     unsigned ThisVal = NumBytes & ARM_AM::rotr32(0xFF, RotAmt);
923     assert(ThisVal && "Didn't extract field correctly");
924
925     // We will handle these bits from offset, clear them.
926     NumBytes &= ~ThisVal;
927
928     assert(ARM_AM::getSOImmVal(ThisVal) != -1 && "Bit extraction didn't work?");
929
930     // Build the new ADD / SUB.
931     unsigned Opc = isSub ? ARM::SUBri : ARM::ADDri;
932     BuildMI(MBB, MBBI, dl, TII.get(Opc), DestReg)
933       .addReg(BaseReg, RegState::Kill).addImm(ThisVal)
934       .addImm((unsigned)Pred).addReg(PredReg).addReg(0);
935     BaseReg = DestReg;
936   }
937 }
938
939 bool llvm::rewriteARMFrameIndex(MachineInstr &MI, unsigned FrameRegIdx,
940                                 unsigned FrameReg, int &Offset,
941                                 const ARMBaseInstrInfo &TII) {
942   unsigned Opcode = MI.getOpcode();
943   const TargetInstrDesc &Desc = MI.getDesc();
944   unsigned AddrMode = (Desc.TSFlags & ARMII::AddrModeMask);
945   bool isSub = false;
946
947   // Memory operands in inline assembly always use AddrMode2.
948   if (Opcode == ARM::INLINEASM)
949     AddrMode = ARMII::AddrMode2;
950
951   if (Opcode == ARM::ADDri) {
952     Offset += MI.getOperand(FrameRegIdx+1).getImm();
953     if (Offset == 0) {
954       // Turn it into a move.
955       MI.setDesc(TII.get(ARM::MOVr));
956       MI.getOperand(FrameRegIdx).ChangeToRegister(FrameReg, false);
957       MI.RemoveOperand(FrameRegIdx+1);
958       Offset = 0;
959       return true;
960     } else if (Offset < 0) {
961       Offset = -Offset;
962       isSub = true;
963       MI.setDesc(TII.get(ARM::SUBri));
964     }
965
966     // Common case: small offset, fits into instruction.
967     if (ARM_AM::getSOImmVal(Offset) != -1) {
968       // Replace the FrameIndex with sp / fp
969       MI.getOperand(FrameRegIdx).ChangeToRegister(FrameReg, false);
970       MI.getOperand(FrameRegIdx+1).ChangeToImmediate(Offset);
971       Offset = 0;
972       return true;
973     }
974
975     // Otherwise, pull as much of the immedidate into this ADDri/SUBri
976     // as possible.
977     unsigned RotAmt = ARM_AM::getSOImmValRotate(Offset);
978     unsigned ThisImmVal = Offset & ARM_AM::rotr32(0xFF, RotAmt);
979
980     // We will handle these bits from offset, clear them.
981     Offset &= ~ThisImmVal;
982
983     // Get the properly encoded SOImmVal field.
984     assert(ARM_AM::getSOImmVal(ThisImmVal) != -1 &&
985            "Bit extraction didn't work?");
986     MI.getOperand(FrameRegIdx+1).ChangeToImmediate(ThisImmVal);
987  } else {
988     unsigned ImmIdx = 0;
989     int InstrOffs = 0;
990     unsigned NumBits = 0;
991     unsigned Scale = 1;
992     switch (AddrMode) {
993     case ARMII::AddrMode2: {
994       ImmIdx = FrameRegIdx+2;
995       InstrOffs = ARM_AM::getAM2Offset(MI.getOperand(ImmIdx).getImm());
996       if (ARM_AM::getAM2Op(MI.getOperand(ImmIdx).getImm()) == ARM_AM::sub)
997         InstrOffs *= -1;
998       NumBits = 12;
999       break;
1000     }
1001     case ARMII::AddrMode3: {
1002       ImmIdx = FrameRegIdx+2;
1003       InstrOffs = ARM_AM::getAM3Offset(MI.getOperand(ImmIdx).getImm());
1004       if (ARM_AM::getAM3Op(MI.getOperand(ImmIdx).getImm()) == ARM_AM::sub)
1005         InstrOffs *= -1;
1006       NumBits = 8;
1007       break;
1008     }
1009     case ARMII::AddrMode4:
1010       // Can't fold any offset even if it's zero.
1011       return false;
1012     case ARMII::AddrMode5: {
1013       ImmIdx = FrameRegIdx+1;
1014       InstrOffs = ARM_AM::getAM5Offset(MI.getOperand(ImmIdx).getImm());
1015       if (ARM_AM::getAM5Op(MI.getOperand(ImmIdx).getImm()) == ARM_AM::sub)
1016         InstrOffs *= -1;
1017       NumBits = 8;
1018       Scale = 4;
1019       break;
1020     }
1021     default:
1022       llvm_unreachable("Unsupported addressing mode!");
1023       break;
1024     }
1025
1026     Offset += InstrOffs * Scale;
1027     assert((Offset & (Scale-1)) == 0 && "Can't encode this offset!");
1028     if (Offset < 0) {
1029       Offset = -Offset;
1030       isSub = true;
1031     }
1032
1033     // Attempt to fold address comp. if opcode has offset bits
1034     if (NumBits > 0) {
1035       // Common case: small offset, fits into instruction.
1036       MachineOperand &ImmOp = MI.getOperand(ImmIdx);
1037       int ImmedOffset = Offset / Scale;
1038       unsigned Mask = (1 << NumBits) - 1;
1039       if ((unsigned)Offset <= Mask * Scale) {
1040         // Replace the FrameIndex with sp
1041         MI.getOperand(FrameRegIdx).ChangeToRegister(FrameReg, false);
1042         if (isSub)
1043           ImmedOffset |= 1 << NumBits;
1044         ImmOp.ChangeToImmediate(ImmedOffset);
1045         Offset = 0;
1046         return true;
1047       }
1048
1049       // Otherwise, it didn't fit. Pull in what we can to simplify the immed.
1050       ImmedOffset = ImmedOffset & Mask;
1051       if (isSub)
1052         ImmedOffset |= 1 << NumBits;
1053       ImmOp.ChangeToImmediate(ImmedOffset);
1054       Offset &= ~(Mask*Scale);
1055     }
1056   }
1057
1058   Offset = (isSub) ? -Offset : Offset;
1059   return Offset == 0;
1060 }