FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/vp8.c
Date: 2025-04-25 22:50:00
Exec Total Coverage
Lines: 1372 1603 85.6%
Functions: 77 84 91.7%
Branches: 817 1159 70.5%

Line Branch Exec Source
1 /*
2 * VP7/VP8 compatible video decoder
3 *
4 * Copyright (C) 2010 David Conrad
5 * Copyright (C) 2010 Ronald S. Bultje
6 * Copyright (C) 2010 Fiona Glaser
7 * Copyright (C) 2012 Daniel Kang
8 * Copyright (C) 2014 Peter Ross
9 *
10 * This file is part of FFmpeg.
11 *
12 * FFmpeg is free software; you can redistribute it and/or
13 * modify it under the terms of the GNU Lesser General Public
14 * License as published by the Free Software Foundation; either
15 * version 2.1 of the License, or (at your option) any later version.
16 *
17 * FFmpeg is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20 * Lesser General Public License for more details.
21 *
22 * You should have received a copy of the GNU Lesser General Public
23 * License along with FFmpeg; if not, write to the Free Software
24 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
25 */
26
27 #include "config_components.h"
28
29 #include "libavutil/mem.h"
30 #include "libavutil/mem_internal.h"
31
32 #include "avcodec.h"
33 #include "codec_internal.h"
34 #include "decode.h"
35 #include "hwaccel_internal.h"
36 #include "hwconfig.h"
37 #include "mathops.h"
38 #include "progressframe.h"
39 #include "libavutil/refstruct.h"
40 #include "thread.h"
41 #include "vp8.h"
42 #include "vp89_rac.h"
43 #include "vp8data.h"
44 #include "vpx_rac.h"
45
46 #if ARCH_ARM
47 # include "arm/vp8.h"
48 #endif
49
50 // fixme: add 1 bit to all the calls to this?
51 6144 static int vp8_rac_get_sint(VPXRangeCoder *c, int bits)
52 {
53 int v;
54
55
2/2
✓ Branch 1 taken 5393 times.
✓ Branch 2 taken 751 times.
6144 if (!vp89_rac_get(c))
56 5393 return 0;
57
58 751 v = vp89_rac_get_uint(c, bits);
59
60
2/2
✓ Branch 1 taken 98 times.
✓ Branch 2 taken 653 times.
751 if (vp89_rac_get(c))
61 98 v = -v;
62
63 751 return v;
64 }
65
66 306 static int vp8_rac_get_nn(VPXRangeCoder *c)
67 {
68 306 int v = vp89_rac_get_uint(c, 7) << 1;
69 306 return v + !v;
70 }
71
72 // DCTextra
73 51167 static int vp8_rac_get_coeff(VPXRangeCoder *c, const uint8_t *prob)
74 {
75 51167 int v = 0;
76
77 do {
78 197155 v = (v<<1) + vpx_rac_get_prob(c, *prob++);
79
2/2
✓ Branch 0 taken 145988 times.
✓ Branch 1 taken 51167 times.
197155 } while (*prob);
80
81 51167 return v;
82 }
83
84 79 static void free_buffers(VP8Context *s)
85 {
86 int i;
87
2/2
✓ Branch 0 taken 40 times.
✓ Branch 1 taken 39 times.
79 if (s->thread_data)
88
2/2
✓ Branch 0 taken 320 times.
✓ Branch 1 taken 40 times.
360 for (i = 0; i < MAX_THREADS; i++) {
89 #if HAVE_THREADS
90 320 pthread_cond_destroy(&s->thread_data[i].cond);
91 320 pthread_mutex_destroy(&s->thread_data[i].lock);
92 #endif
93 320 av_freep(&s->thread_data[i].filter_strength);
94 }
95 79 av_freep(&s->thread_data);
96 79 av_freep(&s->macroblocks_base);
97 79 av_freep(&s->intra4x4_pred_mode_top);
98 79 av_freep(&s->top_nnz);
99 79 av_freep(&s->top_border);
100
101 79 s->macroblocks = NULL;
102 79 }
103
104 1151 static int vp8_alloc_frame(VP8Context *s, VP8Frame *f, int ref)
105 {
106 1151 int ret = ff_progress_frame_get_buffer(s->avctx, &f->tf,
107 ref ? AV_GET_BUFFER_FLAG_REF : 0);
108
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1151 times.
1151 if (ret < 0)
109 return ret;
110 1151 f->seg_map = av_refstruct_allocz(s->mb_width * s->mb_height);
111
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1151 times.
1151 if (!f->seg_map) {
112 ret = AVERROR(ENOMEM);
113 goto fail;
114 }
115 1151 ret = ff_hwaccel_frame_priv_alloc(s->avctx, &f->hwaccel_picture_private);
116
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1151 times.
1151 if (ret < 0)
117 goto fail;
118
119 1151 return 0;
120
121 fail:
122 av_refstruct_unref(&f->seg_map);
123 ff_progress_frame_unref(&f->tf);
124 return ret;
125 }
126
127 1429 static void vp8_release_frame(VP8Frame *f)
128 {
129 1429 av_refstruct_unref(&f->seg_map);
130 1429 av_refstruct_unref(&f->hwaccel_picture_private);
131 1429 ff_progress_frame_unref(&f->tf);
132 1429 }
133
134 79 static av_cold void vp8_decode_flush_impl(AVCodecContext *avctx, int free_mem)
135 {
136 79 VP8Context *s = avctx->priv_data;
137 int i;
138
139
2/2
✓ Branch 0 taken 395 times.
✓ Branch 1 taken 79 times.
474 for (i = 0; i < FF_ARRAY_ELEMS(s->frames); i++)
140 395 vp8_release_frame(&s->frames[i]);
141 79 memset(s->framep, 0, sizeof(s->framep));
142
143
1/2
✓ Branch 0 taken 79 times.
✗ Branch 1 not taken.
79 if (free_mem)
144 79 free_buffers(s);
145
146
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 79 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
79 if (FF_HW_HAS_CB(avctx, flush))
147 FF_HW_SIMPLE_CALL(avctx, flush);
148 79 }
149
150 static av_cold void vp8_decode_flush(AVCodecContext *avctx)
151 {
152 vp8_decode_flush_impl(avctx, 0);
153 }
154
155 1151 static VP8Frame *vp8_find_free_buffer(VP8Context *s)
156 {
157 1151 VP8Frame *frame = NULL;
158 int i;
159
160 // find a free buffer
161
1/2
✓ Branch 0 taken 2810 times.
✗ Branch 1 not taken.
2810 for (i = 0; i < 5; i++)
162
2/2
✓ Branch 0 taken 2214 times.
✓ Branch 1 taken 596 times.
2810 if (&s->frames[i] != s->framep[VP8_FRAME_CURRENT] &&
163
2/2
✓ Branch 0 taken 2213 times.
✓ Branch 1 taken 1 times.
2214 &s->frames[i] != s->framep[VP8_FRAME_PREVIOUS] &&
164
2/2
✓ Branch 0 taken 1609 times.
✓ Branch 1 taken 604 times.
2213 &s->frames[i] != s->framep[VP8_FRAME_GOLDEN] &&
165
2/2
✓ Branch 0 taken 1151 times.
✓ Branch 1 taken 458 times.
1609 &s->frames[i] != s->framep[VP8_FRAME_ALTREF]) {
166 1151 frame = &s->frames[i];
167 1151 break;
168 }
169
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1151 times.
1151 if (i == 5) {
170 av_log(s->avctx, AV_LOG_FATAL, "Ran out of free frames!\n");
171 abort();
172 }
173
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1151 times.
1151 if (frame->tf.f)
174 vp8_release_frame(frame);
175
176 1151 return frame;
177 }
178
179 21 static enum AVPixelFormat get_pixel_format(VP8Context *s)
180 {
181 21 enum AVPixelFormat pix_fmts[] = {
182 #if CONFIG_VP8_VAAPI_HWACCEL
183 AV_PIX_FMT_VAAPI,
184 #endif
185 #if CONFIG_VP8_NVDEC_HWACCEL
186 AV_PIX_FMT_CUDA,
187 #endif
188 AV_PIX_FMT_YUV420P,
189 AV_PIX_FMT_NONE,
190 };
191
192 21 return ff_get_format(s->avctx, pix_fmts);
193 }
194
195 static av_always_inline
196 40 int update_dimensions(VP8Context *s, int width, int height, int is_vp7)
197 {
198 40 AVCodecContext *avctx = s->avctx;
199 40 int i, ret, dim_reset = 0;
200
201
7/8
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 6 times.
✓ Branch 3 taken 24 times.
✓ Branch 4 taken 6 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 24 times.
✓ Branch 7 taken 6 times.
40 if (width != s->avctx->width || ((width+15)/16 != s->mb_width || (height+15)/16 != s->mb_height) && s->macroblocks_base ||
202
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24 times.
24 height != s->avctx->height) {
203 16 vp8_decode_flush_impl(s->avctx, 1);
204
205 16 ret = ff_set_dimensions(s->avctx, width, height);
206
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
16 if (ret < 0)
207 return ret;
208
209 16 dim_reset = (s->macroblocks_base != NULL);
210 }
211
212
3/4
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 28 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 12 times.
40 if ((s->pix_fmt == AV_PIX_FMT_NONE || dim_reset) &&
213
4/4
✓ Branch 0 taken 22 times.
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 21 times.
✓ Branch 3 taken 1 times.
28 !s->actually_webp && !is_vp7) {
214 21 s->pix_fmt = get_pixel_format(s);
215
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 21 times.
21 if (s->pix_fmt < 0)
216 return AVERROR(EINVAL);
217 21 avctx->pix_fmt = s->pix_fmt;
218 }
219
220 40 s->mb_width = (s->avctx->coded_width + 15) / 16;
221 40 s->mb_height = (s->avctx->coded_height + 15) / 16;
222
223
3/4
✓ Branch 0 taken 39 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 39 times.
40 s->mb_layout = is_vp7 || avctx->active_thread_type == FF_THREAD_SLICE &&
224 avctx->thread_count > 1;
225
2/2
✓ Branch 0 taken 39 times.
✓ Branch 1 taken 1 times.
40 if (!s->mb_layout) { // Frame threading and one thread
226 39 s->macroblocks_base = av_mallocz((s->mb_width + s->mb_height * 2 + 1) *
227 sizeof(*s->macroblocks));
228 39 s->intra4x4_pred_mode_top = av_mallocz(s->mb_width * 4);
229 } else // Sliced threading
230 1 s->macroblocks_base = av_mallocz((s->mb_width + 2) * (s->mb_height + 2) *
231 sizeof(*s->macroblocks));
232 40 s->top_nnz = av_mallocz(s->mb_width * sizeof(*s->top_nnz));
233 40 s->top_border = av_mallocz((s->mb_width + 1) * sizeof(*s->top_border));
234 40 s->thread_data = av_mallocz(MAX_THREADS * sizeof(VP8ThreadData));
235
236
3/6
✓ Branch 0 taken 40 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 40 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 40 times.
✗ Branch 5 not taken.
40 if (!s->macroblocks_base || !s->top_nnz || !s->top_border ||
237
4/6
✓ Branch 0 taken 40 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 39 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 1 times.
40 !s->thread_data || (!s->intra4x4_pred_mode_top && !s->mb_layout)) {
238 free_buffers(s);
239 return AVERROR(ENOMEM);
240 }
241
242
2/2
✓ Branch 0 taken 320 times.
✓ Branch 1 taken 40 times.
360 for (i = 0; i < MAX_THREADS; i++) {
243 640 s->thread_data[i].filter_strength =
244 320 av_mallocz(s->mb_width * sizeof(*s->thread_data[0].filter_strength));
245
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 320 times.
320 if (!s->thread_data[i].filter_strength) {
246 free_buffers(s);
247 return AVERROR(ENOMEM);
248 }
249 #if HAVE_THREADS
250 320 ret = pthread_mutex_init(&s->thread_data[i].lock, NULL);
251
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 320 times.
320 if (ret) {
252 free_buffers(s);
253 return AVERROR(ret);
254 }
255 320 ret = pthread_cond_init(&s->thread_data[i].cond, NULL);
256
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 320 times.
320 if (ret) {
257 free_buffers(s);
258 return AVERROR(ret);
259 }
260 #endif
261 }
262
263 40 s->macroblocks = s->macroblocks_base + 1;
264
265 40 return 0;
266 }
267
268 1 static int vp7_update_dimensions(VP8Context *s, int width, int height)
269 {
270 1 return update_dimensions(s, width, height, IS_VP7);
271 }
272
273 39 static int vp8_update_dimensions(VP8Context *s, int width, int height)
274 {
275 39 return update_dimensions(s, width, height, IS_VP8);
276 }
277
278
279 433 static void parse_segment_info(VP8Context *s)
280 {
281 433 VPXRangeCoder *c = &s->c;
282 int i;
283
284 433 s->segmentation.update_map = vp89_rac_get(c);
285 433 s->segmentation.update_feature_data = vp89_rac_get(c);
286
287
2/2
✓ Branch 0 taken 68 times.
✓ Branch 1 taken 365 times.
433 if (s->segmentation.update_feature_data) {
288 68 s->segmentation.absolute_vals = vp89_rac_get(c);
289
290
2/2
✓ Branch 0 taken 272 times.
✓ Branch 1 taken 68 times.
340 for (i = 0; i < 4; i++)
291 272 s->segmentation.base_quant[i] = vp8_rac_get_sint(c, 7);
292
293
2/2
✓ Branch 0 taken 272 times.
✓ Branch 1 taken 68 times.
340 for (i = 0; i < 4; i++)
294 272 s->segmentation.filter_level[i] = vp8_rac_get_sint(c, 6);
295 }
296
2/2
✓ Branch 0 taken 68 times.
✓ Branch 1 taken 365 times.
433 if (s->segmentation.update_map)
297
2/2
✓ Branch 0 taken 204 times.
✓ Branch 1 taken 68 times.
272 for (i = 0; i < 3; i++)
298
2/2
✓ Branch 1 taken 24 times.
✓ Branch 2 taken 180 times.
204 s->prob->segmentid[i] = vp89_rac_get(c) ? vp89_rac_get_uint(c, 8) : 255;
299 433 }
300
301 45 static void update_lf_deltas(VP8Context *s)
302 {
303 45 VPXRangeCoder *c = &s->c;
304 int i;
305
306
2/2
✓ Branch 0 taken 180 times.
✓ Branch 1 taken 45 times.
225 for (i = 0; i < 4; i++) {
307
2/2
✓ Branch 1 taken 135 times.
✓ Branch 2 taken 45 times.
180 if (vp89_rac_get(c)) {
308 135 s->lf_delta.ref[i] = vp89_rac_get_uint(c, 6);
309
310
2/2
✓ Branch 1 taken 90 times.
✓ Branch 2 taken 45 times.
135 if (vp89_rac_get(c))
311 90 s->lf_delta.ref[i] = -s->lf_delta.ref[i];
312 }
313 }
314
315
2/2
✓ Branch 0 taken 180 times.
✓ Branch 1 taken 45 times.
225 for (i = MODE_I4x4; i <= VP8_MVMODE_SPLIT; i++) {
316
1/2
✓ Branch 1 taken 180 times.
✗ Branch 2 not taken.
180 if (vp89_rac_get(c)) {
317 180 s->lf_delta.mode[i] = vp89_rac_get_uint(c, 6);
318
319
2/2
✓ Branch 1 taken 45 times.
✓ Branch 2 taken 135 times.
180 if (vp89_rac_get(c))
320 45 s->lf_delta.mode[i] = -s->lf_delta.mode[i];
321 }
322 }
323 45 }
324
325 1120 static int setup_partitions(VP8Context *s, const uint8_t *buf, int buf_size)
326 {
327 1120 const uint8_t *sizes = buf;
328 int i;
329 int ret;
330
331 1120 s->num_coeff_partitions = 1 << vp89_rac_get_uint(&s->c, 2);
332
333 1120 buf += 3 * (s->num_coeff_partitions - 1);
334 1120 buf_size -= 3 * (s->num_coeff_partitions - 1);
335
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1120 times.
1120 if (buf_size < 0)
336 return -1;
337
338
2/2
✓ Branch 0 taken 339 times.
✓ Branch 1 taken 1120 times.
1459 for (i = 0; i < s->num_coeff_partitions - 1; i++) {
339 339 int size = AV_RL24(sizes + 3 * i);
340
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 339 times.
339 if (buf_size - size < 0)
341 return -1;
342 339 s->coeff_partition_size[i] = size;
343
344 339 ret = ff_vpx_init_range_decoder(&s->coeff_partition[i], buf, size);
345
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 339 times.
339 if (ret < 0)
346 return ret;
347 339 buf += size;
348 339 buf_size -= size;
349 }
350
351 1120 s->coeff_partition_size[i] = buf_size;
352
353 1120 return ff_vpx_init_range_decoder(&s->coeff_partition[i], buf, buf_size);
354 }
355
356 31 static void vp7_get_quants(VP8Context *s)
357 {
358 31 VPXRangeCoder *c = &s->c;
359
360 31 int yac_qi = vp89_rac_get_uint(c, 7);
361
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 int ydc_qi = vp89_rac_get(c) ? vp89_rac_get_uint(c, 7) : yac_qi;
362
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 int y2dc_qi = vp89_rac_get(c) ? vp89_rac_get_uint(c, 7) : yac_qi;
363
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 int y2ac_qi = vp89_rac_get(c) ? vp89_rac_get_uint(c, 7) : yac_qi;
364
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 int uvdc_qi = vp89_rac_get(c) ? vp89_rac_get_uint(c, 7) : yac_qi;
365
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 int uvac_qi = vp89_rac_get(c) ? vp89_rac_get_uint(c, 7) : yac_qi;
366
367 31 s->qmat[0].luma_qmul[0] = vp7_ydc_qlookup[ydc_qi];
368 31 s->qmat[0].luma_qmul[1] = vp7_yac_qlookup[yac_qi];
369 31 s->qmat[0].luma_dc_qmul[0] = vp7_y2dc_qlookup[y2dc_qi];
370 31 s->qmat[0].luma_dc_qmul[1] = vp7_y2ac_qlookup[y2ac_qi];
371 31 s->qmat[0].chroma_qmul[0] = FFMIN(vp7_ydc_qlookup[uvdc_qi], 132);
372 31 s->qmat[0].chroma_qmul[1] = vp7_yac_qlookup[uvac_qi];
373 31 }
374
375 1120 static void vp8_get_quants(VP8Context *s)
376 {
377 1120 VPXRangeCoder *c = &s->c;
378 int i, base_qi;
379
380 1120 s->quant.yac_qi = vp89_rac_get_uint(c, 7);
381 1120 s->quant.ydc_delta = vp8_rac_get_sint(c, 4);
382 1120 s->quant.y2dc_delta = vp8_rac_get_sint(c, 4);
383 1120 s->quant.y2ac_delta = vp8_rac_get_sint(c, 4);
384 1120 s->quant.uvdc_delta = vp8_rac_get_sint(c, 4);
385 1120 s->quant.uvac_delta = vp8_rac_get_sint(c, 4);
386
387
2/2
✓ Branch 0 taken 4480 times.
✓ Branch 1 taken 1120 times.
5600 for (i = 0; i < 4; i++) {
388
2/2
✓ Branch 0 taken 1732 times.
✓ Branch 1 taken 2748 times.
4480 if (s->segmentation.enabled) {
389 1732 base_qi = s->segmentation.base_quant[i];
390
2/2
✓ Branch 0 taken 1608 times.
✓ Branch 1 taken 124 times.
1732 if (!s->segmentation.absolute_vals)
391 1608 base_qi += s->quant.yac_qi;
392 } else
393 2748 base_qi = s->quant.yac_qi;
394
395 4480 s->qmat[i].luma_qmul[0] = vp8_dc_qlookup[av_clip_uintp2(base_qi + s->quant.ydc_delta, 7)];
396 4480 s->qmat[i].luma_qmul[1] = vp8_ac_qlookup[av_clip_uintp2(base_qi, 7)];
397 4480 s->qmat[i].luma_dc_qmul[0] = vp8_dc_qlookup[av_clip_uintp2(base_qi + s->quant.y2dc_delta, 7)] * 2;
398 /* 101581>>16 is equivalent to 155/100 */
399 4480 s->qmat[i].luma_dc_qmul[1] = vp8_ac_qlookup[av_clip_uintp2(base_qi + s->quant.y2ac_delta, 7)] * 101581 >> 16;
400 4480 s->qmat[i].chroma_qmul[0] = vp8_dc_qlookup[av_clip_uintp2(base_qi + s->quant.uvdc_delta, 7)];
401 4480 s->qmat[i].chroma_qmul[1] = vp8_ac_qlookup[av_clip_uintp2(base_qi + s->quant.uvac_delta, 7)];
402
403 4480 s->qmat[i].luma_dc_qmul[1] = FFMAX(s->qmat[i].luma_dc_qmul[1], 8);
404 4480 s->qmat[i].chroma_qmul[0] = FFMIN(s->qmat[i].chroma_qmul[0], 132);
405 }
406 1120 }
407
408 /**
409 * Determine which buffers golden and altref should be updated with after this frame.
410 * The spec isn't clear here, so I'm going by my understanding of what libvpx does
411 *
412 * Intra frames update all 3 references
413 * Inter frames update VP8_FRAME_PREVIOUS if the update_last flag is set
414 * If the update (golden|altref) flag is set, it's updated with the current frame
415 * if update_last is set, and VP8_FRAME_PREVIOUS otherwise.
416 * If the flag is not set, the number read means:
417 * 0: no update
418 * 1: VP8_FRAME_PREVIOUS
419 * 2: update golden with altref, or update altref with golden
420 */
421 2138 static VP8FrameType ref_to_update(VP8Context *s, int update, VP8FrameType ref)
422 {
423 2138 VPXRangeCoder *c = &s->c;
424
425
2/2
✓ Branch 0 taken 90 times.
✓ Branch 1 taken 2048 times.
2138 if (update)
426 90 return VP8_FRAME_CURRENT;
427
428
3/3
✓ Branch 1 taken 31 times.
✓ Branch 2 taken 84 times.
✓ Branch 3 taken 1933 times.
2048 switch (vp89_rac_get_uint(c, 2)) {
429 31 case 1:
430 31 return VP8_FRAME_PREVIOUS;
431 84 case 2:
432
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 81 times.
84 return (ref == VP8_FRAME_GOLDEN) ? VP8_FRAME_ALTREF : VP8_FRAME_GOLDEN;
433 }
434 1933 return VP8_FRAME_NONE;
435 }
436
437 52 static void vp78_reset_probability_tables(VP8Context *s)
438 {
439 int i, j;
440
2/2
✓ Branch 0 taken 208 times.
✓ Branch 1 taken 52 times.
260 for (i = 0; i < 4; i++)
441
2/2
✓ Branch 0 taken 3328 times.
✓ Branch 1 taken 208 times.
3536 for (j = 0; j < 16; j++)
442 3328 memcpy(s->prob->token[i][j], vp8_token_default_probs[i][vp8_coeff_band[j]],
443 sizeof(s->prob->token[i][j]));
444 52 }
445
446 1151 static void vp78_update_probability_tables(VP8Context *s)
447 {
448 1151 VPXRangeCoder *c = &s->c;
449 int i, j, k, l, m;
450
451
2/2
✓ Branch 0 taken 4604 times.
✓ Branch 1 taken 1151 times.
5755 for (i = 0; i < 4; i++)
452
2/2
✓ Branch 0 taken 36832 times.
✓ Branch 1 taken 4604 times.
41436 for (j = 0; j < 8; j++)
453
2/2
✓ Branch 0 taken 110496 times.
✓ Branch 1 taken 36832 times.
147328 for (k = 0; k < 3; k++)
454
2/2
✓ Branch 0 taken 1215456 times.
✓ Branch 1 taken 110496 times.
1325952 for (l = 0; l < NUM_DCT_TOKENS-1; l++)
455
2/2
✓ Branch 1 taken 7805 times.
✓ Branch 2 taken 1207651 times.
1215456 if (vpx_rac_get_prob_branchy(c, ff_vp8_token_update_probs[i][j][k][l])) {
456 7805 int prob = vp89_rac_get_uint(c, 8);
457
2/2
✓ Branch 0 taken 18093 times.
✓ Branch 1 taken 7805 times.
25898 for (m = 0; vp8_coeff_band_indexes[j][m] >= 0; m++)
458 18093 s->prob->token[i][vp8_coeff_band_indexes[j][m]][k][l] = prob;
459 }
460 1151 }
461
462 #define VP7_MVC_SIZE 17
463 #define VP8_MVC_SIZE 19
464
465 1099 static void vp78_update_pred16x16_pred8x8_mvc_probabilities(VP8Context *s,
466 int mvc_size)
467 {
468 1099 VPXRangeCoder *c = &s->c;
469 int i, j;
470
471
2/2
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 1097 times.
1099 if (vp89_rac_get(c))
472
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 2 times.
10 for (i = 0; i < 4; i++)
473 8 s->prob->pred16x16[i] = vp89_rac_get_uint(c, 8);
474
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 1098 times.
1099 if (vp89_rac_get(c))
475
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 1 times.
4 for (i = 0; i < 3; i++)
476 3 s->prob->pred8x8c[i] = vp89_rac_get_uint(c, 8);
477
478 // 17.2 MV probability update
479
2/2
✓ Branch 0 taken 2198 times.
✓ Branch 1 taken 1099 times.
3297 for (i = 0; i < 2; i++)
480
2/2
✓ Branch 0 taken 41642 times.
✓ Branch 1 taken 2198 times.
43840 for (j = 0; j < mvc_size; j++)
481
2/2
✓ Branch 1 taken 306 times.
✓ Branch 2 taken 41336 times.
41642 if (vpx_rac_get_prob_branchy(c, vp8_mv_update_prob[i][j]))
482 306 s->prob->mvc[i][j] = vp8_rac_get_nn(c);
483 1099 }
484
485 1069 static void update_refs(VP8Context *s)
486 {
487 1069 VPXRangeCoder *c = &s->c;
488
489 1069 int update_golden = vp89_rac_get(c);
490 1069 int update_altref = vp89_rac_get(c);
491
492 1069 s->update_golden = ref_to_update(s, update_golden, VP8_FRAME_GOLDEN);
493 1069 s->update_altref = ref_to_update(s, update_altref, VP8_FRAME_ALTREF);
494 1069 }
495
496 static void copy_chroma(AVFrame *dst, const AVFrame *src, int width, int height)
497 {
498 int i, j;
499
500 for (j = 1; j < 3; j++) {
501 for (i = 0; i < height / 2; i++)
502 memcpy(dst->data[j] + i * dst->linesize[j],
503 src->data[j] + i * src->linesize[j], width / 2);
504 }
505 }
506
507 static void fade(uint8_t *dst, ptrdiff_t dst_linesize,
508 const uint8_t *src, ptrdiff_t src_linesize,
509 int width, int height,
510 int alpha, int beta)
511 {
512 int i, j;
513 for (j = 0; j < height; j++) {
514 const uint8_t *src2 = src + j * src_linesize;
515 uint8_t *dst2 = dst + j * dst_linesize;
516 for (i = 0; i < width; i++) {
517 uint8_t y = src2[i];
518 dst2[i] = av_clip_uint8(y + ((y * beta) >> 8) + alpha);
519 }
520 }
521 }
522
523 31 static int vp7_fade_frame(VP8Context *s, int alpha, int beta)
524 {
525 int ret;
526
527
4/6
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 30 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 30 times.
31 if (!s->keyframe && (alpha || beta)) {
528 int width = s->mb_width * 16;
529 int height = s->mb_height * 16;
530 const AVFrame *src;
531 AVFrame *dst;
532
533 if (!s->framep[VP8_FRAME_PREVIOUS] ||
534 !s->framep[VP8_FRAME_GOLDEN]) {
535 av_log(s->avctx, AV_LOG_WARNING, "Discarding interframe without a prior keyframe!\n");
536 return AVERROR_INVALIDDATA;
537 }
538
539 src =
540 dst = s->framep[VP8_FRAME_PREVIOUS]->tf.f;
541
542 /* preserve the golden frame, write a new previous frame */
543 if (s->framep[VP8_FRAME_GOLDEN] == s->framep[VP8_FRAME_PREVIOUS]) {
544 VP8Frame *prev_frame = vp8_find_free_buffer(s);
545
546 ret = vp8_alloc_frame(s, prev_frame, 1);
547 if (ret < 0)
548 return ret;
549 s->framep[VP8_FRAME_PREVIOUS] = prev_frame;
550
551 dst = s->framep[VP8_FRAME_PREVIOUS]->tf.f;
552
553 copy_chroma(dst, src, width, height);
554 }
555
556 fade(dst->data[0], dst->linesize[0],
557 src->data[0], src->linesize[0],
558 width, height, alpha, beta);
559 }
560
561 31 return 0;
562 }
563
564 31 static int vp7_decode_frame_header(VP8Context *s, const uint8_t *buf, int buf_size)
565 {
566 31 VPXRangeCoder *c = &s->c;
567 int part1_size, hscale, vscale, i, j, ret;
568 31 int width = s->avctx->width;
569 31 int height = s->avctx->height;
570 31 int alpha = 0;
571 31 int beta = 0;
572 31 int fade_present = 1;
573
574
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (buf_size < 4) {
575 return AVERROR_INVALIDDATA;
576 }
577
578 31 s->profile = (buf[0] >> 1) & 7;
579
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (s->profile > 1) {
580 avpriv_request_sample(s->avctx, "Unknown profile %d", s->profile);
581 return AVERROR_INVALIDDATA;
582 }
583
584 31 s->keyframe = !(buf[0] & 1);
585 31 s->invisible = 0;
586 31 part1_size = AV_RL24(buf) >> 4;
587
588
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (buf_size < 4 - s->profile + part1_size) {
589 av_log(s->avctx, AV_LOG_ERROR, "Buffer size %d is too small, needed : %d\n", buf_size, 4 - s->profile + part1_size);
590 return AVERROR_INVALIDDATA;
591 }
592
593 31 buf += 4 - s->profile;
594 31 buf_size -= 4 - s->profile;
595
596 31 memcpy(s->put_pixels_tab, s->vp8dsp.put_vp8_epel_pixels_tab, sizeof(s->put_pixels_tab));
597
598 31 ret = ff_vpx_init_range_decoder(c, buf, part1_size);
599
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (ret < 0)
600 return ret;
601 31 buf += part1_size;
602 31 buf_size -= part1_size;
603
604 /* A. Dimension information (keyframes only) */
605
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 30 times.
31 if (s->keyframe) {
606 1 width = vp89_rac_get_uint(c, 12);
607 1 height = vp89_rac_get_uint(c, 12);
608 1 hscale = vp89_rac_get_uint(c, 2);
609 1 vscale = vp89_rac_get_uint(c, 2);
610
2/4
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
1 if (hscale || vscale)
611 avpriv_request_sample(s->avctx, "Upscaling");
612
613 1 s->update_golden = s->update_altref = VP8_FRAME_CURRENT;
614 1 vp78_reset_probability_tables(s);
615 1 memcpy(s->prob->pred16x16, vp8_pred16x16_prob_inter,
616 sizeof(s->prob->pred16x16));
617 1 memcpy(s->prob->pred8x8c, vp8_pred8x8c_prob_inter,
618 sizeof(s->prob->pred8x8c));
619
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
3 for (i = 0; i < 2; i++)
620 2 memcpy(s->prob->mvc[i], vp7_mv_default_prob[i],
621 sizeof(vp7_mv_default_prob[i]));
622 1 memset(&s->segmentation, 0, sizeof(s->segmentation));
623 1 memset(&s->lf_delta, 0, sizeof(s->lf_delta));
624 1 memcpy(s->prob[0].scan, ff_zigzag_scan, sizeof(s->prob[0].scan));
625 }
626
627
3/4
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 30 times.
31 if (s->keyframe || s->profile > 0)
628 1 memset(s->inter_dc_pred, 0 , sizeof(s->inter_dc_pred));
629
630 /* B. Decoding information for all four macroblock-level features */
631
2/2
✓ Branch 0 taken 124 times.
✓ Branch 1 taken 31 times.
155 for (i = 0; i < 4; i++) {
632 124 s->feature_enabled[i] = vp89_rac_get(c);
633
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 124 times.
124 if (s->feature_enabled[i]) {
634 s->feature_present_prob[i] = vp89_rac_get_uint(c, 8);
635
636 for (j = 0; j < 3; j++)
637 s->feature_index_prob[i][j] =
638 vp89_rac_get(c) ? vp89_rac_get_uint(c, 8) : 255;
639
640 if (vp7_feature_value_size[s->profile][i])
641 for (j = 0; j < 4; j++)
642 s->feature_value[i][j] =
643 vp89_rac_get(c) ? vp89_rac_get_uint(c, vp7_feature_value_size[s->profile][i]) : 0;
644 }
645 }
646
647 31 s->segmentation.enabled = 0;
648 31 s->segmentation.update_map = 0;
649 31 s->lf_delta.enabled = 0;
650
651 31 s->num_coeff_partitions = 1;
652 31 ret = ff_vpx_init_range_decoder(&s->coeff_partition[0], buf, buf_size);
653
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (ret < 0)
654 return ret;
655
656
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 1 times.
31 if (!s->macroblocks_base || /* first frame */
657
2/4
✓ Branch 0 taken 30 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 30 times.
✗ Branch 3 not taken.
30 width != s->avctx->width || height != s->avctx->height ||
658
2/4
✓ Branch 0 taken 30 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 30 times.
30 (width + 15) / 16 != s->mb_width || (height + 15) / 16 != s->mb_height) {
659
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
1 if ((ret = vp7_update_dimensions(s, width, height)) < 0)
660 return ret;
661 }
662
663 /* C. Dequantization indices */
664 31 vp7_get_quants(s);
665
666 /* D. Golden frame update flag (a Flag) for interframes only */
667
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 1 times.
31 if (!s->keyframe) {
668
2/2
✓ Branch 1 taken 3 times.
✓ Branch 2 taken 27 times.
30 s->update_golden = vp89_rac_get(c) ? VP8_FRAME_CURRENT : VP8_FRAME_NONE;
669 30 s->sign_bias[VP8_FRAME_GOLDEN] = 0;
670 }
671
672 31 s->update_last = 1;
673 31 s->update_probabilities = 1;
674
675
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (s->profile > 0) {
676 s->update_probabilities = vp89_rac_get(c);
677 if (!s->update_probabilities)
678 s->prob[1] = s->prob[0];
679
680 if (!s->keyframe)
681 fade_present = vp89_rac_get(c);
682 }
683
684
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 if (vpx_rac_is_end(c))
685 return AVERROR_INVALIDDATA;
686 /* E. Fading information for previous frame */
687
2/4
✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 31 times.
31 if (fade_present && vp89_rac_get(c)) {
688 alpha = (int8_t) vp89_rac_get_uint(c, 8);
689 beta = (int8_t) vp89_rac_get_uint(c, 8);
690 }
691
692 /* F. Loop filter type */
693
1/2
✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
31 if (!s->profile)
694 31 s->filter.simple = vp89_rac_get(c);
695
696 /* G. DCT coefficient ordering specification */
697
2/2
✓ Branch 1 taken 13 times.
✓ Branch 2 taken 18 times.
31 if (vp89_rac_get(c))
698
2/2
✓ Branch 0 taken 195 times.
✓ Branch 1 taken 13 times.
208 for (i = 1; i < 16; i++)
699 195 s->prob[0].scan[i] = ff_zigzag_scan[vp89_rac_get_uint(c, 4)];
700
701 /* H. Loop filter levels */
702
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (s->profile > 0)
703 s->filter.simple = vp89_rac_get(c);
704 31 s->filter.level = vp89_rac_get_uint(c, 6);
705 31 s->filter.sharpness = vp89_rac_get_uint(c, 3);
706
707 /* I. DCT coefficient probability update; 13.3 Token Probability Updates */
708 31 vp78_update_probability_tables(s);
709
710 31 s->mbskip_enabled = 0;
711
712 /* J. The remaining frame header data occurs ONLY FOR INTERFRAMES */
713
2/2
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 1 times.
31 if (!s->keyframe) {
714 30 s->prob->intra = vp89_rac_get_uint(c, 8);
715 30 s->prob->last = vp89_rac_get_uint(c, 8);
716 30 vp78_update_pred16x16_pred8x8_mvc_probabilities(s, VP7_MVC_SIZE);
717 }
718
719
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 if (vpx_rac_is_end(c))
720 return AVERROR_INVALIDDATA;
721
722
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
31 if ((ret = vp7_fade_frame(s, alpha, beta)) < 0)
723 return ret;
724
725 31 return 0;
726 }
727
728 1120 static int vp8_decode_frame_header(VP8Context *s, const uint8_t *buf, int buf_size)
729 {
730 1120 VPXRangeCoder *c = &s->c;
731 int header_size, hscale, vscale, ret;
732 1120 int width = s->avctx->width;
733 1120 int height = s->avctx->height;
734
735
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1120 times.
1120 if (buf_size < 3) {
736 av_log(s->avctx, AV_LOG_ERROR, "Insufficent data (%d) for header\n", buf_size);
737 return AVERROR_INVALIDDATA;
738 }
739
740 1120 s->keyframe = !(buf[0] & 1);
741 1120 s->profile = (buf[0]>>1) & 7;
742 1120 s->invisible = !(buf[0] & 0x10);
743 1120 header_size = AV_RL24(buf) >> 5;
744 1120 buf += 3;
745 1120 buf_size -= 3;
746
747 1120 s->header_partition_size = header_size;
748
749
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1120 times.
1120 if (s->profile > 3)
750 av_log(s->avctx, AV_LOG_WARNING, "Unknown profile %d\n", s->profile);
751
752
2/2
✓ Branch 0 taken 964 times.
✓ Branch 1 taken 156 times.
1120 if (!s->profile)
753 964 memcpy(s->put_pixels_tab, s->vp8dsp.put_vp8_epel_pixels_tab,
754 sizeof(s->put_pixels_tab));
755 else // profile 1-3 use bilinear, 4+ aren't defined so whatever
756 156 memcpy(s->put_pixels_tab, s->vp8dsp.put_vp8_bilinear_pixels_tab,
757 sizeof(s->put_pixels_tab));
758
759
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1120 times.
1120 if (header_size > buf_size - 7 * s->keyframe) {
760 av_log(s->avctx, AV_LOG_ERROR, "Header size larger than data provided\n");
761 return AVERROR_INVALIDDATA;
762 }
763
764
2/2
✓ Branch 0 taken 51 times.
✓ Branch 1 taken 1069 times.
1120 if (s->keyframe) {
765
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 51 times.
51 if (AV_RL24(buf) != 0x2a019d) {
766 av_log(s->avctx, AV_LOG_ERROR,
767 "Invalid start code 0x%x\n", AV_RL24(buf));
768 return AVERROR_INVALIDDATA;
769 }
770 51 width = AV_RL16(buf + 3) & 0x3fff;
771 51 height = AV_RL16(buf + 5) & 0x3fff;
772 51 hscale = buf[4] >> 6;
773 51 vscale = buf[6] >> 6;
774 51 buf += 7;
775 51 buf_size -= 7;
776
777
2/4
✓ Branch 0 taken 51 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 51 times.
51 if (hscale || vscale)
778 avpriv_request_sample(s->avctx, "Upscaling");
779
780 51 s->update_golden = s->update_altref = VP8_FRAME_CURRENT;
781 51 vp78_reset_probability_tables(s);
782 51 memcpy(s->prob->pred16x16, vp8_pred16x16_prob_inter,
783 sizeof(s->prob->pred16x16));
784 51 memcpy(s->prob->pred8x8c, vp8_pred8x8c_prob_inter,
785 sizeof(s->prob->pred8x8c));
786 51 memcpy(s->prob->mvc, vp8_mv_default_prob,
787 sizeof(s->prob->mvc));
788 51 memset(&s->segmentation, 0, sizeof(s->segmentation));
789 51 memset(&s->lf_delta, 0, sizeof(s->lf_delta));
790 }
791
792 1120 ret = ff_vpx_init_range_decoder(c, buf, header_size);
793
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1120 times.
1120 if (ret < 0)
794 return ret;
795 1120 buf += header_size;
796 1120 buf_size -= header_size;
797
798
2/2
✓ Branch 0 taken 51 times.
✓ Branch 1 taken 1069 times.
1120 if (s->keyframe) {
799 51 s->colorspace = vp89_rac_get(c);
800
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 51 times.
51 if (s->colorspace)
801 av_log(s->avctx, AV_LOG_WARNING, "Unspecified colorspace\n");
802 51 s->fullrange = vp89_rac_get(c);
803 }
804
805
2/2
✓ Branch 1 taken 433 times.
✓ Branch 2 taken 687 times.
1120 if ((s->segmentation.enabled = vp89_rac_get(c)))
806 433 parse_segment_info(s);
807 else
808 687 s->segmentation.update_map = 0; // FIXME: move this to some init function?
809
810 1120 s->filter.simple = vp89_rac_get(c);
811 1120 s->filter.level = vp89_rac_get_uint(c, 6);
812 1120 s->filter.sharpness = vp89_rac_get_uint(c, 3);
813
814
2/2
✓ Branch 1 taken 1114 times.
✓ Branch 2 taken 6 times.
1120 if ((s->lf_delta.enabled = vp89_rac_get(c))) {
815 1114 s->lf_delta.update = vp89_rac_get(c);
816
2/2
✓ Branch 0 taken 45 times.
✓ Branch 1 taken 1069 times.
1114 if (s->lf_delta.update)
817 45 update_lf_deltas(s);
818 }
819
820
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1120 times.
1120 if (setup_partitions(s, buf, buf_size)) {
821 av_log(s->avctx, AV_LOG_ERROR, "Invalid partitions\n");
822 return AVERROR_INVALIDDATA;
823 }
824
825
2/2
✓ Branch 0 taken 1093 times.
✓ Branch 1 taken 27 times.
1120 if (!s->macroblocks_base || /* first frame */
826
4/4
✓ Branch 0 taken 1087 times.
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 1081 times.
✓ Branch 3 taken 6 times.
1093 width != s->avctx->width || height != s->avctx->height ||
827
2/4
✓ Branch 0 taken 1081 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1081 times.
1081 (width+15)/16 != s->mb_width || (height+15)/16 != s->mb_height)
828
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 39 times.
39 if ((ret = vp8_update_dimensions(s, width, height)) < 0)
829 return ret;
830
831 1120 vp8_get_quants(s);
832
833
2/2
✓ Branch 0 taken 1069 times.
✓ Branch 1 taken 51 times.
1120 if (!s->keyframe) {
834 1069 update_refs(s);
835 1069 s->sign_bias[VP8_FRAME_GOLDEN] = vp89_rac_get(c);
836 1069 s->sign_bias[VP8_FRAME_ALTREF] = vp89_rac_get(c);
837 }
838
839 // if we aren't saving this frame's probabilities for future frames,
840 // make a copy of the current probabilities
841
2/2
✓ Branch 1 taken 64 times.
✓ Branch 2 taken 1056 times.
1120 if (!(s->update_probabilities = vp89_rac_get(c)))
842 64 s->prob[1] = s->prob[0];
843
844
4/4
✓ Branch 0 taken 1069 times.
✓ Branch 1 taken 51 times.
✓ Branch 3 taken 1058 times.
✓ Branch 4 taken 11 times.
1120 s->update_last = s->keyframe || vp89_rac_get(c);
845
846 1120 vp78_update_probability_tables(s);
847
848
2/2
✓ Branch 1 taken 1114 times.
✓ Branch 2 taken 6 times.
1120 if ((s->mbskip_enabled = vp89_rac_get(c)))
849 1114 s->prob->mbskip = vp89_rac_get_uint(c, 8);
850
851
2/2
✓ Branch 0 taken 1069 times.
✓ Branch 1 taken 51 times.
1120 if (!s->keyframe) {
852 1069 s->prob->intra = vp89_rac_get_uint(c, 8);
853 1069 s->prob->last = vp89_rac_get_uint(c, 8);
854 1069 s->prob->golden = vp89_rac_get_uint(c, 8);
855 1069 vp78_update_pred16x16_pred8x8_mvc_probabilities(s, VP8_MVC_SIZE);
856 }
857
858 // Record the entropy coder state here so that hwaccels can use it.
859 1120 s->c.code_word = vpx_rac_renorm(&s->c);
860 1120 s->coder_state_at_header_end.input = s->c.buffer - (-s->c.bits / 8);
861 1120 s->coder_state_at_header_end.range = s->c.high;
862 1120 s->coder_state_at_header_end.value = s->c.code_word >> 16;
863 1120 s->coder_state_at_header_end.bit_count = -s->c.bits % 8;
864
865 1120 return 0;
866 }
867
868 static av_always_inline
869 57182 void clamp_mv(const VP8mvbounds *s, VP8mv *dst, const VP8mv *src)
870 {
871 57182 dst->x = av_clip(src->x, av_clip(s->mv_min.x, INT16_MIN, INT16_MAX),
872 57182 av_clip(s->mv_max.x, INT16_MIN, INT16_MAX));
873 57182 dst->y = av_clip(src->y, av_clip(s->mv_min.y, INT16_MIN, INT16_MAX),
874 57182 av_clip(s->mv_max.y, INT16_MIN, INT16_MAX));
875 57182 }
876
877 /**
878 * Motion vector coding, 17.1.
879 */
880 84910 static av_always_inline int read_mv_component(VPXRangeCoder *c, const uint8_t *p, int vp7)
881 {
882 84910 int bit, x = 0;
883
884
2/2
✓ Branch 1 taken 19966 times.
✓ Branch 2 taken 64944 times.
84910 if (vpx_rac_get_prob_branchy(c, p[0])) {
885 int i;
886
887
2/2
✓ Branch 0 taken 59898 times.
✓ Branch 1 taken 19966 times.
79864 for (i = 0; i < 3; i++)
888 59898 x += vpx_rac_get_prob(c, p[9 + i]) << i;
889
3/4
✗ Branch 0 not taken.
✓ Branch 1 taken 19966 times.
✓ Branch 2 taken 119796 times.
✓ Branch 3 taken 19966 times.
139762 for (i = (vp7 ? 7 : 9); i > 3; i--)
890 119796 x += vpx_rac_get_prob(c, p[9 + i]) << i;
891
5/6
✗ Branch 0 not taken.
✓ Branch 1 taken 19966 times.
✓ Branch 2 taken 12222 times.
✓ Branch 3 taken 7744 times.
✓ Branch 5 taken 5564 times.
✓ Branch 6 taken 6658 times.
19966 if (!(x & (vp7 ? 0xF0 : 0xFFF0)) || vpx_rac_get_prob(c, p[12]))
892 13308 x += 8;
893 } else {
894 // small_mvtree
895 64944 const uint8_t *ps = p + 2;
896 64944 bit = vpx_rac_get_prob(c, *ps);
897 64944 ps += 1 + 3 * bit;
898 64944 x += 4 * bit;
899 64944 bit = vpx_rac_get_prob(c, *ps);
900 64944 ps += 1 + bit;
901 64944 x += 2 * bit;
902 64944 x += vpx_rac_get_prob(c, *ps);
903 }
904
905
4/4
✓ Branch 0 taken 66588 times.
✓ Branch 1 taken 18322 times.
✓ Branch 3 taken 34958 times.
✓ Branch 4 taken 31630 times.
84910 return (x && vpx_rac_get_prob(c, p[1])) ? -x : x;
906 }
907
908 static int vp7_read_mv_component(VPXRangeCoder *c, const uint8_t *p)
909 {
910 return read_mv_component(c, p, 1);
911 }
912
913 28394 static int vp8_read_mv_component(VPXRangeCoder *c, const uint8_t *p)
914 {
915 28394 return read_mv_component(c, p, 0);
916 }
917
918 static av_always_inline
919 136610 const uint8_t *get_submv_prob(uint32_t left, uint32_t top, int is_vp7)
920 {
921
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 136610 times.
136610 if (is_vp7)
922 return vp7_submv_prob;
923
924
2/2
✓ Branch 0 taken 58719 times.
✓ Branch 1 taken 77891 times.
136610 if (left == top)
925
2/2
✓ Branch 0 taken 23422 times.
✓ Branch 1 taken 35297 times.
58719 return vp8_submv_prob[4 - !!left];
926
2/2
✓ Branch 0 taken 18934 times.
✓ Branch 1 taken 58957 times.
77891 if (!top)
927 18934 return vp8_submv_prob[2];
928 58957 return vp8_submv_prob[1 - !!left];
929 }
930
931 /**
932 * Split motion vector prediction, 16.4.
933 * @returns the number of motion vectors parsed (2, 4 or 16)
934 */
935 static av_always_inline
936 18330 int decode_splitmvs(const VP8Context *s, VPXRangeCoder *c, VP8Macroblock *mb,
937 int layout, int is_vp7)
938 {
939 int part_idx;
940 int n, num;
941 const VP8Macroblock *top_mb;
942 18330 const VP8Macroblock *left_mb = &mb[-1];
943 18330 const uint8_t *mbsplits_left = vp8_mbsplits[left_mb->partitioning];
944 const uint8_t *mbsplits_top, *mbsplits_cur, *firstidx;
945 const VP8mv *top_mv;
946 18330 const VP8mv *left_mv = left_mb->bmv;
947 18330 const VP8mv *cur_mv = mb->bmv;
948
949
1/2
✓ Branch 0 taken 18330 times.
✗ Branch 1 not taken.
18330 if (!layout) // layout is inlined, s->mb_layout is not
950 18330 top_mb = &mb[2];
951 else
952 top_mb = &mb[-s->mb_width - 1];
953 18330 mbsplits_top = vp8_mbsplits[top_mb->partitioning];
954 18330 top_mv = top_mb->bmv;
955
956
2/2
✓ Branch 1 taken 11715 times.
✓ Branch 2 taken 6615 times.
18330 if (vpx_rac_get_prob_branchy(c, vp8_mbsplit_prob[0])) {
957
2/2
✓ Branch 1 taken 8045 times.
✓ Branch 2 taken 3670 times.
11715 if (vpx_rac_get_prob_branchy(c, vp8_mbsplit_prob[1]))
958 8045 part_idx = VP8_SPLITMVMODE_16x8 + vpx_rac_get_prob(c, vp8_mbsplit_prob[2]);
959 else
960 3670 part_idx = VP8_SPLITMVMODE_8x8;
961 } else {
962 6615 part_idx = VP8_SPLITMVMODE_4x4;
963 }
964
965 18330 num = vp8_mbsplit_count[part_idx];
966 18330 mbsplits_cur = vp8_mbsplits[part_idx],
967 18330 firstidx = vp8_mbfirstidx[part_idx];
968 18330 mb->partitioning = part_idx;
969
970
2/2
✓ Branch 0 taken 136610 times.
✓ Branch 1 taken 18330 times.
154940 for (n = 0; n < num; n++) {
971 136610 int k = firstidx[n];
972 uint32_t left, above;
973 const uint8_t *submv_prob;
974
975
2/2
✓ Branch 0 taken 46755 times.
✓ Branch 1 taken 89855 times.
136610 if (!(k & 3))
976 46755 left = AV_RN32A(&left_mv[mbsplits_left[k + 3]]);
977 else
978 89855 left = AV_RN32A(&cur_mv[mbsplits_cur[k - 1]]);
979
2/2
✓ Branch 0 taken 44980 times.
✓ Branch 1 taken 91630 times.
136610 if (k <= 3)
980 44980 above = AV_RN32A(&top_mv[mbsplits_top[k + 12]]);
981 else
982 91630 above = AV_RN32A(&cur_mv[mbsplits_cur[k - 4]]);
983
984 136610 submv_prob = get_submv_prob(left, above, is_vp7);
985
986
2/2
✓ Branch 1 taken 50617 times.
✓ Branch 2 taken 85993 times.
136610 if (vpx_rac_get_prob_branchy(c, submv_prob[0])) {
987
2/2
✓ Branch 1 taken 31746 times.
✓ Branch 2 taken 18871 times.
50617 if (vpx_rac_get_prob_branchy(c, submv_prob[1])) {
988
2/2
✓ Branch 1 taken 28258 times.
✓ Branch 2 taken 3488 times.
31746 if (vpx_rac_get_prob_branchy(c, submv_prob[2])) {
989 56516 mb->bmv[n].y = mb->mv.y +
990 28258 read_mv_component(c, s->prob->mvc[0], is_vp7);
991 28258 mb->bmv[n].x = mb->mv.x +
992 28258 read_mv_component(c, s->prob->mvc[1], is_vp7);
993 } else {
994 3488 AV_ZERO32(&mb->bmv[n]);
995 }
996 } else {
997 18871 AV_WN32A(&mb->bmv[n], above);
998 }
999 } else {
1000 85993 AV_WN32A(&mb->bmv[n], left);
1001 }
1002 }
1003
1004 18330 return num;
1005 }
1006
1007 /**
1008 * The vp7 reference decoder uses a padding macroblock column (added to right
1009 * edge of the frame) to guard against illegal macroblock offsets. The
1010 * algorithm has bugs that permit offsets to straddle the padding column.
1011 * This function replicates those bugs.
1012 *
1013 * @param[out] edge_x macroblock x address
1014 * @param[out] edge_y macroblock y address
1015 *
1016 * @return macroblock offset legal (boolean)
1017 */
1018 79200 static int vp7_calculate_mb_offset(int mb_x, int mb_y, int mb_width,
1019 int xoffset, int yoffset, int boundary,
1020 int *edge_x, int *edge_y)
1021 {
1022 79200 int vwidth = mb_width + 1;
1023 79200 int new = (mb_y + yoffset) * vwidth + mb_x + xoffset;
1024
4/4
✓ Branch 0 taken 69690 times.
✓ Branch 1 taken 9510 times.
✓ Branch 2 taken 2760 times.
✓ Branch 3 taken 66930 times.
79200 if (new < boundary || new % vwidth == vwidth - 1)
1025 12270 return 0;
1026 66930 *edge_y = new / vwidth;
1027 66930 *edge_x = new % vwidth;
1028 66930 return 1;
1029 }
1030
1031 66930 static const VP8mv *get_bmv_ptr(const VP8Macroblock *mb, int subblock)
1032 {
1033
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66930 times.
66930 return &mb->bmv[mb->mode == VP8_MVMODE_SPLIT ? vp8_mbsplits[mb->partitioning][subblock] : 0];
1034 }
1035
1036 static av_always_inline
1037 6600 void vp7_decode_mvs(VP8Context *s, VP8Macroblock *mb,
1038 int mb_x, int mb_y, int layout)
1039 {
1040 enum { CNT_ZERO, CNT_NEAREST, CNT_NEAR };
1041 enum { VP8_EDGE_TOP, VP8_EDGE_LEFT, VP8_EDGE_TOPLEFT };
1042 6600 int idx = CNT_ZERO;
1043 VP8mv near_mv[3];
1044 6600 uint8_t cnt[3] = { 0 };
1045 6600 VPXRangeCoder *c = &s->c;
1046 int i;
1047
1048 6600 AV_ZERO32(&near_mv[0]);
1049 6600 AV_ZERO32(&near_mv[1]);
1050 6600 AV_ZERO32(&near_mv[2]);
1051
1052
2/2
✓ Branch 0 taken 79200 times.
✓ Branch 1 taken 6600 times.
85800 for (i = 0; i < VP7_MV_PRED_COUNT; i++) {
1053 79200 const VP7MVPred * pred = &vp7_mv_pred[i];
1054 int edge_x, edge_y;
1055
1056
2/2
✓ Branch 0 taken 66930 times.
✓ Branch 1 taken 12270 times.
79200 if (vp7_calculate_mb_offset(mb_x, mb_y, s->mb_width, pred->xoffset,
1057 79200 pred->yoffset, !s->profile, &edge_x, &edge_y)) {
1058
1/2
✓ Branch 0 taken 66930 times.
✗ Branch 1 not taken.
66930 const VP8Macroblock *edge = (s->mb_layout == 1)
1059 66930 ? s->macroblocks_base + 1 + edge_x +
1060 66930 (s->mb_width + 1) * (edge_y + 1)
1061 : s->macroblocks + edge_x +
1062 (s->mb_height - edge_y - 1) * 2;
1063 66930 uint32_t mv = AV_RN32A(get_bmv_ptr(edge, vp7_mv_pred[i].subblock));
1064
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66930 times.
66930 if (mv) {
1065 if (AV_RN32A(&near_mv[CNT_NEAREST])) {
1066 if (mv == AV_RN32A(&near_mv[CNT_NEAREST])) {
1067 idx = CNT_NEAREST;
1068 } else if (AV_RN32A(&near_mv[CNT_NEAR])) {
1069 if (mv != AV_RN32A(&near_mv[CNT_NEAR]))
1070 continue;
1071 idx = CNT_NEAR;
1072 } else {
1073 AV_WN32A(&near_mv[CNT_NEAR], mv);
1074 idx = CNT_NEAR;
1075 }
1076 } else {
1077 AV_WN32A(&near_mv[CNT_NEAREST], mv);
1078 idx = CNT_NEAREST;
1079 }
1080 } else {
1081 66930 idx = CNT_ZERO;
1082 }
1083 } else {
1084 12270 idx = CNT_ZERO;
1085 }
1086 79200 cnt[idx] += vp7_mv_pred[i].score;
1087 }
1088
1089 6600 mb->partitioning = VP8_SPLITMVMODE_NONE;
1090
1091
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 6600 times.
6600 if (vpx_rac_get_prob_branchy(c, vp7_mode_contexts[cnt[CNT_ZERO]][0])) {
1092 mb->mode = VP8_MVMODE_MV;
1093
1094 if (vpx_rac_get_prob_branchy(c, vp7_mode_contexts[cnt[CNT_NEAREST]][1])) {
1095
1096 if (vpx_rac_get_prob_branchy(c, vp7_mode_contexts[cnt[CNT_NEAR]][2])) {
1097
1098 if (cnt[CNT_NEAREST] > cnt[CNT_NEAR])
1099 AV_WN32A(&mb->mv, cnt[CNT_ZERO] > cnt[CNT_NEAREST] ? 0 : AV_RN32A(&near_mv[CNT_NEAREST]));
1100 else
1101 AV_WN32A(&mb->mv, cnt[CNT_ZERO] > cnt[CNT_NEAR] ? 0 : AV_RN32A(&near_mv[CNT_NEAR]));
1102
1103 if (vpx_rac_get_prob_branchy(c, vp7_mode_contexts[cnt[CNT_NEAR]][3])) {
1104 mb->mode = VP8_MVMODE_SPLIT;
1105 mb->mv = mb->bmv[decode_splitmvs(s, c, mb, layout, IS_VP7) - 1];
1106 } else {
1107 mb->mv.y += vp7_read_mv_component(c, s->prob->mvc[0]);
1108 mb->mv.x += vp7_read_mv_component(c, s->prob->mvc[1]);
1109 mb->bmv[0] = mb->mv;
1110 }
1111 } else {
1112 mb->mv = near_mv[CNT_NEAR];
1113 mb->bmv[0] = mb->mv;
1114 }
1115 } else {
1116 mb->mv = near_mv[CNT_NEAREST];
1117 mb->bmv[0] = mb->mv;
1118 }
1119 } else {
1120 6600 mb->mode = VP8_MVMODE_ZERO;
1121 6600 AV_ZERO32(&mb->mv);
1122 6600 mb->bmv[0] = mb->mv;
1123 }
1124 6600 }
1125
1126 static av_always_inline
1127 380250 void vp8_decode_mvs(VP8Context *s, const VP8mvbounds *mv_bounds, VP8Macroblock *mb,
1128 int mb_x, int mb_y, int layout)
1129 {
1130 380250 VP8Macroblock *mb_edge[3] = { 0 /* top */,
1131 380250 mb - 1 /* left */,
1132 0 /* top-left */ };
1133 enum { CNT_ZERO, CNT_NEAREST, CNT_NEAR, CNT_SPLITMV };
1134 enum { VP8_EDGE_TOP, VP8_EDGE_LEFT, VP8_EDGE_TOPLEFT };
1135 380250 int idx = CNT_ZERO;
1136 380250 int cur_sign_bias = s->sign_bias[mb->ref_frame];
1137 380250 const int8_t *sign_bias = s->sign_bias;
1138 VP8mv near_mv[4];
1139 380250 uint8_t cnt[4] = { 0 };
1140 380250 VPXRangeCoder *c = &s->c;
1141
1142
1/2
✓ Branch 0 taken 380250 times.
✗ Branch 1 not taken.
380250 if (!layout) { // layout is inlined (s->mb_layout is not)
1143 380250 mb_edge[0] = mb + 2;
1144 380250 mb_edge[2] = mb + 1;
1145 } else {
1146 mb_edge[0] = mb - s->mb_width - 1;
1147 mb_edge[2] = mb - s->mb_width - 2;
1148 }
1149
1150 380250 AV_ZERO32(&near_mv[0]);
1151 380250 AV_ZERO32(&near_mv[1]);
1152 380250 AV_ZERO32(&near_mv[2]);
1153
1154 /* Process MB on top, left and top-left */
1155 #define MV_EDGE_CHECK(n) \
1156 { \
1157 const VP8Macroblock *edge = mb_edge[n]; \
1158 int edge_ref = edge->ref_frame; \
1159 if (edge_ref != VP8_FRAME_CURRENT) { \
1160 uint32_t mv = AV_RN32A(&edge->mv); \
1161 if (mv) { \
1162 if (cur_sign_bias != sign_bias[edge_ref]) { \
1163 /* SWAR negate of the values in mv. */ \
1164 mv = ~mv; \
1165 mv = ((mv & 0x7fff7fff) + \
1166 0x00010001) ^ (mv & 0x80008000); \
1167 } \
1168 if (!n || mv != AV_RN32A(&near_mv[idx])) \
1169 AV_WN32A(&near_mv[++idx], mv); \
1170 cnt[idx] += 1 + (n != 2); \
1171 } else \
1172 cnt[CNT_ZERO] += 1 + (n != 2); \
1173 } \
1174 }
1175
1176
6/6
✓ Branch 0 taken 353348 times.
✓ Branch 1 taken 26902 times.
✓ Branch 2 taken 44594 times.
✓ Branch 3 taken 308754 times.
✓ Branch 4 taken 109 times.
✓ Branch 5 taken 44485 times.
380250 MV_EDGE_CHECK(0)
1177
8/8
✓ Branch 0 taken 361319 times.
✓ Branch 1 taken 18931 times.
✓ Branch 2 taken 44761 times.
✓ Branch 3 taken 316558 times.
✓ Branch 4 taken 87 times.
✓ Branch 5 taken 44674 times.
✓ Branch 6 taken 36130 times.
✓ Branch 7 taken 8631 times.
380250 MV_EDGE_CHECK(1)
1178
8/8
✓ Branch 0 taken 340117 times.
✓ Branch 1 taken 40133 times.
✓ Branch 2 taken 41903 times.
✓ Branch 3 taken 298214 times.
✓ Branch 4 taken 123 times.
✓ Branch 5 taken 41780 times.
✓ Branch 6 taken 24718 times.
✓ Branch 7 taken 17185 times.
380250 MV_EDGE_CHECK(2)
1179
1180 380250 mb->partitioning = VP8_SPLITMVMODE_NONE;
1181
2/2
✓ Branch 1 taken 57182 times.
✓ Branch 2 taken 323068 times.
380250 if (vpx_rac_get_prob_branchy(c, vp8_mode_contexts[cnt[CNT_ZERO]][0])) {
1182 57182 mb->mode = VP8_MVMODE_MV;
1183
1184 /* If we have three distinct MVs, merge first and last if they're the same */
1185
2/2
✓ Branch 0 taken 12759 times.
✓ Branch 1 taken 44423 times.
57182 if (cnt[CNT_SPLITMV] &&
1186
2/2
✓ Branch 0 taken 5465 times.
✓ Branch 1 taken 7294 times.
12759 AV_RN32A(&near_mv[1 + VP8_EDGE_TOP]) == AV_RN32A(&near_mv[1 + VP8_EDGE_TOPLEFT]))
1187 5465 cnt[CNT_NEAREST] += 1;
1188
1189 /* Swap near and nearest if necessary */
1190
2/2
✓ Branch 0 taken 4482 times.
✓ Branch 1 taken 52700 times.
57182 if (cnt[CNT_NEAR] > cnt[CNT_NEAREST]) {
1191 4482 FFSWAP(uint8_t, cnt[CNT_NEAREST], cnt[CNT_NEAR]);
1192 4482 FFSWAP(VP8mv, near_mv[CNT_NEAREST], near_mv[CNT_NEAR]);
1193 }
1194
1195
2/2
✓ Branch 1 taken 37615 times.
✓ Branch 2 taken 19567 times.
57182 if (vpx_rac_get_prob_branchy(c, vp8_mode_contexts[cnt[CNT_NEAREST]][1])) {
1196
2/2
✓ Branch 1 taken 32527 times.
✓ Branch 2 taken 5088 times.
37615 if (vpx_rac_get_prob_branchy(c, vp8_mode_contexts[cnt[CNT_NEAR]][2])) {
1197 /* Choose the best mv out of 0,0 and the nearest mv */
1198 32527 clamp_mv(mv_bounds, &mb->mv, &near_mv[CNT_ZERO + (cnt[CNT_NEAREST] >= cnt[CNT_ZERO])]);
1199 32527 cnt[CNT_SPLITMV] = ((mb_edge[VP8_EDGE_LEFT]->mode == VP8_MVMODE_SPLIT) +
1200 32527 (mb_edge[VP8_EDGE_TOP]->mode == VP8_MVMODE_SPLIT)) * 2 +
1201 32527 (mb_edge[VP8_EDGE_TOPLEFT]->mode == VP8_MVMODE_SPLIT);
1202
1203
2/2
✓ Branch 1 taken 18330 times.
✓ Branch 2 taken 14197 times.
32527 if (vpx_rac_get_prob_branchy(c, vp8_mode_contexts[cnt[CNT_SPLITMV]][3])) {
1204 18330 mb->mode = VP8_MVMODE_SPLIT;
1205 18330 mb->mv = mb->bmv[decode_splitmvs(s, c, mb, layout, IS_VP8) - 1];
1206 } else {
1207 14197 mb->mv.y += vp8_read_mv_component(c, s->prob->mvc[0]);
1208 14197 mb->mv.x += vp8_read_mv_component(c, s->prob->mvc[1]);
1209 14197 mb->bmv[0] = mb->mv;
1210 }
1211 } else {
1212 5088 clamp_mv(mv_bounds, &mb->mv, &near_mv[CNT_NEAR]);
1213 5088 mb->bmv[0] = mb->mv;
1214 }
1215 } else {
1216 19567 clamp_mv(mv_bounds, &mb->mv, &near_mv[CNT_NEAREST]);
1217 19567 mb->bmv[0] = mb->mv;
1218 }
1219 } else {
1220 323068 mb->mode = VP8_MVMODE_ZERO;
1221 323068 AV_ZERO32(&mb->mv);
1222 323068 mb->bmv[0] = mb->mv;
1223 }
1224 380250 }
1225
1226 static av_always_inline
1227 17968 void decode_intra4x4_modes(VP8Context *s, VPXRangeCoder *c, VP8Macroblock *mb,
1228 int mb_x, int keyframe, int layout)
1229 {
1230 17968 uint8_t *intra4x4 = mb->intra4x4_pred_mode_mb;
1231
1232
2/2
✓ Branch 0 taken 33 times.
✓ Branch 1 taken 17935 times.
17968 if (layout) {
1233 33 VP8Macroblock *mb_top = mb - s->mb_width - 1;
1234 33 memcpy(mb->intra4x4_pred_mode_top, mb_top->intra4x4_pred_mode_top, 4);
1235 }
1236
2/2
✓ Branch 0 taken 11054 times.
✓ Branch 1 taken 6914 times.
17968 if (keyframe) {
1237 int x, y;
1238 uint8_t *top;
1239 11054 uint8_t *const left = s->intra4x4_pred_mode_left;
1240
2/2
✓ Branch 0 taken 33 times.
✓ Branch 1 taken 11021 times.
11054 if (layout)
1241 33 top = mb->intra4x4_pred_mode_top;
1242 else
1243 11021 top = s->intra4x4_pred_mode_top + 4 * mb_x;
1244
2/2
✓ Branch 0 taken 44216 times.
✓ Branch 1 taken 11054 times.
55270 for (y = 0; y < 4; y++) {
1245
2/2
✓ Branch 0 taken 176864 times.
✓ Branch 1 taken 44216 times.
221080 for (x = 0; x < 4; x++) {
1246 const uint8_t *ctx;
1247 176864 ctx = vp8_pred4x4_prob_intra[top[x]][left[y]];
1248 176864 *intra4x4 = vp89_rac_get_tree(c, vp8_pred4x4_tree, ctx);
1249 176864 left[y] = top[x] = *intra4x4;
1250 176864 intra4x4++;
1251 }
1252 }
1253 } else {
1254 int i;
1255
2/2
✓ Branch 0 taken 110624 times.
✓ Branch 1 taken 6914 times.
117538 for (i = 0; i < 16; i++)
1256 110624 intra4x4[i] = vp89_rac_get_tree(c, vp8_pred4x4_tree,
1257 vp8_pred4x4_prob_inter);
1258 }
1259 17968 }
1260
1261 static av_always_inline
1262 444989 void decode_mb_mode(VP8Context *s, const VP8mvbounds *mv_bounds,
1263 VP8Macroblock *mb, int mb_x, int mb_y,
1264 uint8_t *segment, const uint8_t *ref, int layout, int is_vp7)
1265 {
1266 444989 VPXRangeCoder *c = &s->c;
1267 static const char * const vp7_feature_name[] = { "q-index",
1268 "lf-delta",
1269 "partial-golden-update",
1270 "blit-pitch" };
1271
2/2
✓ Branch 0 taken 6820 times.
✓ Branch 1 taken 438169 times.
444989 if (is_vp7) {
1272 int i;
1273 6820 *segment = 0;
1274
2/2
✓ Branch 0 taken 27280 times.
✓ Branch 1 taken 6820 times.
34100 for (i = 0; i < 4; i++) {
1275
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27280 times.
27280 if (s->feature_enabled[i]) {
1276 if (vpx_rac_get_prob_branchy(c, s->feature_present_prob[i])) {
1277 int index = vp89_rac_get_tree(c, vp7_feature_index_tree,
1278 s->feature_index_prob[i]);
1279 av_log(s->avctx, AV_LOG_WARNING,
1280 "Feature %s present in macroblock (value 0x%x)\n",
1281 vp7_feature_name[i], s->feature_value[i][index]);
1282 }
1283 }
1284 }
1285
2/2
✓ Branch 0 taken 8289 times.
✓ Branch 1 taken 429880 times.
438169 } else if (s->segmentation.update_map) {
1286 8289 int bit = vpx_rac_get_prob(c, s->prob->segmentid[0]);
1287 8289 *segment = vpx_rac_get_prob(c, s->prob->segmentid[1+bit]) + 2*bit;
1288
2/2
✓ Branch 0 taken 94425 times.
✓ Branch 1 taken 335455 times.
429880 } else if (s->segmentation.enabled)
1289
1/2
✓ Branch 0 taken 94425 times.
✗ Branch 1 not taken.
94425 *segment = ref ? *ref : *segment;
1290 444989 mb->segment = *segment;
1291
1292
2/2
✓ Branch 0 taken 437415 times.
✓ Branch 1 taken 7574 times.
444989 mb->skip = s->mbskip_enabled ? vpx_rac_get_prob(c, s->prob->mbskip) : 0;
1293
1294
2/2
✓ Branch 0 taken 33492 times.
✓ Branch 1 taken 411497 times.
444989 if (s->keyframe) {
1295 33492 mb->mode = vp89_rac_get_tree(c, vp8_pred16x16_tree_intra,
1296 vp8_pred16x16_prob_intra);
1297
1298
2/2
✓ Branch 0 taken 11054 times.
✓ Branch 1 taken 22438 times.
33492 if (mb->mode == MODE_I4x4) {
1299 11054 decode_intra4x4_modes(s, c, mb, mb_x, 1, layout);
1300 } else {
1301 22438 const uint32_t modes = (is_vp7 ? vp7_pred4x4_mode
1302
2/2
✓ Branch 0 taken 187 times.
✓ Branch 1 taken 22251 times.
22438 : vp8_pred4x4_mode)[mb->mode] * 0x01010101u;
1303
2/2
✓ Branch 0 taken 187 times.
✓ Branch 1 taken 22251 times.
22438 if (s->mb_layout)
1304 187 AV_WN32A(mb->intra4x4_pred_mode_top, modes);
1305 else
1306 22251 AV_WN32A(s->intra4x4_pred_mode_top + 4 * mb_x, modes);
1307 22438 AV_WN32A(s->intra4x4_pred_mode_left, modes);
1308 }
1309
1310 33492 mb->chroma_pred_mode = vp89_rac_get_tree(c, vp8_pred8x8c_tree,
1311 vp8_pred8x8c_prob_intra);
1312 33492 mb->ref_frame = VP8_FRAME_CURRENT;
1313
2/2
✓ Branch 1 taken 386850 times.
✓ Branch 2 taken 24647 times.
411497 } else if (vpx_rac_get_prob_branchy(c, s->prob->intra)) {
1314 // inter MB, 16.2
1315
2/2
✓ Branch 1 taken 17281 times.
✓ Branch 2 taken 369569 times.
386850 if (vpx_rac_get_prob_branchy(c, s->prob->last))
1316
2/2
✓ Branch 0 taken 17249 times.
✓ Branch 1 taken 32 times.
34530 mb->ref_frame =
1317
2/2
✓ Branch 1 taken 8904 times.
✓ Branch 2 taken 8345 times.
17249 (!is_vp7 && vpx_rac_get_prob(c, s->prob->golden)) ? VP8_FRAME_ALTREF
1318 : VP8_FRAME_GOLDEN;
1319 else
1320 369569 mb->ref_frame = VP8_FRAME_PREVIOUS;
1321 386850 s->ref_count[mb->ref_frame - 1]++;
1322
1323 // motion vectors, 16.3
1324
2/2
✓ Branch 0 taken 6600 times.
✓ Branch 1 taken 380250 times.
386850 if (is_vp7)
1325 6600 vp7_decode_mvs(s, mb, mb_x, mb_y, layout);
1326 else
1327 380250 vp8_decode_mvs(s, mv_bounds, mb, mb_x, mb_y, layout);
1328 } else {
1329 // intra MB, 16.1
1330 49294 mb->mode = vp89_rac_get_tree(c, vp8_pred16x16_tree_inter,
1331 24647 s->prob->pred16x16);
1332
1333
2/2
✓ Branch 0 taken 6914 times.
✓ Branch 1 taken 17733 times.
24647 if (mb->mode == MODE_I4x4)
1334 6914 decode_intra4x4_modes(s, c, mb, mb_x, 0, layout);
1335
1336 49294 mb->chroma_pred_mode = vp89_rac_get_tree(c, vp8_pred8x8c_tree,
1337 24647 s->prob->pred8x8c);
1338 24647 mb->ref_frame = VP8_FRAME_CURRENT;
1339 24647 mb->partitioning = VP8_SPLITMVMODE_NONE;
1340 24647 AV_ZERO32(&mb->bmv[0]);
1341 }
1342 444989 }
1343
1344 /**
1345 * @param r arithmetic bitstream reader context
1346 * @param block destination for block coefficients
1347 * @param probs probabilities to use when reading trees from the bitstream
1348 * @param i initial coeff index, 0 unless a separate DC block is coded
1349 * @param qmul array holding the dc/ac dequant factor at position 0/1
1350 *
1351 * @return 0 if no coeffs were decoded
1352 * otherwise, the index of the last coeff decoded plus one
1353 */
1354 static av_always_inline
1355 397593 int decode_block_coeffs_internal(VPXRangeCoder *r, int16_t block[16],
1356 uint8_t probs[16][3][NUM_DCT_TOKENS - 1],
1357 int i, const uint8_t *token_prob, const int16_t qmul[2],
1358 const uint8_t scan[16], int vp7)
1359 {
1360 397593 VPXRangeCoder c = *r;
1361 397593 goto skip_eob;
1362 do {
1363 int coeff;
1364 1125704 restart:
1365
2/2
✓ Branch 1 taken 377051 times.
✓ Branch 2 taken 750772 times.
1127823 if (!vpx_rac_get_prob_branchy(&c, token_prob[0])) // DCT_EOB
1366 377051 break;
1367
1368 1597284 skip_eob:
1369
2/2
✓ Branch 1 taken 848631 times.
✓ Branch 2 taken 1146246 times.
1994877 if (!vpx_rac_get_prob_branchy(&c, token_prob[1])) { // DCT_0
1370
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 848631 times.
848631 if (++i == 16)
1371 break; // invalid input; blocks should end with EOB
1372 848631 token_prob = probs[i][0];
1373
2/2
✓ Branch 0 taken 2119 times.
✓ Branch 1 taken 846512 times.
848631 if (vp7)
1374 2119 goto restart;
1375 846512 goto skip_eob;
1376 }
1377
1378
2/2
✓ Branch 1 taken 726759 times.
✓ Branch 2 taken 419487 times.
1146246 if (!vpx_rac_get_prob_branchy(&c, token_prob[2])) { // DCT_1
1379 726759 coeff = 1;
1380 726759 token_prob = probs[i + 1][1];
1381 } else {
1382
2/2
✓ Branch 1 taken 279969 times.
✓ Branch 2 taken 139518 times.
419487 if (!vpx_rac_get_prob_branchy(&c, token_prob[3])) { // DCT 2,3,4
1383 279969 coeff = vpx_rac_get_prob_branchy(&c, token_prob[4]);
1384
2/2
✓ Branch 0 taken 116465 times.
✓ Branch 1 taken 163504 times.
279969 if (coeff)
1385 116465 coeff += vpx_rac_get_prob(&c, token_prob[5]);
1386 279969 coeff += 2;
1387 } else {
1388 // DCT_CAT*
1389
2/2
✓ Branch 1 taken 88351 times.
✓ Branch 2 taken 51167 times.
139518 if (!vpx_rac_get_prob_branchy(&c, token_prob[6])) {
1390
2/2
✓ Branch 1 taken 47345 times.
✓ Branch 2 taken 41006 times.
88351 if (!vpx_rac_get_prob_branchy(&c, token_prob[7])) { // DCT_CAT1
1391 47345 coeff = 5 + vpx_rac_get_prob(&c, vp8_dct_cat1_prob[0]);
1392 } else { // DCT_CAT2
1393 41006 coeff = 7;
1394 41006 coeff += vpx_rac_get_prob(&c, vp8_dct_cat2_prob[0]) << 1;
1395 41006 coeff += vpx_rac_get_prob(&c, vp8_dct_cat2_prob[1]);
1396 }
1397 } else { // DCT_CAT3 and up
1398 51167 int a = vpx_rac_get_prob(&c, token_prob[8]);
1399 51167 int b = vpx_rac_get_prob(&c, token_prob[9 + a]);
1400 51167 int cat = (a << 1) + b;
1401 51167 coeff = 3 + (8 << cat);
1402 51167 coeff += vp8_rac_get_coeff(&c, ff_vp8_dct_cat_prob[cat]);
1403 }
1404 }
1405 419487 token_prob = probs[i + 1][2];
1406 }
1407
4/4
✓ Branch 1 taken 576937 times.
✓ Branch 2 taken 569309 times.
✓ Branch 3 taken 844394 times.
✓ Branch 4 taken 301852 times.
1146246 block[scan[i]] = (vp89_rac_get(&c) ? -coeff : coeff) * qmul[!!i];
1408
2/2
✓ Branch 0 taken 1125704 times.
✓ Branch 1 taken 20542 times.
1146246 } while (++i < 16);
1409
1410 397593 *r = c;
1411 397593 return i;
1412 }
1413
1414 static av_always_inline
1415 6600 int inter_predict_dc(int16_t block[16], int16_t pred[2])
1416 {
1417 6600 int16_t dc = block[0];
1418 6600 int ret = 0;
1419
1420
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6600 times.
6600 if (pred[1] > 3) {
1421 dc += pred[0];
1422 ret = 1;
1423 }
1424
1425
1/2
✓ Branch 0 taken 6600 times.
✗ Branch 1 not taken.
6600 if (!pred[0] | !dc | ((int32_t)pred[0] ^ (int32_t)dc) >> 31) {
1426 6600 block[0] = pred[0] = dc;
1427 6600 pred[1] = 0;
1428 } else {
1429 if (pred[0] == dc)
1430 pred[1]++;
1431 block[0] = pred[0] = dc;
1432 }
1433
1434 6600 return ret;
1435 }
1436
1437 735 static int vp7_decode_block_coeffs_internal(VPXRangeCoder *r,
1438 int16_t block[16],
1439 uint8_t probs[16][3][NUM_DCT_TOKENS - 1],
1440 int i, const uint8_t *token_prob,
1441 const int16_t qmul[2],
1442 const uint8_t scan[16])
1443 {
1444 735 return decode_block_coeffs_internal(r, block, probs, i,
1445 token_prob, qmul, scan, IS_VP7);
1446 }
1447
1448 #ifndef vp8_decode_block_coeffs_internal
1449 396858 static int vp8_decode_block_coeffs_internal(VPXRangeCoder *r,
1450 int16_t block[16],
1451 uint8_t probs[16][3][NUM_DCT_TOKENS - 1],
1452 int i, const uint8_t *token_prob,
1453 const int16_t qmul[2])
1454 {
1455 396858 return decode_block_coeffs_internal(r, block, probs, i,
1456 token_prob, qmul, ff_zigzag_scan, IS_VP8);
1457 }
1458 #endif
1459
1460 /**
1461 * @param c arithmetic bitstream reader context
1462 * @param block destination for block coefficients
1463 * @param probs probabilities to use when reading trees from the bitstream
1464 * @param i initial coeff index, 0 unless a separate DC block is coded
1465 * @param zero_nhood the initial prediction context for number of surrounding
1466 * all-zero blocks (only left/top, so 0-2)
1467 * @param qmul array holding the dc/ac dequant factor at position 0/1
1468 * @param scan scan pattern (VP7 only)
1469 *
1470 * @return 0 if no coeffs were decoded
1471 * otherwise, the index of the last coeff decoded plus one
1472 */
1473 static av_always_inline
1474 2081398 int decode_block_coeffs(VPXRangeCoder *c, int16_t block[16],
1475 uint8_t probs[16][3][NUM_DCT_TOKENS - 1],
1476 int i, int zero_nhood, const int16_t qmul[2],
1477 const uint8_t scan[16], int vp7)
1478 {
1479 2081398 const uint8_t *token_prob = probs[i][zero_nhood];
1480
2/2
✓ Branch 1 taken 1683805 times.
✓ Branch 2 taken 397593 times.
2081398 if (!vpx_rac_get_prob_branchy(c, token_prob[0])) // DCT_EOB
1481 1683805 return 0;
1482 735 return vp7 ? vp7_decode_block_coeffs_internal(c, block, probs, i,
1483 token_prob, qmul, scan)
1484
2/2
✓ Branch 0 taken 735 times.
✓ Branch 1 taken 396858 times.
398328 : vp8_decode_block_coeffs_internal(c, block, probs, i,
1485 token_prob, qmul);
1486 }
1487
1488 static av_always_inline
1489 84558 void decode_mb_coeffs(VP8Context *s, VP8ThreadData *td, VPXRangeCoder *c,
1490 VP8Macroblock *mb, uint8_t t_nnz[9], uint8_t l_nnz[9],
1491 int is_vp7)
1492 {
1493 84558 int i, x, y, luma_start = 0, luma_ctx = 3;
1494 84558 int nnz_pred, nnz, nnz_total = 0;
1495 84558 int segment = mb->segment;
1496 84558 int block_dc = 0;
1497
1498
6/6
✓ Branch 0 taken 67069 times.
✓ Branch 1 taken 17489 times.
✓ Branch 2 taken 60282 times.
✓ Branch 3 taken 6787 times.
✓ Branch 4 taken 45219 times.
✓ Branch 5 taken 15063 times.
84558 if (mb->mode != MODE_I4x4 && (is_vp7 || mb->mode != VP8_MVMODE_SPLIT)) {
1499 52006 nnz_pred = t_nnz[8] + l_nnz[8];
1500
1501 // decode DC values and do hadamard
1502 52006 nnz = decode_block_coeffs(c, td->block_dc, s->prob->token[1], 0,
1503 52006 nnz_pred, s->qmat[segment].luma_dc_qmul,
1504 ff_zigzag_scan, is_vp7);
1505 52006 l_nnz[8] = t_nnz[8] = !!nnz;
1506
1507
4/4
✓ Branch 0 taken 6787 times.
✓ Branch 1 taken 45219 times.
✓ Branch 2 taken 6600 times.
✓ Branch 3 taken 187 times.
52006 if (is_vp7 && mb->mode > MODE_I4x4) {
1508 6600 nnz |= inter_predict_dc(td->block_dc,
1509 6600 s->inter_dc_pred[mb->ref_frame - 1]);
1510 }
1511
1512
2/2
✓ Branch 0 taken 35439 times.
✓ Branch 1 taken 16567 times.
52006 if (nnz) {
1513 35439 nnz_total += nnz;
1514 35439 block_dc = 1;
1515
2/2
✓ Branch 0 taken 10514 times.
✓ Branch 1 taken 24925 times.
35439 if (nnz == 1)
1516 10514 s->vp8dsp.vp8_luma_dc_wht_dc(td->block, td->block_dc);
1517 else
1518 24925 s->vp8dsp.vp8_luma_dc_wht(td->block, td->block_dc);
1519 }
1520 52006 luma_start = 1;
1521 52006 luma_ctx = 0;
1522 }
1523
1524 // luma blocks
1525
2/2
✓ Branch 0 taken 338232 times.
✓ Branch 1 taken 84558 times.
422790 for (y = 0; y < 4; y++)
1526
2/2
✓ Branch 0 taken 1352928 times.
✓ Branch 1 taken 338232 times.
1691160 for (x = 0; x < 4; x++) {
1527 1352928 nnz_pred = l_nnz[y] + t_nnz[x];
1528 1352928 nnz = decode_block_coeffs(c, td->block[y][x],
1529 1352928 s->prob->token[luma_ctx],
1530 luma_start, nnz_pred,
1531 1352928 s->qmat[segment].luma_qmul,
1532 1352928 s->prob[0].scan, is_vp7);
1533 /* nnz+block_dc may be one more than the actual last index,
1534 * but we don't care */
1535 1352928 td->non_zero_count_cache[y][x] = nnz + block_dc;
1536 1352928 t_nnz[x] = l_nnz[y] = !!nnz;
1537 1352928 nnz_total += nnz;
1538 }
1539
1540 // chroma blocks
1541 // TODO: what to do about dimensions? 2nd dim for luma is x,
1542 // but for chroma it's (y<<1)|x
1543
2/2
✓ Branch 0 taken 169116 times.
✓ Branch 1 taken 84558 times.
253674 for (i = 4; i < 6; i++)
1544
2/2
✓ Branch 0 taken 338232 times.
✓ Branch 1 taken 169116 times.
507348 for (y = 0; y < 2; y++)
1545
2/2
✓ Branch 0 taken 676464 times.
✓ Branch 1 taken 338232 times.
1014696 for (x = 0; x < 2; x++) {
1546 676464 nnz_pred = l_nnz[i + 2 * y] + t_nnz[i + 2 * x];
1547 676464 nnz = decode_block_coeffs(c, td->block[i][(y << 1) + x],
1548 676464 s->prob->token[2], 0, nnz_pred,
1549 676464 s->qmat[segment].chroma_qmul,
1550 676464 s->prob[0].scan, is_vp7);
1551 676464 td->non_zero_count_cache[i][(y << 1) + x] = nnz;
1552 676464 t_nnz[i + 2 * x] = l_nnz[i + 2 * y] = !!nnz;
1553 676464 nnz_total += nnz;
1554 }
1555
1556 // if there were no coded coeffs despite the macroblock not being marked skip,
1557 // we MUST not do the inner loop filter and should not do IDCT
1558 // Since skip isn't used for bitstream prediction, just manually set it.
1559
2/2
✓ Branch 0 taken 6676 times.
✓ Branch 1 taken 77882 times.
84558 if (!nnz_total)
1560 6676 mb->skip = 1;
1561 84558 }
1562
1563 static av_always_inline
1564 425389 void backup_mb_border(uint8_t *top_border, const uint8_t *src_y,
1565 const uint8_t *src_cb, const uint8_t *src_cr,
1566 ptrdiff_t linesize, ptrdiff_t uvlinesize, int simple)
1567 {
1568 425389 AV_COPY128(top_border, src_y + 15 * linesize);
1569
2/2
✓ Branch 0 taken 417667 times.
✓ Branch 1 taken 7722 times.
425389 if (!simple) {
1570 417667 AV_COPY64(top_border + 16, src_cb + 7 * uvlinesize);
1571 417667 AV_COPY64(top_border + 24, src_cr + 7 * uvlinesize);
1572 }
1573 425389 }
1574
1575 static av_always_inline
1576 84390 void xchg_mb_border(uint8_t *top_border, uint8_t *src_y, uint8_t *src_cb,
1577 uint8_t *src_cr, ptrdiff_t linesize, ptrdiff_t uvlinesize, int mb_x,
1578 int mb_y, int mb_width, int simple, int xchg)
1579 {
1580 84390 uint8_t *top_border_m1 = top_border - 32; // for TL prediction
1581 84390 src_y -= linesize;
1582 84390 src_cb -= uvlinesize;
1583 84390 src_cr -= uvlinesize;
1584
1585 #define XCHG(a, b, xchg) \
1586 do { \
1587 if (xchg) \
1588 AV_SWAP64(b, a); \
1589 else \
1590 AV_COPY64(b, a); \
1591 } while (0)
1592
1593
2/2
✓ Branch 0 taken 42195 times.
✓ Branch 1 taken 42195 times.
84390 XCHG(top_border_m1 + 8, src_y - 8, xchg);
1594
2/2
✓ Branch 0 taken 42195 times.
✓ Branch 1 taken 42195 times.
84390 XCHG(top_border, src_y, xchg);
1595 84390 XCHG(top_border + 8, src_y + 8, 1);
1596
2/2
✓ Branch 0 taken 80942 times.
✓ Branch 1 taken 3448 times.
84390 if (mb_x < mb_width - 1)
1597 80942 XCHG(top_border + 32, src_y + 16, 1);
1598
1599 // only copy chroma for normal loop filter
1600 // or to initialize the top row to 127
1601
3/4
✓ Branch 0 taken 3082 times.
✓ Branch 1 taken 81308 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3082 times.
84390 if (!simple || !mb_y) {
1602
2/2
✓ Branch 0 taken 40654 times.
✓ Branch 1 taken 40654 times.
81308 XCHG(top_border_m1 + 16, src_cb - 8, xchg);
1603
2/2
✓ Branch 0 taken 40654 times.
✓ Branch 1 taken 40654 times.
81308 XCHG(top_border_m1 + 24, src_cr - 8, xchg);
1604 81308 XCHG(top_border + 16, src_cb, 1);
1605 81308 XCHG(top_border + 24, src_cr, 1);
1606 }
1607 84390 }
1608
1609 static av_always_inline
1610 73882 int check_dc_pred8x8_mode(int mode, int mb_x, int mb_y)
1611 {
1612
2/2
✓ Branch 0 taken 3118 times.
✓ Branch 1 taken 70764 times.
73882 if (!mb_x)
1613
2/2
✓ Branch 0 taken 2915 times.
✓ Branch 1 taken 203 times.
3118 return mb_y ? TOP_DC_PRED8x8 : DC_128_PRED8x8;
1614 else
1615
2/2
✓ Branch 0 taken 65364 times.
✓ Branch 1 taken 5400 times.
70764 return mb_y ? mode : LEFT_DC_PRED8x8;
1616 }
1617
1618 static av_always_inline
1619 1826 int check_tm_pred8x8_mode(int mode, int mb_x, int mb_y, int vp7)
1620 {
1621
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1825 times.
1826 if (!mb_x)
1622
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
1 return mb_y ? VERT_PRED8x8 : (vp7 ? DC_128_PRED8x8 : DC_129_PRED8x8);
1623 else
1624
2/2
✓ Branch 0 taken 1823 times.
✓ Branch 1 taken 2 times.
1825 return mb_y ? mode : HOR_PRED8x8;
1625 }
1626
1627 static av_always_inline
1628 98310 int check_intra_pred8x8_mode_emuedge(int mode, int mb_x, int mb_y, int vp7)
1629 {
1630
4/5
✓ Branch 0 taken 73882 times.
✓ Branch 1 taken 5815 times.
✓ Branch 2 taken 16787 times.
✓ Branch 3 taken 1826 times.
✗ Branch 4 not taken.
98310 switch (mode) {
1631 73882 case DC_PRED8x8:
1632 73882 return check_dc_pred8x8_mode(mode, mb_x, mb_y);
1633 5815 case VERT_PRED8x8:
1634
3/4
✓ Branch 0 taken 67 times.
✓ Branch 1 taken 5748 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 67 times.
5815 return !mb_y ? (vp7 ? DC_128_PRED8x8 : DC_127_PRED8x8) : mode;
1635 16787 case HOR_PRED8x8:
1636
3/4
✓ Branch 0 taken 142 times.
✓ Branch 1 taken 16645 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 142 times.
16787 return !mb_x ? (vp7 ? DC_128_PRED8x8 : DC_129_PRED8x8) : mode;
1637 1826 case PLANE_PRED8x8: /* TM */
1638 1826 return check_tm_pred8x8_mode(mode, mb_x, mb_y, vp7);
1639 }
1640 return mode;
1641 }
1642
1643 static av_always_inline
1644 51007 int check_tm_pred4x4_mode(int mode, int mb_x, int mb_y, int vp7)
1645 {
1646
2/2
✓ Branch 0 taken 1362 times.
✓ Branch 1 taken 49645 times.
51007 if (!mb_x) {
1647
3/4
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1361 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
1362 return mb_y ? VERT_VP8_PRED : (vp7 ? DC_128_PRED : DC_129_PRED);
1648 } else {
1649
2/2
✓ Branch 0 taken 48236 times.
✓ Branch 1 taken 1409 times.
49645 return mb_y ? mode : HOR_VP8_PRED;
1650 }
1651 }
1652
1653 static av_always_inline
1654 287488 int check_intra_pred4x4_mode_emuedge(int mode, int mb_x, int mb_y,
1655 int *copy_buf, int vp7)
1656 {
1657
6/7
✓ Branch 0 taken 13788 times.
✓ Branch 1 taken 21697 times.
✓ Branch 2 taken 19736 times.
✓ Branch 3 taken 12422 times.
✓ Branch 4 taken 51007 times.
✓ Branch 5 taken 168838 times.
✗ Branch 6 not taken.
287488 switch (mode) {
1658 13788 case VERT_PRED:
1659
3/4
✓ Branch 0 taken 95 times.
✓ Branch 1 taken 13693 times.
✓ Branch 2 taken 95 times.
✗ Branch 3 not taken.
13788 if (!mb_x && mb_y) {
1660 95 *copy_buf = 1;
1661 95 return mode;
1662 }
1663 /* fall-through */
1664 case DIAG_DOWN_LEFT_PRED:
1665 case VERT_LEFT_PRED:
1666
3/4
✓ Branch 0 taken 24 times.
✓ Branch 1 taken 35366 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 24 times.
35390 return !mb_y ? (vp7 ? DC_128_PRED : DC_127_PRED) : mode;
1667 19736 case HOR_PRED:
1668
2/2
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 19608 times.
19736 if (!mb_y) {
1669 128 *copy_buf = 1;
1670 128 return mode;
1671 }
1672 /* fall-through */
1673 case HOR_UP_PRED:
1674
3/4
✓ Branch 0 taken 44 times.
✓ Branch 1 taken 31986 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 44 times.
32030 return !mb_x ? (vp7 ? DC_128_PRED : DC_129_PRED) : mode;
1675 51007 case TM_VP8_PRED:
1676 51007 return check_tm_pred4x4_mode(mode, mb_x, mb_y, vp7);
1677 168838 case DC_PRED: /* 4x4 DC doesn't use the same "H.264-style" exceptions
1678 * as 16x16/8x8 DC */
1679 case DIAG_DOWN_RIGHT_PRED:
1680 case VERT_RIGHT_PRED:
1681 case HOR_DOWN_PRED:
1682
4/4
✓ Branch 0 taken 168264 times.
✓ Branch 1 taken 574 times.
✓ Branch 2 taken 566 times.
✓ Branch 3 taken 167698 times.
168838 if (!mb_y || !mb_x)
1683 1140 *copy_buf = 1;
1684 168838 return mode;
1685 }
1686 return mode;
1687 }
1688
1689 static av_always_inline
1690 58139 void intra_predict(VP8Context *s, VP8ThreadData *td, uint8_t *const dst[3],
1691 VP8Macroblock *mb, int mb_x, int mb_y, int is_vp7)
1692 {
1693 int x, y, mode, nnz;
1694 uint32_t tr;
1695
1696 /* for the first row, we need to run xchg_mb_border to init the top edge
1697 * to 127 otherwise, skip it if we aren't going to deblock */
1698
6/8
✓ Branch 0 taken 54708 times.
✓ Branch 1 taken 3431 times.
✓ Branch 2 taken 12513 times.
✓ Branch 3 taken 42195 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 12513 times.
✓ Branch 6 taken 42195 times.
✗ Branch 7 not taken.
58139 if (mb_y && (s->deblock_filter || !mb_y) && td->thread_nr == 0)
1699 42195 xchg_mb_border(s->top_border[mb_x + 1], dst[0], dst[1], dst[2],
1700 42195 s->linesize, s->uvlinesize, mb_x, mb_y, s->mb_width,
1701 42195 s->filter.simple, 1);
1702
1703
2/2
✓ Branch 0 taken 40171 times.
✓ Branch 1 taken 17968 times.
58139 if (mb->mode < MODE_I4x4) {
1704 40171 mode = check_intra_pred8x8_mode_emuedge(mb->mode, mb_x, mb_y, is_vp7);
1705 40171 s->hpc.pred16x16[mode](dst[0], s->linesize);
1706 } else {
1707 17968 uint8_t *ptr = dst[0];
1708 17968 const uint8_t *intra4x4 = mb->intra4x4_pred_mode_mb;
1709
2/2
✓ Branch 0 taken 33 times.
✓ Branch 1 taken 17935 times.
17968 const uint8_t lo = is_vp7 ? 128 : 127;
1710
2/2
✓ Branch 0 taken 33 times.
✓ Branch 1 taken 17935 times.
17968 const uint8_t hi = is_vp7 ? 128 : 129;
1711 17968 const uint8_t tr_top[4] = { lo, lo, lo, lo };
1712
1713 // all blocks on the right edge of the macroblock use bottom edge
1714 // the top macroblock for their topright edge
1715 17968 const uint8_t *tr_right = ptr - s->linesize + 16;
1716
1717 // if we're on the right edge of the frame, said edge is extended
1718 // from the top macroblock
1719
4/4
✓ Branch 0 taken 17337 times.
✓ Branch 1 taken 631 times.
✓ Branch 2 taken 638 times.
✓ Branch 3 taken 16699 times.
17968 if (mb_y && mb_x == s->mb_width - 1) {
1720 638 tr = tr_right[-1] * 0x01010101u;
1721 638 tr_right = (uint8_t *) &tr;
1722 }
1723
1724
2/2
✓ Branch 0 taken 480 times.
✓ Branch 1 taken 17488 times.
17968 if (mb->skip)
1725 480 AV_ZERO128(td->non_zero_count_cache);
1726
1727
2/2
✓ Branch 0 taken 71872 times.
✓ Branch 1 taken 17968 times.
89840 for (y = 0; y < 4; y++) {
1728 71872 const uint8_t *topright = ptr + 4 - s->linesize;
1729
2/2
✓ Branch 0 taken 287488 times.
✓ Branch 1 taken 71872 times.
359360 for (x = 0; x < 4; x++) {
1730 287488 int copy = 0;
1731 287488 ptrdiff_t linesize = s->linesize;
1732 287488 uint8_t *dst = ptr + 4 * x;
1733 287488 LOCAL_ALIGNED(4, uint8_t, copy_dst, [5 * 8]);
1734
1735
6/6
✓ Branch 0 taken 215616 times.
✓ Branch 1 taken 71872 times.
✓ Branch 2 taken 53904 times.
✓ Branch 3 taken 161712 times.
✓ Branch 4 taken 4417 times.
✓ Branch 5 taken 121359 times.
287488 if ((y == 0 || x == 3) && mb_y == 0) {
1736 4417 topright = tr_top;
1737
2/2
✓ Branch 0 taken 69348 times.
✓ Branch 1 taken 213723 times.
283071 } else if (x == 3)
1738 69348 topright = tr_right;
1739
1740 287488 mode = check_intra_pred4x4_mode_emuedge(intra4x4[x], mb_x + x,
1741 mb_y + y, &copy, is_vp7);
1742
2/2
✓ Branch 0 taken 1363 times.
✓ Branch 1 taken 286125 times.
287488 if (copy) {
1743 1363 dst = copy_dst + 12;
1744 1363 linesize = 8;
1745
2/2
✓ Branch 0 taken 702 times.
✓ Branch 1 taken 661 times.
1363 if (!(mb_y + y)) {
1746 702 copy_dst[3] = lo;
1747 702 AV_WN32A(copy_dst + 4, lo * 0x01010101U);
1748 } else {
1749 661 AV_COPY32(copy_dst + 4, ptr + 4 * x - s->linesize);
1750
1/2
✓ Branch 0 taken 661 times.
✗ Branch 1 not taken.
661 if (!(mb_x + x)) {
1751 661 copy_dst[3] = hi;
1752 } else {
1753 copy_dst[3] = ptr[4 * x - s->linesize - 1];
1754 }
1755 }
1756
2/2
✓ Branch 0 taken 702 times.
✓ Branch 1 taken 661 times.
1363 if (!(mb_x + x)) {
1757 702 copy_dst[11] =
1758 702 copy_dst[19] =
1759 702 copy_dst[27] =
1760 702 copy_dst[35] = hi;
1761 } else {
1762 661 copy_dst[11] = ptr[4 * x - 1];
1763 661 copy_dst[19] = ptr[4 * x + s->linesize - 1];
1764 661 copy_dst[27] = ptr[4 * x + s->linesize * 2 - 1];
1765 661 copy_dst[35] = ptr[4 * x + s->linesize * 3 - 1];
1766 }
1767 }
1768 287488 s->hpc.pred4x4[mode](dst, topright, linesize);
1769
2/2
✓ Branch 0 taken 1363 times.
✓ Branch 1 taken 286125 times.
287488 if (copy) {
1770 1363 AV_COPY32(ptr + 4 * x, copy_dst + 12);
1771 1363 AV_COPY32(ptr + 4 * x + s->linesize, copy_dst + 20);
1772 1363 AV_COPY32(ptr + 4 * x + s->linesize * 2, copy_dst + 28);
1773 1363 AV_COPY32(ptr + 4 * x + s->linesize * 3, copy_dst + 36);
1774 }
1775
1776 287488 nnz = td->non_zero_count_cache[y][x];
1777
2/2
✓ Branch 0 taken 124762 times.
✓ Branch 1 taken 162726 times.
287488 if (nnz) {
1778
2/2
✓ Branch 0 taken 35500 times.
✓ Branch 1 taken 89262 times.
124762 if (nnz == 1)
1779 35500 s->vp8dsp.vp8_idct_dc_add(ptr + 4 * x,
1780 35500 td->block[y][x], s->linesize);
1781 else
1782 89262 s->vp8dsp.vp8_idct_add(ptr + 4 * x,
1783 89262 td->block[y][x], s->linesize);
1784 }
1785 287488 topright += 4;
1786 }
1787
1788 71872 ptr += 4 * s->linesize;
1789 71872 intra4x4 += 4;
1790 }
1791 }
1792
1793 58139 mode = check_intra_pred8x8_mode_emuedge(mb->chroma_pred_mode,
1794 mb_x, mb_y, is_vp7);
1795 58139 s->hpc.pred8x8[mode](dst[1], s->uvlinesize);
1796 58139 s->hpc.pred8x8[mode](dst[2], s->uvlinesize);
1797
1798
6/8
✓ Branch 0 taken 54708 times.
✓ Branch 1 taken 3431 times.
✓ Branch 2 taken 12513 times.
✓ Branch 3 taken 42195 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 12513 times.
✓ Branch 6 taken 42195 times.
✗ Branch 7 not taken.
58139 if (mb_y && (s->deblock_filter || !mb_y) && td->thread_nr == 0)
1799 42195 xchg_mb_border(s->top_border[mb_x + 1], dst[0], dst[1], dst[2],
1800 42195 s->linesize, s->uvlinesize, mb_x, mb_y, s->mb_width,
1801 42195 s->filter.simple, 0);
1802 58139 }
1803
1804 static const uint8_t subpel_idx[3][8] = {
1805 { 0, 1, 2, 1, 2, 1, 2, 1 }, // nr. of left extra pixels,
1806 // also function pointer index
1807 { 0, 3, 5, 3, 5, 3, 5, 3 }, // nr. of extra pixels required
1808 { 0, 2, 3, 2, 3, 2, 3, 2 }, // nr. of right extra pixels
1809 };
1810
1811 /**
1812 * luma MC function
1813 *
1814 * @param s VP8 decoding context
1815 * @param dst target buffer for block data at block position
1816 * @param ref reference picture buffer at origin (0, 0)
1817 * @param mv motion vector (relative to block position) to get pixel data from
1818 * @param x_off horizontal position of block from origin (0, 0)
1819 * @param y_off vertical position of block from origin (0, 0)
1820 * @param block_w width of block (16, 8 or 4)
1821 * @param block_h height of block (always same as block_w)
1822 * @param width width of src/dst plane data
1823 * @param height height of src/dst plane data
1824 * @param linesize size of a single line of plane data, including padding
1825 * @param mc_func motion compensation function pointers (bilinear or sixtap MC)
1826 */
1827 static av_always_inline
1828 505130 void vp8_mc_luma(VP8Context *s, VP8ThreadData *td, uint8_t *dst,
1829 const ProgressFrame *ref, const VP8mv *mv,
1830 int x_off, int y_off, int block_w, int block_h,
1831 int width, int height, ptrdiff_t linesize,
1832 vp8_mc_func mc_func[3][3])
1833 {
1834 505130 const uint8_t *src = ref->f->data[0];
1835
1836
2/2
✓ Branch 0 taken 124899 times.
✓ Branch 1 taken 380231 times.
505130 if (AV_RN32A(mv)) {
1837 124899 ptrdiff_t src_linesize = linesize;
1838
1839 124899 int mx = (mv->x * 2) & 7, mx_idx = subpel_idx[0][mx];
1840 124899 int my = (mv->y * 2) & 7, my_idx = subpel_idx[0][my];
1841
1842 124899 x_off += mv->x >> 2;
1843 124899 y_off += mv->y >> 2;
1844
1845 // edge emulation
1846 124899 ff_progress_frame_await(ref, (3 + y_off + block_h + subpel_idx[2][my]) >> 4);
1847 124899 src += y_off * linesize + x_off;
1848
6/6
✓ Branch 0 taken 123700 times.
✓ Branch 1 taken 1199 times.
✓ Branch 2 taken 120199 times.
✓ Branch 3 taken 3501 times.
✓ Branch 4 taken 118706 times.
✓ Branch 5 taken 1493 times.
124899 if (x_off < mx_idx || x_off >= width - block_w - subpel_idx[2][mx] ||
1849
2/2
✓ Branch 0 taken 3043 times.
✓ Branch 1 taken 115663 times.
118706 y_off < my_idx || y_off >= height - block_h - subpel_idx[2][my]) {
1850 9236 s->vdsp.emulated_edge_mc(td->edge_emu_buffer,
1851 9236 src - my_idx * linesize - mx_idx,
1852 EDGE_EMU_LINESIZE, linesize,
1853 9236 block_w + subpel_idx[1][mx],
1854 9236 block_h + subpel_idx[1][my],
1855 x_off - mx_idx, y_off - my_idx,
1856 width, height);
1857 9236 src = td->edge_emu_buffer + mx_idx + EDGE_EMU_LINESIZE * my_idx;
1858 9236 src_linesize = EDGE_EMU_LINESIZE;
1859 }
1860 124899 mc_func[my_idx][mx_idx](dst, linesize, src, src_linesize, block_h, mx, my);
1861 } else {
1862 380231 ff_progress_frame_await(ref, (3 + y_off + block_h) >> 4);
1863 380231 mc_func[0][0](dst, linesize, src + y_off * linesize + x_off,
1864 linesize, block_h, 0, 0);
1865 }
1866 505130 }
1867
1868 /**
1869 * chroma MC function
1870 *
1871 * @param s VP8 decoding context
1872 * @param dst1 target buffer for block data at block position (U plane)
1873 * @param dst2 target buffer for block data at block position (V plane)
1874 * @param ref reference picture buffer at origin (0, 0)
1875 * @param mv motion vector (relative to block position) to get pixel data from
1876 * @param x_off horizontal position of block from origin (0, 0)
1877 * @param y_off vertical position of block from origin (0, 0)
1878 * @param block_w width of block (16, 8 or 4)
1879 * @param block_h height of block (always same as block_w)
1880 * @param width width of src/dst plane data
1881 * @param height height of src/dst plane data
1882 * @param linesize size of a single line of plane data, including padding
1883 * @param mc_func motion compensation function pointers (bilinear or sixtap MC)
1884 */
1885 static av_always_inline
1886 425750 void vp8_mc_chroma(VP8Context *s, VP8ThreadData *td, uint8_t *dst1,
1887 uint8_t *dst2, const ProgressFrame *ref, const VP8mv *mv,
1888 int x_off, int y_off, int block_w, int block_h,
1889 int width, int height, ptrdiff_t linesize,
1890 vp8_mc_func mc_func[3][3])
1891 {
1892 425750 const uint8_t *src1 = ref->f->data[1], *src2 = ref->f->data[2];
1893
1894
2/2
✓ Branch 0 taken 79190 times.
✓ Branch 1 taken 346560 times.
425750 if (AV_RN32A(mv)) {
1895 79190 int mx = mv->x & 7, mx_idx = subpel_idx[0][mx];
1896 79190 int my = mv->y & 7, my_idx = subpel_idx[0][my];
1897
1898 79190 x_off += mv->x >> 3;
1899 79190 y_off += mv->y >> 3;
1900
1901 // edge emulation
1902 79190 src1 += y_off * linesize + x_off;
1903 79190 src2 += y_off * linesize + x_off;
1904 79190 ff_progress_frame_await(ref, (3 + y_off + block_h + subpel_idx[2][my]) >> 3);
1905
6/6
✓ Branch 0 taken 77874 times.
✓ Branch 1 taken 1316 times.
✓ Branch 2 taken 74583 times.
✓ Branch 3 taken 3291 times.
✓ Branch 4 taken 73050 times.
✓ Branch 5 taken 1533 times.
79190 if (x_off < mx_idx || x_off >= width - block_w - subpel_idx[2][mx] ||
1906
2/2
✓ Branch 0 taken 3017 times.
✓ Branch 1 taken 70033 times.
73050 y_off < my_idx || y_off >= height - block_h - subpel_idx[2][my]) {
1907 9157 s->vdsp.emulated_edge_mc(td->edge_emu_buffer,
1908 9157 src1 - my_idx * linesize - mx_idx,
1909 EDGE_EMU_LINESIZE, linesize,
1910 9157 block_w + subpel_idx[1][mx],
1911 9157 block_h + subpel_idx[1][my],
1912 x_off - mx_idx, y_off - my_idx, width, height);
1913 9157 src1 = td->edge_emu_buffer + mx_idx + EDGE_EMU_LINESIZE * my_idx;
1914 9157 mc_func[my_idx][mx_idx](dst1, linesize, src1, EDGE_EMU_LINESIZE, block_h, mx, my);
1915
1916 9157 s->vdsp.emulated_edge_mc(td->edge_emu_buffer,
1917 9157 src2 - my_idx * linesize - mx_idx,
1918 EDGE_EMU_LINESIZE, linesize,
1919 9157 block_w + subpel_idx[1][mx],
1920 9157 block_h + subpel_idx[1][my],
1921 x_off - mx_idx, y_off - my_idx, width, height);
1922 9157 src2 = td->edge_emu_buffer + mx_idx + EDGE_EMU_LINESIZE * my_idx;
1923 9157 mc_func[my_idx][mx_idx](dst2, linesize, src2, EDGE_EMU_LINESIZE, block_h, mx, my);
1924 } else {
1925 70033 mc_func[my_idx][mx_idx](dst1, linesize, src1, linesize, block_h, mx, my);
1926 70033 mc_func[my_idx][mx_idx](dst2, linesize, src2, linesize, block_h, mx, my);
1927 }
1928 } else {
1929 346560 ff_progress_frame_await(ref, (3 + y_off + block_h) >> 3);
1930 346560 mc_func[0][0](dst1, linesize, src1 + y_off * linesize + x_off, linesize, block_h, 0, 0);
1931 346560 mc_func[0][0](dst2, linesize, src2 + y_off * linesize + x_off, linesize, block_h, 0, 0);
1932 }
1933 425750 }
1934
1935 static av_always_inline
1936 399290 void vp8_mc_part(VP8Context *s, VP8ThreadData *td, uint8_t *const dst[3],
1937 const ProgressFrame *ref_frame, int x_off, int y_off,
1938 int bx_off, int by_off, int block_w, int block_h,
1939 int width, int height, const VP8mv *mv)
1940 {
1941 399290 VP8mv uvmv = *mv;
1942
1943 /* Y */
1944 399290 vp8_mc_luma(s, td, dst[0] + by_off * s->linesize + bx_off,
1945 ref_frame, mv, x_off + bx_off, y_off + by_off,
1946 block_w, block_h, width, height, s->linesize,
1947 399290 s->put_pixels_tab[block_w == 8]);
1948
1949 /* U/V */
1950
2/2
✓ Branch 0 taken 4775 times.
✓ Branch 1 taken 394515 times.
399290 if (s->profile == 3) {
1951 /* this block only applies VP8; it is safe to check
1952 * only the profile, as VP7 profile <= 1 */
1953 4775 uvmv.x &= ~7;
1954 4775 uvmv.y &= ~7;
1955 }
1956 399290 x_off >>= 1;
1957 399290 y_off >>= 1;
1958 399290 bx_off >>= 1;
1959 399290 by_off >>= 1;
1960 399290 width >>= 1;
1961 399290 height >>= 1;
1962 399290 block_w >>= 1;
1963 399290 block_h >>= 1;
1964 399290 vp8_mc_chroma(s, td, dst[1] + by_off * s->uvlinesize + bx_off,
1965 399290 dst[2] + by_off * s->uvlinesize + bx_off, ref_frame,
1966 &uvmv, x_off + bx_off, y_off + by_off,
1967 block_w, block_h, width, height, s->uvlinesize,
1968
2/2
✓ Branch 0 taken 20950 times.
✓ Branch 1 taken 378340 times.
399290 s->put_pixels_tab[1 + (block_w == 4)]);
1969 399290 }
1970
1971 /* Fetch pixels for estimated mv 4 macroblocks ahead.
1972 * Optimized for 64-byte cache lines. Inspired by ffh264 prefetch_motion. */
1973 static av_always_inline
1974 1334967 void prefetch_motion(const VP8Context *s, const VP8Macroblock *mb,
1975 int mb_x, int mb_y, int mb_xy, int ref)
1976 {
1977 /* Don't prefetch refs that haven't been used very often this frame. */
1978
2/2
✓ Branch 0 taken 479891 times.
✓ Branch 1 taken 855076 times.
1334967 if (s->ref_count[ref - 1] > (mb_xy >> 5)) {
1979 479891 int x_off = mb_x << 4, y_off = mb_y << 4;
1980 479891 int mx = (mb->mv.x >> 2) + x_off + 8;
1981 479891 int my = (mb->mv.y >> 2) + y_off;
1982 479891 uint8_t **src = s->framep[ref]->tf.f->data;
1983 479891 int off = mx + (my + (mb_x & 3) * 4) * s->linesize + 64;
1984 /* For threading, a ff_thread_await_progress here might be useful, but
1985 * it actually slows down the decoder. Since a bad prefetch doesn't
1986 * generate bad decoder output, we don't run it here. */
1987 479891 s->vdsp.prefetch(src[0] + off, s->linesize, 4);
1988 479891 off = (mx >> 1) + ((my >> 1) + (mb_x & 7)) * s->uvlinesize + 64;
1989 479891 s->vdsp.prefetch(src[1] + off, src[2] - src[1], 2);
1990 }
1991 1334967 }
1992
1993 /**
1994 * Apply motion vectors to prediction buffer, chapter 18.
1995 */
1996 static av_always_inline
1997 386850 void inter_predict(VP8Context *s, VP8ThreadData *td, uint8_t *const dst[3],
1998 VP8Macroblock *mb, int mb_x, int mb_y)
1999 {
2000 386850 int x_off = mb_x << 4, y_off = mb_y << 4;
2001 386850 int width = 16 * s->mb_width, height = 16 * s->mb_height;
2002 386850 const ProgressFrame *ref = &s->framep[mb->ref_frame]->tf;
2003 386850 const VP8mv *bmv = mb->bmv;
2004
2005
5/6
✓ Branch 0 taken 368520 times.
✓ Branch 1 taken 6615 times.
✓ Branch 2 taken 4910 times.
✓ Branch 3 taken 3135 times.
✓ Branch 4 taken 3670 times.
✗ Branch 5 not taken.
386850 switch (mb->partitioning) {
2006 368520 case VP8_SPLITMVMODE_NONE:
2007 368520 vp8_mc_part(s, td, dst, ref, x_off, y_off,
2008 368520 0, 0, 16, 16, width, height, &mb->mv);
2009 368520 break;
2010 6615 case VP8_SPLITMVMODE_4x4: {
2011 int x, y;
2012 VP8mv uvmv;
2013
2014 /* Y */
2015
2/2
✓ Branch 0 taken 26460 times.
✓ Branch 1 taken 6615 times.
33075 for (y = 0; y < 4; y++) {
2016
2/2
✓ Branch 0 taken 105840 times.
✓ Branch 1 taken 26460 times.
132300 for (x = 0; x < 4; x++) {
2017 105840 vp8_mc_luma(s, td, dst[0] + 4 * y * s->linesize + x * 4,
2018 105840 ref, &bmv[4 * y + x],
2019 105840 4 * x + x_off, 4 * y + y_off, 4, 4,
2020 width, height, s->linesize,
2021 105840 s->put_pixels_tab[2]);
2022 }
2023 }
2024
2025 /* U/V */
2026 6615 x_off >>= 1;
2027 6615 y_off >>= 1;
2028 6615 width >>= 1;
2029 6615 height >>= 1;
2030
2/2
✓ Branch 0 taken 13230 times.
✓ Branch 1 taken 6615 times.
19845 for (y = 0; y < 2; y++) {
2031
2/2
✓ Branch 0 taken 26460 times.
✓ Branch 1 taken 13230 times.
39690 for (x = 0; x < 2; x++) {
2032 26460 uvmv.x = mb->bmv[2 * y * 4 + 2 * x ].x +
2033 26460 mb->bmv[2 * y * 4 + 2 * x + 1].x +
2034 26460 mb->bmv[(2 * y + 1) * 4 + 2 * x ].x +
2035 26460 mb->bmv[(2 * y + 1) * 4 + 2 * x + 1].x;
2036 26460 uvmv.y = mb->bmv[2 * y * 4 + 2 * x ].y +
2037 26460 mb->bmv[2 * y * 4 + 2 * x + 1].y +
2038 26460 mb->bmv[(2 * y + 1) * 4 + 2 * x ].y +
2039 26460 mb->bmv[(2 * y + 1) * 4 + 2 * x + 1].y;
2040 26460 uvmv.x = (uvmv.x + 2 + FF_SIGNBIT(uvmv.x)) >> 2;
2041 26460 uvmv.y = (uvmv.y + 2 + FF_SIGNBIT(uvmv.y)) >> 2;
2042
2/2
✓ Branch 0 taken 492 times.
✓ Branch 1 taken 25968 times.
26460 if (s->profile == 3) {
2043 492 uvmv.x &= ~7;
2044 492 uvmv.y &= ~7;
2045 }
2046 26460 vp8_mc_chroma(s, td, dst[1] + 4 * y * s->uvlinesize + x * 4,
2047 26460 dst[2] + 4 * y * s->uvlinesize + x * 4, ref,
2048 26460 &uvmv, 4 * x + x_off, 4 * y + y_off, 4, 4,
2049 width, height, s->uvlinesize,
2050 26460 s->put_pixels_tab[2]);
2051 }
2052 }
2053 6615 break;
2054 }
2055 4910 case VP8_SPLITMVMODE_16x8:
2056 4910 vp8_mc_part(s, td, dst, ref, x_off, y_off,
2057 0, 0, 16, 8, width, height, &bmv[0]);
2058 4910 vp8_mc_part(s, td, dst, ref, x_off, y_off,
2059 0, 8, 16, 8, width, height, &bmv[1]);
2060 4910 break;
2061 3135 case VP8_SPLITMVMODE_8x16:
2062 3135 vp8_mc_part(s, td, dst, ref, x_off, y_off,
2063 0, 0, 8, 16, width, height, &bmv[0]);
2064 3135 vp8_mc_part(s, td, dst, ref, x_off, y_off,
2065 8, 0, 8, 16, width, height, &bmv[1]);
2066 3135 break;
2067 3670 case VP8_SPLITMVMODE_8x8:
2068 3670 vp8_mc_part(s, td, dst, ref, x_off, y_off,
2069 0, 0, 8, 8, width, height, &bmv[0]);
2070 3670 vp8_mc_part(s, td, dst, ref, x_off, y_off,
2071 8, 0, 8, 8, width, height, &bmv[1]);
2072 3670 vp8_mc_part(s, td, dst, ref, x_off, y_off,
2073 0, 8, 8, 8, width, height, &bmv[2]);
2074 3670 vp8_mc_part(s, td, dst, ref, x_off, y_off,
2075 8, 8, 8, 8, width, height, &bmv[3]);
2076 3670 break;
2077 }
2078 386850 }
2079
2080 static av_always_inline
2081 77882 void idct_mb(VP8Context *s, VP8ThreadData *td, uint8_t *const dst[3],
2082 const VP8Macroblock *mb)
2083 {
2084 int x, y, ch;
2085
2086
2/2
✓ Branch 0 taken 60394 times.
✓ Branch 1 taken 17488 times.
77882 if (mb->mode != MODE_I4x4) {
2087 60394 uint8_t *y_dst = dst[0];
2088
2/2
✓ Branch 0 taken 241576 times.
✓ Branch 1 taken 60394 times.
301970 for (y = 0; y < 4; y++) {
2089 241576 uint32_t nnz4 = AV_RL32(td->non_zero_count_cache[y]);
2090
2/2
✓ Branch 0 taken 178713 times.
✓ Branch 1 taken 62863 times.
241576 if (nnz4) {
2091
2/2
✓ Branch 0 taken 45342 times.
✓ Branch 1 taken 133371 times.
178713 if (nnz4 & ~0x01010101) {
2092
1/2
✓ Branch 0 taken 157676 times.
✗ Branch 1 not taken.
157676 for (x = 0; x < 4; x++) {
2093
2/2
✓ Branch 0 taken 54946 times.
✓ Branch 1 taken 102730 times.
157676 if ((uint8_t) nnz4 == 1)
2094 54946 s->vp8dsp.vp8_idct_dc_add(y_dst + 4 * x,
2095 54946 td->block[y][x],
2096 s->linesize);
2097
2/2
✓ Branch 0 taken 75145 times.
✓ Branch 1 taken 27585 times.
102730 else if ((uint8_t) nnz4 > 1)
2098 75145 s->vp8dsp.vp8_idct_add(y_dst + 4 * x,
2099 75145 td->block[y][x],
2100 s->linesize);
2101 157676 nnz4 >>= 8;
2102
2/2
✓ Branch 0 taken 45342 times.
✓ Branch 1 taken 112334 times.
157676 if (!nnz4)
2103 45342 break;
2104 }
2105 } else {
2106 133371 s->vp8dsp.vp8_idct_dc_add4y(y_dst, td->block[y], s->linesize);
2107 }
2108 }
2109 241576 y_dst += 4 * s->linesize;
2110 }
2111 }
2112
2113
2/2
✓ Branch 0 taken 155764 times.
✓ Branch 1 taken 77882 times.
233646 for (ch = 0; ch < 2; ch++) {
2114 155764 uint32_t nnz4 = AV_RL32(td->non_zero_count_cache[4 + ch]);
2115
2/2
✓ Branch 0 taken 95400 times.
✓ Branch 1 taken 60364 times.
155764 if (nnz4) {
2116 60364 uint8_t *ch_dst = dst[1 + ch];
2117
2/2
✓ Branch 0 taken 30220 times.
✓ Branch 1 taken 30144 times.
60364 if (nnz4 & ~0x01010101) {
2118
1/2
✓ Branch 0 taken 55999 times.
✗ Branch 1 not taken.
55999 for (y = 0; y < 2; y++) {
2119
2/2
✓ Branch 0 taken 107235 times.
✓ Branch 1 taken 25779 times.
133014 for (x = 0; x < 2; x++) {
2120
2/2
✓ Branch 0 taken 16098 times.
✓ Branch 1 taken 91137 times.
107235 if ((uint8_t) nnz4 == 1)
2121 16098 s->vp8dsp.vp8_idct_dc_add(ch_dst + 4 * x,
2122 16098 td->block[4 + ch][(y << 1) + x],
2123 s->uvlinesize);
2124
2/2
✓ Branch 0 taken 67411 times.
✓ Branch 1 taken 23726 times.
91137 else if ((uint8_t) nnz4 > 1)
2125 67411 s->vp8dsp.vp8_idct_add(ch_dst + 4 * x,
2126 67411 td->block[4 + ch][(y << 1) + x],
2127 s->uvlinesize);
2128 107235 nnz4 >>= 8;
2129
2/2
✓ Branch 0 taken 30220 times.
✓ Branch 1 taken 77015 times.
107235 if (!nnz4)
2130 30220 goto chroma_idct_end;
2131 }
2132 25779 ch_dst += 4 * s->uvlinesize;
2133 }
2134 } else {
2135 30144 s->vp8dsp.vp8_idct_dc_add4uv(ch_dst, td->block[4 + ch], s->uvlinesize);
2136 }
2137 }
2138 155764 chroma_idct_end:
2139 ;
2140 }
2141 77882 }
2142
2143 static av_always_inline
2144 425389 void filter_level_for_mb(const VP8Context *s, const VP8Macroblock *mb,
2145 VP8FilterStrength *f, int is_vp7)
2146 {
2147 int interior_limit, filter_level;
2148
2149
2/2
✓ Branch 0 taken 102219 times.
✓ Branch 1 taken 323170 times.
425389 if (s->segmentation.enabled) {
2150 102219 filter_level = s->segmentation.filter_level[mb->segment];
2151
2/2
✓ Branch 0 taken 98895 times.
✓ Branch 1 taken 3324 times.
102219 if (!s->segmentation.absolute_vals)
2152 98895 filter_level += s->filter.level;
2153 } else
2154 323170 filter_level = s->filter.level;
2155
2156
2/2
✓ Branch 0 taken 417815 times.
✓ Branch 1 taken 7574 times.
425389 if (s->lf_delta.enabled) {
2157 417815 filter_level += s->lf_delta.ref[mb->ref_frame];
2158 417815 filter_level += s->lf_delta.mode[mb->mode];
2159 }
2160
2161 425389 filter_level = av_clip_uintp2(filter_level, 6);
2162
2163 425389 interior_limit = filter_level;
2164
2/2
✓ Branch 0 taken 2772 times.
✓ Branch 1 taken 422617 times.
425389 if (s->filter.sharpness) {
2165 2772 interior_limit >>= (s->filter.sharpness + 3) >> 2;
2166 2772 interior_limit = FFMIN(interior_limit, 9 - s->filter.sharpness);
2167 }
2168 425389 interior_limit = FFMAX(interior_limit, 1);
2169
2170 425389 f->filter_level = filter_level;
2171 425389 f->inner_limit = interior_limit;
2172
6/6
✓ Branch 0 taken 418569 times.
✓ Branch 1 taken 6820 times.
✓ Branch 2 taken 353604 times.
✓ Branch 3 taken 64965 times.
✓ Branch 4 taken 353316 times.
✓ Branch 5 taken 288 times.
778705 f->inner_filter = is_vp7 || !mb->skip || mb->mode == MODE_I4x4 ||
2173
2/2
✓ Branch 0 taken 3237 times.
✓ Branch 1 taken 350079 times.
353316 mb->mode == VP8_MVMODE_SPLIT;
2174 425389 }
2175
2176 static av_always_inline
2177 417667 void filter_mb(const VP8Context *s, uint8_t *const dst[3], const VP8FilterStrength *f,
2178 int mb_x, int mb_y, int is_vp7)
2179 {
2180 int mbedge_lim, bedge_lim_y, bedge_lim_uv, hev_thresh;
2181 417667 int filter_level = f->filter_level;
2182 417667 int inner_limit = f->inner_limit;
2183 417667 int inner_filter = f->inner_filter;
2184 417667 ptrdiff_t linesize = s->linesize;
2185 417667 ptrdiff_t uvlinesize = s->uvlinesize;
2186 static const uint8_t hev_thresh_lut[2][64] = {
2187 { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1,
2188 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
2189 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
2190 3, 3, 3, 3 },
2191 { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1,
2192 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2193 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
2194 2, 2, 2, 2 }
2195 };
2196
2197
2/2
✓ Branch 0 taken 5877 times.
✓ Branch 1 taken 411790 times.
417667 if (!filter_level)
2198 5877 return;
2199
2200
2/2
✓ Branch 0 taken 6820 times.
✓ Branch 1 taken 404970 times.
411790 if (is_vp7) {
2201 6820 bedge_lim_y = filter_level;
2202 6820 bedge_lim_uv = filter_level * 2;
2203 6820 mbedge_lim = filter_level + 2;
2204 } else {
2205 404970 bedge_lim_y =
2206 404970 bedge_lim_uv = filter_level * 2 + inner_limit;
2207 404970 mbedge_lim = bedge_lim_y + 4;
2208 }
2209
2210 411790 hev_thresh = hev_thresh_lut[s->keyframe][filter_level];
2211
2212
2/2
✓ Branch 0 taken 397619 times.
✓ Branch 1 taken 14171 times.
411790 if (mb_x) {
2213 397619 s->vp8dsp.vp8_h_loop_filter16y(dst[0], linesize,
2214 mbedge_lim, inner_limit, hev_thresh);
2215 397619 s->vp8dsp.vp8_h_loop_filter8uv(dst[1], dst[2], uvlinesize,
2216 mbedge_lim, inner_limit, hev_thresh);
2217 }
2218
2219 #define H_LOOP_FILTER_16Y_INNER(cond) \
2220 if (cond && inner_filter) { \
2221 s->vp8dsp.vp8_h_loop_filter16y_inner(dst[0] + 4, linesize, \
2222 bedge_lim_y, inner_limit, \
2223 hev_thresh); \
2224 s->vp8dsp.vp8_h_loop_filter16y_inner(dst[0] + 8, linesize, \
2225 bedge_lim_y, inner_limit, \
2226 hev_thresh); \
2227 s->vp8dsp.vp8_h_loop_filter16y_inner(dst[0] + 12, linesize, \
2228 bedge_lim_y, inner_limit, \
2229 hev_thresh); \
2230 s->vp8dsp.vp8_h_loop_filter8uv_inner(dst[1] + 4, dst[2] + 4, \
2231 uvlinesize, bedge_lim_uv, \
2232 inner_limit, hev_thresh); \
2233 }
2234
2235
4/4
✓ Branch 0 taken 404970 times.
✓ Branch 1 taken 6820 times.
✓ Branch 2 taken 63240 times.
✓ Branch 3 taken 341730 times.
411790 H_LOOP_FILTER_16Y_INNER(!is_vp7)
2236
2237
2/2
✓ Branch 0 taken 390676 times.
✓ Branch 1 taken 21114 times.
411790 if (mb_y) {
2238 390676 s->vp8dsp.vp8_v_loop_filter16y(dst[0], linesize,
2239 mbedge_lim, inner_limit, hev_thresh);
2240 390676 s->vp8dsp.vp8_v_loop_filter8uv(dst[1], dst[2], uvlinesize,
2241 mbedge_lim, inner_limit, hev_thresh);
2242 }
2243
2244
2/2
✓ Branch 0 taken 70060 times.
✓ Branch 1 taken 341730 times.
411790 if (inner_filter) {
2245 70060 s->vp8dsp.vp8_v_loop_filter16y_inner(dst[0] + 4 * linesize,
2246 linesize, bedge_lim_y,
2247 inner_limit, hev_thresh);
2248 70060 s->vp8dsp.vp8_v_loop_filter16y_inner(dst[0] + 8 * linesize,
2249 linesize, bedge_lim_y,
2250 inner_limit, hev_thresh);
2251 70060 s->vp8dsp.vp8_v_loop_filter16y_inner(dst[0] + 12 * linesize,
2252 linesize, bedge_lim_y,
2253 inner_limit, hev_thresh);
2254 70060 s->vp8dsp.vp8_v_loop_filter8uv_inner(dst[1] + 4 * uvlinesize,
2255 70060 dst[2] + 4 * uvlinesize,
2256 uvlinesize, bedge_lim_uv,
2257 inner_limit, hev_thresh);
2258 }
2259
2260
3/4
✓ Branch 0 taken 6820 times.
✓ Branch 1 taken 404970 times.
✓ Branch 2 taken 6820 times.
✗ Branch 3 not taken.
411790 H_LOOP_FILTER_16Y_INNER(is_vp7)
2261 }
2262
2263 static av_always_inline
2264 7722 void filter_mb_simple(const VP8Context *s, uint8_t *dst, const VP8FilterStrength *f,
2265 int mb_x, int mb_y)
2266 {
2267 int mbedge_lim, bedge_lim;
2268 7722 int filter_level = f->filter_level;
2269 7722 int inner_limit = f->inner_limit;
2270 7722 int inner_filter = f->inner_filter;
2271 7722 ptrdiff_t linesize = s->linesize;
2272
2273
2/2
✓ Branch 0 taken 332 times.
✓ Branch 1 taken 7390 times.
7722 if (!filter_level)
2274 332 return;
2275
2276 7390 bedge_lim = 2 * filter_level + inner_limit;
2277 7390 mbedge_lim = bedge_lim + 4;
2278
2279
2/2
✓ Branch 0 taken 6713 times.
✓ Branch 1 taken 677 times.
7390 if (mb_x)
2280 6713 s->vp8dsp.vp8_h_loop_filter_simple(dst, linesize, mbedge_lim);
2281
2/2
✓ Branch 0 taken 3523 times.
✓ Branch 1 taken 3867 times.
7390 if (inner_filter) {
2282 3523 s->vp8dsp.vp8_h_loop_filter_simple(dst + 4, linesize, bedge_lim);
2283 3523 s->vp8dsp.vp8_h_loop_filter_simple(dst + 8, linesize, bedge_lim);
2284 3523 s->vp8dsp.vp8_h_loop_filter_simple(dst + 12, linesize, bedge_lim);
2285 }
2286
2287
2/2
✓ Branch 0 taken 6609 times.
✓ Branch 1 taken 781 times.
7390 if (mb_y)
2288 6609 s->vp8dsp.vp8_v_loop_filter_simple(dst, linesize, mbedge_lim);
2289
2/2
✓ Branch 0 taken 3523 times.
✓ Branch 1 taken 3867 times.
7390 if (inner_filter) {
2290 3523 s->vp8dsp.vp8_v_loop_filter_simple(dst + 4 * linesize, linesize, bedge_lim);
2291 3523 s->vp8dsp.vp8_v_loop_filter_simple(dst + 8 * linesize, linesize, bedge_lim);
2292 3523 s->vp8dsp.vp8_v_loop_filter_simple(dst + 12 * linesize, linesize, bedge_lim);
2293 }
2294 }
2295
2296 #define MARGIN (16 << 2)
2297 static av_always_inline
2298 31 int vp78_decode_mv_mb_modes(AVCodecContext *avctx, VP8Frame *curframe,
2299 const VP8Frame *prev_frame, int is_vp7)
2300 {
2301 31 VP8Context *s = avctx->priv_data;
2302 int mb_x, mb_y;
2303
2304 31 s->mv_bounds.mv_min.y = -MARGIN;
2305 31 s->mv_bounds.mv_max.y = ((s->mb_height - 1) << 6) + MARGIN;
2306
2/2
✓ Branch 0 taken 341 times.
✓ Branch 1 taken 31 times.
372 for (mb_y = 0; mb_y < s->mb_height; mb_y++) {
2307 341 VP8Macroblock *mb = s->macroblocks_base +
2308 341 ((s->mb_width + 1) * (mb_y + 1) + 1);
2309 341 int mb_xy = mb_y * s->mb_width;
2310
2311 341 AV_WN32A(s->intra4x4_pred_mode_left, DC_PRED * 0x01010101);
2312
2313 341 s->mv_bounds.mv_min.x = -MARGIN;
2314 341 s->mv_bounds.mv_max.x = ((s->mb_width - 1) << 6) + MARGIN;
2315
2316
2/2
✓ Branch 0 taken 6820 times.
✓ Branch 1 taken 341 times.
7161 for (mb_x = 0; mb_x < s->mb_width; mb_x++, mb_xy++, mb++) {
2317
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 6820 times.
6820 if (vpx_rac_is_end(&s->c)) {
2318 return AVERROR_INVALIDDATA;
2319 }
2320
2/2
✓ Branch 0 taken 620 times.
✓ Branch 1 taken 6200 times.
6820 if (mb_y == 0)
2321 620 AV_WN32A((mb - s->mb_width - 1)->intra4x4_pred_mode_top,
2322 DC_PRED * 0x01010101);
2323
2/2
✓ Branch 0 taken 6600 times.
✓ Branch 1 taken 220 times.
13420 decode_mb_mode(s, &s->mv_bounds, mb, mb_x, mb_y, curframe->seg_map + mb_xy,
2324
1/2
✓ Branch 0 taken 6600 times.
✗ Branch 1 not taken.
6600 prev_frame && prev_frame->seg_map ?
2325 6600 prev_frame->seg_map + mb_xy : NULL, 1, is_vp7);
2326 6820 s->mv_bounds.mv_min.x -= 64;
2327 6820 s->mv_bounds.mv_max.x -= 64;
2328 }
2329 341 s->mv_bounds.mv_min.y -= 64;
2330 341 s->mv_bounds.mv_max.y -= 64;
2331 }
2332 31 return 0;
2333 }
2334
2335 31 static int vp7_decode_mv_mb_modes(AVCodecContext *avctx, VP8Frame *cur_frame,
2336 const VP8Frame *prev_frame)
2337 {
2338 31 return vp78_decode_mv_mb_modes(avctx, cur_frame, prev_frame, IS_VP7);
2339 }
2340
2341 static int vp8_decode_mv_mb_modes(AVCodecContext *avctx, VP8Frame *cur_frame,
2342 const VP8Frame *prev_frame)
2343 {
2344 return vp78_decode_mv_mb_modes(avctx, cur_frame, prev_frame, IS_VP8);
2345 }
2346
2347 #if HAVE_THREADS
2348 #define check_thread_pos(td, otd, mb_x_check, mb_y_check) \
2349 do { \
2350 int tmp = (mb_y_check << 16) | (mb_x_check & 0xFFFF); \
2351 if (atomic_load(&otd->thread_mb_pos) < tmp) { \
2352 pthread_mutex_lock(&otd->lock); \
2353 atomic_store(&td->wait_mb_pos, tmp); \
2354 do { \
2355 if (atomic_load(&otd->thread_mb_pos) >= tmp) \
2356 break; \
2357 pthread_cond_wait(&otd->cond, &otd->lock); \
2358 } while (1); \
2359 atomic_store(&td->wait_mb_pos, INT_MAX); \
2360 pthread_mutex_unlock(&otd->lock); \
2361 } \
2362 } while (0)
2363
2364 #define update_pos(td, mb_y, mb_x) \
2365 do { \
2366 int pos = (mb_y << 16) | (mb_x & 0xFFFF); \
2367 int sliced_threading = (avctx->active_thread_type == FF_THREAD_SLICE) && \
2368 (num_jobs > 1); \
2369 int is_null = !next_td || !prev_td; \
2370 int pos_check = (is_null) ? 1 : \
2371 (next_td != td && pos >= atomic_load(&next_td->wait_mb_pos)) || \
2372 (prev_td != td && pos >= atomic_load(&prev_td->wait_mb_pos)); \
2373 atomic_store(&td->thread_mb_pos, pos); \
2374 if (sliced_threading && pos_check) { \
2375 pthread_mutex_lock(&td->lock); \
2376 pthread_cond_broadcast(&td->cond); \
2377 pthread_mutex_unlock(&td->lock); \
2378 } \
2379 } while (0)
2380 #else
2381 #define check_thread_pos(td, otd, mb_x_check, mb_y_check) while(0)
2382 #define update_pos(td, mb_y, mb_x) while(0)
2383 #endif
2384
2385 16282 static av_always_inline int decode_mb_row_no_filter(AVCodecContext *avctx, void *tdata,
2386 int jobnr, int threadnr, int is_vp7)
2387 {
2388 16282 VP8Context *s = avctx->priv_data;
2389 16282 VP8ThreadData *prev_td, *next_td, *td = &s->thread_data[threadnr];
2390 16282 int mb_y = atomic_load(&td->thread_mb_pos) >> 16;
2391 16282 int mb_x, mb_xy = mb_y * s->mb_width;
2392 16282 int num_jobs = s->num_jobs;
2393 16282 const VP8Frame *prev_frame = s->prev_frame;
2394 16282 VP8Frame *curframe = s->curframe;
2395 16282 VPXRangeCoder *coeff_c = &s->coeff_partition[mb_y & (s->num_coeff_partitions - 1)];
2396
2397 VP8Macroblock *mb;
2398 16282 uint8_t *dst[3] = {
2399 16282 curframe->tf.f->data[0] + 16 * mb_y * s->linesize,
2400 16282 curframe->tf.f->data[1] + 8 * mb_y * s->uvlinesize,
2401 16282 curframe->tf.f->data[2] + 8 * mb_y * s->uvlinesize
2402 };
2403
2404
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 16282 times.
16282 if (vpx_rac_is_end(&s->c))
2405 return AVERROR_INVALIDDATA;
2406
2407
2/2
✓ Branch 0 taken 1151 times.
✓ Branch 1 taken 15131 times.
16282 if (mb_y == 0)
2408 1151 prev_td = td;
2409 else
2410 15131 prev_td = &s->thread_data[(jobnr + num_jobs - 1) % num_jobs];
2411
2/2
✓ Branch 0 taken 1151 times.
✓ Branch 1 taken 15131 times.
16282 if (mb_y == s->mb_height - 1)
2412 1151 next_td = td;
2413 else
2414 15131 next_td = &s->thread_data[(jobnr + 1) % num_jobs];
2415
2/2
✓ Branch 0 taken 341 times.
✓ Branch 1 taken 15941 times.
16282 if (s->mb_layout == 1)
2416 341 mb = s->macroblocks_base + ((s->mb_width + 1) * (mb_y + 1) + 1);
2417 else {
2418 // Make sure the previous frame has read its segmentation map,
2419 // if we re-use the same map.
2420
4/4
✓ Branch 0 taken 15311 times.
✓ Branch 1 taken 630 times.
✓ Branch 2 taken 5598 times.
✓ Branch 3 taken 9713 times.
15941 if (prev_frame && s->segmentation.enabled &&
2421
2/2
✓ Branch 0 taken 5025 times.
✓ Branch 1 taken 573 times.
5598 !s->segmentation.update_map)
2422 5025 ff_progress_frame_await(&prev_frame->tf, mb_y);
2423 15941 mb = s->macroblocks + (s->mb_height - mb_y - 1) * 2;
2424 15941 memset(mb - 1, 0, sizeof(*mb)); // zero left macroblock
2425 15941 AV_WN32A(s->intra4x4_pred_mode_left, DC_PRED * 0x01010101);
2426 }
2427
2428
4/4
✓ Branch 0 taken 341 times.
✓ Branch 1 taken 15941 times.
✓ Branch 2 taken 31 times.
✓ Branch 3 taken 310 times.
16282 if (!is_vp7 || mb_y == 0)
2429 15972 memset(td->left_nnz, 0, sizeof(td->left_nnz));
2430
2431 16282 td->mv_bounds.mv_min.x = -MARGIN;
2432 16282 td->mv_bounds.mv_max.x = ((s->mb_width - 1) << 6) + MARGIN;
2433
2434
2/2
✓ Branch 0 taken 444989 times.
✓ Branch 1 taken 16282 times.
461271 for (mb_x = 0; mb_x < s->mb_width; mb_x++, mb_xy++, mb++) {
2435
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 444989 times.
444989 if (vpx_rac_is_end(&s->c))
2436 return AVERROR_INVALIDDATA;
2437 // Wait for previous thread to read mb_x+2, and reach mb_y-1.
2438
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 444989 times.
444989 if (prev_td != td) {
2439 if (threadnr != 0) {
2440 check_thread_pos(td, prev_td,
2441 mb_x + (is_vp7 ? 2 : 1),
2442 mb_y - (is_vp7 ? 2 : 1));
2443 } else {
2444 check_thread_pos(td, prev_td,
2445 mb_x + (is_vp7 ? 2 : 1) + s->mb_width + 3,
2446 mb_y - (is_vp7 ? 2 : 1));
2447 }
2448 }
2449
2450 444989 s->vdsp.prefetch(dst[0] + (mb_x & 3) * 4 * s->linesize + 64,
2451 s->linesize, 4);
2452 444989 s->vdsp.prefetch(dst[1] + (mb_x & 7) * s->uvlinesize + 64,
2453 444989 dst[2] - dst[1], 2);
2454
2455
2/2
✓ Branch 0 taken 438169 times.
✓ Branch 1 taken 6820 times.
444989 if (!s->mb_layout)
2456
2/2
✓ Branch 0 taken 406889 times.
✓ Branch 1 taken 31280 times.
845058 decode_mb_mode(s, &td->mv_bounds, mb, mb_x, mb_y, curframe->seg_map + mb_xy,
2457
1/2
✓ Branch 0 taken 406889 times.
✗ Branch 1 not taken.
406889 prev_frame && prev_frame->seg_map ?
2458 406889 prev_frame->seg_map + mb_xy : NULL, 0, is_vp7);
2459
2460 444989 prefetch_motion(s, mb, mb_x, mb_y, mb_xy, VP8_FRAME_PREVIOUS);
2461
2462
2/2
✓ Branch 0 taken 84558 times.
✓ Branch 1 taken 360431 times.
444989 if (!mb->skip) {
2463
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 84558 times.
84558 if (vpx_rac_is_end(coeff_c))
2464 return AVERROR_INVALIDDATA;
2465 84558 decode_mb_coeffs(s, td, coeff_c, mb, s->top_nnz[mb_x], td->left_nnz, is_vp7);
2466 }
2467
2468
2/2
✓ Branch 0 taken 58139 times.
✓ Branch 1 taken 386850 times.
444989 if (mb->mode <= MODE_I4x4)
2469 58139 intra_predict(s, td, dst, mb, mb_x, mb_y, is_vp7);
2470 else
2471 386850 inter_predict(s, td, dst, mb, mb_x, mb_y);
2472
2473 444989 prefetch_motion(s, mb, mb_x, mb_y, mb_xy, VP8_FRAME_GOLDEN);
2474
2475
2/2
✓ Branch 0 taken 77882 times.
✓ Branch 1 taken 367107 times.
444989 if (!mb->skip) {
2476 77882 idct_mb(s, td, dst, mb);
2477 } else {
2478 367107 AV_ZERO64(td->left_nnz);
2479 367107 AV_WN64(s->top_nnz[mb_x], 0); // array of 9, so unaligned
2480
2481 /* Reset DC block predictors if they would exist
2482 * if the mb had coefficients */
2483
4/4
✓ Branch 0 taken 366627 times.
✓ Branch 1 taken 480 times.
✓ Branch 2 taken 363360 times.
✓ Branch 3 taken 3267 times.
367107 if (mb->mode != MODE_I4x4 && mb->mode != VP8_MVMODE_SPLIT) {
2484 363360 td->left_nnz[8] = 0;
2485 363360 s->top_nnz[mb_x][8] = 0;
2486 }
2487 }
2488
2489
2/2
✓ Branch 0 taken 425389 times.
✓ Branch 1 taken 19600 times.
444989 if (s->deblock_filter)
2490 425389 filter_level_for_mb(s, mb, &td->filter_strength[mb_x], is_vp7);
2491
2492
3/6
✓ Branch 0 taken 425389 times.
✓ Branch 1 taken 19600 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 425389 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
444989 if (s->deblock_filter && num_jobs != 1 && threadnr == num_jobs - 1) {
2493 if (s->filter.simple)
2494 backup_mb_border(s->top_border[mb_x + 1], dst[0],
2495 NULL, NULL, s->linesize, 0, 1);
2496 else
2497 backup_mb_border(s->top_border[mb_x + 1], dst[0],
2498 dst[1], dst[2], s->linesize, s->uvlinesize, 0);
2499 }
2500
2501 444989 prefetch_motion(s, mb, mb_x, mb_y, mb_xy, VP8_FRAME_ALTREF);
2502
2503 444989 dst[0] += 16;
2504 444989 dst[1] += 8;
2505 444989 dst[2] += 8;
2506 444989 td->mv_bounds.mv_min.x -= 64;
2507 444989 td->mv_bounds.mv_max.x -= 64;
2508
2509
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 444989 times.
444989 if (mb_x == s->mb_width + 1) {
2510 update_pos(td, mb_y, s->mb_width + 3);
2511 } else {
2512
7/22
✗ Branch 0 not taken.
✓ Branch 1 taken 444989 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 444989 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 444989 times.
✓ Branch 8 taken 444989 times.
✗ Branch 9 not taken.
✗ Branch 10 not taken.
✓ Branch 11 taken 444989 times.
✗ Branch 12 not taken.
✗ Branch 13 not taken.
✗ Branch 14 not taken.
✓ Branch 15 taken 444989 times.
✗ Branch 16 not taken.
✗ Branch 17 not taken.
✗ Branch 18 not taken.
✓ Branch 19 taken 444989 times.
✗ Branch 20 not taken.
✗ Branch 21 not taken.
444989 update_pos(td, mb_y, mb_x);
2513 }
2514 }
2515 16282 return 0;
2516 }
2517
2518 15181 static av_always_inline void filter_mb_row(AVCodecContext *avctx, void *tdata,
2519 int jobnr, int threadnr, int is_vp7)
2520 {
2521 15181 VP8Context *s = avctx->priv_data;
2522 15181 VP8ThreadData *td = &s->thread_data[threadnr];
2523 15181 int mb_x, mb_y = atomic_load(&td->thread_mb_pos) >> 16, num_jobs = s->num_jobs;
2524 15181 AVFrame *curframe = s->curframe->tf.f;
2525 VP8Macroblock *mb;
2526 VP8ThreadData *prev_td, *next_td;
2527 15181 uint8_t *dst[3] = {
2528 15181 curframe->data[0] + 16 * mb_y * s->linesize,
2529 15181 curframe->data[1] + 8 * mb_y * s->uvlinesize,
2530 15181 curframe->data[2] + 8 * mb_y * s->uvlinesize
2531 };
2532
2533
2/2
✓ Branch 0 taken 341 times.
✓ Branch 1 taken 14840 times.
15181 if (s->mb_layout == 1)
2534 341 mb = s->macroblocks_base + ((s->mb_width + 1) * (mb_y + 1) + 1);
2535 else
2536 14840 mb = s->macroblocks + (s->mb_height - mb_y - 1) * 2;
2537
2538
2/2
✓ Branch 0 taken 1040 times.
✓ Branch 1 taken 14141 times.
15181 if (mb_y == 0)
2539 1040 prev_td = td;
2540 else
2541 14141 prev_td = &s->thread_data[(jobnr + num_jobs - 1) % num_jobs];
2542
2/2
✓ Branch 0 taken 1040 times.
✓ Branch 1 taken 14141 times.
15181 if (mb_y == s->mb_height - 1)
2543 1040 next_td = td;
2544 else
2545 14141 next_td = &s->thread_data[(jobnr + 1) % num_jobs];
2546
2547
2/2
✓ Branch 0 taken 425389 times.
✓ Branch 1 taken 15181 times.
440570 for (mb_x = 0; mb_x < s->mb_width; mb_x++, mb++) {
2548 425389 const VP8FilterStrength *f = &td->filter_strength[mb_x];
2549
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 425389 times.
425389 if (prev_td != td)
2550 check_thread_pos(td, prev_td,
2551 (mb_x + 1) + (s->mb_width + 3), mb_y - 1);
2552
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 425389 times.
425389 if (next_td != td)
2553 if (next_td != &s->thread_data[0])
2554 check_thread_pos(td, next_td, mb_x + 1, mb_y + 1);
2555
2556
1/2
✓ Branch 0 taken 425389 times.
✗ Branch 1 not taken.
425389 if (num_jobs == 1) {
2557
2/2
✓ Branch 0 taken 7722 times.
✓ Branch 1 taken 417667 times.
425389 if (s->filter.simple)
2558 7722 backup_mb_border(s->top_border[mb_x + 1], dst[0],
2559 NULL, NULL, s->linesize, 0, 1);
2560 else
2561 417667 backup_mb_border(s->top_border[mb_x + 1], dst[0],
2562 417667 dst[1], dst[2], s->linesize, s->uvlinesize, 0);
2563 }
2564
2565
2/2
✓ Branch 0 taken 7722 times.
✓ Branch 1 taken 417667 times.
425389 if (s->filter.simple)
2566 7722 filter_mb_simple(s, dst[0], f, mb_x, mb_y);
2567 else
2568 417667 filter_mb(s, dst, f, mb_x, mb_y, is_vp7);
2569 425389 dst[0] += 16;
2570 425389 dst[1] += 8;
2571 425389 dst[2] += 8;
2572
2573
7/22
✗ Branch 0 not taken.
✓ Branch 1 taken 425389 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 425389 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 425389 times.
✓ Branch 8 taken 425389 times.
✗ Branch 9 not taken.
✗ Branch 10 not taken.
✓ Branch 11 taken 425389 times.
✗ Branch 12 not taken.
✗ Branch 13 not taken.
✗ Branch 14 not taken.
✓ Branch 15 taken 425389 times.
✗ Branch 16 not taken.
✗ Branch 17 not taken.
✗ Branch 18 not taken.
✓ Branch 19 taken 425389 times.
✗ Branch 20 not taken.
✗ Branch 21 not taken.
425389 update_pos(td, mb_y, (s->mb_width + 3) + mb_x);
2574 }
2575 15181 }
2576
2577 static av_always_inline
2578 1151 int vp78_decode_mb_row_sliced(AVCodecContext *avctx, void *tdata, int jobnr,
2579 int threadnr, int is_vp7)
2580 {
2581 1151 const VP8Context *s = avctx->priv_data;
2582 1151 VP8ThreadData *td = &s->thread_data[jobnr];
2583 1151 VP8ThreadData *next_td = NULL, *prev_td = NULL;
2584 1151 VP8Frame *curframe = s->curframe;
2585 1151 int mb_y, num_jobs = s->num_jobs;
2586 int ret;
2587
2588 1151 td->thread_nr = threadnr;
2589 1151 td->mv_bounds.mv_min.y = -MARGIN - 64 * threadnr;
2590 1151 td->mv_bounds.mv_max.y = ((s->mb_height - 1) << 6) + MARGIN - 64 * threadnr;
2591
2/2
✓ Branch 0 taken 16282 times.
✓ Branch 1 taken 1151 times.
17433 for (mb_y = jobnr; mb_y < s->mb_height; mb_y += num_jobs) {
2592 16282 atomic_store(&td->thread_mb_pos, mb_y << 16);
2593 16282 ret = s->decode_mb_row_no_filter(avctx, tdata, jobnr, threadnr);
2594
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16282 times.
16282 if (ret < 0) {
2595 update_pos(td, s->mb_height, INT_MAX & 0xFFFF);
2596 return ret;
2597 }
2598
2/2
✓ Branch 0 taken 15181 times.
✓ Branch 1 taken 1101 times.
16282 if (s->deblock_filter)
2599 15181 s->filter_mb_row(avctx, tdata, jobnr, threadnr);
2600
4/22
✗ Branch 0 not taken.
✓ Branch 1 taken 16282 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 16282 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✓ Branch 9 taken 16282 times.
✗ Branch 10 not taken.
✗ Branch 11 not taken.
✗ Branch 12 not taken.
✗ Branch 13 not taken.
✗ Branch 14 not taken.
✗ Branch 15 not taken.
✗ Branch 16 not taken.
✗ Branch 17 not taken.
✗ Branch 18 not taken.
✓ Branch 19 taken 16282 times.
✗ Branch 20 not taken.
✗ Branch 21 not taken.
16282 update_pos(td, mb_y, INT_MAX & 0xFFFF);
2601
2602 16282 td->mv_bounds.mv_min.y -= 64 * num_jobs;
2603 16282 td->mv_bounds.mv_max.y -= 64 * num_jobs;
2604
2605
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16282 times.
16282 if (avctx->active_thread_type == FF_THREAD_FRAME)
2606 ff_progress_frame_report(&curframe->tf, mb_y);
2607 }
2608
2609 1151 return 0;
2610 }
2611
2612 31 static int vp7_decode_mb_row_sliced(AVCodecContext *avctx, void *tdata,
2613 int jobnr, int threadnr)
2614 {
2615 31 return vp78_decode_mb_row_sliced(avctx, tdata, jobnr, threadnr, IS_VP7);
2616 }
2617
2618 1120 static int vp8_decode_mb_row_sliced(AVCodecContext *avctx, void *tdata,
2619 int jobnr, int threadnr)
2620 {
2621 1120 return vp78_decode_mb_row_sliced(avctx, tdata, jobnr, threadnr, IS_VP8);
2622 }
2623
2624 static av_always_inline
2625 1151 int vp78_decode_frame(AVCodecContext *avctx, AVFrame *rframe, int *got_frame,
2626 const AVPacket *avpkt, int is_vp7)
2627 {
2628 1151 VP8Context *s = avctx->priv_data;
2629 int ret, i, referenced, num_jobs;
2630 enum AVDiscard skip_thresh;
2631 1151 VP8Frame *av_uninit(curframe), *prev_frame;
2632
2633
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 1120 times.
1151 if (is_vp7)
2634 31 ret = vp7_decode_frame_header(s, avpkt->data, avpkt->size);
2635 else
2636 1120 ret = vp8_decode_frame_header(s, avpkt->data, avpkt->size);
2637
2638
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1151 times.
1151 if (ret < 0)
2639 goto err;
2640
2641
4/4
✓ Branch 0 taken 1120 times.
✓ Branch 1 taken 31 times.
✓ Branch 2 taken 6 times.
✓ Branch 3 taken 1114 times.
1151 if (!is_vp7 && s->actually_webp) {
2642 // VP8 in WebP is supposed to be intra-only. Enforce this here
2643 // to ensure that output is reproducible with frame-threading.
2644
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
6 if (!s->keyframe)
2645 return AVERROR_INVALIDDATA;
2646 // avctx->pix_fmt already set in caller.
2647
3/4
✓ Branch 0 taken 1114 times.
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1114 times.
1145 } else if (!is_vp7 && s->pix_fmt == AV_PIX_FMT_NONE) {
2648 s->pix_fmt = get_pixel_format(s);
2649 if (s->pix_fmt < 0) {
2650 ret = AVERROR(EINVAL);
2651 goto err;
2652 }
2653 avctx->pix_fmt = s->pix_fmt;
2654 }
2655
2656 1151 prev_frame = s->framep[VP8_FRAME_CURRENT];
2657
2658
3/4
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 1140 times.
✓ Branch 2 taken 11 times.
✗ Branch 3 not taken.
1162 referenced = s->update_last || s->update_golden == VP8_FRAME_CURRENT ||
2659
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 2 times.
11 s->update_altref == VP8_FRAME_CURRENT;
2660
2661 1151 skip_thresh = !referenced ? AVDISCARD_NONREF
2662
4/4
✓ Branch 0 taken 1149 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 1097 times.
✓ Branch 3 taken 52 times.
1151 : !s->keyframe ? AVDISCARD_NONKEY
2663 : AVDISCARD_ALL;
2664
2665
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1151 times.
1151 if (avctx->skip_frame >= skip_thresh) {
2666 s->invisible = 1;
2667 memcpy(&s->next_framep[0], &s->framep[0], sizeof(s->framep[0]) * 4);
2668 goto skip_decode;
2669 }
2670
3/4
✓ Branch 0 taken 1040 times.
✓ Branch 1 taken 111 times.
✓ Branch 2 taken 1040 times.
✗ Branch 3 not taken.
1151 s->deblock_filter = s->filter.level && avctx->skip_loop_filter < skip_thresh;
2671
2672 // release no longer referenced frames
2673
2/2
✓ Branch 0 taken 5755 times.
✓ Branch 1 taken 1151 times.
6906 for (i = 0; i < 5; i++)
2674
2/2
✓ Branch 0 taken 3836 times.
✓ Branch 1 taken 1919 times.
5755 if (s->frames[i].tf.f &&
2675
2/2
✓ Branch 0 taken 2725 times.
✓ Branch 1 taken 1111 times.
3836 &s->frames[i] != prev_frame &&
2676
2/2
✓ Branch 0 taken 2714 times.
✓ Branch 1 taken 11 times.
2725 &s->frames[i] != s->framep[VP8_FRAME_PREVIOUS] &&
2677
2/2
✓ Branch 0 taken 1742 times.
✓ Branch 1 taken 972 times.
2714 &s->frames[i] != s->framep[VP8_FRAME_GOLDEN] &&
2678
2/2
✓ Branch 0 taken 1034 times.
✓ Branch 1 taken 708 times.
1742 &s->frames[i] != s->framep[VP8_FRAME_ALTREF])
2679 1034 vp8_release_frame(&s->frames[i]);
2680
2681
1/2
✓ Branch 0 taken 1151 times.
✗ Branch 1 not taken.
1151 if (!s->colorspace)
2682 1151 avctx->colorspace = AVCOL_SPC_BT470BG;
2683
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1151 times.
1151 if (s->fullrange)
2684 avctx->color_range = AVCOL_RANGE_JPEG;
2685 else
2686 1151 avctx->color_range = AVCOL_RANGE_MPEG;
2687
2688 /* Given that arithmetic probabilities are updated every frame, it's quite
2689 * likely that the values we have on a random interframe are complete
2690 * junk if we didn't start decode on a keyframe. So just don't display
2691 * anything rather than junk. */
2692
3/4
✓ Branch 0 taken 1099 times.
✓ Branch 1 taken 52 times.
✓ Branch 2 taken 1099 times.
✗ Branch 3 not taken.
1151 if (!s->keyframe && (!s->framep[VP8_FRAME_PREVIOUS] ||
2693
1/2
✓ Branch 0 taken 1099 times.
✗ Branch 1 not taken.
1099 !s->framep[VP8_FRAME_GOLDEN] ||
2694
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1099 times.
1099 !s->framep[VP8_FRAME_ALTREF])) {
2695 av_log(avctx, AV_LOG_WARNING,
2696 "Discarding interframe without a prior keyframe!\n");
2697 ret = AVERROR_INVALIDDATA;
2698 goto err;
2699 }
2700
2701 1151 curframe = vp8_find_free_buffer(s);
2702
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1151 times.
1151 if ((ret = vp8_alloc_frame(s, curframe, referenced)) < 0)
2703 goto err;
2704 1151 s->framep[VP8_FRAME_CURRENT] = curframe;
2705
2/2
✓ Branch 0 taken 52 times.
✓ Branch 1 taken 1099 times.
1151 if (s->keyframe)
2706 52 curframe->tf.f->flags |= AV_FRAME_FLAG_KEY;
2707 else
2708 1099 curframe->tf.f->flags &= ~AV_FRAME_FLAG_KEY;
2709 2302 curframe->tf.f->pict_type = s->keyframe ? AV_PICTURE_TYPE_I
2710
2/2
✓ Branch 0 taken 52 times.
✓ Branch 1 taken 1099 times.
1151 : AV_PICTURE_TYPE_P;
2711
2712 // check if golden and altref are swapped
2713
2/2
✓ Branch 0 taken 200 times.
✓ Branch 1 taken 951 times.
1151 if (s->update_altref != VP8_FRAME_NONE)
2714 200 s->next_framep[VP8_FRAME_ALTREF] = s->framep[s->update_altref];
2715 else
2716 951 s->next_framep[VP8_FRAME_ALTREF] = s->framep[VP8_FRAME_ALTREF];
2717
2718
2/2
✓ Branch 0 taken 142 times.
✓ Branch 1 taken 1009 times.
1151 if (s->update_golden != VP8_FRAME_NONE)
2719 142 s->next_framep[VP8_FRAME_GOLDEN] = s->framep[s->update_golden];
2720 else
2721 1009 s->next_framep[VP8_FRAME_GOLDEN] = s->framep[VP8_FRAME_GOLDEN];
2722
2723
2/2
✓ Branch 0 taken 1140 times.
✓ Branch 1 taken 11 times.
1151 if (s->update_last)
2724 1140 s->next_framep[VP8_FRAME_PREVIOUS] = curframe;
2725 else
2726 11 s->next_framep[VP8_FRAME_PREVIOUS] = s->framep[VP8_FRAME_PREVIOUS];
2727
2728 1151 s->next_framep[VP8_FRAME_CURRENT] = curframe;
2729
2730
4/4
✓ Branch 0 taken 1120 times.
✓ Branch 1 taken 31 times.
✓ Branch 2 taken 1114 times.
✓ Branch 3 taken 6 times.
1151 if (!is_vp7 && !s->actually_webp)
2731 1114 ff_thread_finish_setup(avctx);
2732
2733
3/4
✓ Branch 0 taken 1120 times.
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1120 times.
1151 if (!is_vp7 && avctx->hwaccel) {
2734 const FFHWAccel *hwaccel = ffhwaccel(avctx->hwaccel);
2735 ret = hwaccel->start_frame(avctx, avpkt->buf, avpkt->data, avpkt->size);
2736 if (ret < 0)
2737 goto err;
2738
2739 ret = hwaccel->decode_slice(avctx, avpkt->data, avpkt->size);
2740 if (ret < 0)
2741 goto err;
2742
2743 ret = hwaccel->end_frame(avctx);
2744 if (ret < 0)
2745 goto err;
2746
2747 } else {
2748 1151 s->linesize = curframe->tf.f->linesize[0];
2749 1151 s->uvlinesize = curframe->tf.f->linesize[1];
2750
2751 1151 memset(s->top_nnz, 0, s->mb_width * sizeof(*s->top_nnz));
2752 /* Zero macroblock structures for top/top-left prediction
2753 * from outside the frame. */
2754
2/2
✓ Branch 0 taken 1120 times.
✓ Branch 1 taken 31 times.
1151 if (!s->mb_layout)
2755 1120 memset(s->macroblocks + s->mb_height * 2 - 1, 0,
2756 1120 (s->mb_width + 1) * sizeof(*s->macroblocks));
2757
4/4
✓ Branch 0 taken 1120 times.
✓ Branch 1 taken 31 times.
✓ Branch 2 taken 51 times.
✓ Branch 3 taken 1069 times.
1151 if (!s->mb_layout && s->keyframe)
2758 51 memset(s->intra4x4_pred_mode_top, DC_PRED, s->mb_width * 4);
2759
2760 1151 memset(s->ref_count, 0, sizeof(s->ref_count));
2761
2762
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 1120 times.
1151 if (s->mb_layout == 1) {
2763 // Make sure the previous frame has read its segmentation map,
2764 // if we re-use the same map.
2765
3/4
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 30 times.
31 if (prev_frame && s->segmentation.enabled &&
2766 !s->segmentation.update_map)
2767 ff_progress_frame_await(&prev_frame->tf, 1);
2768
1/2
✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
31 if (is_vp7)
2769 31 ret = vp7_decode_mv_mb_modes(avctx, curframe, prev_frame);
2770 else
2771 ret = vp8_decode_mv_mb_modes(avctx, curframe, prev_frame);
2772
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
31 if (ret < 0)
2773 goto err;
2774 }
2775
2776
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1151 times.
1151 if (avctx->active_thread_type == FF_THREAD_FRAME)
2777 num_jobs = 1;
2778 else
2779 1151 num_jobs = FFMIN(s->num_coeff_partitions, avctx->thread_count);
2780 1151 s->num_jobs = num_jobs;
2781 1151 s->curframe = curframe;
2782 1151 s->prev_frame = prev_frame;
2783 1151 s->mv_bounds.mv_min.y = -MARGIN;
2784 1151 s->mv_bounds.mv_max.y = ((s->mb_height - 1) << 6) + MARGIN;
2785
2/2
✓ Branch 0 taken 9208 times.
✓ Branch 1 taken 1151 times.
10359 for (i = 0; i < MAX_THREADS; i++) {
2786 9208 VP8ThreadData *td = &s->thread_data[i];
2787 9208 atomic_init(&td->thread_mb_pos, 0);
2788 9208 atomic_init(&td->wait_mb_pos, INT_MAX);
2789 }
2790
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 1120 times.
1151 if (is_vp7)
2791 31 avctx->execute2(avctx, vp7_decode_mb_row_sliced, s->thread_data, NULL,
2792 num_jobs);
2793 else
2794 1120 avctx->execute2(avctx, vp8_decode_mb_row_sliced, s->thread_data, NULL,
2795 num_jobs);
2796 }
2797
2798 1151 ff_progress_frame_report(&curframe->tf, INT_MAX);
2799 1151 memcpy(&s->framep[0], &s->next_framep[0], sizeof(s->framep[0]) * 4);
2800
2801 1151 skip_decode:
2802 // if future frames don't use the updated probabilities,
2803 // reset them to the values we saved
2804
2/2
✓ Branch 0 taken 64 times.
✓ Branch 1 taken 1087 times.
1151 if (!s->update_probabilities)
2805 64 s->prob[0] = s->prob[1];
2806
2807
2/2
✓ Branch 0 taken 1142 times.
✓ Branch 1 taken 9 times.
1151 if (!s->invisible) {
2808
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1142 times.
1142 if ((ret = av_frame_ref(rframe, curframe->tf.f)) < 0)
2809 return ret;
2810 1142 *got_frame = 1;
2811 }
2812
2813 1151 return avpkt->size;
2814 err:
2815 memcpy(&s->next_framep[0], &s->framep[0], sizeof(s->framep[0]) * 4);
2816 return ret;
2817 }
2818
2819 63 av_cold int ff_vp8_decode_free(AVCodecContext *avctx)
2820 {
2821 63 vp8_decode_flush_impl(avctx, 1);
2822
2823 63 return 0;
2824 }
2825
2826 63 static av_cold void vp78_decode_init(AVCodecContext *avctx)
2827 {
2828 63 VP8Context *s = avctx->priv_data;
2829
2830 63 s->avctx = avctx;
2831 63 s->pix_fmt = AV_PIX_FMT_NONE;
2832 63 avctx->pix_fmt = AV_PIX_FMT_YUV420P;
2833
2834 63 ff_videodsp_init(&s->vdsp, 8);
2835
2836 63 ff_vp78dsp_init(&s->vp8dsp);
2837
2838 /* does not change for VP8 */
2839 63 memcpy(s->prob[0].scan, ff_zigzag_scan, sizeof(s->prob[0].scan));
2840 63 }
2841
2842 #if CONFIG_VP8_DECODER
2843 15941 static int vp8_decode_mb_row_no_filter(AVCodecContext *avctx, void *tdata,
2844 int jobnr, int threadnr)
2845 {
2846 15941 return decode_mb_row_no_filter(avctx, tdata, jobnr, threadnr, 0);
2847 }
2848
2849 14840 static void vp8_filter_mb_row(AVCodecContext *avctx, void *tdata,
2850 int jobnr, int threadnr)
2851 {
2852 14840 filter_mb_row(avctx, tdata, jobnr, threadnr, 0);
2853 14840 }
2854
2855 1120 int ff_vp8_decode_frame(AVCodecContext *avctx, AVFrame *frame,
2856 int *got_frame, AVPacket *avpkt)
2857 {
2858 1120 return vp78_decode_frame(avctx, frame, got_frame, avpkt, IS_VP8);
2859 }
2860
2861 60 av_cold int ff_vp8_decode_init(AVCodecContext *avctx)
2862 {
2863 60 VP8Context *s = avctx->priv_data;
2864
2865 60 vp78_decode_init(avctx);
2866 60 ff_h264_pred_init(&s->hpc, AV_CODEC_ID_VP8, 8, 1);
2867 60 ff_vp8dsp_init(&s->vp8dsp);
2868 60 s->decode_mb_row_no_filter = vp8_decode_mb_row_no_filter;
2869 60 s->filter_mb_row = vp8_filter_mb_row;
2870
2871 60 return 0;
2872 }
2873
2874 #if HAVE_THREADS
2875 static void vp8_replace_frame(VP8Frame *dst, const VP8Frame *src)
2876 {
2877 ff_progress_frame_replace(&dst->tf, &src->tf);
2878 av_refstruct_replace(&dst->seg_map, src->seg_map);
2879 av_refstruct_replace(&dst->hwaccel_picture_private,
2880 src->hwaccel_picture_private);
2881 }
2882
2883 #define REBASE(pic) ((pic) ? (pic) - &s_src->frames[0] + &s->frames[0] : NULL)
2884
2885 static int vp8_decode_update_thread_context(AVCodecContext *dst,
2886 const AVCodecContext *src)
2887 {
2888 VP8Context *s = dst->priv_data, *s_src = src->priv_data;
2889
2890 if (s->macroblocks_base &&
2891 (s_src->mb_width != s->mb_width || s_src->mb_height != s->mb_height)) {
2892 free_buffers(s);
2893 s->mb_width = s_src->mb_width;
2894 s->mb_height = s_src->mb_height;
2895 }
2896
2897 s->pix_fmt = s_src->pix_fmt;
2898 s->prob[0] = s_src->prob[!s_src->update_probabilities];
2899 s->segmentation = s_src->segmentation;
2900 s->lf_delta = s_src->lf_delta;
2901 memcpy(s->sign_bias, s_src->sign_bias, sizeof(s->sign_bias));
2902
2903 for (int i = 0; i < FF_ARRAY_ELEMS(s_src->frames); i++)
2904 vp8_replace_frame(&s->frames[i], &s_src->frames[i]);
2905
2906 s->framep[0] = REBASE(s_src->next_framep[0]);
2907 s->framep[1] = REBASE(s_src->next_framep[1]);
2908 s->framep[2] = REBASE(s_src->next_framep[2]);
2909 s->framep[3] = REBASE(s_src->next_framep[3]);
2910
2911 return 0;
2912 }
2913 #endif /* HAVE_THREADS */
2914 #endif /* CONFIG_VP8_DECODER */
2915
2916 #if CONFIG_VP7_DECODER
2917 341 static int vp7_decode_mb_row_no_filter(AVCodecContext *avctx, void *tdata,
2918 int jobnr, int threadnr)
2919 {
2920 341 return decode_mb_row_no_filter(avctx, tdata, jobnr, threadnr, 1);
2921 }
2922
2923 341 static void vp7_filter_mb_row(AVCodecContext *avctx, void *tdata,
2924 int jobnr, int threadnr)
2925 {
2926 341 filter_mb_row(avctx, tdata, jobnr, threadnr, 1);
2927 341 }
2928
2929 31 static int vp7_decode_frame(AVCodecContext *avctx, AVFrame *frame,
2930 int *got_frame, AVPacket *avpkt)
2931 {
2932 31 return vp78_decode_frame(avctx, frame, got_frame, avpkt, IS_VP7);
2933 }
2934
2935 3 av_cold static int vp7_decode_init(AVCodecContext *avctx)
2936 {
2937 3 VP8Context *s = avctx->priv_data;
2938
2939 3 vp78_decode_init(avctx);
2940 3 ff_h264_pred_init(&s->hpc, AV_CODEC_ID_VP7, 8, 1);
2941 3 ff_vp7dsp_init(&s->vp8dsp);
2942 3 s->decode_mb_row_no_filter = vp7_decode_mb_row_no_filter;
2943 3 s->filter_mb_row = vp7_filter_mb_row;
2944
2945 3 return 0;
2946 }
2947
2948 const FFCodec ff_vp7_decoder = {
2949 .p.name = "vp7",
2950 CODEC_LONG_NAME("On2 VP7"),
2951 .p.type = AVMEDIA_TYPE_VIDEO,
2952 .p.id = AV_CODEC_ID_VP7,
2953 .priv_data_size = sizeof(VP8Context),
2954 .init = vp7_decode_init,
2955 .close = ff_vp8_decode_free,
2956 FF_CODEC_DECODE_CB(vp7_decode_frame),
2957 .p.capabilities = AV_CODEC_CAP_DR1,
2958 .flush = vp8_decode_flush,
2959 .caps_internal = FF_CODEC_CAP_USES_PROGRESSFRAMES,
2960 };
2961 #endif /* CONFIG_VP7_DECODER */
2962
2963 #if CONFIG_VP8_DECODER
2964 const FFCodec ff_vp8_decoder = {
2965 .p.name = "vp8",
2966 CODEC_LONG_NAME("On2 VP8"),
2967 .p.type = AVMEDIA_TYPE_VIDEO,
2968 .p.id = AV_CODEC_ID_VP8,
2969 .priv_data_size = sizeof(VP8Context),
2970 .init = ff_vp8_decode_init,
2971 .close = ff_vp8_decode_free,
2972 FF_CODEC_DECODE_CB(ff_vp8_decode_frame),
2973 .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS |
2974 AV_CODEC_CAP_SLICE_THREADS,
2975 .caps_internal = FF_CODEC_CAP_USES_PROGRESSFRAMES,
2976 .flush = vp8_decode_flush,
2977 UPDATE_THREAD_CONTEXT(vp8_decode_update_thread_context),
2978 .hw_configs = (const AVCodecHWConfigInternal *const []) {
2979 #if CONFIG_VP8_VAAPI_HWACCEL
2980 HWACCEL_VAAPI(vp8),
2981 #endif
2982 #if CONFIG_VP8_NVDEC_HWACCEL
2983 HWACCEL_NVDEC(vp8),
2984 #endif
2985 NULL
2986 },
2987 };
2988 #endif /* CONFIG_VP8_DECODER */
2989