Commit e4155f15 authored by Anton Khirnov's avatar Anton Khirnov

eamad: use the AVFrame API properly.

parent 4e7e8055
...@@ -45,7 +45,7 @@ ...@@ -45,7 +45,7 @@
typedef struct MadContext { typedef struct MadContext {
AVCodecContext *avctx; AVCodecContext *avctx;
DSPContext dsp; DSPContext dsp;
AVFrame last_frame; AVFrame *last_frame;
GetBitContext gb; GetBitContext gb;
void *bitstream_buf; void *bitstream_buf;
unsigned int bitstream_buf_size; unsigned int bitstream_buf_size;
...@@ -65,6 +65,11 @@ static av_cold int decode_init(AVCodecContext *avctx) ...@@ -65,6 +65,11 @@ static av_cold int decode_init(AVCodecContext *avctx)
ff_init_scantable_permutation(s->dsp.idct_permutation, FF_NO_IDCT_PERM); ff_init_scantable_permutation(s->dsp.idct_permutation, FF_NO_IDCT_PERM);
ff_init_scantable(s->dsp.idct_permutation, &s->scantable, ff_zigzag_direct); ff_init_scantable(s->dsp.idct_permutation, &s->scantable, ff_zigzag_direct);
ff_mpeg12_init_vlcs(); ff_mpeg12_init_vlcs();
s->last_frame = av_frame_alloc();
if (!s->last_frame)
return AVERROR(ENOMEM);
return 0; return 0;
} }
...@@ -84,14 +89,14 @@ static inline void comp_block(MadContext *t, AVFrame *frame, ...@@ -84,14 +89,14 @@ static inline void comp_block(MadContext *t, AVFrame *frame,
if (j < 4) { if (j < 4) {
comp(frame->data[0] + (mb_y*16 + ((j&2)<<2))*frame->linesize[0] + mb_x*16 + ((j&1)<<3), comp(frame->data[0] + (mb_y*16 + ((j&2)<<2))*frame->linesize[0] + mb_x*16 + ((j&1)<<3),
frame->linesize[0], frame->linesize[0],
t->last_frame.data[0] + (mb_y*16 + ((j&2)<<2) + mv_y)*t->last_frame.linesize[0] + mb_x*16 + ((j&1)<<3) + mv_x, t->last_frame->data[0] + (mb_y*16 + ((j&2)<<2) + mv_y)*t->last_frame->linesize[0] + mb_x*16 + ((j&1)<<3) + mv_x,
t->last_frame.linesize[0], add); t->last_frame->linesize[0], add);
} else if (!(t->avctx->flags & CODEC_FLAG_GRAY)) { } else if (!(t->avctx->flags & CODEC_FLAG_GRAY)) {
int index = j - 3; int index = j - 3;
comp(frame->data[index] + (mb_y*8)*frame->linesize[index] + mb_x * 8, comp(frame->data[index] + (mb_y*8)*frame->linesize[index] + mb_x * 8,
frame->linesize[index], frame->linesize[index],
t->last_frame.data[index] + (mb_y * 8 + (mv_y/2))*t->last_frame.linesize[index] + mb_x * 8 + (mv_x/2), t->last_frame->data[index] + (mb_y * 8 + (mv_y/2))*t->last_frame->linesize[index] + mb_x * 8 + (mv_x/2),
t->last_frame.linesize[index], add); t->last_frame->linesize[index], add);
} }
} }
...@@ -250,7 +255,7 @@ static int decode_frame(AVCodecContext *avctx, ...@@ -250,7 +255,7 @@ static int decode_frame(AVCodecContext *avctx,
buf += 16; buf += 16;
if (avctx->width != width || avctx->height != height) { if (avctx->width != width || avctx->height != height) {
av_frame_unref(&s->last_frame); av_frame_unref(s->last_frame);
if ((ret = ff_set_dimensions(avctx, width, height)) < 0) if ((ret = ff_set_dimensions(avctx, width, height)) < 0)
return ret; return ret;
} }
...@@ -260,17 +265,17 @@ static int decode_frame(AVCodecContext *avctx, ...@@ -260,17 +265,17 @@ static int decode_frame(AVCodecContext *avctx,
return ret; return ret;
} }
if (inter && !s->last_frame.data[0]) { if (inter && !s->last_frame->data[0]) {
av_log(avctx, AV_LOG_WARNING, "Missing reference frame.\n"); av_log(avctx, AV_LOG_WARNING, "Missing reference frame.\n");
ret = ff_get_buffer(avctx, &s->last_frame, AV_GET_BUFFER_FLAG_REF); ret = ff_get_buffer(avctx, s->last_frame, AV_GET_BUFFER_FLAG_REF);
if (ret < 0) if (ret < 0)
return ret; return ret;
memset(s->last_frame.data[0], 0, s->last_frame.height * memset(s->last_frame->data[0], 0, s->last_frame->height *
s->last_frame.linesize[0]); s->last_frame->linesize[0]);
memset(s->last_frame.data[1], 0x80, s->last_frame.height / 2 * memset(s->last_frame->data[1], 0x80, s->last_frame->height / 2 *
s->last_frame.linesize[1]); s->last_frame->linesize[1]);
memset(s->last_frame.data[2], 0x80, s->last_frame.height / 2 * memset(s->last_frame->data[2], 0x80, s->last_frame->height / 2 *
s->last_frame.linesize[2]); s->last_frame->linesize[2]);
} }
av_fast_padded_malloc(&s->bitstream_buf, &s->bitstream_buf_size, av_fast_padded_malloc(&s->bitstream_buf, &s->bitstream_buf_size,
...@@ -287,8 +292,8 @@ static int decode_frame(AVCodecContext *avctx, ...@@ -287,8 +292,8 @@ static int decode_frame(AVCodecContext *avctx,
*got_frame = 1; *got_frame = 1;
if (chunk_type != MADe_TAG) { if (chunk_type != MADe_TAG) {
av_frame_unref(&s->last_frame); av_frame_unref(s->last_frame);
if ((ret = av_frame_ref(&s->last_frame, frame)) < 0) if ((ret = av_frame_ref(s->last_frame, frame)) < 0)
return ret; return ret;
} }
...@@ -298,7 +303,7 @@ static int decode_frame(AVCodecContext *avctx, ...@@ -298,7 +303,7 @@ static int decode_frame(AVCodecContext *avctx,
static av_cold int decode_end(AVCodecContext *avctx) static av_cold int decode_end(AVCodecContext *avctx)
{ {
MadContext *t = avctx->priv_data; MadContext *t = avctx->priv_data;
av_frame_unref(&t->last_frame); av_frame_free(&t->last_frame);
av_free(t->bitstream_buf); av_free(t->bitstream_buf);
return 0; return 0;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment