[FFmpeg-devel] [PATCH] avcodec/hevc: export chroma sample location

[Hendrik Leppkes]

---
 libavcodec/hevcdec.c | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/libavcodec/hevcdec.c b/libavcodec/hevcdec.c
index c8877626d2..e8227a510b 100644
--- a/libavcodec/hevcdec.c
+++ b/libavcodec/hevcdec.c
@@ -344,6 +344,17 @@ static void export_stream_params(AVCodecContext *avctx, const HEVCParamSets *ps,
         avctx->colorspace      = AVCOL_SPC_UNSPECIFIED;
     }
 
+    if (sps->chroma_format_idc == 1) {
+        if (sps->vui.chroma_loc_info_present_flag) {
+            if (sps->vui.chroma_sample_loc_type_top_field <= 5)
+                avctx->chroma_sample_location = sps->vui.chroma_sample_loc_type_top_field + 1;
+            else
+                avctx->chroma_sample_location = AVCHROMA_LOC_UNSPECIFIED;
+        } else
+            avctx->chroma_sample_location = AVCHROMA_LOC_LEFT;
+    } else
+        avctx->chroma_sample_location = AVCHROMA_LOC_UNSPECIFIED;
+
     if (vps->vps_timing_info_present_flag) {
         num = vps->vps_num_units_in_tick;
         den = vps->vps_time_scale;
-- 
2.16.1.windows.4