/**
  *
  *                  OOAS Compiler - C++ AST
  *
  * Copyright 2015, AIT Austrian Institute of Technology.
  * All rights reserved.
  *
  * SEE THE "LICENSE" FILE FOR THE TERMS UNDER WHICH THIS FILE IS PROVIDED.
  *
  * If you modify the file please update the list of contributors below to in-
  * clude your name. Please also stick to the coding convention of using TABs
  * to do the basic (block-level) indentation and spaces for anything after
  * that. (Enable the display of special chars and it should be pretty obvious
  * what this means.) Also, remove all trailing whitespace.
  *
  * Contributors:
  *               Willibald Krenn (AIT)
  *               Stephan Zimmerer (AIT)
  *               Christoph Czurda (AIT)
  *
  */




#pragma once

#include <ast/CompleteAstTraversalVisitor.hpp>
#include <ast/identifiers/Identifier.hpp>
#include <ast/TextEmitter.hpp>
#include <ast/identifiers/IdentifierBlock.hpp>
#include <ast/statements/Skip.hpp>

namespace Ast {

class PrintVisitor final:
	public CompleteAstTraversalVisitor
{
protected:
	bool m_makeParens = true;
	bool m_writeTypes;
	Identifier* m_currentTypeDef;
	TextEmitter m_output;

	template<typename T> void printSubElementOrNull(T* anElem);
	template <typename T> void printEnumeration(const T& iarg);
	void printOperator(Expression* expression);
	void printType(Type* atype, const std::function<void()>& anAction);

public:
	PrintVisitor(CallContext* context);
	virtual ~PrintVisitor();

	void visit(MainModule* mainModule) override;
	void visit(TypeIdentifier* typeIdentifier) override;
	void visit(BoolType* boolType) override;
	void visit(IntType* intType) override;
	void visit(EnumIdentifier* enumIdentifier) override;
	void visit(EnumType* enumType) override;
	void visit(ListType* listType) override;
	void visit(TupleType* tupleType) override;
	void visit(ActionSystemType* actionSystemType) override;
	void visit(NullType* nullType) override;
	void visit(AttributeIdentifier* attributeIdentifier) override;
	void visit(BoolValueExpression* valueExpression) override;
	void visit(IntValueExpression* valueExpression) override;
	void visit(RefValueExpression* valueExpression) override;
//	void visit(AbortStatement* abortStatement) override;
	void visit(Assignment* assignment) override;
	void visit(AccessExpression* accessExpression) override;
	void visit(TypeExpression* typeExpression) override;
	void visit(BinaryOperator* binaryOperator) override;
	void visit(UnaryOperator* unaryOperator) override;
	void visit(CallExpression* callExpression) override;
	void visit(TupleMapAccessExpression* tupleMapAccessExpression) override;
	void visit(TupleConstructor* tupleConstructor) override;
	void visit(ListConstructor* listConstructor) override;
	void visit(NondetBlock* nondetBlock) override;
	void visit(PrioBlock* prioBlock) override;
	void visit(SeqBlock* seqBlock) override;
	void visit(Call* call) override;
	void visit(GuardedCommand* guardedCommand) override;
	void visit(IdentifierExpression* identifierExpression) override;
	void visit(NondetIdentifierBlock* nondetIdentifierList) override;
	void visit(PrioIdentifierBlock* prioIdentifierList) override;
	void visit(SeqIdentifierBlock* seqIdentifierList) override;
	void visit(Skip* skipStatement) override;
	void visit(TernaryOperator* ternaryOperator) override;
	void visit(MethodIdentifier* methodIdentifier) override;
	void visit(NamedActionIdentifier* namedActionIdentifier) override;
	void visit(SetConstructor* setConstructor) override;
	void visit(ExpressionVariableIdentifier* expressionVariableIdentifier) override;
	void visit(ObjectConstructor* objectConstructor) override;
	void visit(PointerType* pointerType) override;

	std::string returnVisitorName() {
		return std::string("Print OOAS");
	}

	std::string toString()
	{
		return m_output.toString();
	}
};

} /* namespace Ast */