File indexing completed on 2024-04-28 03:55:31

 /*
     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"
 
 #ifndef KIO_ANDROID_STUB
 #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()
 {
 #ifndef KIO_ANDROID_STUB
     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()
 {
 #ifndef KIO_ANDROID_STUB
     // 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;
 }
 
 }