2 * Mpeg video formats-related picture management functions
4 * This file is part of FFmpeg.
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 #include "libavutil/avassert.h"
24 #include "libavutil/common.h"
25 #include "libavutil/pixdesc.h"
28 #include "motion_est.h"
29 #include "mpegpicture.h"
30 #include "mpegutils.h"
32 static int make_tables_writable(Picture *pic)
35 #define MAKE_WRITABLE(table) \
38 (ret = av_buffer_make_writable(&pic->table)) < 0)\
42 MAKE_WRITABLE(mb_var_buf);
43 MAKE_WRITABLE(mc_mb_var_buf);
44 MAKE_WRITABLE(mb_mean_buf);
45 MAKE_WRITABLE(mbskip_table_buf);
46 MAKE_WRITABLE(qscale_table_buf);
47 MAKE_WRITABLE(mb_type_buf);
49 for (i = 0; i < 2; i++) {
50 MAKE_WRITABLE(motion_val_buf[i]);
51 MAKE_WRITABLE(ref_index_buf[i]);
57 int ff_mpeg_framesize_alloc(AVCodecContext *avctx, MotionEstContext *me,
58 ScratchpadContext *sc, int linesize)
60 int alloc_size = FFALIGN(FFABS(linesize) + 64, 32);
66 av_log(avctx, AV_LOG_ERROR, "Image too small, temporary buffers cannot function\n");
67 return AVERROR_PATCHWELCOME;
70 // edge emu needs blocksize + filter length - 1
71 // (= 17x17 for halfpel / 21x21 for H.264)
72 // VC-1 computes luma and chroma simultaneously and needs 19X19 + 9x9
73 // at uvlinesize. It supports only YUV420 so 24x24 is enough
74 // linesize * interlaced * MBsize
75 // we also use this buffer for encoding in encode_mb_internal() needig an additional 32 lines
76 FF_ALLOCZ_ARRAY_OR_GOTO(avctx, sc->edge_emu_buffer, alloc_size, 4 * 70,
79 FF_ALLOCZ_ARRAY_OR_GOTO(avctx, me->scratchpad, alloc_size, 4 * 16 * 2,
81 me->temp = me->scratchpad;
82 sc->rd_scratchpad = me->scratchpad;
83 sc->b_scratchpad = me->scratchpad;
84 sc->obmc_scratchpad = me->scratchpad + 16;
88 av_freep(&sc->edge_emu_buffer);
89 return AVERROR(ENOMEM);
93 * Allocate a frame buffer
95 static int alloc_frame_buffer(AVCodecContext *avctx, Picture *pic,
96 MotionEstContext *me, ScratchpadContext *sc,
97 int chroma_x_shift, int chroma_y_shift,
98 int linesize, int uvlinesize)
100 int edges_needed = av_codec_is_encoder(avctx->codec);
104 if (avctx->codec_id != AV_CODEC_ID_WMV3IMAGE &&
105 avctx->codec_id != AV_CODEC_ID_VC1IMAGE &&
106 avctx->codec_id != AV_CODEC_ID_MSS2) {
108 pic->f->width = avctx->width + 2 * EDGE_WIDTH;
109 pic->f->height = avctx->height + 2 * EDGE_WIDTH;
112 r = ff_thread_get_buffer(avctx, &pic->tf,
113 pic->reference ? AV_GET_BUFFER_FLAG_REF : 0);
115 pic->f->width = avctx->width;
116 pic->f->height = avctx->height;
117 pic->f->format = avctx->pix_fmt;
118 r = avcodec_default_get_buffer2(avctx, pic->f, 0);
121 if (r < 0 || !pic->f->buf[0]) {
122 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed (%d %p)\n",
129 for (i = 0; pic->f->data[i]; i++) {
130 int offset = (EDGE_WIDTH >> (i ? chroma_y_shift : 0)) *
131 pic->f->linesize[i] +
132 (EDGE_WIDTH >> (i ? chroma_x_shift : 0));
133 pic->f->data[i] += offset;
135 pic->f->width = avctx->width;
136 pic->f->height = avctx->height;
139 if (avctx->hwaccel) {
140 assert(!pic->hwaccel_picture_private);
141 if (avctx->hwaccel->frame_priv_data_size) {
142 pic->hwaccel_priv_buf = av_buffer_allocz(avctx->hwaccel->frame_priv_data_size);
143 if (!pic->hwaccel_priv_buf) {
144 av_log(avctx, AV_LOG_ERROR, "alloc_frame_buffer() failed (hwaccel private data allocation)\n");
147 pic->hwaccel_picture_private = pic->hwaccel_priv_buf->data;
151 if (linesize && (linesize != pic->f->linesize[0] ||
152 uvlinesize != pic->f->linesize[1])) {
153 av_log(avctx, AV_LOG_ERROR,
154 "get_buffer() failed (stride changed)\n");
155 ff_mpeg_unref_picture(avctx, pic);
159 if (av_pix_fmt_count_planes(pic->f->format) > 2 &&
160 pic->f->linesize[1] != pic->f->linesize[2]) {
161 av_log(avctx, AV_LOG_ERROR,
162 "get_buffer() failed (uv stride mismatch)\n");
163 ff_mpeg_unref_picture(avctx, pic);
167 if (!sc->edge_emu_buffer &&
168 (ret = ff_mpeg_framesize_alloc(avctx, me, sc,
169 pic->f->linesize[0])) < 0) {
170 av_log(avctx, AV_LOG_ERROR,
171 "get_buffer() failed to allocate context scratch buffers.\n");
172 ff_mpeg_unref_picture(avctx, pic);
179 static int alloc_picture_tables(AVCodecContext *avctx, Picture *pic, int encoding, int out_format,
180 int mb_stride, int mb_width, int mb_height, int b8_stride)
182 const int big_mb_num = mb_stride * (mb_height + 1) + 1;
183 const int mb_array_size = mb_stride * mb_height;
184 const int b8_array_size = b8_stride * mb_height * 2;
188 pic->mbskip_table_buf = av_buffer_allocz(mb_array_size + 2);
189 pic->qscale_table_buf = av_buffer_allocz(big_mb_num + mb_stride);
190 pic->mb_type_buf = av_buffer_allocz((big_mb_num + mb_stride) *
192 if (!pic->mbskip_table_buf || !pic->qscale_table_buf || !pic->mb_type_buf)
193 return AVERROR(ENOMEM);
196 pic->mb_var_buf = av_buffer_allocz(mb_array_size * sizeof(int16_t));
197 pic->mc_mb_var_buf = av_buffer_allocz(mb_array_size * sizeof(int16_t));
198 pic->mb_mean_buf = av_buffer_allocz(mb_array_size);
199 if (!pic->mb_var_buf || !pic->mc_mb_var_buf || !pic->mb_mean_buf)
200 return AVERROR(ENOMEM);
203 if (out_format == FMT_H263 || encoding ||
207 (avctx->flags2 & AV_CODEC_FLAG2_EXPORT_MVS)) {
208 int mv_size = 2 * (b8_array_size + 4) * sizeof(int16_t);
209 int ref_index_size = 4 * mb_array_size;
211 for (i = 0; mv_size && i < 2; i++) {
212 pic->motion_val_buf[i] = av_buffer_allocz(mv_size);
213 pic->ref_index_buf[i] = av_buffer_allocz(ref_index_size);
214 if (!pic->motion_val_buf[i] || !pic->ref_index_buf[i])
215 return AVERROR(ENOMEM);
219 pic->alloc_mb_width = mb_width;
220 pic->alloc_mb_height = mb_height;
226 * Allocate a Picture.
227 * The pixels are allocated/set by calling get_buffer() if shared = 0
229 int ff_alloc_picture(AVCodecContext *avctx, Picture *pic, MotionEstContext *me,
230 ScratchpadContext *sc, int shared, int encoding,
231 int chroma_x_shift, int chroma_y_shift, int out_format,
232 int mb_stride, int mb_width, int mb_height, int b8_stride,
233 ptrdiff_t *linesize, ptrdiff_t *uvlinesize)
237 if (pic->qscale_table_buf)
238 if ( pic->alloc_mb_width != mb_width
239 || pic->alloc_mb_height != mb_height)
240 ff_free_picture_tables(pic);
243 av_assert0(pic->f->data[0]);
246 av_assert0(!pic->f->buf[0]);
247 if (alloc_frame_buffer(avctx, pic, me, sc,
248 chroma_x_shift, chroma_y_shift,
249 *linesize, *uvlinesize) < 0)
252 *linesize = pic->f->linesize[0];
253 *uvlinesize = pic->f->linesize[1];
256 if (!pic->qscale_table_buf)
257 ret = alloc_picture_tables(avctx, pic, encoding, out_format,
258 mb_stride, mb_width, mb_height, b8_stride);
260 ret = make_tables_writable(pic);
265 pic->mb_var = (uint16_t*)pic->mb_var_buf->data;
266 pic->mc_mb_var = (uint16_t*)pic->mc_mb_var_buf->data;
267 pic->mb_mean = pic->mb_mean_buf->data;
270 pic->mbskip_table = pic->mbskip_table_buf->data;
271 pic->qscale_table = pic->qscale_table_buf->data + 2 * mb_stride + 1;
272 pic->mb_type = (uint32_t*)pic->mb_type_buf->data + 2 * mb_stride + 1;
274 if (pic->motion_val_buf[0]) {
275 for (i = 0; i < 2; i++) {
276 pic->motion_val[i] = (int16_t (*)[2])pic->motion_val_buf[i]->data + 4;
277 pic->ref_index[i] = pic->ref_index_buf[i]->data;
283 av_log(avctx, AV_LOG_ERROR, "Error allocating a picture.\n");
284 ff_mpeg_unref_picture(avctx, pic);
285 ff_free_picture_tables(pic);
286 return AVERROR(ENOMEM);
290 * Deallocate a picture.
292 void ff_mpeg_unref_picture(AVCodecContext *avctx, Picture *pic)
294 int off = offsetof(Picture, mb_mean) + sizeof(pic->mb_mean);
297 /* WM Image / Screen codecs allocate internal buffers with different
298 * dimensions / colorspaces; ignore user-defined callbacks for these. */
299 if (avctx->codec_id != AV_CODEC_ID_WMV3IMAGE &&
300 avctx->codec_id != AV_CODEC_ID_VC1IMAGE &&
301 avctx->codec_id != AV_CODEC_ID_MSS2)
302 ff_thread_release_buffer(avctx, &pic->tf);
304 av_frame_unref(pic->f);
306 av_buffer_unref(&pic->hwaccel_priv_buf);
308 if (pic->needs_realloc)
309 ff_free_picture_tables(pic);
311 memset((uint8_t*)pic + off, 0, sizeof(*pic) - off);
314 int ff_update_picture_tables(Picture *dst, Picture *src)
318 #define UPDATE_TABLE(table) \
321 (!dst->table || dst->table->buffer != src->table->buffer)) { \
322 av_buffer_unref(&dst->table); \
323 dst->table = av_buffer_ref(src->table); \
325 ff_free_picture_tables(dst); \
326 return AVERROR(ENOMEM); \
331 UPDATE_TABLE(mb_var_buf);
332 UPDATE_TABLE(mc_mb_var_buf);
333 UPDATE_TABLE(mb_mean_buf);
334 UPDATE_TABLE(mbskip_table_buf);
335 UPDATE_TABLE(qscale_table_buf);
336 UPDATE_TABLE(mb_type_buf);
337 for (i = 0; i < 2; i++) {
338 UPDATE_TABLE(motion_val_buf[i]);
339 UPDATE_TABLE(ref_index_buf[i]);
342 dst->mb_var = src->mb_var;
343 dst->mc_mb_var = src->mc_mb_var;
344 dst->mb_mean = src->mb_mean;
345 dst->mbskip_table = src->mbskip_table;
346 dst->qscale_table = src->qscale_table;
347 dst->mb_type = src->mb_type;
348 for (i = 0; i < 2; i++) {
349 dst->motion_val[i] = src->motion_val[i];
350 dst->ref_index[i] = src->ref_index[i];
353 dst->alloc_mb_width = src->alloc_mb_width;
354 dst->alloc_mb_height = src->alloc_mb_height;
359 int ff_mpeg_ref_picture(AVCodecContext *avctx, Picture *dst, Picture *src)
363 av_assert0(!dst->f->buf[0]);
364 av_assert0(src->f->buf[0]);
368 ret = ff_thread_ref_frame(&dst->tf, &src->tf);
372 ret = ff_update_picture_tables(dst, src);
376 if (src->hwaccel_picture_private) {
377 dst->hwaccel_priv_buf = av_buffer_ref(src->hwaccel_priv_buf);
378 if (!dst->hwaccel_priv_buf) {
379 ret = AVERROR(ENOMEM);
382 dst->hwaccel_picture_private = dst->hwaccel_priv_buf->data;
385 dst->field_picture = src->field_picture;
386 dst->mb_var_sum = src->mb_var_sum;
387 dst->mc_mb_var_sum = src->mc_mb_var_sum;
388 dst->b_frame_score = src->b_frame_score;
389 dst->needs_realloc = src->needs_realloc;
390 dst->reference = src->reference;
391 dst->shared = src->shared;
393 memcpy(dst->encoding_error, src->encoding_error,
394 sizeof(dst->encoding_error));
398 ff_mpeg_unref_picture(avctx, dst);
402 static inline int pic_is_unused(Picture *pic)
406 if (pic->needs_realloc && !(pic->reference & DELAYED_PIC_REF))
411 static int find_unused_picture(AVCodecContext *avctx, Picture *picture, int shared)
416 for (i = 0; i < MAX_PICTURE_COUNT; i++) {
417 if (!picture[i].f->buf[0])
421 for (i = 0; i < MAX_PICTURE_COUNT; i++) {
422 if (pic_is_unused(&picture[i]))
427 av_log(avctx, AV_LOG_FATAL,
428 "Internal error, picture buffer overflow\n");
429 /* We could return -1, but the codec would crash trying to draw into a
430 * non-existing frame anyway. This is safer than waiting for a random crash.
431 * Also the return of this is never useful, an encoder must only allocate
432 * as much as allowed in the specification. This has no relationship to how
433 * much libavcodec could allocate (and MAX_PICTURE_COUNT is always large
434 * enough for such valid streams).
435 * Plus, a decoder has to check stream validity and remove frames if too
436 * many reference frames are around. Waiting for "OOM" is not correct at
437 * all. Similarly, missing reference frames have to be replaced by
438 * interpolated/MC frames, anything else is a bug in the codec ...
444 int ff_find_unused_picture(AVCodecContext *avctx, Picture *picture, int shared)
446 int ret = find_unused_picture(avctx, picture, shared);
448 if (ret >= 0 && ret < MAX_PICTURE_COUNT) {
449 if (picture[ret].needs_realloc) {
450 picture[ret].needs_realloc = 0;
451 ff_free_picture_tables(&picture[ret]);
452 ff_mpeg_unref_picture(avctx, &picture[ret]);
458 void ff_free_picture_tables(Picture *pic)
462 pic->alloc_mb_width =
463 pic->alloc_mb_height = 0;
465 av_buffer_unref(&pic->mb_var_buf);
466 av_buffer_unref(&pic->mc_mb_var_buf);
467 av_buffer_unref(&pic->mb_mean_buf);
468 av_buffer_unref(&pic->mbskip_table_buf);
469 av_buffer_unref(&pic->qscale_table_buf);
470 av_buffer_unref(&pic->mb_type_buf);
472 for (i = 0; i < 2; i++) {
473 av_buffer_unref(&pic->motion_val_buf[i]);
474 av_buffer_unref(&pic->ref_index_buf[i]);