123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867 |
- #include <windows.h>
- #include <obs-module.h>
- #include <util/dstr.h>
- #include <util/threading.h>
- #include "cursor-capture.h"
- #ifdef OBS_LEGACY
- #include "../../libobs/util/platform.h"
- #include "../../libobs-winrt/winrt-capture.h"
- #else
- #include <util/platform.h>
- #include <winrt-capture.h>
- #endif
- #define do_log(level, format, ...) \
- blog(level, "[duplicator-monitor-capture: '%s'] " format, obs_source_get_name(capture->source), ##__VA_ARGS__)
- #define warn(format, ...) do_log(LOG_WARNING, format, ##__VA_ARGS__)
- #define info(format, ...) do_log(LOG_INFO, format, ##__VA_ARGS__)
- #define debug(format, ...) do_log(LOG_DEBUG, format, ##__VA_ARGS__)
- /* clang-format off */
- #define TEXT_MONITOR_CAPTURE obs_module_text("MonitorCapture")
- #define TEXT_CAPTURE_CURSOR obs_module_text("CaptureCursor")
- #define TEXT_COMPATIBILITY obs_module_text("Compatibility")
- #define TEXT_MONITOR obs_module_text("Monitor")
- #define TEXT_PRIMARY_MONITOR obs_module_text("PrimaryMonitor")
- #define TEXT_METHOD obs_module_text("Method")
- #define TEXT_METHOD_AUTO obs_module_text("WindowCapture.Method.Auto")
- #define TEXT_METHOD_DXGI obs_module_text("Method.DXGI")
- #define TEXT_METHOD_WGC obs_module_text("Method.WindowsGraphicsCapture")
- #define TEXT_FORCE_SDR obs_module_text("ForceSdr")
- /* clang-format on */
- #define RESET_INTERVAL_SEC 3.0f
- #define INVALID_DISPLAY "DUMMY"
- typedef BOOL (*PFN_winrt_capture_supported)();
- typedef BOOL (*PFN_winrt_capture_cursor_toggle_supported)();
- typedef struct winrt_capture *(*PFN_winrt_capture_init_monitor)(BOOL cursor, HMONITOR monitor, BOOL force_sdr);
- typedef void (*PFN_winrt_capture_free)(struct winrt_capture *capture);
- typedef BOOL (*PFN_winrt_capture_active)(const struct winrt_capture *capture);
- typedef enum gs_color_space (*PFN_winrt_capture_get_color_space)(const struct winrt_capture *capture);
- typedef void (*PFN_winrt_capture_render)(struct winrt_capture *capture);
- typedef uint32_t (*PFN_winrt_capture_width)(const struct winrt_capture *capture);
- typedef uint32_t (*PFN_winrt_capture_height)(const struct winrt_capture *capture);
- struct winrt_exports {
- PFN_winrt_capture_supported winrt_capture_supported;
- PFN_winrt_capture_cursor_toggle_supported winrt_capture_cursor_toggle_supported;
- PFN_winrt_capture_init_monitor winrt_capture_init_monitor;
- PFN_winrt_capture_free winrt_capture_free;
- PFN_winrt_capture_active winrt_capture_active;
- PFN_winrt_capture_get_color_space winrt_capture_get_color_space;
- PFN_winrt_capture_render winrt_capture_render;
- PFN_winrt_capture_width winrt_capture_width;
- PFN_winrt_capture_height winrt_capture_height;
- };
- enum display_capture_method {
- METHOD_AUTO,
- METHOD_DXGI,
- METHOD_WGC,
- };
- struct duplicator_capture {
- obs_source_t *source;
- pthread_mutex_t update_mutex;
- char monitor_id[128];
- char id[128];
- char alt_id[128];
- char monitor_name[64];
- enum display_capture_method method;
- bool reset_wgc;
- HMONITOR handle;
- bool capture_cursor;
- bool force_sdr;
- bool showing;
- LONG logged_width;
- LONG logged_height;
- long x;
- long y;
- int rot;
- uint32_t width;
- uint32_t height;
- gs_duplicator_t *duplicator;
- float reset_timeout;
- struct cursor_data cursor_data;
- void *winrt_module;
- struct winrt_exports exports;
- struct winrt_capture *capture_winrt;
- };
- struct duplicator_monitor_info {
- char device_id[128];
- char id[128];
- char alt_id[128];
- char name[128];
- RECT rect;
- HMONITOR handle;
- };
- /* ------------------------------------------------------------------------- */
- static const char *get_method_name(int method)
- {
- const char *method_name = "";
- switch (method) {
- case METHOD_DXGI:
- method_name = "DXGI";
- break;
- case METHOD_WGC:
- method_name = "WGC";
- break;
- }
- return method_name;
- }
- static bool GetMonitorTarget(LPCWSTR device, DISPLAYCONFIG_TARGET_DEVICE_NAME *target)
- {
- bool found = false;
- UINT32 numPath, numMode;
- if (GetDisplayConfigBufferSizes(QDC_ONLY_ACTIVE_PATHS, &numPath, &numMode) == ERROR_SUCCESS) {
- if (!numPath || !numMode) {
- return false;
- }
- DISPLAYCONFIG_PATH_INFO *paths = bmalloc(numPath * sizeof(DISPLAYCONFIG_PATH_INFO));
- DISPLAYCONFIG_MODE_INFO *modes = bmalloc(numMode * sizeof(DISPLAYCONFIG_MODE_INFO));
- if (QueryDisplayConfig(QDC_ONLY_ACTIVE_PATHS, &numPath, paths, &numMode, modes, NULL) ==
- ERROR_SUCCESS) {
- for (size_t i = 0; i < numPath; ++i) {
- const DISPLAYCONFIG_PATH_INFO *const path = &paths[i];
- DISPLAYCONFIG_SOURCE_DEVICE_NAME
- source;
- source.header.type = DISPLAYCONFIG_DEVICE_INFO_GET_SOURCE_NAME;
- source.header.size = sizeof(source);
- source.header.adapterId = path->sourceInfo.adapterId;
- source.header.id = path->sourceInfo.id;
- if (DisplayConfigGetDeviceInfo(&source.header) == ERROR_SUCCESS &&
- wcscmp(device, source.viewGdiDeviceName) == 0) {
- target->header.type = DISPLAYCONFIG_DEVICE_INFO_GET_TARGET_NAME;
- target->header.size = sizeof(*target);
- target->header.adapterId = path->sourceInfo.adapterId;
- target->header.id = path->targetInfo.id;
- found = DisplayConfigGetDeviceInfo(&target->header) == ERROR_SUCCESS;
- break;
- }
- }
- }
- bfree(modes);
- bfree(paths);
- }
- return found;
- }
- static void GetMonitorName(HMONITOR handle, char *name, size_t count)
- {
- MONITORINFOEXW mi;
- DISPLAYCONFIG_TARGET_DEVICE_NAME target;
- mi.cbSize = sizeof(mi);
- if (GetMonitorInfoW(handle, (LPMONITORINFO)&mi) && GetMonitorTarget(mi.szDevice, &target)) {
- char *friendly_name;
- os_wcs_to_utf8_ptr(target.monitorFriendlyDeviceName, 0, &friendly_name);
- strcpy_s(name, count, friendly_name);
- bfree(friendly_name);
- } else {
- strcpy_s(name, count, "[OBS: Unknown]");
- }
- }
- static BOOL CALLBACK enum_monitor(HMONITOR handle, HDC hdc, LPRECT rect, LPARAM param)
- {
- UNUSED_PARAMETER(hdc);
- struct duplicator_monitor_info *monitor = (struct duplicator_monitor_info *)param;
- bool match = false;
- MONITORINFOEXA mi;
- mi.cbSize = sizeof(mi);
- if (GetMonitorInfoA(handle, (LPMONITORINFO)&mi)) {
- DISPLAY_DEVICEA device;
- device.cb = sizeof(device);
- if (EnumDisplayDevicesA(mi.szDevice, 0, &device, EDD_GET_DEVICE_INTERFACE_NAME)) {
- match = strcmp(monitor->device_id, device.DeviceID) == 0;
- if (match) {
- strcpy_s(monitor->id, _countof(monitor->id), device.DeviceID);
- strcpy_s(monitor->alt_id, _countof(monitor->alt_id), mi.szDevice);
- GetMonitorName(handle, monitor->name, _countof(monitor->name));
- monitor->rect = *rect;
- monitor->handle = handle;
- }
- }
- }
- return !match;
- }
- static BOOL CALLBACK enum_monitor_fallback(HMONITOR handle, HDC hdc, LPRECT rect, LPARAM param)
- {
- UNUSED_PARAMETER(hdc);
- struct duplicator_monitor_info *monitor = (struct duplicator_monitor_info *)param;
- bool match = false;
- MONITORINFOEXA mi;
- mi.cbSize = sizeof(mi);
- if (GetMonitorInfoA(handle, (LPMONITORINFO)&mi)) {
- match = strcmp(monitor->device_id, mi.szDevice) == 0;
- if (match) {
- strcpy_s(monitor->alt_id, _countof(monitor->alt_id), mi.szDevice);
- GetMonitorName(handle, monitor->name, _countof(monitor->name));
- monitor->rect = *rect;
- monitor->handle = handle;
- }
- }
- return !match;
- }
- static void log_settings(struct duplicator_capture *capture, const char *monitor, LONG width, LONG height)
- {
- info("update settings:\n"
- "\tdisplay: %s (%ldx%ld)\n"
- "\tcursor: %s\n"
- "\tmethod: %s\n"
- "\tid: %s\n"
- "\talt_id: %s\n"
- "\tsetting_id: %s\n"
- "\tforce SDR: %s",
- monitor, width, height, capture->capture_cursor ? "true" : "false", get_method_name(capture->method),
- capture->id, capture->alt_id, capture->monitor_id, capture->force_sdr ? "true" : "false");
- }
- static enum display_capture_method choose_method(enum display_capture_method method, bool wgc_supported,
- HMONITOR monitor)
- {
- if (!wgc_supported)
- method = METHOD_DXGI;
- if (method == METHOD_AUTO) {
- method = METHOD_DXGI;
- obs_enter_graphics();
- const int dxgi_index = gs_duplicator_get_monitor_index(monitor);
- obs_leave_graphics();
- if (dxgi_index == -1) {
- method = METHOD_WGC;
- } else {
- SYSTEM_POWER_STATUS status;
- if (GetSystemPowerStatus(&status) && status.BatteryFlag < 128) {
- obs_enter_graphics();
- const uint32_t count = gs_get_adapter_count();
- obs_leave_graphics();
- if (count >= 2)
- method = METHOD_WGC;
- }
- }
- }
- return method;
- }
- extern bool wgc_supported;
- static struct duplicator_monitor_info find_monitor(const char *monitor_id)
- {
- struct duplicator_monitor_info monitor = {0};
- strcpy_s(monitor.device_id, _countof(monitor.device_id), monitor_id);
- EnumDisplayMonitors(NULL, NULL, &enum_monitor, (LPARAM)&monitor);
- if (monitor.handle == NULL) {
- EnumDisplayMonitors(NULL, NULL, &enum_monitor_fallback, (LPARAM)&monitor);
- }
- return monitor;
- }
- static inline void update_settings(struct duplicator_capture *capture, obs_data_t *settings)
- {
- pthread_mutex_lock(&capture->update_mutex);
- struct duplicator_monitor_info monitor = find_monitor(obs_data_get_string(settings, "monitor_id"));
- capture->method = choose_method((int)obs_data_get_int(settings, "method"), wgc_supported, monitor.handle);
- strcpy_s(capture->monitor_id, _countof(capture->monitor_id), monitor.device_id);
- strcpy_s(capture->id, _countof(capture->id), monitor.id);
- strcpy_s(capture->alt_id, _countof(capture->alt_id), monitor.alt_id);
- strcpy_s(capture->monitor_name, _countof(capture->monitor_name), monitor.name);
- capture->handle = monitor.handle;
- capture->capture_cursor = obs_data_get_bool(settings, "capture_cursor");
- capture->force_sdr = obs_data_get_bool(settings, "force_sdr");
- capture->logged_width = monitor.rect.right - monitor.rect.left;
- capture->logged_height = monitor.rect.bottom - monitor.rect.top;
- if (capture->duplicator) {
- obs_enter_graphics();
- gs_duplicator_destroy(capture->duplicator);
- capture->duplicator = NULL;
- obs_leave_graphics();
- }
- capture->width = 0;
- capture->height = 0;
- capture->x = 0;
- capture->y = 0;
- capture->rot = 0;
- capture->reset_timeout = RESET_INTERVAL_SEC;
- pthread_mutex_unlock(&capture->update_mutex);
- }
- /* ------------------------------------------------------------------------- */
- static const char *duplicator_capture_getname(void *unused)
- {
- UNUSED_PARAMETER(unused);
- return TEXT_MONITOR_CAPTURE;
- }
- static void duplicator_actual_destroy(void *data)
- {
- struct duplicator_capture *capture = data;
- if (capture->capture_winrt) {
- capture->exports.winrt_capture_free(capture->capture_winrt);
- capture->capture_winrt = NULL;
- }
- obs_enter_graphics();
- if (capture->duplicator) {
- gs_duplicator_destroy(capture->duplicator);
- capture->duplicator = NULL;
- }
- cursor_data_free(&capture->cursor_data);
- obs_leave_graphics();
- if (capture->winrt_module) {
- os_dlclose(capture->winrt_module);
- capture->winrt_module = NULL;
- }
- pthread_mutex_destroy(&capture->update_mutex);
- bfree(capture);
- }
- static void duplicator_capture_destroy(void *data)
- {
- obs_queue_task(OBS_TASK_GRAPHICS, duplicator_actual_destroy, data, false);
- }
- static void duplicator_capture_defaults(obs_data_t *settings)
- {
- obs_data_set_default_int(settings, "method", METHOD_AUTO);
- obs_data_set_default_string(settings, "monitor_id", INVALID_DISPLAY);
- obs_data_set_default_int(settings, "monitor_wgc", 0);
- obs_data_set_default_bool(settings, "capture_cursor", true);
- obs_data_set_default_bool(settings, "force_sdr", false);
- }
- static void duplicator_capture_update(void *data, obs_data_t *settings)
- {
- struct duplicator_capture *mc = data;
- update_settings(mc, settings);
- log_settings(mc, mc->monitor_name, mc->logged_width, mc->logged_height);
- mc->reset_wgc = true;
- }
- #define WINRT_IMPORT(func) \
- do { \
- exports->func = (PFN_##func)os_dlsym(module, #func); \
- if (!exports->func) { \
- success = false; \
- blog(LOG_ERROR, \
- "Could not load function '%s' from " \
- "module '%s'", \
- #func, module_name); \
- } \
- } while (false)
- static bool load_winrt_imports(struct winrt_exports *exports, void *module, const char *module_name)
- {
- bool success = true;
- WINRT_IMPORT(winrt_capture_supported);
- WINRT_IMPORT(winrt_capture_cursor_toggle_supported);
- WINRT_IMPORT(winrt_capture_init_monitor);
- WINRT_IMPORT(winrt_capture_free);
- WINRT_IMPORT(winrt_capture_active);
- WINRT_IMPORT(winrt_capture_get_color_space);
- WINRT_IMPORT(winrt_capture_render);
- WINRT_IMPORT(winrt_capture_width);
- WINRT_IMPORT(winrt_capture_height);
- return success;
- }
- extern bool graphics_uses_d3d11;
- static void *duplicator_capture_create(obs_data_t *settings, obs_source_t *source)
- {
- struct duplicator_capture *capture;
- capture = bzalloc(sizeof(struct duplicator_capture));
- capture->source = source;
- pthread_mutex_init(&capture->update_mutex, NULL);
- if (graphics_uses_d3d11) {
- static const char *const module = "libobs-winrt";
- capture->winrt_module = os_dlopen(module);
- if (capture->winrt_module) {
- load_winrt_imports(&capture->exports, capture->winrt_module, module);
- }
- }
- update_settings(capture, settings);
- log_settings(capture, capture->monitor_name, capture->logged_width, capture->logged_height);
- return capture;
- }
- static void reset_capture_data(struct duplicator_capture *capture)
- {
- struct gs_monitor_info monitor_info = {0};
- gs_texture_t *texture = gs_duplicator_get_texture(capture->duplicator);
- const int dxgi_index = gs_duplicator_get_monitor_index(capture->handle);
- gs_get_duplicator_monitor_info(dxgi_index, &monitor_info);
- if (texture) {
- capture->width = gs_texture_get_width(texture);
- capture->height = gs_texture_get_height(texture);
- } else {
- capture->width = 0;
- capture->height = 0;
- }
- capture->x = monitor_info.x;
- capture->y = monitor_info.y;
- capture->rot = monitor_info.rotation_degrees;
- }
- static void free_capture_data(struct duplicator_capture *capture)
- {
- if (capture->capture_winrt) {
- capture->exports.winrt_capture_free(capture->capture_winrt);
- capture->capture_winrt = NULL;
- }
- if (capture->duplicator) {
- gs_duplicator_destroy(capture->duplicator);
- capture->duplicator = NULL;
- }
- cursor_data_free(&capture->cursor_data);
- capture->width = 0;
- capture->height = 0;
- capture->x = 0;
- capture->y = 0;
- capture->rot = 0;
- capture->reset_timeout = 0.0f;
- }
- static void update_monitor_handle(struct duplicator_capture *capture)
- {
- capture->handle = find_monitor(capture->monitor_id).handle;
- }
- static void duplicator_capture_tick(void *data, float seconds)
- {
- struct duplicator_capture *capture = data;
- /* completely shut down monitor capture if not in use, otherwise it can
- * sometimes generate system lag when a game is in fullscreen mode */
- if (!obs_source_showing(capture->source)) {
- if (capture->showing) {
- obs_enter_graphics();
- free_capture_data(capture);
- obs_leave_graphics();
- capture->showing = false;
- }
- return;
- }
- /* always try to load the capture immediately when the source is first
- * shown */
- if (!capture->showing) {
- capture->reset_timeout = RESET_INTERVAL_SEC;
- }
- obs_enter_graphics();
- if (capture->method == METHOD_WGC) {
- if (capture->reset_wgc && capture->capture_winrt) {
- capture->exports.winrt_capture_free(capture->capture_winrt);
- capture->capture_winrt = NULL;
- capture->reset_wgc = false;
- capture->reset_timeout = RESET_INTERVAL_SEC;
- }
- if (!capture->capture_winrt) {
- capture->reset_timeout += seconds;
- if (capture->reset_timeout >= RESET_INTERVAL_SEC) {
- if (!capture->handle)
- update_monitor_handle(capture);
- if (capture->handle) {
- capture->capture_winrt = capture->exports.winrt_capture_init_monitor(
- capture->capture_cursor, capture->handle, capture->force_sdr);
- if (!capture->capture_winrt) {
- update_monitor_handle(capture);
- if (capture->handle) {
- capture->capture_winrt =
- capture->exports.winrt_capture_init_monitor(
- capture->capture_cursor, capture->handle,
- capture->force_sdr);
- }
- }
- }
- capture->reset_timeout = 0.0f;
- }
- }
- } else {
- if (capture->capture_winrt) {
- capture->exports.winrt_capture_free(capture->capture_winrt);
- capture->capture_winrt = NULL;
- }
- if (!capture->duplicator) {
- capture->reset_timeout += seconds;
- if (capture->reset_timeout >= RESET_INTERVAL_SEC) {
- if (!capture->handle)
- update_monitor_handle(capture);
- if (capture->handle) {
- int dxgi_index = gs_duplicator_get_monitor_index(capture->handle);
- if (dxgi_index == -1) {
- update_monitor_handle(capture);
- if (capture->handle) {
- dxgi_index = gs_duplicator_get_monitor_index(capture->handle);
- }
- }
- if (dxgi_index != -1) {
- capture->duplicator = gs_duplicator_create(dxgi_index);
- }
- }
- capture->reset_timeout = 0.0f;
- }
- }
- if (capture->duplicator) {
- if (capture->capture_cursor)
- cursor_capture(&capture->cursor_data);
- if (!gs_duplicator_update_frame(capture->duplicator)) {
- free_capture_data(capture);
- } else if (capture->width == 0) {
- reset_capture_data(capture);
- }
- }
- }
- obs_leave_graphics();
- if (!capture->showing)
- capture->showing = true;
- }
- static uint32_t duplicator_capture_width(void *data)
- {
- struct duplicator_capture *capture = data;
- return (capture->method == METHOD_WGC) ? capture->exports.winrt_capture_width(capture->capture_winrt)
- : (capture->rot % 180 == 0 ? capture->width : capture->height);
- }
- static uint32_t duplicator_capture_height(void *data)
- {
- struct duplicator_capture *capture = data;
- return (capture->method == METHOD_WGC) ? capture->exports.winrt_capture_height(capture->capture_winrt)
- : (capture->rot % 180 == 0 ? capture->height : capture->width);
- }
- static void draw_cursor(struct duplicator_capture *capture)
- {
- cursor_draw(&capture->cursor_data, -capture->x, -capture->y,
- capture->rot % 180 == 0 ? capture->width : capture->height,
- capture->rot % 180 == 0 ? capture->height : capture->width);
- }
- static void duplicator_capture_render(void *data, gs_effect_t *unused)
- {
- UNUSED_PARAMETER(unused);
- struct duplicator_capture *capture = data;
- if (capture->method == METHOD_WGC) {
- if (capture->capture_winrt) {
- if (capture->exports.winrt_capture_active(capture->capture_winrt)) {
- capture->exports.winrt_capture_render(capture->capture_winrt);
- } else {
- capture->exports.winrt_capture_free(capture->capture_winrt);
- capture->capture_winrt = NULL;
- }
- }
- } else {
- if (!capture->duplicator)
- return;
- gs_texture_t *const texture = gs_duplicator_get_texture(capture->duplicator);
- if (!texture)
- return;
- const bool previous = gs_framebuffer_srgb_enabled();
- gs_enable_framebuffer_srgb(true);
- gs_enable_blending(false);
- const int rot = capture->rot;
- if (rot != 0) {
- float x = 0.0f;
- float y = 0.0f;
- switch (rot) {
- case 90:
- x = (float)capture->height;
- break;
- case 180:
- x = (float)capture->width;
- y = (float)capture->height;
- break;
- case 270:
- y = (float)capture->width;
- break;
- }
- gs_matrix_push();
- gs_matrix_translate3f(x, y, 0.0f);
- gs_matrix_rotaa4f(0.0f, 0.0f, 1.0f, RAD((float)rot));
- }
- const char *tech_name = "Draw";
- float multiplier = 1.f;
- const enum gs_color_space current_space = gs_get_color_space();
- if (gs_duplicator_get_color_space(capture->duplicator) == GS_CS_709_SCRGB) {
- if (capture->force_sdr) {
- tech_name = "DrawMultiply";
- const float target_nits =
- (current_space == GS_CS_709_SCRGB) ? obs_get_video_sdr_white_level() : 80.f;
- multiplier = target_nits / gs_duplicator_get_sdr_white_level(capture->duplicator);
- } else {
- switch (current_space) {
- case GS_CS_SRGB:
- case GS_CS_SRGB_16F:
- tech_name = "DrawMultiplyTonemap";
- multiplier = 80.f / obs_get_video_sdr_white_level();
- break;
- case GS_CS_709_EXTENDED:
- tech_name = "DrawMultiply";
- multiplier = 80.f / obs_get_video_sdr_white_level();
- }
- }
- } else if (current_space == GS_CS_709_SCRGB) {
- tech_name = "DrawMultiply";
- multiplier = obs_get_video_sdr_white_level() / 80.f;
- }
- gs_effect_t *const opaque_effect = obs_get_base_effect(OBS_EFFECT_OPAQUE);
- gs_eparam_t *multiplier_param = gs_effect_get_param_by_name(opaque_effect, "multiplier");
- gs_effect_set_float(multiplier_param, multiplier);
- gs_eparam_t *image_param = gs_effect_get_param_by_name(opaque_effect, "image");
- gs_effect_set_texture_srgb(image_param, texture);
- while (gs_effect_loop(opaque_effect, tech_name)) {
- gs_draw_sprite(texture, 0, 0, 0);
- }
- if (rot != 0)
- gs_matrix_pop();
- gs_enable_blending(true);
- gs_enable_framebuffer_srgb(previous);
- if (capture->capture_cursor) {
- gs_effect_t *const default_effect = obs_get_base_effect(OBS_EFFECT_DEFAULT);
- while (gs_effect_loop(default_effect, "Draw")) {
- draw_cursor(capture);
- }
- }
- }
- }
- static BOOL CALLBACK enum_monitor_props(HMONITOR handle, HDC hdc, LPRECT rect, LPARAM param)
- {
- UNUSED_PARAMETER(hdc);
- UNUSED_PARAMETER(rect);
- char monitor_name[64];
- GetMonitorName(handle, monitor_name, sizeof(monitor_name));
- MONITORINFOEXA mi;
- mi.cbSize = sizeof(mi);
- if (GetMonitorInfoA(handle, (LPMONITORINFO)&mi)) {
- obs_property_t *monitor_list = (obs_property_t *)param;
- struct dstr monitor_desc = {0};
- dstr_printf(&monitor_desc, "%s: %dx%d @ %d,%d", monitor_name, mi.rcMonitor.right - mi.rcMonitor.left,
- mi.rcMonitor.bottom - mi.rcMonitor.top, mi.rcMonitor.left, mi.rcMonitor.top);
- if (mi.dwFlags == MONITORINFOF_PRIMARY)
- dstr_catf(&monitor_desc, " (%s)", TEXT_PRIMARY_MONITOR);
- DISPLAY_DEVICEA device;
- device.cb = sizeof(device);
- if (EnumDisplayDevicesA(mi.szDevice, 0, &device, EDD_GET_DEVICE_INTERFACE_NAME)) {
- obs_property_list_add_string(monitor_list, monitor_desc.array, device.DeviceID);
- } else {
- blog(LOG_WARNING,
- "[duplicator-monitor-capture] EnumDisplayDevices failed for monitor (%s), falling back to szDevice",
- monitor_name);
- obs_property_list_add_string(monitor_list, monitor_desc.array, mi.szDevice);
- }
- dstr_free(&monitor_desc);
- } else {
- blog(LOG_WARNING, "[duplicator-monitor-capture] GetMonitorInfo failed for monitor: %s", monitor_name);
- }
- return TRUE;
- }
- static void update_settings_visibility(obs_properties_t *props, struct duplicator_capture *capture)
- {
- pthread_mutex_lock(&capture->update_mutex);
- const enum display_capture_method method = capture->method;
- const bool dxgi_options = method == METHOD_DXGI;
- const bool wgc_options = method == METHOD_WGC;
- const bool wgc_cursor_toggle = wgc_options && capture->exports.winrt_capture_cursor_toggle_supported();
- obs_property_t *p = obs_properties_get(props, "cursor");
- obs_property_set_visible(p, dxgi_options || wgc_cursor_toggle);
- pthread_mutex_unlock(&capture->update_mutex);
- }
- static bool display_capture_method_changed(obs_properties_t *props, obs_property_t *p, obs_data_t *settings)
- {
- UNUSED_PARAMETER(p);
- struct duplicator_capture *capture = obs_properties_get_param(props);
- if (!capture)
- return false;
- update_settings(capture, settings);
- update_settings_visibility(props, capture);
- return true;
- }
- static obs_properties_t *duplicator_capture_properties(void *data)
- {
- struct duplicator_capture *capture = data;
- obs_properties_t *props = obs_properties_create();
- obs_properties_set_param(props, capture, NULL);
- obs_property_t *p =
- obs_properties_add_list(props, "method", TEXT_METHOD, OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_INT);
- obs_property_list_add_int(p, TEXT_METHOD_AUTO, METHOD_AUTO);
- obs_property_list_add_int(p, TEXT_METHOD_DXGI, METHOD_DXGI);
- obs_property_list_add_int(p, TEXT_METHOD_WGC, METHOD_WGC);
- obs_property_list_item_disable(p, 2, !wgc_supported);
- obs_property_set_modified_callback(p, display_capture_method_changed);
- obs_property_t *monitors = obs_properties_add_list(props, "monitor_id", TEXT_MONITOR, OBS_COMBO_TYPE_LIST,
- OBS_COMBO_FORMAT_STRING);
- if (capture && strcmp(capture->monitor_id, INVALID_DISPLAY) == 0) {
- obs_property_list_add_string(monitors, obs_module_text("SelectADisplay"), INVALID_DISPLAY);
- obs_property_list_item_disable(monitors, 0, true);
- }
- obs_properties_add_bool(props, "capture_cursor", TEXT_CAPTURE_CURSOR);
- obs_properties_add_bool(props, "force_sdr", TEXT_FORCE_SDR);
- EnumDisplayMonitors(NULL, NULL, enum_monitor_props, (LPARAM)monitors);
- return props;
- }
- enum gs_color_space duplicator_capture_get_color_space(void *data, size_t count,
- const enum gs_color_space *preferred_spaces)
- {
- enum gs_color_space capture_space = GS_CS_SRGB;
- struct duplicator_capture *capture = data;
- if (capture->method == METHOD_WGC) {
- if (capture->capture_winrt) {
- capture_space = capture->exports.winrt_capture_get_color_space(capture->capture_winrt);
- }
- } else if (capture->duplicator && !capture->force_sdr) {
- capture_space = gs_duplicator_get_color_space(capture->duplicator);
- }
- enum gs_color_space space = capture_space;
- for (size_t i = 0; i < count; ++i) {
- const enum gs_color_space preferred_space = preferred_spaces[i];
- space = preferred_space;
- if (preferred_space == capture_space)
- break;
- }
- return space;
- }
- struct obs_source_info duplicator_capture_info = {
- .id = "monitor_capture",
- .type = OBS_SOURCE_TYPE_INPUT,
- .output_flags = OBS_SOURCE_VIDEO | OBS_SOURCE_CUSTOM_DRAW | OBS_SOURCE_DO_NOT_DUPLICATE | OBS_SOURCE_SRGB,
- .get_name = duplicator_capture_getname,
- .create = duplicator_capture_create,
- .destroy = duplicator_capture_destroy,
- .video_render = duplicator_capture_render,
- .video_tick = duplicator_capture_tick,
- .update = duplicator_capture_update,
- .get_width = duplicator_capture_width,
- .get_height = duplicator_capture_height,
- .get_defaults = duplicator_capture_defaults,
- .get_properties = duplicator_capture_properties,
- .icon_type = OBS_ICON_TYPE_DESKTOP_CAPTURE,
- .video_get_color_space = duplicator_capture_get_color_space,
- };
|