Commit 08667c26 authored by Michael Niedermayer's avatar Michael Niedermayer

Merge commit '0b0a7a75'

* commit '0b0a7a75':
  mpegvideo: move encode-only parts of common_end() to encode_end()

Conflicts:
	libavcodec/mpegvideo.c
Merged-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parents e3b7f0e1 0b0a7a75
...@@ -285,7 +285,7 @@ static int alloc_frame_buffer(MpegEncContext *s, Picture *pic) ...@@ -285,7 +285,7 @@ static int alloc_frame_buffer(MpegEncContext *s, Picture *pic)
return 0; return 0;
} }
static void free_picture_tables(Picture *pic) void ff_free_picture_tables(Picture *pic)
{ {
int i; int i;
...@@ -382,7 +382,7 @@ int ff_alloc_picture(MpegEncContext *s, Picture *pic, int shared) ...@@ -382,7 +382,7 @@ int ff_alloc_picture(MpegEncContext *s, Picture *pic, int shared)
if (pic->qscale_table_buf) if (pic->qscale_table_buf)
if ( pic->alloc_mb_width != s->mb_width if ( pic->alloc_mb_width != s->mb_width
|| pic->alloc_mb_height != s->mb_height) || pic->alloc_mb_height != s->mb_height)
free_picture_tables(pic); ff_free_picture_tables(pic);
if (shared) { if (shared) {
av_assert0(pic->f.data[0]); av_assert0(pic->f.data[0]);
...@@ -425,7 +425,7 @@ int ff_alloc_picture(MpegEncContext *s, Picture *pic, int shared) ...@@ -425,7 +425,7 @@ int ff_alloc_picture(MpegEncContext *s, Picture *pic, int shared)
fail: fail:
av_log(s->avctx, AV_LOG_ERROR, "Error allocating a picture.\n"); av_log(s->avctx, AV_LOG_ERROR, "Error allocating a picture.\n");
ff_mpeg_unref_picture(s, pic); ff_mpeg_unref_picture(s, pic);
free_picture_tables(pic); ff_free_picture_tables(pic);
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
} }
...@@ -449,7 +449,7 @@ void ff_mpeg_unref_picture(MpegEncContext *s, Picture *pic) ...@@ -449,7 +449,7 @@ void ff_mpeg_unref_picture(MpegEncContext *s, Picture *pic)
av_buffer_unref(&pic->hwaccel_priv_buf); av_buffer_unref(&pic->hwaccel_priv_buf);
if (pic->needs_realloc) if (pic->needs_realloc)
free_picture_tables(pic); ff_free_picture_tables(pic);
memset((uint8_t*)pic + off, 0, sizeof(*pic) - off); memset((uint8_t*)pic + off, 0, sizeof(*pic) - off);
} }
...@@ -465,7 +465,7 @@ do {\ ...@@ -465,7 +465,7 @@ do {\
av_buffer_unref(&dst->table);\ av_buffer_unref(&dst->table);\
dst->table = av_buffer_ref(src->table);\ dst->table = av_buffer_ref(src->table);\
if (!dst->table) {\ if (!dst->table) {\
free_picture_tables(dst);\ ff_free_picture_tables(dst);\
return AVERROR(ENOMEM);\ return AVERROR(ENOMEM);\
}\ }\
}\ }\
...@@ -1244,36 +1244,19 @@ void ff_MPV_common_end(MpegEncContext *s) ...@@ -1244,36 +1244,19 @@ void ff_MPV_common_end(MpegEncContext *s)
av_freep(&s->bitstream_buffer); av_freep(&s->bitstream_buffer);
s->allocated_bitstream_buffer_size = 0; s->allocated_bitstream_buffer_size = 0;
av_freep(&s->avctx->stats_out);
av_freep(&s->ac_stats);
if(s->q_chroma_intra_matrix != s->q_intra_matrix ) av_freep(&s->q_chroma_intra_matrix);
if(s->q_chroma_intra_matrix16 != s->q_intra_matrix16) av_freep(&s->q_chroma_intra_matrix16);
s->q_chroma_intra_matrix= NULL;
s->q_chroma_intra_matrix16= NULL;
av_freep(&s->q_intra_matrix);
av_freep(&s->q_inter_matrix);
av_freep(&s->q_intra_matrix16);
av_freep(&s->q_inter_matrix16);
av_freep(&s->input_picture);
av_freep(&s->reordered_input_picture);
av_freep(&s->dct_offset);
if (s->picture) { if (s->picture) {
for (i = 0; i < MAX_PICTURE_COUNT; i++) { for (i = 0; i < MAX_PICTURE_COUNT; i++) {
free_picture_tables(&s->picture[i]); ff_free_picture_tables(&s->picture[i]);
ff_mpeg_unref_picture(s, &s->picture[i]); ff_mpeg_unref_picture(s, &s->picture[i]);
} }
} }
av_freep(&s->picture); av_freep(&s->picture);
free_picture_tables(&s->last_picture); ff_free_picture_tables(&s->last_picture);
ff_mpeg_unref_picture(s, &s->last_picture); ff_mpeg_unref_picture(s, &s->last_picture);
free_picture_tables(&s->current_picture); ff_free_picture_tables(&s->current_picture);
ff_mpeg_unref_picture(s, &s->current_picture); ff_mpeg_unref_picture(s, &s->current_picture);
free_picture_tables(&s->next_picture); ff_free_picture_tables(&s->next_picture);
ff_mpeg_unref_picture(s, &s->next_picture); ff_mpeg_unref_picture(s, &s->next_picture);
free_picture_tables(&s->new_picture);
ff_mpeg_unref_picture(s, &s->new_picture);
free_context_frame(s); free_context_frame(s);
...@@ -1440,7 +1423,7 @@ int ff_find_unused_picture(MpegEncContext *s, int shared) ...@@ -1440,7 +1423,7 @@ int ff_find_unused_picture(MpegEncContext *s, int shared)
if (ret >= 0 && ret < MAX_PICTURE_COUNT) { if (ret >= 0 && ret < MAX_PICTURE_COUNT) {
if (s->picture[ret].needs_realloc) { if (s->picture[ret].needs_realloc) {
s->picture[ret].needs_realloc = 0; s->picture[ret].needs_realloc = 0;
free_picture_tables(&s->picture[ret]); ff_free_picture_tables(&s->picture[ret]);
ff_mpeg_unref_picture(s, &s->picture[ret]); ff_mpeg_unref_picture(s, &s->picture[ret]);
avcodec_get_frame_defaults(&s->picture[ret].f); avcodec_get_frame_defaults(&s->picture[ret].f);
} }
......
...@@ -957,6 +957,7 @@ void ff_wmv2_encode_mb(MpegEncContext * s, ...@@ -957,6 +957,7 @@ void ff_wmv2_encode_mb(MpegEncContext * s,
int ff_mpeg_ref_picture(MpegEncContext *s, Picture *dst, Picture *src); int ff_mpeg_ref_picture(MpegEncContext *s, Picture *dst, Picture *src);
void ff_mpeg_unref_picture(MpegEncContext *s, Picture *picture); void ff_mpeg_unref_picture(MpegEncContext *s, Picture *picture);
void ff_free_picture_tables(Picture *pic);
#endif /* AVCODEC_MPEGVIDEO_H */ #endif /* AVCODEC_MPEGVIDEO_H */
...@@ -963,6 +963,24 @@ av_cold int ff_MPV_encode_end(AVCodecContext *avctx) ...@@ -963,6 +963,24 @@ av_cold int ff_MPV_encode_end(AVCodecContext *avctx)
for (i = 0; i < FF_ARRAY_ELEMS(s->tmp_frames); i++) for (i = 0; i < FF_ARRAY_ELEMS(s->tmp_frames); i++)
av_frame_free(&s->tmp_frames[i]); av_frame_free(&s->tmp_frames[i]);
ff_free_picture_tables(&s->new_picture);
ff_mpeg_unref_picture(s, &s->new_picture);
av_freep(&s->avctx->stats_out);
av_freep(&s->ac_stats);
if(s->q_chroma_intra_matrix != s->q_intra_matrix ) av_freep(&s->q_chroma_intra_matrix);
if(s->q_chroma_intra_matrix16 != s->q_intra_matrix16) av_freep(&s->q_chroma_intra_matrix16);
s->q_chroma_intra_matrix= NULL;
s->q_chroma_intra_matrix16= NULL;
av_freep(&s->q_intra_matrix);
av_freep(&s->q_inter_matrix);
av_freep(&s->q_intra_matrix16);
av_freep(&s->q_inter_matrix16);
av_freep(&s->input_picture);
av_freep(&s->reordered_input_picture);
av_freep(&s->dct_offset);
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