Turns out AnalyzeBranch can modify the mbb being analyzed. This is a nasty
[oota-llvm.git] / lib / Target / XCore / XCoreInstrInfo.cpp
1 //===- XCoreInstrInfo.cpp - XCore 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 XCore implementation of the TargetInstrInfo class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "XCoreMachineFunctionInfo.h"
15 #include "XCoreInstrInfo.h"
16 #include "XCore.h"
17 #include "llvm/ADT/STLExtras.h"
18 #include "llvm/CodeGen/MachineInstrBuilder.h"
19 #include "llvm/CodeGen/MachineFrameInfo.h"
20 #include "llvm/CodeGen/MachineLocation.h"
21 #include "llvm/CodeGen/MachineModuleInfo.h"
22 #include "XCoreGenInstrInfo.inc"
23 #include "llvm/Support/Debug.h"
24
25 namespace llvm {
26 namespace XCore {
27
28   // XCore Condition Codes
29   enum CondCode {
30     COND_TRUE,
31     COND_FALSE,
32     COND_INVALID
33   };
34 }
35 }
36
37 using namespace llvm;
38
39 XCoreInstrInfo::XCoreInstrInfo(void)
40   : TargetInstrInfoImpl(XCoreInsts, array_lengthof(XCoreInsts)),
41     RI(*this) {
42 }
43
44 static bool isZeroImm(const MachineOperand &op) {
45   return op.isImm() && op.getImm() == 0;
46 }
47
48 /// Return true if the instruction is a register to register move and
49 /// leave the source and dest operands in the passed parameters.
50 ///
51 bool XCoreInstrInfo::isMoveInstr(const MachineInstr &MI,
52                                  unsigned &SrcReg, unsigned &DstReg,
53                                  unsigned &SrcSR, unsigned &DstSR) const {
54   SrcSR = DstSR = 0; // No sub-registers.
55
56   // We look for 4 kinds of patterns here:
57   // add dst, src, 0
58   // sub dst, src, 0
59   // or dst, src, src
60   // and dst, src, src
61   if ((MI.getOpcode() == XCore::ADD_2rus || MI.getOpcode() == XCore::SUB_2rus)
62       && isZeroImm(MI.getOperand(2))) {
63     DstReg = MI.getOperand(0).getReg();
64     SrcReg = MI.getOperand(1).getReg();
65     return true;
66   } else if ((MI.getOpcode() == XCore::OR_3r || MI.getOpcode() == XCore::AND_3r)
67       && MI.getOperand(1).getReg() == MI.getOperand(2).getReg()) {
68     DstReg = MI.getOperand(0).getReg();
69     SrcReg = MI.getOperand(1).getReg();
70     return true;
71   }
72   return false;
73 }
74
75 /// isLoadFromStackSlot - If the specified machine instruction is a direct
76 /// load from a stack slot, return the virtual or physical register number of
77 /// the destination along with the FrameIndex of the loaded stack slot.  If
78 /// not, return 0.  This predicate must return 0 if the instruction has
79 /// any side effects other than loading from the stack slot.
80 unsigned
81 XCoreInstrInfo::isLoadFromStackSlot(const MachineInstr *MI, int &FrameIndex) const{
82   int Opcode = MI->getOpcode();
83   if (Opcode == XCore::LDWFI) 
84   {
85     if ((MI->getOperand(1).isFI()) && // is a stack slot
86         (MI->getOperand(2).isImm()) &&  // the imm is zero
87         (isZeroImm(MI->getOperand(2)))) 
88     {
89       FrameIndex = MI->getOperand(1).getIndex();
90       return MI->getOperand(0).getReg();
91     }
92   }
93   return 0;
94 }
95   
96   /// isStoreToStackSlot - If the specified machine instruction is a direct
97   /// store to a stack slot, return the virtual or physical register number of
98   /// the source reg along with the FrameIndex of the loaded stack slot.  If
99   /// not, return 0.  This predicate must return 0 if the instruction has
100   /// any side effects other than storing to the stack slot.
101 unsigned
102 XCoreInstrInfo::isStoreToStackSlot(const MachineInstr *MI,
103                                    int &FrameIndex) const {
104   int Opcode = MI->getOpcode();
105   if (Opcode == XCore::STWFI)
106   {
107     if ((MI->getOperand(1).isFI()) && // is a stack slot
108         (MI->getOperand(2).isImm()) &&  // the imm is zero
109         (isZeroImm(MI->getOperand(2))))
110     {
111       FrameIndex = MI->getOperand(1).getIndex();
112       return MI->getOperand(0).getReg();
113     }
114   }
115   return 0;
116 }
117
118 /// isInvariantLoad - Return true if the specified instruction (which is marked
119 /// mayLoad) is loading from a location whose value is invariant across the
120 /// function.  For example, loading a value from the constant pool or from
121 /// from the argument area of a function if it does not change.  This should
122 /// only return true of *all* loads the instruction does are invariant (if it
123 /// does multiple loads).
124 bool
125 XCoreInstrInfo::isInvariantLoad(const MachineInstr *MI) const {
126   // Loads from constants pools and loads from invariant argument slots are
127   // invariant
128   int Opcode = MI->getOpcode();
129   if (Opcode == XCore::LDWCP_ru6 || Opcode == XCore::LDWCP_lru6) {
130     return MI->getOperand(1).isCPI();
131   }
132   int FrameIndex;
133   if (isLoadFromStackSlot(MI, FrameIndex)) {
134     const MachineFrameInfo &MFI =
135       *MI->getParent()->getParent()->getFrameInfo();
136     return MFI.isFixedObjectIndex(FrameIndex) &&
137            MFI.isImmutableObjectIndex(FrameIndex);
138   }
139   return false;
140 }
141
142 //===----------------------------------------------------------------------===//
143 // Branch Analysis
144 //===----------------------------------------------------------------------===//
145
146 static inline bool IsBRU(unsigned BrOpc) {
147   return BrOpc == XCore::BRFU_u6
148       || BrOpc == XCore::BRFU_lu6
149       || BrOpc == XCore::BRBU_u6
150       || BrOpc == XCore::BRBU_lu6;
151 }
152
153 static inline bool IsBRT(unsigned BrOpc) {
154   return BrOpc == XCore::BRFT_ru6
155       || BrOpc == XCore::BRFT_lru6
156       || BrOpc == XCore::BRBT_ru6
157       || BrOpc == XCore::BRBT_lru6;
158 }
159
160 static inline bool IsBRF(unsigned BrOpc) {
161   return BrOpc == XCore::BRFF_ru6
162       || BrOpc == XCore::BRFF_lru6
163       || BrOpc == XCore::BRBF_ru6
164       || BrOpc == XCore::BRBF_lru6;
165 }
166
167 static inline bool IsCondBranch(unsigned BrOpc) {
168   return IsBRF(BrOpc) || IsBRT(BrOpc);
169 }
170
171 /// GetCondFromBranchOpc - Return the XCore CC that matches 
172 /// the correspondent Branch instruction opcode.
173 static XCore::CondCode GetCondFromBranchOpc(unsigned BrOpc) 
174 {
175   if (IsBRT(BrOpc)) {
176     return XCore::COND_TRUE;
177   } else if (IsBRF(BrOpc)) {
178     return XCore::COND_FALSE;
179   } else {
180     return XCore::COND_INVALID;
181   }
182 }
183
184 /// GetCondBranchFromCond - Return the Branch instruction
185 /// opcode that matches the cc.
186 static inline unsigned GetCondBranchFromCond(XCore::CondCode CC) 
187 {
188   switch (CC) {
189   default: assert(0 && "Illegal condition code!");
190   case XCore::COND_TRUE   : return XCore::BRFT_lru6;
191   case XCore::COND_FALSE  : return XCore::BRFF_lru6;
192   }
193 }
194
195 /// GetOppositeBranchCondition - Return the inverse of the specified 
196 /// condition, e.g. turning COND_E to COND_NE.
197 static inline XCore::CondCode GetOppositeBranchCondition(XCore::CondCode CC)
198 {
199   switch (CC) {
200   default: assert(0 && "Illegal condition code!");
201   case XCore::COND_TRUE   : return XCore::COND_FALSE;
202   case XCore::COND_FALSE  : return XCore::COND_TRUE;
203   }
204 }
205
206 /// AnalyzeBranch - Analyze the branching code at the end of MBB, returning
207 /// true if it cannot be understood (e.g. it's a switch dispatch or isn't
208 /// implemented for a target).  Upon success, this returns false and returns
209 /// with the following information in various cases:
210 ///
211 /// 1. If this block ends with no branches (it just falls through to its succ)
212 ///    just return false, leaving TBB/FBB null.
213 /// 2. If this block ends with only an unconditional branch, it sets TBB to be
214 ///    the destination block.
215 /// 3. If this block ends with an conditional branch and it falls through to
216 ///    an successor block, it sets TBB to be the branch destination block and a
217 ///    list of operands that evaluate the condition. These
218 ///    operands can be passed to other TargetInstrInfo methods to create new
219 ///    branches.
220 /// 4. If this block ends with an conditional branch and an unconditional
221 ///    block, it returns the 'true' destination in TBB, the 'false' destination
222 ///    in FBB, and a list of operands that evaluate the condition. These
223 ///    operands can be passed to other TargetInstrInfo methods to create new
224 ///    branches.
225 ///
226 /// Note that RemoveBranch and InsertBranch must be implemented to support
227 /// cases where this method returns success.
228 ///
229 bool
230 XCoreInstrInfo::AnalyzeBranch(MachineBasicBlock &MBB, MachineBasicBlock *&TBB,
231                               MachineBasicBlock *&FBB,
232                               SmallVectorImpl<MachineOperand> &Cond,
233                               bool AllowModify) const {
234   // If the block has no terminators, it just falls into the block after it.
235   MachineBasicBlock::iterator I = MBB.end();
236   if (I == MBB.begin() || !isUnpredicatedTerminator(--I))
237     return false;
238
239   // Get the last instruction in the block.
240   MachineInstr *LastInst = I;
241   
242   // If there is only one terminator instruction, process it.
243   if (I == MBB.begin() || !isUnpredicatedTerminator(--I)) {
244     if (IsBRU(LastInst->getOpcode())) {
245       TBB = LastInst->getOperand(0).getMBB();
246       return false;
247     }
248     
249     XCore::CondCode BranchCode = GetCondFromBranchOpc(LastInst->getOpcode());
250     if (BranchCode == XCore::COND_INVALID)
251       return true;  // Can't handle indirect branch.
252     
253     // Conditional branch
254     // Block ends with fall-through condbranch.
255
256     TBB = LastInst->getOperand(1).getMBB();
257     Cond.push_back(MachineOperand::CreateImm(BranchCode));
258     Cond.push_back(LastInst->getOperand(0));
259     return false;
260   }
261   
262   // Get the instruction before it if it's a terminator.
263   MachineInstr *SecondLastInst = I;
264
265   // If there are three terminators, we don't know what sort of block this is.
266   if (SecondLastInst && I != MBB.begin() &&
267       isUnpredicatedTerminator(--I))
268     return true;
269   
270   unsigned SecondLastOpc    = SecondLastInst->getOpcode();
271   XCore::CondCode BranchCode = GetCondFromBranchOpc(SecondLastOpc);
272   
273   // If the block ends with conditional branch followed by unconditional,
274   // handle it.
275   if (BranchCode != XCore::COND_INVALID
276     && IsBRU(LastInst->getOpcode())) {
277
278     TBB = SecondLastInst->getOperand(1).getMBB();
279     Cond.push_back(MachineOperand::CreateImm(BranchCode));
280     Cond.push_back(SecondLastInst->getOperand(0));
281
282     FBB = LastInst->getOperand(0).getMBB();
283     return false;
284   }
285   
286   // If the block ends with two unconditional branches, handle it.  The second
287   // one is not executed, so remove it.
288   if (IsBRU(SecondLastInst->getOpcode()) && 
289       IsBRU(LastInst->getOpcode())) {
290     TBB = SecondLastInst->getOperand(0).getMBB();
291     I = LastInst;
292     if (AllowModify)
293       I->eraseFromParent();
294     return false;
295   }
296
297   // Otherwise, can't handle this.
298   return true;
299 }
300
301 unsigned
302 XCoreInstrInfo::InsertBranch(MachineBasicBlock &MBB,MachineBasicBlock *TBB,
303                              MachineBasicBlock *FBB,
304                              const SmallVectorImpl<MachineOperand> &Cond)const{
305   // Shouldn't be a fall through.
306   assert(TBB && "InsertBranch must not be told to insert a fallthrough");
307   assert((Cond.size() == 2 || Cond.size() == 0) &&
308          "Unexpected number of components!");
309   
310   if (FBB == 0) { // One way branch.
311     if (Cond.empty()) {
312       // Unconditional branch
313       BuildMI(&MBB, get(XCore::BRFU_lu6)).addMBB(TBB);
314     } else {
315       // Conditional branch.
316       unsigned Opc = GetCondBranchFromCond((XCore::CondCode)Cond[0].getImm());
317       BuildMI(&MBB, get(Opc)).addReg(Cond[1].getReg())
318                              .addMBB(TBB);
319     }
320     return 1;
321   }
322   
323   // Two-way Conditional branch.
324   assert(Cond.size() == 2 && "Unexpected number of components!");
325   unsigned Opc = GetCondBranchFromCond((XCore::CondCode)Cond[0].getImm());
326   BuildMI(&MBB, get(Opc)).addReg(Cond[1].getReg())
327                          .addMBB(TBB);
328   BuildMI(&MBB, get(XCore::BRFU_lu6)).addMBB(FBB);
329   return 2;
330 }
331
332 unsigned
333 XCoreInstrInfo::RemoveBranch(MachineBasicBlock &MBB) const {
334   MachineBasicBlock::iterator I = MBB.end();
335   if (I == MBB.begin()) return 0;
336   --I;
337   if (!IsBRU(I->getOpcode()) && !IsCondBranch(I->getOpcode()))
338     return 0;
339   
340   // Remove the branch.
341   I->eraseFromParent();
342   
343   I = MBB.end();
344
345   if (I == MBB.begin()) return 1;
346   --I;
347   if (!IsCondBranch(I->getOpcode()))
348     return 1;
349   
350   // Remove the branch.
351   I->eraseFromParent();
352   return 2;
353 }
354
355 bool XCoreInstrInfo::copyRegToReg(MachineBasicBlock &MBB,
356                                      MachineBasicBlock::iterator I,
357                                      unsigned DestReg, unsigned SrcReg,
358                                      const TargetRegisterClass *DestRC,
359                                      const TargetRegisterClass *SrcRC) const {
360   if (DestRC == SrcRC) {
361     if (DestRC == XCore::GRRegsRegisterClass) {
362       BuildMI(MBB, I, get(XCore::ADD_2rus), DestReg).addReg(SrcReg).addImm(0);
363       return true;
364     } else {
365       return false;
366     }
367   }
368   
369   if (SrcRC == XCore::RRegsRegisterClass && SrcReg == XCore::SP &&
370     DestRC == XCore::GRRegsRegisterClass) {
371     BuildMI(MBB, I, get(XCore::LDAWSP_ru6), DestReg).addImm(0);
372     return true;
373   }
374   if (DestRC == XCore::RRegsRegisterClass && DestReg == XCore::SP &&
375     SrcRC == XCore::GRRegsRegisterClass) {
376     BuildMI(MBB, I, get(XCore::SETSP_1r)).addReg(SrcReg);
377     return true;
378   }
379   return false;
380 }
381
382 void XCoreInstrInfo::storeRegToStackSlot(MachineBasicBlock &MBB,
383                                   MachineBasicBlock::iterator I,
384                                   unsigned SrcReg, bool isKill, int FrameIndex,
385                                   const TargetRegisterClass *RC) const
386 {
387   BuildMI(MBB, I, get(XCore::STWFI)).addReg(SrcReg, false, false, isKill)
388                                     .addFrameIndex(FrameIndex).addImm(0);
389 }
390
391 void XCoreInstrInfo::storeRegToAddr(MachineFunction &MF, unsigned SrcReg,
392                             bool isKill, SmallVectorImpl<MachineOperand> &Addr,
393                             const TargetRegisterClass *RC,
394                             SmallVectorImpl<MachineInstr*> &NewMIs) const
395 {
396   assert(0 && "unimplemented\n");
397 }
398
399 void XCoreInstrInfo::loadRegFromStackSlot(MachineBasicBlock &MBB,
400                                   MachineBasicBlock::iterator I,
401                                   unsigned DestReg, int FrameIndex,
402                                   const TargetRegisterClass *RC) const
403 {
404   BuildMI(MBB, I, get(XCore::LDWFI), DestReg).addFrameIndex(FrameIndex)
405                                              .addImm(0);
406 }
407
408 void XCoreInstrInfo::loadRegFromAddr(MachineFunction &MF, unsigned DestReg,
409                               SmallVectorImpl<MachineOperand> &Addr,
410                               const TargetRegisterClass *RC,
411                               SmallVectorImpl<MachineInstr*> &NewMIs) const
412 {
413   assert(0 && "unimplemented\n");
414 }
415
416 bool XCoreInstrInfo::spillCalleeSavedRegisters(MachineBasicBlock &MBB,
417                                 MachineBasicBlock::iterator MI,
418                         const std::vector<CalleeSavedInfo> &CSI) const
419 {
420   if (CSI.empty()) {
421     return true;
422   }
423   MachineFunction *MF = MBB.getParent();
424   const MachineFrameInfo *MFI = MF->getFrameInfo();
425   MachineModuleInfo *MMI = MFI->getMachineModuleInfo();
426   XCoreFunctionInfo *XFI = MF->getInfo<XCoreFunctionInfo>();
427   
428   bool emitFrameMoves = XCoreRegisterInfo::needsFrameMoves(*MF);
429   
430   for (std::vector<CalleeSavedInfo>::const_iterator it = CSI.begin();
431                                                     it != CSI.end(); ++it) {
432     // Add the callee-saved register as live-in. It's killed at the spill.
433     MBB.addLiveIn(it->getReg());
434
435     storeRegToStackSlot(MBB, MI, it->getReg(), true,
436                                    it->getFrameIdx(), it->getRegClass());
437     if (emitFrameMoves) {
438       unsigned SaveLabelId = MMI->NextLabelID();
439       BuildMI(MBB, MI, get(XCore::DBG_LABEL)).addImm(SaveLabelId);
440       XFI->getSpillLabels().push_back(
441           std::pair<unsigned, CalleeSavedInfo>(SaveLabelId, *it));
442     }
443   }
444   return true;
445 }
446
447 bool XCoreInstrInfo::restoreCalleeSavedRegisters(MachineBasicBlock &MBB,
448                                          MachineBasicBlock::iterator MI,
449                                const std::vector<CalleeSavedInfo> &CSI) const
450 {
451   bool AtStart = MI == MBB.begin();
452   MachineBasicBlock::iterator BeforeI = MI;
453   if (!AtStart)
454     --BeforeI;
455   for (std::vector<CalleeSavedInfo>::const_iterator it = CSI.begin();
456                                                     it != CSI.end(); ++it) {
457     
458     loadRegFromStackSlot(MBB, MI, it->getReg(),
459                                   it->getFrameIdx(),
460                                   it->getRegClass());
461     assert(MI != MBB.begin() &&
462            "loadRegFromStackSlot didn't insert any code!");
463     // Insert in reverse order.  loadRegFromStackSlot can insert multiple
464     // instructions.
465     if (AtStart)
466       MI = MBB.begin();
467     else {
468       MI = BeforeI;
469       ++MI;
470     }
471   }
472   return true;
473 }
474
475 /// BlockHasNoFallThrough - Analyse if MachineBasicBlock does not
476 /// fall-through into its successor block.
477 bool XCoreInstrInfo::
478 BlockHasNoFallThrough(const MachineBasicBlock &MBB) const 
479 {
480   if (MBB.empty()) return false;
481   
482   switch (MBB.back().getOpcode()) {
483   case XCore::RETSP_u6:     // Return.
484   case XCore::RETSP_lu6:
485   case XCore::BAU_1r:       // Indirect branch.
486   case XCore::BRFU_u6:      // Uncond branch.
487   case XCore::BRFU_lu6:
488   case XCore::BRBU_u6:
489   case XCore::BRBU_lu6:
490     return true;
491   default: return false;
492   }
493 }
494
495 /// ReverseBranchCondition - Return the inverse opcode of the 
496 /// specified Branch instruction.
497 bool XCoreInstrInfo::
498 ReverseBranchCondition(SmallVectorImpl<MachineOperand> &Cond) const 
499 {
500   assert((Cond.size() == 2) && 
501           "Invalid XCore branch condition!");
502   Cond[0].setImm(GetOppositeBranchCondition((XCore::CondCode)Cond[0].getImm()));
503   return false;
504 }