#ifndef CODEGENVISITOR_H
#define CODEGENVISITOR_H

#include "ASTVisitor.h"
#include "llvm/IRBuilder.h"
#include "llvm/Module.h"
#include "llvm/PassManager.h"
#include <map>
#include <string>

class CodeGenVisitor : public ASTVisitor {
public:
	CodeGenVisitor(llvm::Module* module, llvm::FunctionPassManager* fpm);
	virtual ~CodeGenVisitor();

 	virtual void visit(AssignmentExpression* e);
	virtual void visit(BinOpExpression* e);
	virtual void visit(ConstantExpression* e);
	virtual void visit(ExpressionStatement* e);
	virtual void visit(ForStatement* e);
	virtual void visit(FunctionCallExpression* e);
	virtual void visit(FunctionDefinition* e);
	virtual void visit(IfStatement* e);
	virtual void visit(ParameterList* e);
	virtual void visit(RandomForStatement* e);
	virtual void visit(RandomIfStatement* e);
	virtual void visit(ReturnStatement* e);
	virtual void visit(Scope* e);
	virtual void visit(StatementList* e);
	virtual void visit(ValueList* e);
	virtual void visit(VariableDefinition* e);
private:
	llvm::IRBuilder<>* builder_;
	llvm::FunctionPassManager* fpm_;
	llvm::Module* module_;
	std::map<std::string, llvm::Value*> namedValues_;
};

#endif // CODEGENVISITOR_H