[1/9] avcodec/bsf: Add ff_bsf_get_packet_ref() function

Message ID 20181003191538.33448-2-lu_zero@gentoo.org
State Committed
Commit 87d5686151bcb2ab94bd1c962e94d7e41203e31a
Headers show
Series
  • [1/9] avcodec/bsf: Add ff_bsf_get_packet_ref() function
Related show

Commit Message

Luca Barbato Oct. 3, 2018, 7:15 p.m.
From: Jan Sebechlebsky <sebechlebskyjan@gmail.com>

Use of this function can save unnecessary malloc operation
in bitstream filter.

Signed-off-by: Jan Sebechlebsky <sebechlebskyjan@gmail.com>
Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
Signed-off-by: Luca Barbato <lu_zero@gentoo.org>
---
 libavcodec/bsf.c | 16 ++++++++++++++++
 libavcodec/bsf.h | 11 +++++++++++
 2 files changed, 27 insertions(+)

Patch

diff --git a/libavcodec/bsf.c b/libavcodec/bsf.c
index 05cad546de..f6f894b50a 100644
--- a/libavcodec/bsf.c
+++ b/libavcodec/bsf.c
@@ -227,3 +227,19 @@  int ff_bsf_get_packet(AVBSFContext *ctx, AVPacket **pkt)
 
     return 0;
 }
+
+int ff_bsf_get_packet_ref(AVBSFContext *ctx, AVPacket *pkt)
+{
+    AVBSFInternal *in = ctx->internal;
+
+    if (in->eof)
+        return AVERROR_EOF;
+
+    if (!ctx->internal->buffer_pkt->data &&
+        !ctx->internal->buffer_pkt->side_data_elems)
+        return AVERROR(EAGAIN);
+
+    av_packet_move_ref(pkt, ctx->internal->buffer_pkt);
+
+    return 0;
+}
diff --git a/libavcodec/bsf.h b/libavcodec/bsf.h
index cf35fc8e01..39301a286f 100644
--- a/libavcodec/bsf.h
+++ b/libavcodec/bsf.h
@@ -28,6 +28,17 @@ 
  */
 int ff_bsf_get_packet(AVBSFContext *ctx, AVPacket **pkt);
 
+/**
+ * Called by bitstream filters to get packet for filtering.
+ * The reference to packet is moved to provided packet structure.
+ *
+ * @param ctx pointer to AVBSFContext of filter
+ * @param pkt pointer to packet to move reference to
+ *
+ * @return 0>= on success, negative AVERROR in case of failure
+ */
+int ff_bsf_get_packet_ref(AVBSFContext *ctx, AVPacket *pkt);
+
 const AVClass *ff_bsf_child_class_next(const AVClass *prev);
 
 #endif /* AVCODEC_BSF_H */