---
 libavcodec/libx265.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/libavcodec/libx265.c b/libavcodec/libx265.c
index c04299a..36d8897 100644
--- a/libavcodec/libx265.c
+++ b/libavcodec/libx265.c
@@ -124,6 +124,21 @@ static av_cold int libx265_encode_init(AVCodecContext 
*avctx)
     ctx->params->sourceHeight    = avctx->height;
     ctx->params->bEnablePsnr     = !!(avctx->flags & CODEC_FLAG_PSNR);
 
+    if ((avctx->color_primaries <= AVCOL_PRI_BT2020 &&
+         avctx->color_primaries != AVCOL_PRI_UNSPECIFIED) ||
+        (avctx->color_trc <= AVCOL_TRC_BT2020_12 &&
+         avctx->color_trc != AVCOL_TRC_UNSPECIFIED) ||
+        (avctx->colorspace <= AVCOL_SPC_BT2020_CL &&
+         avctx->colorspace != AVCOL_SPC_UNSPECIFIED)) {
+         ctx->params->vui.bEnableColorDescriptionPresentFlag = 1;
+        if (avctx->color_primaries <= AVCOL_PRI_BT2020)
+            ctx->params->vui.colorPrimaries          = avctx->color_primaries;
+        if (avctx->color_trc <= AVCOL_TRC_BT2020_12)
+            ctx->params->vui.transferCharacteristics = avctx->color_trc;
+        if (avctx->colorspace <= AVCOL_SPC_BT2020_CL)
+            ctx->params->vui.matrixCoeffs            = avctx->colorspace;
+    }
+
     if (avctx->sample_aspect_ratio.num > 0 && avctx->sample_aspect_ratio.den > 
0) {
         char sar[12];
         int sar_num, sar_den;
-- 
1.9.0

_______________________________________________
libav-devel mailing list
[email protected]
https://lists.libav.org/mailman/listinfo/libav-devel

Reply via email to