duplicator-monitor-capture.c 24 KB

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