[FFmpeg-devel] [PATCH 2/2] avcodec/mediacodecenc: pass colorspace info to encoder

[Zhao Zhili]

From: Zhao Zhili <zhilizhao at tencent.com>

Signed-off-by: Zhao Zhili <zhilizhao at tencent.com>
---
 libavcodec/mediacodecenc.c | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/libavcodec/mediacodecenc.c b/libavcodec/mediacodecenc.c
index a92a8dc5a9..2ab56597fe 100644
--- a/libavcodec/mediacodecenc.c
+++ b/libavcodec/mediacodecenc.c
@@ -243,6 +243,16 @@ static av_cold int mediacodec_init(AVCodecContext *avctx)
         }
     }
 
+    ret = ff_AMediaFormatColorRange_from_AVColorRange(avctx->color_range);
+    if (ret != COLOR_RANGE_UNSPECIFIED)
+        ff_AMediaFormat_setInt32(format, "color-range", ret);
+    ret = ff_AMediaFormatColorStandard_from_AVColorSpace(avctx->colorspace);
+    if (ret != COLOR_STANDARD_UNSPECIFIED)
+        ff_AMediaFormat_setInt32(format, "color-standard", ret);
+    ret = ff_AMediaFormatColorTransfer_from_AVColorTransfer(avctx->color_trc);
+    if (ret != COLOR_TRANSFER_UNSPECIFIED)
+        ff_AMediaFormat_setInt32(format, "color-transfer", ret);
+
     if (avctx->bit_rate)
         ff_AMediaFormat_setInt32(format, "bitrate", avctx->bit_rate);
     if (s->bitrate_mode >= 0)
-- 
2.25.1