clang  6.0.0svn
StmtViz.cpp
Go to the documentation of this file.
1 //===--- StmtViz.cpp - Graphviz visualization for Stmt ASTs -----*- C++ -*-===//
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 Stmt::viewAST, which generates a Graphviz DOT file
11 // that depicts the AST and then calls Graphviz/dot+gv on it.
12 //
13 //===----------------------------------------------------------------------===//
14 
16 #include "clang/AST/Decl.h"
17 #include "llvm/Support/GraphWriter.h"
18 
19 using namespace clang;
20 
21 void Stmt::viewAST() const {
22 #ifndef NDEBUG
23  llvm::ViewGraph(this,"AST");
24 #else
25  llvm::errs() << "Stmt::viewAST is only available in debug builds on "
26  << "systems with Graphviz or gv!\n";
27 #endif
28 }
29 
30 namespace llvm {
31 template<>
32 struct DOTGraphTraits<const Stmt*> : public DefaultDOTGraphTraits {
33  DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
34 
35  static std::string getNodeLabel(const Stmt* Node, const Stmt* Graph) {
36 
37 #ifndef NDEBUG
38  std::string OutSStr;
39  llvm::raw_string_ostream Out(OutSStr);
40 
41  if (Node)
42  Out << Node->getStmtClassName();
43  else
44  Out << "<NULL>";
45 
46  std::string OutStr = Out.str();
47  if (OutStr[0] == '\n') OutStr.erase(OutStr.begin());
48 
49  // Process string output to make it nicer...
50  for (unsigned i = 0; i != OutStr.length(); ++i)
51  if (OutStr[i] == '\n') { // Left justify
52  OutStr[i] = '\\';
53  OutStr.insert(OutStr.begin()+i+1, 'l');
54  }
55 
56  return OutStr;
57 #else
58  return "";
59 #endif
60  }
61 };
62 } // end namespace llvm
DominatorTree GraphTraits specialization so the DominatorTree can be iterable by generic graph iterat...
Definition: Dominators.h:26
Stmt - This represents one statement.
Definition: Stmt.h:66
const char * getStmtClassName() const
Definition: Stmt.cpp:74
static std::string getNodeLabel(const Stmt *Node, const Stmt *Graph)
Definition: StmtViz.cpp:35
ast_type_traits::DynTypedNode Node
Dataflow Directional Tag Classes.
DOTGraphTraits(bool isSimple=false)
Definition: StmtViz.cpp:33
void viewAST() const
viewAST - Visualize an AST rooted at this Stmt* using GraphViz.
Definition: StmtViz.cpp:21