49 lines
1.4 KiB
C++
49 lines
1.4 KiB
C++
#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);
|
|
virtual void visit(LoadExpression* e);
|
|
|
|
void JIT(Expression* e);
|
|
private:
|
|
void putNamedValue(const std::string& name, llvm::Value* value);
|
|
llvm::Value* getNamedValue(const std::string& name);
|
|
|
|
llvm::IRBuilder<>* builder_;
|
|
llvm::FunctionPassManager* fpm_;
|
|
llvm::Module* module_;
|
|
|
|
int scope_;
|
|
std::vector<std::map<std::string, llvm::Value*>> namedValues_;
|
|
llvm::Value* value_;
|
|
};
|
|
|
|
#endif // CODEGENVISITOR_H
|