avcodec/vp9_raw_reorder: reset state when flushing
authorJames Almer <jamrial@gmail.com>
Mon, 6 May 2019 03:22:32 +0000 (00:22 -0300)
committerJames Almer <jamrial@gmail.com>
Mon, 6 May 2019 03:22:32 +0000 (00:22 -0300)
Signed-off-by: James Almer <jamrial@gmail.com>
libavcodec/vp9_raw_reorder_bsf.c

index 01f3dad..f19b4c7 100644 (file)
@@ -385,6 +385,16 @@ fail:
     return err;
 }
 
+static void vp9_raw_reorder_flush(AVBSFContext *bsf)
+{
+    VP9RawReorderContext *ctx = bsf->priv_data;
+
+    for (int s = 0; s < FRAME_SLOTS; s++)
+        vp9_raw_reorder_clear_slot(ctx, s);
+    ctx->next_frame = NULL;
+    ctx->sequence = 0;
+}
+
 static void vp9_raw_reorder_close(AVBSFContext *bsf)
 {
     VP9RawReorderContext *ctx = bsf->priv_data;
@@ -402,6 +412,7 @@ const AVBitStreamFilter ff_vp9_raw_reorder_bsf = {
     .name           = "vp9_raw_reorder",
     .priv_data_size = sizeof(VP9RawReorderContext),
     .close          = &vp9_raw_reorder_close,
+    .flush          = &vp9_raw_reorder_flush,
     .filter         = &vp9_raw_reorder_filter,
     .codec_ids      = vp9_raw_reorder_codec_ids,
 };