FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/mpeg12dec.c
Date: 2022-07-04 19:11:22
Exec Total Coverage
Lines: 1159 1742 66.5%
Branches: 641 1137 56.4%

Line Branch Exec Source
1 /*
2 * MPEG-1/2 decoder
3 * Copyright (c) 2000, 2001 Fabrice Bellard
4 * Copyright (c) 2002-2013 Michael Niedermayer <michaelni@gmx.at>
5 *
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23 /**
24 * @file
25 * MPEG-1/2 decoder
26 */
27
28 #include "config_components.h"
29
30 #define UNCHECKED_BITSTREAM_READER 1
31 #include <inttypes.h>
32
33 #include "libavutil/attributes.h"
34 #include "libavutil/imgutils.h"
35 #include "libavutil/internal.h"
36 #include "libavutil/mem_internal.h"
37 #include "libavutil/reverse.h"
38 #include "libavutil/stereo3d.h"
39 #include "libavutil/timecode.h"
40
41 #include "avcodec.h"
42 #include "codec_internal.h"
43 #include "error_resilience.h"
44 #include "hwconfig.h"
45 #include "idctdsp.h"
46 #include "internal.h"
47 #include "mpeg_er.h"
48 #include "mpeg12.h"
49 #include "mpeg12data.h"
50 #include "mpeg12dec.h"
51 #include "mpegutils.h"
52 #include "mpegvideo.h"
53 #include "mpegvideodata.h"
54 #include "mpegvideodec.h"
55 #include "profiles.h"
56 #include "startcode.h"
57 #include "thread.h"
58
59 #define A53_MAX_CC_COUNT 2000
60
61 typedef struct Mpeg1Context {
62 MpegEncContext mpeg_enc_ctx;
63 int mpeg_enc_ctx_allocated; /* true if decoding context allocated */
64 int repeat_field; /* true if we must repeat the field */
65 AVPanScan pan_scan; /* some temporary storage for the panscan */
66 AVStereo3D stereo3d;
67 int has_stereo3d;
68 AVBufferRef *a53_buf_ref;
69 uint8_t afd;
70 int has_afd;
71 int slice_count;
72 unsigned aspect_ratio_info;
73 AVRational save_aspect;
74 int save_width, save_height, save_progressive_seq;
75 int rc_buffer_size;
76 AVRational frame_rate_ext; /* MPEG-2 specific framerate modificator */
77 unsigned frame_rate_index;
78 int sync; /* Did we reach a sync point like a GOP/SEQ/KEYFrame? */
79 int closed_gop;
80 int tmpgexs;
81 int first_slice;
82 int extradata_decoded;
83 int64_t timecode_frame_start; /*< GOP timecode frame start number, in non drop frame format */
84 } Mpeg1Context;
85
86 #define MB_TYPE_ZERO_MV 0x20000000
87
88 static const uint32_t ptype2mb_type[7] = {
89 MB_TYPE_INTRA,
90 MB_TYPE_L0 | MB_TYPE_CBP | MB_TYPE_ZERO_MV | MB_TYPE_16x16,
91 MB_TYPE_L0,
92 MB_TYPE_L0 | MB_TYPE_CBP,
93 MB_TYPE_QUANT | MB_TYPE_INTRA,
94 MB_TYPE_QUANT | MB_TYPE_L0 | MB_TYPE_CBP | MB_TYPE_ZERO_MV | MB_TYPE_16x16,
95 MB_TYPE_QUANT | MB_TYPE_L0 | MB_TYPE_CBP,
96 };
97
98 static const uint32_t btype2mb_type[11] = {
99 MB_TYPE_INTRA,
100 MB_TYPE_L1,
101 MB_TYPE_L1 | MB_TYPE_CBP,
102 MB_TYPE_L0,
103 MB_TYPE_L0 | MB_TYPE_CBP,
104 MB_TYPE_L0L1,
105 MB_TYPE_L0L1 | MB_TYPE_CBP,
106 MB_TYPE_QUANT | MB_TYPE_INTRA,
107 MB_TYPE_QUANT | MB_TYPE_L1 | MB_TYPE_CBP,
108 MB_TYPE_QUANT | MB_TYPE_L0 | MB_TYPE_CBP,
109 MB_TYPE_QUANT | MB_TYPE_L0L1 | MB_TYPE_CBP,
110 };
111
112 /* as H.263, but only 17 codes */
113 4268676 static int mpeg_decode_motion(MpegEncContext *s, int fcode, int pred)
114 {
115 int code, sign, val, shift;
116
117 4268676 code = get_vlc2(&s->gb, ff_mv_vlc.table, MV_VLC_BITS, 2);
118
2/2
✓ Branch 0 taken 1662399 times.
✓ Branch 1 taken 2606277 times.
4268676 if (code == 0)
119 1662399 return pred;
120
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2606277 times.
2606277 if (code < 0)
121 return 0xffff;
122
123 2606277 sign = get_bits1(&s->gb);
124 2606277 shift = fcode - 1;
125 2606277 val = code;
126
2/2
✓ Branch 0 taken 2266640 times.
✓ Branch 1 taken 339637 times.
2606277 if (shift) {
127 2266640 val = (val - 1) << shift;
128 2266640 val |= get_bits(&s->gb, shift);
129 2266640 val++;
130 }
131
2/2
✓ Branch 0 taken 1343990 times.
✓ Branch 1 taken 1262287 times.
2606277 if (sign)
132 1343990 val = -val;
133 2606277 val += pred;
134
135 /* modulo decoding */
136 2606277 return sign_extend(val, 5 + shift);
137 }
138
139 #define MAX_INDEX (64 - 1)
140 #define check_scantable_index(ctx, x) \
141 do { \
142 if ((x) > MAX_INDEX) { \
143 av_log(ctx->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", \
144 ctx->mb_x, ctx->mb_y); \
145 return AVERROR_INVALIDDATA; \
146 } \
147 } while (0)
148
149 403118 static inline int mpeg1_decode_block_inter(MpegEncContext *s,
150 int16_t *block, int n)
151 {
152 int level, i, j, run;
153 403118 RLTable *rl = &ff_rl_mpeg1;
154 403118 uint8_t *const scantable = s->intra_scantable.permutated;
155 403118 const uint16_t *quant_matrix = s->inter_matrix;
156 403118 const int qscale = s->qscale;
157
158 {
159 403118 OPEN_READER(re, &s->gb);
160 403118 i = -1;
161 // special case for first coefficient, no need to add second VLC table
162 403118 UPDATE_CACHE(re, &s->gb);
163
2/2
✓ Branch 0 taken 101724 times.
✓ Branch 1 taken 301394 times.
403118 if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
164 101724 level = (3 * qscale * quant_matrix[0]) >> 5;
165 101724 level = (level - 1) | 1;
166
2/2
✓ Branch 0 taken 56898 times.
✓ Branch 1 taken 44826 times.
101724 if (GET_CACHE(re, &s->gb) & 0x40000000)
167 56898 level = -level;
168 101724 block[0] = level;
169 101724 i++;
170 101724 SKIP_BITS(re, &s->gb, 2);
171
2/2
✓ Branch 0 taken 34285 times.
✓ Branch 1 taken 67439 times.
101724 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
172 34285 goto end;
173 }
174 /* now quantify & encode AC coefficients */
175 for (;;) {
176
2/2
✓ Branch 1 taken 336221 times.
✓ Branch 2 taken 2585187 times.
2921408 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
177 TEX_VLC_BITS, 2, 0);
178
179
2/2
✓ Branch 0 taken 2842322 times.
✓ Branch 1 taken 79086 times.
2921408 if (level != 0) {
180 2842322 i += run;
181
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2842322 times.
2842322 if (i > MAX_INDEX)
182 break;
183 2842322 j = scantable[i];
184 2842322 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
185 2842322 level = (level - 1) | 1;
186 2842322 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
187 2842322 SHOW_SBITS(re, &s->gb, 1);
188 2842322 SKIP_BITS(re, &s->gb, 1);
189 } else {
190 /* escape */
191 79086 run = SHOW_UBITS(re, &s->gb, 6) + 1;
192 79086 LAST_SKIP_BITS(re, &s->gb, 6);
193 79086 UPDATE_CACHE(re, &s->gb);
194 79086 level = SHOW_SBITS(re, &s->gb, 8);
195 79086 SKIP_BITS(re, &s->gb, 8);
196
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 79086 times.
79086 if (level == -128) {
197 level = SHOW_UBITS(re, &s->gb, 8) - 256;
198 SKIP_BITS(re, &s->gb, 8);
199
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 79086 times.
79086 } else if (level == 0) {
200 level = SHOW_UBITS(re, &s->gb, 8);
201 SKIP_BITS(re, &s->gb, 8);
202 }
203 79086 i += run;
204
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 79086 times.
79086 if (i > MAX_INDEX)
205 break;
206 79086 j = scantable[i];
207
2/2
✓ Branch 0 taken 38919 times.
✓ Branch 1 taken 40167 times.
79086 if (level < 0) {
208 38919 level = -level;
209 38919 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
210 38919 level = (level - 1) | 1;
211 38919 level = -level;
212 } else {
213 40167 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
214 40167 level = (level - 1) | 1;
215 }
216 }
217
218 2921408 block[j] = level;
219
2/2
✓ Branch 0 taken 368833 times.
✓ Branch 1 taken 2552575 times.
2921408 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
220 368833 break;
221 2552575 UPDATE_CACHE(re, &s->gb);
222 }
223 403118 end:
224 403118 LAST_SKIP_BITS(re, &s->gb, 2);
225 403118 CLOSE_READER(re, &s->gb);
226 }
227
228
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 403118 times.
403118 check_scantable_index(s, i);
229
230 403118 s->block_last_index[n] = i;
231 403118 return 0;
232 }
233
234 /**
235 * Changing this would eat up any speed benefits it has.
236 * Do not use "fast" flag if you need the code to be robust.
237 */
238 static inline int mpeg1_fast_decode_block_inter(MpegEncContext *s,
239 int16_t *block, int n)
240 {
241 int level, i, j, run;
242 RLTable *rl = &ff_rl_mpeg1;
243 uint8_t *const scantable = s->intra_scantable.permutated;
244 const int qscale = s->qscale;
245
246 {
247 OPEN_READER(re, &s->gb);
248 i = -1;
249 // Special case for first coefficient, no need to add second VLC table.
250 UPDATE_CACHE(re, &s->gb);
251 if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
252 level = (3 * qscale) >> 1;
253 level = (level - 1) | 1;
254 if (GET_CACHE(re, &s->gb) & 0x40000000)
255 level = -level;
256 block[0] = level;
257 i++;
258 SKIP_BITS(re, &s->gb, 2);
259 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
260 goto end;
261 }
262
263 /* now quantify & encode AC coefficients */
264 for (;;) {
265 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
266 TEX_VLC_BITS, 2, 0);
267
268 if (level != 0) {
269 i += run;
270 if (i > MAX_INDEX)
271 break;
272 j = scantable[i];
273 level = ((level * 2 + 1) * qscale) >> 1;
274 level = (level - 1) | 1;
275 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
276 SHOW_SBITS(re, &s->gb, 1);
277 SKIP_BITS(re, &s->gb, 1);
278 } else {
279 /* escape */
280 run = SHOW_UBITS(re, &s->gb, 6) + 1;
281 LAST_SKIP_BITS(re, &s->gb, 6);
282 UPDATE_CACHE(re, &s->gb);
283 level = SHOW_SBITS(re, &s->gb, 8);
284 SKIP_BITS(re, &s->gb, 8);
285 if (level == -128) {
286 level = SHOW_UBITS(re, &s->gb, 8) - 256;
287 SKIP_BITS(re, &s->gb, 8);
288 } else if (level == 0) {
289 level = SHOW_UBITS(re, &s->gb, 8);
290 SKIP_BITS(re, &s->gb, 8);
291 }
292 i += run;
293 if (i > MAX_INDEX)
294 break;
295 j = scantable[i];
296 if (level < 0) {
297 level = -level;
298 level = ((level * 2 + 1) * qscale) >> 1;
299 level = (level - 1) | 1;
300 level = -level;
301 } else {
302 level = ((level * 2 + 1) * qscale) >> 1;
303 level = (level - 1) | 1;
304 }
305 }
306
307 block[j] = level;
308 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
309 break;
310 UPDATE_CACHE(re, &s->gb);
311 }
312 end:
313 LAST_SKIP_BITS(re, &s->gb, 2);
314 CLOSE_READER(re, &s->gb);
315 }
316
317 check_scantable_index(s, i);
318
319 s->block_last_index[n] = i;
320 return 0;
321 }
322
323 3924107 static inline int mpeg2_decode_block_non_intra(MpegEncContext *s,
324 int16_t *block, int n)
325 {
326 int level, i, j, run;
327 3924107 RLTable *rl = &ff_rl_mpeg1;
328 3924107 uint8_t *const scantable = s->intra_scantable.permutated;
329 const uint16_t *quant_matrix;
330 3924107 const int qscale = s->qscale;
331 int mismatch;
332
333 3924107 mismatch = 1;
334
335 {
336 3924107 OPEN_READER(re, &s->gb);
337 3924107 i = -1;
338
2/2
✓ Branch 0 taken 2871886 times.
✓ Branch 1 taken 1052221 times.
3924107 if (n < 4)
339 2871886 quant_matrix = s->inter_matrix;
340 else
341 1052221 quant_matrix = s->chroma_inter_matrix;
342
343 // Special case for first coefficient, no need to add second VLC table.
344 3924107 UPDATE_CACHE(re, &s->gb);
345
2/2
✓ Branch 0 taken 1415133 times.
✓ Branch 1 taken 2508974 times.
3924107 if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
346 1415133 level = (3 * qscale * quant_matrix[0]) >> 5;
347
2/2
✓ Branch 0 taken 885057 times.
✓ Branch 1 taken 530076 times.
1415133 if (GET_CACHE(re, &s->gb) & 0x40000000)
348 885057 level = -level;
349 1415133 block[0] = level;
350 1415133 mismatch ^= level;
351 1415133 i++;
352 1415133 SKIP_BITS(re, &s->gb, 2);
353
2/2
✓ Branch 0 taken 673112 times.
✓ Branch 1 taken 742021 times.
1415133 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
354 673112 goto end;
355 }
356
357 /* now quantify & encode AC coefficients */
358 for (;;) {
359
2/2
✓ Branch 1 taken 1407741 times.
✓ Branch 2 taken 13970288 times.
15378029 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
360 TEX_VLC_BITS, 2, 0);
361
362
2/2
✓ Branch 0 taken 15126107 times.
✓ Branch 1 taken 251922 times.
15378029 if (level != 0) {
363 15126107 i += run;
364
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15126107 times.
15126107 if (i > MAX_INDEX)
365 break;
366 15126107 j = scantable[i];
367 15126107 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
368 15126107 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
369 15126107 SHOW_SBITS(re, &s->gb, 1);
370 15126107 SKIP_BITS(re, &s->gb, 1);
371 } else {
372 /* escape */
373 251922 run = SHOW_UBITS(re, &s->gb, 6) + 1;
374 251922 LAST_SKIP_BITS(re, &s->gb, 6);
375 251922 UPDATE_CACHE(re, &s->gb);
376 251922 level = SHOW_SBITS(re, &s->gb, 12);
377 251922 SKIP_BITS(re, &s->gb, 12);
378
379 251922 i += run;
380
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 251922 times.
251922 if (i > MAX_INDEX)
381 break;
382 251922 j = scantable[i];
383
2/2
✓ Branch 0 taken 123054 times.
✓ Branch 1 taken 128868 times.
251922 if (level < 0) {
384 123054 level = ((-level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
385 123054 level = -level;
386 } else {
387 128868 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
388 }
389 }
390
391 15378029 mismatch ^= level;
392 15378029 block[j] = level;
393
2/2
✓ Branch 0 taken 3250995 times.
✓ Branch 1 taken 12127034 times.
15378029 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
394 3250995 break;
395 12127034 UPDATE_CACHE(re, &s->gb);
396 }
397 3924107 end:
398 3924107 LAST_SKIP_BITS(re, &s->gb, 2);
399 3924107 CLOSE_READER(re, &s->gb);
400 }
401 3924107 block[63] ^= (mismatch & 1);
402
403
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3924107 times.
3924107 check_scantable_index(s, i);
404
405 3924107 s->block_last_index[n] = i;
406 3924107 return 0;
407 }
408
409 /**
410 * Changing this would eat up any speed benefits it has.
411 * Do not use "fast" flag if you need the code to be robust.
412 */
413 static inline int mpeg2_fast_decode_block_non_intra(MpegEncContext *s,
414 int16_t *block, int n)
415 {
416 int level, i, j, run;
417 RLTable *rl = &ff_rl_mpeg1;
418 uint8_t *const scantable = s->intra_scantable.permutated;
419 const int qscale = s->qscale;
420 OPEN_READER(re, &s->gb);
421 i = -1;
422
423 // special case for first coefficient, no need to add second VLC table
424 UPDATE_CACHE(re, &s->gb);
425 if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
426 level = (3 * qscale) >> 1;
427 if (GET_CACHE(re, &s->gb) & 0x40000000)
428 level = -level;
429 block[0] = level;
430 i++;
431 SKIP_BITS(re, &s->gb, 2);
432 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
433 goto end;
434 }
435
436 /* now quantify & encode AC coefficients */
437 for (;;) {
438 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2, 0);
439
440 if (level != 0) {
441 i += run;
442 if (i > MAX_INDEX)
443 break;
444 j = scantable[i];
445 level = ((level * 2 + 1) * qscale) >> 1;
446 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
447 SHOW_SBITS(re, &s->gb, 1);
448 SKIP_BITS(re, &s->gb, 1);
449 } else {
450 /* escape */
451 run = SHOW_UBITS(re, &s->gb, 6) + 1;
452 LAST_SKIP_BITS(re, &s->gb, 6);
453 UPDATE_CACHE(re, &s->gb);
454 level = SHOW_SBITS(re, &s->gb, 12);
455 SKIP_BITS(re, &s->gb, 12);
456
457 i += run;
458 if (i > MAX_INDEX)
459 break;
460 j = scantable[i];
461 if (level < 0) {
462 level = ((-level * 2 + 1) * qscale) >> 1;
463 level = -level;
464 } else {
465 level = ((level * 2 + 1) * qscale) >> 1;
466 }
467 }
468
469 block[j] = level;
470 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF || i > 63)
471 break;
472
473 UPDATE_CACHE(re, &s->gb);
474 }
475 end:
476 LAST_SKIP_BITS(re, &s->gb, 2);
477 CLOSE_READER(re, &s->gb);
478
479 check_scantable_index(s, i);
480
481 s->block_last_index[n] = i;
482 return 0;
483 }
484
485 1654206 static inline int mpeg2_decode_block_intra(MpegEncContext *s,
486 int16_t *block, int n)
487 {
488 int level, dc, diff, i, j, run;
489 int component;
490 RLTable *rl;
491 1654206 uint8_t *const scantable = s->intra_scantable.permutated;
492 const uint16_t *quant_matrix;
493 1654206 const int qscale = s->qscale;
494 int mismatch;
495
496 /* DC coefficient */
497
2/2
✓ Branch 0 taken 1031060 times.
✓ Branch 1 taken 623146 times.
1654206 if (n < 4) {
498 1031060 quant_matrix = s->intra_matrix;
499 1031060 component = 0;
500 } else {
501 623146 quant_matrix = s->chroma_intra_matrix;
502 623146 component = (n & 1) + 1;
503 }
504 1654206 diff = decode_dc(&s->gb, component);
505 1654206 dc = s->last_dc[component];
506 1654206 dc += diff;
507 1654206 s->last_dc[component] = dc;
508 1654206 block[0] = dc * (1 << (3 - s->intra_dc_precision));
509 ff_tlog(s->avctx, "dc=%d\n", block[0]);
510 1654206 mismatch = block[0] ^ 1;
511 1654206 i = 0;
512
2/2
✓ Branch 0 taken 1094802 times.
✓ Branch 1 taken 559404 times.
1654206 if (s->intra_vlc_format)
513 1094802 rl = &ff_rl_mpeg2;
514 else
515 559404 rl = &ff_rl_mpeg1;
516
517 {
518 1654206 OPEN_READER(re, &s->gb);
519 /* now quantify & encode AC coefficients */
520 for (;;) {
521 12867568 UPDATE_CACHE(re, &s->gb);
522
2/2
✓ Branch 1 taken 686026 times.
✓ Branch 2 taken 12181542 times.
12867568 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
523 TEX_VLC_BITS, 2, 0);
524
525
2/2
✓ Branch 0 taken 1654202 times.
✓ Branch 1 taken 11213366 times.
12867568 if (level == 127) {
526 1654202 break;
527
2/2
✓ Branch 0 taken 11091082 times.
✓ Branch 1 taken 122284 times.
11213366 } else if (level != 0) {
528 11091082 i += run;
529
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 11091078 times.
11091082 if (i > MAX_INDEX)
530 4 break;
531 11091078 j = scantable[i];
532 11091078 level = (level * qscale * quant_matrix[j]) >> 4;
533 11091078 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
534 11091078 SHOW_SBITS(re, &s->gb, 1);
535 11091078 LAST_SKIP_BITS(re, &s->gb, 1);
536 } else {
537 /* escape */
538 122284 run = SHOW_UBITS(re, &s->gb, 6) + 1;
539 122284 SKIP_BITS(re, &s->gb, 6);
540 122284 level = SHOW_SBITS(re, &s->gb, 12);
541 122284 LAST_SKIP_BITS(re, &s->gb, 12);
542 122284 i += run;
543
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 122284 times.
122284 if (i > MAX_INDEX)
544 break;
545 122284 j = scantable[i];
546
2/2
✓ Branch 0 taken 53053 times.
✓ Branch 1 taken 69231 times.
122284 if (level < 0) {
547 53053 level = (-level * qscale * quant_matrix[j]) >> 4;
548 53053 level = -level;
549 } else {
550 69231 level = (level * qscale * quant_matrix[j]) >> 4;
551 }
552 }
553
554 11213362 mismatch ^= level;
555 11213362 block[j] = level;
556 }
557 1654206 CLOSE_READER(re, &s->gb);
558 }
559 1654206 block[63] ^= mismatch & 1;
560
561
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 1654202 times.
1654206 check_scantable_index(s, i);
562
563 1654202 s->block_last_index[n] = i;
564 1654202 return 0;
565 }
566
567 /**
568 * Changing this would eat up any speed benefits it has.
569 * Do not use "fast" flag if you need the code to be robust.
570 */
571 static inline int mpeg2_fast_decode_block_intra(MpegEncContext *s,
572 int16_t *block, int n)
573 {
574 int level, dc, diff, i, j, run;
575 int component;
576 RLTable *rl;
577 uint8_t *const scantable = s->intra_scantable.permutated;
578 const uint16_t *quant_matrix;
579 const int qscale = s->qscale;
580
581 /* DC coefficient */
582 if (n < 4) {
583 quant_matrix = s->intra_matrix;
584 component = 0;
585 } else {
586 quant_matrix = s->chroma_intra_matrix;
587 component = (n & 1) + 1;
588 }
589 diff = decode_dc(&s->gb, component);
590 dc = s->last_dc[component];
591 dc += diff;
592 s->last_dc[component] = dc;
593 block[0] = dc * (1 << (3 - s->intra_dc_precision));
594 i = 0;
595 if (s->intra_vlc_format)
596 rl = &ff_rl_mpeg2;
597 else
598 rl = &ff_rl_mpeg1;
599
600 {
601 OPEN_READER(re, &s->gb);
602 /* now quantify & encode AC coefficients */
603 for (;;) {
604 UPDATE_CACHE(re, &s->gb);
605 GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
606 TEX_VLC_BITS, 2, 0);
607
608 if (level >= 64 || i > 63) {
609 break;
610 } else if (level != 0) {
611 i += run;
612 j = scantable[i];
613 level = (level * qscale * quant_matrix[j]) >> 4;
614 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
615 SHOW_SBITS(re, &s->gb, 1);
616 LAST_SKIP_BITS(re, &s->gb, 1);
617 } else {
618 /* escape */
619 run = SHOW_UBITS(re, &s->gb, 6) + 1;
620 SKIP_BITS(re, &s->gb, 6);
621 level = SHOW_SBITS(re, &s->gb, 12);
622 LAST_SKIP_BITS(re, &s->gb, 12);
623 i += run;
624 j = scantable[i];
625 if (level < 0) {
626 level = (-level * qscale * quant_matrix[j]) >> 4;
627 level = -level;
628 } else {
629 level = (level * qscale * quant_matrix[j]) >> 4;
630 }
631 }
632
633 block[j] = level;
634 }
635 CLOSE_READER(re, &s->gb);
636 }
637
638 check_scantable_index(s, i);
639
640 s->block_last_index[n] = i;
641 return 0;
642 }
643
644 /******************************************/
645 /* decoding */
646
647 22848 static inline int get_dmv(MpegEncContext *s)
648 {
649
2/2
✓ Branch 1 taken 21321 times.
✓ Branch 2 taken 1527 times.
22848 if (get_bits1(&s->gb))
650 21321 return 1 - (get_bits1(&s->gb) << 1);
651 else
652 1527 return 0;
653 }
654
655 /* motion type (for MPEG-2) */
656 #define MT_FIELD 1
657 #define MT_FRAME 2
658 #define MT_16X8 2
659 #define MT_DMV 3
660
661 2124178 static int mpeg_decode_mb(MpegEncContext *s, int16_t block[12][64])
662 {
663 int i, j, k, cbp, val, mb_type, motion_type;
664 2124178 const int mb_block_count = 4 + (1 << s->chroma_format);
665 int ret;
666
667 ff_tlog(s->avctx, "decode_mb: x=%d y=%d\n", s->mb_x, s->mb_y);
668
669 av_assert2(s->mb_skipped == 0);
670
671
2/2
✓ Branch 0 taken 156167 times.
✓ Branch 1 taken 1968011 times.
2124178 if (s->mb_skip_run-- != 0) {
672
2/2
✓ Branch 0 taken 51347 times.
✓ Branch 1 taken 104820 times.
156167 if (s->pict_type == AV_PICTURE_TYPE_P) {
673 51347 s->mb_skipped = 1;
674 51347 s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride] =
675 MB_TYPE_SKIP | MB_TYPE_L0 | MB_TYPE_16x16;
676 } else {
677 int mb_type;
678
679
1/2
✓ Branch 0 taken 104820 times.
✗ Branch 1 not taken.
104820 if (s->mb_x)
680 104820 mb_type = s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride - 1];
681 else
682 // FIXME not sure if this is allowed in MPEG at all
683 mb_type = s->current_picture.mb_type[s->mb_width + (s->mb_y - 1) * s->mb_stride - 1];
684
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 104820 times.
104820 if (IS_INTRA(mb_type)) {
685 av_log(s->avctx, AV_LOG_ERROR, "skip with previntra\n");
686 return AVERROR_INVALIDDATA;
687 }
688 104820 s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride] =
689 104820 mb_type | MB_TYPE_SKIP;
690
691
2/2
✓ Branch 0 taken 74214 times.
✓ Branch 1 taken 30606 times.
104820 if ((s->mv[0][0][0] | s->mv[0][0][1] | s->mv[1][0][0] | s->mv[1][0][1]) == 0)
692 74214 s->mb_skipped = 1;
693 }
694
695 156167 return 0;
696 }
697
698
3/3
✓ Branch 0 taken 210510 times.
✓ Branch 1 taken 1018163 times.
✓ Branch 2 taken 739338 times.
1968011 switch (s->pict_type) {
699 210510 default:
700 case AV_PICTURE_TYPE_I:
701
2/2
✓ Branch 1 taken 14380 times.
✓ Branch 2 taken 196130 times.
210510 if (get_bits1(&s->gb) == 0) {
702
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 14380 times.
14380 if (get_bits1(&s->gb) == 0) {
703 av_log(s->avctx, AV_LOG_ERROR,
704 "Invalid mb type in I-frame at %d %d\n",
705 s->mb_x, s->mb_y);
706 return AVERROR_INVALIDDATA;
707 }
708 14380 mb_type = MB_TYPE_QUANT | MB_TYPE_INTRA;
709 } else {
710 196130 mb_type = MB_TYPE_INTRA;
711 }
712 210510 break;
713 1018163 case AV_PICTURE_TYPE_P:
714 1018163 mb_type = get_vlc2(&s->gb, ff_mb_ptype_vlc.table, MB_PTYPE_VLC_BITS, 1);
715
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1018163 times.
1018163 if (mb_type < 0) {
716 av_log(s->avctx, AV_LOG_ERROR,
717 "Invalid mb type in P-frame at %d %d\n", s->mb_x, s->mb_y);
718 return AVERROR_INVALIDDATA;
719 }
720 1018163 mb_type = ptype2mb_type[mb_type];
721 1018163 break;
722 739338 case AV_PICTURE_TYPE_B:
723 739338 mb_type = get_vlc2(&s->gb, ff_mb_btype_vlc.table, MB_BTYPE_VLC_BITS, 1);
724
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 739338 times.
739338 if (mb_type < 0) {
725 av_log(s->avctx, AV_LOG_ERROR,
726 "Invalid mb type in B-frame at %d %d\n", s->mb_x, s->mb_y);
727 return AVERROR_INVALIDDATA;
728 }
729 739338 mb_type = btype2mb_type[mb_type];
730 739338 break;
731 }
732 ff_tlog(s->avctx, "mb_type=%x\n", mb_type);
733 // motion_type = 0; /* avoid warning */
734
2/2
✓ Branch 0 taken 278788 times.
✓ Branch 1 taken 1689223 times.
1968011 if (IS_INTRA(mb_type)) {
735 278788 s->bdsp.clear_blocks(s->block[0]);
736
737
2/2
✓ Branch 0 taken 53810 times.
✓ Branch 1 taken 224978 times.
278788 if (!s->chroma_y_shift)
738 53810 s->bdsp.clear_blocks(s->block[6]);
739
740 /* compute DCT type */
741 // FIXME: add an interlaced_dct coded var?
742
2/2
✓ Branch 0 taken 243197 times.
✓ Branch 1 taken 35591 times.
278788 if (s->picture_structure == PICT_FRAME &&
743
2/2
✓ Branch 0 taken 104089 times.
✓ Branch 1 taken 139108 times.
243197 !s->frame_pred_frame_dct)
744 104089 s->interlaced_dct = get_bits1(&s->gb);
745
746
2/2
✓ Branch 0 taken 19501 times.
✓ Branch 1 taken 259287 times.
278788 if (IS_QUANT(mb_type))
747 19501 s->qscale = mpeg_get_qscale(s);
748
749
2/2
✓ Branch 0 taken 678 times.
✓ Branch 1 taken 278110 times.
278788 if (s->concealment_motion_vectors) {
750 /* just parse them */
751
2/2
✓ Branch 0 taken 675 times.
✓ Branch 1 taken 3 times.
678 if (s->picture_structure != PICT_FRAME)
752 675 skip_bits1(&s->gb); /* field select */
753
754 678 s->mv[0][0][0] =
755 678 s->last_mv[0][0][0] =
756 678 s->last_mv[0][1][0] = mpeg_decode_motion(s, s->mpeg_f_code[0][0],
757 s->last_mv[0][0][0]);
758 678 s->mv[0][0][1] =
759 678 s->last_mv[0][0][1] =
760 678 s->last_mv[0][1][1] = mpeg_decode_motion(s, s->mpeg_f_code[0][1],
761 s->last_mv[0][0][1]);
762
763 678 check_marker(s->avctx, &s->gb, "after concealment_motion_vectors");
764 } else {
765 /* reset mv prediction */
766 278110 memset(s->last_mv, 0, sizeof(s->last_mv));
767 }
768 278788 s->mb_intra = 1;
769
770
2/2
✓ Branch 0 taken 257767 times.
✓ Branch 1 taken 21021 times.
278788 if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
771
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 257767 times.
257767 if (s->avctx->flags2 & AV_CODEC_FLAG2_FAST) {
772 for (i = 0; i < 6; i++)
773 mpeg2_fast_decode_block_intra(s, *s->pblocks[i], i);
774 } else {
775
2/2
✓ Branch 0 taken 1654206 times.
✓ Branch 1 taken 257763 times.
1911969 for (i = 0; i < mb_block_count; i++)
776
2/2
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 1654202 times.
1654206 if ((ret = mpeg2_decode_block_intra(s, *s->pblocks[i], i)) < 0)
777 4 return ret;
778 }
779 } else {
780
2/2
✓ Branch 0 taken 126126 times.
✓ Branch 1 taken 21021 times.
147147 for (i = 0; i < 6; i++) {
781 126126 ret = ff_mpeg1_decode_block_intra(&s->gb,
782 126126 s->intra_matrix,
783 126126 s->intra_scantable.permutated,
784 126126 s->last_dc, *s->pblocks[i],
785 i, s->qscale);
786
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 126126 times.
126126 if (ret < 0) {
787 av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n",
788 s->mb_x, s->mb_y);
789 return ret;
790 }
791
792 126126 s->block_last_index[i] = ret;
793 }
794 }
795 } else {
796
2/2
✓ Branch 0 taken 148485 times.
✓ Branch 1 taken 1540738 times.
1689223 if (mb_type & MB_TYPE_ZERO_MV) {
797 av_assert2(mb_type & MB_TYPE_CBP);
798
799 148485 s->mv_dir = MV_DIR_FORWARD;
800
2/2
✓ Branch 0 taken 75956 times.
✓ Branch 1 taken 72529 times.
148485 if (s->picture_structure == PICT_FRAME) {
801
1/2
✓ Branch 0 taken 75956 times.
✗ Branch 1 not taken.
75956 if (s->picture_structure == PICT_FRAME
802
2/2
✓ Branch 0 taken 46050 times.
✓ Branch 1 taken 29906 times.
75956 && !s->frame_pred_frame_dct)
803 46050 s->interlaced_dct = get_bits1(&s->gb);
804 75956 s->mv_type = MV_TYPE_16X16;
805 } else {
806 72529 s->mv_type = MV_TYPE_FIELD;
807 72529 mb_type |= MB_TYPE_INTERLACED;
808 72529 s->field_select[0][0] = s->picture_structure - 1;
809 }
810
811
2/2
✓ Branch 0 taken 31326 times.
✓ Branch 1 taken 117159 times.
148485 if (IS_QUANT(mb_type))
812 31326 s->qscale = mpeg_get_qscale(s);
813
814 148485 s->last_mv[0][0][0] = 0;
815 148485 s->last_mv[0][0][1] = 0;
816 148485 s->last_mv[0][1][0] = 0;
817 148485 s->last_mv[0][1][1] = 0;
818 148485 s->mv[0][0][0] = 0;
819 148485 s->mv[0][0][1] = 0;
820 } else {
821 av_assert2(mb_type & MB_TYPE_L0L1);
822 // FIXME decide if MBs in field pictures are MB_TYPE_INTERLACED
823 /* get additional motion vector type */
824
4/4
✓ Branch 0 taken 1232826 times.
✓ Branch 1 taken 307912 times.
✓ Branch 2 taken 661748 times.
✓ Branch 3 taken 571078 times.
1540738 if (s->picture_structure == PICT_FRAME && s->frame_pred_frame_dct) {
825 661748 motion_type = MT_FRAME;
826 } else {
827 878990 motion_type = get_bits(&s->gb, 2);
828
4/4
✓ Branch 0 taken 571078 times.
✓ Branch 1 taken 307912 times.
✓ Branch 2 taken 414187 times.
✓ Branch 3 taken 156891 times.
878990 if (s->picture_structure == PICT_FRAME && HAS_CBP(mb_type))
829 414187 s->interlaced_dct = get_bits1(&s->gb);
830 }
831
832
2/2
✓ Branch 0 taken 112590 times.
✓ Branch 1 taken 1428148 times.
1540738 if (IS_QUANT(mb_type))
833 112590 s->qscale = mpeg_get_qscale(s);
834
835 /* motion vectors */
836 1540738 s->mv_dir = (mb_type >> 13) & 3;
837 ff_tlog(s->avctx, "motion_type=%d\n", motion_type);
838
3/4
✓ Branch 0 taken 1316564 times.
✓ Branch 1 taken 212750 times.
✓ Branch 2 taken 11424 times.
✗ Branch 3 not taken.
1540738 switch (motion_type) {
839 1316564 case MT_FRAME: /* or MT_16X8 */
840
2/2
✓ Branch 0 taken 1163356 times.
✓ Branch 1 taken 153208 times.
1316564 if (s->picture_structure == PICT_FRAME) {
841 1163356 mb_type |= MB_TYPE_16x16;
842 1163356 s->mv_type = MV_TYPE_16X16;
843
2/2
✓ Branch 0 taken 2326712 times.
✓ Branch 1 taken 1163356 times.
3490068 for (i = 0; i < 2; i++) {
844
2/2
✓ Branch 0 taken 1494834 times.
✓ Branch 1 taken 831878 times.
2326712 if (USES_LIST(mb_type, i)) {
845 /* MT_FRAME */
846 1494834 s->mv[i][0][0] =
847 1494834 s->last_mv[i][0][0] =
848 1494834 s->last_mv[i][1][0] =
849 1494834 mpeg_decode_motion(s, s->mpeg_f_code[i][0],
850 s->last_mv[i][0][0]);
851 1494834 s->mv[i][0][1] =
852 1494834 s->last_mv[i][0][1] =
853 1494834 s->last_mv[i][1][1] =
854 1494834 mpeg_decode_motion(s, s->mpeg_f_code[i][1],
855 s->last_mv[i][0][1]);
856 /* full_pel: only for MPEG-1 */
857
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1494834 times.
1494834 if (s->full_pel[i]) {
858 s->mv[i][0][0] *= 2;
859 s->mv[i][0][1] *= 2;
860 }
861 }
862 }
863 } else {
864 153208 mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
865 153208 s->mv_type = MV_TYPE_16X8;
866
2/2
✓ Branch 0 taken 306416 times.
✓ Branch 1 taken 153208 times.
459624 for (i = 0; i < 2; i++) {
867
2/2
✓ Branch 0 taken 153889 times.
✓ Branch 1 taken 152527 times.
306416 if (USES_LIST(mb_type, i)) {
868 /* MT_16X8 */
869
2/2
✓ Branch 0 taken 307778 times.
✓ Branch 1 taken 153889 times.
461667 for (j = 0; j < 2; j++) {
870 307778 s->field_select[i][j] = get_bits1(&s->gb);
871
2/2
✓ Branch 0 taken 615556 times.
✓ Branch 1 taken 307778 times.
923334 for (k = 0; k < 2; k++) {
872 615556 val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
873 s->last_mv[i][j][k]);
874 615556 s->last_mv[i][j][k] = val;
875 615556 s->mv[i][j][k] = val;
876 }
877 }
878 }
879 }
880 }
881 1316564 break;
882 212750 case MT_FIELD:
883 212750 s->mv_type = MV_TYPE_FIELD;
884
2/2
✓ Branch 0 taken 69470 times.
✓ Branch 1 taken 143280 times.
212750 if (s->picture_structure == PICT_FRAME) {
885 69470 mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
886
2/2
✓ Branch 0 taken 138940 times.
✓ Branch 1 taken 69470 times.
208410 for (i = 0; i < 2; i++) {
887
2/2
✓ Branch 0 taken 85842 times.
✓ Branch 1 taken 53098 times.
138940 if (USES_LIST(mb_type, i)) {
888
2/2
✓ Branch 0 taken 171684 times.
✓ Branch 1 taken 85842 times.
257526 for (j = 0; j < 2; j++) {
889 171684 s->field_select[i][j] = get_bits1(&s->gb);
890 171684 val = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
891 s->last_mv[i][j][0]);
892 171684 s->last_mv[i][j][0] = val;
893 171684 s->mv[i][j][0] = val;
894 ff_tlog(s->avctx, "fmx=%d\n", val);
895 171684 val = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
896 171684 s->last_mv[i][j][1] >> 1);
897 171684 s->last_mv[i][j][1] = 2 * val;
898 171684 s->mv[i][j][1] = val;
899 ff_tlog(s->avctx, "fmy=%d\n", val);
900 }
901 }
902 }
903 } else {
904
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 143280 times.
143280 av_assert0(!s->progressive_sequence);
905 143280 mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
906
2/2
✓ Branch 0 taken 286560 times.
✓ Branch 1 taken 143280 times.
429840 for (i = 0; i < 2; i++) {
907
2/2
✓ Branch 0 taken 147940 times.
✓ Branch 1 taken 138620 times.
286560 if (USES_LIST(mb_type, i)) {
908 147940 s->field_select[i][0] = get_bits1(&s->gb);
909
2/2
✓ Branch 0 taken 295880 times.
✓ Branch 1 taken 147940 times.
443820 for (k = 0; k < 2; k++) {
910 295880 val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
911 s->last_mv[i][0][k]);
912 295880 s->last_mv[i][0][k] = val;
913 295880 s->last_mv[i][1][k] = val;
914 295880 s->mv[i][0][k] = val;
915 }
916 }
917 }
918 }
919 212750 break;
920 11424 case MT_DMV:
921
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11424 times.
11424 if (s->progressive_sequence){
922 av_log(s->avctx, AV_LOG_ERROR, "MT_DMV in progressive_sequence\n");
923 return AVERROR_INVALIDDATA;
924 }
925 11424 s->mv_type = MV_TYPE_DMV;
926
2/2
✓ Branch 0 taken 22848 times.
✓ Branch 1 taken 11424 times.
34272 for (i = 0; i < 2; i++) {
927
2/2
✓ Branch 0 taken 11424 times.
✓ Branch 1 taken 11424 times.
22848 if (USES_LIST(mb_type, i)) {
928 int dmx, dmy, mx, my, m;
929 11424 const int my_shift = s->picture_structure == PICT_FRAME;
930
931 11424 mx = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
932 s->last_mv[i][0][0]);
933 11424 s->last_mv[i][0][0] = mx;
934 11424 s->last_mv[i][1][0] = mx;
935 11424 dmx = get_dmv(s);
936 11424 my = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
937 11424 s->last_mv[i][0][1] >> my_shift);
938 11424 dmy = get_dmv(s);
939
940
941 11424 s->last_mv[i][0][1] = my * (1 << my_shift);
942 11424 s->last_mv[i][1][1] = my * (1 << my_shift);
943
944 11424 s->mv[i][0][0] = mx;
945 11424 s->mv[i][0][1] = my;
946 11424 s->mv[i][1][0] = mx; // not used
947 11424 s->mv[i][1][1] = my; // not used
948
949
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11424 times.
11424 if (s->picture_structure == PICT_FRAME) {
950 mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
951
952 // m = 1 + 2 * s->top_field_first;
953 m = s->top_field_first ? 1 : 3;
954
955 /* top -> top pred */
956 s->mv[i][2][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
957 s->mv[i][2][1] = ((my * m + (my > 0)) >> 1) + dmy - 1;
958 m = 4 - m;
959 s->mv[i][3][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
960 s->mv[i][3][1] = ((my * m + (my > 0)) >> 1) + dmy + 1;
961 } else {
962 11424 mb_type |= MB_TYPE_16x16;
963
964 11424 s->mv[i][2][0] = ((mx + (mx > 0)) >> 1) + dmx;
965 11424 s->mv[i][2][1] = ((my + (my > 0)) >> 1) + dmy;
966
2/2
✓ Branch 0 taken 9816 times.
✓ Branch 1 taken 1608 times.
11424 if (s->picture_structure == PICT_TOP_FIELD)
967 9816 s->mv[i][2][1]--;
968 else
969 1608 s->mv[i][2][1]++;
970 }
971 }
972 }
973 11424 break;
974 default:
975 av_log(s->avctx, AV_LOG_ERROR,
976 "00 motion_type at %d %d\n", s->mb_x, s->mb_y);
977 return AVERROR_INVALIDDATA;
978 }
979 }
980
981 1689223 s->mb_intra = 0;
982
2/2
✓ Branch 0 taken 1329521 times.
✓ Branch 1 taken 359702 times.
1689223 if (HAS_CBP(mb_type)) {
983 1329521 s->bdsp.clear_blocks(s->block[0]);
984
985 1329521 cbp = get_vlc2(&s->gb, ff_mb_pat_vlc.table, MB_PAT_VLC_BITS, 1);
986
2/2
✓ Branch 0 taken 46345 times.
✓ Branch 1 taken 1283176 times.
1329521 if (mb_block_count > 6) {
987 46345 cbp *= 1 << mb_block_count - 6;
988 46345 cbp |= get_bits(&s->gb, mb_block_count - 6);
989 46345 s->bdsp.clear_blocks(s->block[6]);
990 }
991
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1329521 times.
1329521 if (cbp <= 0) {
992 av_log(s->avctx, AV_LOG_ERROR,
993 "invalid cbp %d at %d %d\n", cbp, s->mb_x, s->mb_y);
994 return AVERROR_INVALIDDATA;
995 }
996
997
2/2
✓ Branch 0 taken 1217779 times.
✓ Branch 1 taken 111742 times.
1329521 if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
998
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1217779 times.
1217779 if (s->avctx->flags2 & AV_CODEC_FLAG2_FAST) {
999 for (i = 0; i < 6; i++) {
1000 if (cbp & 32)
1001 mpeg2_fast_decode_block_non_intra(s, *s->pblocks[i], i);
1002 else
1003 s->block_last_index[i] = -1;
1004 cbp += cbp;
1005 }
1006 } else {
1007 1217779 cbp <<= 12 - mb_block_count;
1008
1009
2/2
✓ Branch 0 taken 7399364 times.
✓ Branch 1 taken 1217779 times.
8617143 for (i = 0; i < mb_block_count; i++) {
1010
2/2
✓ Branch 0 taken 3924107 times.
✓ Branch 1 taken 3475257 times.
7399364 if (cbp & (1 << 11)) {
1011
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3924107 times.
3924107 if ((ret = mpeg2_decode_block_non_intra(s, *s->pblocks[i], i)) < 0)
1012 return ret;
1013 } else {
1014 3475257 s->block_last_index[i] = -1;
1015 }
1016 7399364 cbp += cbp;
1017 }
1018 }
1019 } else {
1020
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 111742 times.
111742 if (s->avctx->flags2 & AV_CODEC_FLAG2_FAST) {
1021 for (i = 0; i < 6; i++) {
1022 if (cbp & 32)
1023 mpeg1_fast_decode_block_inter(s, *s->pblocks[i], i);
1024 else
1025 s->block_last_index[i] = -1;
1026 cbp += cbp;
1027 }
1028 } else {
1029
2/2
✓ Branch 0 taken 670452 times.
✓ Branch 1 taken 111742 times.
782194 for (i = 0; i < 6; i++) {
1030
2/2
✓ Branch 0 taken 403118 times.
✓ Branch 1 taken 267334 times.
670452 if (cbp & 32) {
1031
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 403118 times.
403118 if ((ret = mpeg1_decode_block_inter(s, *s->pblocks[i], i)) < 0)
1032 return ret;
1033 } else {
1034 267334 s->block_last_index[i] = -1;
1035 }
1036 670452 cbp += cbp;
1037 }
1038 }
1039 }
1040 } else {
1041
2/2
✓ Branch 0 taken 4316424 times.
✓ Branch 1 taken 359702 times.
4676126 for (i = 0; i < 12; i++)
1042 4316424 s->block_last_index[i] = -1;
1043 }
1044 }
1045
1046 1968007 s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride] = mb_type;
1047
1048 1968007 return 0;
1049 }
1050
1051 264 static av_cold int mpeg_decode_init(AVCodecContext *avctx)
1052 {
1053 264 Mpeg1Context *s = avctx->priv_data;
1054 264 MpegEncContext *s2 = &s->mpeg_enc_ctx;
1055
1056
2/2
✓ Branch 0 taken 262 times.
✓ Branch 1 taken 2 times.
264 if ( avctx->codec_tag != AV_RL32("VCR2")
1057
1/2
✓ Branch 0 taken 262 times.
✗ Branch 1 not taken.
262 && avctx->codec_tag != AV_RL32("BW10"))
1058 262 avctx->coded_width = avctx->coded_height = 0; // do not trust dimensions from input
1059 264 ff_mpv_decode_init(s2, avctx);
1060
1061 /* we need some permutation to store matrices,
1062 * until the decoder sets the real permutation. */
1063 264 ff_mpv_idct_init(s2);
1064 264 ff_mpeg12_common_init(&s->mpeg_enc_ctx);
1065 264 ff_mpeg12_init_vlcs();
1066
1067 264 s2->chroma_format = 1;
1068 264 s->mpeg_enc_ctx_allocated = 0;
1069 264 s->repeat_field = 0;
1070 264 avctx->color_range = AVCOL_RANGE_MPEG;
1071 264 return 0;
1072 }
1073
1074 #if HAVE_THREADS
1075 static int mpeg_decode_update_thread_context(AVCodecContext *avctx,
1076 const AVCodecContext *avctx_from)
1077 {
1078 Mpeg1Context *ctx = avctx->priv_data, *ctx_from = avctx_from->priv_data;
1079 MpegEncContext *s = &ctx->mpeg_enc_ctx, *s1 = &ctx_from->mpeg_enc_ctx;
1080 int err;
1081
1082 if (avctx == avctx_from ||
1083 !ctx_from->mpeg_enc_ctx_allocated ||
1084 !s1->context_initialized)
1085 return 0;
1086
1087 err = ff_mpeg_update_thread_context(avctx, avctx_from);
1088 if (err)
1089 return err;
1090
1091 if (!ctx->mpeg_enc_ctx_allocated)
1092 memcpy(s + 1, s1 + 1, sizeof(Mpeg1Context) - sizeof(MpegEncContext));
1093
1094 return 0;
1095 }
1096 #endif
1097
1098 960 static void quant_matrix_rebuild(uint16_t *matrix, const uint8_t *old_perm,
1099 const uint8_t *new_perm)
1100 {
1101 uint16_t temp_matrix[64];
1102 int i;
1103
1104 960 memcpy(temp_matrix, matrix, 64 * sizeof(uint16_t));
1105
1106
2/2
✓ Branch 0 taken 61440 times.
✓ Branch 1 taken 960 times.
62400 for (i = 0; i < 64; i++)
1107 61440 matrix[new_perm[i]] = temp_matrix[old_perm[i]];
1108 960 }
1109
1110 static const enum AVPixelFormat mpeg1_hwaccel_pixfmt_list_420[] = {
1111 #if CONFIG_MPEG1_NVDEC_HWACCEL
1112 AV_PIX_FMT_CUDA,
1113 #endif
1114 #if CONFIG_MPEG1_VDPAU_HWACCEL
1115 AV_PIX_FMT_VDPAU,
1116 #endif
1117 AV_PIX_FMT_YUV420P,
1118 AV_PIX_FMT_NONE
1119 };
1120
1121 static const enum AVPixelFormat mpeg2_hwaccel_pixfmt_list_420[] = {
1122 #if CONFIG_MPEG2_NVDEC_HWACCEL
1123 AV_PIX_FMT_CUDA,
1124 #endif
1125 #if CONFIG_MPEG2_VDPAU_HWACCEL
1126 AV_PIX_FMT_VDPAU,
1127 #endif
1128 #if CONFIG_MPEG2_DXVA2_HWACCEL
1129 AV_PIX_FMT_DXVA2_VLD,
1130 #endif
1131 #if CONFIG_MPEG2_D3D11VA_HWACCEL
1132 AV_PIX_FMT_D3D11VA_VLD,
1133 AV_PIX_FMT_D3D11,
1134 #endif
1135 #if CONFIG_MPEG2_VAAPI_HWACCEL
1136 AV_PIX_FMT_VAAPI,
1137 #endif
1138 #if CONFIG_MPEG2_VIDEOTOOLBOX_HWACCEL
1139 AV_PIX_FMT_VIDEOTOOLBOX,
1140 #endif
1141 AV_PIX_FMT_YUV420P,
1142 AV_PIX_FMT_NONE
1143 };
1144
1145 static const enum AVPixelFormat mpeg12_pixfmt_list_422[] = {
1146 AV_PIX_FMT_YUV422P,
1147 AV_PIX_FMT_NONE
1148 };
1149
1150 static const enum AVPixelFormat mpeg12_pixfmt_list_444[] = {
1151 AV_PIX_FMT_YUV444P,
1152 AV_PIX_FMT_NONE
1153 };
1154
1155 242 static enum AVPixelFormat mpeg_get_pixelformat(AVCodecContext *avctx)
1156 {
1157 242 Mpeg1Context *s1 = avctx->priv_data;
1158 242 MpegEncContext *s = &s1->mpeg_enc_ctx;
1159 const enum AVPixelFormat *pix_fmts;
1160
1161 if (CONFIG_GRAY && (avctx->flags & AV_CODEC_FLAG_GRAY))
1162 return AV_PIX_FMT_GRAY8;
1163
1164
2/2
✓ Branch 0 taken 187 times.
✓ Branch 1 taken 55 times.
242 if (s->chroma_format < 2)
1165 187 pix_fmts = avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO ?
1166
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 161 times.
187 mpeg1_hwaccel_pixfmt_list_420 :
1167 mpeg2_hwaccel_pixfmt_list_420;
1168
1/2
✓ Branch 0 taken 55 times.
✗ Branch 1 not taken.
55 else if (s->chroma_format == 2)
1169 55 pix_fmts = mpeg12_pixfmt_list_422;
1170 else
1171 pix_fmts = mpeg12_pixfmt_list_444;
1172
1173 242 return ff_thread_get_format(avctx, pix_fmts);
1174 }
1175
1176 /* Call this function when we know all parameters.
1177 * It may be called in different places for MPEG-1 and MPEG-2. */
1178 4414 static int mpeg_decode_postinit(AVCodecContext *avctx)
1179 {
1180 4414 Mpeg1Context *s1 = avctx->priv_data;
1181 4414 MpegEncContext *s = &s1->mpeg_enc_ctx;
1182 uint8_t old_permutation[64];
1183 int ret;
1184
1185
2/2
✓ Branch 0 taken 494 times.
✓ Branch 1 taken 3920 times.
4414 if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
1186 // MPEG-1 aspect
1187 494 AVRational aspect_inv = av_d2q(ff_mpeg1_aspect[s1->aspect_ratio_info], 255);
1188 494 avctx->sample_aspect_ratio = (AVRational) { aspect_inv.den, aspect_inv.num };
1189 } else { // MPEG-2
1190 // MPEG-2 aspect
1191
2/2
✓ Branch 0 taken 1930 times.
✓ Branch 1 taken 1990 times.
3920 if (s1->aspect_ratio_info > 1) {
1192 AVRational dar =
1193 1930 av_mul_q(av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
1194 1930 (AVRational) { s1->pan_scan.width,
1195 1930 s1->pan_scan.height }),
1196 1930 (AVRational) { s->width, s->height });
1197
1198 /* We ignore the spec here and guess a bit as reality does not
1199 * match the spec, see for example res_change_ffmpeg_aspect.ts
1200 * and sequence-display-aspect.mpg.
1201 * issue1613, 621, 562 */
1202
5/6
✓ Branch 0 taken 432 times.
✓ Branch 1 taken 1498 times.
✓ Branch 2 taken 432 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 14 times.
✓ Branch 5 taken 418 times.
2362 if ((s1->pan_scan.width == 0) || (s1->pan_scan.height == 0) ||
1203
1/2
✓ Branch 1 taken 14 times.
✗ Branch 2 not taken.
446 (av_cmp_q(dar, (AVRational) { 4, 3 }) &&
1204 14 av_cmp_q(dar, (AVRational) { 16, 9 }))) {
1205 1512 s->avctx->sample_aspect_ratio =
1206 1512 av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
1207 1512 (AVRational) { s->width, s->height });
1208 } else {
1209 418 s->avctx->sample_aspect_ratio =
1210 418 av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
1211 418 (AVRational) { s1->pan_scan.width, s1->pan_scan.height });
1212 // issue1613 4/3 16/9 -> 16/9
1213 // res_change_ffmpeg_aspect.ts 4/3 225/44 ->4/3
1214 // widescreen-issue562.mpg 4/3 16/9 -> 16/9
1215 // s->avctx->sample_aspect_ratio = av_mul_q(s->avctx->sample_aspect_ratio, (AVRational) {s->width, s->height});
1216 ff_dlog(avctx, "aspect A %d/%d\n",
1217 ff_mpeg2_aspect[s1->aspect_ratio_info].num,
1218 ff_mpeg2_aspect[s1->aspect_ratio_info].den);
1219 ff_dlog(avctx, "aspect B %d/%d\n", s->avctx->sample_aspect_ratio.num,
1220 s->avctx->sample_aspect_ratio.den);
1221 }
1222 } else {
1223 1990 s->avctx->sample_aspect_ratio =
1224 1990 ff_mpeg2_aspect[s1->aspect_ratio_info];
1225 }
1226 } // MPEG-2
1227
1228
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4414 times.
4414 if (av_image_check_sar(s->width, s->height,
1229 avctx->sample_aspect_ratio) < 0) {
1230 av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
1231 avctx->sample_aspect_ratio.num,
1232 avctx->sample_aspect_ratio.den);
1233 avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
1234 }
1235
1236
2/2
✓ Branch 0 taken 4176 times.
✓ Branch 1 taken 238 times.
4414 if ((s1->mpeg_enc_ctx_allocated == 0) ||
1237
1/2
✓ Branch 0 taken 4176 times.
✗ Branch 1 not taken.
4176 avctx->coded_width != s->width ||
1238
1/2
✓ Branch 0 taken 4176 times.
✗ Branch 1 not taken.
4176 avctx->coded_height != s->height ||
1239
1/2
✓ Branch 0 taken 4176 times.
✗ Branch 1 not taken.
4176 s1->save_width != s->width ||
1240
3/4
✓ Branch 0 taken 4176 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4174 times.
✓ Branch 3 taken 2 times.
8352 s1->save_height != s->height ||
1241 4176 av_cmp_q(s1->save_aspect, s->avctx->sample_aspect_ratio) ||
1242
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4174 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
4174 (s1->save_progressive_seq != s->progressive_sequence && FFALIGN(s->height, 16) != FFALIGN(s->height, 32)) ||
1243 0) {
1244
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 238 times.
240 if (s1->mpeg_enc_ctx_allocated) {
1245 #if FF_API_FLAG_TRUNCATED
1246 2 ParseContext pc = s->parse_context;
1247 2 s->parse_context.buffer = 0;
1248 2 ff_mpv_common_end(s);
1249 2 s->parse_context = pc;
1250 #else
1251 ff_mpv_common_end(s);
1252 #endif
1253 2 s1->mpeg_enc_ctx_allocated = 0;
1254 }
1255
1256 240 ret = ff_set_dimensions(avctx, s->width, s->height);
1257
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 240 times.
240 if (ret < 0)
1258 return ret;
1259
1260
4/4
✓ Branch 0 taken 215 times.
✓ Branch 1 taken 25 times.
✓ Branch 2 taken 213 times.
✓ Branch 3 taken 2 times.
240 if (avctx->codec_id == AV_CODEC_ID_MPEG2VIDEO && s->bit_rate) {
1261 213 avctx->rc_max_rate = s->bit_rate;
1262
3/4
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 25 times.
✗ Branch 3 not taken.
27 } else if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO && s->bit_rate &&
1263
2/4
✓ Branch 0 taken 25 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 25 times.
✗ Branch 3 not taken.
25 (s->bit_rate != 0x3FFFF*400 || s->vbv_delay != 0xFFFF)) {
1264 25 avctx->bit_rate = s->bit_rate;
1265 }
1266 240 s1->save_aspect = s->avctx->sample_aspect_ratio;
1267 240 s1->save_width = s->width;
1268 240 s1->save_height = s->height;
1269 240 s1->save_progressive_seq = s->progressive_sequence;
1270
1271 /* low_delay may be forced, in this case we will have B-frames
1272 * that behave like P-frames. */
1273 240 avctx->has_b_frames = !s->low_delay;
1274
1275
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 215 times.
240 if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
1276 // MPEG-1 fps
1277 25 avctx->framerate = ff_mpeg12_frame_rate_tab[s1->frame_rate_index];
1278 25 avctx->ticks_per_frame = 1;
1279
1280 25 avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
1281 } else { // MPEG-2
1282 // MPEG-2 fps
1283 215 av_reduce(&s->avctx->framerate.num,
1284 215 &s->avctx->framerate.den,
1285 215 ff_mpeg12_frame_rate_tab[s1->frame_rate_index].num * s1->frame_rate_ext.num,
1286 215 ff_mpeg12_frame_rate_tab[s1->frame_rate_index].den * s1->frame_rate_ext.den,
1287 1 << 30);
1288 215 avctx->ticks_per_frame = 2;
1289
1290
2/3
✓ Branch 0 taken 160 times.
✓ Branch 1 taken 55 times.
✗ Branch 2 not taken.
215 switch (s->chroma_format) {
1291 160 case 1: avctx->chroma_sample_location = AVCHROMA_LOC_LEFT; break;
1292 55 case 2:
1293 55 case 3: avctx->chroma_sample_location = AVCHROMA_LOC_TOPLEFT; break;
1294 default: av_assert0(0);
1295 }
1296 } // MPEG-2
1297
1298 240 avctx->pix_fmt = mpeg_get_pixelformat(avctx);
1299
1300 /* Quantization matrices may need reordering
1301 * if DCT permutation is changed. */
1302 240 memcpy(old_permutation, s->idsp.idct_permutation, 64 * sizeof(uint8_t));
1303
1304 240 ff_mpv_idct_init(s);
1305
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 240 times.
240 if ((ret = ff_mpv_common_init(s)) < 0)
1306 return ret;
1307
1308 240 quant_matrix_rebuild(s->intra_matrix, old_permutation, s->idsp.idct_permutation);
1309 240 quant_matrix_rebuild(s->inter_matrix, old_permutation, s->idsp.idct_permutation);
1310 240 quant_matrix_rebuild(s->chroma_intra_matrix, old_permutation, s->idsp.idct_permutation);
1311 240 quant_matrix_rebuild(s->chroma_inter_matrix, old_permutation, s->idsp.idct_permutation);
1312
1313 240 s1->mpeg_enc_ctx_allocated = 1;
1314 }
1315 4414 return 0;
1316 }
1317
1318 4414 static int mpeg1_decode_picture(AVCodecContext *avctx, const uint8_t *buf,
1319 int buf_size)
1320 {
1321 4414 Mpeg1Context *s1 = avctx->priv_data;
1322 4414 MpegEncContext *s = &s1->mpeg_enc_ctx;
1323 int ref, f_code, vbv_delay, ret;
1324
1325 4414 ret = init_get_bits8(&s->gb, buf, buf_size);
1326
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4414 times.
4414 if (ret < 0)
1327 return ret;
1328
1329 4414 ref = get_bits(&s->gb, 10); /* temporal ref */
1330 4414 s->pict_type = get_bits(&s->gb, 3);
1331
2/4
✓ Branch 0 taken 4414 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4414 times.
4414 if (s->pict_type == 0 || s->pict_type > 3)
1332 return AVERROR_INVALIDDATA;
1333
1334 4414 vbv_delay = get_bits(&s->gb, 16);
1335 4414 s->vbv_delay = vbv_delay;
1336
2/2
✓ Branch 0 taken 2187 times.
✓ Branch 1 taken 2227 times.
4414 if (s->pict_type == AV_PICTURE_TYPE_P ||
1337
2/2
✓ Branch 0 taken 1520 times.
✓ Branch 1 taken 667 times.
2187 s->pict_type == AV_PICTURE_TYPE_B) {
1338 3747 s->full_pel[0] = get_bits1(&s->gb);
1339 3747 f_code = get_bits(&s->gb, 3);
1340
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 3747 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
3747 if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1341 return AVERROR_INVALIDDATA;
1342 3747 f_code += !f_code;
1343 3747 s->mpeg_f_code[0][0] = f_code;
1344 3747 s->mpeg_f_code[0][1] = f_code;
1345 }
1346
2/2
✓ Branch 0 taken 1520 times.
✓ Branch 1 taken 2894 times.
4414 if (s->pict_type == AV_PICTURE_TYPE_B) {
1347 1520 s->full_pel[1] = get_bits1(&s->gb);
1348 1520 f_code = get_bits(&s->gb, 3);
1349
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1520 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
1520 if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1350 return AVERROR_INVALIDDATA;
1351 1520 f_code += !f_code;
1352 1520 s->mpeg_f_code[1][0] = f_code;
1353 1520 s->mpeg_f_code[1][1] = f_code;
1354 }
1355 4414 s->current_picture.f->pict_type = s->pict_type;
1356 4414 s->current_picture.f->key_frame = s->pict_type == AV_PICTURE_TYPE_I;
1357
1358
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4414 times.
4414 if (avctx->debug & FF_DEBUG_PICT_INFO)
1359 av_log(avctx, AV_LOG_DEBUG,
1360 "vbv_delay %d, ref %d type:%d\n", vbv_delay, ref, s->pict_type);
1361
1362 4414 s->y_dc_scale = 8;
1363 4414 s->c_dc_scale = 8;
1364 4414 return 0;
1365 }
1366
1367 654 static void mpeg_decode_sequence_extension(Mpeg1Context *s1)
1368 {
1369 654 MpegEncContext *s = &s1->mpeg_enc_ctx;
1370 int horiz_size_ext, vert_size_ext;
1371 int bit_rate_ext;
1372 AVCPBProperties *cpb_props;
1373
1374 654 skip_bits(&s->gb, 1); /* profile and level esc*/
1375 654 s->avctx->profile = get_bits(&s->gb, 3);
1376 654 s->avctx->level = get_bits(&s->gb, 4);
1377 654 s->progressive_sequence = get_bits1(&s->gb); /* progressive_sequence */
1378 654 s->chroma_format = get_bits(&s->gb, 2); /* chroma_format 1=420, 2=422, 3=444 */
1379
1380
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 654 times.
654 if (!s->chroma_format) {
1381 s->chroma_format = 1;
1382 av_log(s->avctx, AV_LOG_WARNING, "Chroma format invalid\n");
1383 }
1384
1385 654 horiz_size_ext = get_bits(&s->gb, 2);
1386 654 vert_size_ext = get_bits(&s->gb, 2);
1387 654 s->width |= (horiz_size_ext << 12);
1388 654 s->height |= (vert_size_ext << 12);
1389 654 bit_rate_ext = get_bits(&s->gb, 12); /* XXX: handle it */
1390 654 s->bit_rate += (bit_rate_ext << 18) * 400LL;
1391 654 check_marker(s->avctx, &s->gb, "after bit rate extension");
1392 654 s1->rc_buffer_size += get_bits(&s->gb, 8) * 1024 * 16 << 10;
1393
1394 654 s->low_delay = get_bits1(&s->gb);
1395
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 654 times.
654 if (s->avctx->flags & AV_CODEC_FLAG_LOW_DELAY)
1396 s->low_delay = 1;
1397
1398 654 s1->frame_rate_ext.num = get_bits(&s->gb, 2) + 1;
1399 654 s1->frame_rate_ext.den = get_bits(&s->gb, 5) + 1;
1400
1401 ff_dlog(s->avctx, "sequence extension\n");
1402 654 s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
1403
1404
1/2
✓ Branch 1 taken 654 times.
✗ Branch 2 not taken.
654 if (cpb_props = ff_add_cpb_side_data(s->avctx)) {
1405 654 cpb_props->buffer_size = s1->rc_buffer_size;
1406
2/2
✓ Branch 0 taken 239 times.
✓ Branch 1 taken 415 times.
654 if (s->bit_rate != 0x3FFFF*400)
1407 239 cpb_props->max_bitrate = s->bit_rate;
1408 }
1409
1410
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 654 times.
654 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1411 av_log(s->avctx, AV_LOG_DEBUG,
1412 "profile: %d, level: %d ps: %d cf:%d vbv buffer: %d, bitrate:%"PRId64"\n",
1413 s->avctx->profile, s->avctx->level, s->progressive_sequence, s->chroma_format,
1414 s1->rc_buffer_size, s->bit_rate);
1415 654 }
1416
1417 76 static void mpeg_decode_sequence_display_extension(Mpeg1Context *s1)
1418 {
1419 76 MpegEncContext *s = &s1->mpeg_enc_ctx;
1420 int color_description, w, h;
1421
1422 76 skip_bits(&s->gb, 3); /* video format */
1423 76 color_description = get_bits1(&s->gb);
1424
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 45 times.
76 if (color_description) {
1425 31 s->avctx->color_primaries = get_bits(&s->gb, 8);
1426 31 s->avctx->color_trc = get_bits(&s->gb, 8);
1427 31 s->avctx->colorspace = get_bits(&s->gb, 8);
1428 }
1429 76 w = get_bits(&s->gb, 14);
1430 76 skip_bits(&s->gb, 1); // marker
1431 76 h = get_bits(&s->gb, 14);
1432 // remaining 3 bits are zero padding
1433
1434 76 s1->pan_scan.width = 16 * w;
1435 76 s1->pan_scan.height = 16 * h;
1436
1437
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 76 times.
76 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1438 av_log(s->avctx, AV_LOG_DEBUG, "sde w:%d, h:%d\n", w, h);
1439 76 }
1440
1441 15 static void mpeg_decode_picture_display_extension(Mpeg1Context *s1)
1442 {
1443 15 MpegEncContext *s = &s1->mpeg_enc_ctx;
1444 int i, nofco;
1445
1446 15 nofco = 1;
1447
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
15 if (s->progressive_sequence) {
1448 if (s->repeat_first_field) {
1449 nofco++;
1450 if (s->top_field_first)
1451 nofco++;
1452 }
1453 } else {
1454
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 10 times.
15 if (s->picture_structure == PICT_FRAME) {
1455 5 nofco++;
1456
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 2 times.
5 if (s->repeat_first_field)
1457 3 nofco++;
1458 }
1459 }
1460
2/2
✓ Branch 0 taken 23 times.
✓ Branch 1 taken 15 times.
38 for (i = 0; i < nofco; i++) {
1461 23 s1->pan_scan.position[i][0] = get_sbits(&s->gb, 16);
1462 23 skip_bits(&s->gb, 1); // marker
1463 23 s1->pan_scan.position[i][1] = get_sbits(&s->gb, 16);
1464 23 skip_bits(&s->gb, 1); // marker
1465 }
1466
1467
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
15 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1468 av_log(s->avctx, AV_LOG_DEBUG,
1469 "pde (%"PRId16",%"PRId16") (%"PRId16",%"PRId16") (%"PRId16",%"PRId16")\n",
1470 s1->pan_scan.position[0][0], s1->pan_scan.position[0][1],
1471 s1->pan_scan.position[1][0], s1->pan_scan.position[1][1],
1472 s1->pan_scan.position[2][0], s1->pan_scan.position[2][1]);
1473 15 }
1474
1475 150 static int load_matrix(MpegEncContext *s, uint16_t matrix0[64],
1476 uint16_t matrix1[64], int intra)
1477 {
1478 int i;
1479
1480
2/2
✓ Branch 0 taken 9600 times.
✓ Branch 1 taken 150 times.
9750 for (i = 0; i < 64; i++) {
1481 9600 int j = s->idsp.idct_permutation[ff_zigzag_direct[i]];
1482 9600 int v = get_bits(&s->gb, 8);
1483
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9600 times.
9600 if (v == 0) {
1484 av_log(s->avctx, AV_LOG_ERROR, "matrix damaged\n");
1485 return AVERROR_INVALIDDATA;
1486 }
1487
5/6
✓ Branch 0 taken 2624 times.
✓ Branch 1 taken 6976 times.
✓ Branch 2 taken 41 times.
✓ Branch 3 taken 2583 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 41 times.
9600 if (intra && i == 0 && v != 8) {
1488 av_log(s->avctx, AV_LOG_DEBUG, "intra matrix specifies invalid DC quantizer %d, ignoring\n", v);
1489 v = 8; // needed by pink.mpg / issue1046
1490 }
1491 9600 matrix0[j] = v;
1492
2/2
✓ Branch 0 taken 9216 times.
✓ Branch 1 taken 384 times.
9600 if (matrix1)
1493 9216 matrix1[j] = v;
1494 }
1495 150 return 0;
1496 }
1497
1498 531 static void mpeg_decode_quant_matrix_extension(MpegEncContext *s)
1499 {
1500 ff_dlog(s->avctx, "matrix extension\n");
1501
1502
2/2
✓ Branch 1 taken 18 times.
✓ Branch 2 taken 513 times.
531 if (get_bits1(&s->gb))
1503 18 load_matrix(s, s->chroma_intra_matrix, s->intra_matrix, 1);
1504
2/2
✓ Branch 1 taken 41 times.
✓ Branch 2 taken 490 times.
531 if (get_bits1(&s->gb))
1505 41 load_matrix(s, s->chroma_inter_matrix, s->inter_matrix, 0);
1506
2/2
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 526 times.
531 if (get_bits1(&s->gb))
1507 5 load_matrix(s, s->chroma_intra_matrix, NULL, 1);
1508
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 530 times.
531 if (get_bits1(&s->gb))
1509 1 load_matrix(s, s->chroma_inter_matrix, NULL, 0);
1510 531 }
1511
1512 3920 static int mpeg_decode_picture_coding_extension(Mpeg1Context *s1)
1513 {
1514 3920 MpegEncContext *s = &s1->mpeg_enc_ctx;
1515
1516 3920 s->full_pel[0] = s->full_pel[1] = 0;
1517 3920 s->mpeg_f_code[0][0] = get_bits(&s->gb, 4);
1518 3920 s->mpeg_f_code[0][1] = get_bits(&s->gb, 4);
1519 3920 s->mpeg_f_code[1][0] = get_bits(&s->gb, 4);
1520 3920 s->mpeg_f_code[1][1] = get_bits(&s->gb, 4);
1521 3920 s->mpeg_f_code[0][0] += !s->mpeg_f_code[0][0];
1522 3920 s->mpeg_f_code[0][1] += !s->mpeg_f_code[0][1];
1523 3920 s->mpeg_f_code[1][0] += !s->mpeg_f_code[1][0];
1524 3920 s->mpeg_f_code[1][1] += !s->mpeg_f_code[1][1];
1525
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 3920 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
3920 if (!s->pict_type && s1->mpeg_enc_ctx_allocated) {
1526 av_log(s->avctx, AV_LOG_ERROR, "Missing picture start code\n");
1527 if (s->avctx->err_recognition & AV_EF_EXPLODE)
1528 return AVERROR_INVALIDDATA;
1529 av_log(s->avctx, AV_LOG_WARNING, "Guessing pict_type from mpeg_f_code\n");
1530 if (s->mpeg_f_code[1][0] == 15 && s->mpeg_f_code[1][1] == 15) {
1531 if (s->mpeg_f_code[0][0] == 15 && s->mpeg_f_code[0][1] == 15)
1532 s->pict_type = AV_PICTURE_TYPE_I;
1533 else
1534 s->pict_type = AV_PICTURE_TYPE_P;
1535 } else
1536 s->pict_type = AV_PICTURE_TYPE_B;
1537 s->current_picture.f->pict_type = s->pict_type;
1538 s->current_picture.f->key_frame = s->pict_type == AV_PICTURE_TYPE_I;
1539 }
1540
1541 3920 s->intra_dc_precision = get_bits(&s->gb, 2);
1542 3920 s->picture_structure = get_bits(&s->gb, 2);
1543 3920 s->top_field_first = get_bits1(&s->gb);
1544 3920 s->frame_pred_frame_dct = get_bits1(&s->gb);
1545 3920 s->concealment_motion_vectors = get_bits1(&s->gb);
1546 3920 s->q_scale_type = get_bits1(&s->gb);
1547 3920 s->intra_vlc_format = get_bits1(&s->gb);
1548 3920 s->alternate_scan = get_bits1(&s->gb);
1549 3920 s->repeat_first_field = get_bits1(&s->gb);
1550 3920 s->chroma_420_type = get_bits1(&s->gb);
1551 3920 s->progressive_frame = get_bits1(&s->gb);
1552
1553
2/2
✓ Branch 0 taken 811 times.
✓ Branch 1 taken 3109 times.
3920 if (s->alternate_scan) {
1554 811 ff_init_scantable(s->idsp.idct_permutation, &s->inter_scantable, ff_alternate_vertical_scan);
1555 811 ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable, ff_alternate_vertical_scan);
1556 } else {
1557 3109 ff_init_scantable(s->idsp.idct_permutation, &s->inter_scantable, ff_zigzag_direct);
1558 3109 ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable, ff_zigzag_direct);
1559 }
1560
1561 /* composite display not parsed */
1562 ff_dlog(s->avctx, "intra_dc_precision=%d\n", s->intra_dc_precision);
1563 ff_dlog(s->avctx, "picture_structure=%d\n", s->picture_structure);
1564 ff_dlog(s->avctx, "top field first=%d\n", s->top_field_first);
1565 ff_dlog(s->avctx, "repeat first field=%d\n", s->repeat_first_field);
1566 ff_dlog(s->avctx, "conceal=%d\n", s->concealment_motion_vectors);
1567 ff_dlog(s->avctx, "intra_vlc_format=%d\n", s->intra_vlc_format);
1568 ff_dlog(s->avctx, "alternate_scan=%d\n", s->alternate_scan);
1569 ff_dlog(s->avctx, "frame_pred_frame_dct=%d\n", s->frame_pred_frame_dct);
1570 ff_dlog(s->avctx, "progressive_frame=%d\n", s->progressive_frame);
1571
1572 3920 return 0;
1573 }
1574
1575 4080 static int mpeg_field_start(MpegEncContext *s, const uint8_t *buf, int buf_size)
1576 {
1577 4080 AVCodecContext *avctx = s->avctx;
1578 4080 Mpeg1Context *s1 = (Mpeg1Context *) s;
1579 int ret;
1580
1581
1/2
✓ Branch 0 taken 4080 times.
✗ Branch 1 not taken.
4080 if (!(avctx->flags2 & AV_CODEC_FLAG2_CHUNKS)) {
1582
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4080 times.
4080 if (s->mb_width * s->mb_height * 11LL / (33 * 2 * 8) > buf_size)
1583 return AVERROR_INVALIDDATA;
1584 }
1585
1586 /* start frame decoding */
1587
4/4
✓ Branch 0 taken 3538 times.
✓ Branch 1 taken 542 times.
✓ Branch 2 taken 2996 times.
✓ Branch 3 taken 542 times.
7618 if (s->first_field || s->picture_structure == PICT_FRAME) {
1588 AVFrameSideData *pan_scan;
1589
1590
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3538 times.
3538 if ((ret = ff_mpv_frame_start(s, avctx)) < 0)
1591 return ret;
1592
1593 3538 ff_mpeg_er_frame_start(s);
1594
1595 /* first check if we must repeat the frame */
1596 3538 s->current_picture_ptr->f->repeat_pict = 0;
1597
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 3535 times.
3538 if (s->repeat_first_field) {
1598
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
3 if (s->progressive_sequence) {
1599 if (s->top_field_first)
1600 s->current_picture_ptr->f->repeat_pict = 4;
1601 else
1602 s->current_picture_ptr->f->repeat_pict = 2;
1603
1/2
✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
3 } else if (s->progressive_frame) {
1604 3 s->current_picture_ptr->f->repeat_pict = 1;
1605 }
1606 }
1607
1608 3538 pan_scan = av_frame_new_side_data(s->current_picture_ptr->f,
1609 AV_FRAME_DATA_PANSCAN,
1610 sizeof(s1->pan_scan));
1611
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3538 times.
3538 if (!pan_scan)
1612 return AVERROR(ENOMEM);
1613 3538 memcpy(pan_scan->data, &s1->pan_scan, sizeof(s1->pan_scan));
1614
1615
2/2
✓ Branch 0 taken 607 times.
✓ Branch 1 taken 2931 times.
3538 if (s1->a53_buf_ref) {
1616 607 AVFrameSideData *sd = av_frame_new_side_data_from_buf(
1617 607 s->current_picture_ptr->f, AV_FRAME_DATA_A53_CC,
1618 s1->a53_buf_ref);
1619
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 607 times.
607 if (!sd)
1620 av_buffer_unref(&s1->a53_buf_ref);
1621 607 s1->a53_buf_ref = NULL;
1622 }
1623
1624
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3538 times.
3538 if (s1->has_stereo3d) {
1625 AVStereo3D *stereo = av_stereo3d_create_side_data(s->current_picture_ptr->f);
1626 if (!stereo)
1627 return AVERROR(ENOMEM);
1628
1629 *stereo = s1->stereo3d;
1630 s1->has_stereo3d = 0;
1631 }
1632
1633
2/2
✓ Branch 0 taken 202 times.
✓ Branch 1 taken 3336 times.
3538 if (s1->has_afd) {
1634 AVFrameSideData *sd =
1635 202 av_frame_new_side_data(s->current_picture_ptr->f,
1636 AV_FRAME_DATA_AFD, 1);
1637
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 202 times.
202 if (!sd)
1638 return AVERROR(ENOMEM);
1639
1640 202 *sd->data = s1->afd;
1641 202 s1->has_afd = 0;
1642 }
1643
1644
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3538 times.
3538 if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_FRAME))
1645 ff_thread_finish_setup(avctx);
1646 } else { // second field
1647 int i;
1648
1649
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 542 times.
542 if (!s->current_picture_ptr) {
1650 av_log(s->avctx, AV_LOG_ERROR, "first field missing\n");
1651 return AVERROR_INVALIDDATA;
1652 }
1653
1654
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 542 times.
542 if (s->avctx->hwaccel) {
1655 if ((ret = s->avctx->hwaccel->end_frame(s->avctx)) < 0) {
1656 av_log(avctx, AV_LOG_ERROR,
1657 "hardware accelerator failed to decode first field\n");
1658 return ret;
1659 }
1660 }
1661
1662
2/2
✓ Branch 0 taken 2168 times.
✓ Branch 1 taken 542 times.
2710 for (i = 0; i < 4; i++) {
1663 2168 s->current_picture.f->data[i] = s->current_picture_ptr->f->data[i];
1664
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 2164 times.
2168 if (s->picture_structure == PICT_BOTTOM_FIELD)
1665 4 s->current_picture.f->data[i] +=
1666 4 s->current_picture_ptr->f->linesize[i];
1667 }
1668 }
1669
1670
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4080 times.
4080 if (avctx->hwaccel) {
1671 if ((ret = avctx->hwaccel->start_frame(avctx, buf, buf_size)) < 0)
1672 return ret;
1673 }
1674
1675 4080 return 0;
1676 }
1677
1678 #define DECODE_SLICE_ERROR -1
1679 #define DECODE_SLICE_OK 0
1680
1681 /**
1682 * Decode a slice.
1683 * MpegEncContext.mb_y must be set to the MB row from the startcode.
1684 * @return DECODE_SLICE_ERROR if the slice is damaged,
1685 * DECODE_SLICE_OK if this slice is OK
1686 */
1687 112865 static int mpeg_decode_slice(MpegEncContext *s, int mb_y,
1688 const uint8_t **buf, int buf_size)
1689 {
1690 112865 AVCodecContext *avctx = s->avctx;
1691 112865 const int lowres = s->avctx->lowres;
1692 112865 const int field_pic = s->picture_structure != PICT_FRAME;
1693 int ret;
1694
1695 112865 s->resync_mb_x =
1696 112865 s->resync_mb_y = -1;
1697
1698
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 112865 times.
112865 av_assert0(mb_y < s->mb_height);
1699
1700 112865 init_get_bits(&s->gb, *buf, buf_size * 8);
1701
3/4
✓ Branch 0 taken 105592 times.
✓ Branch 1 taken 7273 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 105592 times.
112865 if (s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
1702 skip_bits(&s->gb, 3);
1703
1704 112865 ff_mpeg1_clean_buffers(s);
1705 112865 s->interlaced_dct = 0;
1706
1707 112865 s->qscale = mpeg_get_qscale(s);
1708
1709
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 112865 times.
112865 if (s->qscale == 0) {
1710 av_log(s->avctx, AV_LOG_ERROR, "qscale == 0\n");
1711 return AVERROR_INVALIDDATA;
1712 }
1713
1714 /* extra slice info */
1715
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 112865 times.
112865 if (skip_1stop_8data_bits(&s->gb) < 0)
1716 return AVERROR_INVALIDDATA;
1717
1718 112865 s->mb_x = 0;
1719
1720
3/4
✓ Branch 0 taken 5001 times.
✓ Branch 1 taken 107864 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 5001 times.
112865 if (mb_y == 0 && s->codec_tag == AV_RL32("SLIF")) {
1721 skip_bits1(&s->gb);
1722 } else {
1723
1/2
✓ Branch 1 taken 125064 times.
✗ Branch 2 not taken.
125064 while (get_bits_left(&s->gb) > 0) {
1724 125064 int code = get_vlc2(&s->gb, ff_mbincr_vlc.table,
1725 MBINCR_VLC_BITS, 2);
1726
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 125064 times.
125064 if (code < 0) {
1727 av_log(s->avctx, AV_LOG_ERROR, "first mb_incr damaged\n");
1728 return AVERROR_INVALIDDATA;
1729 }
1730
2/2
✓ Branch 0 taken 12199 times.
✓ Branch 1 taken 112865 times.
125064 if (code >= 33) {
1731
1/2
✓ Branch 0 taken 12199 times.
✗ Branch 1 not taken.
12199 if (code == 33)
1732 12199 s->mb_x += 33;
1733 /* otherwise, stuffing, nothing to do */
1734 } else {
1735 112865 s->mb_x += code;
1736 112865 break;
1737 }
1738 }
1739 }
1740
1741
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 112865 times.
112865 if (s->mb_x >= (unsigned) s->mb_width) {
1742 av_log(s->avctx, AV_LOG_ERROR, "initial skip overflow\n");
1743 return AVERROR_INVALIDDATA;
1744 }
1745
1746
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 112865 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
112865 if (avctx->hwaccel && avctx->hwaccel->decode_slice) {
1747 const uint8_t *buf_end, *buf_start = *buf - 4; /* include start_code */
1748 int start_code = -1;
1749 buf_end = avpriv_find_start_code(buf_start + 2, *buf + buf_size, &start_code);
1750 if (buf_end < *buf + buf_size)
1751 buf_end -= 4;
1752 s->mb_y = mb_y;
1753 if (avctx->hwaccel->decode_slice(avctx, buf_start, buf_end - buf_start) < 0)
1754 return DECODE_SLICE_ERROR;
1755 *buf = buf_end;
1756 return DECODE_SLICE_OK;
1757 }
1758
1759 112865 s->resync_mb_x = s->mb_x;
1760 112865 s->resync_mb_y = s->mb_y = mb_y;
1761 112865 s->mb_skip_run = 0;
1762 112865 ff_init_block_index(s);
1763
1764
8/8
✓ Branch 0 taken 5001 times.
✓ Branch 1 taken 107864 times.
✓ Branch 2 taken 3538 times.
✓ Branch 3 taken 1463 times.
✓ Branch 4 taken 3537 times.
✓ Branch 5 taken 1 times.
✓ Branch 6 taken 2996 times.
✓ Branch 7 taken 541 times.
112865 if (s->mb_y == 0 && s->mb_x == 0 && (s->first_field || s->picture_structure == PICT_FRAME)) {
1765
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2997 times.
2997 if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
1766 av_log(s->avctx, AV_LOG_DEBUG,
1767 "qp:%d fc:%2d%2d%2d%2d %c %s %s %s %s dc:%d pstruct:%d fdct:%d cmv:%d qtype:%d ivlc:%d rff:%d %s\n",
1768 s->qscale,
1769 s->mpeg_f_code[0][0], s->mpeg_f_code[0][1],
1770 s->mpeg_f_code[1][0], s->mpeg_f_code[1][1],
1771 s->pict_type == AV_PICTURE_TYPE_I ? 'I' :
1772 (s->pict_type == AV_PICTURE_TYPE_P ? 'P' :
1773 (s->pict_type == AV_PICTURE_TYPE_B ? 'B' : 'S')),
1774 s->progressive_sequence ? "ps" : "",
1775 s->progressive_frame ? "pf" : "",
1776 s->alternate_scan ? "alt" : "",
1777 s->top_field_first ? "top" : "",
1778 s->intra_dc_precision, s->picture_structure,
1779 s->frame_pred_frame_dct, s->concealment_motion_vectors,
1780 s->q_scale_type, s->intra_vlc_format,
1781 s->repeat_first_field, s->chroma_420_type ? "420" : "");
1782 }
1783 }
1784
1785 for (;;) {
1786
2/2
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 2124174 times.
2124178 if ((ret = mpeg_decode_mb(s, s->block)) < 0)
1787 4 return ret;
1788
1789 // Note motion_val is normally NULL unless we want to extract the MVs.
1790
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 2124174 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
2124174 if (s->current_picture.motion_val[0] && !s->encoding) {
1791 const int wrap = s->b8_stride;
1792 int xy = s->mb_x * 2 + s->mb_y * 2 * wrap;
1793 int b8_xy = 4 * (s->mb_x + s->mb_y * s->mb_stride);
1794 int motion_x, motion_y, dir, i;
1795
1796 for (i = 0; i < 2; i++) {
1797 for (dir = 0; dir < 2; dir++) {
1798 if (s->mb_intra ||
1799 (dir == 1 && s->pict_type != AV_PICTURE_TYPE_B)) {
1800 motion_x = motion_y = 0;
1801 } else if (s->mv_type == MV_TYPE_16X16 ||
1802 (s->mv_type == MV_TYPE_FIELD && field_pic)) {
1803 motion_x = s->mv[dir][0][0];
1804 motion_y = s->mv[dir][0][1];
1805 } else { /* if ((s->mv_type == MV_TYPE_FIELD) || (s->mv_type == MV_TYPE_16X8)) */
1806 motion_x = s->mv[dir][i][0];
1807 motion_y = s->mv[dir][i][1];
1808 }
1809
1810 s->current_picture.motion_val[dir][xy][0] = motion_x;
1811 s->current_picture.motion_val[dir][xy][1] = motion_y;
1812 s->current_picture.motion_val[dir][xy + 1][0] = motion_x;
1813 s->current_picture.motion_val[dir][xy + 1][1] = motion_y;
1814 s->current_picture.ref_index [dir][b8_xy] =
1815 s->current_picture.ref_index [dir][b8_xy + 1] = s->field_select[dir][i];
1816 av_assert2(s->field_select[dir][i] == 0 ||
1817 s->field_select[dir][i] == 1);
1818 }
1819 xy += wrap;
1820 b8_xy += 2;
1821 }
1822 }
1823
1824 2124174 s->dest[0] += 16 >> lowres;
1825 2124174 s->dest[1] +=(16 >> lowres) >> s->chroma_x_shift;
1826 2124174 s->dest[2] +=(16 >> lowres) >> s->chroma_x_shift;
1827
1828 2124174 ff_mpv_reconstruct_mb(s, s->block);
1829
1830
2/2
✓ Branch 0 taken 67944 times.
✓ Branch 1 taken 2056230 times.
2124174 if (++s->mb_x >= s->mb_width) {
1831 67944 const int mb_size = 16 >> s->avctx->lowres;
1832 int left;
1833
1834 67944 ff_mpeg_draw_horiz_band(s, mb_size * (s->mb_y >> field_pic), mb_size);
1835 67944 ff_mpv_report_decode_progress(s);
1836
1837 67944 s->mb_x = 0;
1838 67944 s->mb_y += 1 << field_pic;
1839
1840
2/2
✓ Branch 0 taken 4076 times.
✓ Branch 1 taken 63868 times.
67944 if (s->mb_y >= s->mb_height) {
1841 4076 int left = get_bits_left(&s->gb);
1842 8413 int is_d10 = s->chroma_format == 2 &&
1843
2/2
✓ Branch 0 taken 49 times.
✓ Branch 1 taken 212 times.
261 s->pict_type == AV_PICTURE_TYPE_I &&
1844
2/4
✓ Branch 0 taken 49 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 49 times.
✗ Branch 3 not taken.
49 avctx->profile == 0 && avctx->level == 5 &&
1845
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 24 times.
49 s->intra_dc_precision == 2 &&
1846
4/6
✓ Branch 0 taken 261 times.
✓ Branch 1 taken 3815 times.
✓ Branch 2 taken 25 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 25 times.
✗ Branch 5 not taken.
4362 s->q_scale_type == 1 && s->alternate_scan == 0 &&
1847
1/2
✓ Branch 0 taken 25 times.
✗ Branch 1 not taken.
25 s->progressive_frame == 0
1848 /* vbv_delay == 0xBBB || 0xE10 */;
1849
1850
4/4
✓ Branch 0 taken 1069 times.
✓ Branch 1 taken 3007 times.
✓ Branch 2 taken 1044 times.
✓ Branch 3 taken 25 times.
4076 if (left >= 32 && !is_d10) {
1851 1044 GetBitContext gb = s->gb;
1852 1044 align_get_bits(&gb);
1853
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1044 times.
1044 if (show_bits(&gb, 24) == 0x060E2B) {
1854 av_log(avctx, AV_LOG_DEBUG, "Invalid MXF data found in video stream\n");
1855 is_d10 = 1;
1856 }
1857
3/4
✓ Branch 0 taken 1043 times.
✓ Branch 1 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 1043 times.
1044 if (left > 32 && show_bits_long(&gb, 32) == 0x201) {
1858 av_log(avctx, AV_LOG_DEBUG, "skipping m704 alpha (unsupported)\n");
1859 goto eos;
1860 }
1861 }
1862
1863
3/4
✓ Branch 0 taken 4076 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 3783 times.
✓ Branch 3 taken 293 times.
4076 if (left < 0 ||
1864
1/4
✗ Branch 1 not taken.
✓ Branch 2 taken 3783 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
3783 (left && show_bits(&s->gb, FFMIN(left, 23)) && !is_d10) ||
1865
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4076 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
4076 ((avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_AGGRESSIVE)) && left > 8)) {
1866 av_log(avctx, AV_LOG_ERROR, "end mismatch left=%d %0X at %d %d\n",
1867 left, left>0 ? show_bits(&s->gb, FFMIN(left, 23)) : 0, s->mb_x, s->mb_y);
1868 return AVERROR_INVALIDDATA;
1869 } else
1870 4076 goto eos;
1871 }
1872 // There are some files out there which are missing the last slice
1873 // in cases where the slice is completely outside the visible
1874 // area, we detect this here instead of running into the end expecting
1875 // more data
1876 63868 left = get_bits_left(&s->gb);
1877
2/2
✓ Branch 0 taken 200 times.
✓ Branch 1 taken 63668 times.
63868 if (s->mb_y >= ((s->height + 15) >> 4) &&
1878
2/4
✓ Branch 0 taken 200 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 200 times.
200 !s->progressive_sequence &&
1879 left <= 25 &&
1880 left >= 0 &&
1881 s->mb_skip_run == -1 &&
1882 (!left || show_bits(&s->gb, left) == 0))
1883 goto eos;
1884
1885 63868 ff_init_block_index(s);
1886 }
1887
1888 /* skip mb handling */
1889
2/2
✓ Branch 0 taken 1963931 times.
✓ Branch 1 taken 156167 times.
2120098 if (s->mb_skip_run == -1) {
1890 /* read increment again */
1891 1963931 s->mb_skip_run = 0;
1892 1229 for (;;) {
1893 1965160 int code = get_vlc2(&s->gb, ff_mbincr_vlc.table,
1894 MBINCR_VLC_BITS, 2);
1895
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1965160 times.
1965160 if (code < 0) {
1896 av_log(s->avctx, AV_LOG_ERROR, "mb incr damaged\n");
1897 return AVERROR_INVALIDDATA;
1898 }
1899
2/2
✓ Branch 0 taken 110014 times.
✓ Branch 1 taken 1855146 times.
1965160 if (code >= 33) {
1900
2/2
✓ Branch 0 taken 1229 times.
✓ Branch 1 taken 108785 times.
110014 if (code == 33) {
1901 1229 s->mb_skip_run += 33;
1902
1/2
✓ Branch 0 taken 108785 times.
✗ Branch 1 not taken.
108785 } else if (code == 35) {
1903
2/4
✓ Branch 0 taken 108785 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 108785 times.
108785 if (s->mb_skip_run != 0 || show_bits(&s->gb, 15) != 0) {
1904 av_log(s->avctx, AV_LOG_ERROR, "slice mismatch\n");
1905 return AVERROR_INVALIDDATA;
1906 }
1907 108785 goto eos; /* end of slice */
1908 }
1909 /* otherwise, stuffing, nothing to do */
1910 } else {
1911 1855146 s->mb_skip_run += code;
1912 1855146 break;
1913 }
1914 }
1915
2/2
✓ Branch 0 taken 57069 times.
✓ Branch 1 taken 1798077 times.
1855146 if (s->mb_skip_run) {
1916 int i;
1917
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 57069 times.
57069 if (s->pict_type == AV_PICTURE_TYPE_I) {
1918 av_log(s->avctx, AV_LOG_ERROR,
1919 "skipped MB in I-frame at %d %d\n", s->mb_x, s->mb_y);
1920 return AVERROR_INVALIDDATA;
1921 }
1922
1923 /* skip mb */
1924 57069 s->mb_intra = 0;
1925
2/2
✓ Branch 0 taken 684828 times.
✓ Branch 1 taken 57069 times.
741897 for (i = 0; i < 12; i++)
1926 684828 s->block_last_index[i] = -1;
1927
2/2
✓ Branch 0 taken 39925 times.
✓ Branch 1 taken 17144 times.
57069 if (s->picture_structure == PICT_FRAME)
1928 39925 s->mv_type = MV_TYPE_16X16;
1929 else
1930 17144 s->mv_type = MV_TYPE_FIELD;
1931
2/2
✓ Branch 0 taken 20233 times.
✓ Branch 1 taken 36836 times.
57069 if (s->pict_type == AV_PICTURE_TYPE_P) {
1932 /* if P type, zero motion vector is implied */
1933 20233 s->mv_dir = MV_DIR_FORWARD;
1934 20233 s->mv[0][0][0] = s->mv[0][0][1] = 0;
1935 20233 s->last_mv[0][0][0] = s->last_mv[0][0][1] = 0;
1936 20233 s->last_mv[0][1][0] = s->last_mv[0][1][1] = 0;
1937 20233 s->field_select[0][0] = (s->picture_structure - 1) & 1;
1938 } else {
1939 /* if B type, reuse previous vectors and directions */
1940 36836 s->mv[0][0][0] = s->last_mv[0][0][0];
1941 36836 s->mv[0][0][1] = s->last_mv[0][0][1];
1942 36836 s->mv[1][0][0] = s->last_mv[1][0][0];
1943 36836 s->mv[1][0][1] = s->last_mv[1][0][1];
1944 36836 s->field_select[0][0] = (s->picture_structure - 1) & 1;
1945 36836 s->field_select[1][0] = (s->picture_structure - 1) & 1;
1946 }
1947 }
1948 }
1949 }
1950 112861 eos: // end of slice
1951
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 112861 times.
112861 if (get_bits_left(&s->gb) < 0) {
1952 av_log(s, AV_LOG_ERROR, "overread %d\n", -get_bits_left(&s->gb));
1953 return AVERROR_INVALIDDATA;
1954 }
1955 112861 *buf += (get_bits_count(&s->gb) - 1) / 8;
1956 ff_dlog(s, "Slice start:%d %d end:%d %d\n", s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y);
1957 112861 return 0;
1958 }
1959
1960 5463 static int slice_decode_thread(AVCodecContext *c, void *arg)
1961 {
1962 5463 MpegEncContext *s = *(void **) arg;
1963 5463 const uint8_t *buf = s->gb.buffer;
1964 5463 int mb_y = s->start_mb_y;
1965 5463 const int field_pic = s->picture_structure != PICT_FRAME;
1966
1967 5463 s->er.error_count = (3 * (s->end_mb_y - s->start_mb_y) * s->mb_width) >> field_pic;
1968
1969 12747 for (;;) {
1970 uint32_t start_code;
1971 int ret;
1972
1973 18210 ret = mpeg_decode_slice(s, mb_y, &buf, s->gb.buffer_end - buf);
1974 18210 emms_c();
1975 ff_dlog(c, "ret:%d resync:%d/%d mb:%d/%d ts:%d/%d ec:%d\n",
1976 ret, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y,
1977 s->start_mb_y, s->end_mb_y, s->er.error_count);
1978
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18210 times.
18210 if (ret < 0) {
1979 if (c->err_recognition & AV_EF_EXPLODE)
1980 5463 return ret;
1981 if (s->resync_mb_x >= 0 && s->resync_mb_y >= 0)
1982 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y,
1983 s->mb_x, s->mb_y,
1984 ER_AC_ERROR | ER_DC_ERROR | ER_MV_ERROR);
1985 } else {
1986 18210 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y,
1987 18210 s->mb_x - 1, s->mb_y,
1988 ER_AC_END | ER_DC_END | ER_MV_END);
1989 }
1990
1991
2/2
✓ Branch 0 taken 5463 times.
✓ Branch 1 taken 12747 times.
18210 if (s->mb_y == s->end_mb_y)
1992 5463 return 0;
1993
1994 12747 start_code = -1;
1995 12747 buf = avpriv_find_start_code(buf, s->gb.buffer_end, &start_code);
1996
2/4
✓ Branch 0 taken 12747 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 12747 times.
12747 if (start_code < SLICE_MIN_START_CODE || start_code > SLICE_MAX_START_CODE)
1997 return AVERROR_INVALIDDATA;
1998 12747 mb_y = start_code - SLICE_MIN_START_CODE;
1999
2/4
✓ Branch 0 taken 12747 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 12747 times.
12747 if (s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
2000 mb_y += (*buf&0xE0)<<2;
2001 12747 mb_y <<= field_pic;
2002
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12747 times.
12747 if (s->picture_structure == PICT_BOTTOM_FIELD)
2003 mb_y++;
2004
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12747 times.
12747 if (mb_y >= s->end_mb_y)
2005 return AVERROR_INVALIDDATA;
2006 }
2007 }
2008
2009 /**
2010 * Handle slice ends.
2011 * @return 1 if it seems to be the last slice
2012 */
2013 3795 static int slice_end(AVCodecContext *avctx, AVFrame *pict)
2014 {
2015 3795 Mpeg1Context *s1 = avctx->priv_data;
2016 3795 MpegEncContext *s = &s1->mpeg_enc_ctx;
2017
2018
3/4
✓ Branch 0 taken 3539 times.
✓ Branch 1 taken 256 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3539 times.
3795 if (!s1->mpeg_enc_ctx_allocated || !s->current_picture_ptr)
2019 256 return 0;
2020
2021
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3539 times.
3539 if (s->avctx->hwaccel) {
2022 int ret = s->avctx->hwaccel->end_frame(s->avctx);
2023 if (ret < 0) {
2024 av_log(avctx, AV_LOG_ERROR,
2025 "hardware accelerator failed to decode picture\n");
2026 return ret;
2027 }
2028 }
2029
2030 /* end of slice reached */
2031
3/4
✓ Branch 0 taken 3538 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 3538 times.
✗ Branch 3 not taken.
3539 if (/* s->mb_y << field_pic == s->mb_height && */ !s->first_field && !s1->first_slice) {
2032 /* end of image */
2033
2034 3538 ff_er_frame_end(&s->er);
2035
2036 3538 ff_mpv_frame_end(s);
2037
2038
4/4
✓ Branch 0 taken 2215 times.
✓ Branch 1 taken 1323 times.
✓ Branch 2 taken 368 times.
✓ Branch 3 taken 1847 times.
5229 if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {
2039 1691 int ret = av_frame_ref(pict, s->current_picture_ptr->f);
2040
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1691 times.
1691 if (ret < 0)
2041 return ret;
2042 1691 ff_print_debug_info(s, s->current_picture_ptr, pict);
2043 1691 ff_mpv_export_qp_table(s, pict, s->current_picture_ptr, FF_MPV_QSCALE_TYPE_MPEG2);
2044 } else {
2045 /* latency of 1 frame for I- and P-frames */
2046
2/2
✓ Branch 0 taken 1788 times.
✓ Branch 1 taken 59 times.
1847 if (s->last_picture_ptr) {
2047 1788 int ret = av_frame_ref(pict, s->last_picture_ptr->f);
2048
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1788 times.
1788 if (ret < 0)
2049 return ret;
2050 1788 ff_print_debug_info(s, s->last_picture_ptr, pict);
2051 1788 ff_mpv_export_qp_table(s, pict, s->last_picture_ptr, FF_MPV_QSCALE_TYPE_MPEG2);
2052 }
2053 }
2054
2055 3538 return 1;
2056 } else {
2057 1 return 0;
2058 }
2059 }
2060
2061 746 static int mpeg1_decode_sequence(AVCodecContext *avctx,
2062 const uint8_t *buf, int buf_size)
2063 {
2064 746 Mpeg1Context *s1 = avctx->priv_data;
2065 746 MpegEncContext *s = &s1->mpeg_enc_ctx;
2066 int width, height;
2067 int i, v, j;
2068
2069 746 init_get_bits(&s->gb, buf, buf_size * 8);
2070
2071 746 width = get_bits(&s->gb, 12);
2072 746 height = get_bits(&s->gb, 12);
2073
2/4
✓ Branch 0 taken 746 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 746 times.
746 if (width == 0 || height == 0) {
2074 av_log(avctx, AV_LOG_WARNING,
2075 "Invalid horizontal or vertical size value.\n");
2076 if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
2077 return AVERROR_INVALIDDATA;
2078 }
2079 746 s1->aspect_ratio_info = get_bits(&s->gb, 4);
2080
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 746 times.
746 if (s1->aspect_ratio_info == 0) {
2081 av_log(avctx, AV_LOG_ERROR, "aspect ratio has forbidden 0 value\n");
2082 if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
2083 return AVERROR_INVALIDDATA;
2084 }
2085 746 s1->frame_rate_index = get_bits(&s->gb, 4);
2086
2/4
✓ Branch 0 taken 746 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 746 times.
746 if (s1->frame_rate_index == 0 || s1->frame_rate_index > 13) {
2087 av_log(avctx, AV_LOG_WARNING,
2088 "frame_rate_index %d is invalid\n", s1->frame_rate_index);
2089 s1->frame_rate_index = 1;
2090 }
2091 746 s->bit_rate = get_bits(&s->gb, 18) * 400LL;
2092
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 746 times.
746 if (check_marker(s->avctx, &s->gb, "in sequence header") == 0) {
2093 return AVERROR_INVALIDDATA;
2094 }
2095
2096 746 s1->rc_buffer_size = get_bits(&s->gb, 10) * 1024 * 16;
2097 746 skip_bits(&s->gb, 1);
2098
2099 /* get matrix */
2100
2/2
✓ Branch 1 taken 18 times.
✓ Branch 2 taken 728 times.
746 if (get_bits1(&s->gb)) {
2101 18 load_matrix(s, s->chroma_intra_matrix, s->intra_matrix, 1);
2102 } else {
2103
2/2
✓ Branch 0 taken 46592 times.
✓ Branch 1 taken 728 times.
47320 for (i = 0; i < 64; i++) {
2104 46592 j = s->idsp.idct_permutation[i];
2105 46592 v = ff_mpeg1_default_intra_matrix[i];
2106 46592 s->intra_matrix[j] = v;
2107 46592 s->chroma_intra_matrix[j] = v;
2108 }
2109 }
2110
2/2
✓ Branch 1 taken 67 times.
✓ Branch 2 taken 679 times.
746 if (get_bits1(&s->gb)) {
2111 67 load_matrix(s, s->chroma_inter_matrix, s->inter_matrix, 0);
2112 } else {
2113
2/2
✓ Branch 0 taken 43456 times.
✓ Branch 1 taken 679 times.
44135 for (i = 0; i < 64; i++) {
2114 43456 int j = s->idsp.idct_permutation[i];
2115 43456 v = ff_mpeg1_default_non_intra_matrix[i];
2116 43456 s->inter_matrix[j] = v;
2117 43456 s->chroma_inter_matrix[j] = v;
2118 }
2119 }
2120
2121
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 746 times.
746 if (show_bits(&s->gb, 23) != 0) {
2122 av_log(s->avctx, AV_LOG_ERROR, "sequence header damaged\n");
2123 return AVERROR_INVALIDDATA;
2124 }
2125
2126 746 s->width = width;
2127 746 s->height = height;
2128
2129 /* We set MPEG-2 parameters so that it emulates MPEG-1. */
2130 746 s->progressive_sequence = 1;
2131 746 s->progressive_frame = 1;
2132 746 s->picture_structure = PICT_FRAME;
2133 746 s->first_field = 0;
2134 746 s->frame_pred_frame_dct = 1;
2135 746 s->chroma_format = 1;
2136 746 s->codec_id =
2137 746 s->avctx->codec_id = AV_CODEC_ID_MPEG1VIDEO;
2138 746 s->out_format = FMT_MPEG1;
2139
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 746 times.
746 if (s->avctx->flags & AV_CODEC_FLAG_LOW_DELAY)
2140 s->low_delay = 1;
2141
2142
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 746 times.
746 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
2143 av_log(s->avctx, AV_LOG_DEBUG, "vbv buffer: %d, bitrate:%"PRId64", aspect_ratio_info: %d \n",
2144 s1->rc_buffer_size, s->bit_rate, s1->aspect_ratio_info);
2145
2146 746 return 0;
2147 }
2148
2149 2 static int vcr2_init_sequence(AVCodecContext *avctx)
2150 {
2151 2 Mpeg1Context *s1 = avctx->priv_data;
2152 2 MpegEncContext *s = &s1->mpeg_enc_ctx;
2153 int i, v, ret;
2154
2155 /* start new MPEG-1 context decoding */
2156 2 s->out_format = FMT_MPEG1;
2157
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (s1->mpeg_enc_ctx_allocated) {
2158 ff_mpv_common_end(s);
2159 s1->mpeg_enc_ctx_allocated = 0;
2160 }
2161 2 s->width = avctx->coded_width;
2162 2 s->height = avctx->coded_height;
2163 2 avctx->has_b_frames = 0; // true?
2164 2 s->low_delay = 1;
2165
2166 2 avctx->pix_fmt = mpeg_get_pixelformat(avctx);
2167
2168 2 ff_mpv_idct_init(s);
2169
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 if ((ret = ff_mpv_common_init(s)) < 0)
2170 return ret;
2171 2 s1->mpeg_enc_ctx_allocated = 1;
2172
2173
2/2
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 2 times.
130 for (i = 0; i < 64; i++) {
2174 128 int j = s->idsp.idct_permutation[i];
2175 128 v = ff_mpeg1_default_intra_matrix[i];
2176 128 s->intra_matrix[j] = v;
2177 128 s->chroma_intra_matrix[j] = v;
2178
2179 128 v = ff_mpeg1_default_non_intra_matrix[i];
2180 128 s->inter_matrix[j] = v;
2181 128 s->chroma_inter_matrix[j] = v;
2182 }
2183
2184 2 s->progressive_sequence = 1;
2185 2 s->progressive_frame = 1;
2186 2 s->picture_structure = PICT_FRAME;
2187 2 s->first_field = 0;
2188 2 s->frame_pred_frame_dct = 1;
2189 2 s->chroma_format = 1;
2190
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (s->codec_tag == AV_RL32("BW10")) {
2191 s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG1VIDEO;
2192 } else {
2193 2 s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
2194 }
2195 2 s1->save_width = s->width;
2196 2 s1->save_height = s->height;
2197 2 s1->save_progressive_seq = s->progressive_sequence;
2198 2 return 0;
2199 }
2200
2201 659 static int mpeg_decode_a53_cc(AVCodecContext *avctx,
2202 const uint8_t *p, int buf_size)
2203 {
2204 659 Mpeg1Context *s1 = avctx->priv_data;
2205
2206
1/2
✓ Branch 0 taken 659 times.
✗ Branch 1 not taken.
659 if (buf_size >= 6 &&
2207
5/8
✓ Branch 0 taken 415 times.
✓ Branch 1 taken 244 times.
✓ Branch 2 taken 415 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 415 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 415 times.
✗ Branch 7 not taken.
659 p[0] == 'G' && p[1] == 'A' && p[2] == '9' && p[3] == '4' &&
2208
2/4
✓ Branch 0 taken 415 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 415 times.
✗ Branch 3 not taken.
415 p[4] == 3 && (p[5] & 0x40)) {
2209 /* extract A53 Part 4 CC data */
2210 415 int cc_count = p[5] & 0x1f;
2211
2/4
✓ Branch 0 taken 415 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 415 times.
✗ Branch 3 not taken.
415 if (cc_count > 0 && buf_size >= 7 + cc_count * 3) {
2212
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 409 times.
415 int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
2213 415 const uint64_t new_size = (old_size + cc_count
2214 415 * UINT64_C(3));
2215 int ret;
2216
2217
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 415 times.
415 if (new_size > 3*A53_MAX_CC_COUNT)
2218 return AVERROR(EINVAL);
2219
2220 415 ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
2221
1/2
✓ Branch 0 taken 415 times.
✗ Branch 1 not taken.
415 if (ret >= 0)
2222 415 memcpy(s1->a53_buf_ref->data + old_size, p + 7, cc_count * UINT64_C(3));
2223
2224 415 avctx->properties |= FF_CODEC_PROPERTY_CLOSED_CAPTIONS;
2225 }
2226 415 return 1;
2227
1/2
✓ Branch 0 taken 244 times.
✗ Branch 1 not taken.
244 } else if (buf_size >= 2 &&
2228
3/4
✓ Branch 0 taken 205 times.
✓ Branch 1 taken 39 times.
✓ Branch 2 taken 205 times.
✗ Branch 3 not taken.
244 p[0] == 0x03 && (p[1]&0x7f) == 0x01) {
2229 /* extract SCTE-20 CC data */
2230 GetBitContext gb;
2231 205 int cc_count = 0;
2232 int i, ret;
2233
2234 205 init_get_bits8(&gb, p + 2, buf_size - 2);
2235 205 cc_count = get_bits(&gb, 5);
2236
1/2
✓ Branch 0 taken 205 times.
✗ Branch 1 not taken.
205 if (cc_count > 0) {
2237
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 203 times.
205 int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
2238 205 const uint64_t new_size = (old_size + cc_count
2239 205 * UINT64_C(3));
2240
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 205 times.
205 if (new_size > 3*A53_MAX_CC_COUNT)
2241 return AVERROR(EINVAL);
2242
2243 205 ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
2244
1/2
✓ Branch 0 taken 205 times.
✗ Branch 1 not taken.
205 if (ret >= 0) {
2245 uint8_t field, cc1, cc2;
2246 205 uint8_t *cap = s1->a53_buf_ref->data;
2247
2248 205 memset(s1->a53_buf_ref->data + old_size, 0, cc_count * 3);
2249
3/4
✓ Branch 0 taken 410 times.
✓ Branch 1 taken 205 times.
✓ Branch 3 taken 410 times.
✗ Branch 4 not taken.
615 for (i = 0; i < cc_count && get_bits_left(&gb) >= 26; i++) {
2250 410 skip_bits(&gb, 2); // priority
2251 410 field = get_bits(&gb, 2);
2252 410 skip_bits(&gb, 5); // line_offset
2253 410 cc1 = get_bits(&gb, 8);
2254 410 cc2 = get_bits(&gb, 8);
2255 410 skip_bits(&gb, 1); // marker
2256
2257
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 410 times.
410 if (!field) { // forbidden
2258 cap[0] = cap[1] = cap[2] = 0x00;
2259 } else {
2260 410 field = (field == 2 ? 1 : 0);
2261
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 410 times.
410 if (!s1->mpeg_enc_ctx.top_field_first) field = !field;
2262 410 cap[0] = 0x04 | field;
2263 410 cap[1] = ff_reverse[cc1];
2264 410 cap[2] = ff_reverse[cc2];
2265 }
2266 410 cap += 3;
2267 }
2268 }
2269 205 avctx->properties |= FF_CODEC_PROPERTY_CLOSED_CAPTIONS;
2270 }
2271 205 return 1;
2272
1/2
✓ Branch 0 taken 39 times.
✗ Branch 1 not taken.
39 } else if (buf_size >= 11 &&
2273
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 39 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
39 p[0] == 'C' && p[1] == 'C' && p[2] == 0x01 && p[3] == 0xf8) {
2274 /* extract DVD CC data
2275 *
2276 * uint32_t user_data_start_code 0x000001B2 (big endian)
2277 * uint16_t user_identifier 0x4343 "CC"
2278 * uint8_t user_data_type_code 0x01
2279 * uint8_t caption_block_size 0xF8
2280 * uint8_t
2281 * bit 7 caption_odd_field_first 1=odd field (CC1/CC2) first 0=even field (CC3/CC4) first
2282 * bit 6 caption_filler 0
2283 * bit 5:1 caption_block_count number of caption blocks (pairs of caption words = frames). Most DVDs use 15 per start of GOP.
2284 * bit 0 caption_extra_field_added 1=one additional caption word
2285 *
2286 * struct caption_field_block {
2287 * uint8_t
2288 * bit 7:1 caption_filler 0x7F (all 1s)
2289 * bit 0 caption_field_odd 1=odd field (this is CC1/CC2) 0=even field (this is CC3/CC4)
2290 * uint8_t caption_first_byte
2291 * uint8_t caption_second_byte
2292 * } caption_block[(caption_block_count * 2) + caption_extra_field_added];
2293 *
2294 * Some DVDs encode caption data for both fields with caption_field_odd=1. The only way to decode the fields
2295 * correctly is to start on the field indicated by caption_odd_field_first and count between odd/even fields.
2296 * Don't assume that the first caption word is the odd field. There do exist MPEG files in the wild that start
2297 * on the even field. There also exist DVDs in the wild that encode an odd field count and the
2298 * caption_extra_field_added/caption_odd_field_first bits change per packet to allow that. */
2299 int cc_count = 0;
2300 int i, ret;
2301 // There is a caption count field in the data, but it is often
2302 // incorrect. So count the number of captions present.
2303 for (i = 5; i + 6 <= buf_size && ((p[i] & 0xfe) == 0xfe); i += 6)
2304 cc_count++;
2305 // Transform the DVD format into A53 Part 4 format
2306 if (cc_count > 0) {
2307 int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
2308 const uint64_t new_size = (old_size + cc_count
2309 * UINT64_C(6));
2310 if (new_size > 3*A53_MAX_CC_COUNT)
2311 return AVERROR(EINVAL);
2312
2313 ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
2314 if (ret >= 0) {
2315 uint8_t field1 = !!(p[4] & 0x80);
2316 uint8_t *cap = s1->a53_buf_ref->data;
2317 p += 5;
2318 for (i = 0; i < cc_count; i++) {
2319 cap[0] = (p[0] == 0xff && field1) ? 0xfc : 0xfd;
2320 cap[1] = p[1];
2321 cap[2] = p[2];
2322 cap[3] = (p[3] == 0xff && !field1) ? 0xfc : 0xfd;
2323 cap[4] = p[4];
2324 cap[5] = p[5];
2325 cap += 6;
2326 p += 6;
2327 }
2328 }
2329 avctx->properties |= FF_CODEC_PROPERTY_CLOSED_CAPTIONS;
2330 }
2331 return 1;
2332 }
2333 39 return 0;
2334 }
2335
2336 866 static void mpeg_decode_user_data(AVCodecContext *avctx,
2337 const uint8_t *p, int buf_size)
2338 {
2339 866 Mpeg1Context *s = avctx->priv_data;
2340 866 const uint8_t *buf_end = p + buf_size;
2341 866 Mpeg1Context *s1 = avctx->priv_data;
2342
2343 #if 0
2344 int i;
2345 for(i=0; !(!p[i-2] && !p[i-1] && p[i]==1) && i<buf_size; i++){
2346 av_log(avctx, AV_LOG_ERROR, "%c", p[i]);
2347 }
2348 av_log(avctx, AV_LOG_ERROR, "\n");
2349 #endif
2350
2351
1/2
✓ Branch 0 taken 866 times.
✗ Branch 1 not taken.
866 if (buf_size > 29){
2352 int i;
2353
2/2
✓ Branch 0 taken 17320 times.
✓ Branch 1 taken 866 times.
18186 for(i=0; i<20; i++)
2354
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 17320 times.
17320 if (!memcmp(p+i, "\0TMPGEXS\0", 9)){
2355 s->tmpgexs= 1;
2356 }
2357 }
2358 /* we parse the DTG active format information */
2359
1/2
✓ Branch 0 taken 866 times.
✗ Branch 1 not taken.
866 if (buf_end - p >= 5 &&
2360
5/8
✓ Branch 0 taken 207 times.
✓ Branch 1 taken 659 times.
✓ Branch 2 taken 207 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 207 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 207 times.
✗ Branch 7 not taken.
1073 p[0] == 'D' && p[1] == 'T' && p[2] == 'G' && p[3] == '1') {
2361 207 int flags = p[4];
2362 207 p += 5;
2363
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 207 times.
207 if (flags & 0x80) {
2364 /* skip event id */
2365 p += 2;
2366 }
2367
1/2
✓ Branch 0 taken 207 times.
✗ Branch 1 not taken.
207 if (flags & 0x40) {
2368
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 207 times.
207 if (buf_end - p < 1)
2369 return;
2370 207 s1->has_afd = 1;
2371 207 s1->afd = p[0] & 0x0f;
2372 }
2373
1/2
✓ Branch 0 taken 659 times.
✗ Branch 1 not taken.
659 } else if (buf_end - p >= 6 &&
2374
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 659 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
659 p[0] == 'J' && p[1] == 'P' && p[2] == '3' && p[3] == 'D' &&
2375 p[4] == 0x03) { // S3D_video_format_length
2376 // the 0x7F mask ignores the reserved_bit value
2377 const uint8_t S3D_video_format_type = p[5] & 0x7F;
2378
2379 if (S3D_video_format_type == 0x03 ||
2380 S3D_video_format_type == 0x04 ||
2381 S3D_video_format_type == 0x08 ||
2382 S3D_video_format_type == 0x23) {
2383
2384 s1->has_stereo3d = 1;
2385
2386 switch (S3D_video_format_type) {
2387 case 0x03:
2388 s1->stereo3d.type = AV_STEREO3D_SIDEBYSIDE;
2389 break;
2390 case 0x04:
2391 s1->stereo3d.type = AV_STEREO3D_TOPBOTTOM;
2392 break;
2393 case 0x08:
2394 s1->stereo3d.type = AV_STEREO3D_2D;
2395 break;
2396 case 0x23:
2397 s1->stereo3d.type = AV_STEREO3D_SIDEBYSIDE_QUINCUNX;
2398 break;
2399 }
2400 }
2401
2/2
✓ Branch 1 taken 620 times.
✓ Branch 2 taken 39 times.
659 } else if (mpeg_decode_a53_cc(avctx, p, buf_size)) {
2402 620 return;
2403 }
2404 }
2405
2406 517 static void mpeg_decode_gop(AVCodecContext *avctx,
2407 const uint8_t *buf, int buf_size)
2408 {
2409 517 Mpeg1Context *s1 = avctx->priv_data;
2410 517 MpegEncContext *s = &s1->mpeg_enc_ctx;
2411 int broken_link;
2412 int64_t tc;
2413
2414 517 init_get_bits(&s->gb, buf, buf_size * 8);
2415
2416 517 tc = s1->timecode_frame_start = get_bits(&s->gb, 25);
2417
2418 517 s1->closed_gop = get_bits1(&s->gb);
2419 /* broken_link indicates that after editing the
2420 * reference frames of the first B-Frames after GOP I-Frame
2421 * are missing (open gop) */
2422 517 broken_link = get_bits1(&s->gb);
2423
2424
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 517 times.
517 if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
2425 char tcbuf[AV_TIMECODE_STR_SIZE];
2426 av_timecode_make_mpeg_tc_string(tcbuf, tc);
2427 av_log(s->avctx, AV_LOG_DEBUG,
2428 "GOP (%s) closed_gop=%d broken_link=%d\n",
2429 tcbuf, s1->closed_gop, broken_link);
2430 }
2431 517 }
2432
2433 4172 static int decode_chunks(AVCodecContext *avctx, AVFrame *picture,
2434 int *got_output, const uint8_t *buf, int buf_size)
2435 {
2436 4172 Mpeg1Context *s = avctx->priv_data;
2437 4172 MpegEncContext *s2 = &s->mpeg_enc_ctx;
2438 4172 const uint8_t *buf_ptr = buf;
2439 4172 const uint8_t *buf_end = buf + buf_size;
2440 int ret, input_size;
2441 4172 int last_code = 0, skip_frame = 0;
2442 4172 int picture_start_code_seen = 0;
2443
2444 198372 for (;;) {
2445 /* find next start code */
2446 202544 uint32_t start_code = -1;
2447 202544 buf_ptr = avpriv_find_start_code(buf_ptr, buf_end, &start_code);
2448
2/2
✓ Branch 0 taken 4054 times.
✓ Branch 1 taken 198490 times.
202544 if (start_code > 0x1ff) {
2449
2/2
✓ Branch 0 taken 3795 times.
✓ Branch 1 taken 259 times.
4054 if (!skip_frame) {
2450 3795 if (HAVE_THREADS &&
2451
2/2
✓ Branch 0 taken 611 times.
✓ Branch 1 taken 3184 times.
3795 (avctx->active_thread_type & FF_THREAD_SLICE) &&
2452
1/2
✓ Branch 0 taken 611 times.
✗ Branch 1 not taken.
611 !avctx->hwaccel) {
2453 int i;
2454
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 611 times.
611 av_assert0(avctx->thread_count > 1);
2455
2456 611 avctx->execute(avctx, slice_decode_thread,
2457 611 &s2->thread_context[0], NULL,
2458 s->slice_count, sizeof(void *));
2459
2/2
✓ Branch 0 taken 5463 times.
✓ Branch 1 taken 611 times.
6074 for (i = 0; i < s->slice_count; i++)
2460 5463 s2->er.error_count += s2->thread_context[i]->er.error_count;
2461 }
2462
2463 3795 ret = slice_end(avctx, picture);
2464
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3795 times.
3795 if (ret < 0)
2465 4172 return ret;
2466
2/2
✓ Branch 0 taken 3538 times.
✓ Branch 1 taken 257 times.
3795 else if (ret) {
2467 // FIXME: merge with the stuff in mpeg_decode_slice
2468
5/6
✓ Branch 0 taken 68 times.
✓ Branch 1 taken 3470 times.
✓ Branch 2 taken 59 times.
✓ Branch 3 taken 9 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 59 times.
3538 if (s2->last_picture_ptr || s2->low_delay || s2->pict_type == AV_PICTURE_TYPE_B)
2469 3479 *got_output = 1;
2470 }
2471 }
2472 4054 s2->pict_type = 0;
2473
2474
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4054 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
4054 if (avctx->err_recognition & AV_EF_EXPLODE && s2->er.error_count)
2475 return AVERROR_INVALIDDATA;
2476
2477 #if FF_API_FLAG_TRUNCATED
2478 4054 return FFMAX(0, buf_ptr - buf - s2->parse_context.last_index);
2479 #else
2480 return FFMAX(0, buf_ptr - buf);
2481 #endif
2482 }
2483
2484 198490 input_size = buf_end - buf_ptr;
2485
2486
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 198490 times.
198490 if (avctx->debug & FF_DEBUG_STARTCODE)
2487 av_log(avctx, AV_LOG_DEBUG, "%3"PRIX32" at %"PTRDIFF_SPECIFIER" left %d\n",
2488 start_code, buf_ptr - buf, input_size);
2489
2490 /* prepare data for next start code */
2491
6/6
✓ Branch 0 taken 746 times.
✓ Branch 1 taken 4532 times.
✓ Branch 2 taken 5196 times.
✓ Branch 3 taken 866 times.
✓ Branch 4 taken 517 times.
✓ Branch 5 taken 186633 times.
198490 switch (start_code) {
2492 746 case SEQ_START_CODE:
2493
1/2
✓ Branch 0 taken 746 times.
✗ Branch 1 not taken.
746 if (last_code == 0) {
2494 746 mpeg1_decode_sequence(avctx, buf_ptr, input_size);
2495
2/2
✓ Branch 0 taken 508 times.
✓ Branch 1 taken 238 times.
746 if (buf != avctx->extradata)
2496 508 s->sync = 1;
2497 } else {
2498 av_log(avctx, AV_LOG_ERROR,
2499 "ignoring SEQ_START_CODE after %X\n", last_code);
2500 if (avctx->err_recognition & AV_EF_EXPLODE)
2501 return AVERROR_INVALIDDATA;
2502 }
2503 746 break;
2504
2505 4532 case PICTURE_START_CODE:
2506
3/4
✓ Branch 0 taken 616 times.
✓ Branch 1 taken 3916 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 616 times.
4532 if (picture_start_code_seen && s2->picture_structure == PICT_FRAME) {
2507 /* If it's a frame picture, there can't be more than one picture header.
2508 Yet, it does happen and we need to handle it. */
2509 av_log(avctx, AV_LOG_WARNING, "ignoring extra picture following a frame-picture\n");
2510 break;
2511 }
2512 4532 picture_start_code_seen = 1;
2513
2514
3/4
✓ Branch 0 taken 4414 times.
✓ Branch 1 taken 118 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 4414 times.
4532 if (s2->width <= 0 || s2->height <= 0) {
2515 118 av_log(avctx, AV_LOG_ERROR, "Invalid frame dimensions %dx%d.\n",
2516 s2->width, s2->height);
2517 118 return AVERROR_INVALIDDATA;
2518 }
2519
2520
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4414 times.
4414 if (s->tmpgexs){
2521 s2->intra_dc_precision= 3;
2522 s2->intra_matrix[0]= 1;
2523 }
2524
2/2
✓ Branch 0 taken 615 times.
✓ Branch 1 taken 3799 times.
4414 if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE) &&
2525
2/4
✓ Branch 0 taken 615 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 615 times.
615 !avctx->hwaccel && s->slice_count) {
2526 int i;
2527
2528 avctx->execute(avctx, slice_decode_thread,
2529 s2->thread_context, NULL,
2530 s->slice_count, sizeof(void *));
2531 for (i = 0; i < s->slice_count; i++)
2532 s2->er.error_count += s2->thread_context[i]->er.error_count;
2533 s->slice_count = 0;
2534 }
2535
3/4
✓ Branch 0 taken 616 times.
✓ Branch 1 taken 3798 times.
✓ Branch 2 taken 616 times.
✗ Branch 3 not taken.
4414 if (last_code == 0 || last_code == SLICE_MIN_START_CODE) {
2536 4414 ret = mpeg_decode_postinit(avctx);
2537
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4414 times.
4414 if (ret < 0) {
2538 av_log(avctx, AV_LOG_ERROR,
2539 "mpeg_decode_postinit() failure\n");
2540 return ret;
2541 }
2542
2543 /* We have a complete image: we try to decompress it. */
2544
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4414 times.
4414 if (mpeg1_decode_picture(avctx, buf_ptr, input_size) < 0)
2545 s2->pict_type = 0;
2546 4414 s->first_slice = 1;
2547 4414 last_code = PICTURE_START_CODE;
2548 } else {
2549 av_log(avctx, AV_LOG_ERROR,
2550 "ignoring pic after %X\n", last_code);
2551 if (avctx->err_recognition & AV_EF_EXPLODE)
2552 return AVERROR_INVALIDDATA;
2553 }
2554 4414 break;
2555 5196 case EXT_START_CODE:
2556 5196 init_get_bits(&s2->gb, buf_ptr, input_size * 8);
2557
2558 5196 switch (get_bits(&s2->gb, 4)) {
2559 654 case 0x1:
2560
1/2
✓ Branch 0 taken 654 times.
✗ Branch 1 not taken.
654 if (last_code == 0) {
2561 654 mpeg_decode_sequence_extension(s);
2562 } else {
2563 av_log(avctx, AV_LOG_ERROR,
2564 "ignoring seq ext after %X\n", last_code);
2565 if (avctx->err_recognition & AV_EF_EXPLODE)
2566 return AVERROR_INVALIDDATA;
2567 }
2568 654 break;
2569 76 case 0x2:
2570 76 mpeg_decode_sequence_display_extension(s);
2571 76 break;
2572 531 case 0x3:
2573 531 mpeg_decode_quant_matrix_extension(s2);
2574 531 break;
2575 15 case 0x7:
2576 15 mpeg_decode_picture_display_extension(s);
2577 15 break;
2578 3920 case 0x8:
2579
1/2
✓ Branch 0 taken 3920 times.
✗ Branch 1 not taken.
3920 if (last_code == PICTURE_START_CODE) {
2580 3920 int ret = mpeg_decode_picture_coding_extension(s);
2581
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3920 times.
3920 if (ret < 0)
2582 return ret;
2583 } else {
2584 av_log(avctx, AV_LOG_ERROR,
2585 "ignoring pic cod ext after %X\n", last_code);
2586 if (avctx->err_recognition & AV_EF_EXPLODE)
2587 return AVERROR_INVALIDDATA;
2588 }
2589 3920 break;
2590 }
2591 5196 break;
2592 866 case USER_START_CODE:
2593 866 mpeg_decode_user_data(avctx, buf_ptr, input_size);
2594 866 break;
2595 517 case GOP_START_CODE:
2596
1/2
✓ Branch 0 taken 517 times.
✗ Branch 1 not taken.
517 if (last_code == 0) {
2597 517 s2->first_field = 0;
2598 517 mpeg_decode_gop(avctx, buf_ptr, input_size);
2599 517 s->sync = 1;
2600 } else {
2601 av_log(avctx, AV_LOG_ERROR,
2602 "ignoring GOP_START_CODE after %X\n", last_code);
2603 if (avctx->err_recognition & AV_EF_EXPLODE)
2604 return AVERROR_INVALIDDATA;
2605 }
2606 517 break;
2607 186633 default:
2608
2/2
✓ Branch 0 taken 185992 times.
✓ Branch 1 taken 641 times.
186633 if (start_code >= SLICE_MIN_START_CODE &&
2609
4/4
✓ Branch 0 taken 185988 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 4414 times.
✓ Branch 3 taken 181574 times.
185992 start_code <= SLICE_MAX_START_CODE && last_code == PICTURE_START_CODE) {
2610
3/4
✓ Branch 0 taken 1650 times.
✓ Branch 1 taken 2764 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1650 times.
4414 if (s2->progressive_sequence && !s2->progressive_frame) {
2611 s2->progressive_frame = 1;
2612 av_log(s2->avctx, AV_LOG_ERROR,
2613 "interlaced frame in progressive sequence, ignoring\n");
2614 }
2615
2616
1/2
✓ Branch 0 taken 4414 times.
✗ Branch 1 not taken.
4414 if (s2->picture_structure == 0 ||
2617
3/4
✓ Branch 0 taken 1654 times.
✓ Branch 1 taken 2760 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1654 times.
4414 (s2->progressive_frame && s2->picture_structure != PICT_FRAME)) {
2618 av_log(s2->avctx, AV_LOG_ERROR,
2619 "picture_structure %d invalid, ignoring\n",
2620 s2->picture_structure);
2621 s2->picture_structure = PICT_FRAME;
2622 }
2623
2624
3/4
✓ Branch 0 taken 1650 times.
✓ Branch 1 taken 2764 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1650 times.
4414 if (s2->progressive_sequence && !s2->frame_pred_frame_dct)
2625 av_log(s2->avctx, AV_LOG_WARNING, "invalid frame_pred_frame_dct\n");
2626
2627
2/2
✓ Branch 0 taken 3180 times.
✓ Branch 1 taken 1234 times.
4414 if (s2->picture_structure == PICT_FRAME) {
2628 3180 s2->first_field = 0;
2629 3180 s2->v_edge_pos = 16 * s2->mb_height;
2630 } else {
2631 1234 s2->first_field ^= 1;
2632 1234 s2->v_edge_pos = 8 * s2->mb_height;
2633 1234 memset(s2->mbskip_table, 0, s2->mb_stride * s2->mb_height);
2634 }
2635 }
2636
2/2
✓ Branch 0 taken 185992 times.
✓ Branch 1 taken 641 times.
186633 if (start_code >= SLICE_MIN_START_CODE &&
2637
4/4
✓ Branch 0 taken 185988 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 185868 times.
✓ Branch 3 taken 120 times.
185992 start_code <= SLICE_MAX_START_CODE && last_code != 0) {
2638 185868 const int field_pic = s2->picture_structure != PICT_FRAME;
2639 185868 int mb_y = start_code - SLICE_MIN_START_CODE;
2640 185868 last_code = SLICE_MIN_START_CODE;
2641
3/4
✓ Branch 0 taken 178280 times.
✓ Branch 1 taken 7588 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 178280 times.
185868 if (s2->codec_id != AV_CODEC_ID_MPEG1VIDEO && s2->mb_height > 2800/16)
2642 mb_y += (*buf_ptr&0xE0)<<2;
2643
2644 185868 mb_y <<= field_pic;
2645
2/2
✓ Branch 0 taken 8919 times.
✓ Branch 1 taken 176949 times.
185868 if (s2->picture_structure == PICT_BOTTOM_FIELD)
2646 8919 mb_y++;
2647
2648
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 185868 times.
185868 if (buf_end - buf_ptr < 2) {
2649 av_log(s2->avctx, AV_LOG_ERROR, "slice too small\n");
2650 return AVERROR_INVALIDDATA;
2651 }
2652
2653
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 185868 times.
185868 if (mb_y >= s2->mb_height) {
2654 av_log(s2->avctx, AV_LOG_ERROR,
2655 "slice below image (%d >= %d)\n", mb_y, s2->mb_height);
2656 return AVERROR_INVALIDDATA;
2657 }
2658
2659
2/2
✓ Branch 0 taken 76578 times.
✓ Branch 1 taken 109290 times.
185868 if (!s2->last_picture_ptr) {
2660 /* Skip B-frames if we do not have reference frames and
2661 * GOP is not closed. */
2662
2/2
✓ Branch 0 taken 672 times.
✓ Branch 1 taken 75906 times.
76578 if (s2->pict_type == AV_PICTURE_TYPE_B) {
2663
1/2
✓ Branch 0 taken 672 times.
✗ Branch 1 not taken.
672 if (!s->closed_gop) {
2664 672 skip_frame = 1;
2665 672 av_log(s2->avctx, AV_LOG_DEBUG,
2666 "Skipping B slice due to open GOP\n");
2667 672 break;
2668 }
2669 }
2670 }
2671
3/4
✓ Branch 0 taken 65676 times.
✓ Branch 1 taken 119520 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 65676 times.
185196 if (s2->pict_type == AV_PICTURE_TYPE_I || (s2->avctx->flags2 & AV_CODEC_FLAG2_SHOW_ALL))
2672 119520 s->sync = 1;
2673
2/2
✓ Branch 0 taken 72399 times.
✓ Branch 1 taken 112797 times.
185196 if (!s2->next_picture_ptr) {
2674 /* Skip P-frames if we do not have a reference frame or
2675 * we have an invalid header. */
2676
4/4
✓ Branch 0 taken 3011 times.
✓ Branch 1 taken 69388 times.
✓ Branch 2 taken 2412 times.
✓ Branch 3 taken 599 times.
72399 if (s2->pict_type == AV_PICTURE_TYPE_P && !s->sync) {
2677 2412 skip_frame = 1;
2678 2412 av_log(s2->avctx, AV_LOG_DEBUG,
2679 "Skipping P slice due to !sync\n");
2680 2412 break;
2681 }
2682 }
2683
2/2
✓ Branch 0 taken 69919 times.
✓ Branch 1 taken 112865 times.
182784 if ((avctx->skip_frame >= AVDISCARD_NONREF &&
2684
1/2
✓ Branch 0 taken 69919 times.
✗ Branch 1 not taken.
69919 s2->pict_type == AV_PICTURE_TYPE_B) ||
2685
2/2
✓ Branch 0 taken 69919 times.
✓ Branch 1 taken 112865 times.
182784 (avctx->skip_frame >= AVDISCARD_NONKEY &&
2686
2/2
✓ Branch 0 taken 69320 times.
✓ Branch 1 taken 599 times.
69919 s2->pict_type != AV_PICTURE_TYPE_I) ||
2687
2/2
✓ Branch 0 taken 69320 times.
✓ Branch 1 taken 112865 times.
182185 avctx->skip_frame >= AVDISCARD_ALL) {
2688 69919 skip_frame = 1;
2689 69919 break;
2690 }
2691
2692
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 112865 times.
112865 if (!s->mpeg_enc_ctx_allocated)
2693 break;
2694
2695
2/2
✓ Branch 0 taken 105592 times.
✓ Branch 1 taken 7273 times.
112865 if (s2->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
2696
1/2
✓ Branch 0 taken 105592 times.
✗ Branch 1 not taken.
105592 if (mb_y < avctx->skip_top ||
2697
1/2
✓ Branch 0 taken 105592 times.
✗ Branch 1 not taken.
105592 mb_y >= s2->mb_height - avctx->skip_bottom)
2698 break;
2699 }
2700
2701
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 112865 times.
112865 if (!s2->pict_type) {
2702 av_log(avctx, AV_LOG_ERROR, "Missing picture start code\n");
2703 if (avctx->err_recognition & AV_EF_EXPLODE)
2704 return AVERROR_INVALIDDATA;
2705 break;
2706 }
2707
2708
2/2
✓ Branch 0 taken 4080 times.
✓ Branch 1 taken 108785 times.
112865 if (s->first_slice) {
2709 4080 skip_frame = 0;
2710 4080 s->first_slice = 0;
2711
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4080 times.
4080 if ((ret = mpeg_field_start(s2, buf, buf_size)) < 0)
2712 return ret;
2713 }
2714
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 112865 times.
112865 if (!s2->current_picture_ptr) {
2715 av_log(avctx, AV_LOG_ERROR,
2716 "current_picture not initialized\n");
2717 return AVERROR_INVALIDDATA;
2718 }
2719
2720 112865 if (HAVE_THREADS &&
2721
2/2
✓ Branch 0 taken 18210 times.
✓ Branch 1 taken 94655 times.
112865 (avctx->active_thread_type & FF_THREAD_SLICE) &&
2722
1/2
✓ Branch 0 taken 18210 times.
✗ Branch 1 not taken.
18210 !avctx->hwaccel) {
2723 18210 int threshold = (s2->mb_height * s->slice_count +
2724 18210 s2->slice_context_count / 2) /
2725 18210 s2->slice_context_count;
2726
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18210 times.
18210 av_assert0(avctx->thread_count > 1);
2727
2/2
✓ Branch 0 taken 5463 times.
✓ Branch 1 taken 12747 times.
18210 if (threshold <= mb_y) {
2728 5463 MpegEncContext *thread_context = s2->thread_context[s->slice_count];
2729
2730 5463 thread_context->start_mb_y = mb_y;
2731 5463 thread_context->end_mb_y = s2->mb_height;
2732
2/2
✓ Branch 0 taken 4856 times.
✓ Branch 1 taken 607 times.
5463 if (s->slice_count) {
2733 4856 s2->thread_context[s->slice_count - 1]->end_mb_y = mb_y;
2734 4856 ret = ff_update_duplicate_context(thread_context, s2);
2735
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4856 times.
4856 if (ret < 0)
2736 return ret;
2737 }
2738 5463 init_get_bits(&thread_context->gb, buf_ptr, input_size * 8);
2739 5463 s->slice_count++;
2740 }
2741 18210 buf_ptr += 2; // FIXME add minimum number of bytes per slice
2742 } else {
2743 94655 ret = mpeg_decode_slice(s2, mb_y, &buf_ptr, input_size);
2744 94655 emms_c();
2745
2746
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 94651 times.
94655 if (ret < 0) {
2747
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
4 if (avctx->err_recognition & AV_EF_EXPLODE)
2748 return ret;
2749
2/4
✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
✗ Branch 3 not taken.
4 if (s2->resync_mb_x >= 0 && s2->resync_mb_y >= 0)
2750 4 ff_er_add_slice(&s2->er, s2->resync_mb_x,
2751 s2->resync_mb_y, s2->mb_x, s2->mb_y,
2752 ER_AC_ERROR | ER_DC_ERROR | ER_MV_ERROR);
2753 } else {
2754 94651 ff_er_add_slice(&s2->er, s2->resync_mb_x,
2755 94651 s2->resync_mb_y, s2->mb_x - 1, s2->mb_y,
2756 ER_AC_END | ER_DC_END | ER_MV_END);
2757 }
2758 }
2759 }
2760 113630 break;
2761 }
2762 }
2763 }
2764
2765 4065 static int mpeg_decode_frame(AVCodecContext *avctx, AVFrame *picture,
2766 int *got_output, AVPacket *avpkt)
2767 {
2768 4065 const uint8_t *buf = avpkt->data;
2769 int ret;
2770 4065 int buf_size = avpkt->size;
2771 4065 Mpeg1Context *s = avctx->priv_data;
2772 4065 MpegEncContext *s2 = &s->mpeg_enc_ctx;
2773
2774
3/6
✓ Branch 0 taken 3934 times.
✓ Branch 1 taken 131 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3934 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
4065 if (buf_size == 0 || (buf_size == 4 && AV_RB32(buf) == SEQ_END_CODE)) {
2775 /* special case for last picture */
2776
4/4
✓ Branch 0 taken 122 times.
✓ Branch 1 taken 9 times.
✓ Branch 2 taken 59 times.
✓ Branch 3 taken 63 times.
131 if (s2->low_delay == 0 && s2->next_picture_ptr) {
2777 59 int ret = av_frame_ref(picture, s2->next_picture_ptr->f);
2778
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 59 times.
59 if (ret < 0)
2779 return ret;
2780
2781 59 s2->next_picture_ptr = NULL;
2782
2783 59 *got_output = 1;
2784 }
2785 131 return buf_size;
2786 }
2787
2788 #if FF_API_FLAG_TRUNCATED
2789
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3934 times.
3934 if (s2->avctx->flags & AV_CODEC_FLAG_TRUNCATED) {
2790 int next = ff_mpeg1_find_frame_end(&s2->parse_context, buf,
2791 buf_size, NULL);
2792
2793 if (ff_combine_frame(&s2->parse_context, next,
2794 (const uint8_t **) &buf, &buf_size) < 0)
2795 return buf_size;
2796 }
2797 #endif
2798
2799 3934 s2->codec_tag = ff_toupper4(avctx->codec_tag);
2800
4/4
✓ Branch 0 taken 376 times.
✓ Branch 1 taken 3558 times.
✓ Branch 2 taken 374 times.
✓ Branch 3 taken 2 times.
3934 if (s->mpeg_enc_ctx_allocated == 0 && ( s2->codec_tag == AV_RL32("VCR2")
2801
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 374 times.
374 || s2->codec_tag == AV_RL32("BW10")
2802 ))
2803 2 vcr2_init_sequence(avctx);
2804
2805 3934 s->slice_count = 0;
2806
2807
4/4
✓ Branch 0 taken 3648 times.
✓ Branch 1 taken 286 times.
✓ Branch 2 taken 238 times.
✓ Branch 3 taken 3410 times.
3934 if (avctx->extradata && !s->extradata_decoded) {
2808 238 ret = decode_chunks(avctx, picture, got_output,
2809 238 avctx->extradata, avctx->extradata_size);
2810
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 238 times.
238 if (*got_output) {
2811 av_log(avctx, AV_LOG_ERROR, "picture in extradata\n");
2812 av_frame_unref(picture);
2813 *got_output = 0;
2814 }
2815 238 s->extradata_decoded = 1;
2816
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 238 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
238 if (ret < 0 && (avctx->err_recognition & AV_EF_EXPLODE)) {
2817 s2->current_picture_ptr = NULL;
2818 return ret;
2819 }
2820 }
2821
2822 3934 ret = decode_chunks(avctx, picture, got_output, buf, buf_size);
2823
4/4
✓ Branch 0 taken 3816 times.
✓ Branch 1 taken 118 times.
✓ Branch 2 taken 3479 times.
✓ Branch 3 taken 337 times.
3934 if (ret<0 || *got_output) {
2824 3597 s2->current_picture_ptr = NULL;
2825
2826
4/4
✓ Branch 0 taken 462 times.
✓ Branch 1 taken 3135 times.
✓ Branch 2 taken 344 times.
✓ Branch 3 taken 118 times.
3597 if (s->timecode_frame_start != -1 && *got_output) {
2827 char tcbuf[AV_TIMECODE_STR_SIZE];
2828 344 AVFrameSideData *tcside = av_frame_new_side_data(picture,
2829 AV_FRAME_DATA_GOP_TIMECODE,
2830 sizeof(int64_t));
2831
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 344 times.
344 if (!tcside)
2832 return AVERROR(ENOMEM);
2833 344 memcpy(tcside->data, &s->timecode_frame_start, sizeof(int64_t));
2834
2835 344 av_timecode_make_mpeg_tc_string(tcbuf, s->timecode_frame_start);
2836 344 av_dict_set(&picture->metadata, "timecode", tcbuf, 0);
2837
2838 344 s->timecode_frame_start = -1;
2839 }
2840 }
2841
2842 3934 return ret;
2843 }
2844
2845 static void flush(AVCodecContext *avctx)
2846 {
2847 Mpeg1Context *s = avctx->priv_data;
2848
2849 s->sync = 0;
2850 s->closed_gop = 0;
2851
2852 ff_mpeg_flush(avctx);
2853 }
2854
2855 264 static av_cold int mpeg_decode_end(AVCodecContext *avctx)
2856 {
2857 264 Mpeg1Context *s = avctx->priv_data;
2858
2859
2/2
✓ Branch 0 taken 240 times.
✓ Branch 1 taken 24 times.
264 if (s->mpeg_enc_ctx_allocated)
2860 240 ff_mpv_common_end(&s->mpeg_enc_ctx);
2861 264 av_buffer_unref(&s->a53_buf_ref);
2862 264 return 0;
2863 }
2864
2865 const FFCodec ff_mpeg1video_decoder = {
2866 .p.name = "mpeg1video",
2867 .p.long_name = NULL_IF_CONFIG_SMALL("MPEG-1 video"),
2868 .p.type = AVMEDIA_TYPE_VIDEO,
2869 .p.id = AV_CODEC_ID_MPEG1VIDEO,
2870 .priv_data_size = sizeof(Mpeg1Context),
2871 .init = mpeg_decode_init,
2872 .close = mpeg_decode_end,
2873 FF_CODEC_DECODE_CB(mpeg_decode_frame),
2874 .p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
2875 #if FF_API_FLAG_TRUNCATED
2876 AV_CODEC_CAP_TRUNCATED |
2877 #endif
2878 AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
2879 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE |
2880 FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
2881 .flush = flush,
2882 .p.max_lowres = 3,
2883 .update_thread_context = ONLY_IF_THREADS_ENABLED(mpeg_decode_update_thread_context),
2884 .hw_configs = (const AVCodecHWConfigInternal *const []) {
2885 #if CONFIG_MPEG1_NVDEC_HWACCEL
2886 HWACCEL_NVDEC(mpeg1),
2887 #endif
2888 #if CONFIG_MPEG1_VDPAU_HWACCEL
2889 HWACCEL_VDPAU(mpeg1),
2890 #endif
2891 #if CONFIG_MPEG1_VIDEOTOOLBOX_HWACCEL
2892 HWACCEL_VIDEOTOOLBOX(mpeg1),
2893 #endif
2894 NULL
2895 },
2896 };
2897
2898 const FFCodec ff_mpeg2video_decoder = {
2899 .p.name = "mpeg2video",
2900 .p.long_name = NULL_IF_CONFIG_SMALL("MPEG-2 video"),
2901 .p.type = AVMEDIA_TYPE_VIDEO,
2902 .p.id = AV_CODEC_ID_MPEG2VIDEO,
2903 .priv_data_size = sizeof(Mpeg1Context),
2904 .init = mpeg_decode_init,
2905 .close = mpeg_decode_end,
2906 FF_CODEC_DECODE_CB(mpeg_decode_frame),
2907 .p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
2908 #if FF_API_FLAG_TRUNCATED
2909 AV_CODEC_CAP_TRUNCATED |
2910 #endif
2911 AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
2912 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE |
2913 FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
2914 .flush = flush,
2915 .p.max_lowres = 3,
2916 .p.profiles = NULL_IF_CONFIG_SMALL(ff_mpeg2_video_profiles),
2917 .hw_configs = (const AVCodecHWConfigInternal *const []) {
2918 #if CONFIG_MPEG2_DXVA2_HWACCEL
2919 HWACCEL_DXVA2(mpeg2),
2920 #endif
2921 #if CONFIG_MPEG2_D3D11VA_HWACCEL
2922 HWACCEL_D3D11VA(mpeg2),
2923 #endif
2924 #if CONFIG_MPEG2_D3D11VA2_HWACCEL
2925 HWACCEL_D3D11VA2(mpeg2),
2926 #endif
2927 #if CONFIG_MPEG2_NVDEC_HWACCEL
2928 HWACCEL_NVDEC(mpeg2),
2929 #endif
2930 #if CONFIG_MPEG2_VAAPI_HWACCEL
2931 HWACCEL_VAAPI(mpeg2),
2932 #endif
2933 #if CONFIG_MPEG2_VDPAU_HWACCEL
2934 HWACCEL_VDPAU(mpeg2),
2935 #endif
2936 #if CONFIG_MPEG2_VIDEOTOOLBOX_HWACCEL
2937 HWACCEL_VIDEOTOOLBOX(mpeg2),
2938 #endif
2939 NULL
2940 },
2941 };
2942
2943 //legacy decoder
2944 const FFCodec ff_mpegvideo_decoder = {
2945 .p.name = "mpegvideo",
2946 .p.long_name = NULL_IF_CONFIG_SMALL("MPEG-1 video"),
2947 .p.type = AVMEDIA_TYPE_VIDEO,
2948 .p.id = AV_CODEC_ID_MPEG2VIDEO,
2949 .priv_data_size = sizeof(Mpeg1Context),
2950 .init = mpeg_decode_init,
2951 .close = mpeg_decode_end,
2952 FF_CODEC_DECODE_CB(mpeg_decode_frame),
2953 .p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
2954 #if FF_API_FLAG_TRUNCATED
2955 AV_CODEC_CAP_TRUNCATED |
2956 #endif
2957 AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
2958 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE |
2959 FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
2960 .flush = flush,
2961 .p.max_lowres = 3,
2962 };
2963
2964 typedef struct IPUContext {
2965 MpegEncContext m;
2966
2967 int flags;
2968 DECLARE_ALIGNED(32, int16_t, block)[6][64];
2969 } IPUContext;
2970
2971 static int ipu_decode_frame(AVCodecContext *avctx, AVFrame *frame,
2972 int *got_frame, AVPacket *avpkt)
2973 {
2974 IPUContext *s = avctx->priv_data;
2975 MpegEncContext *m = &s->m;
2976 GetBitContext *gb = &m->gb;
2977 int ret;
2978
2979 ret = ff_get_buffer(avctx, frame, 0);
2980 if (ret < 0)
2981 return ret;
2982
2983 ret = init_get_bits8(gb, avpkt->data, avpkt->size);
2984 if (ret < 0)
2985 return ret;
2986
2987 s->flags = get_bits(gb, 8);
2988 m->intra_dc_precision = s->flags & 3;
2989 m->q_scale_type = !!(s->flags & 0x40);
2990 m->intra_vlc_format = !!(s->flags & 0x20);
2991 m->alternate_scan = !!(s->flags & 0x10);
2992
2993 if (s->flags & 0x10) {
2994 ff_init_scantable(m->idsp.idct_permutation, &m->inter_scantable, ff_alternate_vertical_scan);
2995 ff_init_scantable(m->idsp.idct_permutation, &m->intra_scantable, ff_alternate_vertical_scan);
2996 } else {
2997 ff_init_scantable(m->idsp.idct_permutation, &m->inter_scantable, ff_zigzag_direct);
2998 ff_init_scantable(m->idsp.idct_permutation, &m->intra_scantable, ff_zigzag_direct);
2999 }
3000
3001 m->last_dc[0] = m->last_dc[1] = m->last_dc[2] = 1 << (7 + (s->flags & 3));
3002 m->qscale = 1;
3003
3004 for (int y = 0; y < avctx->height; y += 16) {
3005 int intraquant;
3006
3007 for (int x = 0; x < avctx->width; x += 16) {
3008 if (x || y) {
3009 if (!get_bits1(gb))
3010 return AVERROR_INVALIDDATA;
3011 }
3012 if (get_bits1(gb)) {
3013 intraquant = 0;
3014 } else {
3015 if (!get_bits1(gb))
3016 return AVERROR_INVALIDDATA;
3017 intraquant = 1;
3018 }
3019
3020 if (s->flags & 4)
3021 skip_bits1(gb);
3022
3023 if (intraquant)
3024 m->qscale = mpeg_get_qscale(m);
3025
3026 memset(s->block, 0, sizeof(s->block));
3027
3028 for (int n = 0; n < 6; n++) {
3029 if (s->flags & 0x80) {
3030 ret = ff_mpeg1_decode_block_intra(&m->gb,
3031 m->intra_matrix,
3032 m->intra_scantable.permutated,
3033 m->last_dc, s->block[n],
3034 n, m->qscale);
3035 if (ret >= 0)
3036 m->block_last_index[n] = ret;
3037 } else {
3038 ret = mpeg2_decode_block_intra(m, s->block[n], n);
3039 }
3040
3041 if (ret < 0)
3042 return ret;
3043 }
3044
3045 m->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x,
3046 frame->linesize[0], s->block[0]);
3047 m->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x + 8,
3048 frame->linesize[0], s->block[1]);
3049 m->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x,
3050 frame->linesize[0], s->block[2]);
3051 m->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x + 8,
3052 frame->linesize[0], s->block[3]);
3053 m->idsp.idct_put(frame->data[1] + (y >> 1) * frame->linesize[1] + (x >> 1),
3054 frame->linesize[1], s->block[4]);
3055 m->idsp.idct_put(frame->data[2] + (y >> 1) * frame->linesize[2] + (x >> 1),
3056 frame->linesize[2], s->block[5]);
3057 }
3058 }
3059
3060 align_get_bits(gb);
3061 if (get_bits_left(gb) != 32)
3062 return AVERROR_INVALIDDATA;
3063
3064 frame->pict_type = AV_PICTURE_TYPE_I;
3065 frame->key_frame = 1;
3066 *got_frame = 1;
3067
3068 return avpkt->size;
3069 }
3070
3071 static av_cold int ipu_decode_init(AVCodecContext *avctx)
3072 {
3073 IPUContext *s = avctx->priv_data;
3074 MpegEncContext *m = &s->m;
3075
3076 avctx->pix_fmt = AV_PIX_FMT_YUV420P;
3077
3078 ff_mpv_decode_init(m, avctx);
3079 ff_mpv_idct_init(m);
3080 ff_mpeg12_common_init(m);
3081 ff_mpeg12_init_vlcs();
3082
3083 for (int i = 0; i < 64; i++) {
3084 int j = m->idsp.idct_permutation[i];
3085 int v = ff_mpeg1_default_intra_matrix[i];
3086 m->intra_matrix[j] = v;
3087 m->chroma_intra_matrix[j] = v;
3088 }
3089
3090 for (int i = 0; i < 64; i++) {
3091 int j = m->idsp.idct_permutation[i];
3092 int v = ff_mpeg1_default_non_intra_matrix[i];
3093 m->inter_matrix[j] = v;
3094 m->chroma_inter_matrix[j] = v;
3095 }
3096
3097 return 0;
3098 }
3099
3100 static av_cold int ipu_decode_end(AVCodecContext *avctx)
3101 {
3102 IPUContext *s = avctx->priv_data;
3103
3104 ff_mpv_common_end(&s->m);
3105
3106 return 0;
3107 }
3108
3109 const FFCodec ff_ipu_decoder = {
3110 .p.name = "ipu",
3111 .p.long_name = NULL_IF_CONFIG_SMALL("IPU Video"),
3112 .p.type = AVMEDIA_TYPE_VIDEO,
3113 .p.id = AV_CODEC_ID_IPU,
3114 .priv_data_size = sizeof(IPUContext),
3115 .init = ipu_decode_init,
3116 FF_CODEC_DECODE_CB(ipu_decode_frame),
3117 .close = ipu_decode_end,
3118 .p.capabilities = AV_CODEC_CAP_DR1,
3119 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | FF_CODEC_CAP_INIT_CLEANUP,
3120 };
3121