dot_visitor.cpp

Go to the documentation of this file.

#include "dot_visitor.hpp"
#include "ast/ast.hpp"
#include "token.hpp"
#include "util.hpp"
#include "llvm/Support/raw_ostream.h"
#include <algorithm>
#include <memory>
#include <string_view>
 
namespace yume::diagnostic {
static auto xml_escape(std::string_view data) -> std::string {
  std::string str{};
  llvm::raw_string_ostream stream(str);
 
  for (char i : data) {
    switch (i) {
    case '&': stream << "&amp;"; break;
    case '\'': stream << "&apos;"; break;
    case '\"': stream << "&quot;"; break;
    case '<': stream << "&lt;"; break;
    case '>': stream << "&gt;"; break;
    case '\0': stream << "\\\\0"; break;
    default: stream << i; break;
    }
  }
 
  return str;
}
 
inline static auto opt_str(const char* ptr) -> optional<string> {
  if (ptr == nullptr)
    return {};
  return string(ptr);
}
 
auto DotVisitor::add_node(const string& content, const char* label) -> DotNode& {
  return add_node(DotNode(m_index, std::nullopt, std::nullopt, content), label);
}
 
auto DotVisitor::add_node(DotNode&& node, const char* label) -> DotNode& {
  m_index++;
  if (m_parent == nullptr) {
    m_root = std::make_unique<DotNode>(node);
    return *m_root;
  }
  return m_parent->children.emplace_back(opt_str(label), node).child;
}
 
void DotVisitor::DotNode::write(llvm::raw_ostream& stream) const {
  stream << AST_KEY << index << " [label=<";
  if (simple()) {
    stream << "<I>" << content << "</I>";
  } else {
    stream << "<FONT POINT-SIZE=\"9\">" << location->to_string() << "</FONT><BR/>";
    if (type.has_value())
      stream << "<U>" << *type << "</U><BR/>";
    stream << "<B>" << content << "</B>";
  }
 
  bool skip_first_child = false;
  if (!children.empty()) {
    const auto& first = children.at(0);
    if (!first.line_label.has_value() && first.child.simple()) {
      skip_first_child = true;
      stream << "<BR/><I>" << first.child.content << "</I>";
    }
  }
 
  stream << ">];\n";
 
  for (const auto& [line_label, child] : children) {
    if (skip_first_child) {
      skip_first_child = false;
      continue;
    }
    child.write(stream);
    stream << AST_KEY << index << " -> " << AST_KEY << child.index;
    if (line_label.has_value())
      stream << " [label=\"" << *line_label << "\"]";
    stream << ";\n";
  }
}
 
auto DotVisitor::visit(const ast::AST& expr, string_view label) -> DotVisitor& {
  Loc location = expr.location();
  optional<string> type = {};
  if (auto val_ty = expr.val_ty(); val_ty)
    type = xml_escape(val_ty->name());
  auto kind_label = xml_escape(expr.kind_name());
 
  auto& node = add_node(DotNode(m_index, location, type, kind_label), label.data());
 
  auto* restore_parent = std::exchange(m_parent, &node);
  expr.visit(*this);
  m_parent = restore_parent;
 
  // This node had no parent, which means it's a root node.
  if (restore_parent == nullptr) {
    node.write(m_stream);
    m_root.reset();
  }
 
  return *this;
}
 
auto DotVisitor::visit(const string& str, string_view label) -> DotVisitor& {
  add_node(xml_escape(str), label.data());
 
  return *this;
}
 
auto DotVisitor::visit(std::nullptr_t, string_view label) -> DotVisitor& {
  add_node("<FONT COLOR=\"RED\">NULL</FONT>", label.data());
 
  return *this;
}
} // namespace yume::diagnostic