plugin: Add Force HDR option

Closes #199
nowrep · Mar 31, 2024 · 151ef9c · 151ef9c
1 parent 4eae471
commit 151ef9c
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/data/locale/cs-CZ.ini b/data/locale/cs-CZ.ini
@@ -5,3 +5,4 @@ CaptureWindow="Okno"
 CaptureAnyWindow="Zaznamenávat jakékoliv okno"
 CaptureAnyWindowExcept="Zaznamenávat jakékoliv okno kromě"
 AllowTransparency="Povolit průhlednost"
+ForceHDR="Vynutit HDR"
diff --git a/data/locale/en-US.ini b/data/locale/en-US.ini
@@ -5,3 +5,4 @@ CaptureWindow="Window"
 CaptureAnyWindow="Capture any window"
 CaptureAnyWindowExcept="Capture any window except"
 AllowTransparency="Allow Transparency"
+ForceHDR="Force HDR"
diff --git a/src/vkcapture.c b/src/vkcapture.c
@@ -96,6 +96,7 @@ typedef struct {
 #endif
     bool show_cursor;
     bool allow_transparency;
+    bool force_hdr;
     bool window_match;
     bool window_exclude;
     const char *window;
@@ -323,6 +324,7 @@ static void vkcapture_source_update(void *data, obs_data_t *settings)
 
     ctx->show_cursor = obs_data_get_bool(settings, "show_cursor");
     ctx->allow_transparency = obs_data_get_bool(settings, "allow_transparency");
+    ctx->force_hdr = obs_data_get_bool(settings, "force_hdr");
 
     ctx->window_match = false;
     ctx->window_exclude = false;
@@ -618,6 +620,7 @@ static void vkcapture_source_get_defaults(obs_data_t *defaults)
 {
     obs_data_set_default_bool(defaults, "show_cursor", true);
     obs_data_set_default_bool(defaults, "allow_transparency", false);
+    obs_data_set_default_bool(defaults, "force_hdr", false);
 }
 
 static obs_properties_t *vkcapture_source_get_properties(void *data)
@@ -663,6 +666,7 @@ static obs_properties_t *vkcapture_source_get_properties(void *data)
     }
 
     obs_properties_add_bool(props, "allow_transparency", obs_module_text("AllowTransparency"));
+    obs_properties_add_bool(props, "force_hdr", obs_module_text("ForceHDR"));
 
     return props;
 }
@@ -673,6 +677,10 @@ enum gs_color_space vkcapture_get_color_space(void *data, size_t count, const en
 
     enum gs_color_space color_space = ctx->tdata.color_space;
 
+    if (ctx->force_hdr) {
+        color_space = GS_CS_709_EXTENDED;
+    }
+
     UNUSED_PARAMETER(count);
     UNUSED_PARAMETER(preferred_spaces);
     return color_space;