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

In half way of write codegen

Browse Source
This commit is contained in:
ZhuangYumin
2024-08-28 10:52:32 +00:00
parent 07df48d0f1
commit 46f659053b
6 changed files with 356 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/IR/IRBuilder.cpp
View File

@ -72,9 +72,9 @@ void IRBuilder::ActuralVisit(FuncDef_ASTNode *node) {
std::dynamic_pointer_cast<UNConditionJMPAction>(cur_func->init_block->exit_action)->label_full =
current_block->label_full;
is_in_func_def = true;
cur_alloca_block=func_def->init_block;
cur_alloca_block = func_def->init_block;
node->func_body->accept(this);
cur_alloca_block=main_init_block;
cur_alloca_block = main_init_block;
is_in_func_def = false;
if (!(cur_block->exit_action)) {
@ -92,6 +92,7 @@ void IRBuilder::ActuralVisit(FuncDef_ASTNode *node) {
void IRBuilder::ActuralVisit(ClassDef_ASTNode *node) {
is_in_class_def = true;
cur_class_name = node->class_name;
prog->low_level_class_info["%.class." + cur_class_name] = global_scope->fetch_class_info(cur_class_name);
auto tpdf = std::make_shared<TypeDefItem>();
tpdf->class_name_raw = node->class_name;
cur_class = tpdf;
@ -103,7 +104,7 @@ void IRBuilder::ActuralVisit(ClassDef_ASTNode *node) {
}
void IRBuilder::ActuralVisit(Program_ASTNode *node) {
cur_alloca_block=main_init_block;
cur_alloca_block = main_init_block;
for (auto ch : node->sorted_children) {
ch->accept(this);
}
@ -950,6 +951,8 @@ void IRBuilder::ActuralVisit(LAndExpr_ASTNode *node) {
size_t new_block_id = block_counter++;
auto right_cal_block = std::make_shared<BlockItem>();
auto new_block = std::make_shared<BlockItem>();
auto phi_act = std::make_shared<PhiItem>();
right_cal_block->label_full = "label_" + std::to_string(right_cal_block_id);
new_block->label_full = "label_" + std::to_string(new_block_id);
cur_block->exit_action = std::make_shared<BRAction>();
@ -957,15 +960,18 @@ void IRBuilder::ActuralVisit(LAndExpr_ASTNode *node) {
std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->true_label_full = right_cal_block->label_full;
std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->false_label_full = new_block->label_full;
std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->cond = node->left->IR_result_full;
cur_block->exit_action->corresponding_phi = phi_act;
cur_func->basic_blocks.push_back(right_cal_block);
cur_block = right_cal_block;
node->right->accept(this);
cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
auto right_block_end = cur_block;
std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = new_block->label_full;
cur_block->exit_action->corresponding_phi = phi_act;
cur_func->basic_blocks.push_back(new_block);
cur_block = new_block;
auto phi_act = std::make_shared<PhiItem>();
phi_act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
phi_act->ty = LLVMIRIntType(1);
phi_act->values.push_back(std::make_pair("0", src_block->label_full));
@ -989,6 +995,8 @@ void IRBuilder::ActuralVisit(LOrExpr_ASTNode *node) {
size_t new_block_id = block_counter++;
auto right_cal_block = std::make_shared<BlockItem>();
auto new_block = std::make_shared<BlockItem>();
auto phi_act = std::make_shared<PhiItem>();
right_cal_block->label_full = "label_" + std::to_string(right_cal_block_id);
new_block->label_full = "label_" + std::to_string(new_block_id);
cur_block->exit_action = std::make_shared<BRAction>();
@ -996,15 +1004,18 @@ void IRBuilder::ActuralVisit(LOrExpr_ASTNode *node) {
std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->true_label_full = new_block->label_full;
std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->false_label_full = right_cal_block->label_full;
std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->cond = node->left->IR_result_full;
cur_block->exit_action->corresponding_phi = phi_act;
cur_func->basic_blocks.push_back(right_cal_block);
cur_block = right_cal_block;
node->right->accept(this);
cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
auto right_block_end = cur_block;
std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = new_block->label_full;
cur_block->exit_action->corresponding_phi = phi_act;
cur_func->basic_blocks.push_back(new_block);
cur_block = new_block;
auto phi_act = std::make_shared<PhiItem>();
phi_act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
phi_act->ty = LLVMIRIntType(1);
phi_act->values.push_back(std::make_pair("1", src_block->label_full));
@ -1045,6 +1056,8 @@ void IRBuilder::ActuralVisit(TernaryExpr_ASTNode *node) {
ExprTypeInfo void_std = IdentifierType("void");
if (node->expr_type_info != void_std) {
auto phi_act = std::make_shared<PhiItem>();
src1_end_block->exit_action->corresponding_phi = phi_act;
src2_end_block->exit_action->corresponding_phi = phi_act;
phi_act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
phi_act->ty = Type_AST2LLVM(node->expr_type_info);
phi_act->values.push_back(std::make_pair(node->src1->IR_result_full, src1_end_block->label_full));

4
src/main.cpp
View File

@ -28,8 +28,8 @@ int main(int argc, char **argv) {
try {
SemanticCheck(fin, ast);
auto IR = BuildIR(ast);
IR->RecursivePrint(fout);
// GenerateNaiveASM(fout, IR);
IR->RecursivePrint(std::cerr);
GenerateNaiveASM(fout, IR);
} catch (const SemanticError &err) {
std::cout << err.what() << std::endl;
return err.GetErrorCode();

57
src/naivebackend/naivebackend.cpp
View File

@ -1,4 +1,59 @@
#include "naivebackend.h"
#include <unordered_map>
#include "IR/IR_basic.h"
#include "build_layout.hpp"
using namespace NaiveBackend;
void GenerateNaiveASM(std::ostream &os, std::shared_ptr<ModuleItem> prog) {
;
auto riscv = std::make_shared<RISCVProgItem>();
for (auto conststr : prog->const_strs) {
auto asm_item = std::make_shared<RISCVConstStrItem>();
riscv->const_strs.push_back(asm_item);
asm_item->content = conststr->string_raw;
asm_item->full_label = ".str." + std::to_string(conststr->const_str_id);
}
for (auto global_var : prog->global_var_defs) {
auto asm_item = std::make_shared<RISCVGlobalVarItem>();
riscv->global_vars.push_back(asm_item);
asm_item->full_label = ".var.global." + global_var->name_raw + ".addrkp";
}
std::unordered_map<std::string, FuncLayout> func_layouts;
for (auto func_def : prog->function_defs) {
if (func_def->init_block) {
for (auto act : func_def->init_block->actions) {
ScanForVar(func_layouts[func_def->func_name_raw], act);
}
}
for (auto block : func_def->basic_blocks) {
for (auto act : block->actions) {
ScanForVar(func_layouts[func_def->func_name_raw], act);
}
}
FuncLayout &layout = func_layouts[func_def->func_name_raw];
for (size_t i = 0; i < func_def->args_full_name.size(); i++) {
layout.arg_offset[func_def->args_full_name[i]] = i;
}
// debug:
std::cerr << "layout info of function " << func_def->func_name_raw << std::endl;
std::cerr << "\tcur_pos=" << layout.cur_pos << std::endl;
std::cerr << "\ttotal_frame_size=" << layout.total_frame_size << std::endl;
for (const auto &item : layout.local_items) {
std::cerr << "\t" << item.first << " " << item.second << std::endl;
}
}
for (auto func_def : prog->function_defs) {
auto func_asm = std::make_shared<RISCVFuncItem>();
riscv->funcs.push_back(func_asm);
func_asm->full_label = func_def->func_name_raw;
if (func_def->init_block) {
func_asm->code_lines.push_back(".entrylabel." + func_def->init_block->label_full + ":");
}
for (auto block : func_def->basic_blocks) {
func_asm->code_lines.push_back(".entrylabel." + block->label_full + ":");
}
}
riscv->RecursivePrint(os);
}