FFmpeg coverage

GCC Code Coverage Report

Directory:	../../../ffmpeg/		Exec	Total	Coverage
File:	src/libavcodec/mpegpicture.c	Lines:	198	246	80.5 %
Date:	2020-09-25 23:16:12	Branches:	180	252	71.4 %


/*
 * Mpeg video formats-related picture management functions
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg 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.
 *
 * FFmpeg 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 FFmpeg; 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 "libavutil/pixdesc.h"
#include "libavutil/imgutils.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)
{
#   define EMU_EDGE_HEIGHT (4 * 70)
    int alloc_size = FFALIGN(FFABS(linesize) + 64, 32);

    if (avctx->hwaccel)
        return 0;

    if (linesize < 24) {
        av_log(avctx, AV_LOG_ERROR, "Image too small, temporary buffers cannot function\n");
        return AVERROR_PATCHWELCOME;
    }

    if (av_image_check_size2(alloc_size, EMU_EDGE_HEIGHT, avctx->max_pixels, AV_PIX_FMT_NONE, 0, avctx) < 0)
        return AVERROR(ENOMEM);

    // edge emu needs blocksize + filter length - 1
    // (= 17x17 for  halfpel / 21x21 for H.264)
    // VC-1 computes luma and chroma simultaneously and needs 19X19 + 9x9
    // at uvlinesize. It supports only YUV420 so 24x24 is enough
    // linesize * interlaced * MBsize
    // we also use this buffer for encoding in encode_mb_internal() needig an additional 32 lines
    if (!FF_ALLOCZ_TYPED_ARRAY(sc->edge_emu_buffer, alloc_size * EMU_EDGE_HEIGHT) ||
        !FF_ALLOCZ_TYPED_ARRAY(me->scratchpad,      alloc_size * 4 * 16 * 2))
        return AVERROR(ENOMEM);
    me->temp            = me->scratchpad;
    sc->rd_scratchpad   = me->scratchpad;
    sc->b_scratchpad    = me->scratchpad;
    sc->obmc_scratchpad = me->scratchpad + 16;

    return 0;
}

/**
 * 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 && uvlinesize != pic->f->linesize[1])) {
        av_log(avctx, AV_LOG_ERROR,
               "get_buffer() failed (stride changed: linesize=%d/%d uvlinesize=%d/%d)\n",
               linesize,   pic->f->linesize[0],
               uvlinesize, pic->f->linesize[1]);
        ff_mpeg_unref_picture(avctx, pic);
        return -1;
    }

    if (av_pix_fmt_count_planes(pic->f->format) > 2 &&
        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(AVCodecContext *avctx, Picture *pic, int encoding, int out_format,
                                int mb_stride, int mb_width, 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 ||
#if FF_API_DEBUG_MV
        avctx->debug_mv ||
#endif
        (avctx->export_side_data & AV_CODEC_EXPORT_DATA_MVS)) {
        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);
        }
    }

    pic->alloc_mb_width  = mb_width;
    pic->alloc_mb_height = mb_height;

    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_width, int mb_height, int b8_stride,
                     ptrdiff_t *linesize, ptrdiff_t *uvlinesize)
{
    int i, ret;

    if (pic->qscale_table_buf)
        if (   pic->alloc_mb_width  != mb_width
            || pic->alloc_mb_height != mb_height)
            ff_free_picture_tables(pic);

    if (shared) {
        av_assert0(pic->f->data[0]);
        pic->shared = 1;
    } else {
        av_assert0(!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(avctx, pic, encoding, out_format,
                                   mb_stride, mb_width, 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)
{
    int off = offsetof(Picture, mb_mean) + sizeof(pic->mb_mean);

    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);

    memset((uint8_t*)pic + off, 0, sizeof(*pic) - off);
}

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];
    }

    dst->alloc_mb_width  = src->alloc_mb_width;
    dst->alloc_mb_height = src->alloc_mb_height;

    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) {
            ret = AVERROR(ENOMEM);
            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(AVCodecContext *avctx, 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;
        }
    }

    av_log(avctx, AV_LOG_FATAL,
           "Internal error, picture buffer overflow\n");
    /* We could return -1, but the codec would crash trying to draw into a
     * non-existing frame anyway. This is safer than waiting for a random crash.
     * Also the return of this is never useful, an encoder must only allocate
     * as much as allowed in the specification. This has no relationship to how
     * much libavcodec could allocate (and MAX_PICTURE_COUNT is always large
     * enough for such valid streams).
     * Plus, a decoder has to check stream validity and remove frames if too
     * many reference frames are around. Waiting for "OOM" is not correct at
     * all. Similarly, missing reference frames have to be replaced by
     * interpolated/MC frames, anything else is a bug in the codec ...
     */
    abort();
    return -1;
}

int ff_find_unused_picture(AVCodecContext *avctx, Picture *picture, int shared)
{
    int ret = find_unused_picture(avctx, 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;

    pic->alloc_mb_width  =
    pic->alloc_mb_height = 0;

    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]);
    }
}


Generated by: GCOVR (Version 4.2)

Line	Branch	Exec	Source
1			/*
2			* Mpeg video formats-related picture management functions
3			*
4			* This file is part of FFmpeg.
5			*
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.
10			*
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.
15			*
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
19			*/
20
21			#include <stdint.h>
22
23			#include "libavutil/avassert.h"
24			#include "libavutil/common.h"
25			#include "libavutil/pixdesc.h"
26			#include "libavutil/imgutils.h"
27
28			#include "avcodec.h"
29			#include "motion_est.h"
30			#include "mpegpicture.h"
31			#include "mpegutils.h"
32
33		18557	static int make_tables_writable(Picture *pic)
34			{
35			int ret, i;
36			#define MAKE_WRITABLE(table) \
37			do {\
38			if (pic->table &&\
39			(ret = av_buffer_make_writable(&pic->table)) < 0)\
40			return ret;\
41			} while (0)
42
43	✓✓✗✓	18557	MAKE_WRITABLE(mb_var_buf);
44	✓✓✗✓	18557	MAKE_WRITABLE(mc_mb_var_buf);
45	✓✓✗✓	18557	MAKE_WRITABLE(mb_mean_buf);
46	✓✗✗✓	18557	MAKE_WRITABLE(mbskip_table_buf);
47	✓✗✗✓	18557	MAKE_WRITABLE(qscale_table_buf);
48	✓✗✗✓	18557	MAKE_WRITABLE(mb_type_buf);
49
50	✓✓	55671	for (i = 0; i < 2; i++) {
51	✓✓✗✓	37114	MAKE_WRITABLE(motion_val_buf[i]);
52	✓✓✗✓	37114	MAKE_WRITABLE(ref_index_buf[i]);
53			}
54
55		18557	return 0;
56			}
57
58		436	int ff_mpeg_framesize_alloc(AVCodecContext avctx, MotionEstContext me,
59			ScratchpadContext *sc, int linesize)
60			{
61			# define EMU_EDGE_HEIGHT (4 * 70)
62		436	int alloc_size = FFALIGN(FFABS(linesize) + 64, 32);
63
64	✗✓	436	if (avctx->hwaccel)
65			return 0;
66
67	✗✓	436	if (linesize < 24) {
68			av_log(avctx, AV_LOG_ERROR, "Image too small, temporary buffers cannot function\n");
69			return AVERROR_PATCHWELCOME;
70			}
71
72	✗✓	436	if (av_image_check_size2(alloc_size, EMU_EDGE_HEIGHT, avctx->max_pixels, AV_PIX_FMT_NONE, 0, avctx) < 0)
73			return AVERROR(ENOMEM);
74
75			// edge emu needs blocksize + filter length - 1
76			// (= 17x17 for halfpel / 21x21 for H.264)
77			// VC-1 computes luma and chroma simultaneously and needs 19X19 + 9x9
78			// at uvlinesize. It supports only YUV420 so 24x24 is enough
79			// linesize * interlaced * MBsize
80			// we also use this buffer for encoding in encode_mb_internal() needig an additional 32 lines
81	✓✗	436	if (!FF_ALLOCZ_TYPED_ARRAY(sc->edge_emu_buffer, alloc_size * EMU_EDGE_HEIGHT) \|\|
82	✗✓	436	!FF_ALLOCZ_TYPED_ARRAY(me->scratchpad, alloc_size * 4 * 16 * 2))
83			return AVERROR(ENOMEM);
84		436	me->temp = me->scratchpad;
85		436	sc->rd_scratchpad = me->scratchpad;
86		436	sc->b_scratchpad = me->scratchpad;
87		436	sc->obmc_scratchpad = me->scratchpad + 16;
88
89		436	return 0;
90			}
91
92			/**
93			* Allocate a frame buffer
94			*/
95		19208	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)
99			{
100		19208	int edges_needed = av_codec_is_encoder(avctx->codec);
101			int r, ret;
102
103		19208	pic->tf.f = pic->f;
104	✓✗	19208	if (avctx->codec_id != AV_CODEC_ID_WMV3IMAGE &&
105	✓✗	19208	avctx->codec_id != AV_CODEC_ID_VC1IMAGE &&
106	✓✓	19208	avctx->codec_id != AV_CODEC_ID_MSS2) {
107	✓✓	19129	if (edges_needed) {
108		8358	pic->f->width = avctx->width + 2 * EDGE_WIDTH;
109		8358	pic->f->height = avctx->height + 2 * EDGE_WIDTH;
110			}
111
112		19129	r = ff_thread_get_buffer(avctx, &pic->tf,
113		19129	pic->reference ? AV_GET_BUFFER_FLAG_REF : 0);
114			} else {
115		79	pic->f->width = avctx->width;
116		79	pic->f->height = avctx->height;
117		79	pic->f->format = avctx->pix_fmt;
118		79	r = avcodec_default_get_buffer2(avctx, pic->f, 0);
119			}
120
121	✓✗✗✓	19208	if (r < 0 \|\| !pic->f->buf[0]) {
122			av_log(avctx, AV_LOG_ERROR, "get_buffer() failed (%d %p)\n",
123			r, pic->f->data[0]);
124			return -1;
125			}
126
127	✓✓	19208	if (edges_needed) {
128			int i;
129	✓✓	33432	for (i = 0; pic->f->data[i]; i++) {
130	✓✓	25074	int offset = (EDGE_WIDTH >> (i ? chroma_y_shift : 0)) *
131		25074	pic->f->linesize[i] +
132	✓✓	25074	(EDGE_WIDTH >> (i ? chroma_x_shift : 0));
133		25074	pic->f->data[i] += offset;
134			}
135		8358	pic->f->width = avctx->width;
136		8358	pic->f->height = avctx->height;
137			}
138
139	✗✓	19208	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");
145			return -1;
146			}
147			pic->hwaccel_picture_private = pic->hwaccel_priv_buf->data;
148			}
149			}
150
151	✓✓✓✗ ✓✓	19208	if ((linesize && linesize != pic->f->linesize[0]) \|\|
152	✗✓	18808	(uvlinesize && uvlinesize != pic->f->linesize[1])) {
153			av_log(avctx, AV_LOG_ERROR,
154			"get_buffer() failed (stride changed: linesize=%d/%d uvlinesize=%d/%d)\n",
155			linesize, pic->f->linesize[0],
156			uvlinesize, pic->f->linesize[1]);
157			ff_mpeg_unref_picture(avctx, pic);
158			return -1;
159			}
160
161	✓✗	19208	if (av_pix_fmt_count_planes(pic->f->format) > 2 &&
162	✗✓	19208	pic->f->linesize[1] != pic->f->linesize[2]) {
163			av_log(avctx, AV_LOG_ERROR,
164			"get_buffer() failed (uv stride mismatch)\n");
165			ff_mpeg_unref_picture(avctx, pic);
166			return -1;
167			}
168
169	✓✓✗✓	19608	if (!sc->edge_emu_buffer &&
170		400	(ret = ff_mpeg_framesize_alloc(avctx, me, sc,
171		400	pic->f->linesize[0])) < 0) {
172			av_log(avctx, AV_LOG_ERROR,
173			"get_buffer() failed to allocate context scratch buffers.\n");
174			ff_mpeg_unref_picture(avctx, pic);
175			return ret;
176			}
177
178		19208	return 0;
179			}
180
181		1300	static int alloc_picture_tables(AVCodecContext avctx, Picture pic, int encoding, int out_format,
182			int mb_stride, int mb_width, int mb_height, int b8_stride)
183			{
184		1300	const int big_mb_num = mb_stride * (mb_height + 1) + 1;
185		1300	const int mb_array_size = mb_stride * mb_height;
186		1300	const int b8_array_size = b8_stride * mb_height * 2;
187			int i;
188
189
190		1300	pic->mbskip_table_buf = av_buffer_allocz(mb_array_size + 2);
191		1300	pic->qscale_table_buf = av_buffer_allocz(big_mb_num + mb_stride);
192		1300	pic->mb_type_buf = av_buffer_allocz((big_mb_num + mb_stride) *
193			sizeof(uint32_t));
194	✓✗✓✗ ✗✓	1300	if (!pic->mbskip_table_buf \|\| !pic->qscale_table_buf \|\| !pic->mb_type_buf)
195			return AVERROR(ENOMEM);
196
197	✓✓	1300	if (encoding) {
198		687	pic->mb_var_buf = av_buffer_allocz(mb_array_size * sizeof(int16_t));
199		687	pic->mc_mb_var_buf = av_buffer_allocz(mb_array_size * sizeof(int16_t));
200		687	pic->mb_mean_buf = av_buffer_allocz(mb_array_size);
201	✓✗✓✗ ✗✓	687	if (!pic->mb_var_buf \|\| !pic->mc_mb_var_buf \|\| !pic->mb_mean_buf)
202			return AVERROR(ENOMEM);
203			}
204
205	✓✓✓✓	1300	if (out_format == FMT_H263 \|\| encoding \|\|
206			#if FF_API_DEBUG_MV
207			avctx->debug_mv \|\|
208			#endif
209	✗✓	184	(avctx->export_side_data & AV_CODEC_EXPORT_DATA_MVS)) {
210		1116	int mv_size = 2 * (b8_array_size + 4) * sizeof(int16_t);
211		1116	int ref_index_size = 4 * mb_array_size;
212
213	✓✗✓✓	3348	for (i = 0; mv_size && i < 2; i++) {
214		2232	pic->motion_val_buf[i] = av_buffer_allocz(mv_size);
215		2232	pic->ref_index_buf[i] = av_buffer_allocz(ref_index_size);
216	✓✗✗✓	2232	if (!pic->motion_val_buf[i] \|\| !pic->ref_index_buf[i])
217			return AVERROR(ENOMEM);
218			}
219			}
220
221		1300	pic->alloc_mb_width = mb_width;
222		1300	pic->alloc_mb_height = mb_height;
223
224		1300	return 0;
225			}
226
227			/**
228			* Allocate a Picture.
229			* The pixels are allocated/set by calling get_buffer() if shared = 0
230			*/
231		19857	int ff_alloc_picture(AVCodecContext avctx, Picture pic, MotionEstContext *me,
232			ScratchpadContext *sc, int shared, int encoding,
233			int chroma_x_shift, int chroma_y_shift, int out_format,
234			int mb_stride, int mb_width, int mb_height, int b8_stride,
235			ptrdiff_t linesize, ptrdiff_t uvlinesize)
236			{
237			int i, ret;
238
239	✓✓	19857	if (pic->qscale_table_buf)
240	✓✗	18557	if ( pic->alloc_mb_width != mb_width
241	✗✓	18557	\|\| pic->alloc_mb_height != mb_height)
242			ff_free_picture_tables(pic);
243
244	✓✓	19857	if (shared) {
245	✗✓	649	av_assert0(pic->f->data[0]);
246		649	pic->shared = 1;
247			} else {
248	✗✓	19208	av_assert0(!pic->f->buf[0]);
249	✗✓	19208	if (alloc_frame_buffer(avctx, pic, me, sc,
250			chroma_x_shift, chroma_y_shift,
251		19208	linesize, uvlinesize) < 0)
252			return -1;
253
254		19208	*linesize = pic->f->linesize[0];
255		19208	*uvlinesize = pic->f->linesize[1];
256			}
257
258	✓✓	19857	if (!pic->qscale_table_buf)
259		1300	ret = alloc_picture_tables(avctx, pic, encoding, out_format,
260			mb_stride, mb_width, mb_height, b8_stride);
261			else
262		18557	ret = make_tables_writable(pic);
263	✗✓	19857	if (ret < 0)
264			goto fail;
265
266	✓✓	19857	if (encoding) {
267		9007	pic->mb_var = (uint16_t*)pic->mb_var_buf->data;
268		9007	pic->mc_mb_var = (uint16_t*)pic->mc_mb_var_buf->data;
269		9007	pic->mb_mean = pic->mb_mean_buf->data;
270			}
271
272		19857	pic->mbskip_table = pic->mbskip_table_buf->data;
273		19857	pic->qscale_table = pic->qscale_table_buf->data + 2 * mb_stride + 1;
274		19857	pic->mb_type = (uint32_t)pic->mb_type_buf->data + 2 mb_stride + 1;
275
276	✓✓	19857	if (pic->motion_val_buf[0]) {
277	✓✓	48192	for (i = 0; i < 2; i++) {
278		32128	pic->motion_val[i] = (int16_t (*)[2])pic->motion_val_buf[i]->data + 4;
279		32128	pic->ref_index[i] = pic->ref_index_buf[i]->data;
280			}
281			}
282
283		19857	return 0;
284			fail:
285			av_log(avctx, AV_LOG_ERROR, "Error allocating a picture.\n");
286			ff_mpeg_unref_picture(avctx, pic);
287			ff_free_picture_tables(pic);
288			return AVERROR(ENOMEM);
289			}
290
291			/**
292			* Deallocate a picture.
293			*/
294		784681	void ff_mpeg_unref_picture(AVCodecContext avctx, Picture pic)
295			{
296		784681	int off = offsetof(Picture, mb_mean) + sizeof(pic->mb_mean);
297
298		784681	pic->tf.f = pic->f;
299			/* WM Image / Screen codecs allocate internal buffers with different
300			* dimensions / colorspaces; ignore user-defined callbacks for these. */
301	✓✗	784681	if (avctx->codec_id != AV_CODEC_ID_WMV3IMAGE &&
302	✓✗	784681	avctx->codec_id != AV_CODEC_ID_VC1IMAGE &&
303	✓✓	784681	avctx->codec_id != AV_CODEC_ID_MSS2)
304		778119	ff_thread_release_buffer(avctx, &pic->tf);
305	✓✗	6562	else if (pic->f)
306		6562	av_frame_unref(pic->f);
307
308		784681	av_buffer_unref(&pic->hwaccel_priv_buf);
309
310	✓✓	784681	if (pic->needs_realloc)
311		1348	ff_free_picture_tables(pic);
312
313		784681	memset((uint8_t)pic + off, 0, sizeof(pic) - off);
314		784681	}
315
316		73725	int ff_update_picture_tables(Picture dst, Picture src)
317			{
318			int i;
319
320			#define UPDATE_TABLE(table) \
321			do { \
322			if (src->table && \
323			(!dst->table \|\| dst->table->buffer != src->table->buffer)) { \
324			av_buffer_unref(&dst->table); \
325			dst->table = av_buffer_ref(src->table); \
326			if (!dst->table) { \
327			ff_free_picture_tables(dst); \
328			return AVERROR(ENOMEM); \
329			} \
330			} \
331			} while (0)
332
333	✓✓✓✓ ✓✓✗✓	73725	UPDATE_TABLE(mb_var_buf);
334	✓✓✓✓ ✓✓✗✓	73725	UPDATE_TABLE(mc_mb_var_buf);
335	✓✓✓✓ ✓✓✗✓	73725	UPDATE_TABLE(mb_mean_buf);
336	✓✗✓✓ ✓✓✗✓	73725	UPDATE_TABLE(mbskip_table_buf);
337	✓✗✓✓ ✓✓✗✓	73725	UPDATE_TABLE(qscale_table_buf);
338	✓✗✓✓ ✓✓✗✓	73725	UPDATE_TABLE(mb_type_buf);
339	✓✓	221175	for (i = 0; i < 2; i++) {
340	✓✓✓✓ ✓✓✗✓	147450	UPDATE_TABLE(motion_val_buf[i]);
341	✓✓✓✓ ✓✓✗✓	147450	UPDATE_TABLE(ref_index_buf[i]);
342			}
343
344		73725	dst->mb_var = src->mb_var;
345		73725	dst->mc_mb_var = src->mc_mb_var;
346		73725	dst->mb_mean = src->mb_mean;
347		73725	dst->mbskip_table = src->mbskip_table;
348		73725	dst->qscale_table = src->qscale_table;
349		73725	dst->mb_type = src->mb_type;
350	✓✓	221175	for (i = 0; i < 2; i++) {
351		147450	dst->motion_val[i] = src->motion_val[i];
352		147450	dst->ref_index[i] = src->ref_index[i];
353			}
354
355		73725	dst->alloc_mb_width = src->alloc_mb_width;
356		73725	dst->alloc_mb_height = src->alloc_mb_height;
357
358		73725	return 0;
359			}
360
361		73725	int ff_mpeg_ref_picture(AVCodecContext avctx, Picture dst, Picture *src)
362			{
363			int ret;
364
365	✗✓	73725	av_assert0(!dst->f->buf[0]);
366	✗✓	73725	av_assert0(src->f->buf[0]);
367
368		73725	src->tf.f = src->f;
369		73725	dst->tf.f = dst->f;
370		73725	ret = ff_thread_ref_frame(&dst->tf, &src->tf);
371	✗✓	73725	if (ret < 0)
372			goto fail;
373
374		73725	ret = ff_update_picture_tables(dst, src);
375	✗✓	73725	if (ret < 0)
376			goto fail;
377
378	✗✓	73725	if (src->hwaccel_picture_private) {
379			dst->hwaccel_priv_buf = av_buffer_ref(src->hwaccel_priv_buf);
380			if (!dst->hwaccel_priv_buf) {
381			ret = AVERROR(ENOMEM);
382			goto fail;
383			}
384			dst->hwaccel_picture_private = dst->hwaccel_priv_buf->data;
385			}
386
387		73725	dst->field_picture = src->field_picture;
388		73725	dst->mb_var_sum = src->mb_var_sum;
389		73725	dst->mc_mb_var_sum = src->mc_mb_var_sum;
390		73725	dst->b_frame_score = src->b_frame_score;
391		73725	dst->needs_realloc = src->needs_realloc;
392		73725	dst->reference = src->reference;
393		73725	dst->shared = src->shared;
394
395		73725	memcpy(dst->encoding_error, src->encoding_error,
396			sizeof(dst->encoding_error));
397
398		73725	return 0;
399			fail:
400			ff_mpeg_unref_picture(avctx, dst);
401			return ret;
402			}
403
404		41113	static inline int pic_is_unused(Picture *pic)
405			{
406	✓✓	41113	if (!pic->f->buf[0])
407		18840	return 1;
408	✓✓✓✗	22273	if (pic->needs_realloc && !(pic->reference & DELAYED_PIC_REF))
409		20	return 1;
410		22253	return 0;
411			}
412
413		19509	static int find_unused_picture(AVCodecContext avctx, Picture picture, int shared)
414			{
415			int i;
416
417	✓✓	19509	if (shared) {
418	✓✗	852	for (i = 0; i < MAX_PICTURE_COUNT; i++) {
419	✓✓	852	if (!picture[i].f->buf[0])
420		649	return i;
421			}
422			} else {
423	✓✗	41113	for (i = 0; i < MAX_PICTURE_COUNT; i++) {
424	✓✓	41113	if (pic_is_unused(&picture[i]))
425		18860	return i;
426			}
427			}
428
429			av_log(avctx, AV_LOG_FATAL,
430			"Internal error, picture buffer overflow\n");
431			/* We could return -1, but the codec would crash trying to draw into a
432			* non-existing frame anyway. This is safer than waiting for a random crash.
433			* Also the return of this is never useful, an encoder must only allocate
434			* as much as allowed in the specification. This has no relationship to how
435			* much libavcodec could allocate (and MAX_PICTURE_COUNT is always large
436			* enough for such valid streams).
437			* Plus, a decoder has to check stream validity and remove frames if too
438			* many reference frames are around. Waiting for "OOM" is not correct at
439			* all. Similarly, missing reference frames have to be replaced by
440			* interpolated/MC frames, anything else is a bug in the codec ...
441			*/
442			abort();
443			return -1;
444			}
445
446		19509	int ff_find_unused_picture(AVCodecContext avctx, Picture picture, int shared)
447			{
448		19509	int ret = find_unused_picture(avctx, picture, shared);
449
450	✓✗✓✗	19509	if (ret >= 0 && ret < MAX_PICTURE_COUNT) {
451	✓✓	19509	if (picture[ret].needs_realloc) {
452		20	picture[ret].needs_realloc = 0;
453		20	ff_free_picture_tables(&picture[ret]);
454		20	ff_mpeg_unref_picture(avctx, &picture[ret]);
455			}
456			}
457		19509	return ret;
458			}
459
460		31690	void ff_free_picture_tables(Picture *pic)
461			{
462			int i;
463
464		31690	pic->alloc_mb_width =
465		31690	pic->alloc_mb_height = 0;
466
467		31690	av_buffer_unref(&pic->mb_var_buf);
468		31690	av_buffer_unref(&pic->mc_mb_var_buf);
469		31690	av_buffer_unref(&pic->mb_mean_buf);
470		31690	av_buffer_unref(&pic->mbskip_table_buf);
471		31690	av_buffer_unref(&pic->qscale_table_buf);
472		31690	av_buffer_unref(&pic->mb_type_buf);
473
474	✓✓	95070	for (i = 0; i < 2; i++) {
475		63380	av_buffer_unref(&pic->motion_val_buf[i]);
476		63380	av_buffer_unref(&pic->ref_index_buf[i]);
477			}
478		31690	}