parser.hpp

#ifndef __parser
#define __parser

#include "ast.hpp"
#include "scan.hpp"
#include "global.hpp"
#include "shunting_yard.h"
#include <stack>
#include <llvm/ExecutionEngine/GenericValue.h>

void HandleProgram(Scan_file *, Main_block *);
void HandleFunction(Scan_file *, Main_block *);
void HandleEnd(Scan_file *, Main_block*);
void HandleReturn(Scan_file *, Main_block*);
void HandleTopLevelExpression(Scan_file*, Main_block*);
void HandleStatement(Scan_file*, Main_block*);
void HandleIfBlock(Scan_file*, Main_block*);
CondAST* HandleIfCond(Scan_file*, Main_block*, bool);
int HandleIfOperator(Scan_file*, Main_block*);
void HandleElseBlock(Scan_file*, Main_block*);
void HandleForBlock(Scan_file*, Main_block*);
VariableAST* HandleVariableDeclaration(Scan_file *, Main_block *);
void parser(Scan_file *, Main_block *);

std::vector<BasicAST*> HandleMath(Scan_file*, Main_block*);
StringAST* HandleString(Scan_file*, Main_block*);

void ExternalFunctions(Scan_file*, Main_block*);

BasicAST* convert_tok_ast(int, std::string);

class NBlock;

using namespace llvm;

class CodeGenBlock {
public:
    BasicBlock *block;
    std::map<std::string, Value*> locals;
};

class MainContext {
    std::stack<CodeGenBlock *> blocks;
    Function *mainFunction;
    std::string name;
public:
    Module *module;
    MainContext(std::string name) : name(name) { module = new Module(name, getGlobalContext()); }

    void generateCode(NBlock& root);
    GenericValue runCode();
    std::map<std::string, Value*>& locals() { return blocks.top()->locals; }
    BasicBlock *currentBlock() { return blocks.top()->block; }
    void pushBlock(BasicBlock *block) { blocks.push(new CodeGenBlock()); blocks.top()->block = block; }
    void popBlock() { CodeGenBlock *top = blocks.top(); blocks.pop(); delete top; }
};

#endif // __parser