Home | History | Annotate | Download | only in CodeGen
      1 //===-- PHIEliminationUtils.cpp - Helper functions for PHI elimination ----===//
      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 #include "PHIEliminationUtils.h"
     11 #include "llvm/CodeGen/MachineBasicBlock.h"
     12 #include "llvm/CodeGen/MachineFunction.h"
     13 #include "llvm/CodeGen/MachineRegisterInfo.h"
     14 #include "llvm/ADT/SmallPtrSet.h"
     15 using namespace llvm;
     16 
     17 // findCopyInsertPoint - Find a safe place in MBB to insert a copy from SrcReg
     18 // when following the CFG edge to SuccMBB. This needs to be after any def of
     19 // SrcReg, but before any subsequent point where control flow might jump out of
     20 // the basic block.
     21 MachineBasicBlock::iterator
     22 llvm::findPHICopyInsertPoint(MachineBasicBlock* MBB, MachineBasicBlock* SuccMBB,
     23                              unsigned SrcReg) {
     24   // Handle the trivial case trivially.
     25   if (MBB->empty())
     26     return MBB->begin();
     27 
     28   // Usually, we just want to insert the copy before the first terminator
     29   // instruction. However, for the edge going to a landing pad, we must insert
     30   // the copy before the call/invoke instruction.
     31   if (!SuccMBB->isLandingPad())
     32     return MBB->getFirstTerminator();
     33 
     34   // Discover any defs/uses in this basic block.
     35   SmallPtrSet<MachineInstr*, 8> DefUsesInMBB;
     36   MachineRegisterInfo& MRI = MBB->getParent()->getRegInfo();
     37   for (MachineRegisterInfo::reg_iterator RI = MRI.reg_begin(SrcReg),
     38          RE = MRI.reg_end(); RI != RE; ++RI) {
     39     MachineInstr* DefUseMI = &*RI;
     40     if (DefUseMI->getParent() == MBB)
     41       DefUsesInMBB.insert(DefUseMI);
     42   }
     43 
     44   MachineBasicBlock::iterator InsertPoint;
     45   if (DefUsesInMBB.empty()) {
     46     // No defs.  Insert the copy at the start of the basic block.
     47     InsertPoint = MBB->begin();
     48   } else if (DefUsesInMBB.size() == 1) {
     49     // Insert the copy immediately after the def/use.
     50     InsertPoint = *DefUsesInMBB.begin();
     51     ++InsertPoint;
     52   } else {
     53     // Insert the copy immediately after the last def/use.
     54     InsertPoint = MBB->end();
     55     while (!DefUsesInMBB.count(&*--InsertPoint)) {}
     56     ++InsertPoint;
     57   }
     58 
     59   // Make sure the copy goes after any phi nodes however.
     60   return MBB->SkipPHIsAndLabels(InsertPoint);
     61 }
     62