From 24ee9f8cf3736c5cd946eb6df8bd3189f8a62ce0 Mon Sep 17 00:00:00 2001
From: Carl Hetherington <cth@carlh.net>
Date: Tue, 14 Oct 2025 21:14:04 +0200
Subject: Fix MPEG2 DCP gamma (#3026).

---
 src/lib/mpeg2_encoder.cc | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

(limited to 'src/lib/mpeg2_encoder.cc')

diff --git a/src/lib/mpeg2_encoder.cc b/src/lib/mpeg2_encoder.cc
index 38388431d..c9ee6c491 100644
--- a/src/lib/mpeg2_encoder.cc
+++ b/src/lib/mpeg2_encoder.cc
@@ -22,6 +22,7 @@
 #include "mpeg2_encoder.h"
 #include "writer.h"
 #include <dcp/ffmpeg_image.h>
+#include <dcp/transfer_function.h>
 extern "C" {
 #include <libavutil/pixfmt.h>
 }
@@ -43,7 +44,24 @@ MPEG2Encoder::encode(shared_ptr<PlayerVideo> pv, dcpomatic::DCPTime time)
 {
 	VideoEncoder::encode(pv, time);
 
-	auto image = pv->image(force(AV_PIX_FMT_YUV420P), VideoRange::VIDEO, false);
+	shared_ptr<Image> image;
+
+	if (auto conversion = pv->colour_conversion()) {
+		if (!conversion->in()->about_equal(conversion->out_mpeg2(), 0.05)) {
+			image = pv->image([](AVPixelFormat) { return AV_PIX_FMT_RGB24; }, VideoRange::VIDEO, Image::Alignment::PADDED, false);
+			dcp::convert_rgb_gamma(image->data()[0], image->size(), image->stride()[0], *conversion);
+			image = image->convert_pixel_format(conversion->yuv_to_rgb(), AV_PIX_FMT_YUV420P, Image::Alignment::COMPACT, false);
+		}
+	}
+
+	if (!image) {
+		image = pv->image(
+			[](AVPixelFormat) { return AV_PIX_FMT_YUV420P; },
+			VideoRange::VIDEO,
+			Image::Alignment::COMPACT,
+			false
+			);
+	}
 
 	dcp::FFmpegImage ffmpeg_image(time.get() * _film->video_frame_rate() / dcpomatic::DCPTime::HZ);
 
-- 
cgit v1.2.3