qsv: Join the derived session to the parent

[ffmpeg] / libavcodec / vp9_superframe_bsf.c

diff --git a/libavcodec/vp9_superframe_bsf.c b/libavcodec/vp9_superframe_bsf.c
index be83ba3e75be306e90ef33f6b9df1f2c0c49cc40..3669216009d7a63d48ac0b86a07ff645238b35dc 100644 (file)
--- a/libavcodec/vp9_superframe_bsf.c
+++ b/libavcodec/vp9_superframe_bsf.c
@@ -21,26 +21,23 @@
 
 #include "libavutil/avassert.h"
 #include "avcodec.h"
+#include "bitstream.h"
 #include "bsf.h"
-#include "get_bits.h"
 
 #define MAX_CACHE 8
 typedef struct VP9BSFContext {
     int n_cache;
-    struct CachedBuf {
-        uint8_t *data;
-        int size;
-    } cache[MAX_CACHE];
+    AVPacket *cache[MAX_CACHE];
 } VP9BSFContext;
 
-static void stats(const struct CachedBuf *in, int n_in,
+static void stats(AVPacket * const *in, int n_in,
                   unsigned *_max, unsigned *_sum)
 {
     int n;
     unsigned max = 0, sum = 0;
 
     for (n = 0; n < n_in; n++) {
-        unsigned sz = in[n].size;
+        unsigned sz = in[n]->size;
 
         if (sz > max)
             max = sz;
@@ -51,7 +48,7 @@ static void stats(const struct CachedBuf *in, int n_in,
     *_sum = sum;
 }
 
-static int merge_superframe(const struct CachedBuf *in, int n_in, AVPacket *out)
+static int merge_superframe(AVPacket * const *in, int n_in, AVPacket *out)
 {
     unsigned max, sum, mag, marker, n, sz;
     uint8_t *ptr;
@@ -66,8 +63,8 @@ static int merge_superframe(const struct CachedBuf *in, int n_in, AVPacket *out)
         return res;
     ptr = out->data;
     for (n = 0; n < n_in; n++) {
-        memcpy(ptr, in[n].data, in[n].size);
-        ptr += in[n].size;
+        memcpy(ptr, in[n]->data, in[n]->size);
+        ptr += in[n]->size;
     }
 
 #define wloop(mag, wr) do { \
@@ -81,16 +78,16 @@ static int merge_superframe(const struct CachedBuf *in, int n_in, AVPacket *out)
     *ptr++ = marker;
     switch (mag) {
     case 0:
-        wloop(mag, *ptr = in[n].size);
+        wloop(mag, *ptr = in[n]->size);
         break;
     case 1:
-        wloop(mag, AV_WL16(ptr, in[n].size));
+        wloop(mag, AV_WL16(ptr, in[n]->size));
         break;
     case 2:
-        wloop(mag, AV_WL24(ptr, in[n].size));
+        wloop(mag, AV_WL24(ptr, in[n]->size));
         break;
     case 3:
-        wloop(mag, AV_WL32(ptr, in[n].size));
+        wloop(mag, AV_WL32(ptr, in[n]->size));
         break;
     }
     *ptr++ = marker;
@@ -101,7 +98,7 @@ static int merge_superframe(const struct CachedBuf *in, int n_in, AVPacket *out)
 
 static int vp9_superframe_filter(AVBSFContext *ctx, AVPacket *out)
 {
-    GetBitContext gb;
+    BitstreamContext bc;
     VP9BSFContext *s = ctx->priv_data;
     AVPacket *in;
     int res, invisible, profile, marker, uses_superframe_syntax = 0, n;
@@ -118,19 +115,21 @@ static int vp9_superframe_filter(AVBSFContext *ctx, AVPacket *out)
         uses_superframe_syntax = in->size >= idx_sz && in->data[in->size - idx_sz] == marker;
     }
 
-    if ((res = init_get_bits8(&gb, in->data, in->size)) < 0)
+    res = bitstream_init8(&bc, in->data, in->size);
+    if (res < 0)
         goto done;
 
-    get_bits(&gb, 2); // frame marker
-    profile  = get_bits1(&gb);
-    profile |= get_bits1(&gb) << 1;
-    if (profile == 3) profile += get_bits1(&gb);
+    bitstream_read(&bc, 2); // frame marker
+    profile  = bitstream_read(&bc, 1);
+    profile |= bitstream_read(&bc, 1) << 1;
+    if (profile == 3)
+        profile += bitstream_read(&bc, 1);
 
-    if (get_bits1(&gb)) {
+    if (bitstream_read(&bc, 1)) {
         invisible = 0;
     } else {
-        get_bits1(&gb); // keyframe
-        invisible = !get_bits1(&gb);
+        bitstream_read(&bc, 1); // keyframe
+        invisible = !bitstream_read(&bc, 1);
     }
 
     if (uses_superframe_syntax && s->n_cache > 0) {
@@ -149,34 +148,26 @@ static int vp9_superframe_filter(AVBSFContext *ctx, AVPacket *out)
         goto done;
     }
 
+    s->cache[s->n_cache++] = in;
+    in                     = NULL;
     if (invisible) {
-        s->cache[s->n_cache].data = av_malloc(in->size);
-        if (!s->cache[s->n_cache].data) {
-            res = AVERROR(ENOMEM);
-            goto done;
-        }
-        memcpy(s->cache[s->n_cache].data, in->data, in->size);
-        s->cache[s->n_cache++].size = in->size;
         res = AVERROR(EAGAIN);
         goto done;
     }
     av_assert0(s->n_cache > 0);
 
-    s->cache[s->n_cache].data = in->data;
-    s->cache[s->n_cache].size = in->size;
-
     // build superframe
-    if ((res = merge_superframe(s->cache, s->n_cache + 1, out)) < 0)
+    if ((res = merge_superframe(s->cache, s->n_cache, out)) < 0)
         goto done;
 
-    for (n = 0; n < s->n_cache; n++)
-        av_freep(&s->cache[n].data);
-    s->n_cache = 0;
-
-    res = av_packet_copy_props(out, in);
+    res = av_packet_copy_props(out, s->cache[s->n_cache - 1]);
     if (res < 0)
         goto done;
 
+    for (n = 0; n < s->n_cache; n++)
+        av_packet_free(&s->cache[n]);
+    s->n_cache = 0;
+
 done:
     if (res < 0)
         av_packet_unref(out);
@@ -191,7 +182,7 @@ static void vp9_superframe_close(AVBSFContext *ctx)
 
     // free cached data
     for (n = 0; n < s->n_cache; n++)
-        av_freep(&s->cache[n].data);
+        av_packet_free(&s->cache[n]);
 }
 
 static const enum AVCodecID codec_ids[] = {