vasilito
cf12defd28
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
120 lines
3.1 KiB
C++
120 lines
3.1 KiB
C++
/*
|
|
This file is part of the KDE project
|
|
SPDX-FileCopyrightText: 2023 Dave Vasilevsky <dave@vasilevsky.ca>
|
|
|
|
SPDX-License-Identifier: LGPL-2.0-only OR LGPL-3.0-only
|
|
*/
|
|
|
|
#include "gpudetection_p.h"
|
|
|
|
#ifdef WITH_QTDBUS
|
|
#include <QDBusArgument>
|
|
#include <QDBusConnection>
|
|
#include <QDBusInterface>
|
|
#include <QDBusReply>
|
|
#include <QDBusVariant>
|
|
#endif
|
|
#include <QMap>
|
|
#include <QString>
|
|
|
|
#include <KProcess>
|
|
|
|
static void checkGpu();
|
|
// Returns true if switcheroo present
|
|
static bool checkGpuWithSwitcheroo();
|
|
static void checkGpuWithSolid();
|
|
|
|
// TODO: GPUs are hot-swappable, watch for changes using dbus PropertiesChanged
|
|
enum class GpuCheck {
|
|
NotChecked,
|
|
Present,
|
|
Absent,
|
|
};
|
|
static GpuCheck s_gpuCheck = GpuCheck::NotChecked;
|
|
static QProcessEnvironment s_gpuEnv;
|
|
|
|
static void checkGpu()
|
|
{
|
|
if (s_gpuCheck == GpuCheck::NotChecked) {
|
|
if (!checkGpuWithSwitcheroo()) {
|
|
checkGpuWithSolid();
|
|
}
|
|
}
|
|
}
|
|
|
|
static bool checkGpuWithSwitcheroo()
|
|
{
|
|
#ifdef WITH_QTDBUS
|
|
QDBusInterface switcheroo(QStringLiteral("net.hadess.SwitcherooControl"),
|
|
QStringLiteral("/net/hadess/SwitcherooControl"),
|
|
QStringLiteral("org.freedesktop.DBus.Properties"),
|
|
QDBusConnection::systemBus());
|
|
if (!switcheroo.isValid()) {
|
|
return false;
|
|
}
|
|
|
|
QDBusReply<QDBusVariant> reply = switcheroo.call(QStringLiteral("Get"), QStringLiteral("net.hadess.SwitcherooControl"), QStringLiteral("GPUs"));
|
|
if (!reply.isValid()) {
|
|
return false;
|
|
}
|
|
|
|
QDBusArgument arg = qvariant_cast<QDBusArgument>(reply.value().variant());
|
|
QList<QVariantMap> gpus;
|
|
arg >> gpus;
|
|
|
|
for (const auto &gpu : gpus) {
|
|
bool defaultGpu = qvariant_cast<bool>(gpu[QStringLiteral("Default")]);
|
|
if (!defaultGpu) {
|
|
s_gpuCheck = GpuCheck::Present;
|
|
QStringList envList = qvariant_cast<QStringList>(gpu[QStringLiteral("Environment")]);
|
|
for (int i = 0; i + 1 < envList.size(); i += 2) {
|
|
s_gpuEnv.insert(envList[i], envList[i + 1]);
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
// No non-default GPU found
|
|
s_gpuCheck = GpuCheck::Absent;
|
|
return true;
|
|
}
|
|
|
|
static void checkGpuWithSolid()
|
|
{
|
|
#ifdef WITH_QTDBUS
|
|
// TODO: Consider moving this check into kio, instead of using Solid
|
|
QDBusInterface iface(QStringLiteral("org.kde.Solid.PowerManagement"),
|
|
QStringLiteral("/org/kde/Solid/PowerManagement"),
|
|
QStringLiteral("org.kde.Solid.PowerManagement"),
|
|
QDBusConnection::sessionBus());
|
|
if (iface.isValid()) {
|
|
QDBusReply<bool> reply = iface.call(QStringLiteral("hasDualGpu"));
|
|
if (reply.isValid() && reply.value()) {
|
|
s_gpuCheck = GpuCheck::Present;
|
|
s_gpuEnv.insert(QStringLiteral("DRI_PRIME"), QStringLiteral("1"));
|
|
return;
|
|
}
|
|
}
|
|
|
|
s_gpuCheck = GpuCheck::Absent;
|
|
#endif
|
|
}
|
|
|
|
namespace KIO
|
|
{
|
|
|
|
bool hasDiscreteGpu()
|
|
{
|
|
checkGpu();
|
|
return s_gpuCheck == GpuCheck::Present;
|
|
}
|
|
|
|
QProcessEnvironment discreteGpuEnvironment()
|
|
{
|
|
checkGpu();
|
|
return s_gpuEnv;
|
|
}
|
|
|
|
}
|