/* * Mpeg video formats-related picture management functions * * This file is part of Libav. * * Libav is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * Libav is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with Libav; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include <stdint.h> #include "libavutil/avassert.h" #include "libavutil/common.h" #include "avcodec.h" #include "motion_est.h" #include "mpegpicture.h" #include "mpegutils.h" static int make_tables_writable(Picture *pic) { int ret, i; #define MAKE_WRITABLE(table) \ do {\ if (pic->table &&\ (ret = av_buffer_make_writable(&pic->table)) < 0)\ return ret;\ } while (0) MAKE_WRITABLE(mb_var_buf); MAKE_WRITABLE(mc_mb_var_buf); MAKE_WRITABLE(mb_mean_buf); MAKE_WRITABLE(mbskip_table_buf); MAKE_WRITABLE(qscale_table_buf); MAKE_WRITABLE(mb_type_buf); for (i = 0; i < 2; i++) { MAKE_WRITABLE(motion_val_buf[i]); MAKE_WRITABLE(ref_index_buf[i]); } return 0; } int ff_mpeg_framesize_alloc(AVCodecContext *avctx, MotionEstContext *me, ScratchpadContext *sc, int linesize) { int alloc_size = FFALIGN(FFABS(linesize) + 32, 32); // edge emu needs blocksize + filter length - 1 // (= 17x17 for halfpel / 21x21 for h264) // VC1 computes luma and chroma simultaneously and needs 19X19 + 9x9 // at uvlinesize. It supports only YUV420 so 24x24 is enough // linesize * interlaced * MBsize FF_ALLOCZ_OR_GOTO(avctx, sc->edge_emu_buffer, alloc_size * 2 * 24, fail); FF_ALLOCZ_OR_GOTO(avctx, me->scratchpad, alloc_size * 2 * 16 * 3, fail) me->temp = me->scratchpad; sc->rd_scratchpad = me->scratchpad; sc->b_scratchpad = me->scratchpad; sc->obmc_scratchpad = me->scratchpad + 16; return 0; fail: av_freep(&sc->edge_emu_buffer); return AVERROR(ENOMEM); } /** * Allocate a frame buffer */ static int alloc_frame_buffer(AVCodecContext *avctx, Picture *pic, MotionEstContext *me, ScratchpadContext *sc, int chroma_x_shift, int chroma_y_shift, int linesize, int uvlinesize) { int edges_needed = av_codec_is_encoder(avctx->codec); int r, ret; pic->tf.f = pic->f; if (avctx->codec_id != AV_CODEC_ID_WMV3IMAGE && avctx->codec_id != AV_CODEC_ID_VC1IMAGE && avctx->codec_id != AV_CODEC_ID_MSS2) { if (edges_needed) { pic->f->width = avctx->width + 2 * EDGE_WIDTH; pic->f->height = avctx->height + 2 * EDGE_WIDTH; } r = ff_thread_get_buffer(avctx, &pic->tf, pic->reference ? AV_GET_BUFFER_FLAG_REF : 0); } else { pic->f->width = avctx->width; pic->f->height = avctx->height; pic->f->format = avctx->pix_fmt; r = avcodec_default_get_buffer2(avctx, pic->f, 0); } if (r < 0 || !pic->f->buf[0]) { av_log(avctx, AV_LOG_ERROR, "get_buffer() failed (%d %p)\n", r, pic->f->data[0]); return -1; } if (edges_needed) { int i; for (i = 0; pic->f->data[i]; i++) { int offset = (EDGE_WIDTH >> (i ? chroma_y_shift : 0)) * pic->f->linesize[i] + (EDGE_WIDTH >> (i ? chroma_x_shift : 0)); pic->f->data[i] += offset; } pic->f->width = avctx->width; pic->f->height = avctx->height; } if (avctx->hwaccel) { assert(!pic->hwaccel_picture_private); if (avctx->hwaccel->frame_priv_data_size) { pic->hwaccel_priv_buf = av_buffer_allocz(avctx->hwaccel->frame_priv_data_size); if (!pic->hwaccel_priv_buf) { av_log(avctx, AV_LOG_ERROR, "alloc_frame_buffer() failed (hwaccel private data allocation)\n"); return -1; } pic->hwaccel_picture_private = pic->hwaccel_priv_buf->data; } } if (linesize && (linesize != pic->f->linesize[0] || uvlinesize != pic->f->linesize[1])) { av_log(avctx, AV_LOG_ERROR, "get_buffer() failed (stride changed)\n"); ff_mpeg_unref_picture(avctx, pic); return -1; } if (pic->f->linesize[1] != pic->f->linesize[2]) { av_log(avctx, AV_LOG_ERROR, "get_buffer() failed (uv stride mismatch)\n"); ff_mpeg_unref_picture(avctx, pic); return -1; } if (!sc->edge_emu_buffer && (ret = ff_mpeg_framesize_alloc(avctx, me, sc, pic->f->linesize[0])) < 0) { av_log(avctx, AV_LOG_ERROR, "get_buffer() failed to allocate context scratch buffers.\n"); ff_mpeg_unref_picture(avctx, pic); return ret; } return 0; } static int alloc_picture_tables(Picture *pic, int encoding, int out_format, int mb_stride, int mb_height, int b8_stride) { const int big_mb_num = mb_stride * (mb_height + 1) + 1; const int mb_array_size = mb_stride * mb_height; const int b8_array_size = b8_stride * mb_height * 2; int i; pic->mbskip_table_buf = av_buffer_allocz(mb_array_size + 2); pic->qscale_table_buf = av_buffer_allocz(big_mb_num + mb_stride); pic->mb_type_buf = av_buffer_allocz((big_mb_num + mb_stride) * sizeof(uint32_t)); if (!pic->mbskip_table_buf || !pic->qscale_table_buf || !pic->mb_type_buf) return AVERROR(ENOMEM); if (encoding) { pic->mb_var_buf = av_buffer_allocz(mb_array_size * sizeof(int16_t)); pic->mc_mb_var_buf = av_buffer_allocz(mb_array_size * sizeof(int16_t)); pic->mb_mean_buf = av_buffer_allocz(mb_array_size); if (!pic->mb_var_buf || !pic->mc_mb_var_buf || !pic->mb_mean_buf) return AVERROR(ENOMEM); } if (out_format == FMT_H263 || encoding) { int mv_size = 2 * (b8_array_size + 4) * sizeof(int16_t); int ref_index_size = 4 * mb_array_size; for (i = 0; mv_size && i < 2; i++) { pic->motion_val_buf[i] = av_buffer_allocz(mv_size); pic->ref_index_buf[i] = av_buffer_allocz(ref_index_size); if (!pic->motion_val_buf[i] || !pic->ref_index_buf[i]) return AVERROR(ENOMEM); } } return 0; } /** * Allocate a Picture. * The pixels are allocated/set by calling get_buffer() if shared = 0 */ int ff_alloc_picture(AVCodecContext *avctx, Picture *pic, MotionEstContext *me, ScratchpadContext *sc, int shared, int encoding, int chroma_x_shift, int chroma_y_shift, int out_format, int mb_stride, int mb_height, int b8_stride, ptrdiff_t *linesize, ptrdiff_t *uvlinesize) { int i, ret; if (shared) { assert(pic->f->data[0]); pic->shared = 1; } else { assert(!pic->f->buf[0]); if (alloc_frame_buffer(avctx, pic, me, sc, chroma_x_shift, chroma_y_shift, *linesize, *uvlinesize) < 0) return -1; *linesize = pic->f->linesize[0]; *uvlinesize = pic->f->linesize[1]; } if (!pic->qscale_table_buf) ret = alloc_picture_tables(pic, encoding, out_format, mb_stride, mb_height, b8_stride); else ret = make_tables_writable(pic); if (ret < 0) goto fail; if (encoding) { pic->mb_var = (uint16_t*)pic->mb_var_buf->data; pic->mc_mb_var = (uint16_t*)pic->mc_mb_var_buf->data; pic->mb_mean = pic->mb_mean_buf->data; } pic->mbskip_table = pic->mbskip_table_buf->data; pic->qscale_table = pic->qscale_table_buf->data + 2 * mb_stride + 1; pic->mb_type = (uint32_t*)pic->mb_type_buf->data + 2 * mb_stride + 1; if (pic->motion_val_buf[0]) { for (i = 0; i < 2; i++) { pic->motion_val[i] = (int16_t (*)[2])pic->motion_val_buf[i]->data + 4; pic->ref_index[i] = pic->ref_index_buf[i]->data; } } return 0; fail: av_log(avctx, AV_LOG_ERROR, "Error allocating a picture.\n"); ff_mpeg_unref_picture(avctx, pic); ff_free_picture_tables(pic); return AVERROR(ENOMEM); } /** * Deallocate a picture. */ void ff_mpeg_unref_picture(AVCodecContext *avctx, Picture *pic) { pic->tf.f = pic->f; /* WM Image / Screen codecs allocate internal buffers with different * dimensions / colorspaces; ignore user-defined callbacks for these. */ if (avctx->codec_id != AV_CODEC_ID_WMV3IMAGE && avctx->codec_id != AV_CODEC_ID_VC1IMAGE && avctx->codec_id != AV_CODEC_ID_MSS2) ff_thread_release_buffer(avctx, &pic->tf); else if (pic->f) av_frame_unref(pic->f); av_buffer_unref(&pic->hwaccel_priv_buf); if (pic->needs_realloc) ff_free_picture_tables(pic); } int ff_update_picture_tables(Picture *dst, Picture *src) { int i; #define UPDATE_TABLE(table) \ do { \ if (src->table && \ (!dst->table || dst->table->buffer != src->table->buffer)) { \ av_buffer_unref(&dst->table); \ dst->table = av_buffer_ref(src->table); \ if (!dst->table) { \ ff_free_picture_tables(dst); \ return AVERROR(ENOMEM); \ } \ } \ } while (0) UPDATE_TABLE(mb_var_buf); UPDATE_TABLE(mc_mb_var_buf); UPDATE_TABLE(mb_mean_buf); UPDATE_TABLE(mbskip_table_buf); UPDATE_TABLE(qscale_table_buf); UPDATE_TABLE(mb_type_buf); for (i = 0; i < 2; i++) { UPDATE_TABLE(motion_val_buf[i]); UPDATE_TABLE(ref_index_buf[i]); } dst->mb_var = src->mb_var; dst->mc_mb_var = src->mc_mb_var; dst->mb_mean = src->mb_mean; dst->mbskip_table = src->mbskip_table; dst->qscale_table = src->qscale_table; dst->mb_type = src->mb_type; for (i = 0; i < 2; i++) { dst->motion_val[i] = src->motion_val[i]; dst->ref_index[i] = src->ref_index[i]; } return 0; } int ff_mpeg_ref_picture(AVCodecContext *avctx, Picture *dst, Picture *src) { int ret; av_assert0(!dst->f->buf[0]); av_assert0(src->f->buf[0]); src->tf.f = src->f; dst->tf.f = dst->f; ret = ff_thread_ref_frame(&dst->tf, &src->tf); if (ret < 0) goto fail; ret = ff_update_picture_tables(dst, src); if (ret < 0) goto fail; if (src->hwaccel_picture_private) { dst->hwaccel_priv_buf = av_buffer_ref(src->hwaccel_priv_buf); if (!dst->hwaccel_priv_buf) goto fail; dst->hwaccel_picture_private = dst->hwaccel_priv_buf->data; } dst->field_picture = src->field_picture; dst->mb_var_sum = src->mb_var_sum; dst->mc_mb_var_sum = src->mc_mb_var_sum; dst->b_frame_score = src->b_frame_score; dst->needs_realloc = src->needs_realloc; dst->reference = src->reference; dst->shared = src->shared; memcpy(dst->encoding_error, src->encoding_error, sizeof(dst->encoding_error)); return 0; fail: ff_mpeg_unref_picture(avctx, dst); return ret; } static inline int pic_is_unused(Picture *pic) { if (!pic->f->buf[0]) return 1; if (pic->needs_realloc && !(pic->reference & DELAYED_PIC_REF)) return 1; return 0; } static int find_unused_picture(Picture *picture, int shared) { int i; if (shared) { for (i = 0; i < MAX_PICTURE_COUNT; i++) { if (!picture[i].f->buf[0]) return i; } } else { for (i = 0; i < MAX_PICTURE_COUNT; i++) { if (pic_is_unused(&picture[i])) return i; } } return AVERROR_INVALIDDATA; } int ff_find_unused_picture(AVCodecContext *avctx, Picture *picture, int shared) { int ret = find_unused_picture(picture, shared); if (ret >= 0 && ret < MAX_PICTURE_COUNT) { if (picture[ret].needs_realloc) { picture[ret].needs_realloc = 0; ff_free_picture_tables(&picture[ret]); ff_mpeg_unref_picture(avctx, &picture[ret]); } } return ret; } void ff_free_picture_tables(Picture *pic) { int i; av_buffer_unref(&pic->mb_var_buf); av_buffer_unref(&pic->mc_mb_var_buf); av_buffer_unref(&pic->mb_mean_buf); av_buffer_unref(&pic->mbskip_table_buf); av_buffer_unref(&pic->qscale_table_buf); av_buffer_unref(&pic->mb_type_buf); for (i = 0; i < 2; i++) { av_buffer_unref(&pic->motion_val_buf[i]); av_buffer_unref(&pic->ref_index_buf[i]); } }