Break DominatorTree from ETNode.
[oota-llvm.git] / include / llvm / Analysis / PostDominators.h
1 //=- llvm/Analysis/PostDominators.h - Post Dominator Calculation-*- C++ -*-===//
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 exposes interfaces to post dominance information.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #ifndef LLVM_ANALYSIS_POST_DOMINATORS_H
15 #define LLVM_ANALYSIS_POST_DOMINATORS_H
16
17 #include "llvm/Analysis/Dominators.h"
18
19 namespace llvm {
20
21 /// PostDominatorTree Class - Concrete subclass of DominatorTree that is used to
22 /// compute the a post-dominator tree.
23 ///
24 struct PostDominatorTree : public DominatorTreeBase {
25   static char ID; // Pass identification, replacement for typeid
26
27   PostDominatorTree() : 
28     DominatorTreeBase((intptr_t)&ID, true) {}
29
30   virtual bool runOnFunction(Function &F) {
31     reset();     // Reset from the last time we were run...
32     calculate(F);
33     return false;
34   }
35
36   virtual void getAnalysisUsage(AnalysisUsage &AU) const {
37     AU.setPreservesAll();
38   }
39 private:
40   void calculate(Function &F);
41   DomTreeNode *getNodeForBlock(BasicBlock *BB);
42   unsigned DFSPass(BasicBlock *V, InfoRec &VInfo,unsigned N);
43   void Compress(BasicBlock *V, InfoRec &VInfo);
44   BasicBlock *Eval(BasicBlock *V);
45   void Link(BasicBlock *V, BasicBlock *W, InfoRec &WInfo);
46
47   inline BasicBlock *getIDom(BasicBlock *BB) const {
48     std::map<BasicBlock*, BasicBlock*>::const_iterator I = IDoms.find(BB);
49     return I != IDoms.end() ? I->second : 0;
50   }
51 };
52
53
54 /// PostDominanceFrontier Class - Concrete subclass of DominanceFrontier that is
55 /// used to compute the a post-dominance frontier.
56 ///
57 struct PostDominanceFrontier : public DominanceFrontierBase {
58   static char ID;
59   PostDominanceFrontier() 
60     : DominanceFrontierBase((intptr_t) &ID, true) {}
61
62   virtual bool runOnFunction(Function &) {
63     Frontiers.clear();
64     PostDominatorTree &DT = getAnalysis<PostDominatorTree>();
65     Roots = DT.getRoots();
66     if (const DomTreeNode *Root = DT.getRootNode())
67       calculate(DT, Root);
68     return false;
69   }
70
71   virtual void getAnalysisUsage(AnalysisUsage &AU) const {
72     AU.setPreservesAll();
73     AU.addRequired<PostDominatorTree>();
74   }
75
76 private:
77   const DomSetType &calculate(const PostDominatorTree &DT,
78                               const DomTreeNode *Node);
79 };
80
81 } // End llvm namespace
82
83 // Make sure that any clients of this file link in PostDominators.cpp
84 FORCE_DEFINING_FILE_TO_BE_LINKED(PostDominanceFrontier)
85
86 #endif