[lab2][night release]temporary stable version 0.1

src/SysYIRGenerator.cpp

@@ -1,7 +1,11 @@
 // SysYIRGenerator.cpp
+// TODO：类型转换及其检查
+// TODO：sysy库函数处理
+// TODO：数组处理
+// TODO：对while、continue、break的测试
 #include "SysYIRGenerator.h"
 #include <iomanip>
-// #TODO浮点数精度还是有问题
+
 std::string SysYIRGenerator::generateIR(SysYParser::CompUnitContext* unit) {
     visitCompUnit(unit);
     return irStream.str();
@@ -26,22 +30,83 @@ std::any SysYIRGenerator::visitCompUnit(SysYParser::CompUnitContext* ctx) {
         decl->accept(this);
     }
     for (auto funcDef : ctx->funcDef()) {
+        inFunction = true; // 进入函数定义
         funcDef->accept(this);
+        inFunction = false; // 离开函数定义
+    }
+    return nullptr;
+}
+
+std::any SysYIRGenerator::visitVarDecl(SysYParser::VarDeclContext* ctx) {
+    // TODO：数组初始化
+    std::string type = ctx->bType()->getText();
+    currentVarType = getLLVMType(type);
+
+    for (auto varDef : ctx->varDef()) {
+        if (!inFunction) {
+            // 全局变量声明
+            std::string varName = varDef->Ident()->getText();
+            std::string llvmType = getLLVMType(type);
+            std::string value = "0"; // 默认值为 0
+
+            if (varDef->ASSIGN()) {
+                value = std::any_cast<std::string>(varDef->initVal()->accept(this));
+            }
+
+            irStream << "@" << varName << " = dso_local global " << llvmType << " " << value << ", align 4\n";
+            globalVars.push_back(varName); // 记录全局变量
+        } else {
+            // 局部变量声明
+            varDef->accept(this);
+        }
     }
     return nullptr;
 }
 
 std::any SysYIRGenerator::visitConstDecl(SysYParser::ConstDeclContext* ctx) {
+    // TODO：数组初始化
     std::string type = ctx->bType()->getText();
     for (auto constDef : ctx->constDef()) {
-        std::string varName = constDef->Ident()->getText();
-        std::string llvmType = getLLVMType(type);
-        std::string allocaName = getNextTemp();
-        
-        irStream << "  " << allocaName << " = alloca " << llvmType << ", align 4\n";
-        
-        if (constDef->constInitVal()) {
-            std::string value = std::any_cast<std::string>(constDef->constInitVal()->accept(this));
+        if (!inFunction) {
+            // 全局常量声明
+            std::string varName = constDef->Ident()->getText();
+            std::string llvmType = getLLVMType(type);
+            std::string value = "0"; // 默认值为 0
+
+            try {
+                value = std::any_cast<std::string>(constDef->constInitVal()->accept(this));
+            } catch (...) {
+                throw std::runtime_error("Const value must be initialized upon definition.");
+            }
+            
+            if (llvmType == "float") {
+                try {
+                    double floatValue = std::stod(value); 
+                    uint64_t hexValue = reinterpret_cast<uint64_t&>(floatValue);
+                    std::stringstream ss;
+                    ss << "0x" << std::hex << std::uppercase << hexValue;
+                    value = ss.str();
+                } catch (...) {
+                    throw std::runtime_error("Invalid float literal: " + value);
+                }
+            }
+
+            irStream << "@" << varName << " = dso_local constant " << llvmType << " " << value << ", align 4\n";
+            globalVars.push_back(varName); // 记录全局变量
+        } else {
+            // 局部常量声明
+            std::string varName = constDef->Ident()->getText();
+            std::string llvmType = getLLVMType(type);
+            std::string allocaName = getNextTemp();
+            std::string value = "0"; // 默认值为 0
+            
+            try {
+                value = std::any_cast<std::string>(constDef->constInitVal()->accept(this));
+            } catch (...) {
+                throw std::runtime_error("Const value must be initialized upon definition.");
+            }
+
+            irStream << "  " << allocaName << " = alloca " << llvmType << ", align 4\n";
             if (llvmType == "float") {
                 try {
                     double floatValue = std::stod(value); 
@@ -54,24 +119,17 @@ std::any SysYIRGenerator::visitConstDecl(SysYParser::ConstDeclContext* ctx) {
                 }
             }
             irStream << "  store " << llvmType << " " << value << ", " << llvmType 
-                     << "* " << allocaName << ", align 4\n";
-        }
-        symbolTable[varName] = {allocaName, llvmType};
-        tmpTable[allocaName] = llvmType;
-    }
-    return nullptr;
-}
-
-std::any SysYIRGenerator::visitVarDecl(SysYParser::VarDeclContext* ctx) {
-    std::string type = ctx->bType()->getText();
-    currentVarType = getLLVMType(type);
-    for (auto varDef : ctx->varDef()) {
-        varDef->accept(this);
+                    << "* " << allocaName << ", align 4\n";
+            
+            symbolTable[varName] = {allocaName, llvmType};
+            tmpTable[allocaName] = llvmType;
+            }
     }
     return nullptr;
 }
 
 std::any SysYIRGenerator::visitVarDef(SysYParser::VarDefContext* ctx) {
+    // TODO：数组初始化
     std::string varName = ctx->Ident()->getText();
     std::string type = currentVarType;
     std::string llvmType = getLLVMType(type);
@@ -331,7 +389,6 @@ std::any SysYIRGenerator::visitAddExp(SysYParser::AddExpContext* ctx) {
     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));
-        irStream << "right is " << right << "\n";
         std::string op = ctx->children[2*i-1]->getText();
         std::string temp = getNextTemp();
         std::string type = tmpTable[left];