[midend-llvmirprint]检查并修复了初始值的打印逻辑

src/midend/IR.cpp

@@ -164,6 +164,21 @@ void Type::print(ostream &os) const {
   }
 }
 
+void Use::print(std::ostream& os) const {
+  os << "Use[" << index << "]: ";
+  if (value) {
+    printVarName(os, value);
+  } else {
+    os << "null";
+  }
+  os << " used by ";
+  if (user) {
+    os << "User@" << user;
+  } else {
+    os << "null";
+  }
+}
+
 PointerType* PointerType::get(Type *baseType) {
   static std::map<Type *, std::unique_ptr<PointerType>> pointerTypes;
   auto iter = pointerTypes.find(baseType);
@@ -217,25 +232,63 @@ void GlobalValue::print(std::ostream& os) const {
   // 输出全局变量的LLVM IR格式
   os << "@" << getName() << " = global ";
   
+  auto baseType = getType()->as<PointerType>()->getBaseType();
+  os << *baseType << " ";
+  
   // 输出初始化值
-  if (initValues.size() > 0) {
+  if (initValues.size() == 0) {
-    os << *getType()->as<PointerType>()->getBaseType() << " ";
+    // 没有初始化值，使用zeroinitializer
-    if (initValues.size() == 1) {
+    os << "zeroinitializer";
-      // 单个初始值
+  } else {
-      initValues.getValue(0)->print(os);
+    // 检查是否所有值都是零值
+    bool allZero = true;
+    const auto& values = initValues.getValues();
+    for (Value* val : values) {
+      if (auto constVal = dynamic_cast<ConstantValue*>(val)) {
+        if (!constVal->isZero()) {
+          allZero = false;
+          break;
+        }
+      } else {
+        allZero = false;
+        break;
+      }
+    }
+    
+    if (allZero) {
+      // 所有值都是零，使用zeroinitializer
+      os << "zeroinitializer";
+    } else if (initValues.size() == 1) {
+      // 单个初始值 - 如果是标量零值也考虑使用zeroinitializer
+      auto singleVal = initValues.getValue(0);
+      if (auto constVal = dynamic_cast<ConstantValue*>(singleVal)) {
+        if (constVal->isZero() && (baseType->isInt() || baseType->isFloat())) {
+          // 标量零值使用zeroinitializer
+          os << "zeroinitializer";
+        } else {
+          // 非零值或非基本类型，打印实际值
+          singleVal->print(os);
+        }
+      } else {
+        // 非常量值，打印实际值
+        singleVal->print(os);
+      }
     } else {
-      // 数组初始值 
+      // 数组初始值 - 需要展开ValueCounter中的压缩表示
       os << "[";
-      for (unsigned i = 0; i < initValues.size(); ++i) {
+      bool first = true;
-        if (i > 0) os << ", ";
+      const auto& numbers = initValues.getNumbers();
-        auto value = initValues.getValue(i);
+      
-        os << *value->getType() << " ";
+      for (size_t i = 0; i < values.size(); ++i) {
-        value->print(os);
+        for (unsigned j = 0; j < numbers[i]; ++j) {
+          if (!first) os << ", ";
+          os << *values[i]->getType() << " ";
+          values[i]->print(os);
+          first = false;
+        }
       }
       os << "]";
     }
-  } else {
-    os << *getType()->as<PointerType>()->getBaseType() << " zeroinitializer";
   }
 }
 
@@ -243,25 +296,91 @@ void ConstantVariable::print(std::ostream& os) const {
   // 输出常量的LLVM IR格式 
   os << "@" << getName() << " = constant ";
   
+  auto baseType = getType()->as<PointerType>()->getBaseType();
+  os << *baseType << " ";
+  
   // 输出初始化值
+  if (initValues.size() == 0) {
+    // 没有初始化值，使用zeroinitializer
+    os << "zeroinitializer";
+  } else {
+    // 检查是否所有值都是零值
+    bool allZero = true;
+    const auto& values = initValues.getValues();
+    for (Value* val : values) {
+      if (auto constVal = dynamic_cast<ConstantValue*>(val)) {
+        if (!constVal->isZero()) {
+          allZero = false;
+          break;
+        }
+      } else {
+        allZero = false;
+        break;
+      }
+    }
+    
+    if (allZero) {
+      // 所有值都是零，使用zeroinitializer
+      os << "zeroinitializer";
+    } else if (initValues.size() == 1) {
+      // 单个初始值 - 如果是标量零值也考虑使用zeroinitializer
+      auto singleVal = initValues.getValue(0);
+      if (auto constVal = dynamic_cast<ConstantValue*>(singleVal)) {
+        if (constVal->isZero() && (baseType->isInt() || baseType->isFloat())) {
+          // 标量零值使用zeroinitializer
+          os << "zeroinitializer";
+        } else {
+          // 非零值或非基本类型，打印实际值
+          singleVal->print(os);
+        }
+      } else {
+        // 非常量值，打印实际值
+        singleVal->print(os);
+      }
+    } else {
+      // 数组初始值 - 需要展开ValueCounter中的压缩表示
+      os << "[";
+      bool first = true;
+      const auto& numbers = initValues.getNumbers();
+      
+      for (size_t i = 0; i < values.size(); ++i) {
+        for (unsigned j = 0; j < numbers[i]; ++j) {
+          if (!first) os << ", ";
+          os << *values[i]->getType() << " ";
+          values[i]->print(os);
+          first = false;
+        }
+      }
+      os << "]";
+    }
+  }
+}
+
+void ConstantVariable::print_init(std::ostream& os) const {
+  // 只输出初始化值部分
   if (initValues.size() > 0) {
-    os << *getType()->as<PointerType>()->getBaseType() << " ";
     if (initValues.size() == 1) {
       // 单个初始值
       initValues.getValue(0)->print(os);
     } else {
-      // 数组初始值
+      // 数组初始值 - 需要展开ValueCounter中的压缩表示
       os << "[";
-      for (unsigned i = 0; i < initValues.size(); ++i) {
+      bool first = true;
-        if (i > 0) os << ", ";
+      const auto& values = initValues.getValues();
-        auto value = initValues.getValue(i);
+      const auto& numbers = initValues.getNumbers();
-        os << *value->getType() << " ";
+      
-        value->print(os);
+      for (size_t i = 0; i < values.size(); ++i) {
+        for (unsigned j = 0; j < numbers[i]; ++j) {
+          if (!first) os << ", ";
+          os << *values[i]->getType() << " ";
+          values[i]->print(os);
+          first = false;
+        }
       }
       os << "]";
     }
   } else {
-    os << *getType()->as<PointerType>()->getBaseType() << " zeroinitializer";
+    os << "zeroinitializer";
   }
 }
 
@@ -926,39 +1045,13 @@ void Function::print(std::ostream& os) const {
 void Module::print(std::ostream& os) const {
   // 打印全局变量声明
   for (auto& globalVar : const_cast<Module*>(this)->getGlobals()) {
-    printVarName(os, globalVar.get());
+    globalVar->print(os);
-    os << " = global " << *globalVar->getType()->as<PointerType>()->getBaseType();
-    
-    // 打印初始值
-    const auto& initValues = globalVar->getInitValues();
-    if (initValues.size() > 0) {
-      os << " ";
-      // 简化处理：只打印第一个初始值
-      initValues.getValue(0)->print(os);
-    } else {
-      // 默认初始化
-      if (globalVar->getType()->as<PointerType>()->getBaseType()->isInt()) {
-        os << " 0";
-      } else if (globalVar->getType()->as<PointerType>()->getBaseType()->isFloat()) {
-        os << " 0.0";
-      } else {
-        os << " zeroinitializer";
-      }
-    }
     os << "\n";
   }
   
   // 打印常量声明
   for (auto& constVar : getConsts()) {
-    printVarName(os, constVar.get());
+    constVar->print(os);
-    os << " = constant " << *constVar->getType()->as<PointerType>()->getBaseType();
-    os << " ";
-    const auto& initValues = constVar->getInitValues();
-    if (initValues.size() > 0) {
-      initValues.getValue(0)->print(os);
-    } else {
-      os << "0";
-    }
     os << "\n";
   }