Skip to content
GitLab
Select Git revision
  • loop
  • submit2
  • final-gvn
  • gvn
  • main default
  • nodic
  • submit
  • cfg-test
  • cfg
  • backend-reg
  • backend-dev
  • frontend
Find file BlameHistoryPermalink
ast2irp.hpp 5.13 KiB
1 
#ifndef AST2IRP_HPP
2 
#define AST2IRP_HPP
3
4 
#include "../utils/dsa/ast.hpp"
5 
#include "../utils/dsa/env.hpp"
6 
#include "../utils/dsa/irp.hpp"
7 
#include "../utils/dsa/table.hpp"
8 
#include "../utils/dsa/temp.hpp"
9 
#include <stack>
10
11 
namespace irpgen {
12 
class TrExp;
13 
class TrExpList;
14 
class ExpTy;
15 
class LoopInfo;
16 
class ConstInitValInfo;
17 
class InitValInfo;
18
19 
class LocalVarInfo {
20 
public:
21 
    temp::Temp *temp;
22 
};
23
24 
class FuncInfo {
25 
public:
26 
    irp::FuncDef *funcDef;
27 
    temp::Label *label;
28 
};
29
30 
class AST2IRP {
31 
public:
32 
    AST2IRP(ast::Prog *root) : root(root) {}
33 
    irp::Prog *result;
34 
    void translate();
35
36 
private:
37 
    ast::Prog *root;
38
39 
    table::SymbolTable<env::GlobalVar *> globalTable;
40 
    table::SymbolTable<env::LocalVar *> localTable;
41 
    table::SymbolTable<env::Func *> funcTable;
42
43 
    std::stack<LoopInfo *> loopStack;
44
45 
    std::string *curFuncName;
46 
    int curFuncStackOffset;
47 
    TrExp *curFuncAllocas;
48 
    std::vector<irp::FormatDef *> *curFuncFormatsPtr;
49 
    std::unordered_map<std::string, irp::FormatDef *> formatTable;
50
51 
    TrExp *beforeMainDefs;
52 
    TrExp *mainAllocas;
53
54 
    irp::FuncDefList *transCompUnitList(ast::CompUnitList *compUnitList);
55 
    irp::FuncDef *transFuncDef(ast::FuncDef *funcDef);
56 
    std::pair<std::vector<env::TyKind> *, temp::TempList *> transFuncFParamList(ast::FuncFParamList *paramList);
57 
    std::pair<env::TyKind, temp::Temp *> transFuncFParam(ast::FuncFParam *param);
58 
    std::vector<int> *getArrayDimList(irp::ExpList *dimList);
59
60 
    void transGlobalDecl(ast::Decl *decl);
61 
    void transGlobalConstDecl(ast::ConstDecl *constDecl);
62 
    void transGlobalConstDef(ast::ConstDef *constDef, ast::BaseType baseType);
63 
    TrExp *transGlobalVarDecl(ast::VarDecl *varDecl);
64 
    TrExp *transGlobalVarDef(ast::VarDef *varDef, ast::BaseType baseType);
65
66 
    irp::GlobalDef *transGlobalArrConstInitVal(ast::InitVal *initVal, ast::BaseType baseType, temp::Label *label,
67 
                                               std::vector<int> *dims);
68 
    ConstInitValInfo *transConstInitValRecur(ast::InitVal *initVal, ast::BaseType baseType, std::vector<int> *dims,
69 
                                             int curDim, int step, int align, int &offset);
70 
    std::pair<irp::GlobalDef *, TrExp *> transGlobalVarInitVal(ast::InitVal *initVal, ast::BaseType baseType,
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
temp::Label *label); std::pair<irp::GlobalDef *, TrExp *> transGlobalArrInitVal(ast::InitVal *initVal, ast::BaseType baseType, temp::Label *label, std::vector<int> *dims); InitValInfo *transGlobalInitValRecur(ast::InitVal *initVal, ast::BaseType baseType, temp::Label *label, std::vector<int> *dims, int curDim, int step, int align, int &offset); TrExp *transBlock(ast::Block *block); TrExp *transBlockItemList(ast::BlockItemList *blockItemList); TrExp *transBlockItem(ast::BlockItem *blockItem); TrExp *transDecl(ast::Decl *decl); void transConstDecl(ast::ConstDecl *constDecl); void transConstDef(ast::ConstDef *constDef, ast::BaseType baseType); irp::LocalConstArrayDef *transLocalArrConstInitVal(ast::InitVal *initVal, ast::BaseType baseType, temp::Label *label, std::vector<int> *dims); TrExp *transVarDecl(ast::VarDecl *varDecl); TrExp *transVarDef(ast::VarDef *varDef, ast::BaseType baseType); TrExp *transInitVal(ast::InitVal *initVal, ast::BaseType baseType, temp::Temp *temp, std::vector<int> *dims); TrExp *transInitValRecur(ast::InitVal *initVal, ast::BaseType baseType, temp::Temp *temp, std::vector<int> *dims, int curDim, int step, int align, int &offset); TrExp *transStmt(ast::Stmt *stmt); TrExp *transAssignStmt(ast::Stmt *stmt); TrExp *transExpStmt(ast::Stmt *stmt); TrExp *transBlockStmt(ast::Stmt *stmt); TrExp *transIfStmt(ast::Stmt *stmt); TrExp *transWhileStmt(ast::Stmt *stmt); TrExp *transBreakStmt(ast::Stmt *stmt); TrExp *transContinueStmt(ast::Stmt *stmt); TrExp *transReturnStmt(ast::Stmt *stmt); TrExp *transPutintStmt(ast::Stmt *stmt); TrExp *transPutchStmt(ast::Stmt *stmt); TrExp *transPutfloatStmt(ast::Stmt *stmt); TrExp *transPutarrayStmt(ast::Stmt *stmt); TrExp *transPutfarrayStmt(ast::Stmt *stmt); TrExp *transPutfStmt(ast::Stmt *stmt); TrExp *transStarttimeStmt(); TrExp *transStoptimeStmt(); ExpTy *transExp(ast::Exp *exp); ExpTy *transBinOpExp(ast::Exp *exp); ExpTy *transCondExp(ast::Exp *exp); ExpTy *transIdExp(ast::Exp *exp); ExpTy *transIntConstExp(ast::Exp *exp); ExpTy *transFloatConstExp(ast::Exp *exp); ExpTy *transNotExp(ast::Exp *exp); ExpTy *transMinusExp(ast::Exp *exp); ExpTy *transCallExp(ast::Exp *exp); ExpTy *transArrayExp(ast::Exp *exp); ExpTy *transGetintExp(); ExpTy *transGetchExp(); ExpTy *transGetfloatExp(); ExpTy *transGetarrayExp(ast::Exp *exp); ExpTy *transGetfarrayExp(ast::Exp *exp); TrExpList *transExpList(ast::ExpList *expList); TrExpList *transArgs(ast::ExpList *args, std::vector<env::TyKind> *paramTypes); }; } // namespace irpgen #endif // AST2IRP_HPP