[lab2] runnable

.gitignore

@@ -40,4 +40,8 @@ doxygen
 build
 .antlr
 
-tmp
+tmp
+
+GPATH
+GRTAGS
+GTAGS

src/SysYIRGenerator.cpp

@@ -1,30 +1,239 @@
-#include "IR.h"
-#include <any>
-#include <memory>
-using namespace std;
 #include "SysYIRGenerator.h"
+#include <iomanip>
 
-namespace sysy {
+std::string SysYIRGenerator::generateIR(SysYParser::CompUnitContext* unit) {
-
+    visitCompUnit(unit);
-any SysYIRGenerator::visitCompUnit(SysYParser::CompUnitContext *ctx) {
+    return irStream.str();
-  // create the IR module
-  auto pModule = new Module();
-  assert(pModule);
-  module.reset(pModule);
-  // generates globals and functions
-  visitChildren(ctx);
-  // return the IR module
-  return pModule;
-}
-std::any
-SysYIRGenerator::visitFuncRParams(SysYParser::FuncRParamsContext *ctx) {
-  return visitChildren(ctx);
-}
-std::any SysYIRGenerator::visitNumber(SysYParser::NumberContext *ctx) {
-  return visitChildren(ctx);
-}
-std::any SysYIRGenerator::visitString(SysYParser::StringContext *ctx) {
-  return visitChildren(ctx);
 }
 
-} // namespace sysy
+std::string SysYIRGenerator::getNextTemp() {
+    return "%" + std::to_string(tempCounter++);
+}
+
+std::string SysYIRGenerator::getLLVMType(const std::string& type) {
+    if (type == "int") return "i32";
+    if (type == "float") return "float";
+    if (type.find("[]") != std::string::npos)
+        return getLLVMType(type.substr(0, type.size() - 2)) + "*";
+    return "i32";
+}
+
+std::any SysYIRGenerator::visitCompUnit(SysYParser::CompUnitContext* ctx) {
+    for (auto decl : ctx->decl()) {
+        decl->accept(this);
+    }
+    for (auto funcDef : ctx->funcDef()) {
+        funcDef->accept(this);
+    }
+    return nullptr;
+}
+
+std::any SysYIRGenerator::visitConstDecl(SysYParser::ConstDeclContext* ctx) {
+    // 常量声明暂不处理（LLVM IR 中常量通常内联）
+    return nullptr;
+}
+
+std::any SysYIRGenerator::visitVarDecl(SysYParser::VarDeclContext* ctx) {
+    for (auto varDef : ctx->varDef()) {
+        varDef->accept(this);
+    }
+    return nullptr;
+}
+
+std::any SysYIRGenerator::visitFuncDef(SysYParser::FuncDefContext* ctx) {
+    currentFunction = ctx->Ident()->getText();
+    symbolTable.clear();
+
+    // 函数头
+    std::string returnType = getLLVMType(ctx->funcType()->getText());
+    irStream << "define " << returnType << " @" << currentFunction << "(";
+
+    // 参数
+    auto paramsCtx = ctx->funcFParams();
+    if (paramsCtx) {
+        auto params = paramsCtx->funcFParam();
+        for (size_t i = 0; i < params.size(); ++i) {
+            if (i > 0) irStream << ", ";
+            auto param = params[i];
+            std::string paramName = "%" + std::to_string(i);
+            std::string paramType = getLLVMType(param->bType()->getText());
+            irStream << paramType << " " << paramName;
+
+            // 分配参数
+            std::string allocaName = getNextTemp();
+            symbolTable[param->Ident()->getText()] = allocaName;
+            irStream << "\n  " << allocaName << " = alloca " << paramType;
+            irStream << "\n  store " << paramType << " %" << i << ", " << paramType << "* " << allocaName;
+        }
+    }
+    irStream << ") {\nentry:\n";
+
+    // 函数体
+    ctx->blockStmt()->accept(this);
+
+    // 默认返回值
+    if (returnType == "void") {
+        irStream << "  ret void\n";
+    } else {
+        irStream << "  ret " << returnType << " 0\n";
+    }
+    irStream << "}\n\n";
+    return nullptr;
+}
+
+std::any SysYIRGenerator::visitBlockStmt(SysYParser::BlockStmtContext* ctx) {
+    for (auto item : ctx->blockItem()) {
+        item->accept(this);
+    }
+    return nullptr;
+}
+
+std::any SysYIRGenerator::visitStmt(SysYParser::StmtContext* ctx) {
+    if (ctx->lValue() && ctx->exp()) {
+        // 赋值语句
+        std::string lhs = std::any_cast<std::string>(ctx->lValue()->accept(this));
+        std::string rhs = std::any_cast<std::string>(ctx->exp()->accept(this));
+        irStream << "  store " << getLLVMType("") << " " << rhs << ", " << getLLVMType("") << "* " << lhs << "\n";
+    } else if (ctx->RETURN()) {
+        // 返回语句
+        if (ctx->exp()) {
+            std::string value = std::any_cast<std::string>(ctx->exp()->accept(this));
+            irStream << "  ret " << getLLVMType("") << " " << value << "\n";
+        } else {
+            irStream << "  ret void\n";
+        }
+    }
+    return nullptr;
+}
+
+std::any SysYIRGenerator::visitLValue(SysYParser::LValueContext* ctx) {
+    std::string varName = ctx->Ident()->getText();
+    if (symbolTable.find(varName) == symbolTable.end()) {
+        std::string allocaName = getNextTemp();
+        symbolTable[varName] = allocaName;
+        irStream << "  " << allocaName << " = alloca " << getLLVMType("") << "\n";
+    }
+    return symbolTable[varName];
+}
+
+std::any SysYIRGenerator::visitNumber(SysYParser::NumberContext* ctx) {
+    if (ctx->ILITERAL()) {
+        return "i32 " + ctx->ILITERAL()->getText();
+    } else if (ctx->FLITERAL()) {
+        return "float " + ctx->FLITERAL()->getText();
+    }
+    return "";
+}
+
+std::any SysYIRGenerator::visitUnaryExp(SysYParser::UnaryExpContext* ctx) {
+    if (ctx->unaryOp()) {
+        std::string operand = std::any_cast<std::string>(ctx->unaryExp()->accept(this));
+        std::string op = ctx->unaryOp()->getText();
+        std::string temp = getNextTemp();
+        if (op == "-") {
+            irStream << "  " << temp << " = sub " << getLLVMType("") << " 0, " << operand << "\n";
+        } else if (op == "!") {
+            irStream << "  " << temp << " = xor " << getLLVMType("") << " " << operand << ", 1\n";
+        }
+        return temp;
+    }
+    return ctx->primaryExp()->accept(this);
+}
+
+std::any SysYIRGenerator::visitMulExp(SysYParser::MulExpContext* ctx) {
+    auto unaryExps = ctx->unaryExp();
+    std::string left = std::any_cast<std::string>(unaryExps[0]->accept(this));
+    for (size_t i = 1; i < unaryExps.size(); ++i) {
+        std::string right = std::any_cast<std::string>(unaryExps[i]->accept(this));
+        std::string op = ctx->children[2 * i - 1]->getText();
+        std::string temp = getNextTemp();
+        if (op == "*") {
+            irStream << "  " << temp << " = mul " << getLLVMType("") << " " << left << ", " << right << "\n";
+        } else if (op == "/") {
+            irStream << "  " << temp << " = sdiv " << getLLVMType("") << " " << left << ", " << right << "\n";
+        } else if (op == "%") {
+            irStream << "  " << temp << " = srem " << getLLVMType("") << " " << left << ", " << right << "\n";
+        }
+        left = temp;
+    }
+    return left;
+}
+
+std::any SysYIRGenerator::visitAddExp(SysYParser::AddExpContext* ctx) {
+    auto mulExps = ctx->mulExp();
+    std::string left = std::any_cast<std::string>(mulExps[0]->accept(this));
+    for (size_t i = 1; i < mulExps.size(); ++i) {
+        std::string right = std::any_cast<std::string>(mulExps[i]->accept(this));
+        std::string op = ctx->children[2 * i - 1]->getText();
+        std::string temp = getNextTemp();
+        if (op == "+") {
+            irStream << "  " << temp << " = add " << getLLVMType("") << " " << left << ", " << right << "\n";
+        } else if (op == "-") {
+            irStream << "  " << temp << " = sub " << getLLVMType("") << " " << left << ", " << right << "\n";
+        }
+        left = temp;
+    }
+    return left;
+}
+
+std::any SysYIRGenerator::visitRelExp(SysYParser::RelExpContext* ctx) {
+    auto addExps = ctx->addExp();
+    std::string left = std::any_cast<std::string>(addExps[0]->accept(this));
+    for (size_t i = 1; i < addExps.size(); ++i) {
+        std::string right = std::any_cast<std::string>(addExps[i]->accept(this));
+        std::string op = ctx->children[2 * i - 1]->getText();
+        std::string temp = getNextTemp();
+        if (op == "<") {
+            irStream << "  " << temp << " = icmp slt " << getLLVMType("") << " " << left << ", " << right << "\n";
+        } else if (op == ">") {
+            irStream << "  " << temp << " = icmp sgt " << getLLVMType("") << " " << left << ", " << right << "\n";
+        } else if (op == "<=") {
+            irStream << "  " << temp << " = icmp sle " << getLLVMType("") << " " << left << ", " << right << "\n";
+        } else if (op == ">=") {
+            irStream << "  " << temp << " = icmp sge " << getLLVMType("") << " " << left << ", " << right << "\n";
+        }
+        left = temp;
+    }
+    return left;
+}
+
+std::any SysYIRGenerator::visitEqExp(SysYParser::EqExpContext* ctx) {
+    auto relExps = ctx->relExp();
+    std::string left = std::any_cast<std::string>(relExps[0]->accept(this));
+    for (size_t i = 1; i < relExps.size(); ++i) {
+        std::string right = std::any_cast<std::string>(relExps[i]->accept(this));
+        std::string op = ctx->children[2 * i - 1]->getText();
+        std::string temp = getNextTemp();
+        if (op == "==") {
+            irStream << "  " << temp << " = icmp eq " << getLLVMType("") << " " << left << ", " << right << "\n";
+        } else if (op == "!=") {
+            irStream << "  " << temp << " = icmp ne " << getLLVMType("") << " " << left << ", " << right << "\n";
+        }
+        left = temp;
+    }
+    return left;
+}
+
+std::any SysYIRGenerator::visitLAndExp(SysYParser::LAndExpContext* ctx) {
+    auto eqExps = ctx->eqExp();
+    std::string left = std::any_cast<std::string>(eqExps[0]->accept(this));
+    for (size_t i = 1; i < eqExps.size(); ++i) {
+        std::string right = std::any_cast<std::string>(eqExps[i]->accept(this));
+        std::string temp = getNextTemp();
+        irStream << "  " << temp << " = and " << getLLVMType("") << " " << left << ", " << right << "\n";
+        left = temp;
+    }
+    return left;
+}
+
+std::any SysYIRGenerator::visitLOrExp(SysYParser::LOrExpContext* ctx) {
+    auto lAndExps = ctx->lAndExp();
+    std::string left = std::any_cast<std::string>(lAndExps[0]->accept(this));
+    for (size_t i = 1; i < lAndExps.size(); ++i) {
+        std::string right = std::any_cast<std::string>(lAndExps[i]->accept(this));
+        std::string temp = getNextTemp();
+        irStream << "  " << temp << " = or " << getLLVMType("") << " " << left << ", " << right << "\n";
+        left = temp;
+    }
+    return left;
+}

src/SysYIRGenerator.h

@@ -1,29 +1,41 @@
 #pragma once
-
-#include "IR.h"
-#include "IRBuilder.h"
 #include "SysYBaseVisitor.h"
 #include "SysYParser.h"
-#include <memory>
+#include <sstream>
-
+#include <map>
-namespace sysy {
+#include <vector>
 
 class SysYIRGenerator : public SysYBaseVisitor {
+public:
+    std::string generateIR(SysYParser::CompUnitContext* unit); // 公共接口，用于生成 IR
+    std::string getIR() const { return irStream.str(); } // 获取生成的 IR
+
 private:
-  std::unique_ptr<Module> module;
+    std::stringstream irStream;
-  IRBuilder builder;
+    int tempCounter = 0;
+    std::map<std::string, std::string> symbolTable; // 符号表
+    std::vector<std::string> globalVars; // 全局变量
+    std::string currentFunction; // 当前函数名
+    std::vector<std::string> breakStack; // break 目标标签栈
+    std::vector<std::string> continueStack; // continue 目标标签栈
 
-public:
+    std::string getNextTemp(); // 获取下一个临时变量名
-  SysYIRGenerator() = default;
+    std::string getLLVMType(const std::string& type); // 获取 LLVM 类型
 
-public:
+    // 访问方法
-  Module *get() const { return module.get(); }
+    std::any visitCompUnit(SysYParser::CompUnitContext* ctx) override;
-
+    std::any visitConstDecl(SysYParser::ConstDeclContext* ctx) override;
-public:
+    std::any visitVarDecl(SysYParser::VarDeclContext* ctx) override;
-  std::any visitCompUnit(SysYParser::CompUnitContext *ctx) override;
+    std::any visitFuncDef(SysYParser::FuncDefContext* ctx) override;
-  std::any visitFuncRParams(SysYParser::FuncRParamsContext *ctx) override;
+    std::any visitBlockStmt(SysYParser::BlockStmtContext* ctx) override;
-  std::any visitNumber(SysYParser::NumberContext *ctx) override;
+    std::any visitStmt(SysYParser::StmtContext* ctx) override;
-  std::any visitString(SysYParser::StringContext *ctx) override;
+    std::any visitLValue(SysYParser::LValueContext* ctx) override;
-}; // class SysYIRGenerator
+    std::any visitNumber(SysYParser::NumberContext* ctx) override;
-
+    std::any visitUnaryExp(SysYParser::UnaryExpContext* ctx) override;
-} // namespace sysy
+    std::any visitMulExp(SysYParser::MulExpContext* ctx) override;
+    std::any visitAddExp(SysYParser::AddExpContext* ctx) override;
+    std::any visitRelExp(SysYParser::RelExpContext* ctx) override;
+    std::any visitEqExp(SysYParser::EqExpContext* ctx) override;
+    std::any visitLAndExp(SysYParser::LAndExpContext* ctx) override;
+    std::any visitLOrExp(SysYParser::LOrExpContext* ctx) override;
+};

src/sysyc.cpp

@@ -18,9 +18,9 @@ static bool argFormat = false;
 void usage(int code = EXIT_FAILURE) {
   const char *msg = "Usage: sysyc [options] inputfile\n\n"
                     "Supported options:\n"
-                    "  -h \tprint help message and exit\n";
+                    "  -h \tprint help message and exit\n"
-  "  -f \tpretty-format the input file\n";
+                    "  -f \tpretty-format the input file\n"
-  "  -s {ast,ir,asm}\tstop after generating AST/IR/Assembly\n";
+                    "  -s {ast,ir,asm}\tstop after generating AST/IR/Assembly\n";
   cerr << msg;
   exit(code);
 }
@@ -51,14 +51,14 @@ void parseArgs(int argc, char **argv) {
 int main(int argc, char **argv) {
   parseArgs(argc, argv);
 
-  // open the input file
+  // 打开输入文件
   ifstream fin(argInputFile);
   if (not fin) {
     cerr << "Failed to open file " << argv[1];
     return EXIT_FAILURE;
   }
 
-  // parse sysy source to AST
+  // 解析 SysY 源码为 AST
   ANTLRInputStream input(fin);
   SysYLexer lexer(&input);
   CommonTokenStream tokens(&lexer);
@@ -69,28 +69,27 @@ int main(int argc, char **argv) {
     return EXIT_SUCCESS;
   }
 
-  // pretty format the input file
+  // 格式化输入文件
   if (argFormat) {
     ASTPrinter printer;
     printer.visitCompUnit(moduleAST);
     return EXIT_SUCCESS;
   }
 
-  // visit AST to generate IR
+  // 遍历 AST 生成 IR
   SysYIRGenerator generator;
-  generator.visitCompUnit(moduleAST);
+  generator.generateIR(moduleAST); // 使用公共接口生成 IR
-  auto moduleIR = generator.get();
   if (argStopAfter == "ir") {
-    moduleIR->print(cout);
+    cout << generator.getIR(); // 输出生成的 IR
     return EXIT_SUCCESS;
   }
 
-  // generate assembly
+  // // 生成汇编代码
-  CodeGen codegen(moduleIR);
+  // CodeGen codegen(generator.getIR()); // 假设 CodeGen 接受字符串作为输入
-  string asmCode = codegen.code_gen();
+  // string asmCode = codegen.code_gen();
-  cout << asmCode << endl;
+  // cout << asmCode << endl;
-  if (argStopAfter == "asm")
+  // if (argStopAfter == "asm")
-    return EXIT_SUCCESS;
+  //   return EXIT_SUCCESS;
 
   return EXIT_SUCCESS;
 }