Procházet zdrojové kódy

libobs: Add funcs to get windows video adapter LUIDs

Useful for providing adapter order to encoder test subprocesses
jp9000 před 2 roky
rodič
revize
78f3389d17

+ 3 - 1
libobs/CMakeLists.txt

@@ -290,6 +290,8 @@ if(OS_WINDOWS)
             util/threading-windows.h
             util/pipe-windows.c
             util/platform-windows.c
+            util/windows/device-enum.c
+            util/windows/device-enum.h
             util/windows/obfuscate.c
             util/windows/obfuscate.h
             util/windows/win-registry.h
@@ -310,7 +312,7 @@ if(OS_WINDOWS)
     libobs PRIVATE UNICODE _UNICODE _CRT_SECURE_NO_WARNINGS
                    _CRT_NONSTDC_NO_WARNINGS)
 
-  target_link_libraries(libobs PRIVATE Avrt Dwmapi winmm)
+  target_link_libraries(libobs PRIVATE dxgi Avrt Dwmapi winmm)
 
   if(MSVC)
     target_link_libraries(libobs PUBLIC OBS::w32-pthreads)

+ 32 - 0
libobs/util/windows/device-enum.c

@@ -0,0 +1,32 @@
+#include "device-enum.h"
+#include "../dstr.h"
+
+#include <dxgi.h>
+
+void enum_graphics_device_luids(device_luid_cb device_luid, void *param)
+{
+	IDXGIFactory1 *factory;
+	IDXGIAdapter1 *adapter;
+	HRESULT hr;
+
+	hr = CreateDXGIFactory1(&IID_IDXGIFactory1, (void **)&factory);
+	if (FAILED(hr))
+		return;
+
+	for (UINT i = 0;
+	     factory->lpVtbl->EnumAdapters1(factory, i, &adapter) == S_OK;
+	     i++) {
+		DXGI_ADAPTER_DESC desc;
+
+		hr = adapter->lpVtbl->GetDesc(adapter, &desc);
+		adapter->lpVtbl->Release(adapter);
+		if (FAILED(hr))
+			continue;
+
+		uint64_t luid64 = *(uint64_t *)&desc.AdapterLuid;
+		if (!device_luid(param, i, luid64))
+			break;
+	}
+
+	factory->lpVtbl->Release(factory);
+}

+ 14 - 0
libobs/util/windows/device-enum.h

@@ -0,0 +1,14 @@
+#pragma once
+
+#include "../c99defs.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+typedef bool (*device_luid_cb)(void *param, uint32_t idx, uint64_t luid);
+EXPORT void enum_graphics_device_luids(device_luid_cb device_luid, void *param);
+
+#ifdef __cplusplus
+}
+#endif