On 4/13/2021 9:14 PM, James Almer wrote:
Should fix ticket #9180

Signed-off-by: James Almer <jamr...@gmail.com>
---
  libavcodec/libaomdec.c | 14 ++++++++++++++
  1 file changed, 14 insertions(+)

diff --git a/libavcodec/libaomdec.c b/libavcodec/libaomdec.c
index 1fc0a0001d..6de3bcc5c3 100644
--- a/libavcodec/libaomdec.c
+++ b/libavcodec/libaomdec.c
@@ -161,6 +161,7 @@ static int aom_decode(AVCodecContext *avctx, void *data, 
int *got_frame,
      AVFrame *picture      = data;
      const void *iter      = NULL;
      struct aom_image *img;
+    aom_codec_frame_flags_t av_unused flags;
      int ret;
if (aom_codec_decode(&ctx->decoder, avpkt->data, avpkt->size, NULL) !=
@@ -198,6 +199,19 @@ static int aom_decode(AVCodecContext *avctx, void *data, 
int *got_frame,
          if ((ret = ff_get_buffer(avctx, picture, 0)) < 0)
              return ret;
+#ifdef AOM_CTRL_AOMD_GET_FRAME_FLAGS
+        ret = aom_codec_control(&ctx->decoder, AOMD_GET_FRAME_FLAGS, &flags);
+        if (ret == AOM_CODEC_OK) {
+            picture->key_frame = !!(flags & AOM_FRAME_IS_KEY);
+            if (flags & (AOM_FRAME_IS_KEY | AOM_FRAME_IS_INTRAONLY))
+                picture->pict_type = AV_PICTURE_TYPE_I;
+            else if (flags & AOM_FRAME_IS_SWITCH)
+                picture->pict_type = AV_PICTURE_TYPE_SP;
+            else
+                picture->pict_type = AV_PICTURE_TYPE_P;
+        }
+#endif
+
          av_reduce(&picture->sample_aspect_ratio.num,
                    &picture->sample_aspect_ratio.den,
                    picture->height * img->r_w,

Will apply.
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".

Reply via email to