scully/src/AST/PrintVisitor.cpp

#include "AST/PrintVisitor.h"
#include <iostream>
#include <sstream>

PrintVisitor::PrintVisitor() : level_(0) {
	//
}

PrintVisitor::~PrintVisitor() {
	//
}

void PrintVisitor::visit(AssignmentExpression* e) {
	println("AssignmentExpression");
	level_++;
	std::stringstream ss;
	ss << e->getId() << " = ";
	println(ss.str());
	e->getExpr()->accept(this);
	level_--;
}

void PrintVisitor::visit(BinOpExpression* e) {
	println("BinOpExpression");
	level_++;
	std::stringstream ss;
	ss << "Operator: " << e->getOp();
	println(ss.str());
	println("LHS:");
	e->getLeftExp()->accept(this);
	println("RHS:");
	e->getRightExp()->accept(this);
	level_--;
}

void PrintVisitor::visit(ConstantExpression* e) {
	println("ConstantExpression");
	level_++;
	println(e->getValue());
	level_--;
}

void PrintVisitor::visit(ExpressionStatement* e) {
	println("ExpressionStatement");
	level_++;
	e->getExpr()->accept(this);
	level_--;
}

void PrintVisitor::visit(ForStatement* e) {
	println("ForStatement");
	level_++;
	println("Init:");
	e->getInit()->accept(this);
	println("Cond:");
	e->getCond()->accept(this);
	println("Step:");
	e->getStep()->accept(this);
	println("Stmt:");
	e->getStmt()->accept(this);
	level_--;
}

void PrintVisitor::visit(FunctionCallExpression* e) {
	println("FunctionCallExpression");
	level_++;
	std::stringstream ss;
	ss << "Name: " << e->getId();
	println(ss.str());
	println("Values:");
	e->getValues()->accept(this);
	level_--;
}

void PrintVisitor::visit(FunctionDefinition* e) {
	println("FunctionDefinition");
	level_++;
	std::stringstream ss;
	ss << "Name: " << e->getName();
	println(ss.str());
	ss.str("");
	ss.clear();
	ss << "Type: " << e->getType()->getName();
	println(ss.str());
	ParameterList* params = e->getParams();
	if (params) {
		params->accept(this);
	}
	e->getSl()->accept(this);
	level_--;
}

void PrintVisitor::visit(IfStatement* e) {
	println("IfStatement");
	level_++;
	println("Cond:");
	e->getCond()->accept(this);
	println("Stmt:");
	e->getStmt()->accept(this);
	level_--;
}

void PrintVisitor::visit(ParameterList* e) {
	// TODO implement
}

void PrintVisitor::visit(RandomForStatement* e) {
	// TODO implement
}

void PrintVisitor::visit(RandomIfStatement* e) {
	// TODO implement
}

void PrintVisitor::visit(ReturnStatement* e) {
	println("ReturnStatement");
	level_++;
	e->getExpr()->accept(this);
	level_--;
}

void PrintVisitor::visit(Scope* e) {
	println("Scope");
	level_++;
	e->getSl()->accept(this);
	level_--;
}

void PrintVisitor::visit(StatementList* e) {
	println("StatementList");
	level_++;
	auto statements = e->getStatements();
	auto iter = statements.begin();
	auto end = statements.end();
	for (; iter != end; ++iter) {
		(*iter)->accept(this);
	}
	level_--;
}

void PrintVisitor::visit(ValueList* e) {
	// TODO implement
}

void PrintVisitor::visit(VariableDefinition* e) {
	println("VariableDefinition");
	level_++;
	std::stringstream ss;
	ss << "Name: " << e->getName();
	println(ss.str());
	ss.str("");
	ss.clear();
	ss << "Type: " << e->getType()->getName();
	println(ss.str());
	level_--;
}

void PrintVisitor::indent() {
	for (int i = 0; i < level_; i++) {
		std::cout << " ";
	}
}

void PrintVisitor::print(const std::string& s) {
	indent();
	std::cout << s;
}

void PrintVisitor::println(const std::string& s) {
	print(s);
	std::cout << std::endl;
}