FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/mpeg12dec.c
Date: 2024-04-16 15:12:51
Exec Total Coverage
Lines: 1154 1576 73.2%
Functions: 29 33 87.9%
Branches: 662 1091 60.7%

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/emms.h"
35 #include "libavutil/imgutils.h"
36 #include "libavutil/internal.h"
37 #include "libavutil/mem_internal.h"
38 #include "libavutil/reverse.h"
39 #include "libavutil/stereo3d.h"
40 #include "libavutil/timecode.h"
41
42 #include "avcodec.h"
43 #include "codec_internal.h"
44 #include "decode.h"
45 #include "error_resilience.h"
46 #include "hwaccel_internal.h"
47 #include "hwconfig.h"
48 #include "idctdsp.h"
49 #include "internal.h"
50 #include "mpeg_er.h"
51 #include "mpeg12.h"
52 #include "mpeg12codecs.h"
53 #include "mpeg12data.h"
54 #include "mpeg12dec.h"
55 #include "mpegutils.h"
56 #include "mpegvideo.h"
57 #include "mpegvideodata.h"
58 #include "mpegvideodec.h"
59 #include "profiles.h"
60 #include "startcode.h"
61 #include "thread.h"
62
63 #define A53_MAX_CC_COUNT 2000
64
65 enum Mpeg2ClosedCaptionsFormat {
66 CC_FORMAT_AUTO,
67 CC_FORMAT_A53_PART4,
68 CC_FORMAT_SCTE20,
69 CC_FORMAT_DVD
70 };
71
72 typedef struct Mpeg1Context {
73 MpegEncContext mpeg_enc_ctx;
74 int repeat_field; /* true if we must repeat the field */
75 AVPanScan pan_scan; /* some temporary storage for the panscan */
76 AVStereo3D stereo3d;
77 int has_stereo3d;
78 AVBufferRef *a53_buf_ref;
79 enum Mpeg2ClosedCaptionsFormat cc_format;
80 uint8_t afd;
81 int has_afd;
82 int slice_count;
83 unsigned aspect_ratio_info;
84 AVRational save_aspect;
85 int save_width, save_height, save_progressive_seq;
86 AVRational frame_rate_ext; /* MPEG-2 specific framerate modificator */
87 unsigned frame_rate_index;
88 int sync; /* Did we reach a sync point like a GOP/SEQ/KEYFrame? */
89 int closed_gop;
90 int tmpgexs;
91 int first_slice;
92 int extradata_decoded;
93 int64_t timecode_frame_start; /*< GOP timecode frame start number, in non drop frame format */
94 } Mpeg1Context;
95
96 #define MB_TYPE_ZERO_MV 0x20000000
97
98 static const uint32_t ptype2mb_type[7] = {
99 MB_TYPE_INTRA,
100 MB_TYPE_L0 | MB_TYPE_CBP | MB_TYPE_ZERO_MV | MB_TYPE_16x16,
101 MB_TYPE_L0,
102 MB_TYPE_L0 | MB_TYPE_CBP,
103 MB_TYPE_QUANT | MB_TYPE_INTRA,
104 MB_TYPE_QUANT | MB_TYPE_L0 | MB_TYPE_CBP | MB_TYPE_ZERO_MV | MB_TYPE_16x16,
105 MB_TYPE_QUANT | MB_TYPE_L0 | MB_TYPE_CBP,
106 };
107
108 static const uint32_t btype2mb_type[11] = {
109 MB_TYPE_INTRA,
110 MB_TYPE_L1,
111 MB_TYPE_L1 | MB_TYPE_CBP,
112 MB_TYPE_L0,
113 MB_TYPE_L0 | MB_TYPE_CBP,
114 MB_TYPE_L0L1,
115 MB_TYPE_L0L1 | MB_TYPE_CBP,
116 MB_TYPE_QUANT | MB_TYPE_INTRA,
117 MB_TYPE_QUANT | MB_TYPE_L1 | MB_TYPE_CBP,
118 MB_TYPE_QUANT | MB_TYPE_L0 | MB_TYPE_CBP,
119 MB_TYPE_QUANT | MB_TYPE_L0L1 | MB_TYPE_CBP,
120 };
121
122 /* as H.263, but only 17 codes */
123 5378442 static int mpeg_decode_motion(MpegEncContext *s, int fcode, int pred)
124 {
125 int code, sign, val, shift;
126
127 5378442 code = get_vlc2(&s->gb, ff_mv_vlc, MV_VLC_BITS, 2);
128
2/2
✓ Branch 0 taken 1999099 times.
✓ Branch 1 taken 3379343 times.
5378442 if (code == 0)
129 1999099 return pred;
130
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3379343 times.
3379343 if (code < 0)
131 return 0xffff;
132
133 3379343 sign = get_bits1(&s->gb);
134 3379343 shift = fcode - 1;
135 3379343 val = code;
136
2/2
✓ Branch 0 taken 3039611 times.
✓ Branch 1 taken 339732 times.
3379343 if (shift) {
137 3039611 val = (val - 1) << shift;
138 3039611 val |= get_bits(&s->gb, shift);
139 3039611 val++;
140 }
141
2/2
✓ Branch 0 taken 1747067 times.
✓ Branch 1 taken 1632276 times.
3379343 if (sign)
142 1747067 val = -val;
143 3379343 val += pred;
144
145 /* modulo decoding */
146 3379343 return sign_extend(val, 5 + shift);
147 }
148
149 #define MAX_INDEX (64 - 1)
150 #define check_scantable_index(ctx, x) \
151 do { \
152 if ((x) > MAX_INDEX) { \
153 av_log(ctx->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", \
154 ctx->mb_x, ctx->mb_y); \
155 return AVERROR_INVALIDDATA; \
156 } \
157 } while (0)
158
159 403182 static inline int mpeg1_decode_block_inter(MpegEncContext *s,
160 int16_t *block, int n)
161 {
162 int level, i, j, run;
163 403182 uint8_t *const scantable = s->intra_scantable.permutated;
164 403182 const uint16_t *quant_matrix = s->inter_matrix;
165 403182 const int qscale = s->qscale;
166
167 {
168 403182 OPEN_READER(re, &s->gb);
169 403182 i = -1;
170 // special case for first coefficient, no need to add second VLC table
171 403182 UPDATE_CACHE(re, &s->gb);
172
2/2
✓ Branch 0 taken 301257 times.
✓ Branch 1 taken 101925 times.
403182 if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
173 101925 level = (3 * qscale * quant_matrix[0]) >> 5;
174 101925 level = (level - 1) | 1;
175
2/2
✓ Branch 0 taken 57008 times.
✓ Branch 1 taken 44917 times.
101925 if (GET_CACHE(re, &s->gb) & 0x40000000)
176 57008 level = -level;
177 101925 block[0] = level;
178 101925 i++;
179 101925 SKIP_BITS(re, &s->gb, 2);
180
2/2
✓ Branch 0 taken 67516 times.
✓ Branch 1 taken 34409 times.
101925 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
181 34409 goto end;
182 }
183 /* now quantify & encode AC coefficients */
184 for (;;) {
185
2/2
✓ Branch 1 taken 335978 times.
✓ Branch 2 taken 2584676 times.
2920654 GET_RL_VLC(level, run, re, &s->gb, ff_mpeg1_rl_vlc,
186 TEX_VLC_BITS, 2, 0);
187
188
2/2
✓ Branch 0 taken 2841520 times.
✓ Branch 1 taken 79134 times.
2920654 if (level != 0) {
189 2841520 i += run;
190
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2841520 times.
2841520 if (i > MAX_INDEX)
191 break;
192 2841520 j = scantable[i];
193 2841520 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
194 2841520 level = (level - 1) | 1;
195 2841520 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
196 2841520 SHOW_SBITS(re, &s->gb, 1);
197 2841520 SKIP_BITS(re, &s->gb, 1);
198 } else {
199 /* escape */
200 79134 run = SHOW_UBITS(re, &s->gb, 6) + 1;
201 79134 LAST_SKIP_BITS(re, &s->gb, 6);
202 79134 UPDATE_CACHE(re, &s->gb);
203 79134 level = SHOW_SBITS(re, &s->gb, 8);
204 79134 SKIP_BITS(re, &s->gb, 8);
205
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 79134 times.
79134 if (level == -128) {
206 level = SHOW_UBITS(re, &s->gb, 8) - 256;
207 SKIP_BITS(re, &s->gb, 8);
208
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 79134 times.
79134 } else if (level == 0) {
209 level = SHOW_UBITS(re, &s->gb, 8);
210 SKIP_BITS(re, &s->gb, 8);
211 }
212 79134 i += run;
213
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 79134 times.
79134 if (i > MAX_INDEX)
214 break;
215 79134 j = scantable[i];
216
2/2
✓ Branch 0 taken 38931 times.
✓ Branch 1 taken 40203 times.
79134 if (level < 0) {
217 38931 level = -level;
218 38931 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
219 38931 level = (level - 1) | 1;
220 38931 level = -level;
221 } else {
222 40203 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
223 40203 level = (level - 1) | 1;
224 }
225 }
226
227 2920654 block[j] = level;
228
2/2
✓ Branch 0 taken 368773 times.
✓ Branch 1 taken 2551881 times.
2920654 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
229 368773 break;
230 2551881 UPDATE_CACHE(re, &s->gb);
231 }
232 403182 end:
233 403182 LAST_SKIP_BITS(re, &s->gb, 2);
234 403182 CLOSE_READER(re, &s->gb);
235 }
236
237
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 403182 times.
403182 check_scantable_index(s, i);
238
239 403182 s->block_last_index[n] = i;
240 403182 return 0;
241 }
242
243 5108383 static inline int mpeg2_decode_block_non_intra(MpegEncContext *s,
244 int16_t *block, int n)
245 {
246 int level, i, j, run;
247 5108383 uint8_t *const scantable = s->intra_scantable.permutated;
248 const uint16_t *quant_matrix;
249 5108383 const int qscale = s->qscale;
250 int mismatch;
251
252 5108383 mismatch = 1;
253
254 {
255 5108383 OPEN_READER(re, &s->gb);
256 5108383 i = -1;
257
2/2
✓ Branch 0 taken 3820871 times.
✓ Branch 1 taken 1287512 times.
5108383 if (n < 4)
258 3820871 quant_matrix = s->inter_matrix;
259 else
260 1287512 quant_matrix = s->chroma_inter_matrix;
261
262 // Special case for first coefficient, no need to add second VLC table.
263 5108383 UPDATE_CACHE(re, &s->gb);
264
2/2
✓ Branch 0 taken 3240237 times.
✓ Branch 1 taken 1868146 times.
5108383 if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
265 1868146 level = (3 * qscale * quant_matrix[0]) >> 5;
266
2/2
✓ Branch 0 taken 1180512 times.
✓ Branch 1 taken 687634 times.
1868146 if (GET_CACHE(re, &s->gb) & 0x40000000)
267 1180512 level = -level;
268 1868146 block[0] = level;
269 1868146 mismatch ^= level;
270 1868146 i++;
271 1868146 SKIP_BITS(re, &s->gb, 2);
272
2/2
✓ Branch 0 taken 978716 times.
✓ Branch 1 taken 889430 times.
1868146 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
273 889430 goto end;
274 }
275
276 /* now quantify & encode AC coefficients */
277 for (;;) {
278
2/2
✓ Branch 1 taken 1616368 times.
✓ Branch 2 taken 16616338 times.
18232706 GET_RL_VLC(level, run, re, &s->gb, ff_mpeg1_rl_vlc,
279 TEX_VLC_BITS, 2, 0);
280
281
2/2
✓ Branch 0 taken 17962280 times.
✓ Branch 1 taken 270426 times.
18232706 if (level != 0) {
282 17962280 i += run;
283
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 17962280 times.
17962280 if (i > MAX_INDEX)
284 break;
285 17962280 j = scantable[i];
286 17962280 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
287 17962280 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
288 17962280 SHOW_SBITS(re, &s->gb, 1);
289 17962280 SKIP_BITS(re, &s->gb, 1);
290 } else {
291 /* escape */
292 270426 run = SHOW_UBITS(re, &s->gb, 6) + 1;
293 270426 LAST_SKIP_BITS(re, &s->gb, 6);
294 270426 UPDATE_CACHE(re, &s->gb);
295 270426 level = SHOW_SBITS(re, &s->gb, 12);
296 270426 SKIP_BITS(re, &s->gb, 12);
297
298 270426 i += run;
299
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 270426 times.
270426 if (i > MAX_INDEX)
300 break;
301 270426 j = scantable[i];
302
2/2
✓ Branch 0 taken 131388 times.
✓ Branch 1 taken 139038 times.
270426 if (level < 0) {
303 131388 level = ((-level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
304 131388 level = -level;
305 } else {
306 139038 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
307 }
308 }
309
310 18232706 mismatch ^= level;
311 18232706 block[j] = level;
312
2/2
✓ Branch 0 taken 4218953 times.
✓ Branch 1 taken 14013753 times.
18232706 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
313 4218953 break;
314 14013753 UPDATE_CACHE(re, &s->gb);
315 }
316 5108383 end:
317 5108383 LAST_SKIP_BITS(re, &s->gb, 2);
318 5108383 CLOSE_READER(re, &s->gb);
319 }
320 5108383 block[63] ^= (mismatch & 1);
321
322
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5108383 times.
5108383 check_scantable_index(s, i);
323
324 5108383 s->block_last_index[n] = i;
325 5108383 return 0;
326 }
327
328 1911446 static inline int mpeg2_decode_block_intra(MpegEncContext *s,
329 int16_t *block, int n)
330 {
331 int level, dc, diff, i, j, run;
332 int component;
333 const RL_VLC_ELEM *rl_vlc;
334 1911446 uint8_t *const scantable = s->intra_scantable.permutated;
335 const uint16_t *quant_matrix;
336 1911446 const int qscale = s->qscale;
337 int mismatch;
338
339 /* DC coefficient */
340
2/2
✓ Branch 0 taken 1202092 times.
✓ Branch 1 taken 709354 times.
1911446 if (n < 4) {
341 1202092 quant_matrix = s->intra_matrix;
342 1202092 component = 0;
343 } else {
344 709354 quant_matrix = s->chroma_intra_matrix;
345 709354 component = (n & 1) + 1;
346 }
347 1911446 diff = decode_dc(&s->gb, component);
348 1911446 dc = s->last_dc[component];
349 1911446 dc += diff;
350 1911446 s->last_dc[component] = dc;
351 1911446 block[0] = dc * (1 << (3 - s->intra_dc_precision));
352 ff_tlog(s->avctx, "dc=%d\n", block[0]);
353 1911446 mismatch = block[0] ^ 1;
354 1911446 i = 0;
355
2/2
✓ Branch 0 taken 1329530 times.
✓ Branch 1 taken 581916 times.
1911446 if (s->intra_vlc_format)
356 1329530 rl_vlc = ff_mpeg2_rl_vlc;
357 else
358 581916 rl_vlc = ff_mpeg1_rl_vlc;
359
360 {
361 1911446 OPEN_READER(re, &s->gb);
362 /* now quantify & encode AC coefficients */
363 for (;;) {
364 26544818 UPDATE_CACHE(re, &s->gb);
365
2/2
✓ Branch 1 taken 728929 times.
✓ Branch 2 taken 13499203 times.
14228132 GET_RL_VLC(level, run, re, &s->gb, rl_vlc,
366 TEX_VLC_BITS, 2, 0);
367
368
2/2
✓ Branch 0 taken 1911436 times.
✓ Branch 1 taken 12316696 times.
14228132 if (level == 127) {
369 1911436 break;
370
2/2
✓ Branch 0 taken 12185519 times.
✓ Branch 1 taken 131177 times.
12316696 } else if (level != 0) {
371 12185519 i += run;
372
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 12185509 times.
12185519 if (i > MAX_INDEX)
373 10 break;
374 12185509 j = scantable[i];
375 12185509 level = (level * qscale * quant_matrix[j]) >> 4;
376 12185509 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
377 12185509 SHOW_SBITS(re, &s->gb, 1);
378 12185509 LAST_SKIP_BITS(re, &s->gb, 1);
379 } else {
380 /* escape */
381 131177 run = SHOW_UBITS(re, &s->gb, 6) + 1;
382 131177 SKIP_BITS(re, &s->gb, 6);
383 131177 level = SHOW_SBITS(re, &s->gb, 12);
384 131177 LAST_SKIP_BITS(re, &s->gb, 12);
385 131177 i += run;
386
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 131177 times.
131177 if (i > MAX_INDEX)
387 break;
388 131177 j = scantable[i];
389
2/2
✓ Branch 0 taken 58424 times.
✓ Branch 1 taken 72753 times.
131177 if (level < 0) {
390 58424 level = (-level * qscale * quant_matrix[j]) >> 4;
391 58424 level = -level;
392 } else {
393 72753 level = (level * qscale * quant_matrix[j]) >> 4;
394 }
395 }
396
397 12316686 mismatch ^= level;
398 12316686 block[j] = level;
399 }
400 1911446 CLOSE_READER(re, &s->gb);
401 }
402 1911446 block[63] ^= mismatch & 1;
403
404
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 1911436 times.
1911446 check_scantable_index(s, i);
405
406 1911436 s->block_last_index[n] = i;
407 1911436 return 0;
408 }
409
410 /******************************************/
411 /* decoding */
412
413 42876 static inline int get_dmv(MpegEncContext *s)
414 {
415
2/2
✓ Branch 1 taken 39969 times.
✓ Branch 2 taken 2907 times.
42876 if (get_bits1(&s->gb))
416 39969 return 1 - (get_bits1(&s->gb) << 1);
417 else
418 2907 return 0;
419 }
420
421 /* motion type (for MPEG-2) */
422 #define MT_FIELD 1
423 #define MT_FRAME 2
424 #define MT_16X8 2
425 #define MT_DMV 3
426
427 2600716 static int mpeg_decode_mb(MpegEncContext *s, int16_t block[12][64])
428 {
429 int i, j, k, cbp, val, mb_type, motion_type;
430 2600716 const int mb_block_count = 4 + (1 << s->chroma_format);
431 int ret;
432
433 ff_tlog(s->avctx, "decode_mb: x=%d y=%d\n", s->mb_x, s->mb_y);
434
435 av_assert2(s->mb_skipped == 0);
436
437
2/2
✓ Branch 0 taken 165331 times.
✓ Branch 1 taken 2435385 times.
2600716 if (s->mb_skip_run-- != 0) {
438
2/2
✓ Branch 0 taken 57361 times.
✓ Branch 1 taken 107970 times.
165331 if (s->pict_type == AV_PICTURE_TYPE_P) {
439 57361 s->mb_skipped = 1;
440 57361 s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride] =
441 MB_TYPE_SKIP | MB_TYPE_L0 | MB_TYPE_16x16;
442 } else {
443 int mb_type;
444
445
1/2
✓ Branch 0 taken 107970 times.
✗ Branch 1 not taken.
107970 if (s->mb_x)
446 107970 mb_type = s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride - 1];
447 else
448 // FIXME not sure if this is allowed in MPEG at all
449 mb_type = s->current_picture.mb_type[s->mb_width + (s->mb_y - 1) * s->mb_stride - 1];
450
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 107970 times.
107970 if (IS_INTRA(mb_type)) {
451 av_log(s->avctx, AV_LOG_ERROR, "skip with previntra\n");
452 return AVERROR_INVALIDDATA;
453 }
454 107970 s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride] =
455 107970 mb_type | MB_TYPE_SKIP;
456
457
2/2
✓ Branch 0 taken 75569 times.
✓ Branch 1 taken 32401 times.
107970 if ((s->mv[0][0][0] | s->mv[0][0][1] | s->mv[1][0][0] | s->mv[1][0][1]) == 0)
458 75569 s->mb_skipped = 1;
459 }
460
461 165331 return 0;
462 }
463
464
3/3
✓ Branch 0 taken 239190 times.
✓ Branch 1 taken 1336029 times.
✓ Branch 2 taken 860166 times.
2435385 switch (s->pict_type) {
465 239190 default:
466 case AV_PICTURE_TYPE_I:
467
2/2
✓ Branch 1 taken 17897 times.
✓ Branch 2 taken 221293 times.
239190 if (get_bits1(&s->gb) == 0) {
468
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 17897 times.
17897 if (get_bits1(&s->gb) == 0) {
469 av_log(s->avctx, AV_LOG_ERROR,
470 "Invalid mb type in I-frame at %d %d\n",
471 s->mb_x, s->mb_y);
472 return AVERROR_INVALIDDATA;
473 }
474 17897 mb_type = MB_TYPE_QUANT | MB_TYPE_INTRA;
475 } else {
476 221293 mb_type = MB_TYPE_INTRA;
477 }
478 239190 break;
479 1336029 case AV_PICTURE_TYPE_P:
480 1336029 mb_type = get_vlc2(&s->gb, ff_mb_ptype_vlc, MB_PTYPE_VLC_BITS, 1);
481
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1336029 times.
1336029 if (mb_type < 0) {
482 av_log(s->avctx, AV_LOG_ERROR,
483 "Invalid mb type in P-frame at %d %d\n", s->mb_x, s->mb_y);
484 return AVERROR_INVALIDDATA;
485 }
486 1336029 mb_type = ptype2mb_type[mb_type];
487 1336029 break;
488 860166 case AV_PICTURE_TYPE_B:
489 860166 mb_type = get_vlc2(&s->gb, ff_mb_btype_vlc, MB_BTYPE_VLC_BITS, 1);
490
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 860166 times.
860166 if (mb_type < 0) {
491 av_log(s->avctx, AV_LOG_ERROR,
492 "Invalid mb type in B-frame at %d %d\n", s->mb_x, s->mb_y);
493 return AVERROR_INVALIDDATA;
494 }
495 860166 mb_type = btype2mb_type[mb_type];
496 860166 break;
497 }
498 ff_tlog(s->avctx, "mb_type=%x\n", mb_type);
499 // motion_type = 0; /* avoid warning */
500
2/2
✓ Branch 0 taken 321532 times.
✓ Branch 1 taken 2113853 times.
2435385 if (IS_INTRA(mb_type)) {
501 321532 s->bdsp.clear_blocks(s->block[0]);
502
503
2/2
✓ Branch 0 taken 54159 times.
✓ Branch 1 taken 267373 times.
321532 if (!s->chroma_y_shift)
504 54159 s->bdsp.clear_blocks(s->block[6]);
505
506 /* compute DCT type */
507 // FIXME: add an interlaced_dct coded var?
508
2/2
✓ Branch 0 taken 263886 times.
✓ Branch 1 taken 57646 times.
321532 if (s->picture_structure == PICT_FRAME &&
509
2/2
✓ Branch 0 taken 112468 times.
✓ Branch 1 taken 151418 times.
263886 !s->frame_pred_frame_dct)
510 112468 s->interlaced_dct = get_bits1(&s->gb);
511
512
2/2
✓ Branch 0 taken 25984 times.
✓ Branch 1 taken 295548 times.
321532 if (IS_QUANT(mb_type))
513 25984 s->qscale = mpeg_get_qscale(s);
514
515
2/2
✓ Branch 0 taken 1356 times.
✓ Branch 1 taken 320176 times.
321532 if (s->concealment_motion_vectors) {
516 /* just parse them */
517
2/2
✓ Branch 0 taken 1350 times.
✓ Branch 1 taken 6 times.
1356 if (s->picture_structure != PICT_FRAME)
518 1350 skip_bits1(&s->gb); /* field select */
519
520 1356 s->mv[0][0][0] =
521 1356 s->last_mv[0][0][0] =
522 1356 s->last_mv[0][1][0] = mpeg_decode_motion(s, s->mpeg_f_code[0][0],
523 s->last_mv[0][0][0]);
524 1356 s->mv[0][0][1] =
525 1356 s->last_mv[0][0][1] =
526 1356 s->last_mv[0][1][1] = mpeg_decode_motion(s, s->mpeg_f_code[0][1],
527 s->last_mv[0][0][1]);
528
529 1356 check_marker(s->avctx, &s->gb, "after concealment_motion_vectors");
530 } else {
531 /* reset mv prediction */
532 320176 memset(s->last_mv, 0, sizeof(s->last_mv));
533 }
534 321532 s->mb_intra = 1;
535
536
2/2
✓ Branch 0 taken 300528 times.
✓ Branch 1 taken 21004 times.
321532 if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
537
2/2
✓ Branch 0 taken 1911446 times.
✓ Branch 1 taken 300518 times.
2211964 for (i = 0; i < mb_block_count; i++)
538
2/2
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 1911436 times.
1911446 if ((ret = mpeg2_decode_block_intra(s, *s->pblocks[i], i)) < 0)
539 10 return ret;
540 } else {
541
2/2
✓ Branch 0 taken 126024 times.
✓ Branch 1 taken 21004 times.
147028 for (i = 0; i < 6; i++) {
542 126024 ret = ff_mpeg1_decode_block_intra(&s->gb,
543 126024 s->intra_matrix,
544 126024 s->intra_scantable.permutated,
545 126024 s->last_dc, *s->pblocks[i],
546 i, s->qscale);
547
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 126024 times.
126024 if (ret < 0) {
548 av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n",
549 s->mb_x, s->mb_y);
550 return ret;
551 }
552
553 126024 s->block_last_index[i] = ret;
554 }
555 }
556 } else {
557
2/2
✓ Branch 0 taken 211965 times.
✓ Branch 1 taken 1901888 times.
2113853 if (mb_type & MB_TYPE_ZERO_MV) {
558 av_assert2(mb_type & MB_TYPE_CBP);
559
560 211965 s->mv_dir = MV_DIR_FORWARD;
561
2/2
✓ Branch 0 taken 87903 times.
✓ Branch 1 taken 124062 times.
211965 if (s->picture_structure == PICT_FRAME) {
562
1/2
✓ Branch 0 taken 87903 times.
✗ Branch 1 not taken.
87903 if (s->picture_structure == PICT_FRAME
563
2/2
✓ Branch 0 taken 52116 times.
✓ Branch 1 taken 35787 times.
87903 && !s->frame_pred_frame_dct)
564 52116 s->interlaced_dct = get_bits1(&s->gb);
565 87903 s->mv_type = MV_TYPE_16X16;
566 } else {
567 124062 s->mv_type = MV_TYPE_FIELD;
568 124062 mb_type |= MB_TYPE_INTERLACED;
569 124062 s->field_select[0][0] = s->picture_structure - 1;
570 }
571
572
2/2
✓ Branch 0 taken 51734 times.
✓ Branch 1 taken 160231 times.
211965 if (IS_QUANT(mb_type))
573 51734 s->qscale = mpeg_get_qscale(s);
574
575 211965 s->last_mv[0][0][0] = 0;
576 211965 s->last_mv[0][0][1] = 0;
577 211965 s->last_mv[0][1][0] = 0;
578 211965 s->last_mv[0][1][1] = 0;
579 211965 s->mv[0][0][0] = 0;
580 211965 s->mv[0][0][1] = 0;
581 } else {
582 av_assert2(mb_type & MB_TYPE_L0L1);
583 // FIXME decide if MBs in field pictures are MB_TYPE_INTERLACED
584 /* get additional motion vector type */
585
4/4
✓ Branch 0 taken 1401576 times.
✓ Branch 1 taken 500312 times.
✓ Branch 2 taken 750644 times.
✓ Branch 3 taken 650932 times.
1901888 if (s->picture_structure == PICT_FRAME && s->frame_pred_frame_dct) {
586 750644 motion_type = MT_FRAME;
587 } else {
588 1151244 motion_type = get_bits(&s->gb, 2);
589
4/4
✓ Branch 0 taken 650932 times.
✓ Branch 1 taken 500312 times.
✓ Branch 2 taken 471878 times.
✓ Branch 3 taken 179054 times.
1151244 if (s->picture_structure == PICT_FRAME && HAS_CBP(mb_type))
590 471878 s->interlaced_dct = get_bits1(&s->gb);
591 }
592
593
2/2
✓ Branch 0 taken 173435 times.
✓ Branch 1 taken 1728453 times.
1901888 if (IS_QUANT(mb_type))
594 173435 s->qscale = mpeg_get_qscale(s);
595
596 /* motion vectors */
597 1901888 s->mv_dir = (mb_type >> 13) & 3;
598 ff_tlog(s->avctx, "motion_type=%d\n", motion_type);
599
3/4
✓ Branch 0 taken 1618543 times.
✓ Branch 1 taken 261907 times.
✓ Branch 2 taken 21438 times.
✗ Branch 3 not taken.
1901888 switch (motion_type) {
600 1618543 case MT_FRAME: /* or MT_16X8 */
601
2/2
✓ Branch 0 taken 1331022 times.
✓ Branch 1 taken 287521 times.
1618543 if (s->picture_structure == PICT_FRAME) {
602 1331022 mb_type |= MB_TYPE_16x16;
603 1331022 s->mv_type = MV_TYPE_16X16;
604
2/2
✓ Branch 0 taken 2662044 times.
✓ Branch 1 taken 1331022 times.
3993066 for (i = 0; i < 2; i++) {
605
2/2
✓ Branch 0 taken 1715222 times.
✓ Branch 1 taken 946822 times.
2662044 if (USES_LIST(mb_type, i)) {
606 /* MT_FRAME */
607 1715222 s->mv[i][0][0] =
608 1715222 s->last_mv[i][0][0] =
609 1715222 s->last_mv[i][1][0] =
610 1715222 mpeg_decode_motion(s, s->mpeg_f_code[i][0],
611 s->last_mv[i][0][0]);
612 1715222 s->mv[i][0][1] =
613 1715222 s->last_mv[i][0][1] =
614 1715222 s->last_mv[i][1][1] =
615 1715222 mpeg_decode_motion(s, s->mpeg_f_code[i][1],
616 s->last_mv[i][0][1]);
617 /* full_pel: only for MPEG-1 */
618
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1715222 times.
1715222 if (s->full_pel[i]) {
619 s->mv[i][0][0] *= 2;
620 s->mv[i][0][1] *= 2;
621 }
622 }
623 }
624 } else {
625 287521 mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
626 287521 s->mv_type = MV_TYPE_16X8;
627
2/2
✓ Branch 0 taken 575042 times.
✓ Branch 1 taken 287521 times.
862563 for (i = 0; i < 2; i++) {
628
2/2
✓ Branch 0 taken 288883 times.
✓ Branch 1 taken 286159 times.
575042 if (USES_LIST(mb_type, i)) {
629 /* MT_16X8 */
630
2/2
✓ Branch 0 taken 577766 times.
✓ Branch 1 taken 288883 times.
866649 for (j = 0; j < 2; j++) {
631 577766 s->field_select[i][j] = get_bits1(&s->gb);
632
2/2
✓ Branch 0 taken 1155532 times.
✓ Branch 1 taken 577766 times.
1733298 for (k = 0; k < 2; k++) {
633 1155532 val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
634 s->last_mv[i][j][k]);
635 1155532 s->last_mv[i][j][k] = val;
636 1155532 s->mv[i][j][k] = val;
637 }
638 }
639 }
640 }
641 }
642 1618543 break;
643 261907 case MT_FIELD:
644 261907 s->mv_type = MV_TYPE_FIELD;
645
2/2
✓ Branch 0 taken 70554 times.
✓ Branch 1 taken 191353 times.
261907 if (s->picture_structure == PICT_FRAME) {
646 70554 mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
647
2/2
✓ Branch 0 taken 141108 times.
✓ Branch 1 taken 70554 times.
211662 for (i = 0; i < 2; i++) {
648
2/2
✓ Branch 0 taken 88706 times.
✓ Branch 1 taken 52402 times.
141108 if (USES_LIST(mb_type, i)) {
649
2/2
✓ Branch 0 taken 177412 times.
✓ Branch 1 taken 88706 times.
266118 for (j = 0; j < 2; j++) {
650 177412 s->field_select[i][j] = get_bits1(&s->gb);
651 177412 val = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
652 s->last_mv[i][j][0]);
653 177412 s->last_mv[i][j][0] = val;
654 177412 s->mv[i][j][0] = val;
655 ff_tlog(s->avctx, "fmx=%d\n", val);
656 177412 val = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
657 177412 s->last_mv[i][j][1] >> 1);
658 177412 s->last_mv[i][j][1] = 2 * val;
659 177412 s->mv[i][j][1] = val;
660 ff_tlog(s->avctx, "fmy=%d\n", val);
661 }
662 }
663 }
664 } else {
665
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 191353 times.
191353 av_assert0(!s->progressive_sequence);
666 191353 mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
667
2/2
✓ Branch 0 taken 382706 times.
✓ Branch 1 taken 191353 times.
574059 for (i = 0; i < 2; i++) {
668
2/2
✓ Branch 0 taken 196027 times.
✓ Branch 1 taken 186679 times.
382706 if (USES_LIST(mb_type, i)) {
669 196027 s->field_select[i][0] = get_bits1(&s->gb);
670
2/2
✓ Branch 0 taken 392054 times.
✓ Branch 1 taken 196027 times.
588081 for (k = 0; k < 2; k++) {
671 392054 val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
672 s->last_mv[i][0][k]);
673 392054 s->last_mv[i][0][k] = val;
674 392054 s->last_mv[i][1][k] = val;
675 392054 s->mv[i][0][k] = val;
676 }
677 }
678 }
679 }
680 261907 break;
681 21438 case MT_DMV:
682
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 21438 times.
21438 if (s->progressive_sequence){
683 av_log(s->avctx, AV_LOG_ERROR, "MT_DMV in progressive_sequence\n");
684 return AVERROR_INVALIDDATA;
685 }
686 21438 s->mv_type = MV_TYPE_DMV;
687
2/2
✓ Branch 0 taken 42876 times.
✓ Branch 1 taken 21438 times.
64314 for (i = 0; i < 2; i++) {
688
2/2
✓ Branch 0 taken 21438 times.
✓ Branch 1 taken 21438 times.
42876 if (USES_LIST(mb_type, i)) {
689 int dmx, dmy, mx, my, m;
690 21438 const int my_shift = s->picture_structure == PICT_FRAME;
691
692 21438 mx = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
693 s->last_mv[i][0][0]);
694 21438 s->last_mv[i][0][0] = mx;
695 21438 s->last_mv[i][1][0] = mx;
696 21438 dmx = get_dmv(s);
697 21438 my = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
698 21438 s->last_mv[i][0][1] >> my_shift);
699 21438 dmy = get_dmv(s);
700
701
702 21438 s->last_mv[i][0][1] = my * (1 << my_shift);
703 21438 s->last_mv[i][1][1] = my * (1 << my_shift);
704
705 21438 s->mv[i][0][0] = mx;
706 21438 s->mv[i][0][1] = my;
707 21438 s->mv[i][1][0] = mx; // not used
708 21438 s->mv[i][1][1] = my; // not used
709
710
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 21438 times.
21438 if (s->picture_structure == PICT_FRAME) {
711 mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
712
713 // m = 1 + 2 * s->top_field_first;
714 m = s->top_field_first ? 1 : 3;
715
716 /* top -> top pred */
717 s->mv[i][2][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
718 s->mv[i][2][1] = ((my * m + (my > 0)) >> 1) + dmy - 1;
719 m = 4 - m;
720 s->mv[i][3][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
721 s->mv[i][3][1] = ((my * m + (my > 0)) >> 1) + dmy + 1;
722 } else {
723 21438 mb_type |= MB_TYPE_16x16;
724
725 21438 s->mv[i][2][0] = ((mx + (mx > 0)) >> 1) + dmx;
726 21438 s->mv[i][2][1] = ((my + (my > 0)) >> 1) + dmy;
727
2/2
✓ Branch 0 taken 18359 times.
✓ Branch 1 taken 3079 times.
21438 if (s->picture_structure == PICT_TOP_FIELD)
728 18359 s->mv[i][2][1]--;
729 else
730 3079 s->mv[i][2][1]++;
731 }
732 }
733 }
734 21438 break;
735 default:
736 av_log(s->avctx, AV_LOG_ERROR,
737 "00 motion_type at %d %d\n", s->mb_x, s->mb_y);
738 return AVERROR_INVALIDDATA;
739 }
740 }
741
742 2113853 s->mb_intra = 0;
743
2/2
✓ Branch 0 taken 1692921 times.
✓ Branch 1 taken 420932 times.
2113853 if (HAS_CBP(mb_type)) {
744 1692921 s->bdsp.clear_blocks(s->block[0]);
745
746 1692921 cbp = get_vlc2(&s->gb, ff_mb_pat_vlc, MB_PAT_VLC_BITS, 1);
747
2/2
✓ Branch 0 taken 43689 times.
✓ Branch 1 taken 1649232 times.
1692921 if (mb_block_count > 6) {
748 43689 cbp *= 1 << mb_block_count - 6;
749 43689 cbp |= get_bits(&s->gb, mb_block_count - 6);
750 43689 s->bdsp.clear_blocks(s->block[6]);
751 }
752
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1692921 times.
1692921 if (cbp <= 0) {
753 av_log(s->avctx, AV_LOG_ERROR,
754 "invalid cbp %d at %d %d\n", cbp, s->mb_x, s->mb_y);
755 return AVERROR_INVALIDDATA;
756 }
757
758
2/2
✓ Branch 0 taken 1581185 times.
✓ Branch 1 taken 111736 times.
1692921 if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
759 1581185 cbp <<= 12 - mb_block_count;
760
761
2/2
✓ Branch 0 taken 9574488 times.
✓ Branch 1 taken 1581185 times.
11155673 for (i = 0; i < mb_block_count; i++) {
762
2/2
✓ Branch 0 taken 5108383 times.
✓ Branch 1 taken 4466105 times.
9574488 if (cbp & (1 << 11)) {
763
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 5108383 times.
5108383 if ((ret = mpeg2_decode_block_non_intra(s, *s->pblocks[i], i)) < 0)
764 return ret;
765 } else {
766 4466105 s->block_last_index[i] = -1;
767 }
768 9574488 cbp += cbp;
769 }
770 } else {
771
2/2
✓ Branch 0 taken 670416 times.
✓ Branch 1 taken 111736 times.
782152 for (i = 0; i < 6; i++) {
772
2/2
✓ Branch 0 taken 403182 times.
✓ Branch 1 taken 267234 times.
670416 if (cbp & 32) {
773
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 403182 times.
403182 if ((ret = mpeg1_decode_block_inter(s, *s->pblocks[i], i)) < 0)
774 return ret;
775 } else {
776 267234 s->block_last_index[i] = -1;
777 }
778 670416 cbp += cbp;
779 }
780 }
781 } else {
782
2/2
✓ Branch 0 taken 5051184 times.
✓ Branch 1 taken 420932 times.
5472116 for (i = 0; i < 12; i++)
783 5051184 s->block_last_index[i] = -1;
784 }
785 }
786
787 2435375 s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride] = mb_type;
788
789 2435375 return 0;
790 }
791
792 286 static av_cold int mpeg_decode_init(AVCodecContext *avctx)
793 {
794 286 Mpeg1Context *s = avctx->priv_data;
795 286 MpegEncContext *s2 = &s->mpeg_enc_ctx;
796
797
2/2
✓ Branch 0 taken 284 times.
✓ Branch 1 taken 2 times.
286 if ( avctx->codec_tag != AV_RL32("VCR2")
798
1/2
✓ Branch 0 taken 284 times.
✗ Branch 1 not taken.
284 && avctx->codec_tag != AV_RL32("BW10"))
799 284 avctx->coded_width = avctx->coded_height = 0; // do not trust dimensions from input
800 286 ff_mpv_decode_init(s2, avctx);
801
802 286 ff_mpeg12_init_vlcs();
803
804 286 s2->chroma_format = 1;
805 286 s->repeat_field = 0;
806 286 avctx->color_range = AVCOL_RANGE_MPEG;
807 286 return 0;
808 }
809
810 #if HAVE_THREADS
811 static int mpeg_decode_update_thread_context(AVCodecContext *avctx,
812 const AVCodecContext *avctx_from)
813 {
814 Mpeg1Context *ctx = avctx->priv_data, *ctx_from = avctx_from->priv_data;
815 MpegEncContext *s = &ctx->mpeg_enc_ctx, *s1 = &ctx_from->mpeg_enc_ctx;
816 int err;
817
818 if (avctx == avctx_from || !s1->context_initialized)
819 return 0;
820
821 err = ff_mpeg_update_thread_context(avctx, avctx_from);
822 if (err)
823 return err;
824
825 if (!s->context_initialized)
826 memcpy(s + 1, s1 + 1, sizeof(Mpeg1Context) - sizeof(MpegEncContext));
827
828 return 0;
829 }
830 #endif
831
832 static const enum AVPixelFormat mpeg1_hwaccel_pixfmt_list_420[] = {
833 #if CONFIG_MPEG1_NVDEC_HWACCEL
834 AV_PIX_FMT_CUDA,
835 #endif
836 #if CONFIG_MPEG1_VDPAU_HWACCEL
837 AV_PIX_FMT_VDPAU,
838 #endif
839 AV_PIX_FMT_YUV420P,
840 AV_PIX_FMT_NONE
841 };
842
843 static const enum AVPixelFormat mpeg2_hwaccel_pixfmt_list_420[] = {
844 #if CONFIG_MPEG2_NVDEC_HWACCEL
845 AV_PIX_FMT_CUDA,
846 #endif
847 #if CONFIG_MPEG2_VDPAU_HWACCEL
848 AV_PIX_FMT_VDPAU,
849 #endif
850 #if CONFIG_MPEG2_DXVA2_HWACCEL
851 AV_PIX_FMT_DXVA2_VLD,
852 #endif
853 #if CONFIG_MPEG2_D3D11VA_HWACCEL
854 AV_PIX_FMT_D3D11VA_VLD,
855 AV_PIX_FMT_D3D11,
856 #endif
857 #if CONFIG_MPEG2_D3D12VA_HWACCEL
858 AV_PIX_FMT_D3D12,
859 #endif
860 #if CONFIG_MPEG2_VAAPI_HWACCEL
861 AV_PIX_FMT_VAAPI,
862 #endif
863 #if CONFIG_MPEG2_VIDEOTOOLBOX_HWACCEL
864 AV_PIX_FMT_VIDEOTOOLBOX,
865 #endif
866 AV_PIX_FMT_YUV420P,
867 AV_PIX_FMT_NONE
868 };
869
870 static const enum AVPixelFormat mpeg12_pixfmt_list_422[] = {
871 AV_PIX_FMT_YUV422P,
872 AV_PIX_FMT_NONE
873 };
874
875 static const enum AVPixelFormat mpeg12_pixfmt_list_444[] = {
876 AV_PIX_FMT_YUV444P,
877 AV_PIX_FMT_NONE
878 };
879
880 263 static enum AVPixelFormat mpeg_get_pixelformat(AVCodecContext *avctx)
881 {
882 263 Mpeg1Context *s1 = avctx->priv_data;
883 263 MpegEncContext *s = &s1->mpeg_enc_ctx;
884 const enum AVPixelFormat *pix_fmts;
885
886 if (CONFIG_GRAY && (avctx->flags & AV_CODEC_FLAG_GRAY))
887 return AV_PIX_FMT_GRAY8;
888
889
2/2
✓ Branch 0 taken 202 times.
✓ Branch 1 taken 61 times.
263 if (s->chroma_format < 2)
890 202 pix_fmts = avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO ?
891
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 176 times.
202 mpeg1_hwaccel_pixfmt_list_420 :
892 mpeg2_hwaccel_pixfmt_list_420;
893
1/2
✓ Branch 0 taken 61 times.
✗ Branch 1 not taken.
61 else if (s->chroma_format == 2)
894 61 pix_fmts = mpeg12_pixfmt_list_422;
895 else
896 pix_fmts = mpeg12_pixfmt_list_444;
897
898 263 return ff_get_format(avctx, pix_fmts);
899 }
900
901 /* Call this function when we know all parameters.
902 * It may be called in different places for MPEG-1 and MPEG-2. */
903 5053 static int mpeg_decode_postinit(AVCodecContext *avctx)
904 {
905 5053 Mpeg1Context *s1 = avctx->priv_data;
906 5053 MpegEncContext *s = &s1->mpeg_enc_ctx;
907 int ret;
908
909
2/2
✓ Branch 0 taken 494 times.
✓ Branch 1 taken 4559 times.
5053 if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
910 // MPEG-1 aspect
911 494 AVRational aspect_inv = av_d2q(ff_mpeg1_aspect[s1->aspect_ratio_info], 255);
912 494 avctx->sample_aspect_ratio = (AVRational) { aspect_inv.den, aspect_inv.num };
913 } else { // MPEG-2
914 // MPEG-2 aspect
915
2/2
✓ Branch 0 taken 2519 times.
✓ Branch 1 taken 2040 times.
4559 if (s1->aspect_ratio_info > 1) {
916 AVRational dar =
917 2519 av_mul_q(av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
918 2519 (AVRational) { s1->pan_scan.width,
919 2519 s1->pan_scan.height }),
920 2519 (AVRational) { s->width, s->height });
921
922 /* We ignore the spec here and guess a bit as reality does not
923 * match the spec, see for example res_change_ffmpeg_aspect.ts
924 * and sequence-display-aspect.mpg.
925 * issue1613, 621, 562 */
926
5/6
✓ Branch 0 taken 583 times.
✓ Branch 1 taken 1936 times.
✓ Branch 2 taken 583 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 27 times.
✓ Branch 5 taken 556 times.
3102 if ((s1->pan_scan.width == 0) || (s1->pan_scan.height == 0) ||
927
2/2
✓ Branch 1 taken 26 times.
✓ Branch 2 taken 1 times.
610 (av_cmp_q(dar, (AVRational) { 4, 3 }) &&
928 27 av_cmp_q(dar, (AVRational) { 16, 9 }))) {
929 1962 s->avctx->sample_aspect_ratio =
930 1962 av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
931 1962 (AVRational) { s->width, s->height });
932 } else {
933 557 s->avctx->sample_aspect_ratio =
934 557 av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
935 557 (AVRational) { s1->pan_scan.width, s1->pan_scan.height });
936 // issue1613 4/3 16/9 -> 16/9
937 // res_change_ffmpeg_aspect.ts 4/3 225/44 ->4/3
938 // widescreen-issue562.mpg 4/3 16/9 -> 16/9
939 // s->avctx->sample_aspect_ratio = av_mul_q(s->avctx->sample_aspect_ratio, (AVRational) {s->width, s->height});
940 ff_dlog(avctx, "aspect A %d/%d\n",
941 ff_mpeg2_aspect[s1->aspect_ratio_info].num,
942 ff_mpeg2_aspect[s1->aspect_ratio_info].den);
943 ff_dlog(avctx, "aspect B %d/%d\n", s->avctx->sample_aspect_ratio.num,
944 s->avctx->sample_aspect_ratio.den);
945 }
946 } else {
947 2040 s->avctx->sample_aspect_ratio =
948 2040 ff_mpeg2_aspect[s1->aspect_ratio_info];
949 }
950 } // MPEG-2
951
952
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 5053 times.
5053 if (av_image_check_sar(s->width, s->height,
953 avctx->sample_aspect_ratio) < 0) {
954 av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
955 avctx->sample_aspect_ratio.num,
956 avctx->sample_aspect_ratio.den);
957 avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
958 }
959
960
2/2
✓ Branch 0 taken 4794 times.
✓ Branch 1 taken 259 times.
5053 if (!s->context_initialized ||
961
1/2
✓ Branch 0 taken 4794 times.
✗ Branch 1 not taken.
4794 avctx->coded_width != s->width ||
962
1/2
✓ Branch 0 taken 4794 times.
✗ Branch 1 not taken.
4794 avctx->coded_height != s->height ||
963
1/2
✓ Branch 0 taken 4794 times.
✗ Branch 1 not taken.
4794 s1->save_width != s->width ||
964
3/4
✓ Branch 0 taken 4794 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4792 times.
✓ Branch 3 taken 2 times.
9588 s1->save_height != s->height ||
965 4794 av_cmp_q(s1->save_aspect, s->avctx->sample_aspect_ratio) ||
966
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4792 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
4792 (s1->save_progressive_seq != s->progressive_sequence && FFALIGN(s->height, 16) != FFALIGN(s->height, 32)) ||
967 0) {
968
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 259 times.
261 if (s->context_initialized)
969 2 ff_mpv_common_end(s);
970
971 261 ret = ff_set_dimensions(avctx, s->width, s->height);
972
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 261 times.
261 if (ret < 0)
973 return ret;
974
975
4/4
✓ Branch 0 taken 236 times.
✓ Branch 1 taken 25 times.
✓ Branch 2 taken 234 times.
✓ Branch 3 taken 2 times.
261 if (avctx->codec_id == AV_CODEC_ID_MPEG2VIDEO && s->bit_rate &&
976
2/2
✓ Branch 0 taken 99 times.
✓ Branch 1 taken 135 times.
234 (s->bit_rate != 0x3FFFF*400)) {
977 99 avctx->rc_max_rate = s->bit_rate;
978
3/4
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 137 times.
✓ Branch 2 taken 25 times.
✗ Branch 3 not taken.
162 } else if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO && s->bit_rate &&
979
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)) {
980 25 avctx->bit_rate = s->bit_rate;
981 }
982 261 s1->save_aspect = s->avctx->sample_aspect_ratio;
983 261 s1->save_width = s->width;
984 261 s1->save_height = s->height;
985 261 s1->save_progressive_seq = s->progressive_sequence;
986
987 /* low_delay may be forced, in this case we will have B-frames
988 * that behave like P-frames. */
989 261 avctx->has_b_frames = !s->low_delay;
990
991
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 236 times.
261 if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
992 // MPEG-1 fps
993 25 avctx->framerate = ff_mpeg12_frame_rate_tab[s1->frame_rate_index];
994 #if FF_API_TICKS_PER_FRAME
995 FF_DISABLE_DEPRECATION_WARNINGS
996 25 avctx->ticks_per_frame = 1;
997 FF_ENABLE_DEPRECATION_WARNINGS
998 #endif
999
1000 25 avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
1001 } else { // MPEG-2
1002 // MPEG-2 fps
1003 236 av_reduce(&s->avctx->framerate.num,
1004 236 &s->avctx->framerate.den,
1005 236 ff_mpeg12_frame_rate_tab[s1->frame_rate_index].num * s1->frame_rate_ext.num,
1006 236 ff_mpeg12_frame_rate_tab[s1->frame_rate_index].den * s1->frame_rate_ext.den,
1007 1 << 30);
1008 #if FF_API_TICKS_PER_FRAME
1009 FF_DISABLE_DEPRECATION_WARNINGS
1010 236 avctx->ticks_per_frame = 2;
1011 FF_ENABLE_DEPRECATION_WARNINGS
1012 #endif
1013
1014
2/3
✓ Branch 0 taken 175 times.
✓ Branch 1 taken 61 times.
✗ Branch 2 not taken.
236 switch (s->chroma_format) {
1015 175 case 1: avctx->chroma_sample_location = AVCHROMA_LOC_LEFT; break;
1016 61 case 2:
1017 61 case 3: avctx->chroma_sample_location = AVCHROMA_LOC_TOPLEFT; break;
1018 default: av_assert0(0);
1019 }
1020 } // MPEG-2
1021
1022 261 avctx->pix_fmt = mpeg_get_pixelformat(avctx);
1023
1024
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 261 times.
261 if ((ret = ff_mpv_common_init(s)) < 0)
1025 return ret;
1026 }
1027 5053 return 0;
1028 }
1029
1030 5053 static int mpeg1_decode_picture(AVCodecContext *avctx, const uint8_t *buf,
1031 int buf_size)
1032 {
1033 5053 Mpeg1Context *s1 = avctx->priv_data;
1034 5053 MpegEncContext *s = &s1->mpeg_enc_ctx;
1035 int ref, f_code, vbv_delay, ret;
1036
1037 5053 ret = init_get_bits8(&s->gb, buf, buf_size);
1038
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5053 times.
5053 if (ret < 0)
1039 return ret;
1040
1041 5053 ref = get_bits(&s->gb, 10); /* temporal ref */
1042 5053 s->pict_type = get_bits(&s->gb, 3);
1043
2/4
✓ Branch 0 taken 5053 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 5053 times.
5053 if (s->pict_type == 0 || s->pict_type > 3)
1044 return AVERROR_INVALIDDATA;
1045
1046 5053 vbv_delay = get_bits(&s->gb, 16);
1047 5053 s->vbv_delay = vbv_delay;
1048
2/2
✓ Branch 0 taken 2329 times.
✓ Branch 1 taken 2724 times.
5053 if (s->pict_type == AV_PICTURE_TYPE_P ||
1049
2/2
✓ Branch 0 taken 1617 times.
✓ Branch 1 taken 712 times.
2329 s->pict_type == AV_PICTURE_TYPE_B) {
1050 4341 s->full_pel[0] = get_bits1(&s->gb);
1051 4341 f_code = get_bits(&s->gb, 3);
1052
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4341 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
4341 if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1053 return AVERROR_INVALIDDATA;
1054 4341 f_code += !f_code;
1055 4341 s->mpeg_f_code[0][0] = f_code;
1056 4341 s->mpeg_f_code[0][1] = f_code;
1057 }
1058
2/2
✓ Branch 0 taken 1617 times.
✓ Branch 1 taken 3436 times.
5053 if (s->pict_type == AV_PICTURE_TYPE_B) {
1059 1617 s->full_pel[1] = get_bits1(&s->gb);
1060 1617 f_code = get_bits(&s->gb, 3);
1061
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1617 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
1617 if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1062 return AVERROR_INVALIDDATA;
1063 1617 f_code += !f_code;
1064 1617 s->mpeg_f_code[1][0] = f_code;
1065 1617 s->mpeg_f_code[1][1] = f_code;
1066 }
1067
1068
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5053 times.
5053 if (avctx->debug & FF_DEBUG_PICT_INFO)
1069 av_log(avctx, AV_LOG_DEBUG,
1070 "vbv_delay %d, ref %d type:%d\n", vbv_delay, ref, s->pict_type);
1071
1072 5053 s->y_dc_scale = 8;
1073 5053 s->c_dc_scale = 8;
1074 5053 return 0;
1075 }
1076
1077 718 static void mpeg_decode_sequence_extension(Mpeg1Context *s1)
1078 {
1079 718 MpegEncContext *s = &s1->mpeg_enc_ctx;
1080 int horiz_size_ext, vert_size_ext;
1081 int bit_rate_ext;
1082
1083 718 skip_bits(&s->gb, 1); /* profile and level esc*/
1084 718 s->avctx->profile = get_bits(&s->gb, 3);
1085 718 s->avctx->level = get_bits(&s->gb, 4);
1086 718 s->progressive_sequence = get_bits1(&s->gb); /* progressive_sequence */
1087 718 s->chroma_format = get_bits(&s->gb, 2); /* chroma_format 1=420, 2=422, 3=444 */
1088
1089
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 718 times.
718 if (!s->chroma_format) {
1090 s->chroma_format = 1;
1091 av_log(s->avctx, AV_LOG_WARNING, "Chroma format invalid\n");
1092 }
1093
1094 718 horiz_size_ext = get_bits(&s->gb, 2);
1095 718 vert_size_ext = get_bits(&s->gb, 2);
1096 718 s->width |= (horiz_size_ext << 12);
1097 718 s->height |= (vert_size_ext << 12);
1098 718 bit_rate_ext = get_bits(&s->gb, 12); /* XXX: handle it */
1099 718 s->bit_rate += (bit_rate_ext << 18) * 400LL;
1100 718 check_marker(s->avctx, &s->gb, "after bit rate extension");
1101 718 s->avctx->rc_buffer_size += get_bits(&s->gb, 8) * 1024 * 16 << 10;
1102
1103 718 s->low_delay = get_bits1(&s->gb);
1104
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 718 times.
718 if (s->avctx->flags & AV_CODEC_FLAG_LOW_DELAY)
1105 s->low_delay = 1;
1106
1107 718 s1->frame_rate_ext.num = get_bits(&s->gb, 2) + 1;
1108 718 s1->frame_rate_ext.den = get_bits(&s->gb, 5) + 1;
1109
1110 ff_dlog(s->avctx, "sequence extension\n");
1111 718 s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
1112
1113
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 718 times.
718 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1114 av_log(s->avctx, AV_LOG_DEBUG,
1115 "profile: %d, level: %d ps: %d cf:%d vbv buffer: %d, bitrate:%"PRId64"\n",
1116 s->avctx->profile, s->avctx->level, s->progressive_sequence, s->chroma_format,
1117 s->avctx->rc_buffer_size, s->bit_rate);
1118 718 }
1119
1120 95 static void mpeg_decode_sequence_display_extension(Mpeg1Context *s1)
1121 {
1122 95 MpegEncContext *s = &s1->mpeg_enc_ctx;
1123 int color_description, w, h;
1124
1125 95 skip_bits(&s->gb, 3); /* video format */
1126 95 color_description = get_bits1(&s->gb);
1127
2/2
✓ Branch 0 taken 33 times.
✓ Branch 1 taken 62 times.
95 if (color_description) {
1128 33 s->avctx->color_primaries = get_bits(&s->gb, 8);
1129 33 s->avctx->color_trc = get_bits(&s->gb, 8);
1130 33 s->avctx->colorspace = get_bits(&s->gb, 8);
1131 }
1132 95 w = get_bits(&s->gb, 14);
1133 95 skip_bits(&s->gb, 1); // marker
1134 95 h = get_bits(&s->gb, 14);
1135 // remaining 3 bits are zero padding
1136
1137 95 s1->pan_scan.width = 16 * w;
1138 95 s1->pan_scan.height = 16 * h;
1139
1140
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 95 times.
95 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1141 av_log(s->avctx, AV_LOG_DEBUG, "sde w:%d, h:%d\n", w, h);
1142 95 }
1143
1144 27 static void mpeg_decode_picture_display_extension(Mpeg1Context *s1)
1145 {
1146 27 MpegEncContext *s = &s1->mpeg_enc_ctx;
1147 int i, nofco;
1148
1149 27 nofco = 1;
1150
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27 times.
27 if (s->progressive_sequence) {
1151 if (s->repeat_first_field) {
1152 nofco++;
1153 if (s->top_field_first)
1154 nofco++;
1155 }
1156 } else {
1157
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 18 times.
27 if (s->picture_structure == PICT_FRAME) {
1158 9 nofco++;
1159
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 3 times.
9 if (s->repeat_first_field)
1160 6 nofco++;
1161 }
1162 }
1163
2/2
✓ Branch 0 taken 42 times.
✓ Branch 1 taken 27 times.
69 for (i = 0; i < nofco; i++) {
1164 42 s1->pan_scan.position[i][0] = get_sbits(&s->gb, 16);
1165 42 skip_bits(&s->gb, 1); // marker
1166 42 s1->pan_scan.position[i][1] = get_sbits(&s->gb, 16);
1167 42 skip_bits(&s->gb, 1); // marker
1168 }
1169
1170
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27 times.
27 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1171 av_log(s->avctx, AV_LOG_DEBUG,
1172 "pde (%"PRId16",%"PRId16") (%"PRId16",%"PRId16") (%"PRId16",%"PRId16")\n",
1173 s1->pan_scan.position[0][0], s1->pan_scan.position[0][1],
1174 s1->pan_scan.position[1][0], s1->pan_scan.position[1][1],
1175 s1->pan_scan.position[2][0], s1->pan_scan.position[2][1]);
1176 27 }
1177
1178 223 static int load_matrix(MpegEncContext *s, uint16_t matrix0[64],
1179 uint16_t matrix1[64], int intra)
1180 {
1181 int i;
1182
1183
2/2
✓ Branch 0 taken 14272 times.
✓ Branch 1 taken 223 times.
14495 for (i = 0; i < 64; i++) {
1184 14272 int j = s->idsp.idct_permutation[ff_zigzag_direct[i]];
1185 14272 int v = get_bits(&s->gb, 8);
1186
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14272 times.
14272 if (v == 0) {
1187 av_log(s->avctx, AV_LOG_ERROR, "matrix damaged\n");
1188 return AVERROR_INVALIDDATA;
1189 }
1190
5/6
✓ Branch 0 taken 4160 times.
✓ Branch 1 taken 10112 times.
✓ Branch 2 taken 65 times.
✓ Branch 3 taken 4095 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 65 times.
14272 if (intra && i == 0 && v != 8) {
1191 av_log(s->avctx, AV_LOG_DEBUG, "intra matrix specifies invalid DC quantizer %d, ignoring\n", v);
1192 v = 8; // needed by pink.mpg / issue1046
1193 }
1194 14272 matrix0[j] = v;
1195
2/2
✓ Branch 0 taken 13632 times.
✓ Branch 1 taken 640 times.
14272 if (matrix1)
1196 13632 matrix1[j] = v;
1197 }
1198 223 return 0;
1199 }
1200
1201 977 static void mpeg_decode_quant_matrix_extension(MpegEncContext *s)
1202 {
1203 ff_dlog(s->avctx, "matrix extension\n");
1204
1205
2/2
✓ Branch 1 taken 34 times.
✓ Branch 2 taken 943 times.
977 if (get_bits1(&s->gb))
1206 34 load_matrix(s, s->chroma_intra_matrix, s->intra_matrix, 1);
1207
2/2
✓ Branch 1 taken 73 times.
✓ Branch 2 taken 904 times.
977 if (get_bits1(&s->gb))
1208 73 load_matrix(s, s->chroma_inter_matrix, s->inter_matrix, 0);
1209
2/2
✓ Branch 1 taken 9 times.
✓ Branch 2 taken 968 times.
977 if (get_bits1(&s->gb))
1210 9 load_matrix(s, s->chroma_intra_matrix, NULL, 1);
1211
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 976 times.
977 if (get_bits1(&s->gb))
1212 1 load_matrix(s, s->chroma_inter_matrix, NULL, 0);
1213 977 }
1214
1215 4559 static int mpeg_decode_picture_coding_extension(Mpeg1Context *s1)
1216 {
1217 4559 MpegEncContext *s = &s1->mpeg_enc_ctx;
1218
1219 4559 s->full_pel[0] = s->full_pel[1] = 0;
1220 4559 s->mpeg_f_code[0][0] = get_bits(&s->gb, 4);
1221 4559 s->mpeg_f_code[0][1] = get_bits(&s->gb, 4);
1222 4559 s->mpeg_f_code[1][0] = get_bits(&s->gb, 4);
1223 4559 s->mpeg_f_code[1][1] = get_bits(&s->gb, 4);
1224 4559 s->mpeg_f_code[0][0] += !s->mpeg_f_code[0][0];
1225 4559 s->mpeg_f_code[0][1] += !s->mpeg_f_code[0][1];
1226 4559 s->mpeg_f_code[1][0] += !s->mpeg_f_code[1][0];
1227 4559 s->mpeg_f_code[1][1] += !s->mpeg_f_code[1][1];
1228
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4559 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
4559 if (!s->pict_type && s->context_initialized) {
1229 av_log(s->avctx, AV_LOG_ERROR, "Missing picture start code\n");
1230 if (s->avctx->err_recognition & AV_EF_EXPLODE)
1231 return AVERROR_INVALIDDATA;
1232 av_log(s->avctx, AV_LOG_WARNING, "Guessing pict_type from mpeg_f_code\n");
1233 if (s->mpeg_f_code[1][0] == 15 && s->mpeg_f_code[1][1] == 15) {
1234 if (s->mpeg_f_code[0][0] == 15 && s->mpeg_f_code[0][1] == 15)
1235 s->pict_type = AV_PICTURE_TYPE_I;
1236 else
1237 s->pict_type = AV_PICTURE_TYPE_P;
1238 } else
1239 s->pict_type = AV_PICTURE_TYPE_B;
1240 }
1241
1242 4559 s->intra_dc_precision = get_bits(&s->gb, 2);
1243 4559 s->picture_structure = get_bits(&s->gb, 2);
1244 4559 s->top_field_first = get_bits1(&s->gb);
1245 4559 s->frame_pred_frame_dct = get_bits1(&s->gb);
1246 4559 s->concealment_motion_vectors = get_bits1(&s->gb);
1247 4559 s->q_scale_type = get_bits1(&s->gb);
1248 4559 s->intra_vlc_format = get_bits1(&s->gb);
1249 4559 s->alternate_scan = get_bits1(&s->gb);
1250 4559 s->repeat_first_field = get_bits1(&s->gb);
1251 4559 s->chroma_420_type = get_bits1(&s->gb);
1252 4559 s->progressive_frame = get_bits1(&s->gb);
1253
1254
2/2
✓ Branch 0 taken 1278 times.
✓ Branch 1 taken 3281 times.
4559 if (s->alternate_scan) {
1255 1278 ff_init_scantable(s->idsp.idct_permutation, &s->inter_scantable, ff_alternate_vertical_scan);
1256 1278 ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable, ff_alternate_vertical_scan);
1257 } else {
1258 3281 ff_init_scantable(s->idsp.idct_permutation, &s->inter_scantable, ff_zigzag_direct);
1259 3281 ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable, ff_zigzag_direct);
1260 }
1261
1262 /* composite display not parsed */
1263 ff_dlog(s->avctx, "intra_dc_precision=%d\n", s->intra_dc_precision);
1264 ff_dlog(s->avctx, "picture_structure=%d\n", s->picture_structure);
1265 ff_dlog(s->avctx, "top field first=%d\n", s->top_field_first);
1266 ff_dlog(s->avctx, "repeat first field=%d\n", s->repeat_first_field);
1267 ff_dlog(s->avctx, "conceal=%d\n", s->concealment_motion_vectors);
1268 ff_dlog(s->avctx, "intra_vlc_format=%d\n", s->intra_vlc_format);
1269 ff_dlog(s->avctx, "alternate_scan=%d\n", s->alternate_scan);
1270 ff_dlog(s->avctx, "frame_pred_frame_dct=%d\n", s->frame_pred_frame_dct);
1271 ff_dlog(s->avctx, "progressive_frame=%d\n", s->progressive_frame);
1272
1273 4559 return 0;
1274 }
1275
1276 4608 static int mpeg_field_start(MpegEncContext *s, const uint8_t *buf, int buf_size)
1277 {
1278 4608 AVCodecContext *avctx = s->avctx;
1279 4608 Mpeg1Context *s1 = (Mpeg1Context *) s;
1280 int ret;
1281
1282
1/2
✓ Branch 0 taken 4608 times.
✗ Branch 1 not taken.
4608 if (!(avctx->flags2 & AV_CODEC_FLAG2_CHUNKS)) {
1283
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4608 times.
4608 if (s->mb_width * s->mb_height * 11LL / (33 * 2 * 8) > buf_size)
1284 return AVERROR_INVALIDDATA;
1285 }
1286
1287 /* start frame decoding */
1288
4/4
✓ Branch 0 taken 3898 times.
✓ Branch 1 taken 710 times.
✓ Branch 2 taken 3188 times.
✓ Branch 3 taken 710 times.
8506 if (s->first_field || s->picture_structure == PICT_FRAME) {
1289 AVFrameSideData *pan_scan;
1290
1291
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3898 times.
3898 if ((ret = ff_mpv_frame_start(s, avctx)) < 0)
1292 return ret;
1293
1294
2/2
✓ Branch 0 taken 710 times.
✓ Branch 1 taken 3188 times.
3898 if (s->picture_structure != PICT_FRAME) {
1295 710 s->current_picture_ptr->f->flags |= AV_FRAME_FLAG_TOP_FIELD_FIRST *
1296
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 708 times.
710 (s->picture_structure == PICT_TOP_FIELD);
1297
1298
2/2
✓ Branch 0 taken 2130 times.
✓ Branch 1 taken 710 times.
2840 for (int i = 0; i < 3; i++) {
1299
2/2
✓ Branch 0 taken 2124 times.
✓ Branch 1 taken 6 times.
2130 if (s->picture_structure == PICT_BOTTOM_FIELD) {
1300
1/2
✓ Branch 0 taken 2124 times.
✗ Branch 1 not taken.
2124 s->current_picture.f->data[i] = FF_PTR_ADD(s->current_picture.f->data[i],
1301 s->current_picture.f->linesize[i]);
1302 }
1303 2130 s->current_picture.f->linesize[i] *= 2;
1304 2130 s->last_picture.f->linesize[i] *= 2;
1305 2130 s->next_picture.f->linesize[i] *= 2;
1306 }
1307 }
1308
1309 3898 ff_mpeg_er_frame_start(s);
1310
1311 /* first check if we must repeat the frame */
1312 3898 s->current_picture_ptr->f->repeat_pict = 0;
1313
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 3892 times.
3898 if (s->repeat_first_field) {
1314
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
6 if (s->progressive_sequence) {
1315 if (s->top_field_first)
1316 s->current_picture_ptr->f->repeat_pict = 4;
1317 else
1318 s->current_picture_ptr->f->repeat_pict = 2;
1319
1/2
✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
6 } else if (s->progressive_frame) {
1320 6 s->current_picture_ptr->f->repeat_pict = 1;
1321 }
1322 }
1323
1324 3898 ret = ff_frame_new_side_data(s->avctx, s->current_picture_ptr->f,
1325 AV_FRAME_DATA_PANSCAN, sizeof(s1->pan_scan),
1326 &pan_scan);
1327
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3898 times.
3898 if (ret < 0)
1328 return ret;
1329
1/2
✓ Branch 0 taken 3898 times.
✗ Branch 1 not taken.
3898 if (pan_scan)
1330 3898 memcpy(pan_scan->data, &s1->pan_scan, sizeof(s1->pan_scan));
1331
1332
2/2
✓ Branch 0 taken 742 times.
✓ Branch 1 taken 3156 times.
3898 if (s1->a53_buf_ref) {
1333 742 ret = ff_frame_new_side_data_from_buf(
1334 742 s->avctx, s->current_picture_ptr->f, AV_FRAME_DATA_A53_CC,
1335 &s1->a53_buf_ref, NULL);
1336
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 742 times.
742 if (ret < 0)
1337 return ret;
1338 }
1339
1340
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3898 times.
3898 if (s1->has_stereo3d) {
1341 AVStereo3D *stereo = av_stereo3d_create_side_data(s->current_picture_ptr->f);
1342 if (!stereo)
1343 return AVERROR(ENOMEM);
1344
1345 *stereo = s1->stereo3d;
1346 s1->has_stereo3d = 0;
1347 }
1348
1349
2/2
✓ Branch 0 taken 202 times.
✓ Branch 1 taken 3696 times.
3898 if (s1->has_afd) {
1350 AVFrameSideData *sd;
1351 202 ret = ff_frame_new_side_data(s->avctx, s->current_picture_ptr->f,
1352 AV_FRAME_DATA_AFD, 1, &sd);
1353
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 202 times.
202 if (ret < 0)
1354 return ret;
1355
1/2
✓ Branch 0 taken 202 times.
✗ Branch 1 not taken.
202 if (sd)
1356 202 *sd->data = s1->afd;
1357 202 s1->has_afd = 0;
1358 }
1359
1360
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3898 times.
3898 if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_FRAME))
1361 ff_thread_finish_setup(avctx);
1362 } else { // second field
1363
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 710 times.
710 if (!s->current_picture_ptr) {
1364 av_log(s->avctx, AV_LOG_ERROR, "first field missing\n");
1365 return AVERROR_INVALIDDATA;
1366 }
1367
1368
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 710 times.
710 if (s->avctx->hwaccel) {
1369 if ((ret = FF_HW_SIMPLE_CALL(s->avctx, end_frame)) < 0) {
1370 av_log(avctx, AV_LOG_ERROR,
1371 "hardware accelerator failed to decode first field\n");
1372 return ret;
1373 }
1374 }
1375
1376
2/2
✓ Branch 0 taken 2130 times.
✓ Branch 1 taken 710 times.
2840 for (int i = 0; i < 3; i++) {
1377 2130 s->current_picture.f->data[i] = s->current_picture_ptr->f->data[i];
1378
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 2124 times.
2130 if (s->picture_structure == PICT_BOTTOM_FIELD)
1379 6 s->current_picture.f->data[i] +=
1380 6 s->current_picture_ptr->f->linesize[i];
1381 }
1382 }
1383
1384
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4608 times.
4608 if (avctx->hwaccel) {
1385 if ((ret = FF_HW_CALL(avctx, start_frame, buf, buf_size)) < 0)
1386 return ret;
1387 }
1388
1389 4608 return 0;
1390 }
1391
1392 #define DECODE_SLICE_ERROR -1
1393 #define DECODE_SLICE_OK 0
1394
1395 /**
1396 * Decode a slice.
1397 * MpegEncContext.mb_y must be set to the MB row from the startcode.
1398 * @return DECODE_SLICE_ERROR if the slice is damaged,
1399 * DECODE_SLICE_OK if this slice is OK
1400 */
1401 127021 static int mpeg_decode_slice(MpegEncContext *s, int mb_y,
1402 const uint8_t **buf, int buf_size)
1403 {
1404 127021 AVCodecContext *avctx = s->avctx;
1405 127021 const int lowres = s->avctx->lowres;
1406 127021 const int field_pic = s->picture_structure != PICT_FRAME;
1407 int ret;
1408
1409 127021 s->resync_mb_x =
1410 127021 s->resync_mb_y = -1;
1411
1412
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 127021 times.
127021 av_assert0(mb_y < s->mb_height);
1413
1414 127021 ret = init_get_bits8(&s->gb, *buf, buf_size);
1415
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 127021 times.
127021 if (ret < 0)
1416 return ret;
1417
1418
3/4
✓ Branch 0 taken 119748 times.
✓ Branch 1 taken 7273 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 119748 times.
127021 if (s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
1419 skip_bits(&s->gb, 3);
1420
1421 127021 ff_mpeg1_clean_buffers(s);
1422 127021 s->interlaced_dct = 0;
1423
1424 127021 s->qscale = mpeg_get_qscale(s);
1425
1426
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 127021 times.
127021 if (s->qscale == 0) {
1427 av_log(s->avctx, AV_LOG_ERROR, "qscale == 0\n");
1428 return AVERROR_INVALIDDATA;
1429 }
1430
1431 /* extra slice info */
1432
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 127021 times.
127021 if (skip_1stop_8data_bits(&s->gb) < 0)
1433 return AVERROR_INVALIDDATA;
1434
1435 127021 s->mb_x = 0;
1436
1437
3/4
✓ Branch 0 taken 5393 times.
✓ Branch 1 taken 121628 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 5393 times.
127021 if (mb_y == 0 && s->codec_tag == AV_RL32("SLIF")) {
1438 skip_bits1(&s->gb);
1439 } else {
1440
1/2
✓ Branch 1 taken 140019 times.
✗ Branch 2 not taken.
267040 while (get_bits_left(&s->gb) > 0) {
1441 140019 int code = get_vlc2(&s->gb, ff_mbincr_vlc,
1442 MBINCR_VLC_BITS, 2);
1443
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 140019 times.
140019 if (code < 0) {
1444 av_log(s->avctx, AV_LOG_ERROR, "first mb_incr damaged\n");
1445 return AVERROR_INVALIDDATA;
1446 }
1447
2/2
✓ Branch 0 taken 12998 times.
✓ Branch 1 taken 127021 times.
140019 if (code >= 33) {
1448
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12998 times.
12998 if (code == 33)
1449 12998 s->mb_x += 33;
1450 /* otherwise, stuffing, nothing to do */
1451 } else {
1452 127021 s->mb_x += code;
1453 127021 break;
1454 }
1455 }
1456 }
1457
1458
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 127021 times.
127021 if (s->mb_x >= (unsigned) s->mb_width) {
1459 av_log(s->avctx, AV_LOG_ERROR, "initial skip overflow\n");
1460 return AVERROR_INVALIDDATA;
1461 }
1462
1463
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 127021 times.
127021 if (avctx->hwaccel) {
1464 const uint8_t *buf_end, *buf_start = *buf - 4; /* include start_code */
1465 int start_code = -1;
1466 buf_end = avpriv_find_start_code(buf_start + 2, *buf + buf_size, &start_code);
1467 if (buf_end < *buf + buf_size)
1468 buf_end -= 4;
1469 s->mb_y = mb_y;
1470 if (FF_HW_CALL(avctx, decode_slice, buf_start, buf_end - buf_start) < 0)
1471 return DECODE_SLICE_ERROR;
1472 *buf = buf_end;
1473 return DECODE_SLICE_OK;
1474 }
1475
1476 127021 s->resync_mb_x = s->mb_x;
1477 127021 s->resync_mb_y = s->mb_y = mb_y;
1478 127021 s->mb_skip_run = 0;
1479 127021 ff_init_block_index(s);
1480
1481
8/8
✓ Branch 0 taken 5393 times.
✓ Branch 1 taken 121628 times.
✓ Branch 2 taken 3898 times.
✓ Branch 3 taken 1495 times.
✓ Branch 4 taken 3896 times.
✓ Branch 5 taken 2 times.
✓ Branch 6 taken 3188 times.
✓ Branch 7 taken 708 times.
127021 if (s->mb_y == 0 && s->mb_x == 0 && (s->first_field || s->picture_structure == PICT_FRAME)) {
1482
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3190 times.
3190 if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
1483 av_log(s->avctx, AV_LOG_DEBUG,
1484 "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",
1485 s->qscale,
1486 s->mpeg_f_code[0][0], s->mpeg_f_code[0][1],
1487 s->mpeg_f_code[1][0], s->mpeg_f_code[1][1],
1488 s->pict_type == AV_PICTURE_TYPE_I ? 'I' :
1489 (s->pict_type == AV_PICTURE_TYPE_P ? 'P' :
1490 (s->pict_type == AV_PICTURE_TYPE_B ? 'B' : 'S')),
1491 s->progressive_sequence ? "ps" : "",
1492 s->progressive_frame ? "pf" : "",
1493 s->alternate_scan ? "alt" : "",
1494 s->top_field_first ? "top" : "",
1495 s->intra_dc_precision, s->picture_structure,
1496 s->frame_pred_frame_dct, s->concealment_motion_vectors,
1497 s->q_scale_type, s->intra_vlc_format,
1498 s->repeat_first_field, s->chroma_420_type ? "420" : "");
1499 }
1500 }
1501
1502 for (;;) {
1503
2/2
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 2600706 times.
2600716 if ((ret = mpeg_decode_mb(s, s->block)) < 0)
1504 10 return ret;
1505
1506 // Note motion_val is normally NULL unless we want to extract the MVs.
1507
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2600706 times.
2600706 if (s->current_picture.motion_val[0]) {
1508 const int wrap = s->b8_stride;
1509 int xy = s->mb_x * 2 + s->mb_y * 2 * wrap;
1510 int b8_xy = 4 * (s->mb_x + s->mb_y * s->mb_stride);
1511 int motion_x, motion_y, dir, i;
1512
1513 for (i = 0; i < 2; i++) {
1514 for (dir = 0; dir < 2; dir++) {
1515 if (s->mb_intra ||
1516 (dir == 1 && s->pict_type != AV_PICTURE_TYPE_B)) {
1517 motion_x = motion_y = 0;
1518 } else if (s->mv_type == MV_TYPE_16X16 ||
1519 (s->mv_type == MV_TYPE_FIELD && field_pic)) {
1520 motion_x = s->mv[dir][0][0];
1521 motion_y = s->mv[dir][0][1];
1522 } else { /* if ((s->mv_type == MV_TYPE_FIELD) || (s->mv_type == MV_TYPE_16X8)) */
1523 motion_x = s->mv[dir][i][0];
1524 motion_y = s->mv[dir][i][1];
1525 }
1526
1527 s->current_picture.motion_val[dir][xy][0] = motion_x;
1528 s->current_picture.motion_val[dir][xy][1] = motion_y;
1529 s->current_picture.motion_val[dir][xy + 1][0] = motion_x;
1530 s->current_picture.motion_val[dir][xy + 1][1] = motion_y;
1531 s->current_picture.ref_index [dir][b8_xy] =
1532 s->current_picture.ref_index [dir][b8_xy + 1] = s->field_select[dir][i];
1533 av_assert2(s->field_select[dir][i] == 0 ||
1534 s->field_select[dir][i] == 1);
1535 }
1536 xy += wrap;
1537 b8_xy += 2;
1538 }
1539 }
1540
1541 2600706 s->dest[0] += 16 >> lowres;
1542 2600706 s->dest[1] +=(16 >> lowres) >> s->chroma_x_shift;
1543 2600706 s->dest[2] +=(16 >> lowres) >> s->chroma_x_shift;
1544
1545 2600706 ff_mpv_reconstruct_mb(s, s->block);
1546
1547
2/2
✓ Branch 0 taken 79110 times.
✓ Branch 1 taken 2521596 times.
2600706 if (++s->mb_x >= s->mb_width) {
1548 79110 const int mb_size = 16 >> s->avctx->lowres;
1549 int left;
1550
1551 79110 ff_mpeg_draw_horiz_band(s, mb_size * (s->mb_y >> field_pic), mb_size);
1552 79110 ff_mpv_report_decode_progress(s);
1553
1554 79110 s->mb_x = 0;
1555 79110 s->mb_y += 1 << field_pic;
1556
1557
2/2
✓ Branch 0 taken 4598 times.
✓ Branch 1 taken 74512 times.
79110 if (s->mb_y >= s->mb_height) {
1558 4598 int left = get_bits_left(&s->gb);
1559 9457 int is_d10 = s->chroma_format == 2 &&
1560
2/2
✓ Branch 0 taken 49 times.
✓ Branch 1 taken 212 times.
261 s->pict_type == AV_PICTURE_TYPE_I &&
1561
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 &&
1562
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 24 times.
49 s->intra_dc_precision == 2 &&
1563
4/6
✓ Branch 0 taken 261 times.
✓ Branch 1 taken 4337 times.
✓ Branch 2 taken 25 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 25 times.
✗ Branch 5 not taken.
4884 s->q_scale_type == 1 && s->alternate_scan == 0 &&
1564
1/2
✓ Branch 0 taken 25 times.
✗ Branch 1 not taken.
25 s->progressive_frame == 0
1565 /* vbv_delay == 0xBBB || 0xE10 */;
1566
1567
4/4
✓ Branch 0 taken 1466 times.
✓ Branch 1 taken 3132 times.
✓ Branch 2 taken 1441 times.
✓ Branch 3 taken 25 times.
4598 if (left >= 32 && !is_d10) {
1568 1441 GetBitContext gb = s->gb;
1569 1441 align_get_bits(&gb);
1570
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1441 times.
1441 if (show_bits(&gb, 24) == 0x060E2B) {
1571 av_log(avctx, AV_LOG_DEBUG, "Invalid MXF data found in video stream\n");
1572 is_d10 = 1;
1573 }
1574
3/4
✓ Branch 0 taken 1440 times.
✓ Branch 1 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 1440 times.
1441 if (left > 32 && show_bits_long(&gb, 32) == 0x201) {
1575 av_log(avctx, AV_LOG_DEBUG, "skipping m704 alpha (unsupported)\n");
1576 goto eos;
1577 }
1578 }
1579
1580
3/4
✓ Branch 0 taken 4598 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4296 times.
✓ Branch 3 taken 302 times.
4598 if (left < 0 ||
1581
1/4
✗ Branch 1 not taken.
✓ Branch 2 taken 4296 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
4296 (left && show_bits(&s->gb, FFMIN(left, 23)) && !is_d10) ||
1582
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4598 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
4598 ((avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_AGGRESSIVE)) && left > 8)) {
1583 av_log(avctx, AV_LOG_ERROR, "end mismatch left=%d %0X at %d %d\n",
1584 left, left>0 ? show_bits(&s->gb, FFMIN(left, 23)) : 0, s->mb_x, s->mb_y);
1585 return AVERROR_INVALIDDATA;
1586 } else
1587 4598 goto eos;
1588 }
1589 // There are some files out there which are missing the last slice
1590 // in cases where the slice is completely outside the visible
1591 // area, we detect this here instead of running into the end expecting
1592 // more data
1593 74512 left = get_bits_left(&s->gb);
1594
2/2
✓ Branch 0 taken 200 times.
✓ Branch 1 taken 74312 times.
74512 if (s->mb_y >= ((s->height + 15) >> 4) &&
1595
2/4
✓ Branch 0 taken 200 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 200 times.
200 !s->progressive_sequence &&
1596 left <= 25 &&
1597 left >= 0 &&
1598 s->mb_skip_run == -1 &&
1599 (!left || show_bits(&s->gb, left) == 0))
1600 goto eos;
1601
1602 74512 ff_init_block_index(s);
1603 }
1604
1605 /* skip mb handling */
1606
2/2
✓ Branch 0 taken 2430777 times.
✓ Branch 1 taken 165331 times.
2596108 if (s->mb_skip_run == -1) {
1607 /* read increment again */
1608 2430777 s->mb_skip_run = 0;
1609 1230 for (;;) {
1610 2432007 int code = get_vlc2(&s->gb, ff_mbincr_vlc,
1611 MBINCR_VLC_BITS, 2);
1612
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2432007 times.
2432007 if (code < 0) {
1613 av_log(s->avctx, AV_LOG_ERROR, "mb incr damaged\n");
1614 return AVERROR_INVALIDDATA;
1615 }
1616
2/2
✓ Branch 0 taken 123643 times.
✓ Branch 1 taken 2308364 times.
2432007 if (code >= 33) {
1617
2/2
✓ Branch 0 taken 1230 times.
✓ Branch 1 taken 122413 times.
123643 if (code == 33) {
1618 1230 s->mb_skip_run += 33;
1619
1/2
✓ Branch 0 taken 122413 times.
✗ Branch 1 not taken.
122413 } else if (code == 35) {
1620
2/4
✓ Branch 0 taken 122413 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 122413 times.
122413 if (s->mb_skip_run != 0 || show_bits(&s->gb, 15) != 0) {
1621 av_log(s->avctx, AV_LOG_ERROR, "slice mismatch\n");
1622 return AVERROR_INVALIDDATA;
1623 }
1624 122413 goto eos; /* end of slice */
1625 }
1626 /* otherwise, stuffing, nothing to do */
1627 } else {
1628 2308364 s->mb_skip_run += code;
1629 2308364 break;
1630 }
1631 }
1632
2/2
✓ Branch 0 taken 61871 times.
✓ Branch 1 taken 2246493 times.
2308364 if (s->mb_skip_run) {
1633 int i;
1634
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 61871 times.
61871 if (s->pict_type == AV_PICTURE_TYPE_I) {
1635 av_log(s->avctx, AV_LOG_ERROR,
1636 "skipped MB in I-frame at %d %d\n", s->mb_x, s->mb_y);
1637 return AVERROR_INVALIDDATA;
1638 }
1639
1640 /* skip mb */
1641 61871 s->mb_intra = 0;
1642
2/2
✓ Branch 0 taken 742452 times.
✓ Branch 1 taken 61871 times.
804323 for (i = 0; i < 12; i++)
1643 742452 s->block_last_index[i] = -1;
1644
2/2
✓ Branch 0 taken 42166 times.
✓ Branch 1 taken 19705 times.
61871 if (s->picture_structure == PICT_FRAME)
1645 42166 s->mv_type = MV_TYPE_16X16;
1646 else
1647 19705 s->mv_type = MV_TYPE_FIELD;
1648
2/2
✓ Branch 0 taken 23749 times.
✓ Branch 1 taken 38122 times.
61871 if (s->pict_type == AV_PICTURE_TYPE_P) {
1649 /* if P type, zero motion vector is implied */
1650 23749 s->mv_dir = MV_DIR_FORWARD;
1651 23749 s->mv[0][0][0] = s->mv[0][0][1] = 0;
1652 23749 s->last_mv[0][0][0] = s->last_mv[0][0][1] = 0;
1653 23749 s->last_mv[0][1][0] = s->last_mv[0][1][1] = 0;
1654 23749 s->field_select[0][0] = (s->picture_structure - 1) & 1;
1655 } else {
1656 /* if B type, reuse previous vectors and directions */
1657 38122 s->mv[0][0][0] = s->last_mv[0][0][0];
1658 38122 s->mv[0][0][1] = s->last_mv[0][0][1];
1659 38122 s->mv[1][0][0] = s->last_mv[1][0][0];
1660 38122 s->mv[1][0][1] = s->last_mv[1][0][1];
1661 38122 s->field_select[0][0] = (s->picture_structure - 1) & 1;
1662 38122 s->field_select[1][0] = (s->picture_structure - 1) & 1;
1663 }
1664 }
1665 }
1666 }
1667 127011 eos: // end of slice
1668
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 127011 times.
127011 if (get_bits_left(&s->gb) < 0) {
1669 av_log(s, AV_LOG_ERROR, "overread %d\n", -get_bits_left(&s->gb));
1670 return AVERROR_INVALIDDATA;
1671 }
1672 127011 *buf += (get_bits_count(&s->gb) - 1) / 8;
1673 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);
1674 127011 return 0;
1675 }
1676
1677 6678 static int slice_decode_thread(AVCodecContext *c, void *arg)
1678 {
1679 6678 MpegEncContext *s = *(void **) arg;
1680 6678 const uint8_t *buf = s->gb.buffer;
1681 6678 int mb_y = s->start_mb_y;
1682 6678 const int field_pic = s->picture_structure != PICT_FRAME;
1683
1684 6678 s->er.error_count = (3 * (s->end_mb_y - s->start_mb_y) * s->mb_width) >> field_pic;
1685
1686 15582 for (;;) {
1687 uint32_t start_code;
1688 int ret;
1689
1690 22260 ret = mpeg_decode_slice(s, mb_y, &buf, s->gb.buffer_end - buf);
1691 22260 emms_c();
1692 ff_dlog(c, "ret:%d resync:%d/%d mb:%d/%d ts:%d/%d ec:%d\n",
1693 ret, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y,
1694 s->start_mb_y, s->end_mb_y, s->er.error_count);
1695
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22260 times.
22260 if (ret < 0) {
1696 if (c->err_recognition & AV_EF_EXPLODE)
1697 6678 return ret;
1698 if (s->resync_mb_x >= 0 && s->resync_mb_y >= 0)
1699 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y,
1700 s->mb_x, s->mb_y,
1701 ER_AC_ERROR | ER_DC_ERROR | ER_MV_ERROR);
1702 } else {
1703 22260 ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y,
1704 22260 s->mb_x - 1, s->mb_y,
1705 ER_AC_END | ER_DC_END | ER_MV_END);
1706 }
1707
1708
2/2
✓ Branch 0 taken 6678 times.
✓ Branch 1 taken 15582 times.
22260 if (s->mb_y == s->end_mb_y)
1709 6678 return 0;
1710
1711 15582 start_code = -1;
1712 15582 buf = avpriv_find_start_code(buf, s->gb.buffer_end, &start_code);
1713
2/4
✓ Branch 0 taken 15582 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 15582 times.
15582 if (start_code < SLICE_MIN_START_CODE || start_code > SLICE_MAX_START_CODE)
1714 return AVERROR_INVALIDDATA;
1715 15582 mb_y = start_code - SLICE_MIN_START_CODE;
1716
2/4
✓ Branch 0 taken 15582 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 15582 times.
15582 if (s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
1717 mb_y += (*buf&0xE0)<<2;
1718 15582 mb_y <<= field_pic;
1719
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15582 times.
15582 if (s->picture_structure == PICT_BOTTOM_FIELD)
1720 mb_y++;
1721
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15582 times.
15582 if (mb_y >= s->end_mb_y)
1722 return AVERROR_INVALIDDATA;
1723 }
1724 }
1725
1726 /**
1727 * Handle slice ends.
1728 * @return 1 if it seems to be the last slice
1729 */
1730 4185 static int slice_end(AVCodecContext *avctx, AVFrame *pict)
1731 {
1732 4185 Mpeg1Context *s1 = avctx->priv_data;
1733 4185 MpegEncContext *s = &s1->mpeg_enc_ctx;
1734
1735
3/4
✓ Branch 0 taken 3899 times.
✓ Branch 1 taken 286 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3899 times.
4185 if (!s->context_initialized || !s->current_picture_ptr)
1736 286 return 0;
1737
1738
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3899 times.
3899 if (s->avctx->hwaccel) {
1739 int ret = FF_HW_SIMPLE_CALL(s->avctx, end_frame);
1740 if (ret < 0) {
1741 av_log(avctx, AV_LOG_ERROR,
1742 "hardware accelerator failed to decode picture\n");
1743 return ret;
1744 }
1745 }
1746
1747 /* end of slice reached */
1748
3/4
✓ Branch 0 taken 3898 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 3898 times.
✗ Branch 3 not taken.
3899 if (/* s->mb_y << field_pic == s->mb_height && */ !s->first_field && !s1->first_slice) {
1749 /* end of image */
1750
1751 3898 ff_er_frame_end(&s->er, NULL);
1752
1753 3898 ff_mpv_frame_end(s);
1754
1755
4/4
✓ Branch 0 taken 2481 times.
✓ Branch 1 taken 1417 times.
✓ Branch 2 taken 533 times.
✓ Branch 3 taken 1948 times.
5848 if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {
1756 1950 int ret = av_frame_ref(pict, s->current_picture_ptr->f);
1757
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1950 times.
1950 if (ret < 0)
1758 return ret;
1759 1950 ff_print_debug_info(s, s->current_picture_ptr, pict);
1760 1950 ff_mpv_export_qp_table(s, pict, s->current_picture_ptr, FF_MPV_QSCALE_TYPE_MPEG2);
1761 } else {
1762 /* latency of 1 frame for I- and P-frames */
1763
2/2
✓ Branch 0 taken 1886 times.
✓ Branch 1 taken 62 times.
1948 if (s->last_picture_ptr) {
1764 1886 int ret = av_frame_ref(pict, s->last_picture_ptr->f);
1765
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1886 times.
1886 if (ret < 0)
1766 return ret;
1767 1886 ff_print_debug_info(s, s->last_picture_ptr, pict);
1768 1886 ff_mpv_export_qp_table(s, pict, s->last_picture_ptr, FF_MPV_QSCALE_TYPE_MPEG2);
1769 }
1770 }
1771
1772 3898 return 1;
1773 } else {
1774 1 return 0;
1775 }
1776 }
1777
1778 810 static int mpeg1_decode_sequence(AVCodecContext *avctx,
1779 const uint8_t *buf, int buf_size)
1780 {
1781 810 Mpeg1Context *s1 = avctx->priv_data;
1782 810 MpegEncContext *s = &s1->mpeg_enc_ctx;
1783 int width, height;
1784 int i, v, j;
1785
1786 810 int ret = init_get_bits8(&s->gb, buf, buf_size);
1787
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 810 times.
810 if (ret < 0)
1788 return ret;
1789
1790 810 width = get_bits(&s->gb, 12);
1791 810 height = get_bits(&s->gb, 12);
1792
2/4
✓ Branch 0 taken 810 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 810 times.
810 if (width == 0 || height == 0) {
1793 av_log(avctx, AV_LOG_WARNING,
1794 "Invalid horizontal or vertical size value.\n");
1795 if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
1796 return AVERROR_INVALIDDATA;
1797 }
1798 810 s1->aspect_ratio_info = get_bits(&s->gb, 4);
1799
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 810 times.
810 if (s1->aspect_ratio_info == 0) {
1800 av_log(avctx, AV_LOG_ERROR, "aspect ratio has forbidden 0 value\n");
1801 if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
1802 return AVERROR_INVALIDDATA;
1803 }
1804 810 s1->frame_rate_index = get_bits(&s->gb, 4);
1805
2/4
✓ Branch 0 taken 810 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 810 times.
810 if (s1->frame_rate_index == 0 || s1->frame_rate_index > 13) {
1806 av_log(avctx, AV_LOG_WARNING,
1807 "frame_rate_index %d is invalid\n", s1->frame_rate_index);
1808 s1->frame_rate_index = 1;
1809 }
1810 810 s->bit_rate = get_bits(&s->gb, 18) * 400LL;
1811
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 810 times.
810 if (check_marker(s->avctx, &s->gb, "in sequence header") == 0) {
1812 return AVERROR_INVALIDDATA;
1813 }
1814
1815 810 s->avctx->rc_buffer_size = get_bits(&s->gb, 10) * 1024 * 16;
1816 810 skip_bits(&s->gb, 1);
1817
1818 /* get matrix */
1819
2/2
✓ Branch 1 taken 22 times.
✓ Branch 2 taken 788 times.
810 if (get_bits1(&s->gb)) {
1820 22 load_matrix(s, s->chroma_intra_matrix, s->intra_matrix, 1);
1821 } else {
1822
2/2
✓ Branch 0 taken 50432 times.
✓ Branch 1 taken 788 times.
51220 for (i = 0; i < 64; i++) {
1823 50432 j = s->idsp.idct_permutation[i];
1824 50432 v = ff_mpeg1_default_intra_matrix[i];
1825 50432 s->intra_matrix[j] = v;
1826 50432 s->chroma_intra_matrix[j] = v;
1827 }
1828 }
1829
2/2
✓ Branch 1 taken 84 times.
✓ Branch 2 taken 726 times.
810 if (get_bits1(&s->gb)) {
1830 84 load_matrix(s, s->chroma_inter_matrix, s->inter_matrix, 0);
1831 } else {
1832
2/2
✓ Branch 0 taken 46464 times.
✓ Branch 1 taken 726 times.
47190 for (i = 0; i < 64; i++) {
1833 46464 int j = s->idsp.idct_permutation[i];
1834 46464 v = ff_mpeg1_default_non_intra_matrix[i];
1835 46464 s->inter_matrix[j] = v;
1836 46464 s->chroma_inter_matrix[j] = v;
1837 }
1838 }
1839
1840
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 810 times.
810 if (show_bits(&s->gb, 23) != 0) {
1841 av_log(s->avctx, AV_LOG_ERROR, "sequence header damaged\n");
1842 return AVERROR_INVALIDDATA;
1843 }
1844
1845 810 s->width = width;
1846 810 s->height = height;
1847
1848 /* We set MPEG-2 parameters so that it emulates MPEG-1. */
1849 810 s->progressive_sequence = 1;
1850 810 s->progressive_frame = 1;
1851 810 s->picture_structure = PICT_FRAME;
1852 810 s->first_field = 0;
1853 810 s->frame_pred_frame_dct = 1;
1854 810 s->chroma_format = 1;
1855 810 s->codec_id =
1856 810 s->avctx->codec_id = AV_CODEC_ID_MPEG1VIDEO;
1857 810 s->out_format = FMT_MPEG1;
1858
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 810 times.
810 if (s->avctx->flags & AV_CODEC_FLAG_LOW_DELAY)
1859 s->low_delay = 1;
1860
1861
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 810 times.
810 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1862 av_log(s->avctx, AV_LOG_DEBUG, "vbv buffer: %d, bitrate:%"PRId64", aspect_ratio_info: %d \n",
1863 s->avctx->rc_buffer_size, s->bit_rate, s1->aspect_ratio_info);
1864
1865 810 return 0;
1866 }
1867
1868 2 static int vcr2_init_sequence(AVCodecContext *avctx)
1869 {
1870 2 Mpeg1Context *s1 = avctx->priv_data;
1871 2 MpegEncContext *s = &s1->mpeg_enc_ctx;
1872 int i, v, ret;
1873
1874 /* start new MPEG-1 context decoding */
1875 2 s->out_format = FMT_MPEG1;
1876
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (s->context_initialized)
1877 ff_mpv_common_end(s);
1878
1879 2 s->width = avctx->coded_width;
1880 2 s->height = avctx->coded_height;
1881 2 avctx->has_b_frames = 0; // true?
1882 2 s->low_delay = 1;
1883
1884 2 avctx->pix_fmt = mpeg_get_pixelformat(avctx);
1885
1886
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
2 if ((ret = ff_mpv_common_init(s)) < 0)
1887 return ret;
1888
1889
2/2
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 2 times.
130 for (i = 0; i < 64; i++) {
1890 128 int j = s->idsp.idct_permutation[i];
1891 128 v = ff_mpeg1_default_intra_matrix[i];
1892 128 s->intra_matrix[j] = v;
1893 128 s->chroma_intra_matrix[j] = v;
1894
1895 128 v = ff_mpeg1_default_non_intra_matrix[i];
1896 128 s->inter_matrix[j] = v;
1897 128 s->chroma_inter_matrix[j] = v;
1898 }
1899
1900 2 s->progressive_sequence = 1;
1901 2 s->progressive_frame = 1;
1902 2 s->picture_structure = PICT_FRAME;
1903 2 s->first_field = 0;
1904 2 s->frame_pred_frame_dct = 1;
1905 2 s->chroma_format = 1;
1906
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (s->codec_tag == AV_RL32("BW10")) {
1907 s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG1VIDEO;
1908 } else {
1909 2 s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
1910 }
1911 2 s1->save_width = s->width;
1912 2 s1->save_height = s->height;
1913 2 s1->save_progressive_seq = s->progressive_sequence;
1914 2 return 0;
1915 }
1916
1917 758 static void mpeg_set_cc_format(AVCodecContext *avctx, enum Mpeg2ClosedCaptionsFormat format,
1918 const char *label)
1919 {
1920 758 Mpeg1Context *s1 = avctx->priv_data;
1921
1922 av_assert2(format != CC_FORMAT_AUTO);
1923
1924
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 747 times.
758 if (!s1->cc_format) {
1925 11 s1->cc_format = format;
1926
1927 11 av_log(avctx, AV_LOG_DEBUG, "CC: first seen substream is %s format\n", label);
1928 }
1929 758 }
1930
1931 830 static int mpeg_decode_a53_cc(AVCodecContext *avctx,
1932 const uint8_t *p, int buf_size)
1933 {
1934 830 Mpeg1Context *s1 = avctx->priv_data;
1935
1936
5/6
✓ Branch 0 taken 747 times.
✓ Branch 1 taken 83 times.
✓ Branch 2 taken 544 times.
✓ Branch 3 taken 203 times.
✓ Branch 4 taken 627 times.
✗ Branch 5 not taken.
830 if ((!s1->cc_format || s1->cc_format == CC_FORMAT_A53_PART4) &&
1937 627 buf_size >= 6 &&
1938
5/8
✓ Branch 0 taken 553 times.
✓ Branch 1 taken 74 times.
✓ Branch 2 taken 553 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 553 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 553 times.
✗ Branch 7 not taken.
627 p[0] == 'G' && p[1] == 'A' && p[2] == '9' && p[3] == '4' &&
1939
2/4
✓ Branch 0 taken 553 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 553 times.
✗ Branch 3 not taken.
553 p[4] == 3 && (p[5] & 0x40)) {
1940 /* extract A53 Part 4 CC data */
1941 553 int cc_count = p[5] & 0x1f;
1942
2/4
✓ Branch 0 taken 553 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 553 times.
✗ Branch 3 not taken.
553 if (cc_count > 0 && buf_size >= 7 + cc_count * 3) {
1943
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 545 times.
553 int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
1944 553 const uint64_t new_size = (old_size + cc_count
1945 553 * UINT64_C(3));
1946 int ret;
1947
1948
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 553 times.
553 if (new_size > 3*A53_MAX_CC_COUNT)
1949 return AVERROR(EINVAL);
1950
1951 553 ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
1952
1/2
✓ Branch 0 taken 553 times.
✗ Branch 1 not taken.
553 if (ret >= 0)
1953 553 memcpy(s1->a53_buf_ref->data + old_size, p + 7, cc_count * UINT64_C(3));
1954
1955 553 avctx->properties |= FF_CODEC_PROPERTY_CLOSED_CAPTIONS;
1956 553 mpeg_set_cc_format(avctx, CC_FORMAT_A53_PART4, "A/53 Part 4");
1957 }
1958 553 return 1;
1959
4/6
✓ Branch 0 taken 203 times.
✓ Branch 1 taken 74 times.
✓ Branch 2 taken 203 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 277 times.
✗ Branch 5 not taken.
277 } else if ((!s1->cc_format || s1->cc_format == CC_FORMAT_SCTE20) &&
1960 277 buf_size >= 2 &&
1961
3/4
✓ Branch 0 taken 205 times.
✓ Branch 1 taken 72 times.
✓ Branch 2 taken 205 times.
✗ Branch 3 not taken.
277 p[0] == 0x03 && (p[1]&0x7f) == 0x01) {
1962 /* extract SCTE-20 CC data */
1963 GetBitContext gb;
1964 205 int cc_count = 0;
1965 int i, ret;
1966
1967 205 ret = init_get_bits8(&gb, p + 2, buf_size - 2);
1968
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 205 times.
205 if (ret < 0)
1969 return ret;
1970 205 cc_count = get_bits(&gb, 5);
1971
1/2
✓ Branch 0 taken 205 times.
✗ Branch 1 not taken.
205 if (cc_count > 0) {
1972
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;
1973 205 const uint64_t new_size = (old_size + cc_count
1974 205 * UINT64_C(3));
1975
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 205 times.
205 if (new_size > 3*A53_MAX_CC_COUNT)
1976 return AVERROR(EINVAL);
1977
1978 205 ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
1979
1/2
✓ Branch 0 taken 205 times.
✗ Branch 1 not taken.
205 if (ret >= 0) {
1980 uint8_t field, cc1, cc2;
1981 205 uint8_t *cap = s1->a53_buf_ref->data;
1982
1983 205 memset(s1->a53_buf_ref->data + old_size, 0, cc_count * 3);
1984
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++) {
1985 410 skip_bits(&gb, 2); // priority
1986 410 field = get_bits(&gb, 2);
1987 410 skip_bits(&gb, 5); // line_offset
1988 410 cc1 = get_bits(&gb, 8);
1989 410 cc2 = get_bits(&gb, 8);
1990 410 skip_bits(&gb, 1); // marker
1991
1992
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 410 times.
410 if (!field) { // forbidden
1993 cap[0] = cap[1] = cap[2] = 0x00;
1994 } else {
1995 410 field = (field == 2 ? 1 : 0);
1996
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 410 times.
410 if (!s1->mpeg_enc_ctx.top_field_first) field = !field;
1997 410 cap[0] = 0x04 | field;
1998 410 cap[1] = ff_reverse[cc1];
1999 410 cap[2] = ff_reverse[cc2];
2000 }
2001 410 cap += 3;
2002 }
2003 }
2004
2005 205 avctx->properties |= FF_CODEC_PROPERTY_CLOSED_CAPTIONS;
2006 205 mpeg_set_cc_format(avctx, CC_FORMAT_SCTE20, "SCTE-20");
2007 }
2008 205 return 1;
2009
2/6
✗ Branch 0 not taken.
✓ Branch 1 taken 72 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 72 times.
✗ Branch 5 not taken.
72 } else if ((!s1->cc_format || s1->cc_format == CC_FORMAT_DVD) &&
2010 72 buf_size >= 11 &&
2011
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 72 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.
72 p[0] == 'C' && p[1] == 'C' && p[2] == 0x01 && p[3] == 0xf8) {
2012 /* extract DVD CC data
2013 *
2014 * uint32_t user_data_start_code 0x000001B2 (big endian)
2015 * uint16_t user_identifier 0x4343 "CC"
2016 * uint8_t user_data_type_code 0x01
2017 * uint8_t caption_block_size 0xF8
2018 * uint8_t
2019 * bit 7 caption_odd_field_first 1=odd field (CC1/CC2) first 0=even field (CC3/CC4) first
2020 * bit 6 caption_filler 0
2021 * bit 5:1 caption_block_count number of caption blocks (pairs of caption words = frames). Most DVDs use 15 per start of GOP.
2022 * bit 0 caption_extra_field_added 1=one additional caption word
2023 *
2024 * struct caption_field_block {
2025 * uint8_t
2026 * bit 7:1 caption_filler 0x7F (all 1s)
2027 * bit 0 caption_field_odd 1=odd field (this is CC1/CC2) 0=even field (this is CC3/CC4)
2028 * uint8_t caption_first_byte
2029 * uint8_t caption_second_byte
2030 * } caption_block[(caption_block_count * 2) + caption_extra_field_added];
2031 *
2032 * Some DVDs encode caption data for both fields with caption_field_odd=1. The only way to decode the fields
2033 * correctly is to start on the field indicated by caption_odd_field_first and count between odd/even fields.
2034 * Don't assume that the first caption word is the odd field. There do exist MPEG files in the wild that start
2035 * on the even field. There also exist DVDs in the wild that encode an odd field count and the
2036 * caption_extra_field_added/caption_odd_field_first bits change per packet to allow that. */
2037 int cc_count = 0;
2038 int i, ret;
2039 // There is a caption count field in the data, but it is often
2040 // incorrect. So count the number of captions present.
2041 for (i = 5; i + 6 <= buf_size && ((p[i] & 0xfe) == 0xfe); i += 6)
2042 cc_count++;
2043 // Transform the DVD format into A53 Part 4 format
2044 if (cc_count > 0) {
2045 int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
2046 const uint64_t new_size = (old_size + cc_count
2047 * UINT64_C(6));
2048 if (new_size > 3*A53_MAX_CC_COUNT)
2049 return AVERROR(EINVAL);
2050
2051 ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
2052 if (ret >= 0) {
2053 uint8_t field1 = !!(p[4] & 0x80);
2054 uint8_t *cap = s1->a53_buf_ref->data;
2055 p += 5;
2056 for (i = 0; i < cc_count; i++) {
2057 cap[0] = (p[0] == 0xff && field1) ? 0xfc : 0xfd;
2058 cap[1] = p[1];
2059 cap[2] = p[2];
2060 cap[3] = (p[3] == 0xff && !field1) ? 0xfc : 0xfd;
2061 cap[4] = p[4];
2062 cap[5] = p[5];
2063 cap += 6;
2064 p += 6;
2065 }
2066 }
2067
2068 avctx->properties |= FF_CODEC_PROPERTY_CLOSED_CAPTIONS;
2069 mpeg_set_cc_format(avctx, CC_FORMAT_DVD, "DVD");
2070 }
2071 return 1;
2072 }
2073 72 return 0;
2074 }
2075
2076 1037 static void mpeg_decode_user_data(AVCodecContext *avctx,
2077 const uint8_t *p, int buf_size)
2078 {
2079 1037 Mpeg1Context *s = avctx->priv_data;
2080 1037 const uint8_t *buf_end = p + buf_size;
2081 1037 Mpeg1Context *s1 = avctx->priv_data;
2082
2083 #if 0
2084 int i;
2085 for(i=0; !(!p[i-2] && !p[i-1] && p[i]==1) && i<buf_size; i++){
2086 av_log(avctx, AV_LOG_ERROR, "%c", p[i]);
2087 }
2088 av_log(avctx, AV_LOG_ERROR, "\n");
2089 #endif
2090
2091
1/2
✓ Branch 0 taken 1037 times.
✗ Branch 1 not taken.
1037 if (buf_size > 29){
2092 int i;
2093
2/2
✓ Branch 0 taken 20740 times.
✓ Branch 1 taken 1037 times.
21777 for(i=0; i<20; i++)
2094
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 20740 times.
20740 if (!memcmp(p+i, "\0TMPGEXS\0", 9)){
2095 s->tmpgexs= 1;
2096 }
2097 }
2098 /* we parse the DTG active format information */
2099
1/2
✓ Branch 0 taken 1037 times.
✗ Branch 1 not taken.
1037 if (buf_end - p >= 5 &&
2100
5/8
✓ Branch 0 taken 207 times.
✓ Branch 1 taken 830 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.
1244 p[0] == 'D' && p[1] == 'T' && p[2] == 'G' && p[3] == '1') {
2101 207 int flags = p[4];
2102 207 p += 5;
2103
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 207 times.
207 if (flags & 0x80) {
2104 /* skip event id */
2105 p += 2;
2106 }
2107
1/2
✓ Branch 0 taken 207 times.
✗ Branch 1 not taken.
207 if (flags & 0x40) {
2108
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 207 times.
207 if (buf_end - p < 1)
2109 return;
2110 207 s1->has_afd = 1;
2111 207 s1->afd = p[0] & 0x0f;
2112 }
2113
1/2
✓ Branch 0 taken 830 times.
✗ Branch 1 not taken.
830 } else if (buf_end - p >= 6 &&
2114
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 830 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.
830 p[0] == 'J' && p[1] == 'P' && p[2] == '3' && p[3] == 'D' &&
2115 p[4] == 0x03) { // S3D_video_format_length
2116 // the 0x7F mask ignores the reserved_bit value
2117 const uint8_t S3D_video_format_type = p[5] & 0x7F;
2118
2119 if (S3D_video_format_type == 0x03 ||
2120 S3D_video_format_type == 0x04 ||
2121 S3D_video_format_type == 0x08 ||
2122 S3D_video_format_type == 0x23) {
2123
2124 s1->has_stereo3d = 1;
2125
2126 switch (S3D_video_format_type) {
2127 case 0x03:
2128 s1->stereo3d.type = AV_STEREO3D_SIDEBYSIDE;
2129 break;
2130 case 0x04:
2131 s1->stereo3d.type = AV_STEREO3D_TOPBOTTOM;
2132 break;
2133 case 0x08:
2134 s1->stereo3d.type = AV_STEREO3D_2D;
2135 break;
2136 case 0x23:
2137 s1->stereo3d.type = AV_STEREO3D_SIDEBYSIDE_QUINCUNX;
2138 break;
2139 }
2140 }
2141
2/2
✓ Branch 1 taken 758 times.
✓ Branch 2 taken 72 times.
830 } else if (mpeg_decode_a53_cc(avctx, p, buf_size)) {
2142 758 return;
2143 }
2144 }
2145
2146 561 static int mpeg_decode_gop(AVCodecContext *avctx,
2147 const uint8_t *buf, int buf_size)
2148 {
2149 561 Mpeg1Context *s1 = avctx->priv_data;
2150 561 MpegEncContext *s = &s1->mpeg_enc_ctx;
2151 int broken_link;
2152 int64_t tc;
2153
2154 561 int ret = init_get_bits8(&s->gb, buf, buf_size);
2155
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 561 times.
561 if (ret < 0)
2156 return ret;
2157
2158 561 tc = s1->timecode_frame_start = get_bits(&s->gb, 25);
2159
2160 561 s1->closed_gop = get_bits1(&s->gb);
2161 /* broken_link indicates that after editing the
2162 * reference frames of the first B-Frames after GOP I-Frame
2163 * are missing (open gop) */
2164 561 broken_link = get_bits1(&s->gb);
2165
2166
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 561 times.
561 if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
2167 char tcbuf[AV_TIMECODE_STR_SIZE];
2168 av_timecode_make_mpeg_tc_string(tcbuf, tc);
2169 av_log(s->avctx, AV_LOG_DEBUG,
2170 "GOP (%s) closed_gop=%d broken_link=%d\n",
2171 tcbuf, s1->closed_gop, broken_link);
2172 }
2173
2174 561 return 0;
2175 }
2176
2177 4667 static int decode_chunks(AVCodecContext *avctx, AVFrame *picture,
2178 int *got_output, const uint8_t *buf, int buf_size)
2179 {
2180 4667 Mpeg1Context *s = avctx->priv_data;
2181 4667 MpegEncContext *s2 = &s->mpeg_enc_ctx;
2182 4667 const uint8_t *buf_ptr = buf;
2183 4667 const uint8_t *buf_end = buf + buf_size;
2184 int ret, input_size;
2185 4667 int last_code = 0, skip_frame = 0;
2186 4667 int picture_start_code_seen = 0;
2187
2188 224183 for (;;) {
2189 /* find next start code */
2190 228850 uint32_t start_code = -1;
2191 228850 buf_ptr = avpriv_find_start_code(buf_ptr, buf_end, &start_code);
2192
2/2
✓ Branch 0 taken 4504 times.
✓ Branch 1 taken 224346 times.
228850 if (start_code > 0x1ff) {
2193
2/2
✓ Branch 0 taken 4185 times.
✓ Branch 1 taken 319 times.
4504 if (!skip_frame) {
2194 4185 if (HAVE_THREADS &&
2195
2/2
✓ Branch 0 taken 747 times.
✓ Branch 1 taken 3438 times.
4185 (avctx->active_thread_type & FF_THREAD_SLICE) &&
2196
1/2
✓ Branch 0 taken 747 times.
✗ Branch 1 not taken.
747 !avctx->hwaccel) {
2197 int i;
2198
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 747 times.
747 av_assert0(avctx->thread_count > 1);
2199
2200 747 avctx->execute(avctx, slice_decode_thread,
2201 747 &s2->thread_context[0], NULL,
2202 s->slice_count, sizeof(void *));
2203
2/2
✓ Branch 0 taken 6678 times.
✓ Branch 1 taken 747 times.
7425 for (i = 0; i < s->slice_count; i++)
2204 6678 s2->er.error_count += s2->thread_context[i]->er.error_count;
2205 }
2206
2207 4185 ret = slice_end(avctx, picture);
2208
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4185 times.
4185 if (ret < 0)
2209 4667 return ret;
2210
2/2
✓ Branch 0 taken 3898 times.
✓ Branch 1 taken 287 times.
4185 else if (ret) {
2211 // FIXME: merge with the stuff in mpeg_decode_slice
2212
5/6
✓ Branch 0 taken 76 times.
✓ Branch 1 taken 3822 times.
✓ Branch 2 taken 62 times.
✓ Branch 3 taken 14 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 62 times.
3898 if (s2->last_picture_ptr || s2->low_delay || s2->pict_type == AV_PICTURE_TYPE_B)
2213 3836 *got_output = 1;
2214 }
2215 }
2216 4504 s2->pict_type = 0;
2217
2218
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4504 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
4504 if (avctx->err_recognition & AV_EF_EXPLODE && s2->er.error_count)
2219 return AVERROR_INVALIDDATA;
2220
2221 4504 return FFMAX(0, buf_ptr - buf);
2222 }
2223
2224 224346 input_size = buf_end - buf_ptr;
2225
2226
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 224346 times.
224346 if (avctx->debug & FF_DEBUG_STARTCODE)
2227 av_log(avctx, AV_LOG_DEBUG, "%3"PRIX32" at %"PTRDIFF_SPECIFIER" left %d\n",
2228 start_code, buf_ptr - buf, input_size);
2229
2230 /* prepare data for next start code */
2231
6/6
✓ Branch 0 taken 810 times.
✓ Branch 1 taken 5216 times.
✓ Branch 2 taken 6376 times.
✓ Branch 3 taken 1037 times.
✓ Branch 4 taken 561 times.
✓ Branch 5 taken 210346 times.
224346 switch (start_code) {
2232 810 case SEQ_START_CODE:
2233
1/2
✓ Branch 0 taken 810 times.
✗ Branch 1 not taken.
810 if (last_code == 0) {
2234 810 mpeg1_decode_sequence(avctx, buf_ptr, input_size);
2235
2/2
✓ Branch 0 taken 552 times.
✓ Branch 1 taken 258 times.
810 if (buf != avctx->extradata)
2236 552 s->sync = 1;
2237 } else {
2238 av_log(avctx, AV_LOG_ERROR,
2239 "ignoring SEQ_START_CODE after %X\n", last_code);
2240 if (avctx->err_recognition & AV_EF_EXPLODE)
2241 return AVERROR_INVALIDDATA;
2242 }
2243 810 break;
2244
2245 5216 case PICTURE_START_CODE:
2246
3/4
✓ Branch 0 taken 835 times.
✓ Branch 1 taken 4381 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 835 times.
5216 if (picture_start_code_seen && s2->picture_structure == PICT_FRAME) {
2247 /* If it's a frame picture, there can't be more than one picture header.
2248 Yet, it does happen and we need to handle it. */
2249 av_log(avctx, AV_LOG_WARNING, "ignoring extra picture following a frame-picture\n");
2250 break;
2251 }
2252 5216 picture_start_code_seen = 1;
2253
2254
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 5216 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
5216 if (buf == avctx->extradata && avctx->codec_tag == AV_RL32("AVmp")) {
2255 av_log(avctx, AV_LOG_WARNING, "ignoring picture start code in AVmp extradata\n");
2256 break;
2257 }
2258
2259
3/4
✓ Branch 0 taken 5053 times.
✓ Branch 1 taken 163 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 5053 times.
5216 if (s2->width <= 0 || s2->height <= 0) {
2260 163 av_log(avctx, AV_LOG_ERROR, "Invalid frame dimensions %dx%d.\n",
2261 s2->width, s2->height);
2262 163 return AVERROR_INVALIDDATA;
2263 }
2264
2265
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5053 times.
5053 if (s->tmpgexs){
2266 s2->intra_dc_precision= 3;
2267 s2->intra_matrix[0]= 1;
2268 }
2269
2/2
✓ Branch 0 taken 752 times.
✓ Branch 1 taken 4301 times.
5053 if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE) &&
2270
2/4
✓ Branch 0 taken 752 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 752 times.
752 !avctx->hwaccel && s->slice_count) {
2271 int i;
2272
2273 avctx->execute(avctx, slice_decode_thread,
2274 s2->thread_context, NULL,
2275 s->slice_count, sizeof(void *));
2276 for (i = 0; i < s->slice_count; i++)
2277 s2->er.error_count += s2->thread_context[i]->er.error_count;
2278 s->slice_count = 0;
2279 }
2280
3/4
✓ Branch 0 taken 835 times.
✓ Branch 1 taken 4218 times.
✓ Branch 2 taken 835 times.
✗ Branch 3 not taken.
5053 if (last_code == 0 || last_code == SLICE_MIN_START_CODE) {
2281 5053 ret = mpeg_decode_postinit(avctx);
2282
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5053 times.
5053 if (ret < 0) {
2283 av_log(avctx, AV_LOG_ERROR,
2284 "mpeg_decode_postinit() failure\n");
2285 return ret;
2286 }
2287
2288 /* We have a complete image: we try to decompress it. */
2289
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 5053 times.
5053 if (mpeg1_decode_picture(avctx, buf_ptr, input_size) < 0)
2290 s2->pict_type = 0;
2291 5053 s->first_slice = 1;
2292 5053 last_code = PICTURE_START_CODE;
2293 } else {
2294 av_log(avctx, AV_LOG_ERROR,
2295 "ignoring pic after %X\n", last_code);
2296 if (avctx->err_recognition & AV_EF_EXPLODE)
2297 return AVERROR_INVALIDDATA;
2298 }
2299 5053 break;
2300 6376 case EXT_START_CODE:
2301 6376 ret = init_get_bits8(&s2->gb, buf_ptr, input_size);
2302
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6376 times.
6376 if (ret < 0)
2303 return ret;
2304
2305 6376 switch (get_bits(&s2->gb, 4)) {
2306 718 case 0x1:
2307
1/2
✓ Branch 0 taken 718 times.
✗ Branch 1 not taken.
718 if (last_code == 0) {
2308 718 mpeg_decode_sequence_extension(s);
2309 } else {
2310 av_log(avctx, AV_LOG_ERROR,
2311 "ignoring seq ext after %X\n", last_code);
2312 if (avctx->err_recognition & AV_EF_EXPLODE)
2313 return AVERROR_INVALIDDATA;
2314 }
2315 718 break;
2316 95 case 0x2:
2317 95 mpeg_decode_sequence_display_extension(s);
2318 95 break;
2319 977 case 0x3:
2320 977 mpeg_decode_quant_matrix_extension(s2);
2321 977 break;
2322 27 case 0x7:
2323 27 mpeg_decode_picture_display_extension(s);
2324 27 break;
2325 4559 case 0x8:
2326
1/2
✓ Branch 0 taken 4559 times.
✗ Branch 1 not taken.
4559 if (last_code == PICTURE_START_CODE) {
2327 4559 int ret = mpeg_decode_picture_coding_extension(s);
2328
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4559 times.
4559 if (ret < 0)
2329 return ret;
2330 } else {
2331 av_log(avctx, AV_LOG_ERROR,
2332 "ignoring pic cod ext after %X\n", last_code);
2333 if (avctx->err_recognition & AV_EF_EXPLODE)
2334 return AVERROR_INVALIDDATA;
2335 }
2336 4559 break;
2337 }
2338 6376 break;
2339 1037 case USER_START_CODE:
2340 1037 mpeg_decode_user_data(avctx, buf_ptr, input_size);
2341 1037 break;
2342 561 case GOP_START_CODE:
2343
1/2
✓ Branch 0 taken 561 times.
✗ Branch 1 not taken.
561 if (last_code == 0) {
2344 561 s2->first_field = 0;
2345 561 ret = mpeg_decode_gop(avctx, buf_ptr, input_size);
2346
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 561 times.
561 if (ret < 0)
2347 return ret;
2348 561 s->sync = 1;
2349 } else {
2350 av_log(avctx, AV_LOG_ERROR,
2351 "ignoring GOP_START_CODE after %X\n", last_code);
2352 if (avctx->err_recognition & AV_EF_EXPLODE)
2353 return AVERROR_INVALIDDATA;
2354 }
2355 561 break;
2356 210346 default:
2357
2/2
✓ Branch 0 taken 209444 times.
✓ Branch 1 taken 902 times.
210346 if (start_code >= SLICE_MIN_START_CODE &&
2358
4/4
✓ Branch 0 taken 209439 times.
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 5053 times.
✓ Branch 3 taken 204386 times.
209444 start_code <= SLICE_MAX_START_CODE && last_code == PICTURE_START_CODE) {
2359
3/4
✓ Branch 0 taken 1701 times.
✓ Branch 1 taken 3352 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1701 times.
5053 if (s2->progressive_sequence && !s2->progressive_frame) {
2360 s2->progressive_frame = 1;
2361 av_log(s2->avctx, AV_LOG_ERROR,
2362 "interlaced frame in progressive sequence, ignoring\n");
2363 }
2364
2365
1/2
✓ Branch 0 taken 5053 times.
✗ Branch 1 not taken.
5053 if (s2->picture_structure == 0 ||
2366
3/4
✓ Branch 0 taken 1708 times.
✓ Branch 1 taken 3345 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1708 times.
5053 (s2->progressive_frame && s2->picture_structure != PICT_FRAME)) {
2367 av_log(s2->avctx, AV_LOG_ERROR,
2368 "picture_structure %d invalid, ignoring\n",
2369 s2->picture_structure);
2370 s2->picture_structure = PICT_FRAME;
2371 }
2372
2373
3/4
✓ Branch 0 taken 1701 times.
✓ Branch 1 taken 3352 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1701 times.
5053 if (s2->progressive_sequence && !s2->frame_pred_frame_dct)
2374 av_log(s2->avctx, AV_LOG_WARNING, "invalid frame_pred_frame_dct\n");
2375
2376
2/2
✓ Branch 0 taken 3381 times.
✓ Branch 1 taken 1672 times.
5053 if (s2->picture_structure == PICT_FRAME) {
2377 3381 s2->first_field = 0;
2378 3381 s2->v_edge_pos = 16 * s2->mb_height;
2379 } else {
2380 1672 s2->first_field ^= 1;
2381 1672 s2->v_edge_pos = 8 * s2->mb_height;
2382 1672 memset(s2->mbskip_table, 0, s2->mb_stride * s2->mb_height);
2383 }
2384 }
2385
2/2
✓ Branch 0 taken 209444 times.
✓ Branch 1 taken 902 times.
210346 if (start_code >= SLICE_MIN_START_CODE &&
2386
4/4
✓ Branch 0 taken 209439 times.
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 209269 times.
✓ Branch 3 taken 170 times.
209444 start_code <= SLICE_MAX_START_CODE && last_code != 0) {
2387 209269 const int field_pic = s2->picture_structure != PICT_FRAME;
2388 209269 int mb_y = start_code - SLICE_MIN_START_CODE;
2389 209269 last_code = SLICE_MIN_START_CODE;
2390
3/4
✓ Branch 0 taken 201681 times.
✓ Branch 1 taken 7588 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 201681 times.
209269 if (s2->codec_id != AV_CODEC_ID_MPEG1VIDEO && s2->mb_height > 2800/16)
2391 mb_y += (*buf_ptr&0xE0)<<2;
2392
2393 209269 mb_y <<= field_pic;
2394
2/2
✓ Branch 0 taken 13673 times.
✓ Branch 1 taken 195596 times.
209269 if (s2->picture_structure == PICT_BOTTOM_FIELD)
2395 13673 mb_y++;
2396
2397
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 209269 times.
209269 if (buf_end - buf_ptr < 2) {
2398 av_log(s2->avctx, AV_LOG_ERROR, "slice too small\n");
2399 return AVERROR_INVALIDDATA;
2400 }
2401
2402
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 209269 times.
209269 if (mb_y >= s2->mb_height) {
2403 av_log(s2->avctx, AV_LOG_ERROR,
2404 "slice below image (%d >= %d)\n", mb_y, s2->mb_height);
2405 return AVERROR_INVALIDDATA;
2406 }
2407
2408
2/2
✓ Branch 0 taken 86489 times.
✓ Branch 1 taken 122780 times.
209269 if (!s2->last_picture_ptr) {
2409 /* Skip B-frames if we do not have reference frames and
2410 * GOP is not closed. */
2411
2/2
✓ Branch 0 taken 732 times.
✓ Branch 1 taken 85757 times.
86489 if (s2->pict_type == AV_PICTURE_TYPE_B) {
2412
1/2
✓ Branch 0 taken 732 times.
✗ Branch 1 not taken.
732 if (!s->closed_gop) {
2413 732 skip_frame = 1;
2414 732 av_log(s2->avctx, AV_LOG_DEBUG,
2415 "Skipping B slice due to open GOP\n");
2416 732 break;
2417 }
2418 }
2419 }
2420
3/4
✓ Branch 0 taken 80823 times.
✓ Branch 1 taken 127714 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 80823 times.
208537 if (s2->pict_type == AV_PICTURE_TYPE_I || (s2->avctx->flags2 & AV_CODEC_FLAG2_SHOW_ALL))
2421 127714 s->sync = 1;
2422
2/2
✓ Branch 0 taken 81592 times.
✓ Branch 1 taken 126945 times.
208537 if (!s2->next_picture_ptr) {
2423 /* Skip P-frames if we do not have a reference frame or
2424 * we have an invalid header. */
2425
4/4
✓ Branch 0 taken 4944 times.
✓ Branch 1 taken 76648 times.
✓ Branch 2 taken 4032 times.
✓ Branch 3 taken 912 times.
81592 if (s2->pict_type == AV_PICTURE_TYPE_P && !s->sync) {
2426 4032 skip_frame = 1;
2427 4032 av_log(s2->avctx, AV_LOG_DEBUG,
2428 "Skipping P slice due to !sync\n");
2429 4032 break;
2430 }
2431 }
2432
2/2
✓ Branch 0 taken 77484 times.
✓ Branch 1 taken 127021 times.
204505 if ((avctx->skip_frame >= AVDISCARD_NONREF &&
2433
1/2
✓ Branch 0 taken 77484 times.
✗ Branch 1 not taken.
77484 s2->pict_type == AV_PICTURE_TYPE_B) ||
2434
2/2
✓ Branch 0 taken 77484 times.
✓ Branch 1 taken 127021 times.
204505 (avctx->skip_frame >= AVDISCARD_NONKEY &&
2435
2/2
✓ Branch 0 taken 76572 times.
✓ Branch 1 taken 912 times.
77484 s2->pict_type != AV_PICTURE_TYPE_I) ||
2436
2/2
✓ Branch 0 taken 76572 times.
✓ Branch 1 taken 127021 times.
203593 avctx->skip_frame >= AVDISCARD_ALL) {
2437 77484 skip_frame = 1;
2438 77484 break;
2439 }
2440
2441
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 127021 times.
127021 if (!s2->context_initialized)
2442 break;
2443
2444
2/2
✓ Branch 0 taken 119748 times.
✓ Branch 1 taken 7273 times.
127021 if (s2->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
2445
1/2
✓ Branch 0 taken 119748 times.
✗ Branch 1 not taken.
119748 if (mb_y < avctx->skip_top ||
2446
1/2
✓ Branch 0 taken 119748 times.
✗ Branch 1 not taken.
119748 mb_y >= s2->mb_height - avctx->skip_bottom)
2447 break;
2448 }
2449
2450
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 127021 times.
127021 if (!s2->pict_type) {
2451 av_log(avctx, AV_LOG_ERROR, "Missing picture start code\n");
2452 if (avctx->err_recognition & AV_EF_EXPLODE)
2453 return AVERROR_INVALIDDATA;
2454 break;
2455 }
2456
2457
2/2
✓ Branch 0 taken 4608 times.
✓ Branch 1 taken 122413 times.
127021 if (s->first_slice) {
2458 4608 skip_frame = 0;
2459 4608 s->first_slice = 0;
2460
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4608 times.
4608 if ((ret = mpeg_field_start(s2, buf, buf_size)) < 0)
2461 return ret;
2462 }
2463
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 127021 times.
127021 if (!s2->current_picture_ptr) {
2464 av_log(avctx, AV_LOG_ERROR,
2465 "current_picture not initialized\n");
2466 return AVERROR_INVALIDDATA;
2467 }
2468
2469 127021 if (HAVE_THREADS &&
2470
2/2
✓ Branch 0 taken 22260 times.
✓ Branch 1 taken 104761 times.
127021 (avctx->active_thread_type & FF_THREAD_SLICE) &&
2471
1/2
✓ Branch 0 taken 22260 times.
✗ Branch 1 not taken.
22260 !avctx->hwaccel) {
2472 22260 int threshold = (s2->mb_height * s->slice_count +
2473 22260 s2->slice_context_count / 2) /
2474 22260 s2->slice_context_count;
2475
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22260 times.
22260 av_assert0(avctx->thread_count > 1);
2476
2/2
✓ Branch 0 taken 6678 times.
✓ Branch 1 taken 15582 times.
22260 if (threshold <= mb_y) {
2477 6678 MpegEncContext *thread_context = s2->thread_context[s->slice_count];
2478
2479 6678 thread_context->start_mb_y = mb_y;
2480 6678 thread_context->end_mb_y = s2->mb_height;
2481
2/2
✓ Branch 0 taken 5936 times.
✓ Branch 1 taken 742 times.
6678 if (s->slice_count) {
2482 5936 s2->thread_context[s->slice_count - 1]->end_mb_y = mb_y;
2483 5936 ret = ff_update_duplicate_context(thread_context, s2);
2484
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5936 times.
5936 if (ret < 0)
2485 return ret;
2486 }
2487 6678 ret = init_get_bits8(&thread_context->gb, buf_ptr, input_size);
2488
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6678 times.
6678 if (ret < 0)
2489 return ret;
2490 6678 s->slice_count++;
2491 }
2492 22260 buf_ptr += 2; // FIXME add minimum number of bytes per slice
2493 } else {
2494 104761 ret = mpeg_decode_slice(s2, mb_y, &buf_ptr, input_size);
2495 104761 emms_c();
2496
2497
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 104751 times.
104761 if (ret < 0) {
2498
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
10 if (avctx->err_recognition & AV_EF_EXPLODE)
2499 return ret;
2500
2/4
✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
10 if (s2->resync_mb_x >= 0 && s2->resync_mb_y >= 0)
2501 10 ff_er_add_slice(&s2->er, s2->resync_mb_x,
2502 s2->resync_mb_y, s2->mb_x, s2->mb_y,
2503 ER_AC_ERROR | ER_DC_ERROR | ER_MV_ERROR);
2504 } else {
2505 104751 ff_er_add_slice(&s2->er, s2->resync_mb_x,
2506 104751 s2->resync_mb_y, s2->mb_x - 1, s2->mb_y,
2507 ER_AC_END | ER_DC_END | ER_MV_END);
2508 }
2509 }
2510 }
2511 128098 break;
2512 }
2513 }
2514 }
2515
2516 4547 static int mpeg_decode_frame(AVCodecContext *avctx, AVFrame *picture,
2517 int *got_output, AVPacket *avpkt)
2518 {
2519 4547 const uint8_t *buf = avpkt->data;
2520 int ret;
2521 4547 int buf_size = avpkt->size;
2522 4547 Mpeg1Context *s = avctx->priv_data;
2523 4547 MpegEncContext *s2 = &s->mpeg_enc_ctx;
2524
2525
3/6
✓ Branch 0 taken 4409 times.
✓ Branch 1 taken 138 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 4409 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
4547 if (buf_size == 0 || (buf_size == 4 && AV_RB32(buf) == SEQ_END_CODE)) {
2526 /* special case for last picture */
2527
4/4
✓ Branch 0 taken 127 times.
✓ Branch 1 taken 11 times.
✓ Branch 2 taken 61 times.
✓ Branch 3 taken 66 times.
138 if (s2->low_delay == 0 && s2->next_picture_ptr) {
2528 61 int ret = av_frame_ref(picture, s2->next_picture_ptr->f);
2529
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 61 times.
61 if (ret < 0)
2530 return ret;
2531
2532 61 s2->next_picture_ptr = NULL;
2533
2534 61 *got_output = 1;
2535 }
2536 138 return buf_size;
2537 }
2538
2539
2/2
✓ Branch 0 taken 452 times.
✓ Branch 1 taken 3957 times.
4409 if (!s2->context_initialized &&
2540
3/4
✓ Branch 0 taken 450 times.
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 450 times.
452 (s2->codec_tag == AV_RL32("VCR2") || s2->codec_tag == AV_RL32("BW10")))
2541 2 vcr2_init_sequence(avctx);
2542
2543 4409 s->slice_count = 0;
2544
2545
4/4
✓ Branch 0 taken 4023 times.
✓ Branch 1 taken 386 times.
✓ Branch 2 taken 258 times.
✓ Branch 3 taken 3765 times.
4409 if (avctx->extradata && !s->extradata_decoded) {
2546 258 ret = decode_chunks(avctx, picture, got_output,
2547 258 avctx->extradata, avctx->extradata_size);
2548
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 258 times.
258 if (*got_output) {
2549 av_log(avctx, AV_LOG_ERROR, "picture in extradata\n");
2550 av_frame_unref(picture);
2551 *got_output = 0;
2552 }
2553 258 s->extradata_decoded = 1;
2554
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 258 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
258 if (ret < 0 && (avctx->err_recognition & AV_EF_EXPLODE)) {
2555 s2->current_picture_ptr = NULL;
2556 return ret;
2557 }
2558 }
2559
2560 4409 ret = decode_chunks(avctx, picture, got_output, buf, buf_size);
2561
4/4
✓ Branch 0 taken 4246 times.
✓ Branch 1 taken 163 times.
✓ Branch 2 taken 3836 times.
✓ Branch 3 taken 410 times.
4409 if (ret<0 || *got_output) {
2562 3999 s2->current_picture_ptr = NULL;
2563
2564
4/4
✓ Branch 0 taken 538 times.
✓ Branch 1 taken 3461 times.
✓ Branch 2 taken 375 times.
✓ Branch 3 taken 163 times.
3999 if (s->timecode_frame_start != -1 && *got_output) {
2565 char tcbuf[AV_TIMECODE_STR_SIZE];
2566 375 AVFrameSideData *tcside = av_frame_new_side_data(picture,
2567 AV_FRAME_DATA_GOP_TIMECODE,
2568 sizeof(int64_t));
2569
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 375 times.
375 if (!tcside)
2570 return AVERROR(ENOMEM);
2571 375 memcpy(tcside->data, &s->timecode_frame_start, sizeof(int64_t));
2572
2573 375 av_timecode_make_mpeg_tc_string(tcbuf, s->timecode_frame_start);
2574 375 av_dict_set(&picture->metadata, "timecode", tcbuf, 0);
2575
2576 375 s->timecode_frame_start = -1;
2577 }
2578 }
2579
2580 4409 return ret;
2581 }
2582
2583 static void flush(AVCodecContext *avctx)
2584 {
2585 Mpeg1Context *s = avctx->priv_data;
2586
2587 s->sync = 0;
2588 s->closed_gop = 0;
2589
2590 av_buffer_unref(&s->a53_buf_ref);
2591 ff_mpeg_flush(avctx);
2592 }
2593
2594 286 static av_cold int mpeg_decode_end(AVCodecContext *avctx)
2595 {
2596 286 Mpeg1Context *s = avctx->priv_data;
2597
2598 286 ff_mpv_common_end(&s->mpeg_enc_ctx);
2599 286 av_buffer_unref(&s->a53_buf_ref);
2600 286 return 0;
2601 }
2602
2603 const FFCodec ff_mpeg1video_decoder = {
2604 .p.name = "mpeg1video",
2605 CODEC_LONG_NAME("MPEG-1 video"),
2606 .p.type = AVMEDIA_TYPE_VIDEO,
2607 .p.id = AV_CODEC_ID_MPEG1VIDEO,
2608 .priv_data_size = sizeof(Mpeg1Context),
2609 .init = mpeg_decode_init,
2610 .close = mpeg_decode_end,
2611 FF_CODEC_DECODE_CB(mpeg_decode_frame),
2612 .p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
2613 AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
2614 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
2615 .flush = flush,
2616 .p.max_lowres = 3,
2617 UPDATE_THREAD_CONTEXT(mpeg_decode_update_thread_context),
2618 .hw_configs = (const AVCodecHWConfigInternal *const []) {
2619 #if CONFIG_MPEG1_NVDEC_HWACCEL
2620 HWACCEL_NVDEC(mpeg1),
2621 #endif
2622 #if CONFIG_MPEG1_VDPAU_HWACCEL
2623 HWACCEL_VDPAU(mpeg1),
2624 #endif
2625 #if CONFIG_MPEG1_VIDEOTOOLBOX_HWACCEL
2626 HWACCEL_VIDEOTOOLBOX(mpeg1),
2627 #endif
2628 NULL
2629 },
2630 };
2631
2632 #define M2V_OFFSET(x) offsetof(Mpeg1Context, x)
2633 #define M2V_PARAM AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
2634
2635 static const AVOption mpeg2video_options[] = {
2636 { "cc_format", "extract a specific Closed Captions format",
2637 M2V_OFFSET(cc_format), AV_OPT_TYPE_INT, { .i64 = CC_FORMAT_AUTO },
2638 CC_FORMAT_AUTO, CC_FORMAT_DVD, M2V_PARAM, .unit = "cc_format" },
2639
2640 { "auto", "pick first seen CC substream", 0, AV_OPT_TYPE_CONST,
2641 { .i64 = CC_FORMAT_AUTO }, .flags = M2V_PARAM, .unit = "cc_format" },
2642 { "a53", "pick A/53 Part 4 CC substream", 0, AV_OPT_TYPE_CONST,
2643 { .i64 = CC_FORMAT_A53_PART4 }, .flags = M2V_PARAM, .unit = "cc_format" },
2644 { "scte20", "pick SCTE-20 CC substream", 0, AV_OPT_TYPE_CONST,
2645 { .i64 = CC_FORMAT_SCTE20 }, .flags = M2V_PARAM, .unit = "cc_format" },
2646 { "dvd", "pick DVD CC substream", 0, AV_OPT_TYPE_CONST,
2647 { .i64 = CC_FORMAT_DVD }, .flags = M2V_PARAM, .unit = "cc_format" },
2648 { NULL }
2649 };
2650
2651 static const AVClass mpeg2video_class = {
2652 .class_name = "MPEG-2 video",
2653 .item_name = av_default_item_name,
2654 .option = mpeg2video_options,
2655 .version = LIBAVUTIL_VERSION_INT,
2656 .category = AV_CLASS_CATEGORY_DECODER,
2657 };
2658
2659 const FFCodec ff_mpeg2video_decoder = {
2660 .p.name = "mpeg2video",
2661 CODEC_LONG_NAME("MPEG-2 video"),
2662 .p.type = AVMEDIA_TYPE_VIDEO,
2663 .p.id = AV_CODEC_ID_MPEG2VIDEO,
2664 .p.priv_class = &mpeg2video_class,
2665 .priv_data_size = sizeof(Mpeg1Context),
2666 .init = mpeg_decode_init,
2667 .close = mpeg_decode_end,
2668 FF_CODEC_DECODE_CB(mpeg_decode_frame),
2669 .p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
2670 AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
2671 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
2672 .flush = flush,
2673 .p.max_lowres = 3,
2674 .p.profiles = NULL_IF_CONFIG_SMALL(ff_mpeg2_video_profiles),
2675 .hw_configs = (const AVCodecHWConfigInternal *const []) {
2676 #if CONFIG_MPEG2_DXVA2_HWACCEL
2677 HWACCEL_DXVA2(mpeg2),
2678 #endif
2679 #if CONFIG_MPEG2_D3D11VA_HWACCEL
2680 HWACCEL_D3D11VA(mpeg2),
2681 #endif
2682 #if CONFIG_MPEG2_D3D11VA2_HWACCEL
2683 HWACCEL_D3D11VA2(mpeg2),
2684 #endif
2685 #if CONFIG_MPEG2_D3D12VA_HWACCEL
2686 HWACCEL_D3D12VA(mpeg2),
2687 #endif
2688 #if CONFIG_MPEG2_NVDEC_HWACCEL
2689 HWACCEL_NVDEC(mpeg2),
2690 #endif
2691 #if CONFIG_MPEG2_VAAPI_HWACCEL
2692 HWACCEL_VAAPI(mpeg2),
2693 #endif
2694 #if CONFIG_MPEG2_VDPAU_HWACCEL
2695 HWACCEL_VDPAU(mpeg2),
2696 #endif
2697 #if CONFIG_MPEG2_VIDEOTOOLBOX_HWACCEL
2698 HWACCEL_VIDEOTOOLBOX(mpeg2),
2699 #endif
2700 NULL
2701 },
2702 };
2703
2704 //legacy decoder
2705 const FFCodec ff_mpegvideo_decoder = {
2706 .p.name = "mpegvideo",
2707 CODEC_LONG_NAME("MPEG-1 video"),
2708 .p.type = AVMEDIA_TYPE_VIDEO,
2709 .p.id = AV_CODEC_ID_MPEG2VIDEO,
2710 .priv_data_size = sizeof(Mpeg1Context),
2711 .init = mpeg_decode_init,
2712 .close = mpeg_decode_end,
2713 FF_CODEC_DECODE_CB(mpeg_decode_frame),
2714 .p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
2715 AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
2716 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
2717 .flush = flush,
2718 .p.max_lowres = 3,
2719 };
2720
2721 typedef struct IPUContext {
2722 MpegEncContext m;
2723
2724 int flags;
2725 DECLARE_ALIGNED(32, int16_t, block)[6][64];
2726 } IPUContext;
2727
2728 static int ipu_decode_frame(AVCodecContext *avctx, AVFrame *frame,
2729 int *got_frame, AVPacket *avpkt)
2730 {
2731 IPUContext *s = avctx->priv_data;
2732 MpegEncContext *m = &s->m;
2733 GetBitContext *gb = &m->gb;
2734 int ret;
2735
2736 // Check for minimal intra MB size (considering mb header, luma & chroma dc VLC, ac EOB VLC)
2737 if (avpkt->size*8LL < (avctx->width+15)/16 * ((avctx->height+15)/16) * (2 + 3*4 + 2*2 + 2*6))
2738 return AVERROR_INVALIDDATA;
2739
2740 ret = ff_get_buffer(avctx, frame, 0);
2741 if (ret < 0)
2742 return ret;
2743
2744 ret = init_get_bits8(gb, avpkt->data, avpkt->size);
2745 if (ret < 0)
2746 return ret;
2747
2748 s->flags = get_bits(gb, 8);
2749 m->intra_dc_precision = s->flags & 3;
2750 m->q_scale_type = !!(s->flags & 0x40);
2751 m->intra_vlc_format = !!(s->flags & 0x20);
2752 m->alternate_scan = !!(s->flags & 0x10);
2753
2754 if (s->flags & 0x10) {
2755 ff_init_scantable(m->idsp.idct_permutation, &m->inter_scantable, ff_alternate_vertical_scan);
2756 ff_init_scantable(m->idsp.idct_permutation, &m->intra_scantable, ff_alternate_vertical_scan);
2757 } else {
2758 ff_init_scantable(m->idsp.idct_permutation, &m->inter_scantable, ff_zigzag_direct);
2759 ff_init_scantable(m->idsp.idct_permutation, &m->intra_scantable, ff_zigzag_direct);
2760 }
2761
2762 m->last_dc[0] = m->last_dc[1] = m->last_dc[2] = 1 << (7 + (s->flags & 3));
2763 m->qscale = 1;
2764
2765 for (int y = 0; y < avctx->height; y += 16) {
2766 int intraquant;
2767
2768 for (int x = 0; x < avctx->width; x += 16) {
2769 if (x || y) {
2770 if (!get_bits1(gb))
2771 return AVERROR_INVALIDDATA;
2772 }
2773 if (get_bits1(gb)) {
2774 intraquant = 0;
2775 } else {
2776 if (!get_bits1(gb))
2777 return AVERROR_INVALIDDATA;
2778 intraquant = 1;
2779 }
2780
2781 if (s->flags & 4)
2782 skip_bits1(gb);
2783
2784 if (intraquant)
2785 m->qscale = mpeg_get_qscale(m);
2786
2787 memset(s->block, 0, sizeof(s->block));
2788
2789 for (int n = 0; n < 6; n++) {
2790 if (s->flags & 0x80) {
2791 ret = ff_mpeg1_decode_block_intra(&m->gb,
2792 m->intra_matrix,
2793 m->intra_scantable.permutated,
2794 m->last_dc, s->block[n],
2795 n, m->qscale);
2796 if (ret >= 0)
2797 m->block_last_index[n] = ret;
2798 } else {
2799 ret = mpeg2_decode_block_intra(m, s->block[n], n);
2800 }
2801
2802 if (ret < 0)
2803 return ret;
2804 }
2805
2806 m->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x,
2807 frame->linesize[0], s->block[0]);
2808 m->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x + 8,
2809 frame->linesize[0], s->block[1]);
2810 m->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x,
2811 frame->linesize[0], s->block[2]);
2812 m->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x + 8,
2813 frame->linesize[0], s->block[3]);
2814 m->idsp.idct_put(frame->data[1] + (y >> 1) * frame->linesize[1] + (x >> 1),
2815 frame->linesize[1], s->block[4]);
2816 m->idsp.idct_put(frame->data[2] + (y >> 1) * frame->linesize[2] + (x >> 1),
2817 frame->linesize[2], s->block[5]);
2818 }
2819 }
2820
2821 align_get_bits(gb);
2822 if (get_bits_left(gb) != 32)
2823 return AVERROR_INVALIDDATA;
2824
2825 frame->pict_type = AV_PICTURE_TYPE_I;
2826 frame->flags |= AV_FRAME_FLAG_KEY;
2827 *got_frame = 1;
2828
2829 return avpkt->size;
2830 }
2831
2832 static av_cold int ipu_decode_init(AVCodecContext *avctx)
2833 {
2834 IPUContext *s = avctx->priv_data;
2835 MpegEncContext *m = &s->m;
2836
2837 avctx->pix_fmt = AV_PIX_FMT_YUV420P;
2838
2839 ff_mpv_decode_init(m, avctx);
2840 ff_mpeg12_init_vlcs();
2841
2842 for (int i = 0; i < 64; i++) {
2843 int j = m->idsp.idct_permutation[i];
2844 int v = ff_mpeg1_default_intra_matrix[i];
2845 m->intra_matrix[j] = v;
2846 m->chroma_intra_matrix[j] = v;
2847 }
2848
2849 for (int i = 0; i < 64; i++) {
2850 int j = m->idsp.idct_permutation[i];
2851 int v = ff_mpeg1_default_non_intra_matrix[i];
2852 m->inter_matrix[j] = v;
2853 m->chroma_inter_matrix[j] = v;
2854 }
2855
2856 return 0;
2857 }
2858
2859 const FFCodec ff_ipu_decoder = {
2860 .p.name = "ipu",
2861 CODEC_LONG_NAME("IPU Video"),
2862 .p.type = AVMEDIA_TYPE_VIDEO,
2863 .p.id = AV_CODEC_ID_IPU,
2864 .priv_data_size = sizeof(IPUContext),
2865 .init = ipu_decode_init,
2866 FF_CODEC_DECODE_CB(ipu_decode_frame),
2867 .p.capabilities = AV_CODEC_CAP_DR1,
2868 };
2869