From d6680b35b4f8df6ef91fab69f475dbf599c4f9e1 Mon Sep 17 00:00:00 2001
From: Seungha Yang <seungha.yang@navercorp.com>
Date: Wed, 17 Jul 2019 22:42:10 +0900
Subject: x265enc: Specify colorimetry related VUI parameters

Set the colorimetry config for the information to be embedded in encodec bitstream.
---
 ext/x265/gstx265enc.c | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'ext/x265')

diff --git a/ext/x265/gstx265enc.c b/ext/x265/gstx265enc.c
index b790247ee..c8fbc688c 100644
--- a/ext/x265/gstx265enc.c
+++ b/ext/x265/gstx265enc.c
@@ -676,6 +676,23 @@ gst_x265_enc_init_encoder (GstX265Enc * encoder)
     encoder->x265param.vui.sarHeight = info->par_d;
   }
 
+  encoder->x265param.vui.bEnableVideoSignalTypePresentFlag = 1;
+  /* Unspecified video format (5) */
+  encoder->x265param.vui.videoFormat = 5;
+  if (info->colorimetry.range == GST_VIDEO_COLOR_RANGE_0_255) {
+    encoder->x265param.vui.bEnableVideoFullRangeFlag = 1;
+  } else {
+    encoder->x265param.vui.bEnableVideoFullRangeFlag = 0;
+  }
+
+  encoder->x265param.vui.bEnableColorDescriptionPresentFlag = 1;
+  encoder->x265param.vui.matrixCoeffs =
+      gst_video_color_matrix_to_iso (info->colorimetry.matrix);
+  encoder->x265param.vui.colorPrimaries =
+      gst_video_color_primaries_to_iso (info->colorimetry.primaries);
+  encoder->x265param.vui.transferCharacteristics =
+      gst_video_color_transfer_to_iso (info->colorimetry.transfer);
+
   if (encoder->qp != -1) {
     /* CQP */
     encoder->x265param.rc.qp = encoder->qp;
-- 
cgit v1.2.1