1 //===- SimplifyCFG.cpp - Code to perform CFG simplification ---------------===//
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 // Peephole optimize the CFG.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #define DEBUG_TYPE "simplifycfg"
15 #include "llvm/Transforms/Utils/Local.h"
16 #include "llvm/Constants.h"
17 #include "llvm/Instructions.h"
18 #include "llvm/IntrinsicInst.h"
19 #include "llvm/Type.h"
20 #include "llvm/DerivedTypes.h"
21 #include "llvm/GlobalVariable.h"
22 #include "llvm/Support/CFG.h"
23 #include "llvm/Support/Debug.h"
24 #include "llvm/Analysis/ConstantFolding.h"
25 #include "llvm/Transforms/Utils/BasicBlockUtils.h"
26 #include "llvm/ADT/SmallVector.h"
27 #include "llvm/ADT/SmallPtrSet.h"
28 #include "llvm/ADT/Statistic.h"
29 #include <algorithm>
30 #include <functional>
31 #include <set>
32 #include <map>
33 using namespace llvm;
34 
35 STATISTIC(NumSpeculations, "Number of speculative executed instructions");
36 
37 /// SafeToMergeTerminators - Return true if it is safe to merge these two
38 /// terminator instructions together.
39 ///
40 static bool SafeToMergeTerminators(TerminatorInst *SI1, TerminatorInst *SI2) {
41   if (SI1 == SI2) return false;  // Can't merge with self!
42 
43   // It is not safe to merge these two switch instructions if they have a common
44   // successor, and if that successor has a PHI node, and if *that* PHI node has
45   // conflicting incoming values from the two switch blocks.
46   BasicBlock *SI1BB = SI1->getParent();
47   BasicBlock *SI2BB = SI2->getParent();
48   SmallPtrSet<BasicBlock*, 16> SI1Succs(succ_begin(SI1BB), succ_end(SI1BB));
49 
50   for (succ_iterator I = succ_begin(SI2BB), E = succ_end(SI2BB); I != E; ++I)
51     if (SI1Succs.count(*I))
52       for (BasicBlock::iterator BBI = (*I)->begin();
53            isa<PHINode>(BBI); ++BBI) {
54         PHINode *PN = cast<PHINode>(BBI);
55         if (PN->getIncomingValueForBlock(SI1BB) !=
56             PN->getIncomingValueForBlock(SI2BB))
57           return false;
58       }
59 
60   return true;
61 }
62 
63 /// AddPredecessorToBlock - Update PHI nodes in Succ to indicate that there will
64 /// now be entries in it from the 'NewPred' block.  The values that will be
65 /// flowing into the PHI nodes will be the same as those coming in from
66 /// ExistPred, an existing predecessor of Succ.
67 static void AddPredecessorToBlock(BasicBlock *Succ, BasicBlock *NewPred,
68                                   BasicBlock *ExistPred) {
69   assert(std::find(succ_begin(ExistPred), succ_end(ExistPred), Succ) !=
70          succ_end(ExistPred) && "ExistPred is not a predecessor of Succ!");
71   if (!isa<PHINode>(Succ->begin())) return; // Quick exit if nothing to do
72 
73   PHINode *PN;
74   for (BasicBlock::iterator I = Succ->begin();
75        (PN = dyn_cast<PHINode>(I)); ++I)
76     PN->addIncoming(PN->getIncomingValueForBlock(ExistPred), NewPred);
77 }
78 
79 /// CanPropagatePredecessorsForPHIs - Return true if we can fold BB, an
80 /// almost-empty BB ending in an unconditional branch to Succ, into succ.
81 ///
82 /// Assumption: Succ is the single successor for BB.
83 ///
84 static bool CanPropagatePredecessorsForPHIs(BasicBlock *BB, BasicBlock *Succ) {
85   assert(*succ_begin(BB) == Succ && "Succ is not successor of BB!");
86 
87   DOUT << "Looking to fold " << BB->getNameStart() << " into "
88        << Succ->getNameStart() << "\n";
89   // Shortcut, if there is only a single predecessor it must be BB and merging
90   // is always safe
91   if (Succ->getSinglePredecessor()) return true;
92 
93   typedef SmallPtrSet<Instruction*, 16> InstrSet;
94   InstrSet BBPHIs;
95 
96   // Make a list of all phi nodes in BB
97   BasicBlock::iterator BBI = BB->begin();
98   while (isa<PHINode>(*BBI)) BBPHIs.insert(BBI++);
99 
100   // Make a list of the predecessors of BB
101   typedef SmallPtrSet<BasicBlock*, 16> BlockSet;
102   BlockSet BBPreds(pred_begin(BB), pred_end(BB));
103 
104   // Use that list to make another list of common predecessors of BB and Succ
105   BlockSet CommonPreds;
106   for (pred_iterator PI = pred_begin(Succ), PE = pred_end(Succ);
107         PI != PE; ++PI)
108     if (BBPreds.count(*PI))
109       CommonPreds.insert(*PI);
110 
111   // Shortcut, if there are no common predecessors, merging is always safe
112   if (CommonPreds.empty())
113     return true;
114 
115   // Look at all the phi nodes in Succ, to see if they present a conflict when
116   // merging these blocks
117   for (BasicBlock::iterator I = Succ->begin(); isa<PHINode>(I); ++I) {
118     PHINode *PN = cast<PHINode>(I);
119 
120     // If the incoming value from BB is again a PHINode in
121     // BB which has the same incoming value for *PI as PN does, we can
122     // merge the phi nodes and then the blocks can still be merged
123     PHINode *BBPN = dyn_cast<PHINode>(PN->getIncomingValueForBlock(BB));
124     if (BBPN && BBPN->getParent() == BB) {
125       for (BlockSet::iterator PI = CommonPreds.begin(), PE = CommonPreds.end();
126             PI != PE; PI++) {
127         if (BBPN->getIncomingValueForBlock(*PI)
128               != PN->getIncomingValueForBlock(*PI)) {
129           DOUT << "Can't fold, phi node " << *PN->getNameStart() << " in "
130                << Succ->getNameStart() << " is conflicting with "
131                << BBPN->getNameStart() << " with regard to common predecessor "
132                << (*PI)->getNameStart() << "\n";
133           return false;
134         }
135       }
136       // Remove this phinode from the list of phis in BB, since it has been
137       // handled.
138       BBPHIs.erase(BBPN);
139     } else {
140       Value* Val = PN->getIncomingValueForBlock(BB);
141       for (BlockSet::iterator PI = CommonPreds.begin(), PE = CommonPreds.end();
142             PI != PE; PI++) {
143         // See if the incoming value for the common predecessor is equal to the
144         // one for BB, in which case this phi node will not prevent the merging
145         // of the block.
146         if (Val != PN->getIncomingValueForBlock(*PI)) {
147           DOUT << "Can't fold, phi node " << *PN->getNameStart() << " in "
148           << Succ->getNameStart() << " is conflicting with regard to common "
149           << "predecessor " << (*PI)->getNameStart() << "\n";
150           return false;
151         }
152       }
153     }
154   }
155 
156   // If there are any other phi nodes in BB that don't have a phi node in Succ
157   // to merge with, they must be moved to Succ completely. However, for any
158   // predecessors of Succ, branches will be added to the phi node that just
159   // point to itself. So, for any common predecessors, this must not cause
160   // conflicts.
161   for (InstrSet::iterator I = BBPHIs.begin(), E = BBPHIs.end();
162         I != E; I++) {
163     PHINode *PN = cast<PHINode>(*I);
164     for (BlockSet::iterator PI = CommonPreds.begin(), PE = CommonPreds.end();
165           PI != PE; PI++)
166       if (PN->getIncomingValueForBlock(*PI) != PN) {
167         DOUT << "Can't fold, phi node " << *PN->getNameStart() << " in "
168              << BB->getNameStart() << " is conflicting with regard to common "
169              << "predecessor " << (*PI)->getNameStart() << "\n";
170         return false;
171       }
172   }
173 
174   return true;
175 }
176 
177 /// TryToSimplifyUncondBranchFromEmptyBlock - BB contains an unconditional
178 /// branch to Succ, and contains no instructions other than PHI nodes and the
179 /// branch.  If possible, eliminate BB.
180 static bool TryToSimplifyUncondBranchFromEmptyBlock(BasicBlock *BB,
181                                                     BasicBlock *Succ) {
182   // Check to see if merging these blocks would cause conflicts for any of the
183   // phi nodes in BB or Succ. If not, we can safely merge.
184   if (!CanPropagatePredecessorsForPHIs(BB, Succ)) return false;
185 
186   DOUT << "Killing Trivial BB: \n" << *BB;
187 
188   if (isa<PHINode>(Succ->begin())) {
189     // If there is more than one pred of succ, and there are PHI nodes in
190     // the successor, then we need to add incoming edges for the PHI nodes
191     //
192     const SmallVector<BasicBlock*, 16> BBPreds(pred_begin(BB), pred_end(BB));
193 
194     // Loop over all of the PHI nodes in the successor of BB.
195     for (BasicBlock::iterator I = Succ->begin(); isa<PHINode>(I); ++I) {
196       PHINode *PN = cast<PHINode>(I);
197       Value *OldVal = PN->removeIncomingValue(BB, false);
198       assert(OldVal && "No entry in PHI for Pred BB!");
199 
200       // If this incoming value is one of the PHI nodes in BB, the new entries
201       // in the PHI node are the entries from the old PHI.
202       if (isa<PHINode>(OldVal) && cast<PHINode>(OldVal)->getParent() == BB) {
203         PHINode *OldValPN = cast<PHINode>(OldVal);
204         for (unsigned i = 0, e = OldValPN->getNumIncomingValues(); i != e; ++i)
205           // Note that, since we are merging phi nodes and BB and Succ might
206           // have common predecessors, we could end up with a phi node with
207           // identical incoming branches. This will be cleaned up later (and
208           // will trigger asserts if we try to clean it up now, without also
209           // simplifying the corresponding conditional branch).
210           PN->addIncoming(OldValPN->getIncomingValue(i),
211                           OldValPN->getIncomingBlock(i));
212       } else {
213         // Add an incoming value for each of the new incoming values.
214         for (unsigned i = 0, e = BBPreds.size(); i != e; ++i)
215           PN->addIncoming(OldVal, BBPreds[i]);
216       }
217     }
218   }
219 
220   if (isa<PHINode>(&BB->front())) {
221     SmallVector<BasicBlock*, 16>
222     OldSuccPreds(pred_begin(Succ), pred_end(Succ));
223 
224     // Move all PHI nodes in BB to Succ if they are alive, otherwise
225     // delete them.
226     while (PHINode *PN = dyn_cast<PHINode>(&BB->front())) {
227       if (PN->use_empty()) {
228         // Just remove the dead phi.  This happens if Succ's PHIs were the only
229         // users of the PHI nodes.
230         PN->eraseFromParent();
231         continue;
232       }
233 
234       // The instruction is alive, so this means that BB must dominate all
235       // predecessors of Succ (Since all uses of the PN are after its
236       // definition, so in Succ or a block dominated by Succ. If a predecessor
237       // of Succ would not be dominated by BB, PN would violate the def before
238       // use SSA demand). Therefore, we can simply move the phi node to the
239       // next block.
240       Succ->getInstList().splice(Succ->begin(),
241                                  BB->getInstList(), BB->begin());
242 
243       // We need to add new entries for the PHI node to account for
244       // predecessors of Succ that the PHI node does not take into
245       // account.  At this point, since we know that BB dominated succ and all
246       // of its predecessors, this means that we should any newly added
247       // incoming edges should use the PHI node itself as the value for these
248       // edges, because they are loop back edges.
249       for (unsigned i = 0, e = OldSuccPreds.size(); i != e; ++i)
250         if (OldSuccPreds[i] != BB)
251           PN->addIncoming(PN, OldSuccPreds[i]);
252     }
253   }
254 
255   // Everything that jumped to BB now goes to Succ.
256   BB->replaceAllUsesWith(Succ);
257   if (!Succ->hasName()) Succ->takeName(BB);
258   BB->eraseFromParent();              // Delete the old basic block.
259   return true;
260 }
261 
262 /// GetIfCondition - Given a basic block (BB) with two predecessors (and
263 /// presumably PHI nodes in it), check to see if the merge at this block is due
264 /// to an "if condition".  If so, return the boolean condition that determines
265 /// which entry into BB will be taken.  Also, return by references the block
266 /// that will be entered from if the condition is true, and the block that will
267 /// be entered if the condition is false.
268 ///
269 ///
270 static Value *GetIfCondition(BasicBlock *BB,
271                              BasicBlock *&IfTrue, BasicBlock *&IfFalse) {
272   assert(std::distance(pred_begin(BB), pred_end(BB)) == 2 &&
273          "Function can only handle blocks with 2 predecessors!");
274   BasicBlock *Pred1 = *pred_begin(BB);
275   BasicBlock *Pred2 = *++pred_begin(BB);
276 
277   // We can only handle branches.  Other control flow will be lowered to
278   // branches if possible anyway.
279   if (!isa<BranchInst>(Pred1->getTerminator()) ||
280       !isa<BranchInst>(Pred2->getTerminator()))
281     return 0;
282   BranchInst *Pred1Br = cast<BranchInst>(Pred1->getTerminator());
283   BranchInst *Pred2Br = cast<BranchInst>(Pred2->getTerminator());
284 
285   // Eliminate code duplication by ensuring that Pred1Br is conditional if
286   // either are.
287   if (Pred2Br->isConditional()) {
288     // If both branches are conditional, we don't have an "if statement".  In
289     // reality, we could transform this case, but since the condition will be
290     // required anyway, we stand no chance of eliminating it, so the xform is
291     // probably not profitable.
292     if (Pred1Br->isConditional())
293       return 0;
294 
295     std::swap(Pred1, Pred2);
296     std::swap(Pred1Br, Pred2Br);
297   }
298 
299   if (Pred1Br->isConditional()) {
300     // If we found a conditional branch predecessor, make sure that it branches
301     // to BB and Pred2Br.  If it doesn't, this isn't an "if statement".
302     if (Pred1Br->getSuccessor(0) == BB &&
303         Pred1Br->getSuccessor(1) == Pred2) {
304       IfTrue = Pred1;
305       IfFalse = Pred2;
306     } else if (Pred1Br->getSuccessor(0) == Pred2 &&
307                Pred1Br->getSuccessor(1) == BB) {
308       IfTrue = Pred2;
309       IfFalse = Pred1;
310     } else {
311       // We know that one arm of the conditional goes to BB, so the other must
312       // go somewhere unrelated, and this must not be an "if statement".
313       return 0;
314     }
315 
316     // The only thing we have to watch out for here is to make sure that Pred2
317     // doesn't have incoming edges from other blocks.  If it does, the condition
318     // doesn't dominate BB.
319     if (++pred_begin(Pred2) != pred_end(Pred2))
320       return 0;
321 
322     return Pred1Br->getCondition();
323   }
324 
325   // Ok, if we got here, both predecessors end with an unconditional branch to
326   // BB.  Don't panic!  If both blocks only have a single (identical)
327   // predecessor, and THAT is a conditional branch, then we're all ok!
328   if (pred_begin(Pred1) == pred_end(Pred1) ||
329       ++pred_begin(Pred1) != pred_end(Pred1) ||
330       pred_begin(Pred2) == pred_end(Pred2) ||
331       ++pred_begin(Pred2) != pred_end(Pred2) ||
332       *pred_begin(Pred1) != *pred_begin(Pred2))
333     return 0;
334 
335   // Otherwise, if this is a conditional branch, then we can use it!
336   BasicBlock *CommonPred = *pred_begin(Pred1);
337   if (BranchInst *BI = dyn_cast<BranchInst>(CommonPred->getTerminator())) {
338     assert(BI->isConditional() && "Two successors but not conditional?");
339     if (BI->getSuccessor(0) == Pred1) {
340       IfTrue = Pred1;
341       IfFalse = Pred2;
342     } else {
343       IfTrue = Pred2;
344       IfFalse = Pred1;
345     }
346     return BI->getCondition();
347   }
348   return 0;
349 }
350 
351 /// DominatesMergePoint - If we have a merge point of an "if condition" as
352 /// accepted above, return true if the specified value dominates the block.  We
353 /// don't handle the true generality of domination here, just a special case
354 /// which works well enough for us.
355 ///
356 /// If AggressiveInsts is non-null, and if V does not dominate BB, we check to
357 /// see if V (which must be an instruction) is cheap to compute and is
358 /// non-trapping.  If both are true, the instruction is inserted into the set
359 /// and true is returned.
360 static bool DominatesMergePoint(Value *V, BasicBlock *BB,
361                                 std::set<Instruction*> *AggressiveInsts) {
362   Instruction *I = dyn_cast<Instruction>(V);
363   if (!I) {
364     // Non-instructions all dominate instructions, but not all constantexprs
365     // can be executed unconditionally.
366     if (ConstantExpr *C = dyn_cast<ConstantExpr>(V))
367       if (C->canTrap())
368         return false;
369     return true;
370   }
371   BasicBlock *PBB = I->getParent();
372 
373   // We don't want to allow weird loops that might have the "if condition" in
374   // the bottom of this block.
375   if (PBB == BB) return false;
376 
377   // If this instruction is defined in a block that contains an unconditional
378   // branch to BB, then it must be in the 'conditional' part of the "if
379   // statement".
380   if (BranchInst *BI = dyn_cast<BranchInst>(PBB->getTerminator()))
381     if (BI->isUnconditional() && BI->getSuccessor(0) == BB) {
382       if (!AggressiveInsts) return false;
383       // Okay, it looks like the instruction IS in the "condition".  Check to
384       // see if its a cheap instruction to unconditionally compute, and if it
385       // only uses stuff defined outside of the condition.  If so, hoist it out.
386       switch (I->getOpcode()) {
387       default: return false;  // Cannot hoist this out safely.
388       case Instruction::Load: {
389         // We can hoist loads that are non-volatile and obviously cannot trap.
390         if (cast<LoadInst>(I)->isVolatile())
391           return false;
392         // FIXME: A computation of a constant can trap!
393         if (!isa<AllocaInst>(I->getOperand(0)) &&
394             !isa<Constant>(I->getOperand(0)))
395           return false;
396         // External weak globals may have address 0, so we can't load them.
397         Value *V2 = I->getOperand(0)->getUnderlyingObject();
398         if (V2) {
399           GlobalVariable* GV = dyn_cast<GlobalVariable>(V2);
400           if (GV && GV->hasExternalWeakLinkage())
401             return false;
402         }
403         // Finally, we have to check to make sure there are no instructions
404         // before the load in its basic block, as we are going to hoist the loop
405         // out to its predecessor.
406         BasicBlock::iterator IP = PBB->begin();
407         while (isa<DbgInfoIntrinsic>(IP))
408           IP++;
409         if (IP != BasicBlock::iterator(I))
410           return false;
411         break;
412       }
413       case Instruction::Add:
414       case Instruction::Sub:
415       case Instruction::And:
416       case Instruction::Or:
417       case Instruction::Xor:
418       case Instruction::Shl:
419       case Instruction::LShr:
420       case Instruction::AShr:
421       case Instruction::ICmp:
422         break;   // These are all cheap and non-trapping instructions.
423       }
424 
425       // Okay, we can only really hoist these out if their operands are not
426       // defined in the conditional region.
427       for (User::op_iterator i = I->op_begin(), e = I->op_end(); i != e; ++i)
428         if (!DominatesMergePoint(*i, BB, 0))
429           return false;
430       // Okay, it's safe to do this!  Remember this instruction.
431       AggressiveInsts->insert(I);
432     }
433 
434   return true;
435 }
436 
437 /// GatherConstantSetEQs - Given a potentially 'or'd together collection of
438 /// icmp_eq instructions that compare a value against a constant, return the
439 /// value being compared, and stick the constant into the Values vector.
440 static Value *GatherConstantSetEQs(Value *V, std::vector<ConstantInt*> &Values){
441   if (Instruction *Inst = dyn_cast<Instruction>(V)) {
442     if (Inst->getOpcode() == Instruction::ICmp &&
443         cast<ICmpInst>(Inst)->getPredicate() == ICmpInst::ICMP_EQ) {
444       if (ConstantInt *C = dyn_cast<ConstantInt>(Inst->getOperand(1))) {
445         Values.push_back(C);
446         return Inst->getOperand(0);
447       } else if (ConstantInt *C = dyn_cast<ConstantInt>(Inst->getOperand(0))) {
448         Values.push_back(C);
449         return Inst->getOperand(1);
450       }
451     } else if (Inst->getOpcode() == Instruction::Or) {
452       if (Value *LHS = GatherConstantSetEQs(Inst->getOperand(0), Values))
453         if (Value *RHS = GatherConstantSetEQs(Inst->getOperand(1), Values))
454           if (LHS == RHS)
455             return LHS;
456     }
457   }
458   return 0;
459 }
460 
461 /// GatherConstantSetNEs - Given a potentially 'and'd together collection of
462 /// setne instructions that compare a value against a constant, return the value
463 /// being compared, and stick the constant into the Values vector.
464 static Value *GatherConstantSetNEs(Value *V, std::vector<ConstantInt*> &Values){
465   if (Instruction *Inst = dyn_cast<Instruction>(V)) {
466     if (Inst->getOpcode() == Instruction::ICmp &&
467                cast<ICmpInst>(Inst)->getPredicate() == ICmpInst::ICMP_NE) {
468       if (ConstantInt *C = dyn_cast<ConstantInt>(Inst->getOperand(1))) {
469         Values.push_back(C);
470         return Inst->getOperand(0);
471       } else if (ConstantInt *C = dyn_cast<ConstantInt>(Inst->getOperand(0))) {
472         Values.push_back(C);
473         return Inst->getOperand(1);
474       }
475     } else if (Inst->getOpcode() == Instruction::And) {
476       if (Value *LHS = GatherConstantSetNEs(Inst->getOperand(0), Values))
477         if (Value *RHS = GatherConstantSetNEs(Inst->getOperand(1), Values))
478           if (LHS == RHS)
479             return LHS;
480     }
481   }
482   return 0;
483 }
484 
485 /// GatherValueComparisons - If the specified Cond is an 'and' or 'or' of a
486 /// bunch of comparisons of one value against constants, return the value and
487 /// the constants being compared.
488 static bool GatherValueComparisons(Instruction *Cond, Value *&CompVal,
489                                    std::vector<ConstantInt*> &Values) {
490   if (Cond->getOpcode() == Instruction::Or) {
491     CompVal = GatherConstantSetEQs(Cond, Values);
492 
493     // Return true to indicate that the condition is true if the CompVal is
494     // equal to one of the constants.
495     return true;
496   } else if (Cond->getOpcode() == Instruction::And) {
497     CompVal = GatherConstantSetNEs(Cond, Values);
498 
499     // Return false to indicate that the condition is false if the CompVal is
500     // equal to one of the constants.
501     return false;
502   }
503   return false;
504 }
505 
506 static void EraseTerminatorInstAndDCECond(TerminatorInst *TI) {
507   Instruction* Cond = 0;
508   if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) {
509     Cond = dyn_cast<Instruction>(SI->getCondition());
510   } else if (BranchInst *BI = dyn_cast<BranchInst>(TI)) {
511     if (BI->isConditional())
512       Cond = dyn_cast<Instruction>(BI->getCondition());
513   }
514 
515   TI->eraseFromParent();
516   if (Cond) RecursivelyDeleteTriviallyDeadInstructions(Cond);
517 }
518 
519 /// isValueEqualityComparison - Return true if the specified terminator checks
520 /// to see if a value is equal to constant integer value.
521 static Value *isValueEqualityComparison(TerminatorInst *TI) {
522   if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) {
523     // Do not permit merging of large switch instructions into their
524     // predecessors unless there is only one predecessor.
525     if (SI->getNumSuccessors() * std::distance(pred_begin(SI->getParent()),
526                                                pred_end(SI->getParent())) > 128)
527       return 0;
528 
529     return SI->getCondition();
530   }
531   if (BranchInst *BI = dyn_cast<BranchInst>(TI))
532     if (BI->isConditional() && BI->getCondition()->hasOneUse())
533       if (ICmpInst *ICI = dyn_cast<ICmpInst>(BI->getCondition()))
534         if ((ICI->getPredicate() == ICmpInst::ICMP_EQ ||
535              ICI->getPredicate() == ICmpInst::ICMP_NE) &&
536             isa<ConstantInt>(ICI->getOperand(1)))
537           return ICI->getOperand(0);
538   return 0;
539 }
540 
541 /// GetValueEqualityComparisonCases - Given a value comparison instruction,
542 /// decode all of the 'cases' that it represents and return the 'default' block.
543 static BasicBlock *
544 GetValueEqualityComparisonCases(TerminatorInst *TI,
545                                 std::vector<std::pair<ConstantInt*,
546                                                       BasicBlock*> > &Cases) {
547   if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) {
548     Cases.reserve(SI->getNumCases());
549     for (unsigned i = 1, e = SI->getNumCases(); i != e; ++i)
550       Cases.push_back(std::make_pair(SI->getCaseValue(i), SI->getSuccessor(i)));
551     return SI->getDefaultDest();
552   }
553 
554   BranchInst *BI = cast<BranchInst>(TI);
555   ICmpInst *ICI = cast<ICmpInst>(BI->getCondition());
556   Cases.push_back(std::make_pair(cast<ConstantInt>(ICI->getOperand(1)),
557                                  BI->getSuccessor(ICI->getPredicate() ==
558                                                   ICmpInst::ICMP_NE)));
559   return BI->getSuccessor(ICI->getPredicate() == ICmpInst::ICMP_EQ);
560 }
561 
562 
563 /// EliminateBlockCases - Given a vector of bb/value pairs, remove any entries
564 /// in the list that match the specified block.
565 static void EliminateBlockCases(BasicBlock *BB,
566                std::vector<std::pair<ConstantInt*, BasicBlock*> > &Cases) {
567   for (unsigned i = 0, e = Cases.size(); i != e; ++i)
568     if (Cases[i].second == BB) {
569       Cases.erase(Cases.begin()+i);
570       --i; --e;
571     }
572 }
573 
574 /// ValuesOverlap - Return true if there are any keys in C1 that exist in C2 as
575 /// well.
576 static bool
577 ValuesOverlap(std::vector<std::pair<ConstantInt*, BasicBlock*> > &C1,
578               std::vector<std::pair<ConstantInt*, BasicBlock*> > &C2) {
579   std::vector<std::pair<ConstantInt*, BasicBlock*> > *V1 = &C1, *V2 = &C2;
580 
581   // Make V1 be smaller than V2.
582   if (V1->size() > V2->size())
583     std::swap(V1, V2);
584 
585   if (V1->size() == 0) return false;
586   if (V1->size() == 1) {
587     // Just scan V2.
588     ConstantInt *TheVal = (*V1)[0].first;
589     for (unsigned i = 0, e = V2->size(); i != e; ++i)
590       if (TheVal == (*V2)[i].first)
591         return true;
592   }
593 
594   // Otherwise, just sort both lists and compare element by element.
595   std::sort(V1->begin(), V1->end());
596   std::sort(V2->begin(), V2->end());
597   unsigned i1 = 0, i2 = 0, e1 = V1->size(), e2 = V2->size();
598   while (i1 != e1 && i2 != e2) {
599     if ((*V1)[i1].first == (*V2)[i2].first)
600       return true;
601     if ((*V1)[i1].first < (*V2)[i2].first)
602       ++i1;
603     else
604       ++i2;
605   }
606   return false;
607 }
608 
609 /// SimplifyEqualityComparisonWithOnlyPredecessor - If TI is known to be a
610 /// terminator instruction and its block is known to only have a single
611 /// predecessor block, check to see if that predecessor is also a value
612 /// comparison with the same value, and if that comparison determines the
613 /// outcome of this comparison.  If so, simplify TI.  This does a very limited
614 /// form of jump threading.
615 static bool SimplifyEqualityComparisonWithOnlyPredecessor(TerminatorInst *TI,
616                                                           BasicBlock *Pred) {
617   Value *PredVal = isValueEqualityComparison(Pred->getTerminator());
618   if (!PredVal) return false;  // Not a value comparison in predecessor.
619 
620   Value *ThisVal = isValueEqualityComparison(TI);
621   assert(ThisVal && "This isn't a value comparison!!");
622   if (ThisVal != PredVal) return false;  // Different predicates.
623 
624   // Find out information about when control will move from Pred to TI's block.
625   std::vector<std::pair<ConstantInt*, BasicBlock*> > PredCases;
626   BasicBlock *PredDef = GetValueEqualityComparisonCases(Pred->getTerminator(),
627                                                         PredCases);
628   EliminateBlockCases(PredDef, PredCases);  // Remove default from cases.
629 
630   // Find information about how control leaves this block.
631   std::vector<std::pair<ConstantInt*, BasicBlock*> > ThisCases;
632   BasicBlock *ThisDef = GetValueEqualityComparisonCases(TI, ThisCases);
633   EliminateBlockCases(ThisDef, ThisCases);  // Remove default from cases.
634 
635   // If TI's block is the default block from Pred's comparison, potentially
636   // simplify TI based on this knowledge.
637   if (PredDef == TI->getParent()) {
638     // If we are here, we know that the value is none of those cases listed in
639     // PredCases.  If there are any cases in ThisCases that are in PredCases, we
640     // can simplify TI.
641     if (ValuesOverlap(PredCases, ThisCases)) {
642       if (isa<BranchInst>(TI)) {
643         // Okay, one of the successors of this condbr is dead.  Convert it to a
644         // uncond br.
645         assert(ThisCases.size() == 1 && "Branch can only have one case!");
646         // Insert the new branch.
647         Instruction *NI = BranchInst::Create(ThisDef, TI);
648 
649         // Remove PHI node entries for the dead edge.
650         ThisCases[0].second->removePredecessor(TI->getParent());
651 
652         DOUT << "Threading pred instr: " << *Pred->getTerminator()
653              << "Through successor TI: " << *TI << "Leaving: " << *NI << "\n";
654 
655         EraseTerminatorInstAndDCECond(TI);
656         return true;
657 
658       } else {
659         SwitchInst *SI = cast<SwitchInst>(TI);
660         // Okay, TI has cases that are statically dead, prune them away.
661         SmallPtrSet<Constant*, 16> DeadCases;
662         for (unsigned i = 0, e = PredCases.size(); i != e; ++i)
663           DeadCases.insert(PredCases[i].first);
664 
665         DOUT << "Threading pred instr: " << *Pred->getTerminator()
666              << "Through successor TI: " << *TI;
667 
668         for (unsigned i = SI->getNumCases()-1; i != 0; --i)
669           if (DeadCases.count(SI->getCaseValue(i))) {
670             SI->getSuccessor(i)->removePredecessor(TI->getParent());
671             SI->removeCase(i);
672           }
673 
674         DOUT << "Leaving: " << *TI << "\n";
675         return true;
676       }
677     }
678 
679   } else {
680     // Otherwise, TI's block must correspond to some matched value.  Find out
681     // which value (or set of values) this is.
682     ConstantInt *TIV = 0;
683     BasicBlock *TIBB = TI->getParent();
684     for (unsigned i = 0, e = PredCases.size(); i != e; ++i)
685       if (PredCases[i].second == TIBB) {
686         if (TIV == 0)
687           TIV = PredCases[i].first;
688         else
689           return false;  // Cannot handle multiple values coming to this block.
690       }
691     assert(TIV && "No edge from pred to succ?");
692 
693     // Okay, we found the one constant that our value can be if we get into TI's
694     // BB.  Find out which successor will unconditionally be branched to.
695     BasicBlock *TheRealDest = 0;
696     for (unsigned i = 0, e = ThisCases.size(); i != e; ++i)
697       if (ThisCases[i].first == TIV) {
698         TheRealDest = ThisCases[i].second;
699         break;
700       }
701 
702     // If not handled by any explicit cases, it is handled by the default case.
703     if (TheRealDest == 0) TheRealDest = ThisDef;
704 
705     // Remove PHI node entries for dead edges.
706     BasicBlock *CheckEdge = TheRealDest;
707     for (succ_iterator SI = succ_begin(TIBB), e = succ_end(TIBB); SI != e; ++SI)
708       if (*SI != CheckEdge)
709         (*SI)->removePredecessor(TIBB);
710       else
711         CheckEdge = 0;
712 
713     // Insert the new branch.
714     Instruction *NI = BranchInst::Create(TheRealDest, TI);
715 
716     DOUT << "Threading pred instr: " << *Pred->getTerminator()
717          << "Through successor TI: " << *TI << "Leaving: " << *NI << "\n";
718 
719     EraseTerminatorInstAndDCECond(TI);
720     return true;
721   }
722   return false;
723 }
724 
725 namespace {
726   /// ConstantIntOrdering - This class implements a stable ordering of constant
727   /// integers that does not depend on their address.  This is important for
728   /// applications that sort ConstantInt's to ensure uniqueness.
729   struct ConstantIntOrdering {
730     bool operator()(const ConstantInt *LHS, const ConstantInt *RHS) const {
731       return LHS->getValue().ult(RHS->getValue());
732     }
733   };
734 }
735 
736 /// FoldValueComparisonIntoPredecessors - The specified terminator is a value
737 /// equality comparison instruction (either a switch or a branch on "X == c").
738 /// See if any of the predecessors of the terminator block are value comparisons
739 /// on the same value.  If so, and if safe to do so, fold them together.
740 static bool FoldValueComparisonIntoPredecessors(TerminatorInst *TI) {
741   BasicBlock *BB = TI->getParent();
742   Value *CV = isValueEqualityComparison(TI);  // CondVal
743   assert(CV && "Not a comparison?");
744   bool Changed = false;
745 
746   SmallVector<BasicBlock*, 16> Preds(pred_begin(BB), pred_end(BB));
747   while (!Preds.empty()) {
748     BasicBlock *Pred = Preds.pop_back_val();
749 
750     // See if the predecessor is a comparison with the same value.
751     TerminatorInst *PTI = Pred->getTerminator();
752     Value *PCV = isValueEqualityComparison(PTI);  // PredCondVal
753 
754     if (PCV == CV && SafeToMergeTerminators(TI, PTI)) {
755       // Figure out which 'cases' to copy from SI to PSI.
756       std::vector<std::pair<ConstantInt*, BasicBlock*> > BBCases;
757       BasicBlock *BBDefault = GetValueEqualityComparisonCases(TI, BBCases);
758 
759       std::vector<std::pair<ConstantInt*, BasicBlock*> > PredCases;
760       BasicBlock *PredDefault = GetValueEqualityComparisonCases(PTI, PredCases);
761 
762       // Based on whether the default edge from PTI goes to BB or not, fill in
763       // PredCases and PredDefault with the new switch cases we would like to
764       // build.
765       SmallVector<BasicBlock*, 8> NewSuccessors;
766 
767       if (PredDefault == BB) {
768         // If this is the default destination from PTI, only the edges in TI
769         // that don't occur in PTI, or that branch to BB will be activated.
770         std::set<ConstantInt*, ConstantIntOrdering> PTIHandled;
771         for (unsigned i = 0, e = PredCases.size(); i != e; ++i)
772           if (PredCases[i].second != BB)
773             PTIHandled.insert(PredCases[i].first);
774           else {
775             // The default destination is BB, we don't need explicit targets.
776             std::swap(PredCases[i], PredCases.back());
777             PredCases.pop_back();
778             --i; --e;
779           }
780 
781         // Reconstruct the new switch statement we will be building.
782         if (PredDefault != BBDefault) {
783           PredDefault->removePredecessor(Pred);
784           PredDefault = BBDefault;
785           NewSuccessors.push_back(BBDefault);
786         }
787         for (unsigned i = 0, e = BBCases.size(); i != e; ++i)
788           if (!PTIHandled.count(BBCases[i].first) &&
789               BBCases[i].second != BBDefault) {
790             PredCases.push_back(BBCases[i]);
791             NewSuccessors.push_back(BBCases[i].second);
792           }
793 
794       } else {
795         // If this is not the default destination from PSI, only the edges
796         // in SI that occur in PSI with a destination of BB will be
797         // activated.
798         std::set<ConstantInt*, ConstantIntOrdering> PTIHandled;
799         for (unsigned i = 0, e = PredCases.size(); i != e; ++i)
800           if (PredCases[i].second == BB) {
801             PTIHandled.insert(PredCases[i].first);
802             std::swap(PredCases[i], PredCases.back());
803             PredCases.pop_back();
804             --i; --e;
805           }
806 
807         // Okay, now we know which constants were sent to BB from the
808         // predecessor.  Figure out where they will all go now.
809         for (unsigned i = 0, e = BBCases.size(); i != e; ++i)
810           if (PTIHandled.count(BBCases[i].first)) {
811             // If this is one we are capable of getting...
812             PredCases.push_back(BBCases[i]);
813             NewSuccessors.push_back(BBCases[i].second);
814             PTIHandled.erase(BBCases[i].first);// This constant is taken care of
815           }
816 
817         // If there are any constants vectored to BB that TI doesn't handle,
818         // they must go to the default destination of TI.
819         for (std::set<ConstantInt*, ConstantIntOrdering>::iterator I =
820                                     PTIHandled.begin(),
821                E = PTIHandled.end(); I != E; ++I) {
822           PredCases.push_back(std::make_pair(*I, BBDefault));
823           NewSuccessors.push_back(BBDefault);
824         }
825       }
826 
827       // Okay, at this point, we know which new successor Pred will get.  Make
828       // sure we update the number of entries in the PHI nodes for these
829       // successors.
830       for (unsigned i = 0, e = NewSuccessors.size(); i != e; ++i)
831         AddPredecessorToBlock(NewSuccessors[i], Pred, BB);
832 
833       // Now that the successors are updated, create the new Switch instruction.
834       SwitchInst *NewSI = SwitchInst::Create(CV, PredDefault,
835                                              PredCases.size(), PTI);
836       for (unsigned i = 0, e = PredCases.size(); i != e; ++i)
837         NewSI->addCase(PredCases[i].first, PredCases[i].second);
838 
839       EraseTerminatorInstAndDCECond(PTI);
840 
841       // Okay, last check.  If BB is still a successor of PSI, then we must
842       // have an infinite loop case.  If so, add an infinitely looping block
843       // to handle the case to preserve the behavior of the code.
844       BasicBlock *InfLoopBlock = 0;
845       for (unsigned i = 0, e = NewSI->getNumSuccessors(); i != e; ++i)
846         if (NewSI->getSuccessor(i) == BB) {
847           if (InfLoopBlock == 0) {
848             // Insert it at the end of the function, because it's either code,
849             // or it won't matter if it's hot. :)
850             InfLoopBlock = BasicBlock::Create("infloop", BB->getParent());
851             BranchInst::Create(InfLoopBlock, InfLoopBlock);
852           }
853           NewSI->setSuccessor(i, InfLoopBlock);
854         }
855 
856       Changed = true;
857     }
858   }
859   return Changed;
860 }
861 
862 /// HoistThenElseCodeToIf - Given a conditional branch that goes to BB1 and
863 /// BB2, hoist any common code in the two blocks up into the branch block.  The
864 /// caller of this function guarantees that BI's block dominates BB1 and BB2.
865 static bool HoistThenElseCodeToIf(BranchInst *BI) {
866   // This does very trivial matching, with limited scanning, to find identical
867   // instructions in the two blocks.  In particular, we don't want to get into
868   // O(M*N) situations here where M and N are the sizes of BB1 and BB2.  As
869   // such, we currently just scan for obviously identical instructions in an
870   // identical order.
871   BasicBlock *BB1 = BI->getSuccessor(0);  // The true destination.
872   BasicBlock *BB2 = BI->getSuccessor(1);  // The false destination
873 
874   BasicBlock::iterator BB1_Itr = BB1->begin();
875   BasicBlock::iterator BB2_Itr = BB2->begin();
876 
877   Instruction *I1 = BB1_Itr++, *I2 = BB2_Itr++;
878   while (isa<DbgInfoIntrinsic>(I1))
879     I1 = BB1_Itr++;
880   while (isa<DbgInfoIntrinsic>(I2))
881     I2 = BB2_Itr++;
882   if (I1->getOpcode() != I2->getOpcode() || isa<PHINode>(I1) ||
883       isa<InvokeInst>(I1) || !I1->isIdenticalTo(I2))
884     return false;
885 
886   // If we get here, we can hoist at least one instruction.
887   BasicBlock *BIParent = BI->getParent();
888 
889   do {
890     // If we are hoisting the terminator instruction, don't move one (making a
891     // broken BB), instead clone it, and remove BI.
892     if (isa<TerminatorInst>(I1))
893       goto HoistTerminator;
894 
895     // For a normal instruction, we just move one to right before the branch,
896     // then replace all uses of the other with the first.  Finally, we remove
897     // the now redundant second instruction.
898     BIParent->getInstList().splice(BI, BB1->getInstList(), I1);
899     if (!I2->use_empty())
900       I2->replaceAllUsesWith(I1);
901     BB2->getInstList().erase(I2);
902 
903     I1 = BB1_Itr++;
904     while (isa<DbgInfoIntrinsic>(I1))
905       I1 = BB1_Itr++;
906     I2 = BB2_Itr++;
907     while (isa<DbgInfoIntrinsic>(I2))
908       I2 = BB2_Itr++;
909   } while (I1->getOpcode() == I2->getOpcode() && I1->isIdenticalTo(I2));
910 
911   return true;
912 
913 HoistTerminator:
914   // Okay, it is safe to hoist the terminator.
915   Instruction *NT = I1->clone();
916   BIParent->getInstList().insert(BI, NT);
917   if (NT->getType() != Type::VoidTy) {
918     I1->replaceAllUsesWith(NT);
919     I2->replaceAllUsesWith(NT);
920     NT->takeName(I1);
921   }
922 
923   // Hoisting one of the terminators from our successor is a great thing.
924   // Unfortunately, the successors of the if/else blocks may have PHI nodes in
925   // them.  If they do, all PHI entries for BB1/BB2 must agree for all PHI
926   // nodes, so we insert select instruction to compute the final result.
927   std::map<std::pair<Value*,Value*>, SelectInst*> InsertedSelects;
928   for (succ_iterator SI = succ_begin(BB1), E = succ_end(BB1); SI != E; ++SI) {
929     PHINode *PN;
930     for (BasicBlock::iterator BBI = SI->begin();
931          (PN = dyn_cast<PHINode>(BBI)); ++BBI) {
932       Value *BB1V = PN->getIncomingValueForBlock(BB1);
933       Value *BB2V = PN->getIncomingValueForBlock(BB2);
934       if (BB1V != BB2V) {
935         // These values do not agree.  Insert a select instruction before NT
936         // that determines the right value.
937         SelectInst *&SI = InsertedSelects[std::make_pair(BB1V, BB2V)];
938         if (SI == 0)
939           SI = SelectInst::Create(BI->getCondition(), BB1V, BB2V,
940                                   BB1V->getName()+"."+BB2V->getName(), NT);
941         // Make the PHI node use the select for all incoming values for BB1/BB2
942         for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i)
943           if (PN->getIncomingBlock(i) == BB1 || PN->getIncomingBlock(i) == BB2)
944             PN->setIncomingValue(i, SI);
945       }
946     }
947   }
948 
949   // Update any PHI nodes in our new successors.
950   for (succ_iterator SI = succ_begin(BB1), E = succ_end(BB1); SI != E; ++SI)
951     AddPredecessorToBlock(*SI, BIParent, BB1);
952 
953   EraseTerminatorInstAndDCECond(BI);
954   return true;
955 }
956 
957 /// SpeculativelyExecuteBB - Given a conditional branch that goes to BB1
958 /// and an BB2 and the only successor of BB1 is BB2, hoist simple code
959 /// (for now, restricted to a single instruction that's side effect free) from
960 /// the BB1 into the branch block to speculatively execute it.
961 static bool SpeculativelyExecuteBB(BranchInst *BI, BasicBlock *BB1) {
962   // Only speculatively execution a single instruction (not counting the
963   // terminator) for now.
964   Instruction *HInst = NULL;
965   Instruction *Term = BB1->getTerminator();
966   for (BasicBlock::iterator BBI = BB1->begin(), BBE = BB1->end();
967        BBI != BBE; ++BBI) {
968     Instruction *I = BBI;
969     // Skip debug info.
970     if (isa<DbgInfoIntrinsic>(I))   continue;
971     if (I == Term)  break;
972 
973     if (!HInst)
974       HInst = I;
975     else
976       return false;
977   }
978   if (!HInst)
979     return false;
980 
981   // Be conservative for now. FP select instruction can often be expensive.
982   Value *BrCond = BI->getCondition();
983   if (isa<Instruction>(BrCond) &&
984       cast<Instruction>(BrCond)->getOpcode() == Instruction::FCmp)
985     return false;
986 
987   // If BB1 is actually on the false edge of the conditional branch, remember
988   // to swap the select operands later.
989   bool Invert = false;
990   if (BB1 != BI->getSuccessor(0)) {
991     assert(BB1 == BI->getSuccessor(1) && "No edge from 'if' block?");
992     Invert = true;
993   }
994 
995   // Turn
996   // BB:
997   //     %t1 = icmp
998   //     br i1 %t1, label %BB1, label %BB2
999   // BB1:
1000   //     %t3 = add %t2, c
1001   //     br label BB2
1002   // BB2:
1003   // =>
1004   // BB:
1005   //     %t1 = icmp
1006   //     %t4 = add %t2, c
1007   //     %t3 = select i1 %t1, %t2, %t3
1008   switch (HInst->getOpcode()) {
1009   default: return false;  // Not safe / profitable to hoist.
1010   case Instruction::Add:
1011   case Instruction::Sub:
1012     // Not worth doing for vector ops.
1013     if (isa<VectorType>(HInst->getType()))
1014       return false;
1015     break;
1016   case Instruction::And:
1017   case Instruction::Or:
1018   case Instruction::Xor:
1019   case Instruction::Shl:
1020   case Instruction::LShr:
1021   case Instruction::AShr:
1022     // Don't mess with vector operations.
1023     if (isa<VectorType>(HInst->getType()))
1024       return false;
1025     break;   // These are all cheap and non-trapping instructions.
1026   }
1027 
1028   // If the instruction is obviously dead, don't try to predicate it.
1029   if (HInst->use_empty()) {
1030     HInst->eraseFromParent();
1031     return true;
1032   }
1033 
1034   // Can we speculatively execute the instruction? And what is the value
1035   // if the condition is false? Consider the phi uses, if the incoming value
1036   // from the "if" block are all the same V, then V is the value of the
1037   // select if the condition is false.
1038   BasicBlock *BIParent = BI->getParent();
1039   SmallVector<PHINode*, 4> PHIUses;
1040   Value *FalseV = NULL;
1041 
1042   BasicBlock *BB2 = BB1->getTerminator()->getSuccessor(0);
1043   for (Value::use_iterator UI = HInst->use_begin(), E = HInst->use_end();
1044        UI != E; ++UI) {
1045     // Ignore any user that is not a PHI node in BB2.  These can only occur in
1046     // unreachable blocks, because they would not be dominated by the instr.
1047     PHINode *PN = dyn_cast<PHINode>(UI);
1048     if (!PN || PN->getParent() != BB2)
1049       return false;
1050     PHIUses.push_back(PN);
1051 
1052     Value *PHIV = PN->getIncomingValueForBlock(BIParent);
1053     if (!FalseV)
1054       FalseV = PHIV;
1055     else if (FalseV != PHIV)
1056       return false;  // Inconsistent value when condition is false.
1057   }
1058 
1059   assert(FalseV && "Must have at least one user, and it must be a PHI");
1060 
1061   // Do not hoist the instruction if any of its operands are defined but not
1062   // used in this BB. The transformation will prevent the operand from
1063   // being sunk into the use block.
1064   for (User::op_iterator i = HInst->op_begin(), e = HInst->op_end();
1065        i != e; ++i) {
1066     Instruction *OpI = dyn_cast<Instruction>(*i);
1067     if (OpI && OpI->getParent() == BIParent &&
1068         !OpI->isUsedInBasicBlock(BIParent))
1069       return false;
1070   }
1071 
1072   // If we get here, we can hoist the instruction. Try to place it
1073   // before the icmp instruction preceding the conditional branch.
1074   BasicBlock::iterator InsertPos = BI;
1075   if (InsertPos != BIParent->begin())
1076     --InsertPos;
1077   // Skip debug info between condition and branch.
1078   while (InsertPos != BIParent->begin() && isa<DbgInfoIntrinsic>(InsertPos))
1079     --InsertPos;
1080   if (InsertPos == BrCond && !isa<PHINode>(BrCond)) {
1081     SmallPtrSet<Instruction *, 4> BB1Insns;
1082     for(BasicBlock::iterator BB1I = BB1->begin(), BB1E = BB1->end();
1083         BB1I != BB1E; ++BB1I)
1084       BB1Insns.insert(BB1I);
1085     for(Value::use_iterator UI = BrCond->use_begin(), UE = BrCond->use_end();
1086         UI != UE; ++UI) {
1087       Instruction *Use = cast<Instruction>(*UI);
1088       if (BB1Insns.count(Use)) {
1089         // If BrCond uses the instruction that place it just before
1090         // branch instruction.
1091         InsertPos = BI;
1092         break;
1093       }
1094     }
1095   } else
1096     InsertPos = BI;
1097   BIParent->getInstList().splice(InsertPos, BB1->getInstList(), HInst);
1098 
1099   // Create a select whose true value is the speculatively executed value and
1100   // false value is the previously determined FalseV.
1101   SelectInst *SI;
1102   if (Invert)
1103     SI = SelectInst::Create(BrCond, FalseV, HInst,
1104                             FalseV->getName() + "." + HInst->getName(), BI);
1105   else
1106     SI = SelectInst::Create(BrCond, HInst, FalseV,
1107                             HInst->getName() + "." + FalseV->getName(), BI);
1108 
1109   // Make the PHI node use the select for all incoming values for "then" and
1110   // "if" blocks.
1111   for (unsigned i = 0, e = PHIUses.size(); i != e; ++i) {
1112     PHINode *PN = PHIUses[i];
1113     for (unsigned j = 0, ee = PN->getNumIncomingValues(); j != ee; ++j)
1114       if (PN->getIncomingBlock(j) == BB1 ||
1115           PN->getIncomingBlock(j) == BIParent)
1116         PN->setIncomingValue(j, SI);
1117   }
1118 
1119   ++NumSpeculations;
1120   return true;
1121 }
1122 
1123 /// BlockIsSimpleEnoughToThreadThrough - Return true if we can thread a branch
1124 /// across this block.
1125 static bool BlockIsSimpleEnoughToThreadThrough(BasicBlock *BB) {
1126   BranchInst *BI = cast<BranchInst>(BB->getTerminator());
1127   unsigned Size = 0;
1128 
1129   for (BasicBlock::iterator BBI = BB->begin(); &*BBI != BI; ++BBI) {
1130     if (isa<DbgInfoIntrinsic>(BBI))
1131       continue;
1132     if (Size > 10) return false;  // Don't clone large BB's.
1133     ++Size;
1134 
1135     // We can only support instructions that do not define values that are
1136     // live outside of the current basic block.
1137     for (Value::use_iterator UI = BBI->use_begin(), E = BBI->use_end();
1138          UI != E; ++UI) {
1139       Instruction *U = cast<Instruction>(*UI);
1140       if (U->getParent() != BB || isa<PHINode>(U)) return false;
1141     }
1142 
1143     // Looks ok, continue checking.
1144   }
1145 
1146   return true;
1147 }
1148 
1149 /// FoldCondBranchOnPHI - If we have a conditional branch on a PHI node value
1150 /// that is defined in the same block as the branch and if any PHI entries are
1151 /// constants, thread edges corresponding to that entry to be branches to their
1152 /// ultimate destination.
1153 static bool FoldCondBranchOnPHI(BranchInst *BI) {
1154   BasicBlock *BB = BI->getParent();
1155   PHINode *PN = dyn_cast<PHINode>(BI->getCondition());
1156   // NOTE: we currently cannot transform this case if the PHI node is used
1157   // outside of the block.
1158   if (!PN || PN->getParent() != BB || !PN->hasOneUse())
1159     return false;
1160 
1161   // Degenerate case of a single entry PHI.
1162   if (PN->getNumIncomingValues() == 1) {
1163     FoldSingleEntryPHINodes(PN->getParent());
1164     return true;
1165   }
1166 
1167   // Now we know that this block has multiple preds and two succs.
1168   if (!BlockIsSimpleEnoughToThreadThrough(BB)) return false;
1169 
1170   // Okay, this is a simple enough basic block.  See if any phi values are
1171   // constants.
1172   for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) {
1173     ConstantInt *CB;
1174     if ((CB = dyn_cast<ConstantInt>(PN->getIncomingValue(i))) &&
1175         CB->getType() == Type::Int1Ty) {
1176       // Okay, we now know that all edges from PredBB should be revectored to
1177       // branch to RealDest.
1178       BasicBlock *PredBB = PN->getIncomingBlock(i);
1179       BasicBlock *RealDest = BI->getSuccessor(!CB->getZExtValue());
1180 
1181       if (RealDest == BB) continue;  // Skip self loops.
1182 
1183       // The dest block might have PHI nodes, other predecessors and other
1184       // difficult cases.  Instead of being smart about this, just insert a new
1185       // block that jumps to the destination block, effectively splitting
1186       // the edge we are about to create.
1187       BasicBlock *EdgeBB = BasicBlock::Create(RealDest->getName()+".critedge",
1188                                               RealDest->getParent(), RealDest);
1189       BranchInst::Create(RealDest, EdgeBB);
1190       PHINode *PN;
1191       for (BasicBlock::iterator BBI = RealDest->begin();
1192            (PN = dyn_cast<PHINode>(BBI)); ++BBI) {
1193         Value *V = PN->getIncomingValueForBlock(BB);
1194         PN->addIncoming(V, EdgeBB);
1195       }
1196 
1197       // BB may have instructions that are being threaded over.  Clone these
1198       // instructions into EdgeBB.  We know that there will be no uses of the
1199       // cloned instructions outside of EdgeBB.
1200       BasicBlock::iterator InsertPt = EdgeBB->begin();
1201       std::map<Value*, Value*> TranslateMap;  // Track translated values.
1202       for (BasicBlock::iterator BBI = BB->begin(); &*BBI != BI; ++BBI) {
1203         if (PHINode *PN = dyn_cast<PHINode>(BBI)) {
1204           TranslateMap[PN] = PN->getIncomingValueForBlock(PredBB);
1205         } else {
1206           // Clone the instruction.
1207           Instruction *N = BBI->clone();
1208           if (BBI->hasName()) N->setName(BBI->getName()+".c");
1209 
1210           // Update operands due to translation.
1211           for (User::op_iterator i = N->op_begin(), e = N->op_end();
1212                i != e; ++i) {
1213             std::map<Value*, Value*>::iterator PI =
1214               TranslateMap.find(*i);
1215             if (PI != TranslateMap.end())
1216               *i = PI->second;
1217           }
1218 
1219           // Check for trivial simplification.
1220           if (Constant *C = ConstantFoldInstruction(N)) {
1221             TranslateMap[BBI] = C;
1222             delete N;   // Constant folded away, don't need actual inst
1223           } else {
1224             // Insert the new instruction into its new home.
1225             EdgeBB->getInstList().insert(InsertPt, N);
1226             if (!BBI->use_empty())
1227               TranslateMap[BBI] = N;
1228           }
1229         }
1230       }
1231 
1232       // Loop over all of the edges from PredBB to BB, changing them to branch
1233       // to EdgeBB instead.
1234       TerminatorInst *PredBBTI = PredBB->getTerminator();
1235       for (unsigned i = 0, e = PredBBTI->getNumSuccessors(); i != e; ++i)
1236         if (PredBBTI->getSuccessor(i) == BB) {
1237           BB->removePredecessor(PredBB);
1238           PredBBTI->setSuccessor(i, EdgeBB);
1239         }
1240 
1241       // Recurse, simplifying any other constants.
1242       return FoldCondBranchOnPHI(BI) | true;
1243     }
1244   }
1245 
1246   return false;
1247 }
1248 
1249 /// FoldTwoEntryPHINode - Given a BB that starts with the specified two-entry
1250 /// PHI node, see if we can eliminate it.
1251 static bool FoldTwoEntryPHINode(PHINode *PN) {
1252   // Ok, this is a two entry PHI node.  Check to see if this is a simple "if
1253   // statement", which has a very simple dominance structure.  Basically, we
1254   // are trying to find the condition that is being branched on, which
1255   // subsequently causes this merge to happen.  We really want control
1256   // dependence information for this check, but simplifycfg can't keep it up
1257   // to date, and this catches most of the cases we care about anyway.
1258   //
1259   BasicBlock *BB = PN->getParent();
1260   BasicBlock *IfTrue, *IfFalse;
1261   Value *IfCond = GetIfCondition(BB, IfTrue, IfFalse);
1262   if (!IfCond) return false;
1263 
1264   // Okay, we found that we can merge this two-entry phi node into a select.
1265   // Doing so would require us to fold *all* two entry phi nodes in this block.
1266   // At some point this becomes non-profitable (particularly if the target
1267   // doesn't support cmov's).  Only do this transformation if there are two or
1268   // fewer PHI nodes in this block.
1269   unsigned NumPhis = 0;
1270   for (BasicBlock::iterator I = BB->begin(); isa<PHINode>(I); ++NumPhis, ++I)
1271     if (NumPhis > 2)
1272       return false;
1273 
1274   DOUT << "FOUND IF CONDITION!  " << *IfCond << "  T: "
1275        << IfTrue->getName() << "  F: " << IfFalse->getName() << "\n";
1276 
1277   // Loop over the PHI's seeing if we can promote them all to select
1278   // instructions.  While we are at it, keep track of the instructions
1279   // that need to be moved to the dominating block.
1280   std::set<Instruction*> AggressiveInsts;
1281 
1282   BasicBlock::iterator AfterPHIIt = BB->begin();
1283   while (isa<PHINode>(AfterPHIIt)) {
1284     PHINode *PN = cast<PHINode>(AfterPHIIt++);
1285     if (PN->getIncomingValue(0) == PN->getIncomingValue(1)) {
1286       if (PN->getIncomingValue(0) != PN)
1287         PN->replaceAllUsesWith(PN->getIncomingValue(0));
1288       else
1289         PN->replaceAllUsesWith(UndefValue::get(PN->getType()));
1290     } else if (!DominatesMergePoint(PN->getIncomingValue(0), BB,
1291                                     &AggressiveInsts) ||
1292                !DominatesMergePoint(PN->getIncomingValue(1), BB,
1293                                     &AggressiveInsts)) {
1294       return false;
1295     }
1296   }
1297 
1298   // If we all PHI nodes are promotable, check to make sure that all
1299   // instructions in the predecessor blocks can be promoted as well.  If
1300   // not, we won't be able to get rid of the control flow, so it's not
1301   // worth promoting to select instructions.
1302   BasicBlock *DomBlock = 0, *IfBlock1 = 0, *IfBlock2 = 0;
1303   PN = cast<PHINode>(BB->begin());
1304   BasicBlock *Pred = PN->getIncomingBlock(0);
1305   if (cast<BranchInst>(Pred->getTerminator())->isUnconditional()) {
1306     IfBlock1 = Pred;
1307     DomBlock = *pred_begin(Pred);
1308     for (BasicBlock::iterator I = Pred->begin();
1309          !isa<TerminatorInst>(I); ++I)
1310       if (!AggressiveInsts.count(I) && !isa<DbgInfoIntrinsic>(I)) {
1311         // This is not an aggressive instruction that we can promote.
1312         // Because of this, we won't be able to get rid of the control
1313         // flow, so the xform is not worth it.
1314         return false;
1315       }
1316   }
1317 
1318   Pred = PN->getIncomingBlock(1);
1319   if (cast<BranchInst>(Pred->getTerminator())->isUnconditional()) {
1320     IfBlock2 = Pred;
1321     DomBlock = *pred_begin(Pred);
1322     for (BasicBlock::iterator I = Pred->begin();
1323          !isa<TerminatorInst>(I); ++I)
1324       if (!AggressiveInsts.count(I) && !isa<DbgInfoIntrinsic>(I)) {
1325         // This is not an aggressive instruction that we can promote.
1326         // Because of this, we won't be able to get rid of the control
1327         // flow, so the xform is not worth it.
1328         return false;
1329       }
1330   }
1331 
1332   // If we can still promote the PHI nodes after this gauntlet of tests,
1333   // do all of the PHI's now.
1334 
1335   // Move all 'aggressive' instructions, which are defined in the
1336   // conditional parts of the if's up to the dominating block.
1337   if (IfBlock1) {
1338     DomBlock->getInstList().splice(DomBlock->getTerminator(),
1339                                    IfBlock1->getInstList(),
1340                                    IfBlock1->begin(),
1341                                    IfBlock1->getTerminator());
1342   }
1343   if (IfBlock2) {
1344     DomBlock->getInstList().splice(DomBlock->getTerminator(),
1345                                    IfBlock2->getInstList(),
1346                                    IfBlock2->begin(),
1347                                    IfBlock2->getTerminator());
1348   }
1349 
1350   while (PHINode *PN = dyn_cast<PHINode>(BB->begin())) {
1351     // Change the PHI node into a select instruction.
1352     Value *TrueVal =
1353       PN->getIncomingValue(PN->getIncomingBlock(0) == IfFalse);
1354     Value *FalseVal =
1355       PN->getIncomingValue(PN->getIncomingBlock(0) == IfTrue);
1356 
1357     Value *NV = SelectInst::Create(IfCond, TrueVal, FalseVal, "", AfterPHIIt);
1358     PN->replaceAllUsesWith(NV);
1359     NV->takeName(PN);
1360 
1361     BB->getInstList().erase(PN);
1362   }
1363   return true;
1364 }
1365 
1366 /// isTerminatorFirstRelevantInsn - Return true if Term is very first
1367 /// instruction ignoring Phi nodes and dbg intrinsics.
1368 static bool isTerminatorFirstRelevantInsn(BasicBlock *BB, Instruction *Term) {
1369   BasicBlock::iterator BBI = Term;
1370   while (BBI != BB->begin()) {
1371     --BBI;
1372     if (!isa<DbgInfoIntrinsic>(BBI))
1373       break;
1374   }
1375 
1376   if (isa<PHINode>(BBI) || &*BBI == Term || isa<DbgInfoIntrinsic>(BBI))
1377     return true;
1378   return false;
1379 }
1380 
1381 /// SimplifyCondBranchToTwoReturns - If we found a conditional branch that goes
1382 /// to two returning blocks, try to merge them together into one return,
1383 /// introducing a select if the return values disagree.
1384 static bool SimplifyCondBranchToTwoReturns(BranchInst *BI) {
1385   assert(BI->isConditional() && "Must be a conditional branch");
1386   BasicBlock *TrueSucc = BI->getSuccessor(0);
1387   BasicBlock *FalseSucc = BI->getSuccessor(1);
1388   ReturnInst *TrueRet = cast<ReturnInst>(TrueSucc->getTerminator());
1389   ReturnInst *FalseRet = cast<ReturnInst>(FalseSucc->getTerminator());
1390 
1391   // Check to ensure both blocks are empty (just a return) or optionally empty
1392   // with PHI nodes.  If there are other instructions, merging would cause extra
1393   // computation on one path or the other.
1394   if (!isTerminatorFirstRelevantInsn(TrueSucc, TrueRet))
1395     return false;
1396   if (!isTerminatorFirstRelevantInsn(FalseSucc, FalseRet))
1397     return false;
1398 
1399   // Okay, we found a branch that is going to two return nodes.  If
1400   // there is no return value for this function, just change the
1401   // branch into a return.
1402   if (FalseRet->getNumOperands() == 0) {
1403     TrueSucc->removePredecessor(BI->getParent());
1404     FalseSucc->removePredecessor(BI->getParent());
1405     ReturnInst::Create(0, BI);
1406     EraseTerminatorInstAndDCECond(BI);
1407     return true;
1408   }
1409 
1410   // Otherwise, figure out what the true and false return values are
1411   // so we can insert a new select instruction.
1412   Value *TrueValue = TrueRet->getReturnValue();
1413   Value *FalseValue = FalseRet->getReturnValue();
1414 
1415   // Unwrap any PHI nodes in the return blocks.
1416   if (PHINode *TVPN = dyn_cast_or_null<PHINode>(TrueValue))
1417     if (TVPN->getParent() == TrueSucc)
1418       TrueValue = TVPN->getIncomingValueForBlock(BI->getParent());
1419   if (PHINode *FVPN = dyn_cast_or_null<PHINode>(FalseValue))
1420     if (FVPN->getParent() == FalseSucc)
1421       FalseValue = FVPN->getIncomingValueForBlock(BI->getParent());
1422 
1423   // In order for this transformation to be safe, we must be able to
1424   // unconditionally execute both operands to the return.  This is
1425   // normally the case, but we could have a potentially-trapping
1426   // constant expression that prevents this transformation from being
1427   // safe.
1428   if (ConstantExpr *TCV = dyn_cast_or_null<ConstantExpr>(TrueValue))
1429     if (TCV->canTrap())
1430       return false;
1431   if (ConstantExpr *FCV = dyn_cast_or_null<ConstantExpr>(FalseValue))
1432     if (FCV->canTrap())
1433       return false;
1434 
1435   // Okay, we collected all the mapped values and checked them for sanity, and
1436   // defined to really do this transformation.  First, update the CFG.
1437   TrueSucc->removePredecessor(BI->getParent());
1438   FalseSucc->removePredecessor(BI->getParent());
1439 
1440   // Insert select instructions where needed.
1441   Value *BrCond = BI->getCondition();
1442   if (TrueValue) {
1443     // Insert a select if the results differ.
1444     if (TrueValue == FalseValue || isa<UndefValue>(FalseValue)) {
1445     } else if (isa<UndefValue>(TrueValue)) {
1446       TrueValue = FalseValue;
1447     } else {
1448       TrueValue = SelectInst::Create(BrCond, TrueValue,
1449                                      FalseValue, "retval", BI);
1450     }
1451   }
1452 
1453   Value *RI = !TrueValue ?
1454               ReturnInst::Create(BI) :
1455               ReturnInst::Create(TrueValue, BI);
1456 
1457   DOUT << "\nCHANGING BRANCH TO TWO RETURNS INTO SELECT:"
1458        << "\n  " << *BI << "NewRet = " << *RI
1459        << "TRUEBLOCK: " << *TrueSucc << "FALSEBLOCK: "<< *FalseSucc;
1460 
1461   EraseTerminatorInstAndDCECond(BI);
1462 
1463   return true;
1464 }
1465 
1466 /// FoldBranchToCommonDest - If this basic block is ONLY a setcc and a branch,
1467 /// and if a predecessor branches to us and one of our successors, fold the
1468 /// setcc into the predecessor and use logical operations to pick the right
1469 /// destination.
1470 static bool FoldBranchToCommonDest(BranchInst *BI) {
1471   BasicBlock *BB = BI->getParent();
1472   Instruction *Cond = dyn_cast<Instruction>(BI->getCondition());
1473   if (Cond == 0) return false;
1474 
1475 
1476   // Only allow this if the condition is a simple instruction that can be
1477   // executed unconditionally.  It must be in the same block as the branch, and
1478   // must be at the front of the block.
1479   BasicBlock::iterator FrontIt = BB->front();
1480   // Ignore dbg intrinsics.
1481   while(isa<DbgInfoIntrinsic>(FrontIt))
1482     ++FrontIt;
1483   if ((!isa<CmpInst>(Cond) && !isa<BinaryOperator>(Cond)) ||
1484       Cond->getParent() != BB || &*FrontIt != Cond || !Cond->hasOneUse()) {
1485     return false;
1486   }
1487 
1488   // Make sure the instruction after the condition is the cond branch.
1489   BasicBlock::iterator CondIt = Cond; ++CondIt;
1490   // Ingore dbg intrinsics.
1491   while(isa<DbgInfoIntrinsic>(CondIt))
1492     ++CondIt;
1493   if (&*CondIt != BI) {
1494     assert (!isa<DbgInfoIntrinsic>(CondIt) && "Hey do not forget debug info!");
1495     return false;
1496   }
1497 
1498   // Cond is known to be a compare or binary operator.  Check to make sure that
1499   // neither operand is a potentially-trapping constant expression.
1500   if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Cond->getOperand(0)))
1501     if (CE->canTrap())
1502       return false;
1503   if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Cond->getOperand(1)))
1504     if (CE->canTrap())
1505       return false;
1506 
1507 
1508   // Finally, don't infinitely unroll conditional loops.
1509   BasicBlock *TrueDest  = BI->getSuccessor(0);
1510   BasicBlock *FalseDest = BI->getSuccessor(1);
1511   if (TrueDest == BB || FalseDest == BB)
1512     return false;
1513 
1514   for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI) {
1515     BasicBlock *PredBlock = *PI;
1516     BranchInst *PBI = dyn_cast<BranchInst>(PredBlock->getTerminator());
1517 
1518     // Check that we have two conditional branches.  If there is a PHI node in
1519     // the common successor, verify that the same value flows in from both
1520     // blocks.
1521     if (PBI == 0 || PBI->isUnconditional() ||
1522         !SafeToMergeTerminators(BI, PBI))
1523       continue;
1524 
1525     Instruction::BinaryOps Opc;
1526     bool InvertPredCond = false;
1527 
1528     if (PBI->getSuccessor(0) == TrueDest)
1529       Opc = Instruction::Or;
1530     else if (PBI->getSuccessor(1) == FalseDest)
1531       Opc = Instruction::And;
1532     else if (PBI->getSuccessor(0) == FalseDest)
1533       Opc = Instruction::And, InvertPredCond = true;
1534     else if (PBI->getSuccessor(1) == TrueDest)
1535       Opc = Instruction::Or, InvertPredCond = true;
1536     else
1537       continue;
1538 
1539     DOUT << "FOLDING BRANCH TO COMMON DEST:\n" << *PBI << *BB;
1540 
1541     // If we need to invert the condition in the pred block to match, do so now.
1542     if (InvertPredCond) {
1543       Value *NewCond =
1544         BinaryOperator::CreateNot(PBI->getCondition(),
1545                                   PBI->getCondition()->getName()+".not", PBI);
1546       PBI->setCondition(NewCond);
1547       BasicBlock *OldTrue = PBI->getSuccessor(0);
1548       BasicBlock *OldFalse = PBI->getSuccessor(1);
1549       PBI->setSuccessor(0, OldFalse);
1550       PBI->setSuccessor(1, OldTrue);
1551     }
1552 
1553     // Clone Cond into the predecessor basic block, and or/and the
1554     // two conditions together.
1555     Instruction *New = Cond->clone();
1556     PredBlock->getInstList().insert(PBI, New);
1557     New->takeName(Cond);
1558     Cond->setName(New->getName()+".old");
1559 
1560     Value *NewCond = BinaryOperator::Create(Opc, PBI->getCondition(),
1561                                             New, "or.cond", PBI);
1562     PBI->setCondition(NewCond);
1563     if (PBI->getSuccessor(0) == BB) {
1564       AddPredecessorToBlock(TrueDest, PredBlock, BB);
1565       PBI->setSuccessor(0, TrueDest);
1566     }
1567     if (PBI->getSuccessor(1) == BB) {
1568       AddPredecessorToBlock(FalseDest, PredBlock, BB);
1569       PBI->setSuccessor(1, FalseDest);
1570     }
1571     return true;
1572   }
1573   return false;
1574 }
1575 
1576 /// SimplifyCondBranchToCondBranch - If we have a conditional branch as a
1577 /// predecessor of another block, this function tries to simplify it.  We know
1578 /// that PBI and BI are both conditional branches, and BI is in one of the
1579 /// successor blocks of PBI - PBI branches to BI.
1580 static bool SimplifyCondBranchToCondBranch(BranchInst *PBI, BranchInst *BI) {
1581   assert(PBI->isConditional() && BI->isConditional());
1582   BasicBlock *BB = BI->getParent();
1583 
1584   // If this block ends with a branch instruction, and if there is a
1585   // predecessor that ends on a branch of the same condition, make
1586   // this conditional branch redundant.
1587   if (PBI->getCondition() == BI->getCondition() &&
1588       PBI->getSuccessor(0) != PBI->getSuccessor(1)) {
1589     // Okay, the outcome of this conditional branch is statically
1590     // knowable.  If this block had a single pred, handle specially.
1591     if (BB->getSinglePredecessor()) {
1592       // Turn this into a branch on constant.
1593       bool CondIsTrue = PBI->getSuccessor(0) == BB;
1594       BI->setCondition(ConstantInt::get(Type::Int1Ty, CondIsTrue));
1595       return true;  // Nuke the branch on constant.
1596     }
1597 
1598     // Otherwise, if there are multiple predecessors, insert a PHI that merges
1599     // in the constant and simplify the block result.  Subsequent passes of
1600     // simplifycfg will thread the block.
1601     if (BlockIsSimpleEnoughToThreadThrough(BB)) {
1602       PHINode *NewPN = PHINode::Create(Type::Int1Ty,
1603                                        BI->getCondition()->getName() + ".pr",
1604                                        BB->begin());
1605       // Okay, we're going to insert the PHI node.  Since PBI is not the only
1606       // predecessor, compute the PHI'd conditional value for all of the preds.
1607       // Any predecessor where the condition is not computable we keep symbolic.
1608       for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI)
1609         if ((PBI = dyn_cast<BranchInst>((*PI)->getTerminator())) &&
1610             PBI != BI && PBI->isConditional() &&
1611             PBI->getCondition() == BI->getCondition() &&
1612             PBI->getSuccessor(0) != PBI->getSuccessor(1)) {
1613           bool CondIsTrue = PBI->getSuccessor(0) == BB;
1614           NewPN->addIncoming(ConstantInt::get(Type::Int1Ty,
1615                                               CondIsTrue), *PI);
1616         } else {
1617           NewPN->addIncoming(BI->getCondition(), *PI);
1618         }
1619 
1620       BI->setCondition(NewPN);
1621       return true;
1622     }
1623   }
1624 
1625   // If this is a conditional branch in an empty block, and if any
1626   // predecessors is a conditional branch to one of our destinations,
1627   // fold the conditions into logical ops and one cond br.
1628   BasicBlock::iterator BBI = BB->begin();
1629   // Ignore dbg intrinsics.
1630   while (isa<DbgInfoIntrinsic>(BBI))
1631     ++BBI;
1632   if (&*BBI != BI)
1633     return false;
1634 
1635 
1636   if (ConstantExpr *CE = dyn_cast<ConstantExpr>(BI->getCondition()))
1637     if (CE->canTrap())
1638       return false;
1639 
1640   int PBIOp, BIOp;
1641   if (PBI->getSuccessor(0) == BI->getSuccessor(0))
1642     PBIOp = BIOp = 0;
1643   else if (PBI->getSuccessor(0) == BI->getSuccessor(1))
1644     PBIOp = 0, BIOp = 1;
1645   else if (PBI->getSuccessor(1) == BI->getSuccessor(0))
1646     PBIOp = 1, BIOp = 0;
1647   else if (PBI->getSuccessor(1) == BI->getSuccessor(1))
1648     PBIOp = BIOp = 1;
1649   else
1650     return false;
1651 
1652   // Check to make sure that the other destination of this branch
1653   // isn't BB itself.  If so, this is an infinite loop that will
1654   // keep getting unwound.
1655   if (PBI->getSuccessor(PBIOp) == BB)
1656     return false;
1657 
1658   // Do not perform this transformation if it would require
1659   // insertion of a large number of select instructions. For targets
1660   // without predication/cmovs, this is a big pessimization.
1661   BasicBlock *CommonDest = PBI->getSuccessor(PBIOp);
1662 
1663   unsigned NumPhis = 0;
1664   for (BasicBlock::iterator II = CommonDest->begin();
1665        isa<PHINode>(II); ++II, ++NumPhis)
1666     if (NumPhis > 2) // Disable this xform.
1667       return false;
1668 
1669   // Finally, if everything is ok, fold the branches to logical ops.
1670   BasicBlock *OtherDest  = BI->getSuccessor(BIOp ^ 1);
1671 
1672   DOUT << "FOLDING BRs:" << *PBI->getParent()
1673        << "AND: " << *BI->getParent();
1674 
1675 
1676   // If OtherDest *is* BB, then BB is a basic block with a single conditional
1677   // branch in it, where one edge (OtherDest) goes back to itself but the other
1678   // exits.  We don't *know* that the program avoids the infinite loop
1679   // (even though that seems likely).  If we do this xform naively, we'll end up
1680   // recursively unpeeling the loop.  Since we know that (after the xform is
1681   // done) that the block *is* infinite if reached, we just make it an obviously
1682   // infinite loop with no cond branch.
1683   if (OtherDest == BB) {
1684     // Insert it at the end of the function, because it's either code,
1685     // or it won't matter if it's hot. :)
1686     BasicBlock *InfLoopBlock = BasicBlock::Create("infloop", BB->getParent());
1687     BranchInst::Create(InfLoopBlock, InfLoopBlock);
1688     OtherDest = InfLoopBlock;
1689   }
1690 
1691   DOUT << *PBI->getParent()->getParent();
1692 
1693   // BI may have other predecessors.  Because of this, we leave
1694   // it alone, but modify PBI.
1695 
1696   // Make sure we get to CommonDest on True&True directions.
1697   Value *PBICond = PBI->getCondition();
1698   if (PBIOp)
1699     PBICond = BinaryOperator::CreateNot(PBICond,
1700                                         PBICond->getName()+".not",
1701                                         PBI);
1702   Value *BICond = BI->getCondition();
1703   if (BIOp)
1704     BICond = BinaryOperator::CreateNot(BICond,
1705                                        BICond->getName()+".not",
1706                                        PBI);
1707   // Merge the conditions.
1708   Value *Cond = BinaryOperator::CreateOr(PBICond, BICond, "brmerge", PBI);
1709 
1710   // Modify PBI to branch on the new condition to the new dests.
1711   PBI->setCondition(Cond);
1712   PBI->setSuccessor(0, CommonDest);
1713   PBI->setSuccessor(1, OtherDest);
1714 
1715   // OtherDest may have phi nodes.  If so, add an entry from PBI's
1716   // block that are identical to the entries for BI's block.
1717   PHINode *PN;
1718   for (BasicBlock::iterator II = OtherDest->begin();
1719        (PN = dyn_cast<PHINode>(II)); ++II) {
1720     Value *V = PN->getIncomingValueForBlock(BB);
1721     PN->addIncoming(V, PBI->getParent());
1722   }
1723 
1724   // We know that the CommonDest already had an edge from PBI to
1725   // it.  If it has PHIs though, the PHIs may have different
1726   // entries for BB and PBI's BB.  If so, insert a select to make
1727   // them agree.
1728   for (BasicBlock::iterator II = CommonDest->begin();
1729        (PN = dyn_cast<PHINode>(II)); ++II) {
1730     Value *BIV = PN->getIncomingValueForBlock(BB);
1731     unsigned PBBIdx = PN->getBasicBlockIndex(PBI->getParent());
1732     Value *PBIV = PN->getIncomingValue(PBBIdx);
1733     if (BIV != PBIV) {
1734       // Insert a select in PBI to pick the right value.
1735       Value *NV = SelectInst::Create(PBICond, PBIV, BIV,
1736                                      PBIV->getName()+".mux", PBI);
1737       PN->setIncomingValue(PBBIdx, NV);
1738     }
1739   }
1740 
1741   DOUT << "INTO: " << *PBI->getParent();
1742 
1743   DOUT << *PBI->getParent()->getParent();
1744 
1745   // This basic block is probably dead.  We know it has at least
1746   // one fewer predecessor.
1747   return true;
1748 }
1749 
1750 
1751 /// SimplifyCFG - This function is used to do simplification of a CFG.  For
1752 /// example, it adjusts branches to branches to eliminate the extra hop, it
1753 /// eliminates unreachable basic blocks, and does other "peephole" optimization
1754 /// of the CFG.  It returns true if a modification was made.
1755 ///
1756 /// WARNING:  The entry node of a function may not be simplified.
1757 ///
1758 bool llvm::SimplifyCFG(BasicBlock *BB) {
1759   bool Changed = false;
1760   Function *M = BB->getParent();
1761 
1762   assert(BB && BB->getParent() && "Block not embedded in function!");
1763   assert(BB->getTerminator() && "Degenerate basic block encountered!");
1764   assert(&BB->getParent()->getEntryBlock() != BB &&
1765          "Can't Simplify entry block!");
1766 
1767   // Remove basic blocks that have no predecessors... or that just have themself
1768   // as a predecessor.  These are unreachable.
1769   if (pred_begin(BB) == pred_end(BB) || BB->getSinglePredecessor() == BB) {
1770     DOUT << "Removing BB: \n" << *BB;
1771     DeleteDeadBlock(BB);
1772     return true;
1773   }
1774 
1775   // Check to see if we can constant propagate this terminator instruction
1776   // away...
1777   Changed |= ConstantFoldTerminator(BB);
1778 
1779   // If there is a trivial two-entry PHI node in this basic block, and we can
1780   // eliminate it, do so now.
1781   if (PHINode *PN = dyn_cast<PHINode>(BB->begin()))
1782     if (PN->getNumIncomingValues() == 2)
1783       Changed |= FoldTwoEntryPHINode(PN);
1784 
1785   // If this is a returning block with only PHI nodes in it, fold the return
1786   // instruction into any unconditional branch predecessors.
1787   //
1788   // If any predecessor is a conditional branch that just selects among
1789   // different return values, fold the replace the branch/return with a select
1790   // and return.
1791   if (ReturnInst *RI = dyn_cast<ReturnInst>(BB->getTerminator())) {
1792     if (isTerminatorFirstRelevantInsn(BB, BB->getTerminator())) {
1793       // Find predecessors that end with branches.
1794       SmallVector<BasicBlock*, 8> UncondBranchPreds;
1795       SmallVector<BranchInst*, 8> CondBranchPreds;
1796       for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI) {
1797         TerminatorInst *PTI = (*PI)->getTerminator();
1798         if (BranchInst *BI = dyn_cast<BranchInst>(PTI)) {
1799           if (BI->isUnconditional())
1800             UncondBranchPreds.push_back(*PI);
1801           else
1802             CondBranchPreds.push_back(BI);
1803         }
1804       }
1805 
1806       // If we found some, do the transformation!
1807       if (!UncondBranchPreds.empty()) {
1808         while (!UncondBranchPreds.empty()) {
1809           BasicBlock *Pred = UncondBranchPreds.pop_back_val();
1810           DOUT << "FOLDING: " << *BB
1811                << "INTO UNCOND BRANCH PRED: " << *Pred;
1812           Instruction *UncondBranch = Pred->getTerminator();
1813           // Clone the return and add it to the end of the predecessor.
1814           Instruction *NewRet = RI->clone();
1815           Pred->getInstList().push_back(NewRet);
1816 
1817           BasicBlock::iterator BBI = RI;
1818           if (BBI != BB->begin()) {
1819             // Move region end info into the predecessor.
1820             if (DbgRegionEndInst *DREI = dyn_cast<DbgRegionEndInst>(--BBI))
1821               DREI->moveBefore(NewRet);
1822           }
1823 
1824           // If the return instruction returns a value, and if the value was a
1825           // PHI node in "BB", propagate the right value into the return.
1826           for (User::op_iterator i = NewRet->op_begin(), e = NewRet->op_end();
1827                i != e; ++i)
1828             if (PHINode *PN = dyn_cast<PHINode>(*i))
1829               if (PN->getParent() == BB)
1830                 *i = PN->getIncomingValueForBlock(Pred);
1831 
1832           // Update any PHI nodes in the returning block to realize that we no
1833           // longer branch to them.
1834           BB->removePredecessor(Pred);
1835           Pred->getInstList().erase(UncondBranch);
1836         }
1837 
1838         // If we eliminated all predecessors of the block, delete the block now.
1839         if (pred_begin(BB) == pred_end(BB))
1840           // We know there are no successors, so just nuke the block.
1841           M->getBasicBlockList().erase(BB);
1842 
1843         return true;
1844       }
1845 
1846       // Check out all of the conditional branches going to this return
1847       // instruction.  If any of them just select between returns, change the
1848       // branch itself into a select/return pair.
1849       while (!CondBranchPreds.empty()) {
1850         BranchInst *BI = CondBranchPreds.pop_back_val();
1851 
1852         // Check to see if the non-BB successor is also a return block.
1853         if (isa<ReturnInst>(BI->getSuccessor(0)->getTerminator()) &&
1854             isa<ReturnInst>(BI->getSuccessor(1)->getTerminator()) &&
1855             SimplifyCondBranchToTwoReturns(BI))
1856           return true;
1857       }
1858     }
1859   } else if (isa<UnwindInst>(BB->begin())) {
1860     // Check to see if the first instruction in this block is just an unwind.
1861     // If so, replace any invoke instructions which use this as an exception
1862     // destination with call instructions, and any unconditional branch
1863     // predecessor with an unwind.
1864     //
1865     SmallVector<BasicBlock*, 8> Preds(pred_begin(BB), pred_end(BB));
1866     while (!Preds.empty()) {
1867       BasicBlock *Pred = Preds.back();
1868       if (BranchInst *BI = dyn_cast<BranchInst>(Pred->getTerminator())) {
1869         if (BI->isUnconditional()) {
1870           Pred->getInstList().pop_back();  // nuke uncond branch
1871           new UnwindInst(Pred);            // Use unwind.
1872           Changed = true;
1873         }
1874       } else if (InvokeInst *II = dyn_cast<InvokeInst>(Pred->getTerminator()))
1875         if (II->getUnwindDest() == BB) {
1876           // Insert a new branch instruction before the invoke, because this
1877           // is now a fall through...
1878           BranchInst *BI = BranchInst::Create(II->getNormalDest(), II);
1879           Pred->getInstList().remove(II);   // Take out of symbol table
1880 
1881           // Insert the call now...
1882           SmallVector<Value*,8> Args(II->op_begin()+3, II->op_end());
1883           CallInst *CI = CallInst::Create(II->getCalledValue(),
1884                                           Args.begin(), Args.end(),
1885                                           II->getName(), BI);
1886           CI->setCallingConv(II->getCallingConv());
1887           CI->setAttributes(II->getAttributes());
1888           // If the invoke produced a value, the Call now does instead
1889           II->replaceAllUsesWith(CI);
1890           delete II;
1891           Changed = true;
1892         }
1893 
1894       Preds.pop_back();
1895     }
1896 
1897     // If this block is now dead, remove it.
1898     if (pred_begin(BB) == pred_end(BB)) {
1899       // We know there are no successors, so just nuke the block.
1900       M->getBasicBlockList().erase(BB);
1901       return true;
1902     }
1903 
1904   } else if (SwitchInst *SI = dyn_cast<SwitchInst>(BB->getTerminator())) {
1905     if (isValueEqualityComparison(SI)) {
1906       // If we only have one predecessor, and if it is a branch on this value,
1907       // see if that predecessor totally determines the outcome of this switch.
1908       if (BasicBlock *OnlyPred = BB->getSinglePredecessor())
1909         if (SimplifyEqualityComparisonWithOnlyPredecessor(SI, OnlyPred))
1910           return SimplifyCFG(BB) || 1;
1911 
1912       // If the block only contains the switch, see if we can fold the block
1913       // away into any preds.
1914       BasicBlock::iterator BBI = BB->begin();
1915       // Ignore dbg intrinsics.
1916       while (isa<DbgInfoIntrinsic>(BBI))
1917         ++BBI;
1918       if (SI == &*BBI)
1919         if (FoldValueComparisonIntoPredecessors(SI))
1920           return SimplifyCFG(BB) || 1;
1921     }
1922   } else if (BranchInst *BI = dyn_cast<BranchInst>(BB->getTerminator())) {
1923     if (BI->isUnconditional()) {
1924       BasicBlock::iterator BBI = BB->getFirstNonPHI();
1925 
1926       BasicBlock *Succ = BI->getSuccessor(0);
1927       // Ignore dbg intrinsics.
1928       while (isa<DbgInfoIntrinsic>(BBI))
1929         ++BBI;
1930       if (BBI->isTerminator() &&  // Terminator is the only non-phi instruction!
1931           Succ != BB)             // Don't hurt infinite loops!
1932         if (TryToSimplifyUncondBranchFromEmptyBlock(BB, Succ))
1933           return true;
1934 
1935     } else {  // Conditional branch
1936       if (isValueEqualityComparison(BI)) {
1937         // If we only have one predecessor, and if it is a branch on this value,
1938         // see if that predecessor totally determines the outcome of this
1939         // switch.
1940         if (BasicBlock *OnlyPred = BB->getSinglePredecessor())
1941           if (SimplifyEqualityComparisonWithOnlyPredecessor(BI, OnlyPred))
1942             return SimplifyCFG(BB) || 1;
1943 
1944         // This block must be empty, except for the setcond inst, if it exists.
1945         // Ignore dbg intrinsics.
1946         BasicBlock::iterator I = BB->begin();
1947         // Ignore dbg intrinsics.
1948         while (isa<DbgInfoIntrinsic>(I))
1949           ++I;
1950         if (&*I == BI) {
1951           if (FoldValueComparisonIntoPredecessors(BI))
1952             return SimplifyCFG(BB) | true;
1953         } else if (&*I == cast<Instruction>(BI->getCondition())){
1954           ++I;
1955           // Ignore dbg intrinsics.
1956           while (isa<DbgInfoIntrinsic>(I))
1957             ++I;
1958           if(&*I == BI) {
1959             if (FoldValueComparisonIntoPredecessors(BI))
1960               return SimplifyCFG(BB) | true;
1961           }
1962         }
1963       }
1964 
1965       // If this is a branch on a phi node in the current block, thread control
1966       // through this block if any PHI node entries are constants.
1967       if (PHINode *PN = dyn_cast<PHINode>(BI->getCondition()))
1968         if (PN->getParent() == BI->getParent())
1969           if (FoldCondBranchOnPHI(BI))
1970             return SimplifyCFG(BB) | true;
1971 
1972       // If this basic block is ONLY a setcc and a branch, and if a predecessor
1973       // branches to us and one of our successors, fold the setcc into the
1974       // predecessor and use logical operations to pick the right destination.
1975       if (FoldBranchToCommonDest(BI))
1976         return SimplifyCFG(BB) | 1;
1977 
1978 
1979       // Scan predecessor blocks for conditional branches.
1980       for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI)
1981         if (BranchInst *PBI = dyn_cast<BranchInst>((*PI)->getTerminator()))
1982           if (PBI != BI && PBI->isConditional())
1983             if (SimplifyCondBranchToCondBranch(PBI, BI))
1984               return SimplifyCFG(BB) | true;
1985     }
1986   } else if (isa<UnreachableInst>(BB->getTerminator())) {
1987     // If there are any instructions immediately before the unreachable that can
1988     // be removed, do so.
1989     Instruction *Unreachable = BB->getTerminator();
1990     while (Unreachable != BB->begin()) {
1991       BasicBlock::iterator BBI = Unreachable;
1992       --BBI;
1993       // Do not delete instructions that can have side effects, like calls
1994       // (which may never return) and volatile loads and stores.
1995       if (isa<CallInst>(BBI) && !isa<DbgInfoIntrinsic>(BBI)) break;
1996 
1997       if (StoreInst *SI = dyn_cast<StoreInst>(BBI))
1998         if (SI->isVolatile())
1999           break;
2000 
2001       if (LoadInst *LI = dyn_cast<LoadInst>(BBI))
2002         if (LI->isVolatile())
2003           break;
2004 
2005       // Delete this instruction
2006       BB->getInstList().erase(BBI);
2007       Changed = true;
2008     }
2009 
2010     // If the unreachable instruction is the first in the block, take a gander
2011     // at all of the predecessors of this instruction, and simplify them.
2012     if (&BB->front() == Unreachable) {
2013       SmallVector<BasicBlock*, 8> Preds(pred_begin(BB), pred_end(BB));
2014       for (unsigned i = 0, e = Preds.size(); i != e; ++i) {
2015         TerminatorInst *TI = Preds[i]->getTerminator();
2016 
2017         if (BranchInst *BI = dyn_cast<BranchInst>(TI)) {
2018           if (BI->isUnconditional()) {
2019             if (BI->getSuccessor(0) == BB) {
2020               new UnreachableInst(TI);
2021               TI->eraseFromParent();
2022               Changed = true;
2023             }
2024           } else {
2025             if (BI->getSuccessor(0) == BB) {
2026               BranchInst::Create(BI->getSuccessor(1), BI);
2027               EraseTerminatorInstAndDCECond(BI);
2028             } else if (BI->getSuccessor(1) == BB) {
2029               BranchInst::Create(BI->getSuccessor(0), BI);
2030               EraseTerminatorInstAndDCECond(BI);
2031               Changed = true;
2032             }
2033           }
2034         } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) {
2035           for (unsigned i = 1, e = SI->getNumCases(); i != e; ++i)
2036             if (SI->getSuccessor(i) == BB) {
2037               BB->removePredecessor(SI->getParent());
2038               SI->removeCase(i);
2039               --i; --e;
2040               Changed = true;
2041             }
2042           // If the default value is unreachable, figure out the most popular
2043           // destination and make it the default.
2044           if (SI->getSuccessor(0) == BB) {
2045             std::map<BasicBlock*, unsigned> Popularity;
2046             for (unsigned i = 1, e = SI->getNumCases(); i != e; ++i)
2047               Popularity[SI->getSuccessor(i)]++;
2048 
2049             // Find the most popular block.
2050             unsigned MaxPop = 0;
2051             BasicBlock *MaxBlock = 0;
2052             for (std::map<BasicBlock*, unsigned>::iterator
2053                    I = Popularity.begin(), E = Popularity.end(); I != E; ++I) {
2054               if (I->second > MaxPop) {
2055                 MaxPop = I->second;
2056                 MaxBlock = I->first;
2057               }
2058             }
2059             if (MaxBlock) {
2060               // Make this the new default, allowing us to delete any explicit
2061               // edges to it.
2062               SI->setSuccessor(0, MaxBlock);
2063               Changed = true;
2064 
2065               // If MaxBlock has phinodes in it, remove MaxPop-1 entries from
2066               // it.
2067               if (isa<PHINode>(MaxBlock->begin()))
2068                 for (unsigned i = 0; i != MaxPop-1; ++i)
2069                   MaxBlock->removePredecessor(SI->getParent());
2070 
2071               for (unsigned i = 1, e = SI->getNumCases(); i != e; ++i)
2072                 if (SI->getSuccessor(i) == MaxBlock) {
2073                   SI->removeCase(i);
2074                   --i; --e;
2075                 }
2076             }
2077           }
2078         } else if (InvokeInst *II = dyn_cast<InvokeInst>(TI)) {
2079           if (II->getUnwindDest() == BB) {
2080             // Convert the invoke to a call instruction.  This would be a good
2081             // place to note that the call does not throw though.
2082             BranchInst *BI = BranchInst::Create(II->getNormalDest(), II);
2083             II->removeFromParent();   // Take out of symbol table
2084 
2085             // Insert the call now...
2086             SmallVector<Value*, 8> Args(II->op_begin()+3, II->op_end());
2087             CallInst *CI = CallInst::Create(II->getCalledValue(),
2088                                             Args.begin(), Args.end(),
2089                                             II->getName(), BI);
2090             CI->setCallingConv(II->getCallingConv());
2091             CI->setAttributes(II->getAttributes());
2092             // If the invoke produced a value, the Call does now instead.
2093             II->replaceAllUsesWith(CI);
2094             delete II;
2095             Changed = true;
2096           }
2097         }
2098       }
2099 
2100       // If this block is now dead, remove it.
2101       if (pred_begin(BB) == pred_end(BB)) {
2102         // We know there are no successors, so just nuke the block.
2103         M->getBasicBlockList().erase(BB);
2104         return true;
2105       }
2106     }
2107   }
2108 
2109   // Merge basic blocks into their predecessor if there is only one distinct
2110   // pred, and if there is only one distinct successor of the predecessor, and
2111   // if there are no PHI nodes.
2112   //
2113   if (MergeBlockIntoPredecessor(BB))
2114     return true;
2115 
2116   // Otherwise, if this block only has a single predecessor, and if that block
2117   // is a conditional branch, see if we can hoist any code from this block up
2118   // into our predecessor.
2119   pred_iterator PI(pred_begin(BB)), PE(pred_end(BB));
2120   BasicBlock *OnlyPred = *PI++;
2121   for (; PI != PE; ++PI)  // Search all predecessors, see if they are all same
2122     if (*PI != OnlyPred) {
2123       OnlyPred = 0;       // There are multiple different predecessors...
2124       break;
2125     }
2126 
2127   if (OnlyPred)
2128     if (BranchInst *BI = dyn_cast<BranchInst>(OnlyPred->getTerminator()))
2129       if (BI->isConditional()) {
2130         // Get the other block.
2131         BasicBlock *OtherBB = BI->getSuccessor(BI->getSuccessor(0) == BB);
2132         PI = pred_begin(OtherBB);
2133         ++PI;
2134 
2135         if (PI == pred_end(OtherBB)) {
2136           // We have a conditional branch to two blocks that are only reachable
2137           // from the condbr.  We know that the condbr dominates the two blocks,
2138           // so see if there is any identical code in the "then" and "else"
2139           // blocks.  If so, we can hoist it up to the branching block.
2140           Changed |= HoistThenElseCodeToIf(BI);
2141         } else {
2142           BasicBlock* OnlySucc = NULL;
2143           for (succ_iterator SI = succ_begin(BB), SE = succ_end(BB);
2144                SI != SE; ++SI) {
2145             if (!OnlySucc)
2146               OnlySucc = *SI;
2147             else if (*SI != OnlySucc) {
2148               OnlySucc = 0;     // There are multiple distinct successors!
2149               break;
2150             }
2151           }
2152 
2153           if (OnlySucc == OtherBB) {
2154             // If BB's only successor is the other successor of the predecessor,
2155             // i.e. a triangle, see if we can hoist any code from this block up
2156             // to the "if" block.
2157             Changed |= SpeculativelyExecuteBB(BI, BB);
2158           }
2159         }
2160       }
2161 
2162   for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI)
2163     if (BranchInst *BI = dyn_cast<BranchInst>((*PI)->getTerminator()))
2164       // Change br (X == 0 | X == 1), T, F into a switch instruction.
2165       if (BI->isConditional() && isa<Instruction>(BI->getCondition())) {
2166         Instruction *Cond = cast<Instruction>(BI->getCondition());
2167         // If this is a bunch of seteq's or'd together, or if it's a bunch of
2168         // 'setne's and'ed together, collect them.
2169         Value *CompVal = 0;
2170         std::vector<ConstantInt*> Values;
2171         bool TrueWhenEqual = GatherValueComparisons(Cond, CompVal, Values);
2172         if (CompVal && CompVal->getType()->isInteger()) {
2173           // There might be duplicate constants in the list, which the switch
2174           // instruction can't handle, remove them now.
2175           std::sort(Values.begin(), Values.end(), ConstantIntOrdering());
2176           Values.erase(std::unique(Values.begin(), Values.end()), Values.end());
2177 
2178           // Figure out which block is which destination.
2179           BasicBlock *DefaultBB = BI->getSuccessor(1);
2180           BasicBlock *EdgeBB    = BI->getSuccessor(0);
2181           if (!TrueWhenEqual) std::swap(DefaultBB, EdgeBB);
2182 
2183           // Create the new switch instruction now.
2184           SwitchInst *New = SwitchInst::Create(CompVal, DefaultBB,
2185                                                Values.size(), BI);
2186 
2187           // Add all of the 'cases' to the switch instruction.
2188           for (unsigned i = 0, e = Values.size(); i != e; ++i)
2189             New->addCase(Values[i], EdgeBB);
2190 
2191           // We added edges from PI to the EdgeBB.  As such, if there were any
2192           // PHI nodes in EdgeBB, they need entries to be added corresponding to
2193           // the number of edges added.
2194           for (BasicBlock::iterator BBI = EdgeBB->begin();
2195                isa<PHINode>(BBI); ++BBI) {
2196             PHINode *PN = cast<PHINode>(BBI);
2197             Value *InVal = PN->getIncomingValueForBlock(*PI);
2198             for (unsigned i = 0, e = Values.size()-1; i != e; ++i)
2199               PN->addIncoming(InVal, *PI);
2200           }
2201 
2202           // Erase the old branch instruction.
2203           EraseTerminatorInstAndDCECond(BI);
2204           return true;
2205         }
2206       }
2207 
2208   return Changed;
2209 }
2210