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  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
63