brkstar
3 weeks ago
parent
be143f5326
commit
c2759c27cd
@ -1,200 +1,190 @@
|
||||
#include "sem/Sema.h"
|
||||
|
||||
#include <any>
|
||||
#include <stdexcept>
|
||||
#include <string>
|
||||
|
||||
#include "SysYBaseVisitor.h"
|
||||
#include "sem/SymbolTable.h"
|
||||
#include "utils/Log.h"
|
||||
|
||||
namespace {
|
||||
|
||||
std::string GetLValueName(SysYParser::LValueContext& lvalue) {
|
||||
if (!lvalue.ID()) {
|
||||
throw std::runtime_error(FormatError("sema", "非法左值"));
|
||||
}
|
||||
return lvalue.ID()->getText();
|
||||
}
|
||||
|
||||
class SemaVisitor final : public SysYBaseVisitor {
|
||||
public:
|
||||
std::any visitCompUnit(SysYParser::CompUnitContext* ctx) override {
|
||||
if (!ctx) {
|
||||
throw std::runtime_error(FormatError("sema", "缺少编译单元"));
|
||||
SysYParser::FuncDefContext* FindMainFunc(SysYParser::CompUnitContext& comp_unit) {
|
||||
SysYParser::FuncDefContext* main_func = nullptr;
|
||||
for (auto* func : comp_unit.funcDef()) {
|
||||
if (!func || !func->Ident()) {
|
||||
continue;
|
||||
}
|
||||
auto* func = ctx->funcDef();
|
||||
if (!func || !func->blockStmt()) {
|
||||
throw std::runtime_error(FormatError("sema", "缺少 main 函数定义"));
|
||||
if (func->Ident()->getText() != "main") {
|
||||
continue;
|
||||
}
|
||||
if (!func->ID() || func->ID()->getText() != "main") {
|
||||
throw std::runtime_error(FormatError("sema", "缺少 main 函数定义"));
|
||||
}
|
||||
func->accept(this);
|
||||
if (!seen_return_) {
|
||||
throw std::runtime_error(
|
||||
FormatError("sema", "main 函数必须包含 return 语句"));
|
||||
if (main_func) {
|
||||
throw std::runtime_error(FormatError("sema", "main 函数定义重复"));
|
||||
}
|
||||
return {};
|
||||
main_func = func;
|
||||
}
|
||||
return main_func;
|
||||
}
|
||||
|
||||
std::any visitFuncDef(SysYParser::FuncDefContext* ctx) override {
|
||||
if (!ctx || !ctx->blockStmt()) {
|
||||
throw std::runtime_error(FormatError("sema", "缺少 main 函数定义"));
|
||||
}
|
||||
if (!ctx->funcType() || !ctx->funcType()->INT()) {
|
||||
throw std::runtime_error(FormatError("sema", "当前仅支持 int main"));
|
||||
}
|
||||
const auto& items = ctx->blockStmt()->blockItem();
|
||||
if (items.empty()) {
|
||||
throw std::runtime_error(
|
||||
FormatError("sema", "main 函数不能为空，且必须以 return 结束"));
|
||||
void CheckExpr(SysYParser::ExpContext& exp, const SymbolTable& table,
|
||||
SemanticContext& sema);
|
||||
|
||||
void CheckLVal(SysYParser::LValContext& lval, const SymbolTable& table,
|
||||
SemanticContext& sema) {
|
||||
if (!lval.Ident()) {
|
||||
throw std::runtime_error(FormatError("sema", "左值缺少标识符"));
|
||||
}
|
||||
const std::string name = lval.Ident()->getText();
|
||||
auto* decl = table.Lookup(name);
|
||||
if (!decl) {
|
||||
throw std::runtime_error(FormatError("sema", "使用了未定义的变量: " + name));
|
||||
}
|
||||
sema.BindVarUse(&lval, decl);
|
||||
for (auto* index : lval.exp()) {
|
||||
if (index) {
|
||||
CheckExpr(*index, table, sema);
|
||||
}
|
||||
ctx->blockStmt()->accept(this);
|
||||
return {};
|
||||
}
|
||||
}
|
||||
|
||||
std::any visitBlockStmt(SysYParser::BlockStmtContext* ctx) override {
|
||||
if (!ctx) {
|
||||
throw std::runtime_error(FormatError("sema", "缺少语句块"));
|
||||
}
|
||||
const auto& items = ctx->blockItem();
|
||||
for (size_t i = 0; i < items.size(); ++i) {
|
||||
auto* item = items[i];
|
||||
if (!item) {
|
||||
continue;
|
||||
}
|
||||
if (seen_return_) {
|
||||
throw std::runtime_error(
|
||||
FormatError("sema", "return 必须是 main 函数中的最后一条语句"));
|
||||
}
|
||||
current_item_index_ = i;
|
||||
total_items_ = items.size();
|
||||
item->accept(this);
|
||||
}
|
||||
return {};
|
||||
void CheckPrimary(SysYParser::PrimaryContext& primary, const SymbolTable& table,
|
||||
SemanticContext& sema) {
|
||||
if (primary.Number()) {
|
||||
return;
|
||||
}
|
||||
|
||||
std::any visitBlockItem(SysYParser::BlockItemContext* ctx) override {
|
||||
if (!ctx) {
|
||||
throw std::runtime_error(FormatError("sema", "暂不支持的语句或声明"));
|
||||
}
|
||||
if (ctx->decl()) {
|
||||
ctx->decl()->accept(this);
|
||||
return {};
|
||||
}
|
||||
if (ctx->stmt()) {
|
||||
ctx->stmt()->accept(this);
|
||||
return {};
|
||||
}
|
||||
throw std::runtime_error(FormatError("sema", "暂不支持的语句或声明"));
|
||||
if (primary.lVal()) {
|
||||
CheckLVal(*primary.lVal(), table, sema);
|
||||
return;
|
||||
}
|
||||
|
||||
std::any visitDecl(SysYParser::DeclContext* ctx) override {
|
||||
if (!ctx) {
|
||||
throw std::runtime_error(FormatError("sema", "非法变量声明"));
|
||||
}
|
||||
if (!ctx->btype() || !ctx->btype()->INT()) {
|
||||
throw std::runtime_error(FormatError("sema", "当前仅支持局部 int 变量声明"));
|
||||
}
|
||||
auto* var_def = ctx->varDef();
|
||||
if (!var_def || !var_def->lValue()) {
|
||||
throw std::runtime_error(FormatError("sema", "非法变量声明"));
|
||||
}
|
||||
const std::string name = GetLValueName(*var_def->lValue());
|
||||
if (table_.Contains(name)) {
|
||||
throw std::runtime_error(FormatError("sema", "重复定义变量: " + name));
|
||||
}
|
||||
if (auto* init = var_def->initValue()) {
|
||||
if (!init->exp()) {
|
||||
throw std::runtime_error(FormatError("sema", "当前不支持聚合初始化"));
|
||||
}
|
||||
init->exp()->accept(this);
|
||||
}
|
||||
table_.Add(name, var_def);
|
||||
return {};
|
||||
if (primary.exp()) {
|
||||
CheckExpr(*primary.exp(), table, sema);
|
||||
return;
|
||||
}
|
||||
|
||||
std::any visitStmt(SysYParser::StmtContext* ctx) override {
|
||||
if (!ctx || !ctx->returnStmt()) {
|
||||
throw std::runtime_error(FormatError("sema", "暂不支持的语句或声明"));
|
||||
}
|
||||
ctx->returnStmt()->accept(this);
|
||||
return {};
|
||||
throw std::runtime_error(FormatError("sema", "暂不支持的表达式形式"));
|
||||
}
|
||||
|
||||
void CheckUnaryExpr(SysYParser::UnaryExpContext& unary, const SymbolTable& table,
|
||||
SemanticContext& sema) {
|
||||
if (unary.primary()) {
|
||||
CheckPrimary(*unary.primary(), table, sema);
|
||||
return;
|
||||
}
|
||||
|
||||
std::any visitReturnStmt(SysYParser::ReturnStmtContext* ctx) override {
|
||||
if (!ctx || !ctx->exp()) {
|
||||
throw std::runtime_error(FormatError("sema", "return 缺少表达式"));
|
||||
}
|
||||
ctx->exp()->accept(this);
|
||||
seen_return_ = true;
|
||||
if (current_item_index_ + 1 != total_items_) {
|
||||
throw std::runtime_error(
|
||||
FormatError("sema", "return 必须是 main 函数中的最后一条语句"));
|
||||
}
|
||||
return {};
|
||||
if (unary.unaryExp()) {
|
||||
CheckUnaryExpr(*unary.unaryExp(), table, sema);
|
||||
return;
|
||||
}
|
||||
|
||||
std::any visitParenExp(SysYParser::ParenExpContext* ctx) override {
|
||||
if (!ctx || !ctx->exp()) {
|
||||
throw std::runtime_error(FormatError("sema", "非法括号表达式"));
|
||||
if (unary.funcRParams()) {
|
||||
for (auto* arg : unary.funcRParams()->exp()) {
|
||||
if (arg) {
|
||||
CheckExpr(*arg, table, sema);
|
||||
}
|
||||
}
|
||||
ctx->exp()->accept(this);
|
||||
return {};
|
||||
}
|
||||
}
|
||||
|
||||
std::any visitVarExp(SysYParser::VarExpContext* ctx) override {
|
||||
if (!ctx || !ctx->var()) {
|
||||
throw std::runtime_error(FormatError("sema", "非法变量表达式"));
|
||||
void CheckMulExpr(SysYParser::MulExpContext& mul, const SymbolTable& table,
|
||||
SemanticContext& sema) {
|
||||
for (auto* unary : mul.unaryExp()) {
|
||||
if (unary) {
|
||||
CheckUnaryExpr(*unary, table, sema);
|
||||
}
|
||||
ctx->var()->accept(this);
|
||||
return {};
|
||||
}
|
||||
}
|
||||
|
||||
std::any visitNumberExp(SysYParser::NumberExpContext* ctx) override {
|
||||
if (!ctx || !ctx->number() || !ctx->number()->ILITERAL()) {
|
||||
throw std::runtime_error(FormatError("sema", "当前仅支持整数字面量"));
|
||||
void CheckAddExpr(SysYParser::AddExpContext& add, const SymbolTable& table,
|
||||
SemanticContext& sema) {
|
||||
for (auto* mul : add.mulExp()) {
|
||||
if (mul) {
|
||||
CheckMulExpr(*mul, table, sema);
|
||||
}
|
||||
return {};
|
||||
}
|
||||
}
|
||||
|
||||
std::any visitAdditiveExp(SysYParser::AdditiveExpContext* ctx) override {
|
||||
if (!ctx || !ctx->exp(0) || !ctx->exp(1)) {
|
||||
throw std::runtime_error(FormatError("sema", "暂不支持的表达式形式"));
|
||||
}
|
||||
ctx->exp(0)->accept(this);
|
||||
ctx->exp(1)->accept(this);
|
||||
return {};
|
||||
void CheckExpr(SysYParser::ExpContext& exp, const SymbolTable& table,
|
||||
SemanticContext& sema) {
|
||||
if (!exp.addExp()) {
|
||||
throw std::runtime_error(FormatError("sema", "非法表达式"));
|
||||
}
|
||||
CheckAddExpr(*exp.addExp(), table, sema);
|
||||
}
|
||||
|
||||
} // namespace
|
||||
|
||||
SemanticContext RunSema(SysYParser::CompUnitContext& comp_unit) {
|
||||
auto* func = FindMainFunc(comp_unit);
|
||||
if (!func || !func->block()) {
|
||||
throw std::runtime_error(FormatError("sema", "缺少 main 函数定义"));
|
||||
}
|
||||
|
||||
SymbolTable table;
|
||||
SemanticContext sema;
|
||||
bool seen_return = false;
|
||||
|
||||
const auto& items = func->block()->blockItem();
|
||||
if (items.empty()) {
|
||||
throw std::runtime_error(
|
||||
FormatError("sema", "main 函数不能为空，且必须以 return 结束"));
|
||||
}
|
||||
|
||||
std::any visitVar(SysYParser::VarContext* ctx) override {
|
||||
if (!ctx || !ctx->ID()) {
|
||||
throw std::runtime_error(FormatError("sema", "非法变量引用"));
|
||||
for (size_t i = 0; i < items.size(); ++i) {
|
||||
auto* item = items[i];
|
||||
if (!item) {
|
||||
continue;
|
||||
}
|
||||
const std::string name = ctx->ID()->getText();
|
||||
auto* decl = table_.Lookup(name);
|
||||
if (!decl) {
|
||||
throw std::runtime_error(FormatError("sema", "使用了未定义的变量: " + name));
|
||||
if (seen_return) {
|
||||
throw std::runtime_error(
|
||||
FormatError("sema", "return 必须是 main 函数中的最后一条语句"));
|
||||
}
|
||||
|
||||
if (auto* decl = item->decl() ? item->decl()->varDecl() : nullptr) {
|
||||
for (auto* def : decl->varDef()) {
|
||||
if (!def || !def->Ident()) {
|
||||
continue;
|
||||
}
|
||||
const std::string name = def->Ident()->getText();
|
||||
if (table.Contains(name)) {
|
||||
throw std::runtime_error(FormatError("sema", "重复定义变量: " + name));
|
||||
}
|
||||
if (!def->constExp().empty()) {
|
||||
throw std::runtime_error(
|
||||
FormatError("sema", "当前 IR 仅支持标量局部变量"));
|
||||
}
|
||||
if (auto* init = def->initVal()) {
|
||||
if (!init->exp()) {
|
||||
throw std::runtime_error(
|
||||
FormatError("sema", "当前 IR 仅支持标量表达式初始化"));
|
||||
}
|
||||
CheckExpr(*init->exp(), table, sema);
|
||||
}
|
||||
table.Add(name, def);
|
||||
}
|
||||
continue;
|
||||
}
|
||||
sema_.BindVarUse(ctx, decl);
|
||||
return {};
|
||||
}
|
||||
|
||||
SemanticContext TakeSemanticContext() { return std::move(sema_); }
|
||||
if (auto* stmt = item->stmt(); stmt && stmt->returnStmt()) {
|
||||
auto* ret = stmt->returnStmt();
|
||||
if (!ret->exp()) {
|
||||
throw std::runtime_error(FormatError("sema", "main 函数必须返回一个值"));
|
||||
}
|
||||
CheckExpr(*ret->exp(), table, sema);
|
||||
seen_return = true;
|
||||
if (i + 1 != items.size()) {
|
||||
throw std::runtime_error(
|
||||
FormatError("sema", "return 必须是 main 函数中的最后一条语句"));
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
private:
|
||||
SymbolTable table_;
|
||||
SemanticContext sema_;
|
||||
bool seen_return_ = false;
|
||||
size_t current_item_index_ = 0;
|
||||
size_t total_items_ = 0;
|
||||
};
|
||||
throw std::runtime_error(FormatError("sema", "暂不支持的语句或声明"));
|
||||
}
|
||||
|
||||
} // namespace
|
||||
if (!seen_return) {
|
||||
throw std::runtime_error(FormatError("sema", "main 函数必须包含 return 语句"));
|
||||
}
|
||||
|
||||
SemanticContext RunSema(SysYParser::CompUnitContext& comp_unit) {
|
||||
SemaVisitor visitor;
|
||||
comp_unit.accept(&visitor);
|
||||
return visitor.TakeSemanticContext();
|
||||
return sema;
|
||||
}
|
||||
|
||||
Loading…
Reference in new issue