FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/h264_picture.c
Date: 2024-11-20 23:03:26
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 134404 void ff_h264_unref_picture(H264Picture *pic)
40 {
41 134404 int off = offsetof(H264Picture, f_grain) + sizeof(pic->f_grain);
42 int i;
43
44
3/4
✓ Branch 0 taken 134404 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 81941 times.
✓ Branch 3 taken 52463 times.
134404 if (!pic->f || !pic->f->buf[0])
45 81941 return;
46
47 52463 ff_thread_release_ext_buffer(&pic->tf);
48 52463 av_frame_unref(pic->f_grain);
49 52463 ff_refstruct_unref(&pic->hwaccel_picture_private);
50
51 52463 ff_refstruct_unref(&pic->qscale_table_base);
52 52463 ff_refstruct_unref(&pic->mb_type_base);
53 52463 ff_refstruct_unref(&pic->pps);
54
2/2
✓ Branch 0 taken 104926 times.
✓ Branch 1 taken 52463 times.
157389 for (i = 0; i < 2; i++) {
55 104926 ff_refstruct_unref(&pic->motion_val_base[i]);
56 104926 ff_refstruct_unref(&pic->ref_index[i]);
57 }
58 52463 ff_refstruct_unref(&pic->decode_error_flags);
59
60 52463 memset((uint8_t*)pic + off, 0, sizeof(*pic) - off);
61 }
62
63 27045 static void h264_copy_picture_params(H264Picture *dst, const H264Picture *src)
64 {
65 27045 ff_refstruct_replace(&dst->qscale_table_base, src->qscale_table_base);
66 27045 ff_refstruct_replace(&dst->mb_type_base, src->mb_type_base);
67 27045 ff_refstruct_replace(&dst->pps, src->pps);
68
69
2/2
✓ Branch 0 taken 54090 times.
✓ Branch 1 taken 27045 times.
81135 for (int i = 0; i < 2; i++) {
70 54090 ff_refstruct_replace(&dst->motion_val_base[i], src->motion_val_base[i]);
71 54090 ff_refstruct_replace(&dst->ref_index[i], src->ref_index[i]);
72 }
73
74 27045 ff_refstruct_replace(&dst->hwaccel_picture_private,
75 27045 src->hwaccel_picture_private);
76
77 27045 ff_refstruct_replace(&dst->decode_error_flags, src->decode_error_flags);
78
79 27045 dst->qscale_table = src->qscale_table;
80 27045 dst->mb_type = src->mb_type;
81
82
2/2
✓ Branch 0 taken 54090 times.
✓ Branch 1 taken 27045 times.
81135 for (int i = 0; i < 2; i++)
83 54090 dst->motion_val[i] = src->motion_val[i];
84
85
2/2
✓ Branch 0 taken 54090 times.
✓ Branch 1 taken 27045 times.
81135 for (int i = 0; i < 2; i++)
86 54090 dst->field_poc[i] = src->field_poc[i];
87
88 27045 memcpy(dst->ref_poc, src->ref_poc, sizeof(src->ref_poc));
89 27045 memcpy(dst->ref_count, src->ref_count, sizeof(src->ref_count));
90
91 27045 dst->poc = src->poc;
92 27045 dst->frame_num = src->frame_num;
93 27045 dst->mmco_reset = src->mmco_reset;
94 27045 dst->long_ref = src->long_ref;
95 27045 dst->mbaff = src->mbaff;
96 27045 dst->field_picture = src->field_picture;
97 27045 dst->reference = src->reference;
98 27045 dst->recovered = src->recovered;
99 27045 dst->gray = src->gray;
100 27045 dst->invalid_gap = src->invalid_gap;
101 27045 dst->sei_recovery_frame_cnt = src->sei_recovery_frame_cnt;
102 27045 dst->mb_width = src->mb_width;
103 27045 dst->mb_height = src->mb_height;
104 27045 dst->mb_stride = src->mb_stride;
105 27045 dst->needs_fg = src->needs_fg;
106 27045 }
107
108 26747 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 26747 times.
26747 av_assert0(!dst->f->buf[0]);
113
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26747 times.
26747 av_assert0(src->f->buf[0]);
114
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26747 times.
26747 av_assert0(src->tf.f == src->f);
115
116 26747 dst->tf.f = dst->f;
117 26747 ret = ff_thread_ref_frame(&dst->tf, &src->tf);
118
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26747 times.
26747 if (ret < 0)
119 goto fail;
120
121
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26747 times.
26747 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 26747 h264_copy_picture_params(dst, src);
128
129 26747 return 0;
130 fail:
131 ff_h264_unref_picture(dst);
132 return ret;
133 }
134
135 2146 int ff_h264_replace_picture(H264Picture *dst, const H264Picture *src)
136 {
137 int ret;
138
139
3/4
✓ Branch 0 taken 2146 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1848 times.
✓ Branch 3 taken 298 times.
2146 if (!src->f || !src->f->buf[0]) {
140 1848 ff_h264_unref_picture(dst);
141 1848 return 0;
142 }
143
144
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 298 times.
298 av_assert0(src->tf.f == src->f);
145
146 298 dst->tf.f = dst->f;
147 298 ret = ff_thread_replace_frame(&dst->tf, &src->tf);
148
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 298 times.
298 if (ret < 0)
149 goto fail;
150
151
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 298 times.
298 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 298 h264_copy_picture_params(dst, src);
159
160 298 return 0;
161 fail:
162 ff_h264_unref_picture(dst);
163 return ret;
164 }
165
166 137565 void ff_h264_set_erpic(ERPicture *dst, const H264Picture *src)
167 {
168 #if CONFIG_ERROR_RESILIENCE
169 int i;
170
171 137565 memset(dst, 0, sizeof(*dst));
172
173
2/2
✓ Branch 0 taken 87540 times.
✓ Branch 1 taken 50025 times.
137565 if (!src)
174 87540 return;
175
176 50025 dst->f = src->f;
177 50025 dst->tf = &src->tf;
178
179
2/2
✓ Branch 0 taken 100050 times.
✓ Branch 1 taken 50025 times.
150075 for (i = 0; i < 2; i++) {
180 100050 dst->motion_val[i] = src->motion_val[i];
181 100050 dst->ref_index[i] = src->ref_index[i];
182 }
183
184 50025 dst->mb_type = src->mb_type;
185 50025 dst->field_picture = src->field_picture;
186 #endif /* CONFIG_ERROR_RESILIENCE */
187 }
188
189 28303 int ff_h264_field_end(H264Context *h, H264SliceContext *sl, int in_setup)
190 {
191 28303 AVCodecContext *const avctx = h->avctx;
192 28303 H264Picture *cur = h->cur_pic_ptr;
193 28303 int err = 0;
194 28303 h->mb_y = 0;
195
196
4/4
✓ Branch 0 taken 28109 times.
✓ Branch 1 taken 194 times.
✓ Branch 2 taken 28067 times.
✓ Branch 3 taken 42 times.
28303 if (in_setup || !(avctx->active_thread_type & FF_THREAD_FRAME)) {
197
2/2
✓ Branch 0 taken 18466 times.
✓ Branch 1 taken 9795 times.
28261 if (!h->droppable) {
198 18466 err = ff_h264_execute_ref_pic_marking(h);
199 18466 h->poc.prev_poc_msb = h->poc.poc_msb;
200 18466 h->poc.prev_poc_lsb = h->poc.poc_lsb;
201 }
202 28261 h->poc.prev_frame_num_offset = h->poc.frame_num_offset;
203 28261 h->poc.prev_frame_num = h->poc.frame_num;
204 }
205
206
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28303 times.
28303 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 28109 times.
✓ Branch 1 taken 194 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 28109 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
28303 } 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 28109 times.
✓ Branch 1 taken 194 times.
✓ Branch 2 taken 18424 times.
✓ Branch 3 taken 9685 times.
28303 if (!in_setup && !h->droppable)
227 18424 ff_thread_report_progress(&cur->tf, INT_MAX,
228 18424 h->picture_structure == PICT_BOTTOM_FIELD);
229 28303 emms_c();
230
231 28303 h->current_slice = 0;
232
233 28303 return err;
234 }
235