|
|
// IRGen：语法树 → IR
|
|
|
// 按语法树节点类型分发到 IRGenFunc/IRGenStmt/IRGenExp/IRGenDecl
|
|
|
|
|
|
#pragma once
|
|
|
|
|
|
#include <any>
|
|
|
#include <memory>
|
|
|
#include <string>
|
|
|
#include <unordered_map>
|
|
|
#include <vector>
|
|
|
|
|
|
#include "SysYBaseVisitor.h"
|
|
|
#include "SysYParser.h"
|
|
|
#include "ir/IR.h"
|
|
|
#include "sem/Sema.h"
|
|
|
|
|
|
class IRGenImpl final : public SysYBaseVisitor {
|
|
|
public:
|
|
|
IRGenImpl(ir::Module& module, const SemanticContext& sema);
|
|
|
|
|
|
// ── 顶层 ──────────────────────────────────────────────────────────────────
|
|
|
std::any visitCompUnit(SysYParser::CompUnitContext* ctx) override;
|
|
|
|
|
|
// ── 函数 ──────────────────────────────────────────────────────────────────
|
|
|
std::any visitFuncDef(SysYParser::FuncDefContext* ctx) override;
|
|
|
|
|
|
// ── 块与块内项 ─────────────────────────────────────────────────────────────
|
|
|
std::any visitBlock(SysYParser::BlockContext* ctx) override;
|
|
|
std::any visitBlockItem(SysYParser::BlockItemContext* ctx) override;
|
|
|
|
|
|
// ── 声明 ──────────────────────────────────────────────────────────────────
|
|
|
std::any visitDecl(SysYParser::DeclContext* ctx) override;
|
|
|
std::any visitVarDecl(SysYParser::VarDeclContext* ctx) override;
|
|
|
std::any visitVarDef(SysYParser::VarDefContext* ctx) override;
|
|
|
std::any visitConstDecl(SysYParser::ConstDeclContext* ctx) override;
|
|
|
|
|
|
// ── 语句 ──────────────────────────────────────────────────────────────────
|
|
|
std::any visitStmt(SysYParser::StmtContext* ctx) override;
|
|
|
|
|
|
// ── 表达式（返回 ir::Value*，i32） ─────────────────────────────────────────
|
|
|
std::any visitExp(SysYParser::ExpContext* ctx) override;
|
|
|
std::any visitAddExp(SysYParser::AddExpContext* ctx) override;
|
|
|
std::any visitMulExp(SysYParser::MulExpContext* ctx) override;
|
|
|
std::any visitUnaryExp(SysYParser::UnaryExpContext* ctx) override;
|
|
|
std::any visitPrimaryExp(SysYParser::PrimaryExpContext* ctx) override;
|
|
|
std::any visitLVar(SysYParser::LVarContext* ctx) override;
|
|
|
std::any visitNumber(SysYParser::NumberContext* ctx) override;
|
|
|
|
|
|
// ── 条件（返回 ir::Value*，i1） ────────────────────────────────────────────
|
|
|
std::any visitCond(SysYParser::CondContext* ctx) override;
|
|
|
std::any visitLOrExp(SysYParser::LOrExpContext* ctx) override;
|
|
|
std::any visitLAndExp(SysYParser::LAndExpContext* ctx) override;
|
|
|
std::any visitEqExp(SysYParser::EqExpContext* ctx) override;
|
|
|
std::any visitRelExp(SysYParser::RelExpContext* ctx) override;
|
|
|
|
|
|
private:
|
|
|
// BlockFlow 用于通知调用者当前块是否已终结
|
|
|
enum class BlockFlow { Continue, Terminated };
|
|
|
|
|
|
// ── 辅助函数 ──────────────────────────────────────────────────────────────
|
|
|
// 求值表达式，返回 i32 值
|
|
|
ir::Value* EvalExpr(SysYParser::ExpContext& expr);
|
|
|
ir::Value* EvalExprAdd(SysYParser::AddExpContext& expr);
|
|
|
// 求值条件，返回 i1 值（供 if/while 使用）
|
|
|
ir::Value* EvalCond(SysYParser::CondContext& cond);
|
|
|
// 把整型值转为 i1 条件（icmp ne i32 val, 0）
|
|
|
ir::Value* ToI1(ir::Value* v);
|
|
|
// 把 i1 值零扩展为 i32
|
|
|
ir::Value* ToI32(ir::Value* v);
|
|
|
// 隐式类型转换：确保两个操作数类型一致（int 转 float）
|
|
|
void ImplicitConvert(ir::Value*& lhs, ir::Value*& rhs);
|
|
|
// 转换为 float（如果是 int）
|
|
|
ir::Value* ToFloat(ir::Value* v);
|
|
|
// 转换为 int（如果是 float）
|
|
|
ir::Value* ToInt(ir::Value* v);
|
|
|
|
|
|
// 访问一条块内项，返回流控状态
|
|
|
BlockFlow VisitBlockItemResult(SysYParser::BlockItemContext& item);
|
|
|
// 访问 stmt，返回流控状态
|
|
|
BlockFlow VisitStmt(SysYParser::StmtContext& stmt);
|
|
|
|
|
|
// 向外部函数声明注册（幂等）
|
|
|
void EnsureExternalDecl(const std::string& name);
|
|
|
|
|
|
// ── 状态 ──────────────────────────────────────────────────────────────────
|
|
|
ir::Module& module_;
|
|
|
const SemanticContext& sema_;
|
|
|
ir::Function* func_ = nullptr;
|
|
|
ir::IRBuilder builder_;
|
|
|
|
|
|
// 局部变量/参数的存储槽位：声明上下文 → alloca Value*
|
|
|
std::unordered_map<antlr4::ParserRuleContext*, ir::Value*> storage_map_;
|
|
|
// 全局变量/常量的存储槽位（不随函数切换清空）
|
|
|
std::unordered_map<antlr4::ParserRuleContext*, ir::Value*> global_storage_map_;
|
|
|
|
|
|
// 数组维度信息：声明上下文 → 各维度大小（从外到内）
|
|
|
std::unordered_map<antlr4::ParserRuleContext*, std::vector<int>> array_dims_;
|
|
|
std::unordered_map<antlr4::ParserRuleContext*, std::vector<int>> global_array_dims_;
|
|
|
|
|
|
// 辅助：求常量表达式的整数值（失败返回 0）
|
|
|
int EvalConstExprInt(SysYParser::ConstExpContext* ctx);
|
|
|
// 辅助：计算一个 lVar 的元素地址（支持多维数组）
|
|
|
ir::Value* EvalLVarAddr(SysYParser::LVarContext* ctx);
|
|
|
|
|
|
// 是否处于全局作用域（visitCompUnit 中处理 decl 时）
|
|
|
bool in_global_scope_ = false;
|
|
|
|
|
|
// 循环上下文（break/continue 的目标块）
|
|
|
struct LoopCtx {
|
|
|
ir::BasicBlock* cond_bb; // continue 跳到这里
|
|
|
ir::BasicBlock* after_bb; // break 跳到这里
|
|
|
};
|
|
|
std::vector<LoopCtx> loop_stack_;
|
|
|
};
|
|
|
|
|
|
std::unique_ptr<ir::Module> GenerateIR(SysYParser::CompUnitContext& tree,
|
|
|
const SemanticContext& sema);
|
