GPU fallback and XML + EDID update

GPU fallback and XML + EDID update

Author: Bud

Common/Include/AdapterOption.h

@@ -1,72 +1,129 @@
 #pragma once
 
-// Anonymous namespace for usings
-namespace
-{
-    using namespace std;
-    using namespace Microsoft::IndirectDisp;
-    using namespace Microsoft::WRL;
-
-    struct AdapterOption
-    {
-        bool hasTargetAdapter{};
-        LUID adapterLuid{};
-
-        void load(const char* path)
-        {
-            ifstream ifs{ path };
-
-            if (!ifs.is_open())
-            {
-                return;
-            }
+#include <wrl/client.h> 	// For ComPtr
+#include <dxgi.h>       	// For IDXGIAdapter, IDXGIFactory1
+#include <algorithm>    	// For sort
+
+using namespace std;
+using namespace Microsoft::WRL;
+
+// Structure to vector gpus
+struct GPUInfo {
+    wstring name;                // GPU name
+    ComPtr<IDXGIAdapter> adapter;// COM pointer to the adapter
+    DXGI_ADAPTER_DESC desc;      // Adapter description
+};
+
+// Sort function for GPUs by dedicated video memory
+bool CompareGPUs(const GPUInfo& a, const GPUInfo& b) {
+    return a.desc.DedicatedVideoMemory > b.desc.DedicatedVideoMemory;
+}
+
+// Get a enumerate list of available GPUs
+vector<GPUInfo> getAvailableGPUs() {
+    vector<GPUInfo> gpus; // Vector to hold all GPU's information
+
+    ComPtr<IDXGIFactory1> factory;
+    if (!SUCCEEDED(CreateDXGIFactory1(IID_PPV_ARGS(&factory)))) {
+        return gpus;
+    }
+
+    // Enumerate all adapters (GPUs)
+    for (UINT i = 0;; i++) {
+        ComPtr<IDXGIAdapter> adapter;
+        if (!SUCCEEDED(factory->EnumAdapters(i, &adapter))) {
+            break;
+        }
+
+        DXGI_ADAPTER_DESC desc;
+
+        if (!SUCCEEDED(adapter->GetDesc(&desc))) {
+            continue;
+        }
+
+        // Add the adapter information to the list
+        GPUInfo info{ desc.Description, adapter, desc };
+        gpus.push_back(info);
+    }
+
+    return gpus;
+}
+
+class AdapterOption {
+public:
+    bool hasTargetAdapter{}; // Indicates if a target adapter is selected
+    LUID adapterLuid{};      // Adapter's unique identifier (LUID)
+    wstring target_name{};   // Target adapter name
 
-            std::string line;
+    // Select the best GPU based on dedicated video memory
+    wstring selectBestGPU() {
+        auto gpus = getAvailableGPUs();
+        if (gpus.empty()) {
+            return L""; // Error check for headless / vm
+        }
+
+        // Sort GPUs by dedicated video memory in descending order
+        sort(gpus.begin(), gpus.end(), CompareGPUs);
+        auto bestGPU = gpus.front(); // Get the GPU with the most memory
+
+        return bestGPU.name;
+    }
+
+    // Load friendlyname from a file OR select the best GPU 
+    void load(const wchar_t* path) {
+        ifstream ifs{ path };
+
+        if (!ifs.is_open()) {
+            target_name = selectBestGPU();
+        }
+        else {
+            string line;
             getline(ifs, line);
+            target_name.assign(line.begin(), line.end());
+        }
 
-            std::wstring target_name{ line.begin(), line.end() };
+        // Find and set the adapter based on the target name
+        if (!findAndSetAdapter(target_name)) {
+            // If the adapter is not found, select the best GPU and retry
+            target_name = selectBestGPU();
+            findAndSetAdapter(target_name);
+        }
+    }
 
-            ComPtr<IDXGIFactory1> factory{};
-            if (!SUCCEEDED(CreateDXGIFactory1(IID_PPV_ARGS(&factory))))
-            {
-                return;
-            }
+    // Set the target adapter from a given name and validate it
+    void xmlprovide(const wstring& xtarg) {
+        target_name = xtarg;
+        if (!findAndSetAdapter(target_name)) {
+            // If the adapter is not found, select the best GPU and retry
+            target_name = selectBestGPU();
+            findAndSetAdapter(target_name);
+        }
+    }
 
-            for (UINT i = 0;; i++)
-            {
-                ComPtr<IDXGIAdapter> adapter{};
-                if (!SUCCEEDED(factory->EnumAdapters(i, &adapter)))
-                {
-                    break;
-                }
-
-                DXGI_ADAPTER_DESC desc;
-                if (!SUCCEEDED(adapter->GetDesc(&desc)))
-                {
-                    break;
-                }
-
-                // We found our target
-                if (_wcsicmp(target_name.c_str(), desc.Description) == 0)
-                {
-                    adapterLuid = desc.AdapterLuid;
-                    hasTargetAdapter = true;
-                }
-            }
+    // Apply the adapter settings to the specified adapter
+    void apply(const IDDCX_ADAPTER& adapter) {
+        if (hasTargetAdapter && IDD_IS_FUNCTION_AVAILABLE(IddCxAdapterSetRenderAdapter)) {
+            IDARG_IN_ADAPTERSETRENDERADAPTER arg{};
+            arg.PreferredRenderAdapter = adapterLuid;
+            IddCxAdapterSetRenderAdapter(adapter, &arg);
         }
+    }
+
+private:
+    // Find and set the adapter by its name
+    bool findAndSetAdapter(const wstring& adapterName) {
+        auto gpus = getAvailableGPUs();
 
-        void apply(const IDDCX_ADAPTER& adapter)
-        {
-            if (hasTargetAdapter)
-            {
-                if (IDD_IS_FUNCTION_AVAILABLE(IddCxAdapterSetRenderAdapter))
-                {
-                    IDARG_IN_ADAPTERSETRENDERADAPTER arg{};
-                    arg.PreferredRenderAdapter = adapterLuid;
-
-                    IddCxAdapterSetRenderAdapter(adapter, &arg);
-                }
+        // Iterate through all available GPUs
+        for (const auto& gpu : gpus) {
+            if (_wcsicmp(gpu.name.c_str(), adapterName.c_str()) == 0) {
+                adapterLuid = gpu.desc.AdapterLuid; // Set the adapter LUID
+                hasTargetAdapter = true; // Indicate that a target adapter is selected
+                return true;
             }
         }
-    };
-}
+
+        hasTargetAdapter = false; // Indicate that no target adapter is selected
+        return false;
+    }
+};