[midend]解决部分变量重命名问题

This commit is contained in:
rain2133
2025-08-09 22:30:09 +08:00
parent 6b9ad0566d
commit c867bda9b4
3 changed files with 75 additions and 4 deletions

View File

@@ -1584,6 +1584,8 @@ class SymbolTable {
Value* getVariable(const std::string &name) const; ///< 根据名字name以及当前作用域获取变量 Value* getVariable(const std::string &name) const; ///< 根据名字name以及当前作用域获取变量
Value* addVariable(const std::string &name, Value *variable); ///< 添加变量 Value* addVariable(const std::string &name, Value *variable); ///< 添加变量
void registerParameterName(const std::string &name); ///< 注册函数参数名字避免alloca重名
void addVariableDirectly(const std::string &name, Value *variable); ///< 直接添加变量到当前作用域,不重命名
std::vector<std::unique_ptr<GlobalValue>>& getGlobals(); ///< 获取全局变量列表 std::vector<std::unique_ptr<GlobalValue>>& getGlobals(); ///< 获取全局变量列表
const std::vector<std::unique_ptr<ConstantVariable>>& getConsts() const; ///< 获取全局常量列表 const std::vector<std::unique_ptr<ConstantVariable>>& getConsts() const; ///< 获取全局常量列表
void enterNewScope(); ///< 进入新的作用域 void enterNewScope(); ///< 进入新的作用域
@@ -1646,6 +1648,12 @@ class Module {
void addVariable(const std::string &name, AllocaInst *variable) { void addVariable(const std::string &name, AllocaInst *variable) {
variableTable.addVariable(name, variable); variableTable.addVariable(name, variable);
} ///< 添加变量 } ///< 添加变量
void addVariableDirectly(const std::string &name, AllocaInst *variable) {
variableTable.addVariableDirectly(name, variable);
} ///< 直接添加变量到当前作用域,不重命名
void registerParameterName(const std::string &name) {
variableTable.registerParameterName(name);
} ///< 注册函数参数名字避免alloca重名
Value* getVariable(const std::string &name) { Value* getVariable(const std::string &name) {
return variableTable.getVariable(name); return variableTable.getVariable(name);
} ///< 根据名字name和当前作用域获取变量 } ///< 根据名字name和当前作用域获取变量

View File

@@ -1,6 +1,7 @@
#include "IR.h" #include "IR.h"
#include <algorithm> #include <algorithm>
#include <cassert> #include <cassert>
#include <functional>
#include <memory> #include <memory>
#include <queue> #include <queue>
#include <set> #include <set>
@@ -1033,12 +1034,23 @@ auto SymbolTable::addVariable(const std::string &name, Value *variable) -> Value
if (curNode != nullptr) { if (curNode != nullptr) {
std::stringstream ss; std::stringstream ss;
auto iter = variableIndex.find(name); auto iter = variableIndex.find(name);
// 处理超长变量名超过100字符
std::string displayName = name;
if (name.length() > 100) {
// 计算简单哈希
std::hash<std::string> hasher;
size_t hash = hasher(name);
// 截断到前100个字符 + 哈希后缀
displayName = name.substr(0, 100) + "_hash_" + std::to_string(hash);
}
if (iter != variableIndex.end()) { if (iter != variableIndex.end()) {
ss << name << iter->second ; ss << displayName << iter->second ;
iter->second += 1; iter->second += 1;
} else { } else {
variableIndex.emplace(name, 1); variableIndex.emplace(name, 1);
ss << name << 0 ; ss << displayName << 0 ;
} }
variable->setName(ss.str()); variable->setName(ss.str());
@@ -1056,6 +1068,47 @@ auto SymbolTable::addVariable(const std::string &name, Value *variable) -> Value
return result; return result;
} }
/**
* 注册函数参数名字到符号表确保后续的alloca变量不会使用相同的名字
*/
void SymbolTable::registerParameterName(const std::string &name) {
if (curNode != nullptr) {
// 为当前函数作用域创建一个唯一的参数名标识
std::string scopedName = name + "_param_" + std::to_string(reinterpret_cast<uintptr_t>(curNode));
auto iter = variableIndex.find(scopedName);
if (iter != variableIndex.end()) {
iter->second += 1;
} else {
// 注册带作用域的参数名确保在本作用域中后续的addVariable会避免冲突
variableIndex.emplace(scopedName, 1);
}
// 同时确保原名字也有一个索引这样addVariable会为alloca生成不同的名字
auto iter2 = variableIndex.find(name);
if (iter2 == variableIndex.end()) {
variableIndex.emplace(name, 1); // 设置为1这样addVariable会生成name1而不是name0
} else {
iter2->second += 1;
}
}
}
/**
* 直接添加变量到当前作用域,不进行重命名
*/
void SymbolTable::addVariableDirectly(const std::string &name, Value *variable) {
if (curNode != nullptr) {
curNode->varList.emplace(name, variable);
auto global = dynamic_cast<GlobalValue *>(variable);
auto constvar = dynamic_cast<ConstantVariable *>(variable);
if (global != nullptr) {
globals.emplace_back(global);
} else if (constvar != nullptr) {
globalconsts.emplace_back(constvar);
}
}
}
/** /**
* 获取全局变量 * 获取全局变量
*/ */

View File

@@ -1210,15 +1210,25 @@ std::any SysYIRGenerator::visitFuncDef(SysYParser::FuncDefContext *ctx){
for(int i = 0; i < paramActualTypes.size(); ++i) { for(int i = 0; i < paramActualTypes.size(); ++i) {
Argument* arg = new Argument(paramActualTypes[i], function, i, paramNames[i]); Argument* arg = new Argument(paramActualTypes[i], function, i, paramNames[i]);
function->insertArgument(arg); function->insertArgument(arg);
}
// 先将所有参数名字注册到符号表中确保alloca不会使用相同的名字
for (int i = 0; i < paramNames.size(); ++i) {
// 预先注册参数名字这样addVariable就会使用不同的后缀
module->registerParameterName(paramNames[i]);
} }
auto funcArgs = function->getArguments(); auto funcArgs = function->getArguments();
std::vector<AllocaInst *> allocas; std::vector<AllocaInst *> allocas;
for (int i = 0; i < paramActualTypes.size(); ++i) { for (int i = 0; i < paramActualTypes.size(); ++i) {
AllocaInst *alloca = builder.createAllocaInst(Type::getPointerType(paramActualTypes[i]), paramNames[i]); // 使用函数特定的前缀来确保参数alloca名字唯一
std::string allocaName = name + "_param_" + paramNames[i];
AllocaInst *alloca = builder.createAllocaInst(Type::getPointerType(paramActualTypes[i]), allocaName);
// 直接设置唯一名字不依赖addVariable的命名逻辑
alloca->setName(allocaName);
allocas.push_back(alloca); allocas.push_back(alloca);
module->addVariable(paramNames[i], alloca); // 直接添加到符号表,使用原参数名作为查找键
module->addVariableDirectly(paramNames[i], alloca);
} }
for(int i = 0; i < paramActualTypes.size(); ++i) { for(int i = 0; i < paramActualTypes.size(); ++i) {