lib/Analysis/PostDominators.cpp

   1 //===- PostDominators.cpp - Post-Dominator Calculation --------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the post-dominator construction algorithms.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "llvm/Analysis/PostDominators.h"
  15 #include "llvm/Instructions.h"
  16 #include "llvm/Support/CFG.h"
  17 #include "llvm/ADT/DepthFirstIterator.h"
  18 #include "llvm/ADT/SetOperations.h"
  19 #include "PostDominatorCalculation.h"
  20 using namespace llvm;
  21
  22 //===----------------------------------------------------------------------===//
  23 //  PostDominatorTree Implementation
  24 //===----------------------------------------------------------------------===//
  25
  26 char PostDominatorTree::ID = 0;
  27 char PostDominanceFrontier::ID = 0;
  28 static RegisterPass<PostDominatorTree>
  29 F("postdomtree", "Post-Dominator Tree Construction", true);
  30
  31 bool PostDominatorTree::runOnFunction(Function &F) {
  32   reset();     // Reset from the last time we were run...
  33
  34   // Initialize the roots list
  35   for (Function::iterator I = F.begin(), E = F.end(); I != E; ++I) {
  36     TerminatorInst *Insn = I->getTerminator();
  37     if (Insn->getNumSuccessors() == 0) {
  38       // Unreachable block is not a root node.
  39       if (!isa<UnreachableInst>(Insn))
  40         Roots.push_back(I);
  41     }
  42
  43     // Prepopulate maps so that we don't get iterator invalidation issues later.
  44     IDoms[I] = 0;
  45     DomTreeNodes[I] = 0;
  46   }
  47
  48   Vertex.push_back(0);
  49
  50   PDTcalculate(*this, F);
  51   return false;
  52 }
  53
  54 //===----------------------------------------------------------------------===//
  55 //  PostDominanceFrontier Implementation
  56 //===----------------------------------------------------------------------===//
  57
  58 static RegisterPass<PostDominanceFrontier>
  59 H("postdomfrontier", "Post-Dominance Frontier Construction", true);
  60
  61 const DominanceFrontier::DomSetType &
  62 PostDominanceFrontier::calculate(const PostDominatorTree &DT,
  63                                  const DomTreeNode *Node) {
  64   // Loop over CFG successors to calculate DFlocal[Node]
  65   BasicBlock *BB = Node->getBlock();
  66   DomSetType &S = Frontiers[BB];       // The new set to fill in...
  67   if (getRoots().empty()) return S;
  68
  69   if (BB)
  70     for (pred_iterator SI = pred_begin(BB), SE = pred_end(BB);
  71          SI != SE; ++SI) {
  72       // Does Node immediately dominate this predecessor?
  73       DomTreeNode *SINode = DT[*SI];
  74       if (SINode && SINode->getIDom() != Node)
  75         S.insert(*SI);
  76     }
  77
  78   // At this point, S is DFlocal.  Now we union in DFup's of our children...
  79   // Loop through and visit the nodes that Node immediately dominates (Node's
  80   // children in the IDomTree)
  81   //
  82   for (DomTreeNode::const_iterator
  83          NI = Node->begin(), NE = Node->end(); NI != NE; ++NI) {
  84     DomTreeNode *IDominee = *NI;
  85     const DomSetType &ChildDF = calculate(DT, IDominee);
  86
  87     DomSetType::const_iterator CDFI = ChildDF.begin(), CDFE = ChildDF.end();
  88     for (; CDFI != CDFE; ++CDFI) {
  89       if (!DT.properlyDominates(Node, DT[*CDFI]))
  90         S.insert(*CDFI);
  91     }
  92   }
  93
  94   return S;
  95 }
  96
  97 // Ensure that this .cpp file gets linked when PostDominators.h is used.
  98 DEFINING_FILE_FOR(PostDominanceFrontier)