From 055d5527ef94cdb552252c946ae84be190f3a520 Mon Sep 17 00:00:00 2001
From: Geoffrey McRae <geoff@hostfission.com>
Date: Fri, 23 Feb 2024 08:14:24 +1100
Subject: [PATCH] [host] d12: allow specifying adapter and output to capture

---
 host/platform/Windows/capture/D12/d12.c | 56 ++++++++++++++++++++++++-
 1 file changed, 54 insertions(+), 2 deletions(-)

diff --git a/host/platform/Windows/capture/D12/d12.c b/host/platform/Windows/capture/D12/d12.c
index db51cba6..064ef913 100644
--- a/host/platform/Windows/capture/D12/d12.c
+++ b/host/platform/Windows/capture/D12/d12.c
@@ -117,6 +117,20 @@ static void d12_initOptions(void)
 {
   struct Option options[] =
   {
+    {
+      .module         = "d12",
+      .name           = "adapter",
+      .description    = "The name of the adapter to capture",
+      .type           = OPTION_TYPE_STRING,
+      .value.x_string = NULL
+    },
+    {
+      .module         = "d12",
+      .name           = "output",
+      .description    = "The name of the adapter's output to capture",
+      .type           = OPTION_TYPE_STRING,
+      .value.x_string = NULL
+    },
     {
       .module       = "d12",
       .name         = "allowRGB24",
@@ -557,6 +571,23 @@ static bool d12_enumerateDevices(
   DXGI_ADAPTER_DESC1 adapterDesc;
   DXGI_OUTPUT_DESC   outputDesc;
 
+  const char * _optAdapter = option_get_string("d12", "adapter");
+  const char * _optOutput  = option_get_string("d12", "output" );
+
+  wchar_t * optAdapter = NULL;
+  wchar_t * optOutput  = NULL;
+  if (_optAdapter)
+  {
+    optAdapter = malloc((strlen(_optAdapter) + 1) * 2);
+    mbstowcs(optAdapter, _optAdapter, strlen(_optAdapter));
+  }
+
+  if (_optOutput)
+  {
+    optOutput = malloc((strlen(_optOutput) + 1) * 2);
+    mbstowcs(optOutput, _optOutput, strlen(_optOutput));
+  }
+
   for(
     int i = 0;
     IDXGIFactory2_EnumAdapters1(*factory, i, adapter)
@@ -595,7 +626,15 @@ static bool d12_enumerateDevices(
     if (skip)
       continue;
 
-    // FIXME: Allow specifying the specific adapter
+    if (optAdapter)
+    {
+      if (wcsstr(adapterDesc.Description, optAdapter) == NULL)
+      {
+        DEBUG_INFO("Not using adapter: %ls", adapterDesc.Description);
+        continue;
+      }
+      DEBUG_INFO("Adapter matched, trying: %ls", adapterDesc.Description);
+    }
 
     for(
       int n = 0;
@@ -603,7 +642,17 @@ static bool d12_enumerateDevices(
       ++n, comRef_release(output))
     {
       IDXGIOutput_GetDesc(*output, &outputDesc);
-      // FIXME: Allow specifying the specific output
+
+      if (optOutput)
+      {
+        if (wcsstr(outputDesc.DeviceName, optOutput) == NULL)
+        {
+          DEBUG_INFO("Not using adapter output: %ls", outputDesc.DeviceName);
+          continue;
+        }
+
+        DEBUG_INFO("Adapter output matched, trying: %ls", outputDesc.DeviceName);
+      }
 
       if (outputDesc.AttachedToDesktop)
         break;
@@ -613,6 +662,9 @@ static bool d12_enumerateDevices(
       break;
   }
 
+  free(optAdapter);
+  free(optOutput );
+
   if (!*output)
   {
     DEBUG_ERROR("Failed to locate a valid output device");