using System;
|
using System.Collections.Generic;
|
using System.Linq;
|
using System.Management;
|
using System.Text;
|
using System.Threading.Tasks;
|
|
namespace LB_SmartVisionCommon
|
{
|
public class DiskInfoHelper
|
{
|
public static double GetFreeSpaceGB(string driveName)
|
{
|
try
|
{
|
DriveInfo drive = new DriveInfo(driveName);
|
return drive.IsReady ? Math.Round((double)drive.AvailableFreeSpace / (1024 * 1024 * 1024), 2) : -1;
|
}
|
catch
|
{
|
return -1;
|
}
|
}
|
|
public static Dictionary<string, object> GetDiskDetails(string driveName)
|
{
|
var result = new Dictionary<string, object>();
|
|
try
|
{
|
DriveInfo drive = new DriveInfo(driveName);
|
|
if (drive.IsReady)
|
{
|
result["DriveName"] = drive.Name;
|
result["TotalSizeGB"] = Math.Round((double)drive.TotalSize / (1024 * 1024 * 1024), 2);
|
result["FreeSpaceGB"] = Math.Round((double)drive.AvailableFreeSpace / (1024 * 1024 * 1024), 2);
|
result["DriveFormat"] = drive.DriveFormat;
|
result["DriveType"] = drive.DriveType.ToString();
|
}
|
}
|
catch (Exception ex)
|
{
|
result["Error"] = ex.Message;
|
}
|
|
return result;
|
}
|
|
/// <summary>
|
/// 获取GPU信息，包含系统索引和CUDA标准的NVIDIA索引
|
/// </summary>
|
/// <returns>GPU信息列表</returns>
|
public static List<GpuInfo> GetGpuInfoWithCudaIndex()
|
{
|
var gpuList = new List<GpuInfo>();
|
int systemIndex = 0; // 系统显示适配器索引（包含所有显卡）
|
int cudaIndex = 0; // CUDA专属索引（仅NVIDIA GPU，从0开始）
|
|
// WMI查询所有显示适配器
|
ManagementObjectSearcher searcher = new ManagementObjectSearcher(
|
"root\\CIMV2",
|
"SELECT Name, PNPDeviceID FROM Win32_VideoController");
|
|
foreach (ManagementObject queryObj in searcher.Get())
|
{
|
string gpuName = queryObj["Name"]?.ToString() ?? "未知设备";
|
string pnpId = queryObj["PNPDeviceID"]?.ToString() ?? string.Empty;
|
bool isNvidia = gpuName.Contains("NVIDIA", StringComparison.OrdinalIgnoreCase) ||
|
pnpId.Contains("VEN_10DE", StringComparison.OrdinalIgnoreCase);
|
|
var gpuInfo = new GpuInfo
|
{
|
GpuName = gpuName,
|
SystemIndex = systemIndex,
|
IsNvidia = isNvidia,
|
CudaIndex = -1 // 非NVIDIA GPU默认-1
|
};
|
|
// 仅NVIDIA GPU分配CUDA索引
|
if (isNvidia)
|
{
|
gpuInfo.CudaIndex = cudaIndex;
|
cudaIndex++;
|
}
|
gpuList.Add(gpuInfo);
|
systemIndex++;
|
}
|
|
return gpuList;
|
}
|
|
}
|
/// <summary>
|
/// GPU信息模型
|
/// </summary>
|
public class GpuInfo
|
{
|
public string GpuName { get; set; } // GPU名称
|
public int SystemIndex { get; set; } // 系统显示适配器索引（WMI枚举）
|
public int CudaIndex { get; set; } // CUDA设备索引（软件显示的索引）
|
public bool IsNvidia { get; set; } // 是否为NVIDIA GPU
|
}
|
}
|
