ZYM/MXCompiler
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

set structure for IR

Browse Source
This commit is contained in:
ZhuangYumin
2024-08-21 04:10:36 +00:00
parent 03e33d0a0f
commit ed1ba4b59a
17 changed files with 1423 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/IR/IR.h Normal file
View File

@ -0,0 +1,3 @@
#pragma once
#include "IR_basic.h"
#include "IRBuilder.h"

54
include/IR/IRBuilder.h Normal file
View File

@ -0,0 +1,54 @@
#include <memory>
#include "IR_basic.h"
#include "ast/astnode_visitor.h"
class IRBuilder : public ASTNodeVirturalVisitor {
std::shared_ptr<ModuleItem> prog;
public:
// Structural AST Nodes
void ActuralVisit(FuncDef_ASTNode *node) override;
void ActuralVisit(ClassDef_ASTNode *node) override;
void ActuralVisit(Program_ASTNode *node) override;
// Statement AST Nodes
void ActuralVisit(EmptyStatement_ASTNode *node) override;
void ActuralVisit(DefinitionStatement_ASTNode *node) override;
void ActuralVisit(ExprStatement_ASTNode *node) override;
void ActuralVisit(IfStatement_ASTNode *node) override;
void ActuralVisit(WhileStatement_ASTNode *node) override;
void ActuralVisit(ForStatement_ASTNode *node) override;
void ActuralVisit(JmpStatement_ASTNode *node) override;
void ActuralVisit(SuiteStatement_ASTNode *node) override;
// Expression AST Nodes
void ActuralVisit(NewArrayExpr_ASTNode *node) override;
void ActuralVisit(NewConstructExpr_ASTNode *node) override;
void ActuralVisit(NewExpr_ASTNode *node) override;
void ActuralVisit(AccessExpr_ASTNode *node) override;
void ActuralVisit(IndexExpr_ASTNode *node) override;
void ActuralVisit(SuffixExpr_ASTNode *node) override;
void ActuralVisit(PrefixExpr_ASTNode *node) override;
void ActuralVisit(OppositeExpr_ASTNode *node) override;
void ActuralVisit(LNotExpr_ASTNode *node) override;
void ActuralVisit(BNotExpr_ASTNode *node) override;
void ActuralVisit(MDMExpr_ASTNode *node) override;
void ActuralVisit(PMExpr_ASTNode *node) override;
void ActuralVisit(RLExpr_ASTNode *node) override;
void ActuralVisit(GGLLExpr_ASTNode *node) override;
void ActuralVisit(NEExpr_ASTNode *node) override;
void ActuralVisit(BAndExpr_ASTNode *node) override;
void ActuralVisit(BXorExpr_ASTNode *node) override;
void ActuralVisit(BOrExpr_ASTNode *node) override;
void ActuralVisit(LAndExpr_ASTNode *node) override;
void ActuralVisit(LOrExpr_ASTNode *node) override;
void ActuralVisit(TernaryExpr_ASTNode *node) override;
void ActuralVisit(AssignExpr_ASTNode *node) override;
void ActuralVisit(ThisExpr_ASTNode *node) override;
void ActuralVisit(ParenExpr_ASTNode *node) override;
void ActuralVisit(IDExpr_ASTNode *node) override;
void ActuralVisit(FunctionCallExpr_ASTNode *node) override;
void ActuralVisit(FormattedStringExpr_ASTNode *node) override;
void ActuralVisit(ConstantExpr_ASTNode *node) override;
};
std::shared_ptr<ModuleItem> BuildIR(std::shared_ptr<Program_ASTNode> src);

143
include/IR/IR_basic.h Normal file
View File

@ -0,0 +1,143 @@
#pragma once
#include <ios>
#include <memory>
#include <string>
#include <variant>
#include <vector>
#include "ast/astnode.h"
struct LLVMIRIntType {
size_t bits;
};
struct LLVMIRPTRType {};
struct LLVMIRCLASSTYPE {};
using LLVMType = std::variant<LLVMIRIntType, LLVMIRPTRType, LLVMIRCLASSTYPE>;
class LLVMIRItemBase {
public:
LLVMIRItemBase() = default;
virtual ~LLVMIRItemBase() = default;
virtual void RecursivePrint(std::ostream &os) const;
};
class TypeDefItem : public LLVMIRItemBase {
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class GlobalVarDefItem : public LLVMIRItemBase {
LLVMType type;
std::string name;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class ActionItem : public LLVMIRItemBase {
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class JMPActionItem : public ActionItem {
std::string label;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class BRAction: public JMPActionItem {
std::string cond;
std::string true_label;
std::string false_label;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class UNConditionJMPAction: public JMPActionItem {
std::string label;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class RETAction : public JMPActionItem {
std::string value;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class BinaryOperationAction : public ActionItem {
std::string op;
std::string lhs;
std::string rhs;
std::string result;
LLVMType type;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class AllocaAction : public ActionItem {
std::string name;
LLVMType type;
size_t num;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class LoadAction : public ActionItem {
std::string result;
LLVMType ty;
std::string ptr;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class StoreAction : public ActionItem {
LLVMType ty;
std::string value;
std::string ptr;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class GetElementPtrAction : public ActionItem {
std::string result;
LLVMType ty;
std::string ptr;
std::vector<std::string> indices;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class ICMPAction : public ActionItem {
std::string op;
std::string lhs;
std::string rhs;
std::string result;
LLVMType ty;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class BlockItem : public LLVMIRItemBase {
std::string label;
std::vector<std::shared_ptr<ActionItem>> actions;
std::shared_ptr<JMPActionItem> exit_action;
public:
void RecursivePrint(std::ostream &os) const { ; }
};
class FunctionDefItem : public LLVMIRItemBase {
std::vector<std::shared_ptr<BlockItem>> basic_blocks;
public:
void RecursivePrint(std::ostream &os) const {
for (auto &item : basic_blocks) {
item->RecursivePrint(os);
os << '\n';
}
}
};
class ModuleItem : public LLVMIRItemBase {
std::vector<std::shared_ptr<TypeDefItem>> type_defs;
std::vector<std::shared_ptr<GlobalVarDefItem>> global_var_defs;
std::vector<std::shared_ptr<FunctionDefItem>> function_defs;
public:
void RecursivePrint(std::ostream &os) const {
for (auto &item : type_defs) {
item->RecursivePrint(os);
os << '\n';
}
for (auto &item : global_var_defs) {
item->RecursivePrint(os);
os << '\n';
}
for (auto &item : function_defs) {
item->RecursivePrint(os);
os << '\n';
}
}
};

1
include/ast/ast.h
View File

@ -2,6 +2,7 @@
#define AST_H
#include "astnode.h"
#include "astnode_visitor.h"
#include "semantic_visitor.h"
#include "expr_astnode.h"
#include "statement_astnode.h"
#include "structural_astnode.h"

63
include/ast/astnode_visitor.h
View File

@ -54,67 +54,4 @@ class ASTNodeVirturalVisitor : public ASTNodeVisitorBase {
virtual void ActuralVisit(ConstantExpr_ASTNode *node) = 0;
};
class ASTSemanticCheckVisitor : public ASTNodeVirturalVisitor {
bool is_in_func_def;
bool has_return;
FunctionSchema cur_func_schema;
std::string cur_class_name;
bool is_in_class_def;
size_t loop_level;
std::shared_ptr<GlobalScope> global_scope;
friend std::shared_ptr<Program_ASTNode> CheckAndDecorate(std::shared_ptr<Program_ASTNode> src);
bool ClassExists(const std::string &name) {
if (name == "int" || name == "bool") return true;
return global_scope->classes.find(name) != global_scope->classes.end();
}
public:
ASTSemanticCheckVisitor() : is_in_func_def(false), loop_level(0) {}
// Structural AST Nodes
void ActuralVisit(FuncDef_ASTNode *node) override;
void ActuralVisit(ClassDef_ASTNode *node) override;
void ActuralVisit(Program_ASTNode *node) override;
// Statement AST Nodes
void ActuralVisit(EmptyStatement_ASTNode *node) override;
void ActuralVisit(DefinitionStatement_ASTNode *node) override;
void ActuralVisit(ExprStatement_ASTNode *node) override;
void ActuralVisit(IfStatement_ASTNode *node) override;
void ActuralVisit(WhileStatement_ASTNode *node) override;
void ActuralVisit(ForStatement_ASTNode *node) override;
void ActuralVisit(JmpStatement_ASTNode *node) override;
void ActuralVisit(SuiteStatement_ASTNode *node) override;
// Expression AST Nodes
void ActuralVisit(NewArrayExpr_ASTNode *node) override;
void ActuralVisit(NewConstructExpr_ASTNode *node) override;
void ActuralVisit(NewExpr_ASTNode *node) override;
void ActuralVisit(AccessExpr_ASTNode *node) override;
void ActuralVisit(IndexExpr_ASTNode *node) override;
void ActuralVisit(SuffixExpr_ASTNode *node) override;
void ActuralVisit(PrefixExpr_ASTNode *node) override;
void ActuralVisit(OppositeExpr_ASTNode *node) override;
void ActuralVisit(LNotExpr_ASTNode *node) override;
void ActuralVisit(BNotExpr_ASTNode *node) override;
void ActuralVisit(MDMExpr_ASTNode *node) override;
void ActuralVisit(PMExpr_ASTNode *node) override;
void ActuralVisit(RLExpr_ASTNode *node) override;
void ActuralVisit(GGLLExpr_ASTNode *node) override;
void ActuralVisit(NEExpr_ASTNode *node) override;
void ActuralVisit(BAndExpr_ASTNode *node) override;
void ActuralVisit(BXorExpr_ASTNode *node) override;
void ActuralVisit(BOrExpr_ASTNode *node) override;
void ActuralVisit(LAndExpr_ASTNode *node) override;
void ActuralVisit(LOrExpr_ASTNode *node) override;
void ActuralVisit(TernaryExpr_ASTNode *node) override;
void ActuralVisit(AssignExpr_ASTNode *node) override;
void ActuralVisit(ThisExpr_ASTNode *node) override;
void ActuralVisit(ParenExpr_ASTNode *node) override;
void ActuralVisit(IDExpr_ASTNode *node) override;
void ActuralVisit(FunctionCallExpr_ASTNode *node) override;
void ActuralVisit(FormattedStringExpr_ASTNode *node) override;
void ActuralVisit(ConstantExpr_ASTNode *node) override;
};
#endif // ASTNODE_ActuralVisitOR_H

7
include/ast/scope.hpp
View File

@ -17,6 +17,7 @@ class ScopeBase {
protected:
ScopeBase *parent; // cannot use std::shared_ptr<ScopeBase> because of circular dependency
size_t scope_id;
virtual bool VariableNameAvailable(const std::string &name, int ttl) = 0;
virtual bool add_variable(const std::string &name, const ExprTypeInfo &type) = 0;
virtual ExprTypeInfo fetch_varaible(const std::string &name) = 0;
@ -26,6 +27,11 @@ class ScopeBase {
"for", "while", "break", "continue", "return"};
return keywords.find(name) != keywords.end();
}
public:
ScopeBase() {
static size_t scope_counter=0;
scope_id = scope_counter++;
}
};
class LocalScope : public ScopeBase {
friend class Visitor;
@ -104,6 +110,7 @@ class ClassDefScope : public ScopeBase {
friend std::shared_ptr<Program_ASTNode> CheckAndDecorate(std::shared_ptr<Program_ASTNode> src);
std::unordered_map<std::string, ExprTypeInfo> member_variables;
std::unordered_map<std::string, std::shared_ptr<FunctionScope>> member_functions;
IRClassInfo llvm_class_info;
bool add_variable(const std::string &name, const ExprTypeInfo &type) override {
if (!VariableNameAvailable(name, 0)) {
return false;

69
include/ast/semantic_visitor.h Normal file
View File

@ -0,0 +1,69 @@
#pragma once
#include "astnode.h"
#include "expr_astnode.h"
#include "statement_astnode.h"
#include "structural_astnode.h"
#include "astnode_visitor.h"
class ASTSemanticCheckVisitor : public ASTNodeVirturalVisitor {
bool is_in_func_def;
bool has_return;
FunctionSchema cur_func_schema;
std::string cur_class_name;
bool is_in_class_def;
size_t loop_level;
std::shared_ptr<GlobalScope> global_scope;
friend std::shared_ptr<Program_ASTNode> CheckAndDecorate(std::shared_ptr<Program_ASTNode> src);
bool ClassExists(const std::string &name) {
if (name == "int" || name == "bool") return true;
return global_scope->classes.find(name) != global_scope->classes.end();
}
public:
ASTSemanticCheckVisitor() : is_in_func_def(false), loop_level(0) {}
// Structural AST Nodes
void ActuralVisit(FuncDef_ASTNode *node) override;
void ActuralVisit(ClassDef_ASTNode *node) override;
void ActuralVisit(Program_ASTNode *node) override;
// Statement AST Nodes
void ActuralVisit(EmptyStatement_ASTNode *node) override;
void ActuralVisit(DefinitionStatement_ASTNode *node) override;
void ActuralVisit(ExprStatement_ASTNode *node) override;
void ActuralVisit(IfStatement_ASTNode *node) override;
void ActuralVisit(WhileStatement_ASTNode *node) override;
void ActuralVisit(ForStatement_ASTNode *node) override;
void ActuralVisit(JmpStatement_ASTNode *node) override;
void ActuralVisit(SuiteStatement_ASTNode *node) override;
// Expression AST Nodes
void ActuralVisit(NewArrayExpr_ASTNode *node) override;
void ActuralVisit(NewConstructExpr_ASTNode *node) override;
void ActuralVisit(NewExpr_ASTNode *node) override;
void ActuralVisit(AccessExpr_ASTNode *node) override;
void ActuralVisit(IndexExpr_ASTNode *node) override;
void ActuralVisit(SuffixExpr_ASTNode *node) override;
void ActuralVisit(PrefixExpr_ASTNode *node) override;
void ActuralVisit(OppositeExpr_ASTNode *node) override;
void ActuralVisit(LNotExpr_ASTNode *node) override;
void ActuralVisit(BNotExpr_ASTNode *node) override;
void ActuralVisit(MDMExpr_ASTNode *node) override;
void ActuralVisit(PMExpr_ASTNode *node) override;
void ActuralVisit(RLExpr_ASTNode *node) override;
void ActuralVisit(GGLLExpr_ASTNode *node) override;
void ActuralVisit(NEExpr_ASTNode *node) override;
void ActuralVisit(BAndExpr_ASTNode *node) override;
void ActuralVisit(BXorExpr_ASTNode *node) override;
void ActuralVisit(BOrExpr_ASTNode *node) override;
void ActuralVisit(LAndExpr_ASTNode *node) override;
void ActuralVisit(LOrExpr_ASTNode *node) override;
void ActuralVisit(TernaryExpr_ASTNode *node) override;
void ActuralVisit(AssignExpr_ASTNode *node) override;
void ActuralVisit(ThisExpr_ASTNode *node) override;
void ActuralVisit(ParenExpr_ASTNode *node) override;
void ActuralVisit(IDExpr_ASTNode *node) override;
void ActuralVisit(FunctionCallExpr_ASTNode *node) override;
void ActuralVisit(FormattedStringExpr_ASTNode *node) override;
void ActuralVisit(ConstantExpr_ASTNode *node) override;
};

38
include/tools.h
View File

@ -1,6 +1,8 @@
#pragma once
#include <stdexcept>
#include <unordered_map>
#include <variant>
#include <vector>
enum class ASTNodeType {
// Expression nodes
NewArrayExpr,
@ -98,4 +100,38 @@ inline bool operator==(const ExprTypeInfo &l, const ExprTypeInfo &r) {
}
throw std::runtime_error("something strange happened");
}
inline bool operator!=(const ExprTypeInfo &l, const ExprTypeInfo &r) { return !(l == r); }
inline bool operator!=(const ExprTypeInfo &l, const ExprTypeInfo &r) { return !(l == r); }
class IRClassInfo {
public:
std::string class_name; // This data must be provided by user
std::vector<size_t> member_var_size; // This data must be provided by user. Each of them is the size of a member
// variable, which must be in [1,4]
std::unordered_map<std::string, size_t> member_var_offset; // This data must be provided by user
std::vector<size_t> member_var_pos_after_align;
size_t class_size_after_align;
void ArrangeSpace() {
size_t cur_pos = 0;
size_t align_size = 1;
for (size_t cur_size : member_var_size) {
if (cur_size != 1 && cur_size != 4) throw std::runtime_error("Invalid member variable size");
if (cur_pos % cur_size == 0) {
member_var_pos_after_align.push_back(cur_pos);
cur_pos += cur_size;
} else {
cur_pos += cur_size - (cur_pos % cur_size);
member_var_pos_after_align.push_back(cur_pos);
cur_pos += cur_size;
}
if (cur_size > align_size) align_size = cur_size;
if (cur_pos % align_size != 0) cur_pos += align_size - (cur_pos % align_size);
class_size_after_align = cur_pos;
}
}
};
class IRVariableInfo {
public:
enum class VariableType { global_variable, local_variable, member_variable };
std::string class_name;
std::string variable_name;
};