FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/h264_picture.c
Date: 2024-07-17 14:05:47
Exec Total Coverage
Lines: 101 129 78.3%
Functions: 6 6 100.0%
Branches: 40 66 60.6%

Line Branch Exec Source
1 /*
2 * H.26L/H.264/AVC/JVT/14496-10/... decoder
3 * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 /**
23 * @file
24 * H.264 / AVC / MPEG-4 part10 codec.
25 * @author Michael Niedermayer <michaelni@gmx.at>
26 */
27
28 #include "libavutil/avassert.h"
29 #include "libavutil/emms.h"
30 #include "error_resilience.h"
31 #include "avcodec.h"
32 #include "h264dec.h"
33 #include "hwaccel_internal.h"
34 #include "mpegutils.h"
35 #include "refstruct.h"
36 #include "thread.h"
37 #include "threadframe.h"
38
39 134403 void ff_h264_unref_picture(H264Picture *pic)
40 {
41 134403 int off = offsetof(H264Picture, f_grain) + sizeof(pic->f_grain);
42 int i;
43
44
3/4
✓ Branch 0 taken 134403 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 81964 times.
✓ Branch 3 taken 52439 times.
134403 if (!pic->f || !pic->f->buf[0])
45 81964 return;
46
47 52439 ff_thread_release_ext_buffer(&pic->tf);
48 52439 av_frame_unref(pic->f_grain);
49 52439 ff_refstruct_unref(&pic->hwaccel_picture_private);
50
51 52439 ff_refstruct_unref(&pic->qscale_table_base);
52 52439 ff_refstruct_unref(&pic->mb_type_base);
53 52439 ff_refstruct_unref(&pic->pps);
54
2/2
✓ Branch 0 taken 104878 times.
✓ Branch 1 taken 52439 times.
157317 for (i = 0; i < 2; i++) {
55 104878 ff_refstruct_unref(&pic->motion_val_base[i]);
56 104878 ff_refstruct_unref(&pic->ref_index[i]);
57 }
58 52439 ff_refstruct_unref(&pic->decode_error_flags);
59
60 52439 memset((uint8_t*)pic + off, 0, sizeof(*pic) - off);
61 }
62
63 27052 static void h264_copy_picture_params(H264Picture *dst, const H264Picture *src)
64 {
65 27052 ff_refstruct_replace(&dst->qscale_table_base, src->qscale_table_base);
66 27052 ff_refstruct_replace(&dst->mb_type_base, src->mb_type_base);
67 27052 ff_refstruct_replace(&dst->pps, src->pps);
68
69
2/2
✓ Branch 0 taken 54104 times.
✓ Branch 1 taken 27052 times.
81156 for (int i = 0; i < 2; i++) {
70 54104 ff_refstruct_replace(&dst->motion_val_base[i], src->motion_val_base[i]);
71 54104 ff_refstruct_replace(&dst->ref_index[i], src->ref_index[i]);
72 }
73
74 27052 ff_refstruct_replace(&dst->hwaccel_picture_private,
75 27052 src->hwaccel_picture_private);
76
77 27052 ff_refstruct_replace(&dst->decode_error_flags, src->decode_error_flags);
78
79 27052 dst->qscale_table = src->qscale_table;
80 27052 dst->mb_type = src->mb_type;
81
82
2/2
✓ Branch 0 taken 54104 times.
✓ Branch 1 taken 27052 times.
81156 for (int i = 0; i < 2; i++)
83 54104 dst->motion_val[i] = src->motion_val[i];
84
85
2/2
✓ Branch 0 taken 54104 times.
✓ Branch 1 taken 27052 times.
81156 for (int i = 0; i < 2; i++)
86 54104 dst->field_poc[i] = src->field_poc[i];
87
88 27052 memcpy(dst->ref_poc, src->ref_poc, sizeof(src->ref_poc));
89 27052 memcpy(dst->ref_count, src->ref_count, sizeof(src->ref_count));
90
91 27052 dst->poc = src->poc;
92 27052 dst->frame_num = src->frame_num;
93 27052 dst->mmco_reset = src->mmco_reset;
94 27052 dst->long_ref = src->long_ref;
95 27052 dst->mbaff = src->mbaff;
96 27052 dst->field_picture = src->field_picture;
97 27052 dst->reference = src->reference;
98 27052 dst->recovered = src->recovered;
99 27052 dst->gray = src->gray;
100 27052 dst->invalid_gap = src->invalid_gap;
101 27052 dst->sei_recovery_frame_cnt = src->sei_recovery_frame_cnt;
102 27052 dst->mb_width = src->mb_width;
103 27052 dst->mb_height = src->mb_height;
104 27052 dst->mb_stride = src->mb_stride;
105 27052 dst->needs_fg = src->needs_fg;
106 27052 }
107
108 26734 int ff_h264_ref_picture(H264Picture *dst, const H264Picture *src)
109 {
110 int ret;
111
112
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26734 times.
26734 av_assert0(!dst->f->buf[0]);
113
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26734 times.
26734 av_assert0(src->f->buf[0]);
114
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26734 times.
26734 av_assert0(src->tf.f == src->f);
115
116 26734 dst->tf.f = dst->f;
117 26734 ret = ff_thread_ref_frame(&dst->tf, &src->tf);
118
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26734 times.
26734 if (ret < 0)
119 goto fail;
120
121
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26734 times.
26734 if (src->needs_fg) {
122 ret = av_frame_ref(dst->f_grain, src->f_grain);
123 if (ret < 0)
124 goto fail;
125 }
126
127 26734 h264_copy_picture_params(dst, src);
128
129 26734 return 0;
130 fail:
131 ff_h264_unref_picture(dst);
132 return ret;
133 }
134
135 2294 int ff_h264_replace_picture(H264Picture *dst, const H264Picture *src)
136 {
137 int ret;
138
139
3/4
✓ Branch 0 taken 2294 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1976 times.
✓ Branch 3 taken 318 times.
2294 if (!src->f || !src->f->buf[0]) {
140 1976 ff_h264_unref_picture(dst);
141 1976 return 0;
142 }
143
144
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 318 times.
318 av_assert0(src->tf.f == src->f);
145
146 318 dst->tf.f = dst->f;
147 318 ret = ff_thread_replace_frame(&dst->tf, &src->tf);
148
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 318 times.
318 if (ret < 0)
149 goto fail;
150
151
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 318 times.
318 if (src->needs_fg) {
152 av_frame_unref(dst->f_grain);
153 ret = av_frame_ref(dst->f_grain, src->f_grain);
154 if (ret < 0)
155 goto fail;
156 }
157
158 318 h264_copy_picture_params(dst, src);
159
160 318 return 0;
161 fail:
162 ff_h264_unref_picture(dst);
163 return ret;
164 }
165
166 137516 void ff_h264_set_erpic(ERPicture *dst, const H264Picture *src)
167 {
168 #if CONFIG_ERROR_RESILIENCE
169 int i;
170
171 137516 memset(dst, 0, sizeof(*dst));
172
173
2/2
✓ Branch 0 taken 87505 times.
✓ Branch 1 taken 50011 times.
137516 if (!src)
174 87505 return;
175
176 50011 dst->f = src->f;
177 50011 dst->tf = &src->tf;
178
179
2/2
✓ Branch 0 taken 100022 times.
✓ Branch 1 taken 50011 times.
150033 for (i = 0; i < 2; i++) {
180 100022 dst->motion_val[i] = src->motion_val[i];
181 100022 dst->ref_index[i] = src->ref_index[i];
182 }
183
184 50011 dst->mb_type = src->mb_type;
185 50011 dst->field_picture = src->field_picture;
186 #endif /* CONFIG_ERROR_RESILIENCE */
187 }
188
189 28287 int ff_h264_field_end(H264Context *h, H264SliceContext *sl, int in_setup)
190 {
191 28287 AVCodecContext *const avctx = h->avctx;
192 28287 H264Picture *cur = h->cur_pic_ptr;
193 28287 int err = 0;
194 28287 h->mb_y = 0;
195
196
4/4
✓ Branch 0 taken 28093 times.
✓ Branch 1 taken 194 times.
✓ Branch 2 taken 28051 times.
✓ Branch 3 taken 42 times.
28287 if (in_setup || !(avctx->active_thread_type & FF_THREAD_FRAME)) {
197
2/2
✓ Branch 0 taken 18457 times.
✓ Branch 1 taken 9788 times.
28245 if (!h->droppable) {
198 18457 err = ff_h264_execute_ref_pic_marking(h);
199 18457 h->poc.prev_poc_msb = h->poc.poc_msb;
200 18457 h->poc.prev_poc_lsb = h->poc.poc_lsb;
201 }
202 28245 h->poc.prev_frame_num_offset = h->poc.frame_num_offset;
203 28245 h->poc.prev_frame_num = h->poc.frame_num;
204 }
205
206
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28287 times.
28287 if (avctx->hwaccel) {
207 err = FF_HW_SIMPLE_CALL(avctx, end_frame);
208 if (err < 0)
209 av_log(avctx, AV_LOG_ERROR,
210 "hardware accelerator failed to decode picture\n");
211
3/8
✓ Branch 0 taken 28093 times.
✓ Branch 1 taken 194 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 28093 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
28287 } else if (!in_setup && cur->needs_fg && (!FIELD_PICTURE(h) || !h->first_field)) {
212 const AVFrameSideData *sd = av_frame_get_side_data(cur->f, AV_FRAME_DATA_FILM_GRAIN_PARAMS);
213
214 err = AVERROR_INVALIDDATA;
215 if (sd) // a decoding error may have happened before the side data could be allocated
216 err = ff_h274_apply_film_grain(cur->f_grain, cur->f, &h->h274db,
217 (AVFilmGrainParams *) sd->data);
218 if (err < 0) {
219 av_log(h->avctx, AV_LOG_WARNING, "Failed synthesizing film "
220 "grain, ignoring: %s\n", av_err2str(err));
221 cur->needs_fg = 0;
222 err = 0;
223 }
224 }
225
226
4/4
✓ Branch 0 taken 28093 times.
✓ Branch 1 taken 194 times.
✓ Branch 2 taken 18415 times.
✓ Branch 3 taken 9678 times.
28287 if (!in_setup && !h->droppable)
227 18415 ff_thread_report_progress(&cur->tf, INT_MAX,
228 18415 h->picture_structure == PICT_BOTTOM_FIELD);
229 28287 emms_c();
230
231 28287 h->current_slice = 0;
232
233 28287 return err;
234 }
235