| @@ -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) | |||||
| 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, | |||||
| 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.linesize[0], add); | |||||
| 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); | |||||
| } 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.linesize[index], add); | |||||
| 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); | |||||
| } | } | ||||
| } | } | ||||
| @@ -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, | |||||
| 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 * | |||||
| s->last_frame.linesize[0]); | |||||
| memset(s->last_frame.data[1], 0x80, s->last_frame.height / 2 * | |||||
| s->last_frame.linesize[1]); | |||||
| memset(s->last_frame.data[2], 0x80, s->last_frame.height / 2 * | |||||
| s->last_frame.linesize[2]); | |||||
| memset(s->last_frame->data[0], 0, s->last_frame->height * | |||||
| s->last_frame->linesize[0]); | |||||
| memset(s->last_frame->data[1], 0x80, s->last_frame->height / 2 * | |||||
| s->last_frame->linesize[1]); | |||||
| memset(s->last_frame->data[2], 0x80, s->last_frame->height / 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, | |||||
| *got_frame = 1; | *got_frame = 1; | ||||
| if (chunk_type != MADe_TAG) { | if (chunk_type != MADe_TAG) { | ||||
| av_frame_unref(&s->last_frame); | |||||
| if ((ret = av_frame_ref(&s->last_frame, frame)) < 0) | |||||
| av_frame_unref(s->last_frame); | |||||
| if ((ret = av_frame_ref(s->last_frame, frame)) < 0) | |||||
| return ret; | return ret; | ||||
| } | } | ||||
| @@ -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; | ||||
| } | } | ||||