//===--- 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