duplicator-monitor-capture.c 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867
  1. #include <windows.h>
  2. #include <obs-module.h>
  3. #include <util/dstr.h>
  4. #include <util/threading.h>
  5. #include "cursor-capture.h"
  6. #ifdef OBS_LEGACY
  7. #include "../../libobs/util/platform.h"
  8. #include "../../libobs-winrt/winrt-capture.h"
  9. #else
  10. #include <util/platform.h>
  11. #include <winrt-capture.h>
  12. #endif
  13. #define do_log(level, format, ...) \
  14. blog(level, "[duplicator-monitor-capture: '%s'] " format, obs_source_get_name(capture->source), ##__VA_ARGS__)
  15. #define warn(format, ...) do_log(LOG_WARNING, format, ##__VA_ARGS__)
  16. #define info(format, ...) do_log(LOG_INFO, format, ##__VA_ARGS__)
  17. #define debug(format, ...) do_log(LOG_DEBUG, format, ##__VA_ARGS__)
  18. /* clang-format off */
  19. #define TEXT_MONITOR_CAPTURE obs_module_text("MonitorCapture")
  20. #define TEXT_CAPTURE_CURSOR obs_module_text("CaptureCursor")
  21. #define TEXT_COMPATIBILITY obs_module_text("Compatibility")
  22. #define TEXT_MONITOR obs_module_text("Monitor")
  23. #define TEXT_PRIMARY_MONITOR obs_module_text("PrimaryMonitor")
  24. #define TEXT_METHOD obs_module_text("Method")
  25. #define TEXT_METHOD_AUTO obs_module_text("WindowCapture.Method.Auto")
  26. #define TEXT_METHOD_DXGI obs_module_text("Method.DXGI")
  27. #define TEXT_METHOD_WGC obs_module_text("Method.WindowsGraphicsCapture")
  28. #define TEXT_FORCE_SDR obs_module_text("ForceSdr")
  29. /* clang-format on */
  30. #define RESET_INTERVAL_SEC 3.0f
  31. #define INVALID_DISPLAY "DUMMY"
  32. typedef BOOL (*PFN_winrt_capture_supported)();
  33. typedef BOOL (*PFN_winrt_capture_cursor_toggle_supported)();
  34. typedef struct winrt_capture *(*PFN_winrt_capture_init_monitor)(BOOL cursor, HMONITOR monitor, BOOL force_sdr);
  35. typedef void (*PFN_winrt_capture_free)(struct winrt_capture *capture);
  36. typedef BOOL (*PFN_winrt_capture_active)(const struct winrt_capture *capture);
  37. typedef enum gs_color_space (*PFN_winrt_capture_get_color_space)(const struct winrt_capture *capture);
  38. typedef void (*PFN_winrt_capture_render)(struct winrt_capture *capture);
  39. typedef uint32_t (*PFN_winrt_capture_width)(const struct winrt_capture *capture);
  40. typedef uint32_t (*PFN_winrt_capture_height)(const struct winrt_capture *capture);
  41. struct winrt_exports {
  42. PFN_winrt_capture_supported winrt_capture_supported;
  43. PFN_winrt_capture_cursor_toggle_supported winrt_capture_cursor_toggle_supported;
  44. PFN_winrt_capture_init_monitor winrt_capture_init_monitor;
  45. PFN_winrt_capture_free winrt_capture_free;
  46. PFN_winrt_capture_active winrt_capture_active;
  47. PFN_winrt_capture_get_color_space winrt_capture_get_color_space;
  48. PFN_winrt_capture_render winrt_capture_render;
  49. PFN_winrt_capture_width winrt_capture_width;
  50. PFN_winrt_capture_height winrt_capture_height;
  51. };
  52. enum display_capture_method {
  53. METHOD_AUTO,
  54. METHOD_DXGI,
  55. METHOD_WGC,
  56. };
  57. struct duplicator_capture {
  58. obs_source_t *source;
  59. pthread_mutex_t update_mutex;
  60. char monitor_id[128];
  61. char id[128];
  62. char alt_id[128];
  63. char monitor_name[64];
  64. enum display_capture_method method;
  65. bool reset_wgc;
  66. HMONITOR handle;
  67. bool capture_cursor;
  68. bool force_sdr;
  69. bool showing;
  70. LONG logged_width;
  71. LONG logged_height;
  72. long x;
  73. long y;
  74. int rot;
  75. uint32_t width;
  76. uint32_t height;
  77. gs_duplicator_t *duplicator;
  78. float reset_timeout;
  79. struct cursor_data cursor_data;
  80. void *winrt_module;
  81. struct winrt_exports exports;
  82. struct winrt_capture *capture_winrt;
  83. };
  84. struct duplicator_monitor_info {
  85. char device_id[128];
  86. char id[128];
  87. char alt_id[128];
  88. char name[128];
  89. RECT rect;
  90. HMONITOR handle;
  91. };
  92. /* ------------------------------------------------------------------------- */
  93. static const char *get_method_name(int method)
  94. {
  95. const char *method_name = "";
  96. switch (method) {
  97. case METHOD_DXGI:
  98. method_name = "DXGI";
  99. break;
  100. case METHOD_WGC:
  101. method_name = "WGC";
  102. break;
  103. }
  104. return method_name;
  105. }
  106. static bool GetMonitorTarget(LPCWSTR device, DISPLAYCONFIG_TARGET_DEVICE_NAME *target)
  107. {
  108. bool found = false;
  109. UINT32 numPath, numMode;
  110. if (GetDisplayConfigBufferSizes(QDC_ONLY_ACTIVE_PATHS, &numPath, &numMode) == ERROR_SUCCESS) {
  111. if (!numPath || !numMode) {
  112. return false;
  113. }
  114. DISPLAYCONFIG_PATH_INFO *paths = bmalloc(numPath * sizeof(DISPLAYCONFIG_PATH_INFO));
  115. DISPLAYCONFIG_MODE_INFO *modes = bmalloc(numMode * sizeof(DISPLAYCONFIG_MODE_INFO));
  116. if (QueryDisplayConfig(QDC_ONLY_ACTIVE_PATHS, &numPath, paths, &numMode, modes, NULL) ==
  117. ERROR_SUCCESS) {
  118. for (size_t i = 0; i < numPath; ++i) {
  119. const DISPLAYCONFIG_PATH_INFO *const path = &paths[i];
  120. DISPLAYCONFIG_SOURCE_DEVICE_NAME
  121. source;
  122. source.header.type = DISPLAYCONFIG_DEVICE_INFO_GET_SOURCE_NAME;
  123. source.header.size = sizeof(source);
  124. source.header.adapterId = path->sourceInfo.adapterId;
  125. source.header.id = path->sourceInfo.id;
  126. if (DisplayConfigGetDeviceInfo(&source.header) == ERROR_SUCCESS &&
  127. wcscmp(device, source.viewGdiDeviceName) == 0) {
  128. target->header.type = DISPLAYCONFIG_DEVICE_INFO_GET_TARGET_NAME;
  129. target->header.size = sizeof(*target);
  130. target->header.adapterId = path->sourceInfo.adapterId;
  131. target->header.id = path->targetInfo.id;
  132. found = DisplayConfigGetDeviceInfo(&target->header) == ERROR_SUCCESS;
  133. break;
  134. }
  135. }
  136. }
  137. bfree(modes);
  138. bfree(paths);
  139. }
  140. return found;
  141. }
  142. static void GetMonitorName(HMONITOR handle, char *name, size_t count)
  143. {
  144. MONITORINFOEXW mi;
  145. DISPLAYCONFIG_TARGET_DEVICE_NAME target;
  146. mi.cbSize = sizeof(mi);
  147. if (GetMonitorInfoW(handle, (LPMONITORINFO)&mi) && GetMonitorTarget(mi.szDevice, &target)) {
  148. char *friendly_name;
  149. os_wcs_to_utf8_ptr(target.monitorFriendlyDeviceName, 0, &friendly_name);
  150. strcpy_s(name, count, friendly_name);
  151. bfree(friendly_name);
  152. } else {
  153. strcpy_s(name, count, "[OBS: Unknown]");
  154. }
  155. }
  156. static BOOL CALLBACK enum_monitor(HMONITOR handle, HDC hdc, LPRECT rect, LPARAM param)
  157. {
  158. UNUSED_PARAMETER(hdc);
  159. struct duplicator_monitor_info *monitor = (struct duplicator_monitor_info *)param;
  160. bool match = false;
  161. MONITORINFOEXA mi;
  162. mi.cbSize = sizeof(mi);
  163. if (GetMonitorInfoA(handle, (LPMONITORINFO)&mi)) {
  164. DISPLAY_DEVICEA device;
  165. device.cb = sizeof(device);
  166. if (EnumDisplayDevicesA(mi.szDevice, 0, &device, EDD_GET_DEVICE_INTERFACE_NAME)) {
  167. match = strcmp(monitor->device_id, device.DeviceID) == 0;
  168. if (match) {
  169. strcpy_s(monitor->id, _countof(monitor->id), device.DeviceID);
  170. strcpy_s(monitor->alt_id, _countof(monitor->alt_id), mi.szDevice);
  171. GetMonitorName(handle, monitor->name, _countof(monitor->name));
  172. monitor->rect = *rect;
  173. monitor->handle = handle;
  174. }
  175. }
  176. }
  177. return !match;
  178. }
  179. static BOOL CALLBACK enum_monitor_fallback(HMONITOR handle, HDC hdc, LPRECT rect, LPARAM param)
  180. {
  181. UNUSED_PARAMETER(hdc);
  182. struct duplicator_monitor_info *monitor = (struct duplicator_monitor_info *)param;
  183. bool match = false;
  184. MONITORINFOEXA mi;
  185. mi.cbSize = sizeof(mi);
  186. if (GetMonitorInfoA(handle, (LPMONITORINFO)&mi)) {
  187. match = strcmp(monitor->device_id, mi.szDevice) == 0;
  188. if (match) {
  189. strcpy_s(monitor->alt_id, _countof(monitor->alt_id), mi.szDevice);
  190. GetMonitorName(handle, monitor->name, _countof(monitor->name));
  191. monitor->rect = *rect;
  192. monitor->handle = handle;
  193. }
  194. }
  195. return !match;
  196. }
  197. static void log_settings(struct duplicator_capture *capture, const char *monitor, LONG width, LONG height)
  198. {
  199. info("update settings:\n"
  200. "\tdisplay: %s (%ldx%ld)\n"
  201. "\tcursor: %s\n"
  202. "\tmethod: %s\n"
  203. "\tid: %s\n"
  204. "\talt_id: %s\n"
  205. "\tsetting_id: %s\n"
  206. "\tforce SDR: %s",
  207. monitor, width, height, capture->capture_cursor ? "true" : "false", get_method_name(capture->method),
  208. capture->id, capture->alt_id, capture->monitor_id, capture->force_sdr ? "true" : "false");
  209. }
  210. static enum display_capture_method choose_method(enum display_capture_method method, bool wgc_supported,
  211. HMONITOR monitor)
  212. {
  213. if (!wgc_supported)
  214. method = METHOD_DXGI;
  215. if (method == METHOD_AUTO) {
  216. method = METHOD_DXGI;
  217. obs_enter_graphics();
  218. const int dxgi_index = gs_duplicator_get_monitor_index(monitor);
  219. obs_leave_graphics();
  220. if (dxgi_index == -1) {
  221. method = METHOD_WGC;
  222. } else {
  223. SYSTEM_POWER_STATUS status;
  224. if (GetSystemPowerStatus(&status) && status.BatteryFlag < 128) {
  225. obs_enter_graphics();
  226. const uint32_t count = gs_get_adapter_count();
  227. obs_leave_graphics();
  228. if (count >= 2)
  229. method = METHOD_WGC;
  230. }
  231. }
  232. }
  233. return method;
  234. }
  235. extern bool wgc_supported;
  236. static struct duplicator_monitor_info find_monitor(const char *monitor_id)
  237. {
  238. struct duplicator_monitor_info monitor = {0};
  239. strcpy_s(monitor.device_id, _countof(monitor.device_id), monitor_id);
  240. EnumDisplayMonitors(NULL, NULL, &enum_monitor, (LPARAM)&monitor);
  241. if (monitor.handle == NULL) {
  242. EnumDisplayMonitors(NULL, NULL, &enum_monitor_fallback, (LPARAM)&monitor);
  243. }
  244. return monitor;
  245. }
  246. static inline void update_settings(struct duplicator_capture *capture, obs_data_t *settings)
  247. {
  248. pthread_mutex_lock(&capture->update_mutex);
  249. struct duplicator_monitor_info monitor = find_monitor(obs_data_get_string(settings, "monitor_id"));
  250. capture->method = choose_method((int)obs_data_get_int(settings, "method"), wgc_supported, monitor.handle);
  251. strcpy_s(capture->monitor_id, _countof(capture->monitor_id), monitor.device_id);
  252. strcpy_s(capture->id, _countof(capture->id), monitor.id);
  253. strcpy_s(capture->alt_id, _countof(capture->alt_id), monitor.alt_id);
  254. strcpy_s(capture->monitor_name, _countof(capture->monitor_name), monitor.name);
  255. capture->handle = monitor.handle;
  256. capture->capture_cursor = obs_data_get_bool(settings, "capture_cursor");
  257. capture->force_sdr = obs_data_get_bool(settings, "force_sdr");
  258. capture->logged_width = monitor.rect.right - monitor.rect.left;
  259. capture->logged_height = monitor.rect.bottom - monitor.rect.top;
  260. if (capture->duplicator) {
  261. obs_enter_graphics();
  262. gs_duplicator_destroy(capture->duplicator);
  263. capture->duplicator = NULL;
  264. obs_leave_graphics();
  265. }
  266. capture->width = 0;
  267. capture->height = 0;
  268. capture->x = 0;
  269. capture->y = 0;
  270. capture->rot = 0;
  271. capture->reset_timeout = RESET_INTERVAL_SEC;
  272. pthread_mutex_unlock(&capture->update_mutex);
  273. }
  274. /* ------------------------------------------------------------------------- */
  275. static const char *duplicator_capture_getname(void *unused)
  276. {
  277. UNUSED_PARAMETER(unused);
  278. return TEXT_MONITOR_CAPTURE;
  279. }
  280. static void duplicator_actual_destroy(void *data)
  281. {
  282. struct duplicator_capture *capture = data;
  283. if (capture->capture_winrt) {
  284. capture->exports.winrt_capture_free(capture->capture_winrt);
  285. capture->capture_winrt = NULL;
  286. }
  287. obs_enter_graphics();
  288. if (capture->duplicator) {
  289. gs_duplicator_destroy(capture->duplicator);
  290. capture->duplicator = NULL;
  291. }
  292. cursor_data_free(&capture->cursor_data);
  293. obs_leave_graphics();
  294. if (capture->winrt_module) {
  295. os_dlclose(capture->winrt_module);
  296. capture->winrt_module = NULL;
  297. }
  298. pthread_mutex_destroy(&capture->update_mutex);
  299. bfree(capture);
  300. }
  301. static void duplicator_capture_destroy(void *data)
  302. {
  303. obs_queue_task(OBS_TASK_GRAPHICS, duplicator_actual_destroy, data, false);
  304. }
  305. static void duplicator_capture_defaults(obs_data_t *settings)
  306. {
  307. obs_data_set_default_int(settings, "method", METHOD_AUTO);
  308. obs_data_set_default_string(settings, "monitor_id", INVALID_DISPLAY);
  309. obs_data_set_default_int(settings, "monitor_wgc", 0);
  310. obs_data_set_default_bool(settings, "capture_cursor", true);
  311. obs_data_set_default_bool(settings, "force_sdr", false);
  312. }
  313. static void duplicator_capture_update(void *data, obs_data_t *settings)
  314. {
  315. struct duplicator_capture *mc = data;
  316. update_settings(mc, settings);
  317. log_settings(mc, mc->monitor_name, mc->logged_width, mc->logged_height);
  318. mc->reset_wgc = true;
  319. }
  320. #define WINRT_IMPORT(func) \
  321. do { \
  322. exports->func = (PFN_##func)os_dlsym(module, #func); \
  323. if (!exports->func) { \
  324. success = false; \
  325. blog(LOG_ERROR, \
  326. "Could not load function '%s' from " \
  327. "module '%s'", \
  328. #func, module_name); \
  329. } \
  330. } while (false)
  331. static bool load_winrt_imports(struct winrt_exports *exports, void *module, const char *module_name)
  332. {
  333. bool success = true;
  334. WINRT_IMPORT(winrt_capture_supported);
  335. WINRT_IMPORT(winrt_capture_cursor_toggle_supported);
  336. WINRT_IMPORT(winrt_capture_init_monitor);
  337. WINRT_IMPORT(winrt_capture_free);
  338. WINRT_IMPORT(winrt_capture_active);
  339. WINRT_IMPORT(winrt_capture_get_color_space);
  340. WINRT_IMPORT(winrt_capture_render);
  341. WINRT_IMPORT(winrt_capture_width);
  342. WINRT_IMPORT(winrt_capture_height);
  343. return success;
  344. }
  345. extern bool graphics_uses_d3d11;
  346. static void *duplicator_capture_create(obs_data_t *settings, obs_source_t *source)
  347. {
  348. struct duplicator_capture *capture;
  349. capture = bzalloc(sizeof(struct duplicator_capture));
  350. capture->source = source;
  351. pthread_mutex_init(&capture->update_mutex, NULL);
  352. if (graphics_uses_d3d11) {
  353. static const char *const module = "libobs-winrt";
  354. capture->winrt_module = os_dlopen(module);
  355. if (capture->winrt_module) {
  356. load_winrt_imports(&capture->exports, capture->winrt_module, module);
  357. }
  358. }
  359. update_settings(capture, settings);
  360. log_settings(capture, capture->monitor_name, capture->logged_width, capture->logged_height);
  361. return capture;
  362. }
  363. static void reset_capture_data(struct duplicator_capture *capture)
  364. {
  365. struct gs_monitor_info monitor_info = {0};
  366. gs_texture_t *texture = gs_duplicator_get_texture(capture->duplicator);
  367. const int dxgi_index = gs_duplicator_get_monitor_index(capture->handle);
  368. gs_get_duplicator_monitor_info(dxgi_index, &monitor_info);
  369. if (texture) {
  370. capture->width = gs_texture_get_width(texture);
  371. capture->height = gs_texture_get_height(texture);
  372. } else {
  373. capture->width = 0;
  374. capture->height = 0;
  375. }
  376. capture->x = monitor_info.x;
  377. capture->y = monitor_info.y;
  378. capture->rot = monitor_info.rotation_degrees;
  379. }
  380. static void free_capture_data(struct duplicator_capture *capture)
  381. {
  382. if (capture->capture_winrt) {
  383. capture->exports.winrt_capture_free(capture->capture_winrt);
  384. capture->capture_winrt = NULL;
  385. }
  386. if (capture->duplicator) {
  387. gs_duplicator_destroy(capture->duplicator);
  388. capture->duplicator = NULL;
  389. }
  390. cursor_data_free(&capture->cursor_data);
  391. capture->width = 0;
  392. capture->height = 0;
  393. capture->x = 0;
  394. capture->y = 0;
  395. capture->rot = 0;
  396. capture->reset_timeout = 0.0f;
  397. }
  398. static void update_monitor_handle(struct duplicator_capture *capture)
  399. {
  400. capture->handle = find_monitor(capture->monitor_id).handle;
  401. }
  402. static void duplicator_capture_tick(void *data, float seconds)
  403. {
  404. struct duplicator_capture *capture = data;
  405. /* completely shut down monitor capture if not in use, otherwise it can
  406. * sometimes generate system lag when a game is in fullscreen mode */
  407. if (!obs_source_showing(capture->source)) {
  408. if (capture->showing) {
  409. obs_enter_graphics();
  410. free_capture_data(capture);
  411. obs_leave_graphics();
  412. capture->showing = false;
  413. }
  414. return;
  415. }
  416. /* always try to load the capture immediately when the source is first
  417. * shown */
  418. if (!capture->showing) {
  419. capture->reset_timeout = RESET_INTERVAL_SEC;
  420. }
  421. obs_enter_graphics();
  422. if (capture->method == METHOD_WGC) {
  423. if (capture->reset_wgc && capture->capture_winrt) {
  424. capture->exports.winrt_capture_free(capture->capture_winrt);
  425. capture->capture_winrt = NULL;
  426. capture->reset_wgc = false;
  427. capture->reset_timeout = RESET_INTERVAL_SEC;
  428. }
  429. if (!capture->capture_winrt) {
  430. capture->reset_timeout += seconds;
  431. if (capture->reset_timeout >= RESET_INTERVAL_SEC) {
  432. if (!capture->handle)
  433. update_monitor_handle(capture);
  434. if (capture->handle) {
  435. capture->capture_winrt = capture->exports.winrt_capture_init_monitor(
  436. capture->capture_cursor, capture->handle, capture->force_sdr);
  437. if (!capture->capture_winrt) {
  438. update_monitor_handle(capture);
  439. if (capture->handle) {
  440. capture->capture_winrt =
  441. capture->exports.winrt_capture_init_monitor(
  442. capture->capture_cursor, capture->handle,
  443. capture->force_sdr);
  444. }
  445. }
  446. }
  447. capture->reset_timeout = 0.0f;
  448. }
  449. }
  450. } else {
  451. if (capture->capture_winrt) {
  452. capture->exports.winrt_capture_free(capture->capture_winrt);
  453. capture->capture_winrt = NULL;
  454. }
  455. if (!capture->duplicator) {
  456. capture->reset_timeout += seconds;
  457. if (capture->reset_timeout >= RESET_INTERVAL_SEC) {
  458. if (!capture->handle)
  459. update_monitor_handle(capture);
  460. if (capture->handle) {
  461. int dxgi_index = gs_duplicator_get_monitor_index(capture->handle);
  462. if (dxgi_index == -1) {
  463. update_monitor_handle(capture);
  464. if (capture->handle) {
  465. dxgi_index = gs_duplicator_get_monitor_index(capture->handle);
  466. }
  467. }
  468. if (dxgi_index != -1) {
  469. capture->duplicator = gs_duplicator_create(dxgi_index);
  470. }
  471. }
  472. capture->reset_timeout = 0.0f;
  473. }
  474. }
  475. if (capture->duplicator) {
  476. if (capture->capture_cursor)
  477. cursor_capture(&capture->cursor_data);
  478. if (!gs_duplicator_update_frame(capture->duplicator)) {
  479. free_capture_data(capture);
  480. } else if (capture->width == 0) {
  481. reset_capture_data(capture);
  482. }
  483. }
  484. }
  485. obs_leave_graphics();
  486. if (!capture->showing)
  487. capture->showing = true;
  488. }
  489. static uint32_t duplicator_capture_width(void *data)
  490. {
  491. struct duplicator_capture *capture = data;
  492. return (capture->method == METHOD_WGC) ? capture->exports.winrt_capture_width(capture->capture_winrt)
  493. : (capture->rot % 180 == 0 ? capture->width : capture->height);
  494. }
  495. static uint32_t duplicator_capture_height(void *data)
  496. {
  497. struct duplicator_capture *capture = data;
  498. return (capture->method == METHOD_WGC) ? capture->exports.winrt_capture_height(capture->capture_winrt)
  499. : (capture->rot % 180 == 0 ? capture->height : capture->width);
  500. }
  501. static void draw_cursor(struct duplicator_capture *capture)
  502. {
  503. cursor_draw(&capture->cursor_data, -capture->x, -capture->y,
  504. capture->rot % 180 == 0 ? capture->width : capture->height,
  505. capture->rot % 180 == 0 ? capture->height : capture->width);
  506. }
  507. static void duplicator_capture_render(void *data, gs_effect_t *unused)
  508. {
  509. UNUSED_PARAMETER(unused);
  510. struct duplicator_capture *capture = data;
  511. if (capture->method == METHOD_WGC) {
  512. if (capture->capture_winrt) {
  513. if (capture->exports.winrt_capture_active(capture->capture_winrt)) {
  514. capture->exports.winrt_capture_render(capture->capture_winrt);
  515. } else {
  516. capture->exports.winrt_capture_free(capture->capture_winrt);
  517. capture->capture_winrt = NULL;
  518. }
  519. }
  520. } else {
  521. if (!capture->duplicator)
  522. return;
  523. gs_texture_t *const texture = gs_duplicator_get_texture(capture->duplicator);
  524. if (!texture)
  525. return;
  526. const bool previous = gs_framebuffer_srgb_enabled();
  527. gs_enable_framebuffer_srgb(true);
  528. gs_enable_blending(false);
  529. const int rot = capture->rot;
  530. if (rot != 0) {
  531. float x = 0.0f;
  532. float y = 0.0f;
  533. switch (rot) {
  534. case 90:
  535. x = (float)capture->height;
  536. break;
  537. case 180:
  538. x = (float)capture->width;
  539. y = (float)capture->height;
  540. break;
  541. case 270:
  542. y = (float)capture->width;
  543. break;
  544. }
  545. gs_matrix_push();
  546. gs_matrix_translate3f(x, y, 0.0f);
  547. gs_matrix_rotaa4f(0.0f, 0.0f, 1.0f, RAD((float)rot));
  548. }
  549. const char *tech_name = "Draw";
  550. float multiplier = 1.f;
  551. const enum gs_color_space current_space = gs_get_color_space();
  552. if (gs_duplicator_get_color_space(capture->duplicator) == GS_CS_709_SCRGB) {
  553. if (capture->force_sdr) {
  554. tech_name = "DrawMultiply";
  555. const float target_nits =
  556. (current_space == GS_CS_709_SCRGB) ? obs_get_video_sdr_white_level() : 80.f;
  557. multiplier = target_nits / gs_duplicator_get_sdr_white_level(capture->duplicator);
  558. } else {
  559. switch (current_space) {
  560. case GS_CS_SRGB:
  561. case GS_CS_SRGB_16F:
  562. tech_name = "DrawMultiplyTonemap";
  563. multiplier = 80.f / obs_get_video_sdr_white_level();
  564. break;
  565. case GS_CS_709_EXTENDED:
  566. tech_name = "DrawMultiply";
  567. multiplier = 80.f / obs_get_video_sdr_white_level();
  568. }
  569. }
  570. } else if (current_space == GS_CS_709_SCRGB) {
  571. tech_name = "DrawMultiply";
  572. multiplier = obs_get_video_sdr_white_level() / 80.f;
  573. }
  574. gs_effect_t *const opaque_effect = obs_get_base_effect(OBS_EFFECT_OPAQUE);
  575. gs_eparam_t *multiplier_param = gs_effect_get_param_by_name(opaque_effect, "multiplier");
  576. gs_effect_set_float(multiplier_param, multiplier);
  577. gs_eparam_t *image_param = gs_effect_get_param_by_name(opaque_effect, "image");
  578. gs_effect_set_texture_srgb(image_param, texture);
  579. while (gs_effect_loop(opaque_effect, tech_name)) {
  580. gs_draw_sprite(texture, 0, 0, 0);
  581. }
  582. if (rot != 0)
  583. gs_matrix_pop();
  584. gs_enable_blending(true);
  585. gs_enable_framebuffer_srgb(previous);
  586. if (capture->capture_cursor) {
  587. gs_effect_t *const default_effect = obs_get_base_effect(OBS_EFFECT_DEFAULT);
  588. while (gs_effect_loop(default_effect, "Draw")) {
  589. draw_cursor(capture);
  590. }
  591. }
  592. }
  593. }
  594. static BOOL CALLBACK enum_monitor_props(HMONITOR handle, HDC hdc, LPRECT rect, LPARAM param)
  595. {
  596. UNUSED_PARAMETER(hdc);
  597. UNUSED_PARAMETER(rect);
  598. char monitor_name[64];
  599. GetMonitorName(handle, monitor_name, sizeof(monitor_name));
  600. MONITORINFOEXA mi;
  601. mi.cbSize = sizeof(mi);
  602. if (GetMonitorInfoA(handle, (LPMONITORINFO)&mi)) {
  603. obs_property_t *monitor_list = (obs_property_t *)param;
  604. struct dstr monitor_desc = {0};
  605. dstr_printf(&monitor_desc, "%s: %dx%d @ %d,%d", monitor_name, mi.rcMonitor.right - mi.rcMonitor.left,
  606. mi.rcMonitor.bottom - mi.rcMonitor.top, mi.rcMonitor.left, mi.rcMonitor.top);
  607. if (mi.dwFlags == MONITORINFOF_PRIMARY)
  608. dstr_catf(&monitor_desc, " (%s)", TEXT_PRIMARY_MONITOR);
  609. DISPLAY_DEVICEA device;
  610. device.cb = sizeof(device);
  611. if (EnumDisplayDevicesA(mi.szDevice, 0, &device, EDD_GET_DEVICE_INTERFACE_NAME)) {
  612. obs_property_list_add_string(monitor_list, monitor_desc.array, device.DeviceID);
  613. } else {
  614. blog(LOG_WARNING,
  615. "[duplicator-monitor-capture] EnumDisplayDevices failed for monitor (%s), falling back to szDevice",
  616. monitor_name);
  617. obs_property_list_add_string(monitor_list, monitor_desc.array, mi.szDevice);
  618. }
  619. dstr_free(&monitor_desc);
  620. } else {
  621. blog(LOG_WARNING, "[duplicator-monitor-capture] GetMonitorInfo failed for monitor: %s", monitor_name);
  622. }
  623. return TRUE;
  624. }
  625. static void update_settings_visibility(obs_properties_t *props, struct duplicator_capture *capture)
  626. {
  627. pthread_mutex_lock(&capture->update_mutex);
  628. const enum display_capture_method method = capture->method;
  629. const bool dxgi_options = method == METHOD_DXGI;
  630. const bool wgc_options = method == METHOD_WGC;
  631. const bool wgc_cursor_toggle = wgc_options && capture->exports.winrt_capture_cursor_toggle_supported();
  632. obs_property_t *p = obs_properties_get(props, "cursor");
  633. obs_property_set_visible(p, dxgi_options || wgc_cursor_toggle);
  634. pthread_mutex_unlock(&capture->update_mutex);
  635. }
  636. static bool display_capture_method_changed(obs_properties_t *props, obs_property_t *p, obs_data_t *settings)
  637. {
  638. UNUSED_PARAMETER(p);
  639. struct duplicator_capture *capture = obs_properties_get_param(props);
  640. if (!capture)
  641. return false;
  642. update_settings(capture, settings);
  643. update_settings_visibility(props, capture);
  644. return true;
  645. }
  646. static obs_properties_t *duplicator_capture_properties(void *data)
  647. {
  648. struct duplicator_capture *capture = data;
  649. obs_properties_t *props = obs_properties_create();
  650. obs_properties_set_param(props, capture, NULL);
  651. obs_property_t *p =
  652. obs_properties_add_list(props, "method", TEXT_METHOD, OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_INT);
  653. obs_property_list_add_int(p, TEXT_METHOD_AUTO, METHOD_AUTO);
  654. obs_property_list_add_int(p, TEXT_METHOD_DXGI, METHOD_DXGI);
  655. obs_property_list_add_int(p, TEXT_METHOD_WGC, METHOD_WGC);
  656. obs_property_list_item_disable(p, 2, !wgc_supported);
  657. obs_property_set_modified_callback(p, display_capture_method_changed);
  658. obs_property_t *monitors = obs_properties_add_list(props, "monitor_id", TEXT_MONITOR, OBS_COMBO_TYPE_LIST,
  659. OBS_COMBO_FORMAT_STRING);
  660. if (capture && strcmp(capture->monitor_id, INVALID_DISPLAY) == 0) {
  661. obs_property_list_add_string(monitors, obs_module_text("SelectADisplay"), INVALID_DISPLAY);
  662. obs_property_list_item_disable(monitors, 0, true);
  663. }
  664. obs_properties_add_bool(props, "capture_cursor", TEXT_CAPTURE_CURSOR);
  665. obs_properties_add_bool(props, "force_sdr", TEXT_FORCE_SDR);
  666. EnumDisplayMonitors(NULL, NULL, enum_monitor_props, (LPARAM)monitors);
  667. return props;
  668. }
  669. enum gs_color_space duplicator_capture_get_color_space(void *data, size_t count,
  670. const enum gs_color_space *preferred_spaces)
  671. {
  672. enum gs_color_space capture_space = GS_CS_SRGB;
  673. struct duplicator_capture *capture = data;
  674. if (capture->method == METHOD_WGC) {
  675. if (capture->capture_winrt) {
  676. capture_space = capture->exports.winrt_capture_get_color_space(capture->capture_winrt);
  677. }
  678. } else if (capture->duplicator && !capture->force_sdr) {
  679. capture_space = gs_duplicator_get_color_space(capture->duplicator);
  680. }
  681. enum gs_color_space space = capture_space;
  682. for (size_t i = 0; i < count; ++i) {
  683. const enum gs_color_space preferred_space = preferred_spaces[i];
  684. space = preferred_space;
  685. if (preferred_space == capture_space)
  686. break;
  687. }
  688. return space;
  689. }
  690. struct obs_source_info duplicator_capture_info = {
  691. .id = "monitor_capture",
  692. .type = OBS_SOURCE_TYPE_INPUT,
  693. .output_flags = OBS_SOURCE_VIDEO | OBS_SOURCE_CUSTOM_DRAW | OBS_SOURCE_DO_NOT_DUPLICATE | OBS_SOURCE_SRGB,
  694. .get_name = duplicator_capture_getname,
  695. .create = duplicator_capture_create,
  696. .destroy = duplicator_capture_destroy,
  697. .video_render = duplicator_capture_render,
  698. .video_tick = duplicator_capture_tick,
  699. .update = duplicator_capture_update,
  700. .get_width = duplicator_capture_width,
  701. .get_height = duplicator_capture_height,
  702. .get_defaults = duplicator_capture_defaults,
  703. .get_properties = duplicator_capture_properties,
  704. .icon_type = OBS_ICON_TYPE_DESKTOP_CAPTURE,
  705. .video_get_color_space = duplicator_capture_get_color_space,
  706. };