1f22ef01cSRoman Divacky //===-- llvm/CodeGen/MachineBasicBlock.cpp ----------------------*- C++ -*-===// 2f22ef01cSRoman Divacky // 3f22ef01cSRoman Divacky // The LLVM Compiler Infrastructure 4f22ef01cSRoman Divacky // 5f22ef01cSRoman Divacky // This file is distributed under the University of Illinois Open Source 6f22ef01cSRoman Divacky // License. See LICENSE.TXT for details. 7f22ef01cSRoman Divacky // 8f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 9f22ef01cSRoman Divacky // 10f22ef01cSRoman Divacky // Collect the sequence of machine instructions for a basic block. 11f22ef01cSRoman Divacky // 12f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 13f22ef01cSRoman Divacky 14f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineBasicBlock.h" 15f22ef01cSRoman Divacky #include "llvm/BasicBlock.h" 16ffd1746dSEd Schouten #include "llvm/CodeGen/LiveVariables.h" 17ffd1746dSEd Schouten #include "llvm/CodeGen/MachineDominators.h" 18f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineFunction.h" 19ffd1746dSEd Schouten #include "llvm/CodeGen/MachineLoopInfo.h" 202754fe60SDimitry Andric #include "llvm/CodeGen/SlotIndexes.h" 21f22ef01cSRoman Divacky #include "llvm/MC/MCAsmInfo.h" 22f22ef01cSRoman Divacky #include "llvm/MC/MCContext.h" 23f22ef01cSRoman Divacky #include "llvm/Target/TargetRegisterInfo.h" 24f22ef01cSRoman Divacky #include "llvm/Target/TargetData.h" 25f22ef01cSRoman Divacky #include "llvm/Target/TargetInstrInfo.h" 26f22ef01cSRoman Divacky #include "llvm/Target/TargetMachine.h" 27f22ef01cSRoman Divacky #include "llvm/Assembly/Writer.h" 28f22ef01cSRoman Divacky #include "llvm/ADT/SmallString.h" 29f22ef01cSRoman Divacky #include "llvm/ADT/SmallPtrSet.h" 30f22ef01cSRoman Divacky #include "llvm/Support/Debug.h" 31f22ef01cSRoman Divacky #include "llvm/Support/LeakDetector.h" 32f22ef01cSRoman Divacky #include "llvm/Support/raw_ostream.h" 33f22ef01cSRoman Divacky #include <algorithm> 34f22ef01cSRoman Divacky using namespace llvm; 35f22ef01cSRoman Divacky 36f22ef01cSRoman Divacky MachineBasicBlock::MachineBasicBlock(MachineFunction &mf, const BasicBlock *bb) 37f22ef01cSRoman Divacky : BB(bb), Number(-1), xParent(&mf), Alignment(0), IsLandingPad(false), 38f22ef01cSRoman Divacky AddressTaken(false) { 39f22ef01cSRoman Divacky Insts.Parent = this; 40f22ef01cSRoman Divacky } 41f22ef01cSRoman Divacky 42f22ef01cSRoman Divacky MachineBasicBlock::~MachineBasicBlock() { 43f22ef01cSRoman Divacky LeakDetector::removeGarbageObject(this); 44f22ef01cSRoman Divacky } 45f22ef01cSRoman Divacky 46f22ef01cSRoman Divacky /// getSymbol - Return the MCSymbol for this basic block. 47f22ef01cSRoman Divacky /// 48f22ef01cSRoman Divacky MCSymbol *MachineBasicBlock::getSymbol() const { 49f22ef01cSRoman Divacky const MachineFunction *MF = getParent(); 50f22ef01cSRoman Divacky MCContext &Ctx = MF->getContext(); 51f22ef01cSRoman Divacky const char *Prefix = Ctx.getAsmInfo().getPrivateGlobalPrefix(); 52f22ef01cSRoman Divacky return Ctx.GetOrCreateSymbol(Twine(Prefix) + "BB" + 53f22ef01cSRoman Divacky Twine(MF->getFunctionNumber()) + "_" + 54f22ef01cSRoman Divacky Twine(getNumber())); 55f22ef01cSRoman Divacky } 56f22ef01cSRoman Divacky 57f22ef01cSRoman Divacky 58f22ef01cSRoman Divacky raw_ostream &llvm::operator<<(raw_ostream &OS, const MachineBasicBlock &MBB) { 59f22ef01cSRoman Divacky MBB.print(OS); 60f22ef01cSRoman Divacky return OS; 61f22ef01cSRoman Divacky } 62f22ef01cSRoman Divacky 63f22ef01cSRoman Divacky /// addNodeToList (MBB) - When an MBB is added to an MF, we need to update the 64f22ef01cSRoman Divacky /// parent pointer of the MBB, the MBB numbering, and any instructions in the 65f22ef01cSRoman Divacky /// MBB to be on the right operand list for registers. 66f22ef01cSRoman Divacky /// 67f22ef01cSRoman Divacky /// MBBs start out as #-1. When a MBB is added to a MachineFunction, it 68f22ef01cSRoman Divacky /// gets the next available unique MBB number. If it is removed from a 69f22ef01cSRoman Divacky /// MachineFunction, it goes back to being #-1. 70f22ef01cSRoman Divacky void ilist_traits<MachineBasicBlock>::addNodeToList(MachineBasicBlock *N) { 71f22ef01cSRoman Divacky MachineFunction &MF = *N->getParent(); 72f22ef01cSRoman Divacky N->Number = MF.addToMBBNumbering(N); 73f22ef01cSRoman Divacky 74f22ef01cSRoman Divacky // Make sure the instructions have their operands in the reginfo lists. 75f22ef01cSRoman Divacky MachineRegisterInfo &RegInfo = MF.getRegInfo(); 76dff0c46cSDimitry Andric for (MachineBasicBlock::instr_iterator 77dff0c46cSDimitry Andric I = N->instr_begin(), E = N->instr_end(); I != E; ++I) 78f22ef01cSRoman Divacky I->AddRegOperandsToUseLists(RegInfo); 79f22ef01cSRoman Divacky 80f22ef01cSRoman Divacky LeakDetector::removeGarbageObject(N); 81f22ef01cSRoman Divacky } 82f22ef01cSRoman Divacky 83f22ef01cSRoman Divacky void ilist_traits<MachineBasicBlock>::removeNodeFromList(MachineBasicBlock *N) { 84f22ef01cSRoman Divacky N->getParent()->removeFromMBBNumbering(N->Number); 85f22ef01cSRoman Divacky N->Number = -1; 86f22ef01cSRoman Divacky LeakDetector::addGarbageObject(N); 87f22ef01cSRoman Divacky } 88f22ef01cSRoman Divacky 89f22ef01cSRoman Divacky 90f22ef01cSRoman Divacky /// addNodeToList (MI) - When we add an instruction to a basic block 91f22ef01cSRoman Divacky /// list, we update its parent pointer and add its operands from reg use/def 92f22ef01cSRoman Divacky /// lists if appropriate. 93f22ef01cSRoman Divacky void ilist_traits<MachineInstr>::addNodeToList(MachineInstr *N) { 94f22ef01cSRoman Divacky assert(N->getParent() == 0 && "machine instruction already in a basic block"); 95f22ef01cSRoman Divacky N->setParent(Parent); 96f22ef01cSRoman Divacky 97f22ef01cSRoman Divacky // Add the instruction's register operands to their corresponding 98f22ef01cSRoman Divacky // use/def lists. 99f22ef01cSRoman Divacky MachineFunction *MF = Parent->getParent(); 100f22ef01cSRoman Divacky N->AddRegOperandsToUseLists(MF->getRegInfo()); 101f22ef01cSRoman Divacky 102f22ef01cSRoman Divacky LeakDetector::removeGarbageObject(N); 103f22ef01cSRoman Divacky } 104f22ef01cSRoman Divacky 105f22ef01cSRoman Divacky /// removeNodeFromList (MI) - When we remove an instruction from a basic block 106f22ef01cSRoman Divacky /// list, we update its parent pointer and remove its operands from reg use/def 107f22ef01cSRoman Divacky /// lists if appropriate. 108f22ef01cSRoman Divacky void ilist_traits<MachineInstr>::removeNodeFromList(MachineInstr *N) { 109f22ef01cSRoman Divacky assert(N->getParent() != 0 && "machine instruction not in a basic block"); 110f22ef01cSRoman Divacky 111f22ef01cSRoman Divacky // Remove from the use/def lists. 1127ae0e2c9SDimitry Andric if (MachineFunction *MF = N->getParent()->getParent()) 1137ae0e2c9SDimitry Andric N->RemoveRegOperandsFromUseLists(MF->getRegInfo()); 114f22ef01cSRoman Divacky 115f22ef01cSRoman Divacky N->setParent(0); 116f22ef01cSRoman Divacky 117f22ef01cSRoman Divacky LeakDetector::addGarbageObject(N); 118f22ef01cSRoman Divacky } 119f22ef01cSRoman Divacky 120f22ef01cSRoman Divacky /// transferNodesFromList (MI) - When moving a range of instructions from one 121f22ef01cSRoman Divacky /// MBB list to another, we need to update the parent pointers and the use/def 122f22ef01cSRoman Divacky /// lists. 123f22ef01cSRoman Divacky void ilist_traits<MachineInstr>:: 124f22ef01cSRoman Divacky transferNodesFromList(ilist_traits<MachineInstr> &fromList, 125dff0c46cSDimitry Andric ilist_iterator<MachineInstr> first, 126dff0c46cSDimitry Andric ilist_iterator<MachineInstr> last) { 127f22ef01cSRoman Divacky assert(Parent->getParent() == fromList.Parent->getParent() && 128f22ef01cSRoman Divacky "MachineInstr parent mismatch!"); 129f22ef01cSRoman Divacky 130f22ef01cSRoman Divacky // Splice within the same MBB -> no change. 131f22ef01cSRoman Divacky if (Parent == fromList.Parent) return; 132f22ef01cSRoman Divacky 133f22ef01cSRoman Divacky // If splicing between two blocks within the same function, just update the 134f22ef01cSRoman Divacky // parent pointers. 135f22ef01cSRoman Divacky for (; first != last; ++first) 136f22ef01cSRoman Divacky first->setParent(Parent); 137f22ef01cSRoman Divacky } 138f22ef01cSRoman Divacky 139f22ef01cSRoman Divacky void ilist_traits<MachineInstr>::deleteNode(MachineInstr* MI) { 140f22ef01cSRoman Divacky assert(!MI->getParent() && "MI is still in a block!"); 141f22ef01cSRoman Divacky Parent->getParent()->DeleteMachineInstr(MI); 142f22ef01cSRoman Divacky } 143f22ef01cSRoman Divacky 144ffd1746dSEd Schouten MachineBasicBlock::iterator MachineBasicBlock::getFirstNonPHI() { 145dff0c46cSDimitry Andric instr_iterator I = instr_begin(), E = instr_end(); 146dff0c46cSDimitry Andric while (I != E && I->isPHI()) 147ffd1746dSEd Schouten ++I; 148dff0c46cSDimitry Andric assert(!I->isInsideBundle() && "First non-phi MI cannot be inside a bundle!"); 149ffd1746dSEd Schouten return I; 150ffd1746dSEd Schouten } 151ffd1746dSEd Schouten 1522754fe60SDimitry Andric MachineBasicBlock::iterator 1532754fe60SDimitry Andric MachineBasicBlock::SkipPHIsAndLabels(MachineBasicBlock::iterator I) { 154dff0c46cSDimitry Andric iterator E = end(); 155dff0c46cSDimitry Andric while (I != E && (I->isPHI() || I->isLabel() || I->isDebugValue())) 1562754fe60SDimitry Andric ++I; 157dff0c46cSDimitry Andric // FIXME: This needs to change if we wish to bundle labels / dbg_values 158dff0c46cSDimitry Andric // inside the bundle. 159dff0c46cSDimitry Andric assert(!I->isInsideBundle() && 160dff0c46cSDimitry Andric "First non-phi / non-label instruction is inside a bundle!"); 1612754fe60SDimitry Andric return I; 1622754fe60SDimitry Andric } 1632754fe60SDimitry Andric 164f22ef01cSRoman Divacky MachineBasicBlock::iterator MachineBasicBlock::getFirstTerminator() { 165dff0c46cSDimitry Andric iterator B = begin(), E = end(), I = E; 166dff0c46cSDimitry Andric while (I != B && ((--I)->isTerminator() || I->isDebugValue())) 167f22ef01cSRoman Divacky ; /*noop */ 168dff0c46cSDimitry Andric while (I != E && !I->isTerminator()) 169dff0c46cSDimitry Andric ++I; 170dff0c46cSDimitry Andric return I; 171dff0c46cSDimitry Andric } 172dff0c46cSDimitry Andric 173dff0c46cSDimitry Andric MachineBasicBlock::const_iterator 174dff0c46cSDimitry Andric MachineBasicBlock::getFirstTerminator() const { 175dff0c46cSDimitry Andric const_iterator B = begin(), E = end(), I = E; 176dff0c46cSDimitry Andric while (I != B && ((--I)->isTerminator() || I->isDebugValue())) 177dff0c46cSDimitry Andric ; /*noop */ 178dff0c46cSDimitry Andric while (I != E && !I->isTerminator()) 179dff0c46cSDimitry Andric ++I; 180dff0c46cSDimitry Andric return I; 181dff0c46cSDimitry Andric } 182dff0c46cSDimitry Andric 183dff0c46cSDimitry Andric MachineBasicBlock::instr_iterator MachineBasicBlock::getFirstInstrTerminator() { 184dff0c46cSDimitry Andric instr_iterator B = instr_begin(), E = instr_end(), I = E; 185dff0c46cSDimitry Andric while (I != B && ((--I)->isTerminator() || I->isDebugValue())) 186dff0c46cSDimitry Andric ; /*noop */ 187dff0c46cSDimitry Andric while (I != E && !I->isTerminator()) 1882754fe60SDimitry Andric ++I; 189f22ef01cSRoman Divacky return I; 190f22ef01cSRoman Divacky } 191f22ef01cSRoman Divacky 1922754fe60SDimitry Andric MachineBasicBlock::iterator MachineBasicBlock::getLastNonDebugInstr() { 193dff0c46cSDimitry Andric // Skip over end-of-block dbg_value instructions. 194dff0c46cSDimitry Andric instr_iterator B = instr_begin(), I = instr_end(); 1952754fe60SDimitry Andric while (I != B) { 1962754fe60SDimitry Andric --I; 197dff0c46cSDimitry Andric // Return instruction that starts a bundle. 198dff0c46cSDimitry Andric if (I->isDebugValue() || I->isInsideBundle()) 199dff0c46cSDimitry Andric continue; 200dff0c46cSDimitry Andric return I; 201dff0c46cSDimitry Andric } 202dff0c46cSDimitry Andric // The block is all debug values. 203dff0c46cSDimitry Andric return end(); 204dff0c46cSDimitry Andric } 205dff0c46cSDimitry Andric 206dff0c46cSDimitry Andric MachineBasicBlock::const_iterator 207dff0c46cSDimitry Andric MachineBasicBlock::getLastNonDebugInstr() const { 208dff0c46cSDimitry Andric // Skip over end-of-block dbg_value instructions. 209dff0c46cSDimitry Andric const_instr_iterator B = instr_begin(), I = instr_end(); 210dff0c46cSDimitry Andric while (I != B) { 211dff0c46cSDimitry Andric --I; 212dff0c46cSDimitry Andric // Return instruction that starts a bundle. 213dff0c46cSDimitry Andric if (I->isDebugValue() || I->isInsideBundle()) 2142754fe60SDimitry Andric continue; 2152754fe60SDimitry Andric return I; 2162754fe60SDimitry Andric } 2172754fe60SDimitry Andric // The block is all debug values. 2182754fe60SDimitry Andric return end(); 2192754fe60SDimitry Andric } 2202754fe60SDimitry Andric 2212754fe60SDimitry Andric const MachineBasicBlock *MachineBasicBlock::getLandingPadSuccessor() const { 2222754fe60SDimitry Andric // A block with a landing pad successor only has one other successor. 2232754fe60SDimitry Andric if (succ_size() > 2) 2242754fe60SDimitry Andric return 0; 2252754fe60SDimitry Andric for (const_succ_iterator I = succ_begin(), E = succ_end(); I != E; ++I) 2262754fe60SDimitry Andric if ((*I)->isLandingPad()) 2272754fe60SDimitry Andric return *I; 2282754fe60SDimitry Andric return 0; 2292754fe60SDimitry Andric } 2302754fe60SDimitry Andric 231f22ef01cSRoman Divacky void MachineBasicBlock::dump() const { 232f22ef01cSRoman Divacky print(dbgs()); 233f22ef01cSRoman Divacky } 234f22ef01cSRoman Divacky 235f22ef01cSRoman Divacky StringRef MachineBasicBlock::getName() const { 236f22ef01cSRoman Divacky if (const BasicBlock *LBB = getBasicBlock()) 237f22ef01cSRoman Divacky return LBB->getName(); 238f22ef01cSRoman Divacky else 239f22ef01cSRoman Divacky return "(null)"; 240f22ef01cSRoman Divacky } 241f22ef01cSRoman Divacky 242dff0c46cSDimitry Andric /// Return a hopefully unique identifier for this block. 243dff0c46cSDimitry Andric std::string MachineBasicBlock::getFullName() const { 244dff0c46cSDimitry Andric std::string Name; 245dff0c46cSDimitry Andric if (getParent()) 246dff0c46cSDimitry Andric Name = (getParent()->getFunction()->getName() + ":").str(); 247dff0c46cSDimitry Andric if (getBasicBlock()) 248dff0c46cSDimitry Andric Name += getBasicBlock()->getName(); 249dff0c46cSDimitry Andric else 250dff0c46cSDimitry Andric Name += (Twine("BB") + Twine(getNumber())).str(); 251dff0c46cSDimitry Andric return Name; 252dff0c46cSDimitry Andric } 253dff0c46cSDimitry Andric 2542754fe60SDimitry Andric void MachineBasicBlock::print(raw_ostream &OS, SlotIndexes *Indexes) const { 255f22ef01cSRoman Divacky const MachineFunction *MF = getParent(); 256f22ef01cSRoman Divacky if (!MF) { 257f22ef01cSRoman Divacky OS << "Can't print out MachineBasicBlock because parent MachineFunction" 258f22ef01cSRoman Divacky << " is null\n"; 259f22ef01cSRoman Divacky return; 260f22ef01cSRoman Divacky } 261f22ef01cSRoman Divacky 2622754fe60SDimitry Andric if (Indexes) 2632754fe60SDimitry Andric OS << Indexes->getMBBStartIdx(this) << '\t'; 2642754fe60SDimitry Andric 265f22ef01cSRoman Divacky OS << "BB#" << getNumber() << ": "; 266f22ef01cSRoman Divacky 267f22ef01cSRoman Divacky const char *Comma = ""; 268f22ef01cSRoman Divacky if (const BasicBlock *LBB = getBasicBlock()) { 269f22ef01cSRoman Divacky OS << Comma << "derived from LLVM BB "; 270f22ef01cSRoman Divacky WriteAsOperand(OS, LBB, /*PrintType=*/false); 271f22ef01cSRoman Divacky Comma = ", "; 272f22ef01cSRoman Divacky } 273f22ef01cSRoman Divacky if (isLandingPad()) { OS << Comma << "EH LANDING PAD"; Comma = ", "; } 274f22ef01cSRoman Divacky if (hasAddressTaken()) { OS << Comma << "ADDRESS TAKEN"; Comma = ", "; } 2757ae0e2c9SDimitry Andric if (Alignment) 276dff0c46cSDimitry Andric OS << Comma << "Align " << Alignment << " (" << (1u << Alignment) 277dff0c46cSDimitry Andric << " bytes)"; 278dff0c46cSDimitry Andric 279f22ef01cSRoman Divacky OS << '\n'; 280f22ef01cSRoman Divacky 281f22ef01cSRoman Divacky const TargetRegisterInfo *TRI = MF->getTarget().getRegisterInfo(); 282f22ef01cSRoman Divacky if (!livein_empty()) { 2832754fe60SDimitry Andric if (Indexes) OS << '\t'; 284f22ef01cSRoman Divacky OS << " Live Ins:"; 285f22ef01cSRoman Divacky for (livein_iterator I = livein_begin(),E = livein_end(); I != E; ++I) 2862754fe60SDimitry Andric OS << ' ' << PrintReg(*I, TRI); 287f22ef01cSRoman Divacky OS << '\n'; 288f22ef01cSRoman Divacky } 289f22ef01cSRoman Divacky // Print the preds of this block according to the CFG. 290f22ef01cSRoman Divacky if (!pred_empty()) { 2912754fe60SDimitry Andric if (Indexes) OS << '\t'; 292f22ef01cSRoman Divacky OS << " Predecessors according to CFG:"; 293f22ef01cSRoman Divacky for (const_pred_iterator PI = pred_begin(), E = pred_end(); PI != E; ++PI) 294f22ef01cSRoman Divacky OS << " BB#" << (*PI)->getNumber(); 295f22ef01cSRoman Divacky OS << '\n'; 296f22ef01cSRoman Divacky } 297f22ef01cSRoman Divacky 298dff0c46cSDimitry Andric for (const_instr_iterator I = instr_begin(); I != instr_end(); ++I) { 2992754fe60SDimitry Andric if (Indexes) { 3002754fe60SDimitry Andric if (Indexes->hasIndex(I)) 3012754fe60SDimitry Andric OS << Indexes->getInstructionIndex(I); 3022754fe60SDimitry Andric OS << '\t'; 3032754fe60SDimitry Andric } 304f22ef01cSRoman Divacky OS << '\t'; 305dff0c46cSDimitry Andric if (I->isInsideBundle()) 306dff0c46cSDimitry Andric OS << " * "; 307f22ef01cSRoman Divacky I->print(OS, &getParent()->getTarget()); 308f22ef01cSRoman Divacky } 309f22ef01cSRoman Divacky 310f22ef01cSRoman Divacky // Print the successors of this block according to the CFG. 311f22ef01cSRoman Divacky if (!succ_empty()) { 3122754fe60SDimitry Andric if (Indexes) OS << '\t'; 313f22ef01cSRoman Divacky OS << " Successors according to CFG:"; 3147ae0e2c9SDimitry Andric for (const_succ_iterator SI = succ_begin(), E = succ_end(); SI != E; ++SI) { 315f22ef01cSRoman Divacky OS << " BB#" << (*SI)->getNumber(); 3167ae0e2c9SDimitry Andric if (!Weights.empty()) 3177ae0e2c9SDimitry Andric OS << '(' << *getWeightIterator(SI) << ')'; 3187ae0e2c9SDimitry Andric } 319f22ef01cSRoman Divacky OS << '\n'; 320f22ef01cSRoman Divacky } 321f22ef01cSRoman Divacky } 322f22ef01cSRoman Divacky 323f22ef01cSRoman Divacky void MachineBasicBlock::removeLiveIn(unsigned Reg) { 324f22ef01cSRoman Divacky std::vector<unsigned>::iterator I = 325f22ef01cSRoman Divacky std::find(LiveIns.begin(), LiveIns.end(), Reg); 326dff0c46cSDimitry Andric if (I != LiveIns.end()) 327f22ef01cSRoman Divacky LiveIns.erase(I); 328f22ef01cSRoman Divacky } 329f22ef01cSRoman Divacky 330f22ef01cSRoman Divacky bool MachineBasicBlock::isLiveIn(unsigned Reg) const { 331f22ef01cSRoman Divacky livein_iterator I = std::find(livein_begin(), livein_end(), Reg); 332f22ef01cSRoman Divacky return I != livein_end(); 333f22ef01cSRoman Divacky } 334f22ef01cSRoman Divacky 335f22ef01cSRoman Divacky void MachineBasicBlock::moveBefore(MachineBasicBlock *NewAfter) { 336f22ef01cSRoman Divacky getParent()->splice(NewAfter, this); 337f22ef01cSRoman Divacky } 338f22ef01cSRoman Divacky 339f22ef01cSRoman Divacky void MachineBasicBlock::moveAfter(MachineBasicBlock *NewBefore) { 340f22ef01cSRoman Divacky MachineFunction::iterator BBI = NewBefore; 341f22ef01cSRoman Divacky getParent()->splice(++BBI, this); 342f22ef01cSRoman Divacky } 343f22ef01cSRoman Divacky 344f22ef01cSRoman Divacky void MachineBasicBlock::updateTerminator() { 345f22ef01cSRoman Divacky const TargetInstrInfo *TII = getParent()->getTarget().getInstrInfo(); 346f22ef01cSRoman Divacky // A block with no successors has no concerns with fall-through edges. 347f22ef01cSRoman Divacky if (this->succ_empty()) return; 348f22ef01cSRoman Divacky 349f22ef01cSRoman Divacky MachineBasicBlock *TBB = 0, *FBB = 0; 350f22ef01cSRoman Divacky SmallVector<MachineOperand, 4> Cond; 351ffd1746dSEd Schouten DebugLoc dl; // FIXME: this is nowhere 352f22ef01cSRoman Divacky bool B = TII->AnalyzeBranch(*this, TBB, FBB, Cond); 353f22ef01cSRoman Divacky (void) B; 354f22ef01cSRoman Divacky assert(!B && "UpdateTerminators requires analyzable predecessors!"); 355f22ef01cSRoman Divacky if (Cond.empty()) { 356f22ef01cSRoman Divacky if (TBB) { 357f22ef01cSRoman Divacky // The block has an unconditional branch. If its successor is now 358f22ef01cSRoman Divacky // its layout successor, delete the branch. 359f22ef01cSRoman Divacky if (isLayoutSuccessor(TBB)) 360f22ef01cSRoman Divacky TII->RemoveBranch(*this); 361f22ef01cSRoman Divacky } else { 362f22ef01cSRoman Divacky // The block has an unconditional fallthrough. If its successor is not 363dff0c46cSDimitry Andric // its layout successor, insert a branch. First we have to locate the 364dff0c46cSDimitry Andric // only non-landing-pad successor, as that is the fallthrough block. 365dff0c46cSDimitry Andric for (succ_iterator SI = succ_begin(), SE = succ_end(); SI != SE; ++SI) { 366dff0c46cSDimitry Andric if ((*SI)->isLandingPad()) 367dff0c46cSDimitry Andric continue; 368dff0c46cSDimitry Andric assert(!TBB && "Found more than one non-landing-pad successor!"); 369dff0c46cSDimitry Andric TBB = *SI; 370dff0c46cSDimitry Andric } 371dff0c46cSDimitry Andric 372dff0c46cSDimitry Andric // If there is no non-landing-pad successor, the block has no 373dff0c46cSDimitry Andric // fall-through edges to be concerned with. 374dff0c46cSDimitry Andric if (!TBB) 375dff0c46cSDimitry Andric return; 376dff0c46cSDimitry Andric 377dff0c46cSDimitry Andric // Finally update the unconditional successor to be reached via a branch 378dff0c46cSDimitry Andric // if it would not be reached by fallthrough. 379f22ef01cSRoman Divacky if (!isLayoutSuccessor(TBB)) 380ffd1746dSEd Schouten TII->InsertBranch(*this, TBB, 0, Cond, dl); 381f22ef01cSRoman Divacky } 382f22ef01cSRoman Divacky } else { 383f22ef01cSRoman Divacky if (FBB) { 384f22ef01cSRoman Divacky // The block has a non-fallthrough conditional branch. If one of its 385f22ef01cSRoman Divacky // successors is its layout successor, rewrite it to a fallthrough 386f22ef01cSRoman Divacky // conditional branch. 387f22ef01cSRoman Divacky if (isLayoutSuccessor(TBB)) { 388f22ef01cSRoman Divacky if (TII->ReverseBranchCondition(Cond)) 389f22ef01cSRoman Divacky return; 390f22ef01cSRoman Divacky TII->RemoveBranch(*this); 391ffd1746dSEd Schouten TII->InsertBranch(*this, FBB, 0, Cond, dl); 392f22ef01cSRoman Divacky } else if (isLayoutSuccessor(FBB)) { 393f22ef01cSRoman Divacky TII->RemoveBranch(*this); 394ffd1746dSEd Schouten TII->InsertBranch(*this, TBB, 0, Cond, dl); 395f22ef01cSRoman Divacky } 396f22ef01cSRoman Divacky } else { 397cb4dff85SDimitry Andric // Walk through the successors and find the successor which is not 398cb4dff85SDimitry Andric // a landing pad and is not the conditional branch destination (in TBB) 399cb4dff85SDimitry Andric // as the fallthrough successor. 400cb4dff85SDimitry Andric MachineBasicBlock *FallthroughBB = 0; 401cb4dff85SDimitry Andric for (succ_iterator SI = succ_begin(), SE = succ_end(); SI != SE; ++SI) { 402cb4dff85SDimitry Andric if ((*SI)->isLandingPad() || *SI == TBB) 403cb4dff85SDimitry Andric continue; 404cb4dff85SDimitry Andric assert(!FallthroughBB && "Found more than one fallthrough successor."); 405cb4dff85SDimitry Andric FallthroughBB = *SI; 406cb4dff85SDimitry Andric } 407cb4dff85SDimitry Andric if (!FallthroughBB && canFallThrough()) { 408cb4dff85SDimitry Andric // We fallthrough to the same basic block as the conditional jump 409cb4dff85SDimitry Andric // targets. Remove the conditional jump, leaving unconditional 410cb4dff85SDimitry Andric // fallthrough. 411cb4dff85SDimitry Andric // FIXME: This does not seem like a reasonable pattern to support, but it 412cb4dff85SDimitry Andric // has been seen in the wild coming out of degenerate ARM test cases. 413cb4dff85SDimitry Andric TII->RemoveBranch(*this); 414cb4dff85SDimitry Andric 415cb4dff85SDimitry Andric // Finally update the unconditional successor to be reached via a branch 416cb4dff85SDimitry Andric // if it would not be reached by fallthrough. 417cb4dff85SDimitry Andric if (!isLayoutSuccessor(TBB)) 418cb4dff85SDimitry Andric TII->InsertBranch(*this, TBB, 0, Cond, dl); 419cb4dff85SDimitry Andric return; 420cb4dff85SDimitry Andric } 421cb4dff85SDimitry Andric 422f22ef01cSRoman Divacky // The block has a fallthrough conditional branch. 423f22ef01cSRoman Divacky if (isLayoutSuccessor(TBB)) { 424f22ef01cSRoman Divacky if (TII->ReverseBranchCondition(Cond)) { 425f22ef01cSRoman Divacky // We can't reverse the condition, add an unconditional branch. 426f22ef01cSRoman Divacky Cond.clear(); 427cb4dff85SDimitry Andric TII->InsertBranch(*this, FallthroughBB, 0, Cond, dl); 428f22ef01cSRoman Divacky return; 429f22ef01cSRoman Divacky } 430f22ef01cSRoman Divacky TII->RemoveBranch(*this); 431cb4dff85SDimitry Andric TII->InsertBranch(*this, FallthroughBB, 0, Cond, dl); 432cb4dff85SDimitry Andric } else if (!isLayoutSuccessor(FallthroughBB)) { 433f22ef01cSRoman Divacky TII->RemoveBranch(*this); 434cb4dff85SDimitry Andric TII->InsertBranch(*this, TBB, FallthroughBB, Cond, dl); 435f22ef01cSRoman Divacky } 436f22ef01cSRoman Divacky } 437f22ef01cSRoman Divacky } 438f22ef01cSRoman Divacky } 439f22ef01cSRoman Divacky 44017a519f9SDimitry Andric void MachineBasicBlock::addSuccessor(MachineBasicBlock *succ, uint32_t weight) { 44117a519f9SDimitry Andric 44217a519f9SDimitry Andric // If we see non-zero value for the first time it means we actually use Weight 44317a519f9SDimitry Andric // list, so we fill all Weights with 0's. 44417a519f9SDimitry Andric if (weight != 0 && Weights.empty()) 44517a519f9SDimitry Andric Weights.resize(Successors.size()); 44617a519f9SDimitry Andric 44717a519f9SDimitry Andric if (weight != 0 || !Weights.empty()) 44817a519f9SDimitry Andric Weights.push_back(weight); 44917a519f9SDimitry Andric 450f22ef01cSRoman Divacky Successors.push_back(succ); 451f22ef01cSRoman Divacky succ->addPredecessor(this); 452f22ef01cSRoman Divacky } 453f22ef01cSRoman Divacky 454f22ef01cSRoman Divacky void MachineBasicBlock::removeSuccessor(MachineBasicBlock *succ) { 455f22ef01cSRoman Divacky succ->removePredecessor(this); 456f22ef01cSRoman Divacky succ_iterator I = std::find(Successors.begin(), Successors.end(), succ); 457f22ef01cSRoman Divacky assert(I != Successors.end() && "Not a current successor!"); 45817a519f9SDimitry Andric 45917a519f9SDimitry Andric // If Weight list is empty it means we don't use it (disabled optimization). 46017a519f9SDimitry Andric if (!Weights.empty()) { 46117a519f9SDimitry Andric weight_iterator WI = getWeightIterator(I); 46217a519f9SDimitry Andric Weights.erase(WI); 46317a519f9SDimitry Andric } 46417a519f9SDimitry Andric 465f22ef01cSRoman Divacky Successors.erase(I); 466f22ef01cSRoman Divacky } 467f22ef01cSRoman Divacky 468f22ef01cSRoman Divacky MachineBasicBlock::succ_iterator 469f22ef01cSRoman Divacky MachineBasicBlock::removeSuccessor(succ_iterator I) { 470f22ef01cSRoman Divacky assert(I != Successors.end() && "Not a current successor!"); 47117a519f9SDimitry Andric 47217a519f9SDimitry Andric // If Weight list is empty it means we don't use it (disabled optimization). 47317a519f9SDimitry Andric if (!Weights.empty()) { 47417a519f9SDimitry Andric weight_iterator WI = getWeightIterator(I); 47517a519f9SDimitry Andric Weights.erase(WI); 47617a519f9SDimitry Andric } 47717a519f9SDimitry Andric 478f22ef01cSRoman Divacky (*I)->removePredecessor(this); 479f22ef01cSRoman Divacky return Successors.erase(I); 480f22ef01cSRoman Divacky } 481f22ef01cSRoman Divacky 48217a519f9SDimitry Andric void MachineBasicBlock::replaceSuccessor(MachineBasicBlock *Old, 48317a519f9SDimitry Andric MachineBasicBlock *New) { 4847ae0e2c9SDimitry Andric if (Old == New) 4857ae0e2c9SDimitry Andric return; 48617a519f9SDimitry Andric 4877ae0e2c9SDimitry Andric succ_iterator E = succ_end(); 4887ae0e2c9SDimitry Andric succ_iterator NewI = E; 4897ae0e2c9SDimitry Andric succ_iterator OldI = E; 4907ae0e2c9SDimitry Andric for (succ_iterator I = succ_begin(); I != E; ++I) { 4917ae0e2c9SDimitry Andric if (*I == Old) { 4927ae0e2c9SDimitry Andric OldI = I; 4937ae0e2c9SDimitry Andric if (NewI != E) 4947ae0e2c9SDimitry Andric break; 4957ae0e2c9SDimitry Andric } 4967ae0e2c9SDimitry Andric if (*I == New) { 4977ae0e2c9SDimitry Andric NewI = I; 4987ae0e2c9SDimitry Andric if (OldI != E) 4997ae0e2c9SDimitry Andric break; 5007ae0e2c9SDimitry Andric } 5017ae0e2c9SDimitry Andric } 5027ae0e2c9SDimitry Andric assert(OldI != E && "Old is not a successor of this block"); 5037ae0e2c9SDimitry Andric Old->removePredecessor(this); 5047ae0e2c9SDimitry Andric 5057ae0e2c9SDimitry Andric // If New isn't already a successor, let it take Old's place. 5067ae0e2c9SDimitry Andric if (NewI == E) { 5077ae0e2c9SDimitry Andric New->addPredecessor(this); 5087ae0e2c9SDimitry Andric *OldI = New; 5097ae0e2c9SDimitry Andric return; 51017a519f9SDimitry Andric } 51117a519f9SDimitry Andric 5127ae0e2c9SDimitry Andric // New is already a successor. 5137ae0e2c9SDimitry Andric // Update its weight instead of adding a duplicate edge. 5147ae0e2c9SDimitry Andric if (!Weights.empty()) { 5157ae0e2c9SDimitry Andric weight_iterator OldWI = getWeightIterator(OldI); 5167ae0e2c9SDimitry Andric *getWeightIterator(NewI) += *OldWI; 5177ae0e2c9SDimitry Andric Weights.erase(OldWI); 5187ae0e2c9SDimitry Andric } 5197ae0e2c9SDimitry Andric Successors.erase(OldI); 52017a519f9SDimitry Andric } 52117a519f9SDimitry Andric 522f22ef01cSRoman Divacky void MachineBasicBlock::addPredecessor(MachineBasicBlock *pred) { 523f22ef01cSRoman Divacky Predecessors.push_back(pred); 524f22ef01cSRoman Divacky } 525f22ef01cSRoman Divacky 526f22ef01cSRoman Divacky void MachineBasicBlock::removePredecessor(MachineBasicBlock *pred) { 5273b0f4066SDimitry Andric pred_iterator I = std::find(Predecessors.begin(), Predecessors.end(), pred); 528f22ef01cSRoman Divacky assert(I != Predecessors.end() && "Pred is not a predecessor of this block!"); 529f22ef01cSRoman Divacky Predecessors.erase(I); 530f22ef01cSRoman Divacky } 531f22ef01cSRoman Divacky 532f22ef01cSRoman Divacky void MachineBasicBlock::transferSuccessors(MachineBasicBlock *fromMBB) { 533f22ef01cSRoman Divacky if (this == fromMBB) 534f22ef01cSRoman Divacky return; 535f22ef01cSRoman Divacky 536ffd1746dSEd Schouten while (!fromMBB->succ_empty()) { 537ffd1746dSEd Schouten MachineBasicBlock *Succ = *fromMBB->succ_begin(); 5387ae0e2c9SDimitry Andric uint32_t Weight = 0; 53917a519f9SDimitry Andric 54017a519f9SDimitry Andric // If Weight list is empty it means we don't use it (disabled optimization). 54117a519f9SDimitry Andric if (!fromMBB->Weights.empty()) 5427ae0e2c9SDimitry Andric Weight = *fromMBB->Weights.begin(); 54317a519f9SDimitry Andric 5447ae0e2c9SDimitry Andric addSuccessor(Succ, Weight); 545ffd1746dSEd Schouten fromMBB->removeSuccessor(Succ); 546ffd1746dSEd Schouten } 547ffd1746dSEd Schouten } 548f22ef01cSRoman Divacky 549ffd1746dSEd Schouten void 550ffd1746dSEd Schouten MachineBasicBlock::transferSuccessorsAndUpdatePHIs(MachineBasicBlock *fromMBB) { 551ffd1746dSEd Schouten if (this == fromMBB) 552ffd1746dSEd Schouten return; 553ffd1746dSEd Schouten 554ffd1746dSEd Schouten while (!fromMBB->succ_empty()) { 555ffd1746dSEd Schouten MachineBasicBlock *Succ = *fromMBB->succ_begin(); 5567ae0e2c9SDimitry Andric uint32_t Weight = 0; 5577ae0e2c9SDimitry Andric if (!fromMBB->Weights.empty()) 5587ae0e2c9SDimitry Andric Weight = *fromMBB->Weights.begin(); 5597ae0e2c9SDimitry Andric addSuccessor(Succ, Weight); 560ffd1746dSEd Schouten fromMBB->removeSuccessor(Succ); 561ffd1746dSEd Schouten 562ffd1746dSEd Schouten // Fix up any PHI nodes in the successor. 563dff0c46cSDimitry Andric for (MachineBasicBlock::instr_iterator MI = Succ->instr_begin(), 564dff0c46cSDimitry Andric ME = Succ->instr_end(); MI != ME && MI->isPHI(); ++MI) 565ffd1746dSEd Schouten for (unsigned i = 2, e = MI->getNumOperands()+1; i != e; i += 2) { 566ffd1746dSEd Schouten MachineOperand &MO = MI->getOperand(i); 567ffd1746dSEd Schouten if (MO.getMBB() == fromMBB) 568ffd1746dSEd Schouten MO.setMBB(this); 569ffd1746dSEd Schouten } 570ffd1746dSEd Schouten } 571f22ef01cSRoman Divacky } 572f22ef01cSRoman Divacky 5737ae0e2c9SDimitry Andric bool MachineBasicBlock::isPredecessor(const MachineBasicBlock *MBB) const { 5747ae0e2c9SDimitry Andric return std::find(pred_begin(), pred_end(), MBB) != pred_end(); 5757ae0e2c9SDimitry Andric } 5767ae0e2c9SDimitry Andric 577f22ef01cSRoman Divacky bool MachineBasicBlock::isSuccessor(const MachineBasicBlock *MBB) const { 5787ae0e2c9SDimitry Andric return std::find(succ_begin(), succ_end(), MBB) != succ_end(); 579f22ef01cSRoman Divacky } 580f22ef01cSRoman Divacky 581f22ef01cSRoman Divacky bool MachineBasicBlock::isLayoutSuccessor(const MachineBasicBlock *MBB) const { 582f22ef01cSRoman Divacky MachineFunction::const_iterator I(this); 583f22ef01cSRoman Divacky return llvm::next(I) == MachineFunction::const_iterator(MBB); 584f22ef01cSRoman Divacky } 585f22ef01cSRoman Divacky 586f22ef01cSRoman Divacky bool MachineBasicBlock::canFallThrough() { 587f22ef01cSRoman Divacky MachineFunction::iterator Fallthrough = this; 588f22ef01cSRoman Divacky ++Fallthrough; 589f22ef01cSRoman Divacky // If FallthroughBlock is off the end of the function, it can't fall through. 590f22ef01cSRoman Divacky if (Fallthrough == getParent()->end()) 591f22ef01cSRoman Divacky return false; 592f22ef01cSRoman Divacky 593f22ef01cSRoman Divacky // If FallthroughBlock isn't a successor, no fallthrough is possible. 594f22ef01cSRoman Divacky if (!isSuccessor(Fallthrough)) 595f22ef01cSRoman Divacky return false; 596f22ef01cSRoman Divacky 597f22ef01cSRoman Divacky // Analyze the branches, if any, at the end of the block. 598f22ef01cSRoman Divacky MachineBasicBlock *TBB = 0, *FBB = 0; 599f22ef01cSRoman Divacky SmallVector<MachineOperand, 4> Cond; 600f22ef01cSRoman Divacky const TargetInstrInfo *TII = getParent()->getTarget().getInstrInfo(); 601f22ef01cSRoman Divacky if (TII->AnalyzeBranch(*this, TBB, FBB, Cond)) { 602f22ef01cSRoman Divacky // If we couldn't analyze the branch, examine the last instruction. 603f22ef01cSRoman Divacky // If the block doesn't end in a known control barrier, assume fallthrough 604dff0c46cSDimitry Andric // is possible. The isPredicated check is needed because this code can be 605f22ef01cSRoman Divacky // called during IfConversion, where an instruction which is normally a 606dff0c46cSDimitry Andric // Barrier is predicated and thus no longer an actual control barrier. 607dff0c46cSDimitry Andric return empty() || !back().isBarrier() || TII->isPredicated(&back()); 608f22ef01cSRoman Divacky } 609f22ef01cSRoman Divacky 610f22ef01cSRoman Divacky // If there is no branch, control always falls through. 611f22ef01cSRoman Divacky if (TBB == 0) return true; 612f22ef01cSRoman Divacky 613f22ef01cSRoman Divacky // If there is some explicit branch to the fallthrough block, it can obviously 614f22ef01cSRoman Divacky // reach, even though the branch should get folded to fall through implicitly. 615f22ef01cSRoman Divacky if (MachineFunction::iterator(TBB) == Fallthrough || 616f22ef01cSRoman Divacky MachineFunction::iterator(FBB) == Fallthrough) 617f22ef01cSRoman Divacky return true; 618f22ef01cSRoman Divacky 619f22ef01cSRoman Divacky // If it's an unconditional branch to some block not the fall through, it 620f22ef01cSRoman Divacky // doesn't fall through. 621f22ef01cSRoman Divacky if (Cond.empty()) return false; 622f22ef01cSRoman Divacky 623f22ef01cSRoman Divacky // Otherwise, if it is conditional and has no explicit false block, it falls 624f22ef01cSRoman Divacky // through. 625f22ef01cSRoman Divacky return FBB == 0; 626f22ef01cSRoman Divacky } 627f22ef01cSRoman Divacky 628ffd1746dSEd Schouten MachineBasicBlock * 629ffd1746dSEd Schouten MachineBasicBlock::SplitCriticalEdge(MachineBasicBlock *Succ, Pass *P) { 6307ae0e2c9SDimitry Andric // Splitting the critical edge to a landing pad block is non-trivial. Don't do 6317ae0e2c9SDimitry Andric // it in this generic function. 6327ae0e2c9SDimitry Andric if (Succ->isLandingPad()) 6337ae0e2c9SDimitry Andric return NULL; 6347ae0e2c9SDimitry Andric 635ffd1746dSEd Schouten MachineFunction *MF = getParent(); 636ffd1746dSEd Schouten DebugLoc dl; // FIXME: this is nowhere 637ffd1746dSEd Schouten 6382754fe60SDimitry Andric // We may need to update this's terminator, but we can't do that if 6392754fe60SDimitry Andric // AnalyzeBranch fails. If this uses a jump table, we won't touch it. 640ffd1746dSEd Schouten const TargetInstrInfo *TII = MF->getTarget().getInstrInfo(); 641ffd1746dSEd Schouten MachineBasicBlock *TBB = 0, *FBB = 0; 642ffd1746dSEd Schouten SmallVector<MachineOperand, 4> Cond; 643ffd1746dSEd Schouten if (TII->AnalyzeBranch(*this, TBB, FBB, Cond)) 644ffd1746dSEd Schouten return NULL; 645ffd1746dSEd Schouten 6462754fe60SDimitry Andric // Avoid bugpoint weirdness: A block may end with a conditional branch but 6472754fe60SDimitry Andric // jumps to the same MBB is either case. We have duplicate CFG edges in that 6482754fe60SDimitry Andric // case that we can't handle. Since this never happens in properly optimized 6492754fe60SDimitry Andric // code, just skip those edges. 6502754fe60SDimitry Andric if (TBB && TBB == FBB) { 6512754fe60SDimitry Andric DEBUG(dbgs() << "Won't split critical edge after degenerate BB#" 6522754fe60SDimitry Andric << getNumber() << '\n'); 6532754fe60SDimitry Andric return NULL; 6542754fe60SDimitry Andric } 6552754fe60SDimitry Andric 656ffd1746dSEd Schouten MachineBasicBlock *NMBB = MF->CreateMachineBasicBlock(); 657ffd1746dSEd Schouten MF->insert(llvm::next(MachineFunction::iterator(this)), NMBB); 658e580952dSDimitry Andric DEBUG(dbgs() << "Splitting critical edge:" 659ffd1746dSEd Schouten " BB#" << getNumber() 660ffd1746dSEd Schouten << " -- BB#" << NMBB->getNumber() 661ffd1746dSEd Schouten << " -- BB#" << Succ->getNumber() << '\n'); 662ffd1746dSEd Schouten 663bd5abe19SDimitry Andric // On some targets like Mips, branches may kill virtual registers. Make sure 664bd5abe19SDimitry Andric // that LiveVariables is properly updated after updateTerminator replaces the 665bd5abe19SDimitry Andric // terminators. 666bd5abe19SDimitry Andric LiveVariables *LV = P->getAnalysisIfAvailable<LiveVariables>(); 667bd5abe19SDimitry Andric 668bd5abe19SDimitry Andric // Collect a list of virtual registers killed by the terminators. 669bd5abe19SDimitry Andric SmallVector<unsigned, 4> KilledRegs; 670bd5abe19SDimitry Andric if (LV) 671dff0c46cSDimitry Andric for (instr_iterator I = getFirstInstrTerminator(), E = instr_end(); 672dff0c46cSDimitry Andric I != E; ++I) { 673bd5abe19SDimitry Andric MachineInstr *MI = I; 674bd5abe19SDimitry Andric for (MachineInstr::mop_iterator OI = MI->operands_begin(), 675bd5abe19SDimitry Andric OE = MI->operands_end(); OI != OE; ++OI) { 676dff0c46cSDimitry Andric if (!OI->isReg() || OI->getReg() == 0 || 677dff0c46cSDimitry Andric !OI->isUse() || !OI->isKill() || OI->isUndef()) 678bd5abe19SDimitry Andric continue; 679bd5abe19SDimitry Andric unsigned Reg = OI->getReg(); 680dff0c46cSDimitry Andric if (TargetRegisterInfo::isPhysicalRegister(Reg) || 681bd5abe19SDimitry Andric LV->getVarInfo(Reg).removeKill(MI)) { 682bd5abe19SDimitry Andric KilledRegs.push_back(Reg); 683bd5abe19SDimitry Andric DEBUG(dbgs() << "Removing terminator kill: " << *MI); 684bd5abe19SDimitry Andric OI->setIsKill(false); 685bd5abe19SDimitry Andric } 686bd5abe19SDimitry Andric } 687bd5abe19SDimitry Andric } 688bd5abe19SDimitry Andric 689ffd1746dSEd Schouten ReplaceUsesOfBlockWith(Succ, NMBB); 690ffd1746dSEd Schouten updateTerminator(); 691ffd1746dSEd Schouten 692ffd1746dSEd Schouten // Insert unconditional "jump Succ" instruction in NMBB if necessary. 693ffd1746dSEd Schouten NMBB->addSuccessor(Succ); 694ffd1746dSEd Schouten if (!NMBB->isLayoutSuccessor(Succ)) { 695ffd1746dSEd Schouten Cond.clear(); 696ffd1746dSEd Schouten MF->getTarget().getInstrInfo()->InsertBranch(*NMBB, Succ, NULL, Cond, dl); 697ffd1746dSEd Schouten } 698ffd1746dSEd Schouten 699ffd1746dSEd Schouten // Fix PHI nodes in Succ so they refer to NMBB instead of this 700dff0c46cSDimitry Andric for (MachineBasicBlock::instr_iterator 701dff0c46cSDimitry Andric i = Succ->instr_begin(),e = Succ->instr_end(); 702ffd1746dSEd Schouten i != e && i->isPHI(); ++i) 703ffd1746dSEd Schouten for (unsigned ni = 1, ne = i->getNumOperands(); ni != ne; ni += 2) 704ffd1746dSEd Schouten if (i->getOperand(ni+1).getMBB() == this) 705ffd1746dSEd Schouten i->getOperand(ni+1).setMBB(NMBB); 706ffd1746dSEd Schouten 7076122f3e6SDimitry Andric // Inherit live-ins from the successor 7086122f3e6SDimitry Andric for (MachineBasicBlock::livein_iterator I = Succ->livein_begin(), 7096122f3e6SDimitry Andric E = Succ->livein_end(); I != E; ++I) 7106122f3e6SDimitry Andric NMBB->addLiveIn(*I); 7116122f3e6SDimitry Andric 712bd5abe19SDimitry Andric // Update LiveVariables. 713dff0c46cSDimitry Andric const TargetRegisterInfo *TRI = MF->getTarget().getRegisterInfo(); 714bd5abe19SDimitry Andric if (LV) { 715bd5abe19SDimitry Andric // Restore kills of virtual registers that were killed by the terminators. 716bd5abe19SDimitry Andric while (!KilledRegs.empty()) { 717bd5abe19SDimitry Andric unsigned Reg = KilledRegs.pop_back_val(); 718dff0c46cSDimitry Andric for (instr_iterator I = instr_end(), E = instr_begin(); I != E;) { 719dff0c46cSDimitry Andric if (!(--I)->addRegisterKilled(Reg, TRI, /* addIfNotFound= */ false)) 720bd5abe19SDimitry Andric continue; 721dff0c46cSDimitry Andric if (TargetRegisterInfo::isVirtualRegister(Reg)) 722bd5abe19SDimitry Andric LV->getVarInfo(Reg).Kills.push_back(I); 723bd5abe19SDimitry Andric DEBUG(dbgs() << "Restored terminator kill: " << *I); 724bd5abe19SDimitry Andric break; 725bd5abe19SDimitry Andric } 726bd5abe19SDimitry Andric } 727bd5abe19SDimitry Andric // Update relevant live-through information. 728ffd1746dSEd Schouten LV->addNewBlock(NMBB, this, Succ); 729bd5abe19SDimitry Andric } 730ffd1746dSEd Schouten 731ffd1746dSEd Schouten if (MachineDominatorTree *MDT = 732e580952dSDimitry Andric P->getAnalysisIfAvailable<MachineDominatorTree>()) { 733e580952dSDimitry Andric // Update dominator information. 734e580952dSDimitry Andric MachineDomTreeNode *SucccDTNode = MDT->getNode(Succ); 735ffd1746dSEd Schouten 736e580952dSDimitry Andric bool IsNewIDom = true; 737e580952dSDimitry Andric for (const_pred_iterator PI = Succ->pred_begin(), E = Succ->pred_end(); 738e580952dSDimitry Andric PI != E; ++PI) { 739e580952dSDimitry Andric MachineBasicBlock *PredBB = *PI; 740e580952dSDimitry Andric if (PredBB == NMBB) 741e580952dSDimitry Andric continue; 742e580952dSDimitry Andric if (!MDT->dominates(SucccDTNode, MDT->getNode(PredBB))) { 743e580952dSDimitry Andric IsNewIDom = false; 744e580952dSDimitry Andric break; 745e580952dSDimitry Andric } 746e580952dSDimitry Andric } 747e580952dSDimitry Andric 748e580952dSDimitry Andric // We know "this" dominates the newly created basic block. 749e580952dSDimitry Andric MachineDomTreeNode *NewDTNode = MDT->addNewBlock(NMBB, this); 750e580952dSDimitry Andric 751e580952dSDimitry Andric // If all the other predecessors of "Succ" are dominated by "Succ" itself 752e580952dSDimitry Andric // then the new block is the new immediate dominator of "Succ". Otherwise, 753e580952dSDimitry Andric // the new block doesn't dominate anything. 754e580952dSDimitry Andric if (IsNewIDom) 755e580952dSDimitry Andric MDT->changeImmediateDominator(SucccDTNode, NewDTNode); 756e580952dSDimitry Andric } 757e580952dSDimitry Andric 758e580952dSDimitry Andric if (MachineLoopInfo *MLI = P->getAnalysisIfAvailable<MachineLoopInfo>()) 759ffd1746dSEd Schouten if (MachineLoop *TIL = MLI->getLoopFor(this)) { 760ffd1746dSEd Schouten // If one or the other blocks were not in a loop, the new block is not 761ffd1746dSEd Schouten // either, and thus LI doesn't need to be updated. 762ffd1746dSEd Schouten if (MachineLoop *DestLoop = MLI->getLoopFor(Succ)) { 763ffd1746dSEd Schouten if (TIL == DestLoop) { 764ffd1746dSEd Schouten // Both in the same loop, the NMBB joins loop. 765ffd1746dSEd Schouten DestLoop->addBasicBlockToLoop(NMBB, MLI->getBase()); 766ffd1746dSEd Schouten } else if (TIL->contains(DestLoop)) { 767ffd1746dSEd Schouten // Edge from an outer loop to an inner loop. Add to the outer loop. 768ffd1746dSEd Schouten TIL->addBasicBlockToLoop(NMBB, MLI->getBase()); 769ffd1746dSEd Schouten } else if (DestLoop->contains(TIL)) { 770ffd1746dSEd Schouten // Edge from an inner loop to an outer loop. Add to the outer loop. 771ffd1746dSEd Schouten DestLoop->addBasicBlockToLoop(NMBB, MLI->getBase()); 772ffd1746dSEd Schouten } else { 773ffd1746dSEd Schouten // Edge from two loops with no containment relation. Because these 774ffd1746dSEd Schouten // are natural loops, we know that the destination block must be the 775ffd1746dSEd Schouten // header of its loop (adding a branch into a loop elsewhere would 776ffd1746dSEd Schouten // create an irreducible loop). 777ffd1746dSEd Schouten assert(DestLoop->getHeader() == Succ && 778ffd1746dSEd Schouten "Should not create irreducible loops!"); 779ffd1746dSEd Schouten if (MachineLoop *P = DestLoop->getParentLoop()) 780ffd1746dSEd Schouten P->addBasicBlockToLoop(NMBB, MLI->getBase()); 781ffd1746dSEd Schouten } 782ffd1746dSEd Schouten } 783ffd1746dSEd Schouten } 784ffd1746dSEd Schouten 785ffd1746dSEd Schouten return NMBB; 786ffd1746dSEd Schouten } 787ffd1746dSEd Schouten 788dff0c46cSDimitry Andric MachineBasicBlock::iterator 789dff0c46cSDimitry Andric MachineBasicBlock::erase(MachineBasicBlock::iterator I) { 790dff0c46cSDimitry Andric if (I->isBundle()) { 791dff0c46cSDimitry Andric MachineBasicBlock::iterator E = llvm::next(I); 792dff0c46cSDimitry Andric return Insts.erase(I.getInstrIterator(), E.getInstrIterator()); 793dff0c46cSDimitry Andric } 794dff0c46cSDimitry Andric 795dff0c46cSDimitry Andric return Insts.erase(I.getInstrIterator()); 796dff0c46cSDimitry Andric } 797dff0c46cSDimitry Andric 798dff0c46cSDimitry Andric MachineInstr *MachineBasicBlock::remove(MachineInstr *I) { 799dff0c46cSDimitry Andric if (I->isBundle()) { 800dff0c46cSDimitry Andric instr_iterator MII = llvm::next(I); 801dff0c46cSDimitry Andric iterator E = end(); 802dff0c46cSDimitry Andric while (MII != E && MII->isInsideBundle()) { 803dff0c46cSDimitry Andric MachineInstr *MI = &*MII++; 804dff0c46cSDimitry Andric Insts.remove(MI); 805dff0c46cSDimitry Andric } 806dff0c46cSDimitry Andric } 807dff0c46cSDimitry Andric 808dff0c46cSDimitry Andric return Insts.remove(I); 809dff0c46cSDimitry Andric } 810dff0c46cSDimitry Andric 811dff0c46cSDimitry Andric void MachineBasicBlock::splice(MachineBasicBlock::iterator where, 812dff0c46cSDimitry Andric MachineBasicBlock *Other, 813dff0c46cSDimitry Andric MachineBasicBlock::iterator From) { 814dff0c46cSDimitry Andric if (From->isBundle()) { 815dff0c46cSDimitry Andric MachineBasicBlock::iterator To = llvm::next(From); 816dff0c46cSDimitry Andric Insts.splice(where.getInstrIterator(), Other->Insts, 817dff0c46cSDimitry Andric From.getInstrIterator(), To.getInstrIterator()); 818dff0c46cSDimitry Andric return; 819dff0c46cSDimitry Andric } 820dff0c46cSDimitry Andric 821dff0c46cSDimitry Andric Insts.splice(where.getInstrIterator(), Other->Insts, From.getInstrIterator()); 822dff0c46cSDimitry Andric } 823dff0c46cSDimitry Andric 824f22ef01cSRoman Divacky /// removeFromParent - This method unlinks 'this' from the containing function, 825f22ef01cSRoman Divacky /// and returns it, but does not delete it. 826f22ef01cSRoman Divacky MachineBasicBlock *MachineBasicBlock::removeFromParent() { 827f22ef01cSRoman Divacky assert(getParent() && "Not embedded in a function!"); 828f22ef01cSRoman Divacky getParent()->remove(this); 829f22ef01cSRoman Divacky return this; 830f22ef01cSRoman Divacky } 831f22ef01cSRoman Divacky 832f22ef01cSRoman Divacky 833f22ef01cSRoman Divacky /// eraseFromParent - This method unlinks 'this' from the containing function, 834f22ef01cSRoman Divacky /// and deletes it. 835f22ef01cSRoman Divacky void MachineBasicBlock::eraseFromParent() { 836f22ef01cSRoman Divacky assert(getParent() && "Not embedded in a function!"); 837f22ef01cSRoman Divacky getParent()->erase(this); 838f22ef01cSRoman Divacky } 839f22ef01cSRoman Divacky 840f22ef01cSRoman Divacky 841f22ef01cSRoman Divacky /// ReplaceUsesOfBlockWith - Given a machine basic block that branched to 842f22ef01cSRoman Divacky /// 'Old', change the code and CFG so that it branches to 'New' instead. 843f22ef01cSRoman Divacky void MachineBasicBlock::ReplaceUsesOfBlockWith(MachineBasicBlock *Old, 844f22ef01cSRoman Divacky MachineBasicBlock *New) { 845f22ef01cSRoman Divacky assert(Old != New && "Cannot replace self with self!"); 846f22ef01cSRoman Divacky 847dff0c46cSDimitry Andric MachineBasicBlock::instr_iterator I = instr_end(); 848dff0c46cSDimitry Andric while (I != instr_begin()) { 849f22ef01cSRoman Divacky --I; 850dff0c46cSDimitry Andric if (!I->isTerminator()) break; 851f22ef01cSRoman Divacky 852f22ef01cSRoman Divacky // Scan the operands of this machine instruction, replacing any uses of Old 853f22ef01cSRoman Divacky // with New. 854f22ef01cSRoman Divacky for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i) 855f22ef01cSRoman Divacky if (I->getOperand(i).isMBB() && 856f22ef01cSRoman Divacky I->getOperand(i).getMBB() == Old) 857f22ef01cSRoman Divacky I->getOperand(i).setMBB(New); 858f22ef01cSRoman Divacky } 859f22ef01cSRoman Divacky 860f22ef01cSRoman Divacky // Update the successor information. 86117a519f9SDimitry Andric replaceSuccessor(Old, New); 862f22ef01cSRoman Divacky } 863f22ef01cSRoman Divacky 864f22ef01cSRoman Divacky /// CorrectExtraCFGEdges - Various pieces of code can cause excess edges in the 865f22ef01cSRoman Divacky /// CFG to be inserted. If we have proven that MBB can only branch to DestA and 866f22ef01cSRoman Divacky /// DestB, remove any other MBB successors from the CFG. DestA and DestB can be 867f22ef01cSRoman Divacky /// null. 868f22ef01cSRoman Divacky /// 869f22ef01cSRoman Divacky /// Besides DestA and DestB, retain other edges leading to LandingPads 870f22ef01cSRoman Divacky /// (currently there can be only one; we don't check or require that here). 871f22ef01cSRoman Divacky /// Note it is possible that DestA and/or DestB are LandingPads. 872f22ef01cSRoman Divacky bool MachineBasicBlock::CorrectExtraCFGEdges(MachineBasicBlock *DestA, 873f22ef01cSRoman Divacky MachineBasicBlock *DestB, 874f22ef01cSRoman Divacky bool isCond) { 875f22ef01cSRoman Divacky // The values of DestA and DestB frequently come from a call to the 876f22ef01cSRoman Divacky // 'TargetInstrInfo::AnalyzeBranch' method. We take our meaning of the initial 877f22ef01cSRoman Divacky // values from there. 878f22ef01cSRoman Divacky // 879f22ef01cSRoman Divacky // 1. If both DestA and DestB are null, then the block ends with no branches 880f22ef01cSRoman Divacky // (it falls through to its successor). 881f22ef01cSRoman Divacky // 2. If DestA is set, DestB is null, and isCond is false, then the block ends 882f22ef01cSRoman Divacky // with only an unconditional branch. 883f22ef01cSRoman Divacky // 3. If DestA is set, DestB is null, and isCond is true, then the block ends 884f22ef01cSRoman Divacky // with a conditional branch that falls through to a successor (DestB). 885f22ef01cSRoman Divacky // 4. If DestA and DestB is set and isCond is true, then the block ends with a 886f22ef01cSRoman Divacky // conditional branch followed by an unconditional branch. DestA is the 887f22ef01cSRoman Divacky // 'true' destination and DestB is the 'false' destination. 888f22ef01cSRoman Divacky 889f22ef01cSRoman Divacky bool Changed = false; 890f22ef01cSRoman Divacky 891f22ef01cSRoman Divacky MachineFunction::iterator FallThru = 892f22ef01cSRoman Divacky llvm::next(MachineFunction::iterator(this)); 893f22ef01cSRoman Divacky 894f22ef01cSRoman Divacky if (DestA == 0 && DestB == 0) { 895f22ef01cSRoman Divacky // Block falls through to successor. 896f22ef01cSRoman Divacky DestA = FallThru; 897f22ef01cSRoman Divacky DestB = FallThru; 898f22ef01cSRoman Divacky } else if (DestA != 0 && DestB == 0) { 899f22ef01cSRoman Divacky if (isCond) 900f22ef01cSRoman Divacky // Block ends in conditional jump that falls through to successor. 901f22ef01cSRoman Divacky DestB = FallThru; 902f22ef01cSRoman Divacky } else { 903f22ef01cSRoman Divacky assert(DestA && DestB && isCond && 904f22ef01cSRoman Divacky "CFG in a bad state. Cannot correct CFG edges"); 905f22ef01cSRoman Divacky } 906f22ef01cSRoman Divacky 907f22ef01cSRoman Divacky // Remove superfluous edges. I.e., those which aren't destinations of this 908f22ef01cSRoman Divacky // basic block, duplicate edges, or landing pads. 909f22ef01cSRoman Divacky SmallPtrSet<const MachineBasicBlock*, 8> SeenMBBs; 910f22ef01cSRoman Divacky MachineBasicBlock::succ_iterator SI = succ_begin(); 911f22ef01cSRoman Divacky while (SI != succ_end()) { 912f22ef01cSRoman Divacky const MachineBasicBlock *MBB = *SI; 913f22ef01cSRoman Divacky if (!SeenMBBs.insert(MBB) || 914f22ef01cSRoman Divacky (MBB != DestA && MBB != DestB && !MBB->isLandingPad())) { 915f22ef01cSRoman Divacky // This is a superfluous edge, remove it. 916f22ef01cSRoman Divacky SI = removeSuccessor(SI); 917f22ef01cSRoman Divacky Changed = true; 918f22ef01cSRoman Divacky } else { 919f22ef01cSRoman Divacky ++SI; 920f22ef01cSRoman Divacky } 921f22ef01cSRoman Divacky } 922f22ef01cSRoman Divacky 923f22ef01cSRoman Divacky return Changed; 924f22ef01cSRoman Divacky } 925f22ef01cSRoman Divacky 926f22ef01cSRoman Divacky /// findDebugLoc - find the next valid DebugLoc starting at MBBI, skipping 927f22ef01cSRoman Divacky /// any DBG_VALUE instructions. Return UnknownLoc if there is none. 928f22ef01cSRoman Divacky DebugLoc 929dff0c46cSDimitry Andric MachineBasicBlock::findDebugLoc(instr_iterator MBBI) { 930f22ef01cSRoman Divacky DebugLoc DL; 931dff0c46cSDimitry Andric instr_iterator E = instr_end(); 932dff0c46cSDimitry Andric if (MBBI == E) 933dff0c46cSDimitry Andric return DL; 934dff0c46cSDimitry Andric 935f22ef01cSRoman Divacky // Skip debug declarations, we don't want a DebugLoc from them. 936dff0c46cSDimitry Andric while (MBBI != E && MBBI->isDebugValue()) 937dff0c46cSDimitry Andric MBBI++; 938dff0c46cSDimitry Andric if (MBBI != E) 939dff0c46cSDimitry Andric DL = MBBI->getDebugLoc(); 940f22ef01cSRoman Divacky return DL; 941f22ef01cSRoman Divacky } 942f22ef01cSRoman Divacky 94317a519f9SDimitry Andric /// getSuccWeight - Return weight of the edge from this block to MBB. 94417a519f9SDimitry Andric /// 945dff0c46cSDimitry Andric uint32_t MachineBasicBlock::getSuccWeight(const MachineBasicBlock *succ) const { 94617a519f9SDimitry Andric if (Weights.empty()) 94717a519f9SDimitry Andric return 0; 94817a519f9SDimitry Andric 949dff0c46cSDimitry Andric const_succ_iterator I = std::find(Successors.begin(), Successors.end(), succ); 95017a519f9SDimitry Andric return *getWeightIterator(I); 95117a519f9SDimitry Andric } 95217a519f9SDimitry Andric 95317a519f9SDimitry Andric /// getWeightIterator - Return wight iterator corresonding to the I successor 95417a519f9SDimitry Andric /// iterator 95517a519f9SDimitry Andric MachineBasicBlock::weight_iterator MachineBasicBlock:: 95617a519f9SDimitry Andric getWeightIterator(MachineBasicBlock::succ_iterator I) { 95717a519f9SDimitry Andric assert(Weights.size() == Successors.size() && "Async weight list!"); 95817a519f9SDimitry Andric size_t index = std::distance(Successors.begin(), I); 95917a519f9SDimitry Andric assert(index < Weights.size() && "Not a current successor!"); 96017a519f9SDimitry Andric return Weights.begin() + index; 96117a519f9SDimitry Andric } 96217a519f9SDimitry Andric 963dff0c46cSDimitry Andric /// getWeightIterator - Return wight iterator corresonding to the I successor 964dff0c46cSDimitry Andric /// iterator 965dff0c46cSDimitry Andric MachineBasicBlock::const_weight_iterator MachineBasicBlock:: 966dff0c46cSDimitry Andric getWeightIterator(MachineBasicBlock::const_succ_iterator I) const { 967dff0c46cSDimitry Andric assert(Weights.size() == Successors.size() && "Async weight list!"); 968dff0c46cSDimitry Andric const size_t index = std::distance(Successors.begin(), I); 969dff0c46cSDimitry Andric assert(index < Weights.size() && "Not a current successor!"); 970dff0c46cSDimitry Andric return Weights.begin() + index; 971dff0c46cSDimitry Andric } 972dff0c46cSDimitry Andric 973f22ef01cSRoman Divacky void llvm::WriteAsOperand(raw_ostream &OS, const MachineBasicBlock *MBB, 974f22ef01cSRoman Divacky bool t) { 975f22ef01cSRoman Divacky OS << "BB#" << MBB->getNumber(); 976f22ef01cSRoman Divacky } 977f22ef01cSRoman Divacky 978