Jokul
2570a028ea
第二、三阶段功能： 数据增强（第一阶段）: - GPU频率: NVIDIA nvidia-smi clocks.gr, Intel gt/gt0/rps_act_freq_mhz - 显存频率: NVIDIA nvidia-smi clocks.mem - 功耗: NVIDIA nvidia-smi power.draw, AMD hwmon power1_average - 风扇转速: NVIDIA nvidia-smi fan.speed, AMD hwmon fan1_input - 任务栏图标根据最高温度变色(≥80°C红/≥60°C橙) - readGpuStats 每次刷新前重置统计字段，避免残留旧值 交互提升（第二阶段）: - 温度告警: GPU温度≥80°C时通过D-Bus发送桌面通知，30秒冷却 - 右键菜单: 使用Qt.labs.platform原生菜单+MenuHelper，支持刷新和GPU切换 - 迷你趋势图: Canvas绘制近60秒温度走势折线 - 英文翻译: 新增en.ts，23条全部翻译 高级功能（第三阶段）: - NVIDIA进程列表: nvidia-smi --query-compute-apps查询，弹框中显示进程名和显存占用 - GPU切换: pkexec prime-select set <mode>，弹框按钮+右键菜单，仅PRIME模式显示 其他改进: - 自定义深色主题tooltip替代系统默认白色tooltip - 第一张卡片tooltip显示在下方避免被裁剪 - CMakeLists.txt lupdate添加C++源文件提取tr()翻译 - gpuStats格式扩展为9字段: temp|gpuUsage|memUsage|memUsed|memTotal|gpuClock|memClock|powerDraw|fanSpeed
637 lines
22 KiB
C++
637 lines
22 KiB
C++
// SPDX-FileCopyrightText: 2024 UnionTech Software Technology Co., Ltd.
|
||
//
|
||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||
|
||
#include "graphicsdriverapplet.h"
|
||
#include <pluginfactory.h>
|
||
|
||
#include <QProcess>
|
||
#include <QDir>
|
||
#include <QDebug>
|
||
#include <QFile>
|
||
#include <QTextStream>
|
||
#include <QRegularExpression>
|
||
#include <QFileInfo>
|
||
#include <QDateTime>
|
||
#include <QDBusInterface>
|
||
#include <QDBusMessage>
|
||
#include <QVariantList>
|
||
|
||
DCORE_USE_NAMESPACE
|
||
|
||
DS_BEGIN_NAMESPACE
|
||
|
||
GraphicsDriverApplet::GraphicsDriverApplet(QObject *parent)
|
||
: DApplet(parent)
|
||
, m_ready(false)
|
||
, m_lastAlertTime(0)
|
||
{
|
||
}
|
||
|
||
GraphicsDriverApplet::~GraphicsDriverApplet()
|
||
{
|
||
}
|
||
|
||
bool GraphicsDriverApplet::load()
|
||
{
|
||
return DApplet::load();
|
||
}
|
||
|
||
bool GraphicsDriverApplet::init()
|
||
{
|
||
detectGpus();
|
||
return DApplet::init();
|
||
}
|
||
|
||
QString GraphicsDriverApplet::deviceInfo() const
|
||
{
|
||
return m_deviceInfo;
|
||
}
|
||
|
||
QString GraphicsDriverApplet::currentDriver() const
|
||
{
|
||
return m_currentDriver;
|
||
}
|
||
|
||
QString GraphicsDriverApplet::gpuSummary() const
|
||
{
|
||
return m_gpuSummary;
|
||
}
|
||
|
||
bool GraphicsDriverApplet::ready() const
|
||
{
|
||
return m_ready;
|
||
}
|
||
|
||
QStringList GraphicsDriverApplet::gpuStats() const
|
||
{
|
||
return m_gpuStats;
|
||
}
|
||
|
||
QString GraphicsDriverApplet::gpuMode() const
|
||
{
|
||
return m_gpuMode;
|
||
}
|
||
|
||
QStringList GraphicsDriverApplet::gpuPrimary() const
|
||
{
|
||
return m_gpuPrimary;
|
||
}
|
||
|
||
QVariantList GraphicsDriverApplet::tempHistory() const
|
||
{
|
||
QVariantList result;
|
||
for (const QList<int> &history : m_tempHistory) {
|
||
QVariantList gpuHistory;
|
||
for (int temp : history) {
|
||
gpuHistory << temp;
|
||
}
|
||
result << QVariant(gpuHistory);
|
||
}
|
||
return result;
|
||
}
|
||
|
||
QStringList GraphicsDriverApplet::gpuProcesses() const
|
||
{
|
||
return m_gpuProcesses;
|
||
}
|
||
|
||
void GraphicsDriverApplet::refreshDeviceInfo()
|
||
{
|
||
detectGpus();
|
||
}
|
||
|
||
void GraphicsDriverApplet::refreshStats()
|
||
{
|
||
readGpuStats();
|
||
}
|
||
|
||
void GraphicsDriverApplet::detectGpus()
|
||
{
|
||
m_gpus.clear();
|
||
m_tempHistory.clear();
|
||
|
||
// 通过 lspci 检测显卡
|
||
QStringList lines = runCommand("lspci", {"-mm"});
|
||
for (const QString &line : lines) {
|
||
if (line.contains("VGA compatible controller", Qt::CaseInsensitive) ||
|
||
line.contains("3D controller", Qt::CaseInsensitive) ||
|
||
line.contains("Display controller", Qt::CaseInsensitive)) {
|
||
|
||
GpuInfo gpu;
|
||
// 初始化统计字段
|
||
gpu.drmCard = -1;
|
||
gpu.isPrimary = false;
|
||
gpu.temperature = -1;
|
||
gpu.gpuUsage = -1;
|
||
gpu.memUsage = -1;
|
||
gpu.memTotal = -1;
|
||
gpu.memUsed = -1;
|
||
gpu.gpuClock = -1;
|
||
gpu.memClock = -1;
|
||
gpu.powerDraw = -1;
|
||
gpu.fanSpeed = -1;
|
||
|
||
// lspci -mm 格式: "Class" "Vendor" "Device" ...
|
||
// 提取 PCI ID（行首的 01:00.0 等）
|
||
int spaceIdx = line.indexOf(' ');
|
||
if (spaceIdx > 0) {
|
||
gpu.pciId = line.left(spaceIdx);
|
||
}
|
||
|
||
gpu.name = parseGpuName(line);
|
||
|
||
// 检测驱动：通过 symlink 目标获取真实驱动名
|
||
QString sysPath = "/sys/bus/pci/devices/0000:" + gpu.pciId + "/driver";
|
||
QFileInfo driverLink(sysPath);
|
||
if (driverLink.exists() && driverLink.isSymLink()) {
|
||
QString target = driverLink.symLinkTarget();
|
||
// symLinkTarget 返回如 ".../drivers/nvidia"，提取最后一段
|
||
gpu.driver = target.section('/', -1);
|
||
|
||
// 读取驱动版本
|
||
gpu.driverVersion = readDriverVersion(gpu.driver);
|
||
} else {
|
||
gpu.driver = "unknown";
|
||
}
|
||
|
||
// 查找 DRM 卡号
|
||
gpu.drmCard = findDrmCard(gpu.pciId);
|
||
|
||
// 读取 boot_vga 判断主副显
|
||
QFile bootVgaFile("/sys/bus/pci/devices/0000:" + gpu.pciId + "/boot_vga");
|
||
if (bootVgaFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
gpu.isPrimary = QTextStream(&bootVgaFile).readAll().trimmed() == "1";
|
||
}
|
||
|
||
m_gpus.append(gpu);
|
||
}
|
||
}
|
||
|
||
// 生成摘要信息
|
||
QStringList parts;
|
||
QStringList driverParts;
|
||
for (const GpuInfo &gpu : m_gpus) {
|
||
QString info = gpu.name;
|
||
if (!gpu.driver.isEmpty() && gpu.driver != "unknown") {
|
||
info += " (" + gpu.driver;
|
||
if (!gpu.driverVersion.isEmpty()) {
|
||
info += " " + gpu.driverVersion;
|
||
}
|
||
info += ")";
|
||
}
|
||
parts << info;
|
||
if (!gpu.driver.isEmpty() && !driverParts.contains(gpu.driver)) {
|
||
driverParts << gpu.driver;
|
||
}
|
||
}
|
||
|
||
m_deviceInfo = parts.join("\n");
|
||
m_currentDriver = driverParts.join(", ");
|
||
m_gpuSummary = parts.join(" | ");
|
||
m_ready = !m_gpus.isEmpty();
|
||
|
||
// 生成主副显标识
|
||
m_gpuPrimary.clear();
|
||
for (const GpuInfo &gpu : m_gpus) {
|
||
m_gpuPrimary << (gpu.isPrimary ? "true" : "false");
|
||
}
|
||
|
||
qDebug() << "Detected GPUs:" << m_deviceInfo;
|
||
qDebug() << "Current drivers:" << m_currentDriver;
|
||
|
||
emit deviceInfoChanged();
|
||
emit currentDriverChanged();
|
||
emit gpuSummaryChanged();
|
||
emit gpuStatsChanged();
|
||
emit gpuPrimaryChanged();
|
||
emit readyChanged();
|
||
|
||
// 读取实时统计信息
|
||
readGpuStats();
|
||
|
||
// 检测 GPU 切换模式
|
||
detectGpuMode();
|
||
}
|
||
|
||
QStringList GraphicsDriverApplet::runCommand(const QString &cmd, const QStringList &args)
|
||
{
|
||
QProcess process;
|
||
process.start(cmd, args);
|
||
process.waitForFinished(5000);
|
||
QString output = QString::fromUtf8(process.readAllStandardOutput());
|
||
return output.split('\n', Qt::SkipEmptyParts);
|
||
}
|
||
|
||
QString GraphicsDriverApplet::parseGpuName(const QString &line)
|
||
{
|
||
// lspci -mm 格式示例:
|
||
// 01:00.0 "VGA compatible controller" "NVIDIA Corporation" "GA106M [GeForce RTX 3060 Mobile / Max-Q]" "ASUSTeK Computer Inc." "Device 10a2"
|
||
// 提取引号中的内容
|
||
QStringList parts;
|
||
int pos = 0;
|
||
while (pos < line.length()) {
|
||
int start = line.indexOf('"', pos);
|
||
if (start < 0) break;
|
||
int end = line.indexOf('"', start + 1);
|
||
if (end < 0) break;
|
||
parts << line.mid(start + 1, end - start - 1);
|
||
pos = end + 1;
|
||
}
|
||
|
||
if (parts.size() >= 3) {
|
||
// parts[0] = class, parts[1] = vendor, parts[2] = device
|
||
QString vendor = parts[1];
|
||
QString device = parts[2];
|
||
// 去掉方括号内的别名，保留主要名称
|
||
int bracket = device.indexOf('[');
|
||
if (bracket > 0) {
|
||
// 保留完整名称，方括号内是子系统信息
|
||
}
|
||
return vendor + " " + device;
|
||
}
|
||
|
||
return line;
|
||
}
|
||
|
||
QString GraphicsDriverApplet::readDriverVersion(const QString &driverName)
|
||
{
|
||
// 尝试从 /sys/module/<driver>/version 读取版本
|
||
QString path = "/sys/module/" + driverName + "/version";
|
||
QFile file(path);
|
||
if (file.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
QString version = QTextStream(&file).readAll().trimmed();
|
||
if (!version.isEmpty()) {
|
||
return version;
|
||
}
|
||
}
|
||
|
||
// nvidia 特殊处理：从 /proc/driver/nvidia/version 读取
|
||
if (driverName == "nvidia") {
|
||
QFile nvFile("/proc/driver/nvidia/version");
|
||
if (nvFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
QString content = QTextStream(&nvFile).readAll();
|
||
// 格式: NVRM version: NVIDIA UNIX x86_64 Kernel Module 535.183.01 ...
|
||
QRegularExpression re("Kernel Module\\s+(\\d+\\.\\d+\\.\\d+)");
|
||
auto match = re.match(content);
|
||
if (match.hasMatch()) {
|
||
return match.captured(1);
|
||
}
|
||
}
|
||
}
|
||
|
||
// 开源内核驱动（如 amdgpu, radeon, nouveau, i915）无独立版本号，版本跟随内核
|
||
static const QStringList kernelDrivers = {
|
||
"amdgpu", "radeon", "nouveau", "i915", "i965", "mgag200", "ast", "vmwgfx"
|
||
};
|
||
if (kernelDrivers.contains(driverName)) {
|
||
QFile osrelease("/proc/sys/kernel/osrelease");
|
||
if (osrelease.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
return "Linux " + QTextStream(&osrelease).readAll().trimmed();
|
||
}
|
||
}
|
||
|
||
return QString();
|
||
}
|
||
|
||
void GraphicsDriverApplet::detectGpuMode()
|
||
{
|
||
// 单 GPU
|
||
if (m_gpus.size() <= 1) {
|
||
m_gpuMode = "Solo";
|
||
emit gpuModeChanged();
|
||
return;
|
||
}
|
||
|
||
// 多 GPU：检测 PRIME / Bumblebee
|
||
// PRIME: nvidia-drm-outputclass.conf 存在
|
||
if (QFile::exists("/usr/share/X11/xorg.conf.d/nvidia-drm-outputclass.conf") ||
|
||
QFile::exists("/etc/X11/xorg.conf.d/70-nvidia.conf")) {
|
||
m_gpuMode = "PRIME";
|
||
emit gpuModeChanged();
|
||
return;
|
||
}
|
||
|
||
// Bumblebee: 检查 bumblebeed 服务
|
||
QStringList lsmod = runCommand("lsmod", {});
|
||
bool bumblebeeLoaded = false;
|
||
for (const QString &line : lsmod) {
|
||
if (line.contains("bumblebee", Qt::CaseInsensitive)) {
|
||
bumblebeeLoaded = true;
|
||
break;
|
||
}
|
||
}
|
||
if (bumblebeeLoaded || QFile::exists("/usr/sbin/bumblebeed")) {
|
||
m_gpuMode = "Bumblebee";
|
||
emit gpuModeChanged();
|
||
return;
|
||
}
|
||
|
||
// 多 GPU 但无已知切换方案
|
||
m_gpuMode = "Hybrid";
|
||
emit gpuModeChanged();
|
||
}
|
||
|
||
int GraphicsDriverApplet::findDrmCard(const QString &pciId)
|
||
{
|
||
// 扫描 /sys/bus/pci/devices/0000:XX:XX.X/drm/ 目录，找到 cardN 条目
|
||
QString drmPath = "/sys/bus/pci/devices/0000:" + pciId + "/drm";
|
||
QDir dir(drmPath);
|
||
if (!dir.exists()) return -1;
|
||
|
||
QStringList entries = dir.entryList(QStringList() << "card*", QDir::Dirs | QDir::NoDotAndDotDot);
|
||
for (const QString &entry : entries) {
|
||
// 匹配 "card0" "card1" 等，排除 "card0-DP-1" 等连接器
|
||
QRegularExpression re("^card(\\d+)$");
|
||
auto match = re.match(entry);
|
||
if (match.hasMatch()) {
|
||
return match.captured(1).toInt();
|
||
}
|
||
}
|
||
return -1;
|
||
}
|
||
|
||
void GraphicsDriverApplet::readGpuStats()
|
||
{
|
||
// 先处理 NVIDIA GPU（单次 nvidia-smi 调用获取所有 NVIDIA GPU 数据）
|
||
// nvidia-smi 输出格式: pci.bus_id,temperature.gpu,utilization.gpu,utilization.memory,
|
||
// memory.total,memory.used,clocks.gr,clocks.mem,power.draw,fan.speed
|
||
QMap<QString, QStringList> nvidiaData; // pciId -> stats
|
||
QStringList nvLines = runCommand("nvidia-smi", {
|
||
"--query-gpu=pci.bus_id,temperature.gpu,utilization.gpu,utilization.memory,memory.total,memory.used,clocks.gr,clocks.mem,power.draw,fan.speed",
|
||
"--format=csv,noheader,nounits"
|
||
});
|
||
for (const QString &line : nvLines) {
|
||
QStringList parts = line.split(",");
|
||
if (parts.size() >= 6) {
|
||
// nvidia-smi pci.bus_id 格式: 00000000:01:00.0，提取后半部分匹配
|
||
QString busId = parts[0].trimmed();
|
||
QString shortId = busId.section(':', -2); // "01:00.0"
|
||
nvidiaData[shortId] = parts;
|
||
}
|
||
}
|
||
|
||
// 逐个 GPU 读取统计
|
||
for (GpuInfo &gpu : m_gpus) {
|
||
// 每次刷新前重置实时统计字段，确保数据源不可用时不会残留旧值
|
||
gpu.temperature = -1;
|
||
gpu.gpuUsage = -1;
|
||
gpu.memUsage = -1;
|
||
gpu.memTotal = -1;
|
||
gpu.memUsed = -1;
|
||
gpu.gpuClock = -1;
|
||
gpu.memClock = -1;
|
||
gpu.powerDraw = -1;
|
||
gpu.fanSpeed = -1;
|
||
|
||
if (gpu.driver == "nvidia" && nvidiaData.contains(gpu.pciId)) {
|
||
// NVIDIA: 使用 nvidia-smi 数据
|
||
QStringList &parts = nvidiaData[gpu.pciId];
|
||
gpu.temperature = parts[1].trimmed().toInt();
|
||
gpu.gpuUsage = parts[2].trimmed().toInt();
|
||
gpu.memUsage = parts[3].trimmed().toInt();
|
||
gpu.memTotal = parts[4].trimmed().toInt();
|
||
gpu.memUsed = parts[5].trimmed().toInt();
|
||
// 频率 (parts[6]=GPU, parts[7]=显存)
|
||
if (parts.size() > 6) gpu.gpuClock = parts[6].trimmed().toInt();
|
||
if (parts.size() > 7) gpu.memClock = parts[7].trimmed().toInt();
|
||
// 功耗 (parts[8])，[N/A] 时保持 -1
|
||
if (parts.size() > 8) {
|
||
QString pwr = parts[8].trimmed();
|
||
if (pwr != "[N/A]" && !pwr.isEmpty()) gpu.powerDraw = pwr.toDouble();
|
||
}
|
||
// 风扇转速 (parts[9])，[N/A] 时保持 -1
|
||
if (parts.size() > 9) {
|
||
QString fan = parts[9].trimmed();
|
||
if (fan != "[N/A]" && !fan.isEmpty()) gpu.fanSpeed = fan.toInt();
|
||
}
|
||
} else if (gpu.drmCard >= 0) {
|
||
// AMD / Intel / 其他: 使用 sysfs
|
||
QString cardPath = QString("/sys/class/drm/card%1/device").arg(gpu.drmCard);
|
||
|
||
// GPU 使用率
|
||
QFile busyFile(cardPath + "/gpu_busy_percent");
|
||
if (busyFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
gpu.gpuUsage = QTextStream(&busyFile).readAll().trimmed().toInt();
|
||
}
|
||
|
||
// 显存 (字节，转换为 MB)
|
||
QFile vramTotal(cardPath + "/mem_info_vram_total");
|
||
QFile vramUsed(cardPath + "/mem_info_vram_used");
|
||
if (vramTotal.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
gpu.memTotal = QTextStream(&vramTotal).readAll().trimmed().toLongLong() / 1024 / 1024;
|
||
}
|
||
if (vramUsed.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
gpu.memUsed = QTextStream(&vramUsed).readAll().trimmed().toLongLong() / 1024 / 1024;
|
||
}
|
||
if (gpu.memTotal > 0 && gpu.memUsed >= 0) {
|
||
gpu.memUsage = gpu.memUsed * 100 / gpu.memTotal;
|
||
}
|
||
|
||
// 温度: 查找 hwmon 子目录
|
||
QDir hwmonDir(cardPath + "/hwmon");
|
||
if (hwmonDir.exists()) {
|
||
QStringList hwmons = hwmonDir.entryList(QStringList() << "hwmon*", QDir::Dirs | QDir::NoDotAndDotDot);
|
||
for (const QString &hwmon : hwmons) {
|
||
QFile nameFile(hwmonDir.absoluteFilePath(hwmon + "/name"));
|
||
if (nameFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
QString name = QTextStream(&nameFile).readAll().trimmed();
|
||
if (name == gpu.driver || name == "amdgpu" || name == "radeon") {
|
||
QString hwmonPath = hwmonDir.absoluteFilePath(hwmon);
|
||
// 温度
|
||
QFile tempFile(hwmonPath + "/temp1_input");
|
||
if (tempFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
int tempMilli = QTextStream(&tempFile).readAll().trimmed().toInt();
|
||
gpu.temperature = tempMilli / 1000;
|
||
}
|
||
// 功耗 (µW -> W)
|
||
QFile powerFile(hwmonPath + "/power1_average");
|
||
if (powerFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
qint64 microWatts = QTextStream(&powerFile).readAll().trimmed().toLongLong();
|
||
gpu.powerDraw = microWatts / 1000000.0;
|
||
}
|
||
// 风扇转速 (RPM)
|
||
QFile fanFile(hwmonPath + "/fan1_input");
|
||
if (fanFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
gpu.fanSpeed = QTextStream(&fanFile).readAll().trimmed().toInt();
|
||
}
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// GPU 频率: Intel i915 使用 gt/gt0/rps_act_freq_mhz，AMD 使用 hwmon freq1_input
|
||
QString gtPath = QString("/sys/class/drm/card%1/gt/gt0/rps_act_freq_mhz").arg(gpu.drmCard);
|
||
QFile gtFreqFile(gtPath);
|
||
if (gtFreqFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
gpu.gpuClock = QTextStream(>FreqFile).readAll().trimmed().toInt();
|
||
}
|
||
}
|
||
|
||
// --- 以下为 Intel 核显等无 hwmon/vram sysfs 接口时的回退逻辑 ---
|
||
|
||
// 温度回退：扫描 thermal_zone，匹配 Intel GPU 的 ACPI 标识 (B0D4)
|
||
// B0D4 是 Intel GPU 在 ACPI 规范中的标准热区标识
|
||
if (gpu.temperature < 0) {
|
||
QDir thermalDir("/sys/class/thermal");
|
||
QStringList zones = thermalDir.entryList(QStringList() << "thermal_zone*", QDir::Dirs | QDir::NoDotAndDotDot);
|
||
for (const QString &zone : zones) {
|
||
QFile typeFile(thermalDir.absoluteFilePath(zone + "/type"));
|
||
if (typeFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
QString type = QTextStream(&typeFile).readAll().trimmed();
|
||
if (type == "B0D4") {
|
||
QFile tempFile(thermalDir.absoluteFilePath(zone + "/temp"));
|
||
if (tempFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
int tempMilli = QTextStream(&tempFile).readAll().trimmed().toInt();
|
||
gpu.temperature = tempMilli / 1000;
|
||
}
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 显存回退：Intel 核显使用共享系统内存，读取 /proc/meminfo 作为参考
|
||
// 注意：这是系统内存使用量，非核显独占，仅作参考
|
||
if (gpu.memTotal < 0) {
|
||
QFile meminfo("/proc/meminfo");
|
||
if (meminfo.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
||
QString content = QTextStream(&meminfo).readAll();
|
||
QRegularExpression totalRe("MemTotal:\\s*(\\d+)\\s*kB");
|
||
QRegularExpression availRe("MemAvailable:\\s*(\\d+)\\s*kB");
|
||
auto totalMatch = totalRe.match(content);
|
||
auto availMatch = availRe.match(content);
|
||
if (totalMatch.hasMatch()) {
|
||
gpu.memTotal = totalMatch.captured(1).toLongLong() / 1024; // kB -> MB
|
||
}
|
||
if (availMatch.hasMatch() && gpu.memTotal > 0) {
|
||
qint64 availMB = availMatch.captured(1).toLongLong() / 1024;
|
||
gpu.memUsed = gpu.memTotal - availMB;
|
||
if (gpu.memUsed >= 0) {
|
||
gpu.memUsage = gpu.memUsed * 100 / gpu.memTotal;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 生成统计数据 (格式: temp|gpuUsage|memUsage|memUsed|memTotal|gpuClock|memClock|powerDraw|fanSpeed)
|
||
m_gpuStats.clear();
|
||
for (const GpuInfo &gpu : m_gpus) {
|
||
m_gpuStats << QString("%1|%2|%3|%4|%5|%6|%7|%8|%9")
|
||
.arg(gpu.temperature)
|
||
.arg(gpu.gpuUsage)
|
||
.arg(gpu.memUsage)
|
||
.arg(gpu.memUsed)
|
||
.arg(gpu.memTotal)
|
||
.arg(gpu.gpuClock)
|
||
.arg(gpu.memClock)
|
||
.arg(gpu.powerDraw, 0, 'f', 1)
|
||
.arg(gpu.fanSpeed);
|
||
}
|
||
|
||
emit gpuStatsChanged();
|
||
|
||
// 更新温度历史（保留最近 60 个采样点）
|
||
if (m_tempHistory.size() != m_gpus.size()) {
|
||
m_tempHistory.resize(m_gpus.size());
|
||
}
|
||
for (int i = 0; i < m_gpus.size(); ++i) {
|
||
m_tempHistory[i].append(m_gpus[i].temperature);
|
||
if (m_tempHistory[i].size() > 60) {
|
||
m_tempHistory[i].removeFirst();
|
||
}
|
||
}
|
||
emit tempHistoryChanged();
|
||
|
||
// 温度告警检测
|
||
checkTempAlert();
|
||
|
||
// 读取 NVIDIA 进程列表
|
||
readGpuProcesses();
|
||
}
|
||
|
||
// 温度告警：检测是否有 GPU 温度超过 80°C，30 秒内不重复告警
|
||
void GraphicsDriverApplet::checkTempAlert()
|
||
{
|
||
qint64 now = QDateTime::currentSecsSinceEpoch();
|
||
if (now - m_lastAlertTime < 30) return;
|
||
|
||
for (const GpuInfo &gpu : m_gpus) {
|
||
if (gpu.temperature >= 80) {
|
||
sendNotification(
|
||
tr("GPU Temperature Alert"),
|
||
tr("%1 temperature reached %2°C, please check cooling")
|
||
.arg(gpu.name)
|
||
.arg(gpu.temperature)
|
||
);
|
||
m_lastAlertTime = now;
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 通过 D-Bus 发送桌面通知
|
||
void GraphicsDriverApplet::sendNotification(const QString &summary, const QString &body)
|
||
{
|
||
QDBusInterface notify("org.freedesktop.Notifications",
|
||
"/org/freedesktop/Notifications",
|
||
"org.freedesktop.Notifications",
|
||
QDBusConnection::sessionBus());
|
||
if (notify.isValid()) {
|
||
notify.call("Notify", "dde-shell-graphics-driver", 0u,
|
||
"dialog-warning", summary, body,
|
||
QStringList(), QVariantMap(), 5000);
|
||
}
|
||
}
|
||
|
||
// 读取 NVIDIA GPU 上运行的进程列表
|
||
void GraphicsDriverApplet::readGpuProcesses()
|
||
{
|
||
m_gpuProcesses.clear();
|
||
|
||
// 仅当存在 NVIDIA GPU 时查询
|
||
bool hasNvidia = false;
|
||
for (const GpuInfo &gpu : m_gpus) {
|
||
if (gpu.driver == "nvidia") { hasNvidia = true; break; }
|
||
}
|
||
if (!hasNvidia) {
|
||
emit gpuProcessesChanged();
|
||
return;
|
||
}
|
||
|
||
// nvidia-smi --query-compute-apps=pid,process_name,used_memory --format=csv,noheader,nounits
|
||
QStringList lines = runCommand("nvidia-smi", {
|
||
"--query-compute-apps=pid,process_name,used_memory",
|
||
"--format=csv,noheader,nounits"
|
||
});
|
||
for (const QString &line : lines) {
|
||
QStringList parts = line.split(",");
|
||
if (parts.size() >= 3) {
|
||
m_gpuProcesses << QString("%1|%2|%3")
|
||
.arg(parts[0].trimmed())
|
||
.arg(parts[1].trimmed())
|
||
.arg(parts[2].trimmed());
|
||
}
|
||
}
|
||
|
||
emit gpuProcessesChanged();
|
||
}
|
||
|
||
// GPU 切换：调用 prime-select 切换 PRIME 模式（需要 polkit 提权）
|
||
bool GraphicsDriverApplet::switchGpu(const QString &mode)
|
||
{
|
||
// mode: "nvidia" / "intel" / "on-demand"
|
||
QProcess process;
|
||
process.start("pkexec", {"prime-select", "set", mode});
|
||
process.waitForFinished(30000);
|
||
return process.exitCode() == 0;
|
||
}
|
||
|
||
D_APPLET_CLASS(GraphicsDriverApplet)
|
||
|
||
DS_END_NAMESPACE
|
||
|
||
#include "graphicsdriverapplet.moc"
|