Now with less tabs!
[oota-llvm.git] / lib / VMCore / Dominators.cpp
index 30d170b6698da43bdf5cbd1128e28457ed071c57..5bb25d47d611c0abcf2036838e67939c39377787 100644 (file)
-//===- DominatorSet.cpp - Dominator Set Calculation --------------*- C++ -*--=//
+//===- Dominators.cpp - Dominator Calculation -----------------------------===//
 //
-// This file provides a simple class to calculate the dominator set of a
-// function.
+//                     The LLVM Compiler Infrastructure
+//
+// This file was developed by the LLVM research group and is distributed under
+// the University of Illinois Open Source License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file implements simple dominator construction algorithms for finding
+// forward dominators.  Postdominators are available in libanalysis, but are not
+// included in libvmcore, because it's not needed.  Forward dominators are
+// needed to support the Verifier pass.
 //
 //===----------------------------------------------------------------------===//
 
 #include "llvm/Analysis/Dominators.h"
-#include "llvm/Transforms/UnifyFunctionExitNodes.h"
-#include "llvm/Function.h"
 #include "llvm/Support/CFG.h"
-#include "Support/DepthFirstIterator.h"
-#include "Support/STLExtras.h"
-#include "Support/SetOperations.h"
+#include "llvm/Support/Compiler.h"
+#include "llvm/ADT/DepthFirstIterator.h"
+#include "llvm/ADT/SetOperations.h"
+#include "llvm/ADT/SmallPtrSet.h"
+#include "llvm/ADT/SmallVector.h"
+#include "llvm/Analysis/DominatorInternals.h"
+#include "llvm/Instructions.h"
+#include "llvm/Support/Streams.h"
 #include <algorithm>
-using std::set;
+using namespace llvm;
+
+namespace llvm {
+static std::ostream &operator<<(std::ostream &o,
+                                const std::set<BasicBlock*> &BBs) {
+  for (std::set<BasicBlock*>::const_iterator I = BBs.begin(), E = BBs.end();
+       I != E; ++I)
+    if (*I)
+      WriteAsOperand(o, *I, false);
+    else
+      o << " <<exit node>>";
+  return o;
+}
+}
 
 //===----------------------------------------------------------------------===//
-//  DominatorSet Implementation
+//  DominatorTree Implementation
 //===----------------------------------------------------------------------===//
-
-AnalysisID DominatorSet::ID(AnalysisID::create<DominatorSet>());
-AnalysisID DominatorSet::PostDomID(AnalysisID::create<DominatorSet>());
-
-bool DominatorSet::runOnFunction(Function *F) {
-  Doms.clear();   // Reset from the last time we were run...
-
-  if (isPostDominator())
-    calcPostDominatorSet(F);
-  else
-    calcForwardDominatorSet(F);
-  return false;
-}
-
-
-// calcForwardDominatorSet - This method calculates the forward dominator sets
-// for the specified function.
 //
-void DominatorSet::calcForwardDominatorSet(Function *M) {
-  Root = M->getEntryNode();
-  assert(pred_begin(Root) == pred_end(Root) &&
-        "Root node has predecessors in function!");
+// Provide public access to DominatorTree information.  Implementation details
+// can be found in DominatorCalculation.h.
+//
+//===----------------------------------------------------------------------===//
 
-  bool Changed;
-  do {
-    Changed = false;
+TEMPLATE_INSTANTIATION(class DomTreeNodeBase<BasicBlock>);
+TEMPLATE_INSTANTIATION(class DominatorTreeBase<BasicBlock>);
 
-    DomSetType WorkingSet;
-    df_iterator<Function*> It = df_begin(M), End = df_end(M);
-    for ( ; It != End; ++It) {
-      BasicBlock *BB = *It;
-      pred_iterator PI = pred_begin(BB), PEnd = pred_end(BB);
-      if (PI != PEnd) {                // Is there SOME predecessor?
-       // Loop until we get to a predecessor that has had it's dom set filled
-       // in at least once.  We are guaranteed to have this because we are
-       // traversing the graph in DFO and have handled start nodes specially.
-       //
-       while (Doms[*PI].size() == 0) ++PI;
-       WorkingSet = Doms[*PI];
+char DominatorTree::ID = 0;
+static RegisterPass<DominatorTree>
+E("domtree", "Dominator Tree Construction", true);
 
-       for (++PI; PI != PEnd; ++PI) { // Intersect all of the predecessor sets
-         DomSetType &PredSet = Doms[*PI];
-         if (PredSet.size())
-           set_intersect(WorkingSet, PredSet);
-       }
-      }
-       
-      WorkingSet.insert(BB);           // A block always dominates itself
-      DomSetType &BBSet = Doms[BB];
-      if (BBSet != WorkingSet) {
-       BBSet.swap(WorkingSet);        // Constant time operation!
-       Changed = true;                // The sets changed.
-      }
-      WorkingSet.clear();              // Clear out the set for next iteration
-    }
-  } while (Changed);
+bool DominatorTree::runOnFunction(Function &F) {
+  DT->recalculate(F);
+  
+  return false;
 }
 
-// Postdominator set constructor.  This ctor converts the specified function to
-// only have a single exit node (return stmt), then calculates the post
-// dominance sets for the function.
-//
-void DominatorSet::calcPostDominatorSet(Function *F) {
-  // Since we require that the unify all exit nodes pass has been run, we know
-  // that there can be at most one return instruction in the function left.
-  // Get it.
-  //
-  Root = getAnalysis<UnifyFunctionExitNodes>().getExitNode();
+//===----------------------------------------------------------------------===//
+//  DominanceFrontier Implementation
+//===----------------------------------------------------------------------===//
 
-  if (Root == 0) {  // No exit node for the function?  Postdomsets are all empty
-    for (Function::iterator FI = F->begin(), FE = F->end(); FI != FE; ++FI)
-      Doms[*FI] = DomSetType();
+char DominanceFrontier::ID = 0;
+static RegisterPass<DominanceFrontier>
+G("domfrontier", "Dominance Frontier Construction", true);
+
+// NewBB is split and now it has one successor. Update dominace frontier to
+// reflect this change.
+void DominanceFrontier::splitBlock(BasicBlock *NewBB) {
+  assert(NewBB->getTerminator()->getNumSuccessors() == 1
+         && "NewBB should have a single successor!");
+  BasicBlock *NewBBSucc = NewBB->getTerminator()->getSuccessor(0);
+
+  std::vector<BasicBlock*> PredBlocks;
+  for (pred_iterator PI = pred_begin(NewBB), PE = pred_end(NewBB);
+       PI != PE; ++PI)
+      PredBlocks.push_back(*PI);  
+
+  if (PredBlocks.empty())
+    // If NewBB does not have any predecessors then it is a entry block.
+    // In this case, NewBB and its successor NewBBSucc dominates all
+    // other blocks.
     return;
-  }
 
-  bool Changed;
-  do {
-    Changed = false;
-
-    set<const BasicBlock*> Visited;
-    DomSetType WorkingSet;
-    idf_iterator<BasicBlock*> It = idf_begin(Root), End = idf_end(Root);
-    for ( ; It != End; ++It) {
-      BasicBlock *BB = *It;
-      succ_iterator PI = succ_begin(BB), PEnd = succ_end(BB);
-      if (PI != PEnd) {                // Is there SOME predecessor?
-       // Loop until we get to a successor that has had it's dom set filled
-       // in at least once.  We are guaranteed to have this because we are
-       // traversing the graph in DFO and have handled start nodes specially.
-       //
-       while (Doms[*PI].size() == 0) ++PI;
-       WorkingSet = Doms[*PI];
+  // NewBBSucc inherits original NewBB frontier.
+  DominanceFrontier::iterator NewBBI = find(NewBB);
+  if (NewBBI != end()) {
+    DominanceFrontier::DomSetType NewBBSet = NewBBI->second;
+    DominanceFrontier::DomSetType NewBBSuccSet;
+    NewBBSuccSet.insert(NewBBSet.begin(), NewBBSet.end());
+    addBasicBlock(NewBBSucc, NewBBSuccSet);
+  }
 
-       for (++PI; PI != PEnd; ++PI) { // Intersect all of the successor sets
-         DomSetType &PredSet = Doms[*PI];
-         if (PredSet.size())
-           set_intersect(WorkingSet, PredSet);
-       }
-      }
-       
-      WorkingSet.insert(BB);           // A block always dominates itself
-      DomSetType &BBSet = Doms[BB];
-      if (BBSet != WorkingSet) {
-       BBSet.swap(WorkingSet);        // Constant time operation!
-       Changed = true;                // The sets changed.
+  // If NewBB dominates NewBBSucc, then DF(NewBB) is now going to be the
+  // DF(PredBlocks[0]) without the stuff that the new block does not dominate
+  // a predecessor of.
+  DominatorTree &DT = getAnalysis<DominatorTree>();
+  if (DT.dominates(NewBB, NewBBSucc)) {
+    DominanceFrontier::iterator DFI = find(PredBlocks[0]);
+    if (DFI != end()) {
+      DominanceFrontier::DomSetType Set = DFI->second;
+      // Filter out stuff in Set that we do not dominate a predecessor of.
+      for (DominanceFrontier::DomSetType::iterator SetI = Set.begin(),
+             E = Set.end(); SetI != E;) {
+        bool DominatesPred = false;
+        for (pred_iterator PI = pred_begin(*SetI), E = pred_end(*SetI);
+             PI != E; ++PI)
+          if (DT.dominates(NewBB, *PI))
+            DominatesPred = true;
+        if (!DominatesPred)
+          Set.erase(SetI++);
+        else
+          ++SetI;
       }
-      WorkingSet.clear();              // Clear out the set for next iteration
-    }
-  } while (Changed);
-}
 
-// getAnalysisUsage - This obviously provides a dominator set, but it also
-// uses the UnifyFunctionExitNodes pass if building post-dominators
-//
-void DominatorSet::getAnalysisUsage(AnalysisUsage &AU) const {
-  AU.setPreservesAll();
-  if (isPostDominator()) {
-    AU.addProvided(PostDomID);
-    AU.addRequired(UnifyFunctionExitNodes::ID);
+      if (NewBBI != end()) {
+        for (DominanceFrontier::DomSetType::iterator SetI = Set.begin(),
+               E = Set.end(); SetI != E; ++SetI) {
+          BasicBlock *SB = *SetI;
+          addToFrontier(NewBBI, SB);
+        }
+      } else 
+        addBasicBlock(NewBB, Set);
+    }
+    
   } else {
-    AU.addProvided(ID);
+    // DF(NewBB) is {NewBBSucc} because NewBB does not strictly dominate
+    // NewBBSucc, but it does dominate itself (and there is an edge (NewBB ->
+    // NewBBSucc)).  NewBBSucc is the single successor of NewBB.
+    DominanceFrontier::DomSetType NewDFSet;
+    NewDFSet.insert(NewBBSucc);
+    addBasicBlock(NewBB, NewDFSet);
   }
-}
-
-
-//===----------------------------------------------------------------------===//
-//  ImmediateDominators Implementation
-//===----------------------------------------------------------------------===//
-
-AnalysisID ImmediateDominators::ID(AnalysisID::create<ImmediateDominators>());
-AnalysisID ImmediateDominators::PostDomID(AnalysisID::create<ImmediateDominators>());
-
-// calcIDoms - Calculate the immediate dominator mapping, given a set of
-// dominators for every basic block.
-void ImmediateDominators::calcIDoms(const DominatorSet &DS) {
-  // Loop over all of the nodes that have dominators... figuring out the IDOM
-  // for each node...
+  
+  // Now we must loop over all of the dominance frontiers in the function,
+  // replacing occurrences of NewBBSucc with NewBB in some cases.  All
+  // blocks that dominate a block in PredBlocks and contained NewBBSucc in
+  // their dominance frontier must be updated to contain NewBB instead.
   //
-  for (DominatorSet::const_iterator DI = DS.begin(), DEnd = DS.end(); 
-       DI != DEnd; ++DI) {
-    BasicBlock *BB = DI->first;
-    const DominatorSet::DomSetType &Dominators = DI->second;
-    unsigned DomSetSize = Dominators.size();
-    if (DomSetSize == 1) continue;  // Root node... IDom = null
-
-    // Loop over all dominators of this node.  This corresponds to looping over
-    // nodes in the dominator chain, looking for a node whose dominator set is
-    // equal to the current nodes, except that the current node does not exist
-    // in it.  This means that it is one level higher in the dom chain than the
-    // current node, and it is our idom!
-    //
-    DominatorSet::DomSetType::const_iterator I = Dominators.begin();
-    DominatorSet::DomSetType::const_iterator End = Dominators.end();
-    for (; I != End; ++I) {   // Iterate over dominators...
-      // All of our dominators should form a chain, where the number of elements
-      // in the dominator set indicates what level the node is at in the chain.
-      // We want the node immediately above us, so it will have an identical 
-      // dominator set, except that BB will not dominate it... therefore it's
-      // dominator set size will be one less than BB's...
-      //
-      if (DS.getDominators(*I).size() == DomSetSize - 1) {
-       IDoms[BB] = *I;
-       break;
+  for (Function::iterator FI = NewBB->getParent()->begin(),
+         FE = NewBB->getParent()->end(); FI != FE; ++FI) {
+    DominanceFrontier::iterator DFI = find(FI);
+    if (DFI == end()) continue;  // unreachable block.
+    
+    // Only consider nodes that have NewBBSucc in their dominator frontier.
+    if (!DFI->second.count(NewBBSucc)) continue;
+
+    // Verify whether this block dominates a block in predblocks.  If not, do
+    // not update it.
+    bool BlockDominatesAny = false;
+    for (std::vector<BasicBlock*>::const_iterator BI = PredBlocks.begin(), 
+           BE = PredBlocks.end(); BI != BE; ++BI) {
+      if (DT.dominates(FI, *BI)) {
+        BlockDominatesAny = true;
+        break;
       }
     }
+    
+    if (!BlockDominatesAny)
+      continue;
+    
+    // If NewBBSucc should not stay in our dominator frontier, remove it.
+    // We remove it unless there is a predecessor of NewBBSucc that we
+    // dominate, but we don't strictly dominate NewBBSucc.
+    bool ShouldRemove = true;
+    if ((BasicBlock*)FI == NewBBSucc || !DT.dominates(FI, NewBBSucc)) {
+      // Okay, we know that PredDom does not strictly dominate NewBBSucc.
+      // Check to see if it dominates any predecessors of NewBBSucc.
+      for (pred_iterator PI = pred_begin(NewBBSucc),
+           E = pred_end(NewBBSucc); PI != E; ++PI)
+        if (DT.dominates(FI, *PI)) {
+          ShouldRemove = false;
+          break;
+        }
+    }
+    
+    if (ShouldRemove)
+      removeFromFrontier(DFI, NewBBSucc);
+    addToFrontier(DFI, NewBB);
   }
 }
 
-
-//===----------------------------------------------------------------------===//
-//  DominatorTree Implementation
-//===----------------------------------------------------------------------===//
-
-AnalysisID DominatorTree::ID(AnalysisID::create<DominatorTree>());
-AnalysisID DominatorTree::PostDomID(AnalysisID::create<DominatorTree>());
-
-// DominatorTree::reset - Free all of the tree node memory.
-//
-void DominatorTree::reset() { 
-  for (NodeMapType::iterator I = Nodes.begin(), E = Nodes.end(); I != E; ++I)
-    delete I->second;
-  Nodes.clear();
+namespace {
+  class DFCalculateWorkObject {
+  public:
+    DFCalculateWorkObject(BasicBlock *B, BasicBlock *P, 
+                          const DomTreeNode *N,
+                          const DomTreeNode *PN)
+    : currentBB(B), parentBB(P), Node(N), parentNode(PN) {}
+    BasicBlock *currentBB;
+    BasicBlock *parentBB;
+    const DomTreeNode *Node;
+    const DomTreeNode *parentNode;
+  };
 }
 
+const DominanceFrontier::DomSetType &
+DominanceFrontier::calculate(const DominatorTree &DT,
+                             const DomTreeNode *Node) {
+  BasicBlock *BB = Node->getBlock();
+  DomSetType *Result = NULL;
 
-#if 0
-// Given immediate dominators, we can also calculate the dominator tree
-DominatorTree::DominatorTree(const ImmediateDominators &IDoms) 
-  : DominatorBase(IDoms.getRoot()) {
-  const Function *M = Root->getParent();
-
-  Nodes[Root] = new Node(Root, 0);   // Add a node for the root...
-
-  // Iterate over all nodes in depth first order...
-  for (df_iterator<const Function*> I = df_begin(M), E = df_end(M); I!=E; ++I) {
-    const BasicBlock *BB = *I, *IDom = IDoms[*I];
-
-    if (IDom != 0) {   // Ignore the root node and other nasty nodes
-      // We know that the immediate dominator should already have a node, 
-      // because we are traversing the CFG in depth first order!
-      //
-      assert(Nodes[IDom] && "No node for IDOM?");
-      Node *IDomNode = Nodes[IDom];
-
-      // Add a new tree node for this BasicBlock, and link it as a child of
-      // IDomNode
-      Nodes[BB] = IDomNode->addChild(new Node(BB, IDomNode));
-    }
-  }
-}
-#endif
-
-void DominatorTree::calculate(const DominatorSet &DS) {
-  Nodes[Root] = new Node(Root, 0);   // Add a node for the root...
+  std::vector<DFCalculateWorkObject> workList;
+  SmallPtrSet<BasicBlock *, 32> visited;
 
-  if (!isPostDominator()) {
-    // Iterate over all nodes in depth first order...
-    for (df_iterator<BasicBlock*> I = df_begin(Root), E = df_end(Root);
-         I != E; ++I) {
-      BasicBlock *BB = *I;
-      const DominatorSet::DomSetType &Dominators = DS.getDominators(BB);
-      unsigned DomSetSize = Dominators.size();
-      if (DomSetSize == 1) continue;  // Root node... IDom = null
-      
-      // Loop over all dominators of this node. This corresponds to looping over
-      // nodes in the dominator chain, looking for a node whose dominator set is
-      // equal to the current nodes, except that the current node does not exist
-      // in it. This means that it is one level higher in the dom chain than the
-      // current node, and it is our idom!  We know that we have already added
-      // a DominatorTree node for our idom, because the idom must be a
-      // predecessor in the depth first order that we are iterating through the
-      // function.
-      //
-      DominatorSet::DomSetType::const_iterator I = Dominators.begin();
-      DominatorSet::DomSetType::const_iterator End = Dominators.end();
-      for (; I != End; ++I) {   // Iterate over dominators...
-       // All of our dominators should form a chain, where the number of
-       // elements in the dominator set indicates what level the node is at in
-       // the chain.  We want the node immediately above us, so it will have
-       // an identical dominator set, except that BB will not dominate it...
-       // therefore it's dominator set size will be one less than BB's...
-       //
-       if (DS.getDominators(*I).size() == DomSetSize - 1) {
-         // We know that the immediate dominator should already have a node, 
-         // because we are traversing the CFG in depth first order!
-         //
-         Node *IDomNode = Nodes[*I];
-         assert(IDomNode && "No node for IDOM?");
-         
-         // Add a new tree node for this BasicBlock, and link it as a child of
-         // IDomNode
-         Nodes[BB] = IDomNode->addChild(new Node(BB, IDomNode));
-         break;
-       }
+  workList.push_back(DFCalculateWorkObject(BB, NULL, Node, NULL));
+  do {
+    DFCalculateWorkObject *currentW = &workList.back();
+    assert (currentW && "Missing work object.");
+
+    BasicBlock *currentBB = currentW->currentBB;
+    BasicBlock *parentBB = currentW->parentBB;
+    const DomTreeNode *currentNode = currentW->Node;
+    const DomTreeNode *parentNode = currentW->parentNode;
+    assert (currentBB && "Invalid work object. Missing current Basic Block");
+    assert (currentNode && "Invalid work object. Missing current Node");
+    DomSetType &S = Frontiers[currentBB];
+
+    // Visit each block only once.
+    if (visited.count(currentBB) == 0) {
+      visited.insert(currentBB);
+
+      // Loop over CFG successors to calculate DFlocal[currentNode]
+      for (succ_iterator SI = succ_begin(currentBB), SE = succ_end(currentBB);
+           SI != SE; ++SI) {
+        // Does Node immediately dominate this successor?
+        if (DT[*SI]->getIDom() != currentNode)
+          S.insert(*SI);
       }
     }
-  } else if (Root) {
-    // Iterate over all nodes in depth first order...
-    for (idf_iterator<BasicBlock*> I = idf_begin(Root), E = idf_end(Root);
-         I != E; ++I) {
-      BasicBlock *BB = *I;
-      const DominatorSet::DomSetType &Dominators = DS.getDominators(BB);
-      unsigned DomSetSize = Dominators.size();
-      if (DomSetSize == 1) continue;  // Root node... IDom = null
-      
-      // Loop over all dominators of this node.  This corresponds to looping
-      // over nodes in the dominator chain, looking for a node whose dominator
-      // set is equal to the current nodes, except that the current node does
-      // not exist in it.  This means that it is one level higher in the dom
-      // chain than the current node, and it is our idom!  We know that we have
-      // already added a DominatorTree node for our idom, because the idom must
-      // be a predecessor in the depth first order that we are iterating through
-      // the function.
-      //
-      DominatorSet::DomSetType::const_iterator I = Dominators.begin();
-      DominatorSet::DomSetType::const_iterator End = Dominators.end();
-      for (; I != End; ++I) {   // Iterate over dominators...
-       // All of our dominators should form a chain, where the number
-       // of elements in the dominator set indicates what level the
-       // node is at in the chain.  We want the node immediately
-       // above us, so it will have an identical dominator set,
-       // except that BB will not dominate it... therefore it's
-       // dominator set size will be one less than BB's...
-       //
-       if (DS.getDominators(*I).size() == DomSetSize - 1) {
-         // We know that the immediate dominator should already have a node, 
-         // because we are traversing the CFG in depth first order!
-         //
-         Node *IDomNode = Nodes[*I];
-         assert(IDomNode && "No node for IDOM?");
-         
-         // Add a new tree node for this BasicBlock, and link it as a child of
-         // IDomNode
-         Nodes[BB] = IDomNode->addChild(new Node(BB, IDomNode));
-         break;
-       }
+
+    // At this point, S is DFlocal.  Now we union in DFup's of our children...
+    // Loop through and visit the nodes that Node immediately dominates (Node's
+    // children in the IDomTree)
+    bool visitChild = false;
+    for (DomTreeNode::const_iterator NI = currentNode->begin(), 
+           NE = currentNode->end(); NI != NE; ++NI) {
+      DomTreeNode *IDominee = *NI;
+      BasicBlock *childBB = IDominee->getBlock();
+      if (visited.count(childBB) == 0) {
+        workList.push_back(DFCalculateWorkObject(childBB, currentBB,
+                                                 IDominee, currentNode));
+        visitChild = true;
       }
     }
-  }
-}
 
+    // If all children are visited or there is any child then pop this block
+    // from the workList.
+    if (!visitChild) {
 
+      if (!parentBB) {
+        Result = &S;
+        break;
+      }
 
-//===----------------------------------------------------------------------===//
-//  DominanceFrontier Implementation
-//===----------------------------------------------------------------------===//
-
-AnalysisID DominanceFrontier::ID(AnalysisID::create<DominanceFrontier>());
-AnalysisID DominanceFrontier::PostDomID(AnalysisID::create<DominanceFrontier>());
-
-const DominanceFrontier::DomSetType &
-DominanceFrontier::calcDomFrontier(const DominatorTree &DT, 
-                                   const DominatorTree::Node *Node) {
-  // Loop over CFG successors to calculate DFlocal[Node]
-  BasicBlock *BB = Node->getNode();
-  DomSetType &S = Frontiers[BB];       // The new set to fill in...
-
-  for (succ_iterator SI = succ_begin(BB), SE = succ_end(BB);
-       SI != SE; ++SI) {
-    // Does Node immediately dominate this successor?
-    if (DT[*SI]->getIDom() != Node)
-      S.insert(*SI);
-  }
-
-  // At this point, S is DFlocal.  Now we union in DFup's of our children...
-  // Loop through and visit the nodes that Node immediately dominates (Node's
-  // children in the IDomTree)
-  //
-  for (DominatorTree::Node::const_iterator NI = Node->begin(), NE = Node->end();
-       NI != NE; ++NI) {
-    DominatorTree::Node *IDominee = *NI;
-    const DomSetType &ChildDF = calcDomFrontier(DT, IDominee);
-
-    DomSetType::const_iterator CDFI = ChildDF.begin(), CDFE = ChildDF.end();
-    for (; CDFI != CDFE; ++CDFI) {
-      if (!Node->dominates(DT[*CDFI]))
-       S.insert(*CDFI);
+      DomSetType::const_iterator CDFI = S.begin(), CDFE = S.end();
+      DomSetType &parentSet = Frontiers[parentBB];
+      for (; CDFI != CDFE; ++CDFI) {
+        if (!DT.properlyDominates(parentNode, DT[*CDFI]))
+          parentSet.insert(*CDFI);
+      }
+      workList.pop_back();
     }
-  }
-
-  return S;
-}
-
-const DominanceFrontier::DomSetType &
-DominanceFrontier::calcPostDomFrontier(const DominatorTree &DT, 
-                                       const DominatorTree::Node *Node) {
-  // Loop over CFG successors to calculate DFlocal[Node]
-  BasicBlock *BB = Node->getNode();
-  DomSetType &S = Frontiers[BB];       // The new set to fill in...
-  if (!Root) return S;
 
-  for (pred_iterator SI = pred_begin(BB), SE = pred_end(BB);
-       SI != SE; ++SI) {
-    // Does Node immediately dominate this predeccessor?
-    if (DT[*SI]->getIDom() != Node)
-      S.insert(*SI);
-  }
+  } while (!workList.empty());
 
-  // At this point, S is DFlocal.  Now we union in DFup's of our children...
-  // Loop through and visit the nodes that Node immediately dominates (Node's
-  // children in the IDomTree)
-  //
-  for (DominatorTree::Node::const_iterator NI = Node->begin(), NE = Node->end();
-       NI != NE; ++NI) {
-    DominatorTree::Node *IDominee = *NI;
-    const DomSetType &ChildDF = calcPostDomFrontier(DT, IDominee);
+  return *Result;
+}
 
-    DomSetType::const_iterator CDFI = ChildDF.begin(), CDFE = ChildDF.end();
-    for (; CDFI != CDFE; ++CDFI) {
-      if (!Node->dominates(DT[*CDFI]))
-       S.insert(*CDFI);
-    }
+void DominanceFrontierBase::print(std::ostream &o, const Module* ) const {
+  for (const_iterator I = begin(), E = end(); I != E; ++I) {
+    o << "  DomFrontier for BB";
+    if (I->first)
+      WriteAsOperand(o, I->first, false);
+    else
+      o << " <<exit node>>";
+    o << " is:\t" << I->second << "\n";
   }
+}
 
-  return S;
+void DominanceFrontierBase::dump() {
+  print (llvm::cerr);
 }