1 //===- DomPrinter.cpp - DOT printer for the dominance trees ------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file defines '-dot-dom' and '-dot-postdom' analysis passes, which emit
11 // a dom.<fnname>.dot or postdom.<fnname>.dot file for each function in the
12 // program, with a graph of the dominance/postdominance tree of that
15 // There are also passes available to directly call dotty ('-view-dom' or
16 // '-view-postdom'). By appending '-only' like '-dot-dom-only' only the
17 // names of the bbs are printed, but the content is hidden.
19 //===----------------------------------------------------------------------===//
21 #include "llvm/Analysis/DomPrinter.h"
22 #include "llvm/Analysis/DOTGraphTraitsPass.h"
23 #include "llvm/Analysis/PostDominators.h"
29 struct DOTGraphTraits<DomTreeNode*> : public DefaultDOTGraphTraits {
31 DOTGraphTraits (bool isSimple=false)
32 : DefaultDOTGraphTraits(isSimple) {}
34 std::string getNodeLabel(DomTreeNode *Node, DomTreeNode *Graph) {
36 BasicBlock *BB = Node->getBlock();
39 return "Post dominance root node";
43 return DOTGraphTraits<const Function*>
44 ::getSimpleNodeLabel(BB, BB->getParent());
46 return DOTGraphTraits<const Function*>
47 ::getCompleteNodeLabel(BB, BB->getParent());
52 struct DOTGraphTraits<DominatorTree*> : public DOTGraphTraits<DomTreeNode*> {
54 DOTGraphTraits (bool isSimple=false)
55 : DOTGraphTraits<DomTreeNode*>(isSimple) {}
57 static std::string getGraphName(DominatorTree *DT) {
58 return "Dominator tree";
61 std::string getNodeLabel(DomTreeNode *Node, DominatorTree *G) {
62 return DOTGraphTraits<DomTreeNode*>::getNodeLabel(Node, G->getRootNode());
67 struct DOTGraphTraits<PostDominatorTree*>
68 : public DOTGraphTraits<DomTreeNode*> {
70 DOTGraphTraits (bool isSimple=false)
71 : DOTGraphTraits<DomTreeNode*>(isSimple) {}
73 static std::string getGraphName(PostDominatorTree *DT) {
74 return "Post dominator tree";
77 std::string getNodeLabel(DomTreeNode *Node, PostDominatorTree *G ) {
78 return DOTGraphTraits<DomTreeNode*>::getNodeLabel(Node, G->getRootNode());
84 struct DominatorTreeWrapperPassAnalysisGraphTraits {
85 static DominatorTree *getGraph(DominatorTreeWrapperPass *DTWP) {
86 return &DTWP->getDomTree();
90 struct DomViewer : public DOTGraphTraitsViewer<
91 DominatorTreeWrapperPass, false, DominatorTree *,
92 DominatorTreeWrapperPassAnalysisGraphTraits> {
95 : DOTGraphTraitsViewer<DominatorTreeWrapperPass, false, DominatorTree *,
96 DominatorTreeWrapperPassAnalysisGraphTraits>(
98 initializeDomViewerPass(*PassRegistry::getPassRegistry());
102 struct DomOnlyViewer : public DOTGraphTraitsViewer<
103 DominatorTreeWrapperPass, true, DominatorTree *,
104 DominatorTreeWrapperPassAnalysisGraphTraits> {
107 : DOTGraphTraitsViewer<DominatorTreeWrapperPass, true, DominatorTree *,
108 DominatorTreeWrapperPassAnalysisGraphTraits>(
110 initializeDomOnlyViewerPass(*PassRegistry::getPassRegistry());
115 : public DOTGraphTraitsViewer<PostDominatorTree, false> {
118 DOTGraphTraitsViewer<PostDominatorTree, false>("postdom", ID){
119 initializePostDomViewerPass(*PassRegistry::getPassRegistry());
123 struct PostDomOnlyViewer
124 : public DOTGraphTraitsViewer<PostDominatorTree, true> {
126 PostDomOnlyViewer() :
127 DOTGraphTraitsViewer<PostDominatorTree, true>("postdomonly", ID){
128 initializePostDomOnlyViewerPass(*PassRegistry::getPassRegistry());
131 } // end anonymous namespace
133 char DomViewer::ID = 0;
134 INITIALIZE_PASS(DomViewer, "view-dom",
135 "View dominance tree of function", false, false)
137 char DomOnlyViewer::ID = 0;
138 INITIALIZE_PASS(DomOnlyViewer, "view-dom-only",
139 "View dominance tree of function (with no function bodies)",
142 char PostDomViewer::ID = 0;
143 INITIALIZE_PASS(PostDomViewer, "view-postdom",
144 "View postdominance tree of function", false, false)
146 char PostDomOnlyViewer::ID = 0;
147 INITIALIZE_PASS(PostDomOnlyViewer, "view-postdom-only",
148 "View postdominance tree of function "
149 "(with no function bodies)",
153 struct DomPrinter : public DOTGraphTraitsPrinter<
154 DominatorTreeWrapperPass, false, DominatorTree *,
155 DominatorTreeWrapperPassAnalysisGraphTraits> {
158 : DOTGraphTraitsPrinter<DominatorTreeWrapperPass, false, DominatorTree *,
159 DominatorTreeWrapperPassAnalysisGraphTraits>(
161 initializeDomPrinterPass(*PassRegistry::getPassRegistry());
165 struct DomOnlyPrinter : public DOTGraphTraitsPrinter<
166 DominatorTreeWrapperPass, true, DominatorTree *,
167 DominatorTreeWrapperPassAnalysisGraphTraits> {
170 : DOTGraphTraitsPrinter<DominatorTreeWrapperPass, true, DominatorTree *,
171 DominatorTreeWrapperPassAnalysisGraphTraits>(
173 initializeDomOnlyPrinterPass(*PassRegistry::getPassRegistry());
177 struct PostDomPrinter
178 : public DOTGraphTraitsPrinter<PostDominatorTree, false> {
181 DOTGraphTraitsPrinter<PostDominatorTree, false>("postdom", ID) {
182 initializePostDomPrinterPass(*PassRegistry::getPassRegistry());
186 struct PostDomOnlyPrinter
187 : public DOTGraphTraitsPrinter<PostDominatorTree, true> {
189 PostDomOnlyPrinter() :
190 DOTGraphTraitsPrinter<PostDominatorTree, true>("postdomonly", ID) {
191 initializePostDomOnlyPrinterPass(*PassRegistry::getPassRegistry());
194 } // end anonymous namespace
198 char DomPrinter::ID = 0;
199 INITIALIZE_PASS(DomPrinter, "dot-dom",
200 "Print dominance tree of function to 'dot' file",
203 char DomOnlyPrinter::ID = 0;
204 INITIALIZE_PASS(DomOnlyPrinter, "dot-dom-only",
205 "Print dominance tree of function to 'dot' file "
206 "(with no function bodies)",
209 char PostDomPrinter::ID = 0;
210 INITIALIZE_PASS(PostDomPrinter, "dot-postdom",
211 "Print postdominance tree of function to 'dot' file",
214 char PostDomOnlyPrinter::ID = 0;
215 INITIALIZE_PASS(PostDomOnlyPrinter, "dot-postdom-only",
216 "Print postdominance tree of function to 'dot' file "
217 "(with no function bodies)",
220 // Create methods available outside of this file, to use them
221 // "include/llvm/LinkAllPasses.h". Otherwise the pass would be deleted by
222 // the link time optimization.
224 FunctionPass *llvm::createDomPrinterPass() {
225 return new DomPrinter();
228 FunctionPass *llvm::createDomOnlyPrinterPass() {
229 return new DomOnlyPrinter();
232 FunctionPass *llvm::createDomViewerPass() {
233 return new DomViewer();
236 FunctionPass *llvm::createDomOnlyViewerPass() {
237 return new DomOnlyViewer();
240 FunctionPass *llvm::createPostDomPrinterPass() {
241 return new PostDomPrinter();
244 FunctionPass *llvm::createPostDomOnlyPrinterPass() {
245 return new PostDomOnlyPrinter();
248 FunctionPass *llvm::createPostDomViewerPass() {
249 return new PostDomViewer();
252 FunctionPass *llvm::createPostDomOnlyViewerPass() {
253 return new PostDomOnlyViewer();