| //===--- StmtViz.cpp - Graphviz visualization for Stmt ASTs -----*- C++ -*-===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This file implements Stmt::viewAST, which generates a Graphviz DOT file |
| // that depicts the AST and then calls Graphviz/dot+gv on it. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "clang/AST/StmtGraphTraits.h" |
| #include "clang/AST/Decl.h" |
| #include "llvm/Support/GraphWriter.h" |
| |
| using namespace clang; |
| |
| void Stmt::viewAST() const { |
| #ifndef NDEBUG |
| llvm::ViewGraph(this,"AST"); |
| #else |
| llvm::errs() << "Stmt::viewAST is only available in debug builds on " |
| << "systems with Graphviz or gv!\n"; |
| #endif |
| } |
| |
| namespace llvm { |
| template<> |
| struct DOTGraphTraits<const Stmt*> : public DefaultDOTGraphTraits { |
| DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {} |
| |
| static std::string getNodeLabel(const Stmt* Node, const Stmt* Graph) { |
| |
| #ifndef NDEBUG |
| std::string OutSStr; |
| llvm::raw_string_ostream Out(OutSStr); |
| |
| if (Node) |
| Out << Node->getStmtClassName(); |
| else |
| Out << "<NULL>"; |
| |
| std::string OutStr = Out.str(); |
| if (OutStr[0] == '\n') OutStr.erase(OutStr.begin()); |
| |
| // Process string output to make it nicer... |
| for (unsigned i = 0; i != OutStr.length(); ++i) |
| if (OutStr[i] == '\n') { // Left justify |
| OutStr[i] = '\\'; |
| OutStr.insert(OutStr.begin()+i+1, 'l'); |
| } |
| |
| return OutStr; |
| #else |
| return ""; |
| #endif |
| } |
| }; |
| } // end namespace llvm |