write basic functions for LLVM IR

src/IR/IRBuilder.cpp

@@ -1,51 +1,310 @@
 #include "IRBuilder.h"
 #include <stdexcept>
+#include "IR.h"
+#include "IR_basic.h"
+#include "tools.h"
 
 // Structural AST Nodes
 void IRBuilder::ActuralVisit(FuncDef_ASTNode *node) {
-  // TODO: Implement function body
+  auto declare = std::make_shared<FunctionDeclareItem>();
+  auto func_def = std::make_shared<FunctionDefItem>();
+  // prog->function_declares.push_back(declare);
+  prog->function_defs.push_back(func_def);
+  if (is_in_class_def) {
+    declare->func_name_raw = cur_class_name + "." + node->func_name;
+    declare->args.push_back(LLVMIRPTRType());  // The `self` pointer
+  } else {
+    declare->func_name_raw = node->func_name;
+  }
+  declare->return_type = Type_AST2LLVM(node->return_type);
+  for (auto &arg : node->params) {
+    declare->args.push_back(Type_AST2LLVM(arg.second));
+  }
+  func_def->func_name_raw = declare->func_name_raw;
+  func_def->return_type = declare->return_type;
+  func_def->args = declare->args;
+  size_t scope_id = node->current_scope->scope_id;
+  if (is_in_class_def) {
+    std::string arg_name_raw = "this";
+    std::string full_name = "%.var.local." + std::to_string(scope_id) + "." + arg_name_raw + ".val";
+    func_def->args_full_name.push_back(full_name);
+  }
+  for (auto &arg : node->params) {
+    std::string &arg_name_raw = arg.first;
+    std::string full_name = "%.var.local." + std::to_string(scope_id) + "." + arg_name_raw + ".val";
+    func_def->args_full_name.push_back(full_name);
+  }
+  if (func_def->args.size() != func_def->args_full_name.size()) throw std::runtime_error("args size not match");
+
+  cur_func = func_def;
+  auto current_block = std::make_shared<BlockItem>();
+  cur_block = current_block;
+  cur_func->basic_blocks.push_back(current_block);
+  size_t block_id = block_counter++;
+  current_block->label_full = "label_" + std::to_string(block_id);
+
+  is_in_func_def = true;
+  node->func_body->accept(this);
+  is_in_func_def = false;
+
+  if (!(cur_block->exit_action)) {
+    auto default_ret = std::make_shared<RETAction>();
+    cur_block->exit_action = default_ret;
+    default_ret->type = cur_func->return_type;
+    if (std::holds_alternative<LLVMIRIntType>(default_ret->type)) {
+      default_ret->value = "0";
+    } else {
+      default_ret->value = "null";
+    }
+  }
 }
 
 void IRBuilder::ActuralVisit(ClassDef_ASTNode *node) {
-  // TODO: Implement function body
+  is_in_class_def = true;
+  cur_class_name = node->class_name;
+  auto tpdf = std::make_shared<TypeDefItem>();
+  tpdf->class_name_raw = node->class_name;
+  cur_class = tpdf;
+  prog->type_defs.push_back(tpdf);
+  for (auto ch : node->sorted_children) {
+    ch->accept(this);
+  }
+  is_in_class_def = false;
 }
 
 void IRBuilder::ActuralVisit(Program_ASTNode *node) {
-  // TODO: Implement function body
-  throw std::runtime_error("IRBuilder not implemented");
+  for (auto ch : node->sorted_children) {
+    ch->accept(this);
+  }
 }
 
 // Statement AST Nodes
 void IRBuilder::ActuralVisit(EmptyStatement_ASTNode *node) {
-  // TODO: Implement function body
+  // do nothing
 }
 
 void IRBuilder::ActuralVisit(DefinitionStatement_ASTNode *node) {
-  // TODO: Implement function body
+  if (is_in_class_def) {
+    cur_class->elements.push_back(Type_AST2LLVM(node->var_type));
+  } else if (!is_in_func_def) {
+    for (const auto &var : node->vars) {
+      auto var_def = std::make_shared<GlobalVarDefItem>();
+      prog->global_var_defs.push_back(var_def);
+      var_def->type = Type_AST2LLVM(node->var_type);
+      var_def->name_raw = var.first;
+      // TODO: initial value
+    }
+  } else {
+    for (const auto &var : node->vars) {
+      auto var_def = std::make_shared<AllocaAction>();
+      cur_block->actions.push_back(var_def);
+      var_def->num = 1;
+      var_def->type = Type_AST2LLVM(node->var_type);
+      var_def->name_full = "%.var.local." + std::to_string(node->current_scope->scope_id) + "." + var.first + ".addrkp";
+      if (var.second) {
+        var.second->accept(this);
+        auto act = std::make_shared<StoreAction>();
+        cur_block->actions.push_back(act);
+        act->ty = var_def->type;
+        act->ptr_full = var_def->name_full;
+        act->value_full = var.second->IR_result_full;
+      }
+    }
+  }
+  just_encountered_jmp = false;
 }
 
 void IRBuilder::ActuralVisit(ExprStatement_ASTNode *node) {
-  // TODO: Implement function body
+  // just visit it
+  node->expr->accept(this);
+  just_encountered_jmp = false;
 }
 
 void IRBuilder::ActuralVisit(IfStatement_ASTNode *node) {
-  // TODO: Implement function body
+  node->condition->accept(this);
+  cur_block->exit_action = std::make_shared<BRAction>();
+  if (node->else_clause) {
+    size_t if_block_start_id = block_counter++;
+    size_t else_block_start_id = block_counter++;
+    size_t following_block_id = block_counter++;
+    auto first_if_block = std::make_shared<BlockItem>();
+    first_if_block->label_full = "label_" + std::to_string(if_block_start_id);
+    auto first_else_block = std::make_shared<BlockItem>();
+    first_else_block->label_full = "label_" + std::to_string(else_block_start_id);
+    auto following_block = std::make_shared<BlockItem>();
+    following_block->label_full = "label_" + std::to_string(following_block_id);
+    std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->cond = node->condition->IR_result_full;
+    std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->true_label_full = first_if_block->label_full;
+    std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->false_label_full = first_else_block->label_full;
+    cur_block = first_if_block;
+    cur_func->basic_blocks.push_back(first_if_block);
+    node->if_clause->accept(this);
+    if (!(cur_block->exit_action)) {
+      cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+      std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = following_block->label_full;
+    }
+    cur_block = first_else_block;
+    cur_func->basic_blocks.push_back(first_else_block);
+    node->else_clause->accept(this);
+    if (!(cur_block->exit_action)) {
+      cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+      std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = following_block->label_full;
+    }
+    cur_block = following_block;
+    cur_func->basic_blocks.push_back(following_block);
+  } else {
+    size_t if_block_start_id = block_counter++;
+    size_t following_block_id = block_counter++;
+    auto first_if_block = std::make_shared<BlockItem>();
+    first_if_block->label_full = "label_" + std::to_string(if_block_start_id);
+    auto following_block = std::make_shared<BlockItem>();
+    following_block->label_full = "label_" + std::to_string(following_block_id);
+    std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->cond = node->condition->IR_result_full;
+    std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->true_label_full = first_if_block->label_full;
+    std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->false_label_full = following_block->label_full;
+    cur_block = first_if_block;
+    cur_func->basic_blocks.push_back(first_if_block);
+    node->if_clause->accept(this);
+    if (!(cur_block->exit_action)) {
+      cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+      std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = following_block->label_full;
+    }
+    cur_block = following_block;
+    cur_func->basic_blocks.push_back(following_block);
+  }
+  just_encountered_jmp = false;
 }
 
 void IRBuilder::ActuralVisit(WhileStatement_ASTNode *node) {
-  // TODO: Implement function body
+  std::string break_target_backup = cur_break_target;
+  std::string continue_target_backup = cur_continue_target;
+
+  size_t checker_block_id = block_counter++;
+  size_t first_loop_body_block_id = block_counter++;
+  size_t following_block_id = block_counter++;
+  auto checker_block = std::make_shared<BlockItem>();
+  auto first_loop_body_block = std::make_shared<BlockItem>();
+  auto following_block = std::make_shared<BlockItem>();
+  checker_block->label_full = "label_" + std::to_string(checker_block_id);
+  first_loop_body_block->label_full = "label_" + std::to_string(first_loop_body_block_id);
+  following_block->label_full = "label_" + std::to_string(following_block_id);
+  cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+  std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = checker_block->label_full;
+  cur_block = checker_block;
+  cur_func->basic_blocks.push_back(checker_block);
+  node->condition->accept(this);
+  cur_block->exit_action = std::make_shared<BRAction>();
+  std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->cond = node->condition->IR_result_full;
+  std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->true_label_full = first_loop_body_block->label_full;
+  std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->false_label_full = following_block->label_full;
+
+  cur_block = first_loop_body_block;
+  cur_func->basic_blocks.push_back(first_loop_body_block);
+  cur_break_target = following_block->label_full;
+  cur_continue_target = checker_block->label_full;
+  node->loop_body->accept(this);
+  if (!(cur_block->exit_action)) {
+    cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+    std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = checker_block->label_full;
+  }
+
+  cur_block = following_block;
+  cur_func->basic_blocks.push_back(following_block);
+
+  cur_break_target = break_target_backup;
+  cur_continue_target = continue_target_backup;
+  just_encountered_jmp = false;
 }
 
 void IRBuilder::ActuralVisit(ForStatement_ASTNode *node) {
-  // TODO: Implement function body
+  std::string break_target_backup = cur_break_target;
+  std::string continue_target_backup = cur_continue_target;
+  size_t checker_block_id = block_counter++;
+  size_t first_loop_body_block_id = block_counter++;
+  size_t step_block_id = block_counter++;
+  size_t following_block_id = block_counter++;
+  auto checker_block = std::make_shared<BlockItem>();
+  auto first_loop_body_block = std::make_shared<BlockItem>();
+  auto step_block = std::make_shared<BlockItem>();
+  auto following_block = std::make_shared<BlockItem>();
+  checker_block->label_full = "label_" + std::to_string(checker_block_id);
+  first_loop_body_block->label_full = "label_" + std::to_string(first_loop_body_block_id);
+  step_block->label_full = "label_" + std::to_string(step_block_id);
+  following_block->label_full = "label_" + std::to_string(following_block_id);
+  if (node->initial) {
+    node->initial->accept(this);  // just finish initialization worker in current block
+  }
+  cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+  std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = checker_block->label_full;
+  cur_block = checker_block;
+  cur_func->basic_blocks.push_back(checker_block);
+  if (node->condition) {
+    node->condition->accept(this);
+    cur_block->exit_action = std::make_shared<BRAction>();
+    std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->cond = node->condition->IR_result_full;
+    std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->true_label_full = first_loop_body_block->label_full;
+    std::dynamic_pointer_cast<BRAction>(cur_block->exit_action)->false_label_full = following_block->label_full;
+  } else {
+    cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+    std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full =
+        first_loop_body_block->label_full;
+  }
+  cur_block = first_loop_body_block;
+  cur_func->basic_blocks.push_back(first_loop_body_block);
+  cur_break_target = following_block->label_full;
+  cur_continue_target = step_block->label_full;
+  node->loop_body->accept(this);
+  if (!(cur_block->exit_action)) {
+    cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+    std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = step_block->label_full;
+  }
+  cur_block = step_block;
+  cur_func->basic_blocks.push_back(step_block);
+  if (node->update) {
+    node->update->accept(this);
+  }
+  cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+  std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = checker_block->label_full;
+  cur_block = following_block;
+  cur_func->basic_blocks.push_back(following_block);
+  cur_break_target = break_target_backup;
+  cur_continue_target = continue_target_backup;
+  just_encountered_jmp = false;
 }
 
 void IRBuilder::ActuralVisit(JmpStatement_ASTNode *node) {
-  // TODO: Implement function body
+  if (node->jmp_type == 1) {
+    // break
+    cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+    std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = cur_break_target;
+    just_encountered_jmp = true;
+  } else if (node->jmp_type == 2) {
+    // continue
+    cur_block->exit_action = std::make_shared<UNConditionJMPAction>();
+    std::dynamic_pointer_cast<UNConditionJMPAction>(cur_block->exit_action)->label_full = cur_continue_target;
+    just_encountered_jmp = true;
+  } else if (node->jmp_type == 0) {
+    // return
+    cur_block->exit_action = std::make_shared<RETAction>();
+    if (node->return_value) {
+      node->return_value->accept(this);
+      std::dynamic_pointer_cast<RETAction>(cur_block->exit_action)->value = node->return_value->IR_result_full;
+    }
+    std::dynamic_pointer_cast<RETAction>(cur_block->exit_action)->type = cur_func->return_type;
+    just_encountered_jmp = true;
+  } else {
+    throw std::runtime_error("unknown jmp type");
+  }
 }
 
 void IRBuilder::ActuralVisit(SuiteStatement_ASTNode *node) {
-  // TODO: Implement function body
+  for (auto &stmt : node->statements) {
+    stmt->accept(this);
+    if (just_encountered_jmp) {
+      just_encountered_jmp = false;
+      break;  // no need to continue
+    }
+  }
 }
 
 // Expression AST Nodes
@@ -70,71 +329,286 @@ void IRBuilder::ActuralVisit(IndexExpr_ASTNode *node) {
 }
 
 void IRBuilder::ActuralVisit(SuffixExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->base->is_requiring_lvalue = true;
+  node->base->accept(this);
+  std::string val_backup = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  auto backup_act = std::make_shared<LoadAction>();
+  cur_block->actions.push_back(backup_act);
+  backup_act->ty = Type_AST2LLVM(node->base->expr_type_info);
+  backup_act->ptr_full = node->base->IR_result_full;
+  backup_act->result_full = val_backup;
+  auto op_act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(op_act);
+  if (node->op == "++") {
+    op_act->op = "add";
+  } else {
+    op_act->op = "sub";
+  }
+  op_act->type = LLVMIRIntType(32);
+  op_act->operand1_full = val_backup;
+  op_act->operand2_full = "1";
+  op_act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  auto store_act = std::make_shared<StoreAction>();
+  cur_block->actions.push_back(store_act);
+  store_act->ty = Type_AST2LLVM(node->base->expr_type_info);
+  store_act->ptr_full = node->base->IR_result_full;
+  store_act->value_full = op_act->result_full;
+  node->IR_result_full = val_backup;
 }
 
 void IRBuilder::ActuralVisit(PrefixExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->base->is_requiring_lvalue = true;
+  node->base->accept(this);
+  std::string val_backup = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  auto backup_act = std::make_shared<LoadAction>();
+  cur_block->actions.push_back(backup_act);
+  backup_act->ty = Type_AST2LLVM(node->base->expr_type_info);
+  backup_act->ptr_full = node->base->IR_result_full;
+  backup_act->result_full = val_backup;
+  auto op_act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(op_act);
+  if (node->op == "++") {
+    op_act->op = "add";
+  } else {
+    op_act->op = "sub";
+  }
+  op_act->type = LLVMIRIntType(32);
+  op_act->operand1_full = val_backup;
+  op_act->operand2_full = "1";
+  op_act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  auto store_act = std::make_shared<StoreAction>();
+  cur_block->actions.push_back(store_act);
+  store_act->ty = Type_AST2LLVM(node->base->expr_type_info);
+  store_act->ptr_full = node->base->IR_result_full;
+  store_act->value_full = op_act->result_full;
+  if (node->is_requiring_lvalue) {
+    node->IR_result_full = node->base->IR_result_full;
+  } else {
+    node->IR_result_full = op_act->result_full;
+  }
 }
 
 void IRBuilder::ActuralVisit(OppositeExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->base->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  act->op = "sub";
+  act->operand1_full = "0";
+  act->operand2_full = node->base->IR_result_full;
+  act->type = LLVMIRIntType(32);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
 }
 
 void IRBuilder::ActuralVisit(LNotExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->base->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  act->op = "xor";
+  act->operand1_full = node->base->IR_result_full;
+  act->operand2_full = "1";
+  act->type = LLVMIRIntType(1);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
 }
 
 void IRBuilder::ActuralVisit(BNotExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->base->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  act->op = "xor";
+  act->operand1_full = node->base->IR_result_full;
+  act->operand2_full = "-1";
+  act->type = LLVMIRIntType(32);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
 }
 
 void IRBuilder::ActuralVisit(MDMExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  if (node->op == "*") {
+    act->op = "mul";
+  } else if (node->op == "/") {
+    act->op = "sdiv";
+  } else if (node->op == "%") {
+    act->op = "srem";
+  } else {
+    throw std::runtime_error("unknown MDM operator");
+  }
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = LLVMIRIntType(32);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
 }
 
 void IRBuilder::ActuralVisit(PMExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  if (node->op == "+") {
+    act->op = "add";
+  } else if (node->op == "-") {
+    act->op = "sub";
+  } else {
+    throw std::runtime_error("unknown PM operator");
+  }
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = LLVMIRIntType(32);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
+  // TODO: support for string concatenation
 }
 
 void IRBuilder::ActuralVisit(RLExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  if (node->op == ">>") {
+    act->op = "ashr";
+  } else if (node->op == "<<") {
+    act->op = "shl";
+  } else {
+    throw std::runtime_error("unknown RL operator");
+  }
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = LLVMIRIntType(32);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
 }
 
 void IRBuilder::ActuralVisit(GGLLExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<ICMPAction>();
+  cur_block->actions.push_back(act);
+  if (node->op == ">=") {
+    act->op = "sge";
+  } else if (node->op == ">") {
+    act->op = "sgt";
+  } else if (node->op == "<=") {
+    act->op = "sle";
+  } else if (node->op == "<") {
+    act->op = "slt";
+  } else {
+    throw std::runtime_error("unknown GGLL operator");
+  }
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = Type_AST2LLVM(node->left->expr_type_info);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
+  // TODO: string comparison
 }
 
 void IRBuilder::ActuralVisit(NEExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<ICMPAction>();
+  cur_block->actions.push_back(act);
+  if (node->op == "!=") {
+    act->op = "ne";
+  } else if (node->op == "==") {
+    act->op = "eq";
+  } else {
+    throw std::runtime_error("unknown NE operator");
+  }
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = Type_AST2LLVM(node->left->expr_type_info);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
+  // TODO: string comparison
 }
 
 void IRBuilder::ActuralVisit(BAndExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  act->op = "and";
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = LLVMIRIntType(32);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
 }
 
 void IRBuilder::ActuralVisit(BXorExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  act->op = "xor";
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = LLVMIRIntType(32);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
 }
 
 void IRBuilder::ActuralVisit(BOrExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  act->op = "or";
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = LLVMIRIntType(32);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
 }
 
 void IRBuilder::ActuralVisit(LAndExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  act->op = "and";
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = LLVMIRIntType(1);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
+  // TODO: short-circuit
 }
 
 void IRBuilder::ActuralVisit(LOrExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->left->accept(this);
+  node->right->accept(this);
+  auto act = std::make_shared<BinaryOperationAction>();
+  cur_block->actions.push_back(act);
+  act->op = "or";
+  act->operand1_full = node->left->IR_result_full;
+  act->operand2_full = node->right->IR_result_full;
+  act->type = LLVMIRIntType(1);
+  act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+  node->IR_result_full = act->result_full;
+  // TODO: short-circuit
 }
 
 void IRBuilder::ActuralVisit(TernaryExpr_ASTNode *node) {
   // TODO: Implement function body
+  throw std::runtime_error("ternary operator not supported");
 }
 
 void IRBuilder::ActuralVisit(AssignExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->dest->is_requiring_lvalue = true;
+  node->dest->accept(this);
+  node->src->accept(this);
+  auto act = std::make_shared<StoreAction>();
+  cur_block->actions.push_back(act);
+  act->ptr_full = node->dest->IR_result_full;
+  act->value_full = node->src->IR_result_full;
+  act->ty = Type_AST2LLVM(node->src->expr_type_info);
 }
 
 void IRBuilder::ActuralVisit(ThisExpr_ASTNode *node) {
@@ -142,15 +616,62 @@ void IRBuilder::ActuralVisit(ThisExpr_ASTNode *node) {
 }
 
 void IRBuilder::ActuralVisit(ParenExpr_ASTNode *node) {
-  // TODO: Implement function body
+  node->expr->accept(this);  // just visit it
 }
 
 void IRBuilder::ActuralVisit(IDExpr_ASTNode *node) {
-  // TODO: Implement function body
+  IRVariableInfo var_info = node->current_scope->fetch_variable_for_IR(node->id);
+  if (var_info.variable_type == 0 || var_info.variable_type == 1) {
+    if (node->is_requiring_lvalue) {
+      node->IR_result_full = var_info.GenerateFullName();
+      return;
+    }
+    auto act = std::make_shared<LoadAction>();
+    cur_block->actions.push_back(act);
+    act->ptr_full = var_info.GenerateFullName();
+    act->ty = var_info.ty;
+    act->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+    node->IR_result_full = act->result_full;
+  } else if (var_info.variable_type == 2) {
+    throw std::runtime_error("not support for class member access");
+  } else if (var_info.variable_type == 3) {
+    if (node->is_requiring_lvalue) {
+      throw std::runtime_error("for argument, lvalue support need additional work");
+    }
+    node->IR_result_full = var_info.GenerateFullName();
+    return;
+  } else {
+    throw std::runtime_error("unknown variable type");
+  }
 }
 
 void IRBuilder::ActuralVisit(FunctionCallExpr_ASTNode *node) {
-  // TODO: Implement function body
+  bool is_member_func = false;
+  if (is_in_class_def) {
+    try {
+      auto schema = global_scope->FetchClassMemberFunction(cur_class_name, node->func_name);
+      is_member_func = true;
+    } catch (...) {
+    }
+  }
+  if (is_member_func) {
+    // TODO: member function call
+    throw std::runtime_error("not support for class member function call");
+  } else {
+    auto call = std::make_shared<CallItem>();
+    call->return_type = Type_AST2LLVM(node->expr_type_info);
+    call->func_name_raw = node->func_name;
+    for (auto &arg : node->arguments) {
+      arg->accept(this);
+      call->args_val_full.push_back(arg->IR_result_full);
+      call->args_ty.push_back(Type_AST2LLVM(arg->expr_type_info));
+    }
+    cur_block->actions.push_back(call);
+    if (!std::holds_alternative<LLVMVOIDType>(call->return_type)) {
+      call->result_full = "%.var.tmp." + std::to_string(tmp_var_counter++);
+      node->IR_result_full = call->result_full;
+    }
+  }
 }
 
 void IRBuilder::ActuralVisit(FormattedStringExpr_ASTNode *node) {
@@ -158,10 +679,49 @@ void IRBuilder::ActuralVisit(FormattedStringExpr_ASTNode *node) {
 }
 
 void IRBuilder::ActuralVisit(ConstantExpr_ASTNode *node) {
-  // TODO: Implement function body
+  if ((node->level == 0) != std::holds_alternative<AtomicConstantType>(node->value)) {
+    throw std::runtime_error("ConstantExpr_ASTNode level not match");
+  }
+  if (std::holds_alternative<AtomicConstantType>(node->value)) {
+    auto &val = std::get<AtomicConstantType>(node->value);
+    if (std::holds_alternative<uint32_t>(val)) {
+      node->IR_result_full = std::to_string(std::get<uint32_t>(val));
+    } else if (std::holds_alternative<bool>(val)) {
+      node->IR_result_full = std::to_string(int(std::get<bool>(val)));
+    } else if (std::holds_alternative<std::string>(val)) {
+      // TODO: string constant
+      throw std::runtime_error("String constant not supported");
+    } else if (std::holds_alternative<NullType>(val)) {
+      node->IR_result_full = "null";
+    } else {
+      throw std::runtime_error("unknown constant type");
+    }
+  } else {
+    // TODO: array constant
+    throw std::runtime_error("Array constant not supported");
+  }
 }
 
 std::shared_ptr<ModuleItem> BuildIR(std::shared_ptr<Program_ASTNode> src) {
   IRBuilder visitor;
+  visitor.prog = std::make_shared<ModuleItem>();
+  auto tmp = std::make_shared<FunctionDeclareItem>();
+  tmp->func_name_raw = "malloc";
+  tmp->return_type = LLVMIRPTRType();
+  tmp->args.push_back(LLVMIRIntType(32));
+  visitor.prog->function_declares.push_back(tmp);
+  tmp = std::make_shared<FunctionDeclareItem>();
+  tmp->func_name_raw = "printInt";
+  tmp->return_type = LLVMVOIDType();
+  tmp->args.push_back(LLVMIRIntType(32));
+  visitor.prog->function_declares.push_back(tmp);
+  tmp = std::make_shared<FunctionDeclareItem>();
+  tmp->func_name_raw = "print";
+  tmp->return_type = LLVMVOIDType();
+  tmp->args.push_back(LLVMIRPTRType());
+  visitor.prog->function_declares.push_back(tmp);
+  visitor.global_scope = std::dynamic_pointer_cast<GlobalScope>(src->current_scope);
+  if (!(visitor.global_scope)) throw std::runtime_error("global scope not found");
   visitor.visit(src.get());
+  return visitor.prog;
 }

src/IR/build.sh

@@ -7,4 +7,4 @@ clang-18 -S -emit-llvm --target=riscv32-unknown-elf -O2 -fno-builtin-printf -fno
         builtin.c -o builtin_intermediate.ll
 sed 's/_builtin_/.builtin./g;s/string_/string./g;s/array_/array./g' builtin_intermediate.ll > builtin.ll
 rm builtin_intermediate.ll
-llc-18 -march=riscv32 builtin.ll -o builtin.s -O2
+llc-18 -march=riscv32 -mattr=+m builtin.ll -o builtin.s -O2

src/IR/builtin.s

@@ -3,7 +3,7 @@
 	.attribute	5, "rv32i2p1_m2p0_a2p1_c2p0"
 	.file	"builtin.c"
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	string.length                   # -- Begin function string.length
 	.p2align	1
 	.type	string.length,@function
@@ -23,7 +23,7 @@ string.length:                          # @string.length
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	string.substring                # -- Begin function string.substring
 	.p2align	1
 	.type	string.substring,@function
@@ -63,7 +63,7 @@ string.substring:                       # @string.substring
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	string.parseInt                 # -- Begin function string.parseInt
 	.p2align	1
 	.type	string.parseInt,@function
@@ -84,7 +84,7 @@ string.parseInt:                        # @string.parseInt
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	string.ord                      # -- Begin function string.ord
 	.p2align	1
 	.type	string.ord,@function
@@ -98,7 +98,7 @@ string.ord:                             # @string.ord
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	print                           # -- Begin function print
 	.p2align	1
 	.type	print,@function
@@ -115,7 +115,7 @@ print:                                  # @print
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	println                         # -- Begin function println
 	.p2align	1
 	.type	println,@function
@@ -132,7 +132,7 @@ println:                                # @println
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	printInt                        # -- Begin function printInt
 	.p2align	1
 	.type	printInt,@function
@@ -149,7 +149,7 @@ printInt:                               # @printInt
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	printlnInt                      # -- Begin function printlnInt
 	.p2align	1
 	.type	printlnInt,@function
@@ -166,7 +166,7 @@ printlnInt:                             # @printlnInt
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	toString                        # -- Begin function toString
 	.p2align	1
 	.type	toString,@function
@@ -196,7 +196,7 @@ toString:                               # @toString
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	getString                       # -- Begin function getString
 	.p2align	1
 	.type	getString,@function
@@ -300,7 +300,7 @@ getString:                              # @getString
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	getInt                          # -- Begin function getInt
 	.p2align	1
 	.type	getInt,@function
@@ -321,7 +321,7 @@ getInt:                                 # @getInt
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	.builtin.AllocateClassBody      # -- Begin function .builtin.AllocateClassBody
 	.p2align	1
 	.type	.builtin.AllocateClassBody,@function
@@ -333,7 +333,7 @@ getInt:                                 # @getInt
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	.builtin.GetArrayLength         # -- Begin function .builtin.GetArrayLength
 	.p2align	1
 	.type	.builtin.GetArrayLength,@function
@@ -355,7 +355,7 @@ getInt:                                 # @getInt
                                         # -- End function
 	.option	pop
 	.option	push
-	.option	arch, +a, +c, +m
+	.option	arch, +a, +c
 	.globl	.builtin.RecursiveAllocateArray # -- Begin function .builtin.RecursiveAllocateArray
 	.p2align	1
 	.type	.builtin.RecursiveAllocateArray,@function

src/main.cpp

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

src/semantic/visitor.cpp

@@ -249,7 +249,7 @@ std::any Visitor::visitClass_var_def(MXParser::Class_var_defContext *context) {
 std::any Visitor::visitClass_constructor(MXParser::Class_constructorContext *context) {
   auto construct_func = std::make_shared<FuncDef_ASTNode>();
   construct_func->type = ASTNodeType::Constructor;
-  construct_func->is_constructor = false;
+  construct_func->is_constructor = true;
   construct_func->start_line = context->getStart()->getLine();
   construct_func->start_char_pos = context->getStart()->getCharPositionInLine();
   construct_func->end_line = context->getStop()->getLine();
@@ -259,6 +259,7 @@ std::any Visitor::visitClass_constructor(MXParser::Class_constructorContext *con
   construct_func->current_scope = cur_scope;
   construct_func->func_name = context->ID()->getText();
   cur_scope->schema.return_type = "void";
+  construct_func->return_type = "void";
   nodetype_stk.push_back({ASTNodeType::Constructor, construct_func->current_scope});
 
   construct_func->func_body = std::dynamic_pointer_cast<SuiteStatement_ASTNode>(
@@ -476,8 +477,7 @@ std::any Visitor::visitDefine_statement(MXParser::Define_statementContext *conte
   assert(nodetype_stk.size() > 0);
   def_stmt->current_scope = nodetype_stk.back().second;
   if (nodetype_stk.size() > 0 && (nodetype_stk.back().first == ASTNodeType::IfStatement ||
-                                  nodetype_stk.back().first == ASTNodeType::WhileStatement ||
-                                  nodetype_stk.back().first == ASTNodeType::ForStatement)) {
+                                  nodetype_stk.back().first == ASTNodeType::WhileStatement)) {
     def_stmt->current_scope = std::make_shared<LocalScope>();
     def_stmt->current_scope->parent = nodetype_stk.back().second.get();
   }