ASTVisualizer.h

Go to the documentation of this file.

#pragma once
 
#include "ASTVisitor.h"
#include <SyntaxTree.h>
#include <iostream>
#include <map>
#include <ostream>
#include <stack>
#include <string>
 
class ASTVisualizer {
public:
  ASTVisualizer() : os(std::cout) {}
  ASTVisualizer(std::ostream &os) : os(os) {}
 
  void buildGraph(SyntaxTree &tree);
 
private:
  std::ostream &os;
  std::stack<ASTNode *> parent;
  std::map<ASTNode *, std::string> vertexMap;
 
  void connect_node_to_parent(ASTNode *element);
  void declare_node(ASTNode *element, std::string label = "");
  void pushn(ASTNode *element, int n);
  void process_node(ASTNode *element);
};