StmtViz.cpp revision 198092
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 15#include "clang/AST/StmtGraphTraits.h" 16#include "clang/AST/Decl.h" 17#include "llvm/Support/GraphWriter.h" 18 19using namespace clang; 20 21void 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 30namespace llvm { 31template<> 32struct DOTGraphTraits<const Stmt*> : public DefaultDOTGraphTraits { 33 static std::string getNodeLabel(const Stmt* Node, const Stmt* Graph, 34 bool ShortNames) { 35 36#ifndef NDEBUG 37 std::string OutSStr; 38 llvm::raw_string_ostream Out(OutSStr); 39 40 if (Node) 41 Out << Node->getStmtClassName(); 42 else 43 Out << "<NULL>"; 44 45 std::string OutStr = Out.str(); 46 if (OutStr[0] == '\n') OutStr.erase(OutStr.begin()); 47 48 // Process string output to make it nicer... 49 for (unsigned i = 0; i != OutStr.length(); ++i) 50 if (OutStr[i] == '\n') { // Left justify 51 OutStr[i] = '\\'; 52 OutStr.insert(OutStr.begin()+i+1, 'l'); 53 } 54 55 return OutStr; 56#else 57 return ""; 58#endif 59 } 60}; 61} // end namespace llvm 62