lib/Analysis/PostDominators.cpp

   1 //===- PostDominators.cpp - Post-Dominator Calculation --------------------===//
   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 // This file implements the post-dominator construction algorithms.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #define DEBUG_TYPE "postdomtree"
  15
  16 #include "llvm/Analysis/PostDominators.h"
  17 #include "llvm/Instructions.h"
  18 #include "llvm/Support/CFG.h"
  19 #include "llvm/Support/Debug.h"
  20 #include "llvm/ADT/DepthFirstIterator.h"
  21 #include "llvm/ADT/SetOperations.h"
  22 #include "llvm/Analysis/DominatorInternals.h"
  23 using namespace llvm;
  24
  25 //===----------------------------------------------------------------------===//
  26 //  PostDominatorTree Implementation
  27 //===----------------------------------------------------------------------===//
  28
  29 char PostDominatorTree::ID = 0;
  30 char PostDominanceFrontier::ID = 0;
  31 static RegisterPass<PostDominatorTree>
  32 F("postdomtree", "Post-Dominator Tree Construction", true, true);
  33
  34 bool PostDominatorTree::runOnFunction(Function &F) {
  35   DT->recalculate(F);
  36   DEBUG(DT->print(errs()));
  37   return false;
  38 }
  39
  40 PostDominatorTree::~PostDominatorTree() {
  41   delete DT;
  42 }
  43
  44 void PostDominatorTree::print(std::ostream &OS, const Module *) const {
  45   raw_os_ostream OSS(OS);
  46   DT->print(OSS);
  47 }
  48
  49
  50 FunctionPass* llvm::createPostDomTree() {
  51   return new PostDominatorTree();
  52 }
  53
  54 //===----------------------------------------------------------------------===//
  55 //  PostDominanceFrontier Implementation
  56 //===----------------------------------------------------------------------===//
  57
  58 static RegisterPass<PostDominanceFrontier>
  59 H("postdomfrontier", "Post-Dominance Frontier Construction", true, 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 FunctionPass* llvm::createPostDomFrontier() {
  98   return new PostDominanceFrontier();
  99 }