feat: normalize disk size

only for specific devices.

Log: as above.

Bug: https://pms.uniontech.com/bug-view-329231.html
Task: https://pms.uniontech.com/task-view-368603.html

Author: itsXuSt

service/diskoperation/DeviceStorage.cpp

@@ -608,7 +608,7 @@ void DeviceStorage::getDiskInfoInterface(const QString &devicePath, QString &int
     return;
 }
 
-void DeviceStorage::updateForHWDevice(const QString &devicePath)
+void DeviceStorage::updateForHWDevice(const QString &/*devicePath*/)
 {
     if (m_model != Utils::readContent("/proc/bootdevice/product_name").trimmed())
         return;
@@ -622,6 +622,58 @@ void DeviceStorage::updateForHWDevice(const QString &devicePath)
     // hide mode and vendor
     m_model = "";
     m_vendor = "";
+
+    // Normalize disk size
+    if (!m_size.isEmpty()) {
+        // Convert size string to bytes
+        qint64 bytes = 0;
+        QString sizeStr = m_size.toLower();
+
+        if (sizeStr.contains("bytes")) {
+            sizeStr = sizeStr.split("bytes").first().trimmed();
+            bytes = sizeStr.toLongLong();
+        } else {
+            // Handle GB/TB directly
+            double value = sizeStr.split(" ").first().toDouble();
+            if (sizeStr.contains("gb")) {
+                bytes = value * 1000 * 1000 * 1000;
+            } else if (sizeStr.contains("tb")) {
+                bytes = value * 1000 * 1000 * 1000 * 1000;
+            }
+        }
+
+        // Convert to standardized format
+        if (bytes > 0) {
+            // Convert to GB and round to standard sizes
+            int gb = qRound(bytes / (1000.0 * 1000 * 1000));
+            if (gb <= 300) {
+                m_size = "256 GB";
+            } else if (gb <= 600) {
+                m_size = "512 GB";
+            } else if (gb <= 1200){
+                m_size = "1 TB";
+            } else if (gb <= 2200) {
+                m_size = "2 TB";
+            }
+        }
+    }
+}
+
+void DeviceStorage::formatDeviceSize() 
+{
+    static QRegularExpression numRegex("(\\d+)(?:\\.\\d+)?\\s*(TB|GB|MB)");
+    QRegularExpressionMatch match = numRegex.match(m_size);
+    if (match.hasMatch()) {
+        double number = match.captured(1).toDouble();
+        QString unit = match.captured(2);
+        // 如果小数部分为0,转换为整数显示,否则保留小数
+        if(qFuzzyCompare(number, qRound(number))) {
+            m_size = QString("%1 %2").arg(qRound(number)).arg(unit); 
+        } else {
+            m_size = QString("%1 %2").arg(number, 0, 'f', 2).arg(unit);
+        }
+        return;
+    }
 }
 
 void DeviceStorage::getMapInfoFromSmartctl(QMap<QString, QString> &mapInfo, const QString &info, const QString &ch)

service/diskoperation/DeviceStorage.h

@@ -36,6 +36,7 @@ class DeviceStorage
 
     /**@brief:为HW设备更新信息*/
     void updateForHWDevice(const QString &devicePath);
+    void formatDeviceSize();
 
 
 private:

service/diskoperation/partedcore.cpp

@@ -146,6 +146,7 @@ HardDiskInfo PartedCore::getDeviceHardInfo(const QString &devicepath)
     device.getDiskInfoInterface(devicepath, device.m_interface, device.m_model);
 
     device.updateForHWDevice(devicepath);
+    device.formatDeviceSize();
 
     hdinfo.m_model = device.m_model;
     hdinfo.m_vendor = device.m_vendor;