FFmpeg coverage

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