From ed02c3541eaf6c61127669882168f7e8b17e3af0 Mon Sep 17 00:00:00 2001
From: jpark37 <jpark37@users.noreply.github.com>
Date: Sun, 13 Feb 2022 00:07:28 -0800
Subject: [PATCH] Toggle hardware tonemapper for Elgato 4K60 S+

---
 source/device.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/source/device.cpp b/source/device.cpp
index 0bf69ca..e78549f 100644
--- a/source/device.cpp
+++ b/source/device.cpp
@@ -136,6 +136,9 @@ void HDevice::Receive(bool isVideo, IMediaSample *sample)
 			const bool hdr = IsVendorVideoHDR(propertySet);
 			if (deviceHdrSignal != hdr) {
 				deviceHdrSignal = hdr;
+#ifdef ENABLE_HEVC
+				SetVendorVideoFormat(propertySet, hdr);
+#endif
 				videoConfig.reactivateCallback();
 				reactivatePending = true;
 				return;
@@ -409,6 +412,9 @@ bool HDevice::SetVideoConfig(VideoConfig *config)
 	ComPtr<IKsPropertySet> propertySet = ComQIPtr<IKsPropertySet>(filter);
 	if (propertySet) {
 		const bool hdr = IsVendorVideoHDR(propertySet);
+#ifdef ENABLE_HEVC
+		SetVendorVideoFormat(propertySet, hdr);
+#endif
 		deviceHdrSignal = hdr;
 	}