diff mbox series

[FFmpeg-devel,4/8] avcodec/cbs_vp9: implement a CodedBitstreamType.flush() callback

Message ID 20200925144318.6194-4-jamrial@gmail.com
State Accepted
Commit 421906dddb631420fe82c6248f14100bae19b8b8
Headers show
Series [FFmpeg-devel,1/8] avcodec/cbs: add a flush callback to CodedBitstreamType | expand

Checks

Context Check Description
andriy/default pending
andriy/make success Make finished
andriy/make_fate success Make fate finished

Commit Message

James Almer Sept. 25, 2020, 2:43 p.m. UTC 
Signed-off-by: James Almer <jamrial@gmail.com>
---
 libavcodec/cbs_vp9.c | 10 ++++++++++
 1 file changed, 10 insertions(+)
diff mbox series

Patch

diff --git a/libavcodec/cbs_vp9.c b/libavcodec/cbs_vp9.c
index 6a480b4ce3..e0b8c02ac2 100644
--- a/libavcodec/cbs_vp9.c
+++ b/libavcodec/cbs_vp9.c
@@ -634,6 +634,13 @@  static int cbs_vp9_assemble_fragment(CodedBitstreamContext *ctx,
     return 0;
 }
 
+static void cbs_vp9_flush(CodedBitstreamContext *ctx)
+{
+    CodedBitstreamVP9Context *vp9 = ctx->priv_data;
+
+    memset(vp9->ref, 0, sizeof(vp9->ref));
+}
+
 static const CodedBitstreamUnitTypeDescriptor cbs_vp9_unit_types[] = {
     CBS_UNIT_TYPE_INTERNAL_REF(0, VP9RawFrame, data),
     CBS_UNIT_TYPE_END_OF_LIST
@@ -649,5 +656,8 @@  const CodedBitstreamType ff_cbs_type_vp9 = {
     .split_fragment    = &cbs_vp9_split_fragment,
     .read_unit         = &cbs_vp9_read_unit,
     .write_unit        = &cbs_vp9_write_unit,
+
+    .flush             = &cbs_vp9_flush,
+
     .assemble_fragment = &cbs_vp9_assemble_fragment,
 };