FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/h264_picture.c
Date: 2024-02-16 17:37:06
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 134411 void ff_h264_unref_picture(H264Picture *pic)
40 {
41 134411 int off = offsetof(H264Picture, f_grain) + sizeof(pic->f_grain);
42 int i;
43
44
3/4
✓ Branch 0 taken 134411 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 81968 times.
✓ Branch 3 taken 52443 times.
134411 if (!pic->f || !pic->f->buf[0])
45 81968 return;
46
47 52443 ff_thread_release_ext_buffer(&pic->tf);
48 52443 av_frame_unref(pic->f_grain);
49 52443 ff_refstruct_unref(&pic->hwaccel_picture_private);
50
51 52443 ff_refstruct_unref(&pic->qscale_table_base);
52 52443 ff_refstruct_unref(&pic->mb_type_base);
53 52443 ff_refstruct_unref(&pic->pps);
54
2/2
✓ Branch 0 taken 104886 times.
✓ Branch 1 taken 52443 times.
157329 for (i = 0; i < 2; i++) {
55 104886 ff_refstruct_unref(&pic->motion_val_base[i]);
56 104886 ff_refstruct_unref(&pic->ref_index[i]);
57 }
58 52443 ff_refstruct_unref(&pic->decode_error_flags);
59
60 52443 memset((uint8_t*)pic + off, 0, sizeof(*pic) - off);
61 }
62
63 27054 static void h264_copy_picture_params(H264Picture *dst, const H264Picture *src)
64 {
65 27054 ff_refstruct_replace(&dst->qscale_table_base, src->qscale_table_base);
66 27054 ff_refstruct_replace(&dst->mb_type_base, src->mb_type_base);
67 27054 ff_refstruct_replace(&dst->pps, src->pps);
68
69
2/2
✓ Branch 0 taken 54108 times.
✓ Branch 1 taken 27054 times.
81162 for (int i = 0; i < 2; i++) {
70 54108 ff_refstruct_replace(&dst->motion_val_base[i], src->motion_val_base[i]);
71 54108 ff_refstruct_replace(&dst->ref_index[i], src->ref_index[i]);
72 }
73
74 27054 ff_refstruct_replace(&dst->hwaccel_picture_private,
75 27054 src->hwaccel_picture_private);
76
77 27054 ff_refstruct_replace(&dst->decode_error_flags, src->decode_error_flags);
78
79 27054 dst->qscale_table = src->qscale_table;
80 27054 dst->mb_type = src->mb_type;
81
82
2/2
✓ Branch 0 taken 54108 times.
✓ Branch 1 taken 27054 times.
81162 for (int i = 0; i < 2; i++)
83 54108 dst->motion_val[i] = src->motion_val[i];
84
85
2/2
✓ Branch 0 taken 54108 times.
✓ Branch 1 taken 27054 times.
81162 for (int i = 0; i < 2; i++)
86 54108 dst->field_poc[i] = src->field_poc[i];
87
88 27054 memcpy(dst->ref_poc, src->ref_poc, sizeof(src->ref_poc));
89 27054 memcpy(dst->ref_count, src->ref_count, sizeof(src->ref_count));
90
91 27054 dst->poc = src->poc;
92 27054 dst->frame_num = src->frame_num;
93 27054 dst->mmco_reset = src->mmco_reset;
94 27054 dst->long_ref = src->long_ref;
95 27054 dst->mbaff = src->mbaff;
96 27054 dst->field_picture = src->field_picture;
97 27054 dst->reference = src->reference;
98 27054 dst->recovered = src->recovered;
99 27054 dst->gray = src->gray;
100 27054 dst->invalid_gap = src->invalid_gap;
101 27054 dst->sei_recovery_frame_cnt = src->sei_recovery_frame_cnt;
102 27054 dst->mb_width = src->mb_width;
103 27054 dst->mb_height = src->mb_height;
104 27054 dst->mb_stride = src->mb_stride;
105 27054 dst->needs_fg = src->needs_fg;
106 27054 }
107
108 26736 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 26736 times.
26736 av_assert0(!dst->f->buf[0]);
113
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26736 times.
26736 av_assert0(src->f->buf[0]);
114
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26736 times.
26736 av_assert0(src->tf.f == src->f);
115
116 26736 dst->tf.f = dst->f;
117 26736 ret = ff_thread_ref_frame(&dst->tf, &src->tf);
118
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26736 times.
26736 if (ret < 0)
119 goto fail;
120
121
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26736 times.
26736 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 26736 h264_copy_picture_params(dst, src);
128
129 26736 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 137528 void ff_h264_set_erpic(ERPicture *dst, const H264Picture *src)
167 {
168 #if CONFIG_ERROR_RESILIENCE
169 int i;
170
171 137528 memset(dst, 0, sizeof(*dst));
172
173
2/2
✓ Branch 0 taken 87511 times.
✓ Branch 1 taken 50017 times.
137528 if (!src)
174 87511 return;
175
176 50017 dst->f = src->f;
177 50017 dst->tf = &src->tf;
178
179
2/2
✓ Branch 0 taken 100034 times.
✓ Branch 1 taken 50017 times.
150051 for (i = 0; i < 2; i++) {
180 100034 dst->motion_val[i] = src->motion_val[i];
181 100034 dst->ref_index[i] = src->ref_index[i];
182 }
183
184 50017 dst->mb_type = src->mb_type;
185 50017 dst->field_picture = src->field_picture;
186 #endif /* CONFIG_ERROR_RESILIENCE */
187 }
188
189 28289 int ff_h264_field_end(H264Context *h, H264SliceContext *sl, int in_setup)
190 {
191 28289 AVCodecContext *const avctx = h->avctx;
192 28289 H264Picture *cur = h->cur_pic_ptr;
193 28289 int err = 0;
194 28289 h->mb_y = 0;
195
196
4/4
✓ Branch 0 taken 28095 times.
✓ Branch 1 taken 194 times.
✓ Branch 2 taken 28053 times.
✓ Branch 3 taken 42 times.
28289 if (in_setup || !(avctx->active_thread_type & FF_THREAD_FRAME)) {
197
2/2
✓ Branch 0 taken 18458 times.
✓ Branch 1 taken 9789 times.
28247 if (!h->droppable) {
198 18458 err = ff_h264_execute_ref_pic_marking(h);
199 18458 h->poc.prev_poc_msb = h->poc.poc_msb;
200 18458 h->poc.prev_poc_lsb = h->poc.poc_lsb;
201 }
202 28247 h->poc.prev_frame_num_offset = h->poc.frame_num_offset;
203 28247 h->poc.prev_frame_num = h->poc.frame_num;
204 }
205
206
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28289 times.
28289 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 28095 times.
✓ Branch 1 taken 194 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 28095 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
28289 } 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 28095 times.
✓ Branch 1 taken 194 times.
✓ Branch 2 taken 18416 times.
✓ Branch 3 taken 9679 times.
28289 if (!in_setup && !h->droppable)
227 18416 ff_thread_report_progress(&cur->tf, INT_MAX,
228 18416 h->picture_structure == PICT_BOTTOM_FIELD);
229 28289 emms_c();
230
231 28289 h->current_slice = 0;
232
233 28289 return err;
234 }
235