#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;
*_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;
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 { \
*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;
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;
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) {
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);
// 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[] = {