mpeg12dec.c 107 KB
Newer Older
1 2 3
/*
 * MPEG-1/2 decoder
 * 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 10 11 12
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
13
 * FFmpeg is distributed in the hope that it will be useful,
14 15 16 17 18
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * 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 21 22 23 24 25 26 27
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/**
 * @file
 * MPEG-1/2 decoder
 */

28
#define UNCHECKED_BITSTREAM_READER 1
29 30
#include <inttypes.h>

31
#include "libavutil/attributes.h"
32
#include "libavutil/imgutils.h"
33
#include "libavutil/internal.h"
34
#include "libavutil/stereo3d.h"
35

36
#include "avcodec.h"
37
#include "bytestream.h"
38
#include "error_resilience.h"
39
#include "idctdsp.h"
40
#include "internal.h"
41
#include "mpeg_er.h"
42 43
#include "mpeg12.h"
#include "mpeg12data.h"
44
#include "mpegutils.h"
45
#include "mpegvideo.h"
46
#include "thread.h"
47
#include "version.h"
48
#include "vdpau_internal.h"
49
#include "xvmc_internal.h"
50

51 52 53
typedef struct Mpeg1Context {
    MpegEncContext mpeg_enc_ctx;
    int mpeg_enc_ctx_allocated; /* true if decoding context allocated */
54 55
    int repeat_field;           /* true if we must repeat the field */
    AVPanScan pan_scan;         /* some temporary storage for the panscan */
56 57
    AVStereo3D stereo3d;
    int has_stereo3d;
58 59
    uint8_t *a53_caption;
    int a53_caption_size;
60 61
    uint8_t afd;
    int has_afd;
62
    int slice_count;
63
    AVRational save_aspect;
64
    int save_width, save_height, save_progressive_seq;
65 66
    AVRational frame_rate_ext;  /* MPEG-2 specific framerate modificator */
    int sync;                   /* Did we reach a sync point like a GOP/SEQ/KEYFrame? */
67
    int tmpgexs;
68
    int first_slice;
69 70 71
    int extradata_decoded;
} Mpeg1Context;

72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
#define MB_TYPE_ZERO_MV   0x20000000

static const uint32_t ptype2mb_type[7] = {
                    MB_TYPE_INTRA,
                    MB_TYPE_L0 | MB_TYPE_CBP | MB_TYPE_ZERO_MV | MB_TYPE_16x16,
                    MB_TYPE_L0,
                    MB_TYPE_L0 | MB_TYPE_CBP,
    MB_TYPE_QUANT | MB_TYPE_INTRA,
    MB_TYPE_QUANT | MB_TYPE_L0 | MB_TYPE_CBP | MB_TYPE_ZERO_MV | MB_TYPE_16x16,
    MB_TYPE_QUANT | MB_TYPE_L0 | MB_TYPE_CBP,
};

static const uint32_t btype2mb_type[11] = {
                    MB_TYPE_INTRA,
                    MB_TYPE_L1,
                    MB_TYPE_L1   | MB_TYPE_CBP,
                    MB_TYPE_L0,
                    MB_TYPE_L0   | MB_TYPE_CBP,
                    MB_TYPE_L0L1,
                    MB_TYPE_L0L1 | MB_TYPE_CBP,
    MB_TYPE_QUANT | MB_TYPE_INTRA,
    MB_TYPE_QUANT | MB_TYPE_L1   | MB_TYPE_CBP,
    MB_TYPE_QUANT | MB_TYPE_L0   | MB_TYPE_CBP,
    MB_TYPE_QUANT | MB_TYPE_L0L1 | MB_TYPE_CBP,
};

static const uint8_t non_linear_qscale[32] = {
99 100 101 102
     0,  1,  2,  3,  4,  5,   6,   7,
     8, 10, 12, 14, 16, 18,  20,  22,
    24, 28, 32, 36, 40, 44,  48,  52,
    56, 64, 72, 80, 88, 96, 104, 112,
103 104
};

105 106 107 108 109 110
/* as H.263, but only 17 codes */
static int mpeg_decode_motion(MpegEncContext *s, int fcode, int pred)
{
    int code, sign, val, shift;

    code = get_vlc2(&s->gb, ff_mv_vlc.table, MV_VLC_BITS, 2);
111
    if (code == 0)
112
        return pred;
113
    if (code < 0)
114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
        return 0xffff;

    sign  = get_bits1(&s->gb);
    shift = fcode - 1;
    val   = code;
    if (shift) {
        val  = (val - 1) << shift;
        val |= get_bits(&s->gb, shift);
        val++;
    }
    if (sign)
        val = -val;
    val += pred;

    /* modulo decoding */
    return sign_extend(val, 5 + shift);
}

132 133 134 135 136 137 138 139 140 141 142
#define check_scantable_index(ctx, x)                                         \
    do {                                                                      \
        if ((x) > 63) {                                                       \
            av_log(ctx->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n",     \
                   ctx->mb_x, ctx->mb_y);                                     \
            return AVERROR_INVALIDDATA;                                       \
        }                                                                     \
    } while (0)

static inline int mpeg1_decode_block_intra(MpegEncContext *s,
                                           int16_t *block, int n)
143 144 145
{
    int level, dc, diff, i, j, run;
    int component;
146 147
    RLTable *rl                  = &ff_rl_mpeg1;
    uint8_t *const scantable     = s->intra_scantable.permutated;
148 149 150 151 152 153 154
    const uint16_t *quant_matrix = s->intra_matrix;
    const int qscale             = s->qscale;

    /* DC coefficient */
    component = (n <= 3 ? 0 : n - 4 + 1);
    diff = decode_dc(&s->gb, component);
    if (diff >= 0xffff)
155
        return AVERROR_INVALIDDATA;
156 157 158 159
    dc  = s->last_dc[component];
    dc += diff;
    s->last_dc[component] = dc;
    block[0] = dc * quant_matrix[0];
160
    ff_tlog(s->avctx, "dc=%d diff=%d\n", dc, diff);
161 162 163
    i = 0;
    {
        OPEN_READER(re, &s->gb);
164 165 166 167
        UPDATE_CACHE(re, &s->gb);
        if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
            goto end;

168 169
        /* now quantify & encode AC coefficients */
        for (;;) {
170 171
            GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
                       TEX_VLC_BITS, 2, 0);
172

173
            if (level != 0) {
174
                i += run;
175
                check_scantable_index(s, i);
176 177 178
                j = scantable[i];
                level = (level * qscale * quant_matrix[j]) >> 4;
                level = (level - 1) | 1;
179 180
                level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
                        SHOW_SBITS(re, &s->gb, 1);
181
                SKIP_BITS(re, &s->gb, 1);
182 183
            } else {
                /* escape */
184 185
                run = SHOW_UBITS(re, &s->gb, 6) + 1;
                LAST_SKIP_BITS(re, &s->gb, 6);
186
                UPDATE_CACHE(re, &s->gb);
187 188
                level = SHOW_SBITS(re, &s->gb, 8);
                SKIP_BITS(re, &s->gb, 8);
189
                if (level == -128) {
190
                    level = SHOW_UBITS(re, &s->gb, 8) - 256;
191
                    SKIP_BITS(re, &s->gb, 8);
192
                } else if (level == 0) {
193
                    level = SHOW_UBITS(re, &s->gb, 8);
194
                    SKIP_BITS(re, &s->gb, 8);
195 196
                }
                i += run;
197
                check_scantable_index(s, i);
198 199 200 201 202 203 204 205 206 207 208 209 210
                j = scantable[i];
                if (level < 0) {
                    level = -level;
                    level = (level * qscale * quant_matrix[j]) >> 4;
                    level = (level - 1) | 1;
                    level = -level;
                } else {
                    level = (level * qscale * quant_matrix[j]) >> 4;
                    level = (level - 1) | 1;
                }
            }

            block[j] = level;
211 212 213 214
            if (((int32_t)GET_CACHE(re, &s->gb)) <= (int32_t)0xBFFFFFFF)
               break;

            UPDATE_CACHE(re, &s->gb);
215
        }
216 217
end:
        LAST_SKIP_BITS(re, &s->gb, 2);
218 219 220
        CLOSE_READER(re, &s->gb);
    }
    s->block_last_index[n] = i;
221
    return 0;
222 223 224 225 226 227 228
}

int ff_mpeg1_decode_block_intra(MpegEncContext *s, int16_t *block, int n)
{
    return mpeg1_decode_block_intra(s, block, n);
}

229 230
static inline int mpeg1_decode_block_inter(MpegEncContext *s,
                                           int16_t *block, int n)
231 232
{
    int level, i, j, run;
233 234
    RLTable *rl                  = &ff_rl_mpeg1;
    uint8_t *const scantable     = s->intra_scantable.permutated;
235 236 237 238 239 240 241 242
    const uint16_t *quant_matrix = s->inter_matrix;
    const int qscale             = s->qscale;

    {
        OPEN_READER(re, &s->gb);
        i = -1;
        // special case for first coefficient, no need to add second VLC table
        UPDATE_CACHE(re, &s->gb);
243
        if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
244 245 246 247 248 249 250
            level = (3 * qscale * quant_matrix[0]) >> 5;
            level = (level - 1) | 1;
            if (GET_CACHE(re, &s->gb) & 0x40000000)
                level = -level;
            block[0] = level;
            i++;
            SKIP_BITS(re, &s->gb, 2);
251
            if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
252 253 254 255
                goto end;
        }
        /* now quantify & encode AC coefficients */
        for (;;) {
256 257
            GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
                       TEX_VLC_BITS, 2, 0);
258 259 260

            if (level != 0) {
                i += run;
261
                check_scantable_index(s, i);
262 263 264
                j = scantable[i];
                level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
                level = (level - 1) | 1;
265 266
                level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
                        SHOW_SBITS(re, &s->gb, 1);
267 268 269
                SKIP_BITS(re, &s->gb, 1);
            } else {
                /* escape */
270 271
                run = SHOW_UBITS(re, &s->gb, 6) + 1;
                LAST_SKIP_BITS(re, &s->gb, 6);
272
                UPDATE_CACHE(re, &s->gb);
273 274
                level = SHOW_SBITS(re, &s->gb, 8);
                SKIP_BITS(re, &s->gb, 8);
275
                if (level == -128) {
276 277
                    level = SHOW_UBITS(re, &s->gb, 8) - 256;
                    SKIP_BITS(re, &s->gb, 8);
278
                } else if (level == 0) {
279 280
                    level = SHOW_UBITS(re, &s->gb, 8);
                    SKIP_BITS(re, &s->gb, 8);
281 282
                }
                i += run;
283
                check_scantable_index(s, i);
284 285 286 287 288 289 290 291 292 293 294 295 296
                j = scantable[i];
                if (level < 0) {
                    level = -level;
                    level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
                    level = (level - 1) | 1;
                    level = -level;
                } else {
                    level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
                    level = (level - 1) | 1;
                }
            }

            block[j] = level;
297
            if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
298 299 300 301 302 303 304 305 306 307 308
                break;
            UPDATE_CACHE(re, &s->gb);
        }
end:
        LAST_SKIP_BITS(re, &s->gb, 2);
        CLOSE_READER(re, &s->gb);
    }
    s->block_last_index[n] = i;
    return 0;
}

309 310 311 312 313
/**
 * Note: this function can read out of range and crash for corrupt streams.
 * Changing this would eat up any speed benefits it has.
 * Do not use "fast" flag if you need the code to be robust.
 */
314 315
static inline int mpeg1_fast_decode_block_inter(MpegEncContext *s,
                                                int16_t *block, int n)
316 317
{
    int level, i, j, run;
318 319 320
    RLTable *rl              = &ff_rl_mpeg1;
    uint8_t *const scantable = s->intra_scantable.permutated;
    const int qscale         = s->qscale;
321 322 323 324

    {
        OPEN_READER(re, &s->gb);
        i = -1;
325
        // Special case for first coefficient, no need to add second VLC table.
326
        UPDATE_CACHE(re, &s->gb);
327
        if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
328 329 330 331 332 333 334
            level = (3 * qscale) >> 1;
            level = (level - 1) | 1;
            if (GET_CACHE(re, &s->gb) & 0x40000000)
                level = -level;
            block[0] = level;
            i++;
            SKIP_BITS(re, &s->gb, 2);
335
            if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
336 337 338 339 340
                goto end;
        }

        /* now quantify & encode AC coefficients */
        for (;;) {
341 342
            GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
                       TEX_VLC_BITS, 2, 0);
343 344 345

            if (level != 0) {
                i += run;
346
                check_scantable_index(s, i);
347 348 349
                j = scantable[i];
                level = ((level * 2 + 1) * qscale) >> 1;
                level = (level - 1) | 1;
350 351
                level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
                        SHOW_SBITS(re, &s->gb, 1);
352 353 354
                SKIP_BITS(re, &s->gb, 1);
            } else {
                /* escape */
355 356
                run = SHOW_UBITS(re, &s->gb, 6) + 1;
                LAST_SKIP_BITS(re, &s->gb, 6);
357
                UPDATE_CACHE(re, &s->gb);
358 359
                level = SHOW_SBITS(re, &s->gb, 8);
                SKIP_BITS(re, &s->gb, 8);
360
                if (level == -128) {
361 362
                    level = SHOW_UBITS(re, &s->gb, 8) - 256;
                    SKIP_BITS(re, &s->gb, 8);
363
                } else if (level == 0) {
364 365
                    level = SHOW_UBITS(re, &s->gb, 8);
                    SKIP_BITS(re, &s->gb, 8);
366 367
                }
                i += run;
368
                check_scantable_index(s, i);
369 370 371 372 373 374 375 376 377 378 379 380 381
                j = scantable[i];
                if (level < 0) {
                    level = -level;
                    level = ((level * 2 + 1) * qscale) >> 1;
                    level = (level - 1) | 1;
                    level = -level;
                } else {
                    level = ((level * 2 + 1) * qscale) >> 1;
                    level = (level - 1) | 1;
                }
            }

            block[j] = level;
382
            if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
383 384 385 386 387 388 389 390 391 392 393
                break;
            UPDATE_CACHE(re, &s->gb);
        }
end:
        LAST_SKIP_BITS(re, &s->gb, 2);
        CLOSE_READER(re, &s->gb);
    }
    s->block_last_index[n] = i;
    return 0;
}

394 395
static inline int mpeg2_decode_block_non_intra(MpegEncContext *s,
                                               int16_t *block, int n)
396 397 398
{
    int level, i, j, run;
    RLTable *rl = &ff_rl_mpeg1;
399
    uint8_t *const scantable = s->intra_scantable.permutated;
400 401 402 403 404 405 406 407 408 409 410 411 412 413
    const uint16_t *quant_matrix;
    const int qscale = s->qscale;
    int mismatch;

    mismatch = 1;

    {
        OPEN_READER(re, &s->gb);
        i = -1;
        if (n < 4)
            quant_matrix = s->inter_matrix;
        else
            quant_matrix = s->chroma_inter_matrix;

414
        // Special case for first coefficient, no need to add second VLC table.
415
        UPDATE_CACHE(re, &s->gb);
416 417
        if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
            level = (3 * qscale * quant_matrix[0]) >> 5;
418 419 420 421 422 423
            if (GET_CACHE(re, &s->gb) & 0x40000000)
                level = -level;
            block[0]  = level;
            mismatch ^= level;
            i++;
            SKIP_BITS(re, &s->gb, 2);
424
            if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
425 426 427 428 429
                goto end;
        }

        /* now quantify & encode AC coefficients */
        for (;;) {
430 431
            GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
                       TEX_VLC_BITS, 2, 0);
432 433 434

            if (level != 0) {
                i += run;
435
                check_scantable_index(s, i);
436 437
                j = scantable[i];
                level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
438 439
                level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
                        SHOW_SBITS(re, &s->gb, 1);
440 441 442
                SKIP_BITS(re, &s->gb, 1);
            } else {
                /* escape */
443 444
                run = SHOW_UBITS(re, &s->gb, 6) + 1;
                LAST_SKIP_BITS(re, &s->gb, 6);
445
                UPDATE_CACHE(re, &s->gb);
446 447
                level = SHOW_SBITS(re, &s->gb, 12);
                SKIP_BITS(re, &s->gb, 12);
448 449

                i += run;
450
                check_scantable_index(s, i);
451 452 453 454 455 456 457 458 459 460 461
                j = scantable[i];
                if (level < 0) {
                    level = ((-level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
                    level = -level;
                } else {
                    level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
                }
            }

            mismatch ^= level;
            block[j]  = level;
462
            if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
463 464 465 466 467 468 469 470 471 472 473 474 475
                break;
            UPDATE_CACHE(re, &s->gb);
        }
end:
        LAST_SKIP_BITS(re, &s->gb, 2);
        CLOSE_READER(re, &s->gb);
    }
    block[63] ^= (mismatch & 1);

    s->block_last_index[n] = i;
    return 0;
}

476 477 478 479 480
/**
 * Note: this function can read out of range and crash for corrupt streams.
 * Changing this would eat up any speed benefits it has.
 * Do not use "fast" flag if you need the code to be robust.
 */
481 482 483 484
static inline int mpeg2_fast_decode_block_non_intra(MpegEncContext *s,
                                                    int16_t *block, int n)
{
    int level, i, j, run;
485 486 487
    RLTable *rl              = &ff_rl_mpeg1;
    uint8_t *const scantable = s->intra_scantable.permutated;
    const int qscale         = s->qscale;
488 489 490 491 492
    OPEN_READER(re, &s->gb);
    i = -1;

    // special case for first coefficient, no need to add second VLC table
    UPDATE_CACHE(re, &s->gb);
493
    if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
494 495 496 497 498 499
        level = (3 * qscale) >> 1;
        if (GET_CACHE(re, &s->gb) & 0x40000000)
            level = -level;
        block[0] = level;
        i++;
        SKIP_BITS(re, &s->gb, 2);
500
        if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
501 502 503 504 505 506 507 508 509
            goto end;
    }

    /* now quantify & encode AC coefficients */
    for (;;) {
        GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0], TEX_VLC_BITS, 2, 0);

        if (level != 0) {
            i += run;
510
            j = scantable[i];
511
            level = ((level * 2 + 1) * qscale) >> 1;
512 513
            level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
                    SHOW_SBITS(re, &s->gb, 1);
514 515 516
            SKIP_BITS(re, &s->gb, 1);
        } else {
            /* escape */
517 518
            run = SHOW_UBITS(re, &s->gb, 6) + 1;
            LAST_SKIP_BITS(re, &s->gb, 6);
519
            UPDATE_CACHE(re, &s->gb);
520 521
            level = SHOW_SBITS(re, &s->gb, 12);
            SKIP_BITS(re, &s->gb, 12);
522 523

            i += run;
524
            j = scantable[i];
525 526 527 528 529 530 531 532 533
            if (level < 0) {
                level = ((-level * 2 + 1) * qscale) >> 1;
                level = -level;
            } else {
                level = ((level * 2 + 1) * qscale) >> 1;
            }
        }

        block[j] = level;
534
        if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF || i > 63)
535
            break;
536

537 538 539 540 541 542 543 544 545
        UPDATE_CACHE(re, &s->gb);
    }
end:
    LAST_SKIP_BITS(re, &s->gb, 2);
    CLOSE_READER(re, &s->gb);
    s->block_last_index[n] = i;
    return 0;
}

546 547
static inline int mpeg2_decode_block_intra(MpegEncContext *s,
                                           int16_t *block, int n)
548 549 550 551
{
    int level, dc, diff, i, j, run;
    int component;
    RLTable *rl;
552
    uint8_t *const scantable = s->intra_scantable.permutated;
553 554 555 556 557 558 559
    const uint16_t *quant_matrix;
    const int qscale = s->qscale;
    int mismatch;

    /* DC coefficient */
    if (n < 4) {
        quant_matrix = s->intra_matrix;
560
        component    = 0;
561 562
    } else {
        quant_matrix = s->chroma_intra_matrix;
563
        component    = (n & 1) + 1;
564 565 566
    }
    diff = decode_dc(&s->gb, component);
    if (diff >= 0xffff)
567
        return AVERROR_INVALIDDATA;
568 569 570 571
    dc  = s->last_dc[component];
    dc += diff;
    s->last_dc[component] = dc;
    block[0] = dc << (3 - s->intra_dc_precision);
572
    ff_tlog(s->avctx, "dc=%d\n", block[0]);
573 574 575 576 577 578 579 580 581 582 583 584
    mismatch = block[0] ^ 1;
    i = 0;
    if (s->intra_vlc_format)
        rl = &ff_rl_mpeg2;
    else
        rl = &ff_rl_mpeg1;

    {
        OPEN_READER(re, &s->gb);
        /* now quantify & encode AC coefficients */
        for (;;) {
            UPDATE_CACHE(re, &s->gb);
585 586
            GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
                       TEX_VLC_BITS, 2, 0);
587 588 589 590 591

            if (level == 127) {
                break;
            } else if (level != 0) {
                i += run;
592
                check_scantable_index(s, i);
593
                j = scantable[i];
594
                level = (level * qscale * quant_matrix[j]) >> 4;
595 596
                level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
                        SHOW_SBITS(re, &s->gb, 1);
597 598 599
                LAST_SKIP_BITS(re, &s->gb, 1);
            } else {
                /* escape */
600 601
                run = SHOW_UBITS(re, &s->gb, 6) + 1;
                LAST_SKIP_BITS(re, &s->gb, 6);
602
                UPDATE_CACHE(re, &s->gb);
603 604
                level = SHOW_SBITS(re, &s->gb, 12);
                SKIP_BITS(re, &s->gb, 12);
605
                i += run;
606
                check_scantable_index(s, i);
607
                j = scantable[i];
608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626
                if (level < 0) {
                    level = (-level * qscale * quant_matrix[j]) >> 4;
                    level = -level;
                } else {
                    level = (level * qscale * quant_matrix[j]) >> 4;
                }
            }

            mismatch ^= level;
            block[j]  = level;
        }
        CLOSE_READER(re, &s->gb);
    }
    block[63] ^= mismatch & 1;

    s->block_last_index[n] = i;
    return 0;
}

627 628 629 630 631
/**
 * Note: this function can read out of range and crash for corrupt streams.
 * Changing this would eat up any speed benefits it has.
 * Do not use "fast" flag if you need the code to be robust.
 */
632 633
static inline int mpeg2_fast_decode_block_intra(MpegEncContext *s,
                                                int16_t *block, int n)
634
{
635
    int level, dc, diff, i, j, run;
636 637
    int component;
    RLTable *rl;
638
    uint8_t *const scantable = s->intra_scantable.permutated;
639 640 641 642 643 644
    const uint16_t *quant_matrix;
    const int qscale = s->qscale;

    /* DC coefficient */
    if (n < 4) {
        quant_matrix = s->intra_matrix;
645
        component    = 0;
646 647
    } else {
        quant_matrix = s->chroma_intra_matrix;
648
        component    = (n & 1) + 1;
649 650 651
    }
    diff = decode_dc(&s->gb, component);
    if (diff >= 0xffff)
652
        return AVERROR_INVALIDDATA;
653 654 655 656
    dc = s->last_dc[component];
    dc += diff;
    s->last_dc[component] = dc;
    block[0] = dc << (3 - s->intra_dc_precision);
657
    i = 0;
658 659 660 661 662 663 664 665 666 667
    if (s->intra_vlc_format)
        rl = &ff_rl_mpeg2;
    else
        rl = &ff_rl_mpeg1;

    {
        OPEN_READER(re, &s->gb);
        /* now quantify & encode AC coefficients */
        for (;;) {
            UPDATE_CACHE(re, &s->gb);
668 669
            GET_RL_VLC(level, run, re, &s->gb, rl->rl_vlc[0],
                       TEX_VLC_BITS, 2, 0);
670

671
            if (level >= 64 || i > 63) {
672 673
                break;
            } else if (level != 0) {
674
                i += run;
675
                j = scantable[i];
676
                level = (level * qscale * quant_matrix[j]) >> 4;
677 678
                level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
                        SHOW_SBITS(re, &s->gb, 1);
679 680 681
                LAST_SKIP_BITS(re, &s->gb, 1);
            } else {
                /* escape */
682 683
                run = SHOW_UBITS(re, &s->gb, 6) + 1;
                LAST_SKIP_BITS(re, &s->gb, 6);
684
                UPDATE_CACHE(re, &s->gb);
685 686
                level = SHOW_SBITS(re, &s->gb, 12);
                SKIP_BITS(re, &s->gb, 12);
687
                i += run;
688
                j = scantable[i];
689 690 691 692 693 694 695 696 697 698 699 700 701
                if (level < 0) {
                    level = (-level * qscale * quant_matrix[j]) >> 4;
                    level = -level;
                } else {
                    level = (level * qscale * quant_matrix[j]) >> 4;
                }
            }

            block[j] = level;
        }
        CLOSE_READER(re, &s->gb);
    }

702
    s->block_last_index[n] = i;
703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719
    return 0;
}

/******************************************/
/* decoding */

static inline int get_dmv(MpegEncContext *s)
{
    if (get_bits1(&s->gb))
        return 1 - (get_bits1(&s->gb) << 1);
    else
        return 0;
}

static inline int get_qscale(MpegEncContext *s)
{
    int qscale = get_bits(&s->gb, 5);
720
    if (s->q_scale_type)
721
        return non_linear_qscale[qscale];
722
    else
723 724 725 726 727 728 729 730 731 732 733 734 735 736
        return qscale << 1;
}


/* motion type (for MPEG-2) */
#define MT_FIELD 1
#define MT_FRAME 2
#define MT_16X8  2
#define MT_DMV   3

static int mpeg_decode_mb(MpegEncContext *s, int16_t block[12][64])
{
    int i, j, k, cbp, val, mb_type, motion_type;
    const int mb_block_count = 4 + (1 << s->chroma_format);
737
    int ret;
738

739
    ff_tlog(s->avctx, "decode_mb: x=%d y=%d\n", s->mb_x, s->mb_y);
740

741
    av_assert2(s->mb_skipped == 0);
742 743 744 745

    if (s->mb_skip_run-- != 0) {
        if (s->pict_type == AV_PICTURE_TYPE_P) {
            s->mb_skipped = 1;
746 747
            s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride] =
                MB_TYPE_SKIP | MB_TYPE_L0 | MB_TYPE_16x16;
748 749 750 751 752 753
        } else {
            int mb_type;

            if (s->mb_x)
                mb_type = s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride - 1];
            else
754 755
                // FIXME not sure if this is allowed in MPEG at all
                mb_type = s->current_picture.mb_type[s->mb_width + (s->mb_y - 1) * s->mb_stride - 1];
756 757
            if (IS_INTRA(mb_type)) {
                av_log(s->avctx, AV_LOG_ERROR, "skip with previntra\n");
758
                return AVERROR_INVALIDDATA;
759
            }
760
            s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride] =
761
                mb_type | MB_TYPE_SKIP;
762

763 764 765 766 767 768 769 770 771 772 773 774
            if ((s->mv[0][0][0] | s->mv[0][0][1] | s->mv[1][0][0] | s->mv[1][0][1]) == 0)
                s->mb_skipped = 1;
        }

        return 0;
    }

    switch (s->pict_type) {
    default:
    case AV_PICTURE_TYPE_I:
        if (get_bits1(&s->gb) == 0) {
            if (get_bits1(&s->gb) == 0) {
775 776 777
                av_log(s->avctx, AV_LOG_ERROR,
                       "invalid mb type in I Frame at %d %d\n",
                       s->mb_x, s->mb_y);
778
                return AVERROR_INVALIDDATA;
779 780 781 782 783 784 785 786 787
            }
            mb_type = MB_TYPE_QUANT | MB_TYPE_INTRA;
        } else {
            mb_type = MB_TYPE_INTRA;
        }
        break;
    case AV_PICTURE_TYPE_P:
        mb_type = get_vlc2(&s->gb, ff_mb_ptype_vlc.table, MB_PTYPE_VLC_BITS, 1);
        if (mb_type < 0) {
788 789
            av_log(s->avctx, AV_LOG_ERROR,
                   "invalid mb type in P Frame at %d %d\n", s->mb_x, s->mb_y);
790
            return AVERROR_INVALIDDATA;
791 792 793 794 795 796
        }
        mb_type = ptype2mb_type[mb_type];
        break;
    case AV_PICTURE_TYPE_B:
        mb_type = get_vlc2(&s->gb, ff_mb_btype_vlc.table, MB_BTYPE_VLC_BITS, 1);
        if (mb_type < 0) {
797 798
            av_log(s->avctx, AV_LOG_ERROR,
                   "invalid mb type in B Frame at %d %d\n", s->mb_x, s->mb_y);
799
            return AVERROR_INVALIDDATA;
800 801 802 803
        }
        mb_type = btype2mb_type[mb_type];
        break;
    }
804
    ff_tlog(s->avctx, "mb_type=%x\n", mb_type);
805 806
//    motion_type = 0; /* avoid warning */
    if (IS_INTRA(mb_type)) {
807
        s->bdsp.clear_blocks(s->block[0]);
808

809
        if (!s->chroma_y_shift)
810
            s->bdsp.clear_blocks(s->block[6]);
811 812

        /* compute DCT type */
813 814 815
        // FIXME: add an interlaced_dct coded var?
        if (s->picture_structure == PICT_FRAME &&
            !s->frame_pred_frame_dct)
816 817 818 819 820 821 822 823
            s->interlaced_dct = get_bits1(&s->gb);

        if (IS_QUANT(mb_type))
            s->qscale = get_qscale(s);

        if (s->concealment_motion_vectors) {
            /* just parse them */
            if (s->picture_structure != PICT_FRAME)
824
                skip_bits1(&s->gb);  /* field select */
825

826 827 828 829 830 831 832 833
            s->mv[0][0][0]      =
            s->last_mv[0][0][0] =
            s->last_mv[0][1][0] = mpeg_decode_motion(s, s->mpeg_f_code[0][0],
                                                     s->last_mv[0][0][0]);
            s->mv[0][0][1]      =
            s->last_mv[0][0][1] =
            s->last_mv[0][1][1] = mpeg_decode_motion(s, s->mpeg_f_code[0][1],
                                                     s->last_mv[0][0][1]);
834

835
            check_marker(&s->gb, "after concealment_motion_vectors");
836 837 838 839
        } else {
            /* reset mv prediction */
            memset(s->last_mv, 0, sizeof(s->last_mv));
        }
840 841
        s->mb_intra = 1;
        // if 1, we memcpy blocks in xvmcvideo
842
        if ((CONFIG_MPEG1_XVMC_HWACCEL || CONFIG_MPEG2_XVMC_HWACCEL) && s->pack_pblocks)
843 844 845
            ff_xvmc_pack_pblocks(s, -1); // inter are always full blocks

        if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
846
            if (s->avctx->flags2 & CODEC_FLAG2_FAST) {
847
                for (i = 0; i < 6; i++)
848 849
                    mpeg2_fast_decode_block_intra(s, *s->pblocks[i], i);
            } else {
850
                for (i = 0; i < mb_block_count; i++)
851 852
                    if ((ret = mpeg2_decode_block_intra(s, *s->pblocks[i], i)) < 0)
                        return ret;
853 854
            }
        } else {
855
            for (i = 0; i < 6; i++)
856 857
                if ((ret = mpeg1_decode_block_intra(s, *s->pblocks[i], i)) < 0)
                    return ret;
858 859 860
        }
    } else {
        if (mb_type & MB_TYPE_ZERO_MV) {
861
            av_assert2(mb_type & MB_TYPE_CBP);
862 863 864

            s->mv_dir = MV_DIR_FORWARD;
            if (s->picture_structure == PICT_FRAME) {
865 866
                if (s->picture_structure == PICT_FRAME
                    && !s->frame_pred_frame_dct)
867 868 869
                    s->interlaced_dct = get_bits1(&s->gb);
                s->mv_type = MV_TYPE_16X16;
            } else {
870 871
                s->mv_type            = MV_TYPE_FIELD;
                mb_type              |= MB_TYPE_INTERLACED;
872 873 874 875 876 877 878 879 880 881
                s->field_select[0][0] = s->picture_structure - 1;
            }

            if (IS_QUANT(mb_type))
                s->qscale = get_qscale(s);

            s->last_mv[0][0][0] = 0;
            s->last_mv[0][0][1] = 0;
            s->last_mv[0][1][0] = 0;
            s->last_mv[0][1][1] = 0;
882 883
            s->mv[0][0][0]      = 0;
            s->mv[0][0][1]      = 0;
884
        } else {
885
            av_assert2(mb_type & MB_TYPE_L0L1);
886 887
            // FIXME decide if MBs in field pictures are MB_TYPE_INTERLACED
            /* get additional motion vector type */
888
            if (s->picture_structure == PICT_FRAME && s->frame_pred_frame_dct) {
889
                motion_type = MT_FRAME;
890
            } else {
891 892 893 894 895 896 897 898 899 900
                motion_type = get_bits(&s->gb, 2);
                if (s->picture_structure == PICT_FRAME && HAS_CBP(mb_type))
                    s->interlaced_dct = get_bits1(&s->gb);
            }

            if (IS_QUANT(mb_type))
                s->qscale = get_qscale(s);

            /* motion vectors */
            s->mv_dir = (mb_type >> 13) & 3;
901
            ff_tlog(s->avctx, "motion_type=%d\n", motion_type);
902 903 904
            switch (motion_type) {
            case MT_FRAME: /* or MT_16X8 */
                if (s->picture_structure == PICT_FRAME) {
905
                    mb_type   |= MB_TYPE_16x16;
906 907 908 909
                    s->mv_type = MV_TYPE_16X16;
                    for (i = 0; i < 2; i++) {
                        if (USES_LIST(mb_type, i)) {
                            /* MT_FRAME */
910 911 912 913 914 915 916 917 918 919
                            s->mv[i][0][0]      =
                            s->last_mv[i][0][0] =
                            s->last_mv[i][1][0] =
                                mpeg_decode_motion(s, s->mpeg_f_code[i][0],
                                                   s->last_mv[i][0][0]);
                            s->mv[i][0][1]      =
                            s->last_mv[i][0][1] =
                            s->last_mv[i][1][1] =
                                mpeg_decode_motion(s, s->mpeg_f_code[i][1],
                                                   s->last_mv[i][0][1]);
920 921 922 923 924 925 926 927
                            /* full_pel: only for MPEG-1 */
                            if (s->full_pel[i]) {
                                s->mv[i][0][0] <<= 1;
                                s->mv[i][0][1] <<= 1;
                            }
                        }
                    }
                } else {
928
                    mb_type   |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957
                    s->mv_type = MV_TYPE_16X8;
                    for (i = 0; i < 2; i++) {
                        if (USES_LIST(mb_type, i)) {
                            /* MT_16X8 */
                            for (j = 0; j < 2; j++) {
                                s->field_select[i][j] = get_bits1(&s->gb);
                                for (k = 0; k < 2; k++) {
                                    val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
                                                             s->last_mv[i][j][k]);
                                    s->last_mv[i][j][k] = val;
                                    s->mv[i][j][k]      = val;
                                }
                            }
                        }
                    }
                }
                break;
            case MT_FIELD:
                s->mv_type = MV_TYPE_FIELD;
                if (s->picture_structure == PICT_FRAME) {
                    mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
                    for (i = 0; i < 2; i++) {
                        if (USES_LIST(mb_type, i)) {
                            for (j = 0; j < 2; j++) {
                                s->field_select[i][j] = get_bits1(&s->gb);
                                val = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
                                                         s->last_mv[i][j][0]);
                                s->last_mv[i][j][0] = val;
                                s->mv[i][j][0]      = val;
958
                                ff_tlog(s->avctx, "fmx=%d\n", val);
959 960
                                val = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
                                                         s->last_mv[i][j][1] >> 1);
961
                                s->last_mv[i][j][1] = 2 * val;
962
                                s->mv[i][j][1]      = val;
963
                                ff_tlog(s->avctx, "fmy=%d\n", val);
964 965 966 967
                            }
                        }
                    }
                } else {
968
                    av_assert0(!s->progressive_sequence);
969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984
                    mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
                    for (i = 0; i < 2; i++) {
                        if (USES_LIST(mb_type, i)) {
                            s->field_select[i][0] = get_bits1(&s->gb);
                            for (k = 0; k < 2; k++) {
                                val = mpeg_decode_motion(s, s->mpeg_f_code[i][k],
                                                         s->last_mv[i][0][k]);
                                s->last_mv[i][0][k] = val;
                                s->last_mv[i][1][k] = val;
                                s->mv[i][0][k]      = val;
                            }
                        }
                    }
                }
                break;
            case MT_DMV:
985
                if (s->progressive_sequence){
986
                    av_log(s->avctx, AV_LOG_ERROR, "MT_DMV in progressive_sequence\n");
987
                    return AVERROR_INVALIDDATA;
988
                }
989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038
                s->mv_type = MV_TYPE_DMV;
                for (i = 0; i < 2; i++) {
                    if (USES_LIST(mb_type, i)) {
                        int dmx, dmy, mx, my, m;
                        const int my_shift = s->picture_structure == PICT_FRAME;

                        mx = mpeg_decode_motion(s, s->mpeg_f_code[i][0],
                                                s->last_mv[i][0][0]);
                        s->last_mv[i][0][0] = mx;
                        s->last_mv[i][1][0] = mx;
                        dmx = get_dmv(s);
                        my  = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
                                                 s->last_mv[i][0][1] >> my_shift);
                        dmy = get_dmv(s);


                        s->last_mv[i][0][1] = my << my_shift;
                        s->last_mv[i][1][1] = my << my_shift;

                        s->mv[i][0][0] = mx;
                        s->mv[i][0][1] = my;
                        s->mv[i][1][0] = mx; // not used
                        s->mv[i][1][1] = my; // not used

                        if (s->picture_structure == PICT_FRAME) {
                            mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;

                            // m = 1 + 2 * s->top_field_first;
                            m = s->top_field_first ? 1 : 3;

                            /* top -> top pred */
                            s->mv[i][2][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
                            s->mv[i][2][1] = ((my * m + (my > 0)) >> 1) + dmy - 1;
                            m = 4 - m;
                            s->mv[i][3][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
                            s->mv[i][3][1] = ((my * m + (my > 0)) >> 1) + dmy + 1;
                        } else {
                            mb_type |= MB_TYPE_16x16;

                            s->mv[i][2][0] = ((mx + (mx > 0)) >> 1) + dmx;
                            s->mv[i][2][1] = ((my + (my > 0)) >> 1) + dmy;
                            if (s->picture_structure == PICT_TOP_FIELD)
                                s->mv[i][2][1]--;
                            else
                                s->mv[i][2][1]++;
                        }
                    }
                }
                break;
            default:
1039 1040
                av_log(s->avctx, AV_LOG_ERROR,
                       "00 motion_type at %d %d\n", s->mb_x, s->mb_y);
1041
                return AVERROR_INVALIDDATA;
1042 1043 1044 1045 1046
            }
        }

        s->mb_intra = 0;
        if (HAS_CBP(mb_type)) {
1047
            s->bdsp.clear_blocks(s->block[0]);
1048 1049 1050

            cbp = get_vlc2(&s->gb, ff_mb_pat_vlc.table, MB_PAT_VLC_BITS, 1);
            if (mb_block_count > 6) {
1051 1052
                cbp <<= mb_block_count - 6;
                cbp  |= get_bits(&s->gb, mb_block_count - 6);
1053
                s->bdsp.clear_blocks(s->block[6]);
1054 1055
            }
            if (cbp <= 0) {
1056
                av_log(s->avctx, AV_LOG_ERROR,
1057
                       "invalid cbp %d at %d %d\n", cbp, s->mb_x, s->mb_y);
1058
                return AVERROR_INVALIDDATA;
1059 1060
            }

1061
            // if 1, we memcpy blocks in xvmcvideo
1062
            if ((CONFIG_MPEG1_XVMC_HWACCEL || CONFIG_MPEG2_XVMC_HWACCEL) && s->pack_pblocks)
1063 1064 1065
                ff_xvmc_pack_pblocks(s, cbp);

            if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
1066
                if (s->avctx->flags2 & CODEC_FLAG2_FAST) {
1067
                    for (i = 0; i < 6; i++) {
1068
                        if (cbp & 32)
1069
                            mpeg2_fast_decode_block_non_intra(s, *s->pblocks[i], i);
1070
                        else
1071 1072 1073 1074
                            s->block_last_index[i] = -1;
                        cbp += cbp;
                    }
                } else {
1075
                    cbp <<= 12 - mb_block_count;
1076 1077 1078

                    for (i = 0; i < mb_block_count; i++) {
                        if (cbp & (1 << 11)) {
1079 1080
                            if ((ret = mpeg2_decode_block_non_intra(s, *s->pblocks[i], i)) < 0)
                                return ret;
1081 1082 1083 1084 1085 1086 1087
                        } else {
                            s->block_last_index[i] = -1;
                        }
                        cbp += cbp;
                    }
                }
            } else {
1088
                if (s->avctx->flags2 & CODEC_FLAG2_FAST) {
1089
                    for (i = 0; i < 6; i++) {
1090
                        if (cbp & 32)
1091
                            mpeg1_fast_decode_block_inter(s, *s->pblocks[i], i);
1092
                        else
1093 1094 1095 1096 1097 1098
                            s->block_last_index[i] = -1;
                        cbp += cbp;
                    }
                } else {
                    for (i = 0; i < 6; i++) {
                        if (cbp & 32) {
1099 1100
                            if ((ret = mpeg1_decode_block_inter(s, *s->pblocks[i], i)) < 0)
                                return ret;
1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120
                        } else {
                            s->block_last_index[i] = -1;
                        }
                        cbp += cbp;
                    }
                }
            }
        } else {
            for (i = 0; i < 12; i++)
                s->block_last_index[i] = -1;
        }
    }

    s->current_picture.mb_type[s->mb_x + s->mb_y * s->mb_stride] = mb_type;

    return 0;
}

static av_cold int mpeg_decode_init(AVCodecContext *avctx)
{
1121
    Mpeg1Context *s    = avctx->priv_data;
1122 1123
    MpegEncContext *s2 = &s->mpeg_enc_ctx;

1124
    ff_mpv_decode_defaults(s2);
1125 1126 1127 1128

    if (   avctx->codec_tag != AV_RL32("VCR2")
        && avctx->codec_tag != AV_RL32("BW10"))
        avctx->coded_width = avctx->coded_height = 0; // do not trust dimensions from input
1129
    ff_mpv_decode_init(s2, avctx);
1130 1131

    s->mpeg_enc_ctx.avctx  = avctx;
1132 1133 1134 1135

    /* we need some permutation to store matrices,
     * until the decoder sets the real permutation. */
    ff_mpv_idct_init(s2);
1136 1137 1138 1139 1140 1141 1142
    ff_mpeg12_common_init(&s->mpeg_enc_ctx);
    ff_mpeg12_init_vlcs();

    s->mpeg_enc_ctx_allocated      = 0;
    s->mpeg_enc_ctx.picture_number = 0;
    s->repeat_field                = 0;
    s->mpeg_enc_ctx.codec_id       = avctx->codec->id;
1143
    avctx->color_range             = AVCOL_RANGE_MPEG;
1144 1145 1146
    return 0;
}

1147 1148
static int mpeg_decode_update_thread_context(AVCodecContext *avctx,
                                             const AVCodecContext *avctx_from)
1149 1150 1151 1152 1153
{
    Mpeg1Context *ctx = avctx->priv_data, *ctx_from = avctx_from->priv_data;
    MpegEncContext *s = &ctx->mpeg_enc_ctx, *s1 = &ctx_from->mpeg_enc_ctx;
    int err;

1154 1155 1156
    if (avctx == avctx_from               ||
        !ctx_from->mpeg_enc_ctx_allocated ||
        !s1->context_initialized)
1157 1158 1159
        return 0;

    err = ff_mpeg_update_thread_context(avctx, avctx_from);
1160 1161
    if (err)
        return err;
1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179

    if (!ctx->mpeg_enc_ctx_allocated)
        memcpy(s + 1, s1 + 1, sizeof(Mpeg1Context) - sizeof(MpegEncContext));

    if (!(s->pict_type == AV_PICTURE_TYPE_B || s->low_delay))
        s->picture_number++;

    return 0;
}

static void quant_matrix_rebuild(uint16_t *matrix, const uint8_t *old_perm,
                                 const uint8_t *new_perm)
{
    uint16_t temp_matrix[64];
    int i;

    memcpy(temp_matrix, matrix, 64 * sizeof(uint16_t));

1180
    for (i = 0; i < 64; i++)
1181 1182 1183
        matrix[new_perm[i]] = temp_matrix[old_perm[i]];
}

1184
static const enum AVPixelFormat mpeg1_hwaccel_pixfmt_list_420[] = {
1185 1186
#if CONFIG_MPEG1_XVMC_HWACCEL
    AV_PIX_FMT_XVMC,
1187 1188 1189
#endif
#if CONFIG_MPEG1_VDPAU_HWACCEL
    AV_PIX_FMT_VDPAU_MPEG1,
1190
    AV_PIX_FMT_VDPAU,
1191 1192 1193 1194
#endif
    AV_PIX_FMT_YUV420P,
    AV_PIX_FMT_NONE
};
1195

1196
static const enum AVPixelFormat mpeg2_hwaccel_pixfmt_list_420[] = {
1197 1198
#if CONFIG_MPEG2_XVMC_HWACCEL
    AV_PIX_FMT_XVMC,
1199 1200 1201
#endif
#if CONFIG_MPEG2_VDPAU_HWACCEL
    AV_PIX_FMT_VDPAU_MPEG2,
1202
    AV_PIX_FMT_VDPAU,
1203
#endif
1204 1205 1206 1207 1208 1209 1210 1211 1212 1213
#if CONFIG_MPEG2_DXVA2_HWACCEL
    AV_PIX_FMT_DXVA2_VLD,
#endif
#if CONFIG_MPEG2_VAAPI_HWACCEL
    AV_PIX_FMT_VAAPI_VLD,
#endif
    AV_PIX_FMT_YUV420P,
    AV_PIX_FMT_NONE
};

1214 1215 1216 1217 1218 1219 1220 1221 1222 1223
static const enum AVPixelFormat mpeg12_pixfmt_list_422[] = {
    AV_PIX_FMT_YUV422P,
    AV_PIX_FMT_NONE
};

static const enum AVPixelFormat mpeg12_pixfmt_list_444[] = {
    AV_PIX_FMT_YUV444P,
    AV_PIX_FMT_NONE
};

1224 1225 1226 1227
static inline int uses_vdpau(AVCodecContext *avctx) {
    return avctx->pix_fmt == AV_PIX_FMT_VDPAU_MPEG1 || avctx->pix_fmt == AV_PIX_FMT_VDPAU_MPEG2;
}

1228 1229
static enum AVPixelFormat mpeg_get_pixelformat(AVCodecContext *avctx)
{
1230
    Mpeg1Context *s1  = avctx->priv_data;
1231
    MpegEncContext *s = &s1->mpeg_enc_ctx;
1232
    const enum AVPixelFormat *pix_fmts;
1233

1234 1235 1236
    if (CONFIG_GRAY && (avctx->flags & CODEC_FLAG_GRAY))
        return AV_PIX_FMT_GRAY8;

1237
    if (s->chroma_format < 2)
1238
        pix_fmts = avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO ?
1239
                                mpeg1_hwaccel_pixfmt_list_420 :
1240
                                mpeg2_hwaccel_pixfmt_list_420;
1241
    else if (s->chroma_format == 2)
1242
        pix_fmts = mpeg12_pixfmt_list_422;
1243
    else
1244 1245
        pix_fmts = mpeg12_pixfmt_list_444;

1246
    return ff_thread_get_format(avctx, pix_fmts);
1247 1248 1249 1250 1251
}

static void setup_hwaccel_for_pixfmt(AVCodecContext *avctx)
{
    // until then pix_fmt may be changed right after codec init
1252
    if (avctx->hwaccel || uses_vdpau(avctx))
1253 1254
        if (avctx->idct_algo == FF_IDCT_AUTO)
            avctx->idct_algo = FF_IDCT_SIMPLE;
1255 1256 1257 1258 1259 1260 1261 1262 1263 1264

    if (avctx->hwaccel && avctx->pix_fmt == AV_PIX_FMT_XVMC) {
        Mpeg1Context *s1 = avctx->priv_data;
        MpegEncContext *s = &s1->mpeg_enc_ctx;

        s->pack_pblocks = 1;
#if FF_API_XVMC
        avctx->xvmc_acceleration = 2;
#endif /* FF_API_XVMC */
    }
1265 1266 1267 1268 1269 1270
}

/* Call this function when we know all parameters.
 * It may be called in different places for MPEG-1 and MPEG-2. */
static int mpeg_decode_postinit(AVCodecContext *avctx)
{
1271
    Mpeg1Context *s1  = avctx->priv_data;
1272 1273
    MpegEncContext *s = &s1->mpeg_enc_ctx;
    uint8_t old_permutation[64];
1274
    int ret;
1275

1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305
    if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
        // MPEG-1 aspect
        avctx->sample_aspect_ratio = av_d2q(1.0 / ff_mpeg1_aspect[s->aspect_ratio_info], 255);
    } else { // MPEG-2
        // MPEG-2 aspect
        if (s->aspect_ratio_info > 1) {
            AVRational dar =
                av_mul_q(av_div_q(ff_mpeg2_aspect[s->aspect_ratio_info],
                                  (AVRational) { s1->pan_scan.width,
                                                 s1->pan_scan.height }),
                         (AVRational) { s->width, s->height });

            /* We ignore the spec here and guess a bit as reality does not
             * match the spec, see for example res_change_ffmpeg_aspect.ts
             * and sequence-display-aspect.mpg.
             * issue1613, 621, 562 */
            if ((s1->pan_scan.width == 0) || (s1->pan_scan.height == 0) ||
                (av_cmp_q(dar, (AVRational) { 4, 3 }) &&
                 av_cmp_q(dar, (AVRational) { 16, 9 }))) {
                s->avctx->sample_aspect_ratio =
                    av_div_q(ff_mpeg2_aspect[s->aspect_ratio_info],
                             (AVRational) { s->width, s->height });
            } else {
                s->avctx->sample_aspect_ratio =
                    av_div_q(ff_mpeg2_aspect[s->aspect_ratio_info],
                             (AVRational) { s1->pan_scan.width, s1->pan_scan.height });
// issue1613 4/3 16/9 -> 16/9
// res_change_ffmpeg_aspect.ts 4/3 225/44 ->4/3
// widescreen-issue562.mpg 4/3 16/9 -> 16/9
//                s->avctx->sample_aspect_ratio = av_mul_q(s->avctx->sample_aspect_ratio, (AVRational) {s->width, s->height});
1306
                ff_dlog(avctx, "aspect A %d/%d\n",
1307 1308
                        ff_mpeg2_aspect[s->aspect_ratio_info].num,
                        ff_mpeg2_aspect[s->aspect_ratio_info].den);
1309
                ff_dlog(avctx, "aspect B %d/%d\n", s->avctx->sample_aspect_ratio.num,
1310 1311 1312 1313 1314 1315 1316 1317
                        s->avctx->sample_aspect_ratio.den);
            }
        } else {
            s->avctx->sample_aspect_ratio =
                ff_mpeg2_aspect[s->aspect_ratio_info];
        }
    } // MPEG-2

1318 1319 1320 1321 1322 1323 1324
    if (av_image_check_sar(s->width, s->height,
                           avctx->sample_aspect_ratio) < 0) {
        av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
                avctx->sample_aspect_ratio.num,
                avctx->sample_aspect_ratio.den);
        avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
    }
1325

1326 1327 1328
    if ((s1->mpeg_enc_ctx_allocated == 0)                   ||
        avctx->coded_width       != s->width                ||
        avctx->coded_height      != s->height               ||
1329 1330
        s1->save_width           != s->width                ||
        s1->save_height          != s->height               ||
1331
        av_cmp_q(s1->save_aspect, s->avctx->sample_aspect_ratio) ||
1332
        (s1->save_progressive_seq != s->progressive_sequence && FFALIGN(s->height, 16) != FFALIGN(s->height, 32)) ||
1333
        0) {
1334 1335 1336
        if (s1->mpeg_enc_ctx_allocated) {
            ParseContext pc = s->parse_context;
            s->parse_context.buffer = 0;
1337
            ff_mpv_common_end(s);
1338
            s->parse_context = pc;
1339
            s1->mpeg_enc_ctx_allocated = 0;
1340 1341
        }

1342 1343 1344 1345
        ret = ff_set_dimensions(avctx, s->width, s->height);
        if (ret < 0)
            return ret;

1346 1347 1348 1349 1350 1351
        if (avctx->codec_id == AV_CODEC_ID_MPEG2VIDEO && s->bit_rate) {
            avctx->rc_max_rate = s->bit_rate;
        } else if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO && s->bit_rate &&
                   (s->bit_rate != 0x3FFFF*400 || s->vbv_delay != 0xFFFF)) {
            avctx->bit_rate = s->bit_rate;
        }
1352
        s1->save_aspect          = s->avctx->sample_aspect_ratio;
1353 1354 1355 1356 1357 1358 1359 1360 1361
        s1->save_width           = s->width;
        s1->save_height          = s->height;
        s1->save_progressive_seq = s->progressive_sequence;

        /* low_delay may be forced, in this case we will have B-frames
         * that behave like P-frames. */
        avctx->has_b_frames = !s->low_delay;

        if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
1362
            // MPEG-1 fps
1363
            avctx->framerate = ff_mpeg12_frame_rate_tab[s->frame_rate_index];
1364
            avctx->ticks_per_frame     = 1;
1365 1366

            avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
1367 1368
        } else { // MPEG-2
            // MPEG-2 fps
1369 1370
            av_reduce(&s->avctx->framerate.num,
                      &s->avctx->framerate.den,
1371
                      ff_mpeg12_frame_rate_tab[s->frame_rate_index].num * s1->frame_rate_ext.num,
1372 1373 1374
                      ff_mpeg12_frame_rate_tab[s->frame_rate_index].den * s1->frame_rate_ext.den,
                      1 << 30);
            avctx->ticks_per_frame = 2;
1375 1376 1377 1378 1379 1380

            switch (s->chroma_format) {
            case 1: avctx->chroma_sample_location = AVCHROMA_LOC_LEFT; break;
            case 2:
            case 3: avctx->chroma_sample_location = AVCHROMA_LOC_TOPLEFT; break;
            }
1381 1382 1383
        } // MPEG-2

        avctx->pix_fmt = mpeg_get_pixelformat(avctx);
1384
        setup_hwaccel_for_pixfmt(avctx);
1385 1386 1387

        /* Quantization matrices may need reordering
         * if DCT permutation is changed. */
1388
        memcpy(old_permutation, s->idsp.idct_permutation, 64 * sizeof(uint8_t));
1389

1390
        ff_mpv_idct_init(s);
1391 1392
        if ((ret = ff_mpv_common_init(s)) < 0)
            return ret;
1393

1394 1395 1396 1397
        quant_matrix_rebuild(s->intra_matrix,        old_permutation, s->idsp.idct_permutation);
        quant_matrix_rebuild(s->inter_matrix,        old_permutation, s->idsp.idct_permutation);
        quant_matrix_rebuild(s->chroma_intra_matrix, old_permutation, s->idsp.idct_permutation);
        quant_matrix_rebuild(s->chroma_inter_matrix, old_permutation, s->idsp.idct_permutation);
1398 1399 1400 1401 1402 1403

        s1->mpeg_enc_ctx_allocated = 1;
    }
    return 0;
}

1404 1405
static int mpeg1_decode_picture(AVCodecContext *avctx, const uint8_t *buf,
                                int buf_size)
1406
{
1407
    Mpeg1Context *s1  = avctx->priv_data;
1408 1409 1410
    MpegEncContext *s = &s1->mpeg_enc_ctx;
    int ref, f_code, vbv_delay;

1411
    init_get_bits(&s->gb, buf, buf_size * 8);
1412 1413 1414 1415

    ref = get_bits(&s->gb, 10); /* temporal ref */
    s->pict_type = get_bits(&s->gb, 3);
    if (s->pict_type == 0 || s->pict_type > 3)
1416
        return AVERROR_INVALIDDATA;
1417 1418

    vbv_delay = get_bits(&s->gb, 16);
1419
    s->vbv_delay = vbv_delay;
1420 1421
    if (s->pict_type == AV_PICTURE_TYPE_P ||
        s->pict_type == AV_PICTURE_TYPE_B) {
1422 1423
        s->full_pel[0] = get_bits1(&s->gb);
        f_code = get_bits(&s->gb, 3);
1424
        if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1425
            return AVERROR_INVALIDDATA;
1426
        f_code += !f_code;
1427 1428 1429 1430 1431 1432
        s->mpeg_f_code[0][0] = f_code;
        s->mpeg_f_code[0][1] = f_code;
    }
    if (s->pict_type == AV_PICTURE_TYPE_B) {
        s->full_pel[1] = get_bits1(&s->gb);
        f_code = get_bits(&s->gb, 3);
1433
        if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1434
            return AVERROR_INVALIDDATA;
1435
        f_code += !f_code;
1436 1437 1438
        s->mpeg_f_code[1][0] = f_code;
        s->mpeg_f_code[1][1] = f_code;
    }
1439 1440
    s->current_picture.f->pict_type = s->pict_type;
    s->current_picture.f->key_frame = s->pict_type == AV_PICTURE_TYPE_I;
1441 1442

    if (avctx->debug & FF_DEBUG_PICT_INFO)
1443 1444
        av_log(avctx, AV_LOG_DEBUG,
               "vbv_delay %d, ref %d type:%d\n", vbv_delay, ref, s->pict_type);
1445 1446 1447 1448 1449 1450 1451 1452

    s->y_dc_scale = 8;
    s->c_dc_scale = 8;
    return 0;
}

static void mpeg_decode_sequence_extension(Mpeg1Context *s1)
{
1453
    MpegEncContext *s = &s1->mpeg_enc_ctx;
1454 1455 1456 1457 1458 1459
    int horiz_size_ext, vert_size_ext;
    int bit_rate_ext;

    skip_bits(&s->gb, 1); /* profile and level esc*/
    s->avctx->profile       = get_bits(&s->gb, 3);
    s->avctx->level         = get_bits(&s->gb, 4);
1460
    s->progressive_sequence = get_bits1(&s->gb);   /* progressive_sequence */
1461 1462 1463 1464 1465 1466 1467
    s->chroma_format        = get_bits(&s->gb, 2); /* chroma_format 1=420, 2=422, 3=444 */
    horiz_size_ext          = get_bits(&s->gb, 2);
    vert_size_ext           = get_bits(&s->gb, 2);
    s->width  |= (horiz_size_ext << 12);
    s->height |= (vert_size_ext  << 12);
    bit_rate_ext = get_bits(&s->gb, 12);  /* XXX: handle it */
    s->bit_rate += (bit_rate_ext << 18) * 400;
1468
    check_marker(&s->gb, "after bit rate extension");
1469 1470 1471
    s->avctx->rc_buffer_size += get_bits(&s->gb, 8) * 1024 * 16 << 10;

    s->low_delay = get_bits1(&s->gb);
1472
    if (s->avctx->flags & CODEC_FLAG_LOW_DELAY)
1473 1474 1475 1476 1477
        s->low_delay = 1;

    s1->frame_rate_ext.num = get_bits(&s->gb, 2) + 1;
    s1->frame_rate_ext.den = get_bits(&s->gb, 5) + 1;

1478
    ff_dlog(s->avctx, "sequence extension\n");
1479
    s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
1480 1481

    if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1482
        av_log(s->avctx, AV_LOG_DEBUG,
1483 1484
               "profile: %d, level: %d ps: %d cf:%d vbv buffer: %d, bitrate:%d\n",
               s->avctx->profile, s->avctx->level, s->progressive_sequence, s->chroma_format,
1485
               s->avctx->rc_buffer_size, s->bit_rate);
1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500
}

static void mpeg_decode_sequence_display_extension(Mpeg1Context *s1)
{
    MpegEncContext *s = &s1->mpeg_enc_ctx;
    int color_description, w, h;

    skip_bits(&s->gb, 3); /* video format */
    color_description = get_bits1(&s->gb);
    if (color_description) {
        s->avctx->color_primaries = get_bits(&s->gb, 8);
        s->avctx->color_trc       = get_bits(&s->gb, 8);
        s->avctx->colorspace      = get_bits(&s->gb, 8);
    }
    w = get_bits(&s->gb, 14);
1501
    skip_bits(&s->gb, 1); // marker
1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538
    h = get_bits(&s->gb, 14);
    // remaining 3 bits are zero padding

    s1->pan_scan.width  = 16 * w;
    s1->pan_scan.height = 16 * h;

    if (s->avctx->debug & FF_DEBUG_PICT_INFO)
        av_log(s->avctx, AV_LOG_DEBUG, "sde w:%d, h:%d\n", w, h);
}

static void mpeg_decode_picture_display_extension(Mpeg1Context *s1)
{
    MpegEncContext *s = &s1->mpeg_enc_ctx;
    int i, nofco;

    nofco = 1;
    if (s->progressive_sequence) {
        if (s->repeat_first_field) {
            nofco++;
            if (s->top_field_first)
                nofco++;
        }
    } else {
        if (s->picture_structure == PICT_FRAME) {
            nofco++;
            if (s->repeat_first_field)
                nofco++;
        }
    }
    for (i = 0; i < nofco; i++) {
        s1->pan_scan.position[i][0] = get_sbits(&s->gb, 16);
        skip_bits(&s->gb, 1); // marker
        s1->pan_scan.position[i][1] = get_sbits(&s->gb, 16);
        skip_bits(&s->gb, 1); // marker
    }

    if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1539 1540
        av_log(s->avctx, AV_LOG_DEBUG,
               "pde (%"PRId16",%"PRId16") (%"PRId16",%"PRId16") (%"PRId16",%"PRId16")\n",
1541 1542 1543 1544 1545
               s1->pan_scan.position[0][0], s1->pan_scan.position[0][1],
               s1->pan_scan.position[1][0], s1->pan_scan.position[1][1],
               s1->pan_scan.position[2][0], s1->pan_scan.position[2][1]);
}

1546 1547
static int load_matrix(MpegEncContext *s, uint16_t matrix0[64],
                       uint16_t matrix1[64], int intra)
1548 1549 1550 1551
{
    int i;

    for (i = 0; i < 64; i++) {
1552
        int j = s->idsp.idct_permutation[ff_zigzag_direct[i]];
1553 1554 1555
        int v = get_bits(&s->gb, 8);
        if (v == 0) {
            av_log(s->avctx, AV_LOG_ERROR, "matrix damaged\n");
1556
            return AVERROR_INVALIDDATA;
1557 1558
        }
        if (intra && i == 0 && v != 8) {
1559
            av_log(s->avctx, AV_LOG_DEBUG, "intra matrix specifies invalid DC quantizer %d, ignoring\n", v);
1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570
            v = 8; // needed by pink.mpg / issue1046
        }
        matrix0[j] = v;
        if (matrix1)
            matrix1[j] = v;
    }
    return 0;
}

static void mpeg_decode_quant_matrix_extension(MpegEncContext *s)
{
1571
    ff_dlog(s->avctx, "matrix extension\n");
1572

1573 1574 1575 1576 1577 1578 1579 1580
    if (get_bits1(&s->gb))
        load_matrix(s, s->chroma_intra_matrix, s->intra_matrix, 1);
    if (get_bits1(&s->gb))
        load_matrix(s, s->chroma_inter_matrix, s->inter_matrix, 0);
    if (get_bits1(&s->gb))
        load_matrix(s, s->chroma_intra_matrix, NULL, 1);
    if (get_bits1(&s->gb))
        load_matrix(s, s->chroma_inter_matrix, NULL, 0);
1581 1582 1583 1584 1585 1586
}

static void mpeg_decode_picture_coding_extension(Mpeg1Context *s1)
{
    MpegEncContext *s = &s1->mpeg_enc_ctx;

1587
    s->full_pel[0]       = s->full_pel[1] = 0;
1588 1589 1590 1591 1592
    s->mpeg_f_code[0][0] = get_bits(&s->gb, 4);
    s->mpeg_f_code[0][1] = get_bits(&s->gb, 4);
    s->mpeg_f_code[1][0] = get_bits(&s->gb, 4);
    s->mpeg_f_code[1][1] = get_bits(&s->gb, 4);
    if (!s->pict_type && s1->mpeg_enc_ctx_allocated) {
1593 1594
        av_log(s->avctx, AV_LOG_ERROR,
               "Missing picture start code, guessing missing values\n");
1595 1596 1597 1598 1599 1600 1601
        if (s->mpeg_f_code[1][0] == 15 && s->mpeg_f_code[1][1] == 15) {
            if (s->mpeg_f_code[0][0] == 15 && s->mpeg_f_code[0][1] == 15)
                s->pict_type = AV_PICTURE_TYPE_I;
            else
                s->pict_type = AV_PICTURE_TYPE_P;
        } else
            s->pict_type = AV_PICTURE_TYPE_B;
1602 1603
        s->current_picture.f->pict_type = s->pict_type;
        s->current_picture.f->key_frame = s->pict_type == AV_PICTURE_TYPE_I;
1604
    }
1605 1606 1607 1608 1609
    s->mpeg_f_code[0][0] += !s->mpeg_f_code[0][0];
    s->mpeg_f_code[0][1] += !s->mpeg_f_code[0][1];
    s->mpeg_f_code[1][0] += !s->mpeg_f_code[1][0];
    s->mpeg_f_code[1][1] += !s->mpeg_f_code[1][1];

1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622
    s->intra_dc_precision         = get_bits(&s->gb, 2);
    s->picture_structure          = get_bits(&s->gb, 2);
    s->top_field_first            = get_bits1(&s->gb);
    s->frame_pred_frame_dct       = get_bits1(&s->gb);
    s->concealment_motion_vectors = get_bits1(&s->gb);
    s->q_scale_type               = get_bits1(&s->gb);
    s->intra_vlc_format           = get_bits1(&s->gb);
    s->alternate_scan             = get_bits1(&s->gb);
    s->repeat_first_field         = get_bits1(&s->gb);
    s->chroma_420_type            = get_bits1(&s->gb);
    s->progressive_frame          = get_bits1(&s->gb);

    if (s->alternate_scan) {
1623 1624
        ff_init_scantable(s->idsp.idct_permutation, &s->inter_scantable, ff_alternate_vertical_scan);
        ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable, ff_alternate_vertical_scan);
1625
    } else {
1626 1627
        ff_init_scantable(s->idsp.idct_permutation, &s->inter_scantable, ff_zigzag_direct);
        ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable, ff_zigzag_direct);
1628 1629 1630
    }

    /* composite display not parsed */
1631 1632 1633 1634 1635 1636 1637 1638 1639
    ff_dlog(s->avctx, "intra_dc_precision=%d\n", s->intra_dc_precision);
    ff_dlog(s->avctx, "picture_structure=%d\n", s->picture_structure);
    ff_dlog(s->avctx, "top field first=%d\n", s->top_field_first);
    ff_dlog(s->avctx, "repeat first field=%d\n", s->repeat_first_field);
    ff_dlog(s->avctx, "conceal=%d\n", s->concealment_motion_vectors);
    ff_dlog(s->avctx, "intra_vlc_format=%d\n", s->intra_vlc_format);
    ff_dlog(s->avctx, "alternate_scan=%d\n", s->alternate_scan);
    ff_dlog(s->avctx, "frame_pred_frame_dct=%d\n", s->frame_pred_frame_dct);
    ff_dlog(s->avctx, "progressive_frame=%d\n", s->progressive_frame);
1640 1641 1642 1643 1644
}

static int mpeg_field_start(MpegEncContext *s, const uint8_t *buf, int buf_size)
{
    AVCodecContext *avctx = s->avctx;
1645
    Mpeg1Context *s1      = (Mpeg1Context *) s;
1646
    int ret;
1647 1648 1649 1650 1651

    /* start frame decoding */
    if (s->first_field || s->picture_structure == PICT_FRAME) {
        AVFrameSideData *pan_scan;

1652 1653
        if ((ret = ff_mpv_frame_start(s, avctx)) < 0)
            return ret;
1654 1655 1656 1657

        ff_mpeg_er_frame_start(s);

        /* first check if we must repeat the frame */
1658
        s->current_picture_ptr->f->repeat_pict = 0;
1659 1660 1661
        if (s->repeat_first_field) {
            if (s->progressive_sequence) {
                if (s->top_field_first)
1662
                    s->current_picture_ptr->f->repeat_pict = 4;
1663
                else
1664
                    s->current_picture_ptr->f->repeat_pict = 2;
1665
            } else if (s->progressive_frame) {
1666
                s->current_picture_ptr->f->repeat_pict = 1;
1667 1668 1669
            }
        }

1670
        pan_scan = av_frame_new_side_data(s->current_picture_ptr->f,
1671 1672 1673 1674 1675 1676
                                          AV_FRAME_DATA_PANSCAN,
                                          sizeof(s1->pan_scan));
        if (!pan_scan)
            return AVERROR(ENOMEM);
        memcpy(pan_scan->data, &s1->pan_scan, sizeof(s1->pan_scan));

1677 1678
        if (s1->a53_caption) {
            AVFrameSideData *sd = av_frame_new_side_data(
1679
                s->current_picture_ptr->f, AV_FRAME_DATA_A53_CC,
1680 1681 1682 1683 1684
                s1->a53_caption_size);
            if (sd)
                memcpy(sd->data, s1->a53_caption, s1->a53_caption_size);
            av_freep(&s1->a53_caption);
        }
1685 1686

        if (s1->has_stereo3d) {
1687
            AVStereo3D *stereo = av_stereo3d_create_side_data(s->current_picture_ptr->f);
1688 1689 1690 1691 1692 1693
            if (!stereo)
                return AVERROR(ENOMEM);

            *stereo = s1->stereo3d;
            s1->has_stereo3d = 0;
        }
1694 1695

        if (s1->has_afd) {
1696 1697 1698
            AVFrameSideData *sd =
                av_frame_new_side_data(s->current_picture_ptr->f,
                                       AV_FRAME_DATA_AFD, 1);
1699 1700 1701
            if (!sd)
                return AVERROR(ENOMEM);

1702
            *sd->data   = s1->afd;
1703 1704 1705
            s1->has_afd = 0;
        }

1706 1707 1708 1709 1710 1711 1712
        if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_FRAME))
            ff_thread_finish_setup(avctx);
    } else { // second field
        int i;

        if (!s->current_picture_ptr) {
            av_log(s->avctx, AV_LOG_ERROR, "first field missing\n");
1713
            return AVERROR_INVALIDDATA;
1714 1715 1716 1717 1718
        }

        if (s->avctx->hwaccel &&
            (s->avctx->slice_flags & SLICE_FLAG_ALLOW_FIELD)) {
            if (s->avctx->hwaccel->end_frame(s->avctx) < 0)
1719 1720
                av_log(avctx, AV_LOG_ERROR,
                       "hardware accelerator failed to decode first field\n");
1721 1722 1723
        }

        for (i = 0; i < 4; i++) {
1724
            s->current_picture.f->data[i] = s->current_picture_ptr->f->data[i];
1725
            if (s->picture_structure == PICT_BOTTOM_FIELD)
1726 1727
                s->current_picture.f->data[i] +=
                    s->current_picture_ptr->f->linesize[i];
1728 1729 1730 1731
        }
    }

    if (avctx->hwaccel) {
1732 1733
        if ((ret = avctx->hwaccel->start_frame(avctx, buf, buf_size)) < 0)
            return ret;
1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751
    }

    return 0;
}

#define DECODE_SLICE_ERROR -1
#define DECODE_SLICE_OK     0

/**
 * Decode a slice.
 * MpegEncContext.mb_y must be set to the MB row from the startcode.
 * @return DECODE_SLICE_ERROR if the slice is damaged,
 *         DECODE_SLICE_OK if this slice is OK
 */
static int mpeg_decode_slice(MpegEncContext *s, int mb_y,
                             const uint8_t **buf, int buf_size)
{
    AVCodecContext *avctx = s->avctx;
1752
    const int lowres      = s->avctx->lowres;
1753
    const int field_pic   = s->picture_structure != PICT_FRAME;
1754
    int ret;
1755 1756 1757 1758

    s->resync_mb_x =
    s->resync_mb_y = -1;

1759
    av_assert0(mb_y < s->mb_height);
1760 1761

    init_get_bits(&s->gb, *buf, buf_size * 8);
1762
    if (s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
1763
        skip_bits(&s->gb, 3);
1764 1765 1766 1767 1768 1769 1770 1771

    ff_mpeg1_clean_buffers(s);
    s->interlaced_dct = 0;

    s->qscale = get_qscale(s);

    if (s->qscale == 0) {
        av_log(s->avctx, AV_LOG_ERROR, "qscale == 0\n");
1772
        return AVERROR_INVALIDDATA;
1773 1774 1775
    }

    /* extra slice info */
1776 1777
    if (skip_1stop_8data_bits(&s->gb) < 0)
        return AVERROR_INVALIDDATA;
1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788

    s->mb_x = 0;

    if (mb_y == 0 && s->codec_tag == AV_RL32("SLIF")) {
        skip_bits1(&s->gb);
    } else {
        while (get_bits_left(&s->gb) > 0) {
            int code = get_vlc2(&s->gb, ff_mbincr_vlc.table,
                                MBINCR_VLC_BITS, 2);
            if (code < 0) {
                av_log(s->avctx, AV_LOG_ERROR, "first mb_incr damaged\n");
1789
                return AVERROR_INVALIDDATA;
1790 1791
            }
            if (code >= 33) {
1792
                if (code == 33)
1793 1794 1795 1796 1797 1798 1799 1800 1801
                    s->mb_x += 33;
                /* otherwise, stuffing, nothing to do */
            } else {
                s->mb_x += code;
                break;
            }
        }
    }

1802
    if (s->mb_x >= (unsigned) s->mb_width) {
1803
        av_log(s->avctx, AV_LOG_ERROR, "initial skip overflow\n");
1804
        return AVERROR_INVALIDDATA;
1805 1806
    }

1807
    if (avctx->hwaccel && avctx->hwaccel->decode_slice) {
1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826
        const uint8_t *buf_end, *buf_start = *buf - 4; /* include start_code */
        int start_code = -1;
        buf_end = avpriv_find_start_code(buf_start + 2, *buf + buf_size, &start_code);
        if (buf_end < *buf + buf_size)
            buf_end -= 4;
        s->mb_y = mb_y;
        if (avctx->hwaccel->decode_slice(avctx, buf_start, buf_end - buf_start) < 0)
            return DECODE_SLICE_ERROR;
        *buf = buf_end;
        return DECODE_SLICE_OK;
    }

    s->resync_mb_x = s->mb_x;
    s->resync_mb_y = s->mb_y = mb_y;
    s->mb_skip_run = 0;
    ff_init_block_index(s);

    if (s->mb_y == 0 && s->mb_x == 0 && (s->first_field || s->picture_structure == PICT_FRAME)) {
        if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842
            av_log(s->avctx, AV_LOG_DEBUG,
                   "qp:%d fc:%2d%2d%2d%2d %s %s %s %s %s dc:%d pstruct:%d fdct:%d cmv:%d qtype:%d ivlc:%d rff:%d %s\n",
                   s->qscale,
                   s->mpeg_f_code[0][0], s->mpeg_f_code[0][1],
                   s->mpeg_f_code[1][0], s->mpeg_f_code[1][1],
                   s->pict_type  == AV_PICTURE_TYPE_I ? "I" :
                   (s->pict_type == AV_PICTURE_TYPE_P ? "P" :
                   (s->pict_type == AV_PICTURE_TYPE_B ? "B" : "S")),
                   s->progressive_sequence ? "ps"  : "",
                   s->progressive_frame    ? "pf"  : "",
                   s->alternate_scan       ? "alt" : "",
                   s->top_field_first      ? "top" : "",
                   s->intra_dc_precision, s->picture_structure,
                   s->frame_pred_frame_dct, s->concealment_motion_vectors,
                   s->q_scale_type, s->intra_vlc_format,
                   s->repeat_first_field, s->chroma_420_type ? "420" : "");
1843 1844 1845 1846 1847
        }
    }

    for (;;) {
        // If 1, we memcpy blocks in xvmcvideo.
1848
        if ((CONFIG_MPEG1_XVMC_HWACCEL || CONFIG_MPEG2_XVMC_HWACCEL) && s->pack_pblocks)
1849 1850
            ff_xvmc_init_block(s); // set s->block

1851 1852
        if ((ret = mpeg_decode_mb(s, s->block)) < 0)
            return ret;
1853

1854 1855
        // Note motion_val is normally NULL unless we want to extract the MVs.
        if (s->current_picture.motion_val[0] && !s->encoding) {
1856 1857 1858 1859 1860 1861 1862
            const int wrap = s->b8_stride;
            int xy         = s->mb_x * 2 + s->mb_y * 2 * wrap;
            int b8_xy      = 4 * (s->mb_x + s->mb_y * s->mb_stride);
            int motion_x, motion_y, dir, i;

            for (i = 0; i < 2; i++) {
                for (dir = 0; dir < 2; dir++) {
1863 1864
                    if (s->mb_intra ||
                        (dir == 1 && s->pict_type != AV_PICTURE_TYPE_B)) {
1865
                        motion_x = motion_y = 0;
1866 1867
                    } else if (s->mv_type == MV_TYPE_16X16 ||
                               (s->mv_type == MV_TYPE_FIELD && field_pic)) {
1868 1869
                        motion_x = s->mv[dir][0][0];
                        motion_y = s->mv[dir][0][1];
1870
                    } else { /* if ((s->mv_type == MV_TYPE_FIELD) || (s->mv_type == MV_TYPE_16X8)) */
1871 1872 1873 1874
                        motion_x = s->mv[dir][i][0];
                        motion_y = s->mv[dir][i][1];
                    }

1875 1876
                    s->current_picture.motion_val[dir][xy][0]     = motion_x;
                    s->current_picture.motion_val[dir][xy][1]     = motion_y;
1877 1878
                    s->current_picture.motion_val[dir][xy + 1][0] = motion_x;
                    s->current_picture.motion_val[dir][xy + 1][1] = motion_y;
1879
                    s->current_picture.ref_index [dir][b8_xy]     =
1880
                    s->current_picture.ref_index [dir][b8_xy + 1] = s->field_select[dir][i];
1881 1882
                    av_assert2(s->field_select[dir][i] == 0 ||
                               s->field_select[dir][i] == 1);
1883
                }
1884 1885
                xy    += wrap;
                b8_xy += 2;
1886 1887 1888
            }
        }

1889 1890 1891
        s->dest[0] += 16 >> lowres;
        s->dest[1] +=(16 >> lowres) >> s->chroma_x_shift;
        s->dest[2] +=(16 >> lowres) >> s->chroma_x_shift;
1892

1893
        ff_mpv_decode_mb(s, s->block);
1894 1895

        if (++s->mb_x >= s->mb_width) {
1896
            const int mb_size = 16 >> s->avctx->lowres;
1897

1898
            ff_mpeg_draw_horiz_band(s, mb_size * (s->mb_y >> field_pic), mb_size);
1899
            ff_mpv_report_decode_progress(s);
1900

1901
            s->mb_x  = 0;
1902 1903 1904 1905
            s->mb_y += 1 << field_pic;

            if (s->mb_y >= s->mb_height) {
                int left   = get_bits_left(&s->gb);
1906 1907 1908 1909 1910 1911 1912
                int is_d10 = s->chroma_format == 2 &&
                             s->pict_type == AV_PICTURE_TYPE_I &&
                             avctx->profile == 0 && avctx->level == 5 &&
                             s->intra_dc_precision == 2 &&
                             s->q_scale_type == 1 && s->alternate_scan == 0 &&
                             s->progressive_frame == 0
                             /* vbv_delay == 0xBBB || 0xE10 */;
1913

1914 1915 1916 1917 1918 1919 1920 1921 1922
                if (left >= 32 && !is_d10) {
                    GetBitContext gb = s->gb;
                    align_get_bits(&gb);
                    if (show_bits(&gb, 24) == 0x060E2B) {
                        av_log(avctx, AV_LOG_DEBUG, "Invalid MXF data found in video stream\n");
                        is_d10 = 1;
                    }
                }

1923 1924
                if (left < 0 ||
                    (left && show_bits(&s->gb, FFMIN(left, 23)) && !is_d10) ||
1925
                    ((avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_AGGRESSIVE)) && left > 8)) {
1926 1927
                    av_log(avctx, AV_LOG_ERROR, "end mismatch left=%d %0X\n",
                           left, show_bits(&s->gb, FFMIN(left, 23)));
1928
                    return AVERROR_INVALIDDATA;
1929 1930 1931
                } else
                    goto eos;
            }
1932
            // There are some files out there which are missing the last slice
1933
            // in cases where the slice is completely outside the visible
1934 1935
            // area, we detect this here instead of running into the end expecting
            // more data
1936 1937 1938 1939 1940 1941 1942 1943
            if (s->mb_y >= ((s->height + 15) >> 4) &&
                s->progressive_frame &&
                !s->progressive_sequence &&
                get_bits_left(&s->gb) <= 8 &&
                get_bits_left(&s->gb) >= 0 &&
                s->mb_skip_run == -1 &&
                show_bits(&s->gb, 8) == 0)
                goto eos;
1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956

            ff_init_block_index(s);
        }

        /* skip mb handling */
        if (s->mb_skip_run == -1) {
            /* read increment again */
            s->mb_skip_run = 0;
            for (;;) {
                int code = get_vlc2(&s->gb, ff_mbincr_vlc.table,
                                    MBINCR_VLC_BITS, 2);
                if (code < 0) {
                    av_log(s->avctx, AV_LOG_ERROR, "mb incr damaged\n");
1957
                    return AVERROR_INVALIDDATA;
1958 1959 1960 1961 1962 1963 1964
                }
                if (code >= 33) {
                    if (code == 33) {
                        s->mb_skip_run += 33;
                    } else if (code == 35) {
                        if (s->mb_skip_run != 0 || show_bits(&s->gb, 15) != 0) {
                            av_log(s->avctx, AV_LOG_ERROR, "slice mismatch\n");
1965
                            return AVERROR_INVALIDDATA;
1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977
                        }
                        goto eos; /* end of slice */
                    }
                    /* otherwise, stuffing, nothing to do */
                } else {
                    s->mb_skip_run += code;
                    break;
                }
            }
            if (s->mb_skip_run) {
                int i;
                if (s->pict_type == AV_PICTURE_TYPE_I) {
1978 1979
                    av_log(s->avctx, AV_LOG_ERROR,
                           "skipped MB in I frame at %d %d\n", s->mb_x, s->mb_y);
1980
                    return AVERROR_INVALIDDATA;
1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008
                }

                /* skip mb */
                s->mb_intra = 0;
                for (i = 0; i < 12; i++)
                    s->block_last_index[i] = -1;
                if (s->picture_structure == PICT_FRAME)
                    s->mv_type = MV_TYPE_16X16;
                else
                    s->mv_type = MV_TYPE_FIELD;
                if (s->pict_type == AV_PICTURE_TYPE_P) {
                    /* if P type, zero motion vector is implied */
                    s->mv_dir             = MV_DIR_FORWARD;
                    s->mv[0][0][0]        = s->mv[0][0][1]      = 0;
                    s->last_mv[0][0][0]   = s->last_mv[0][0][1] = 0;
                    s->last_mv[0][1][0]   = s->last_mv[0][1][1] = 0;
                    s->field_select[0][0] = (s->picture_structure - 1) & 1;
                } else {
                    /* if B type, reuse previous vectors and directions */
                    s->mv[0][0][0] = s->last_mv[0][0][0];
                    s->mv[0][0][1] = s->last_mv[0][0][1];
                    s->mv[1][0][0] = s->last_mv[1][0][0];
                    s->mv[1][0][1] = s->last_mv[1][0][1];
                }
            }
        }
    }
eos: // end of slice
2009 2010
    if (get_bits_left(&s->gb) < 0) {
        av_log(s, AV_LOG_ERROR, "overread %d\n", -get_bits_left(&s->gb));
2011
        return AVERROR_INVALIDDATA;
2012
    }
2013
    *buf += (get_bits_count(&s->gb) - 1) / 8;
2014
    ff_dlog(s, "Slice start:%d %d  end:%d %d\n", s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y);
2015 2016 2017 2018 2019
    return 0;
}

static int slice_decode_thread(AVCodecContext *c, void *arg)
{
2020
    MpegEncContext *s   = *(void **) arg;
2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032
    const uint8_t *buf  = s->gb.buffer;
    int mb_y            = s->start_mb_y;
    const int field_pic = s->picture_structure != PICT_FRAME;

    s->er.error_count = (3 * (s->end_mb_y - s->start_mb_y) * s->mb_width) >> field_pic;

    for (;;) {
        uint32_t start_code;
        int ret;

        ret = mpeg_decode_slice(s, mb_y, &buf, s->gb.buffer_end - buf);
        emms_c();
2033
        ff_dlog(c, "ret:%d resync:%d/%d mb:%d/%d ts:%d/%d ec:%d\n",
2034 2035 2036 2037 2038 2039
                ret, s->resync_mb_x, s->resync_mb_y, s->mb_x, s->mb_y,
                s->start_mb_y, s->end_mb_y, s->er.error_count);
        if (ret < 0) {
            if (c->err_recognition & AV_EF_EXPLODE)
                return ret;
            if (s->resync_mb_x >= 0 && s->resync_mb_y >= 0)
2040 2041 2042
                ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y,
                                s->mb_x, s->mb_y,
                                ER_AC_ERROR | ER_DC_ERROR | ER_MV_ERROR);
2043
        } else {
2044 2045 2046
            ff_er_add_slice(&s->er, s->resync_mb_x, s->resync_mb_y,
                            s->mb_x - 1, s->mb_y,
                            ER_AC_END | ER_DC_END | ER_MV_END);
2047 2048 2049 2050 2051 2052
        }

        if (s->mb_y == s->end_mb_y)
            return 0;

        start_code = -1;
2053
        buf        = avpriv_find_start_code(buf, s->gb.buffer_end, &start_code);
2054 2055
        mb_y       = start_code - SLICE_MIN_START_CODE;
        if (s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
2056 2057
            mb_y += (*buf&0xE0)<<2;
        mb_y <<= field_pic;
2058 2059 2060
        if (s->picture_structure == PICT_BOTTOM_FIELD)
            mb_y++;
        if (mb_y < 0 || mb_y >= s->end_mb_y)
2061
            return AVERROR_INVALIDDATA;
2062 2063 2064 2065 2066 2067 2068 2069 2070
    }
}

/**
 * Handle slice ends.
 * @return 1 if it seems to be the last slice
 */
static int slice_end(AVCodecContext *avctx, AVFrame *pict)
{
2071
    Mpeg1Context *s1  = avctx->priv_data;
2072 2073 2074 2075 2076 2077 2078
    MpegEncContext *s = &s1->mpeg_enc_ctx;

    if (!s1->mpeg_enc_ctx_allocated || !s->current_picture_ptr)
        return 0;

    if (s->avctx->hwaccel) {
        if (s->avctx->hwaccel->end_frame(s->avctx) < 0)
2079 2080
            av_log(avctx, AV_LOG_ERROR,
                   "hardware accelerator failed to decode picture\n");
2081 2082 2083
    }

    /* end of slice reached */
2084
    if (/* s->mb_y << field_pic == s->mb_height && */ !s->first_field && !s1->first_slice) {
2085 2086 2087 2088
        /* end of image */

        ff_er_frame_end(&s->er);

2089
        ff_mpv_frame_end(s);
2090 2091

        if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {
2092
            int ret = av_frame_ref(pict, s->current_picture_ptr->f);
2093 2094
            if (ret < 0)
                return ret;
2095 2096
            ff_print_debug_info(s, s->current_picture_ptr, pict);
            ff_mpv_export_qp_table(s, pict, s->current_picture_ptr, FF_QSCALE_TYPE_MPEG2);
2097 2098 2099 2100 2101
        } else {
            if (avctx->active_thread_type & FF_THREAD_FRAME)
                s->picture_number++;
            /* latency of 1 frame for I- and P-frames */
            /* XXX: use another variable than picture_number */
2102
            if (s->last_picture_ptr) {
2103
                int ret = av_frame_ref(pict, s->last_picture_ptr->f);
2104 2105
                if (ret < 0)
                    return ret;
2106 2107
                ff_print_debug_info(s, s->last_picture_ptr, pict);
                ff_mpv_export_qp_table(s, pict, s->last_picture_ptr, FF_QSCALE_TYPE_MPEG2);
2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119
            }
        }

        return 1;
    } else {
        return 0;
    }
}

static int mpeg1_decode_sequence(AVCodecContext *avctx,
                                 const uint8_t *buf, int buf_size)
{
2120
    Mpeg1Context *s1  = avctx->priv_data;
2121 2122 2123 2124
    MpegEncContext *s = &s1->mpeg_enc_ctx;
    int width, height;
    int i, v, j;

2125
    init_get_bits(&s->gb, buf, buf_size * 8);
2126 2127 2128 2129

    width  = get_bits(&s->gb, 12);
    height = get_bits(&s->gb, 12);
    if (width == 0 || height == 0) {
2130 2131
        av_log(avctx, AV_LOG_WARNING,
               "Invalid horizontal or vertical size value.\n");
2132
        if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
2133 2134 2135 2136 2137
            return AVERROR_INVALIDDATA;
    }
    s->aspect_ratio_info = get_bits(&s->gb, 4);
    if (s->aspect_ratio_info == 0) {
        av_log(avctx, AV_LOG_ERROR, "aspect ratio has forbidden 0 value\n");
2138
        if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
2139
            return AVERROR_INVALIDDATA;
2140 2141
    }
    s->frame_rate_index = get_bits(&s->gb, 4);
2142
    if (s->frame_rate_index == 0 || s->frame_rate_index > 13) {
2143 2144
        av_log(avctx, AV_LOG_WARNING,
               "frame_rate_index %d is invalid\n", s->frame_rate_index);
2145
        s->frame_rate_index = 1;
2146
    }
2147
    s->bit_rate = get_bits(&s->gb, 18) * 400;
2148
    if (check_marker(&s->gb, "in sequence header") == 0) {
2149
        return AVERROR_INVALIDDATA;
2150
    }
2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161
    s->width  = width;
    s->height = height;

    s->avctx->rc_buffer_size = get_bits(&s->gb, 10) * 1024 * 16;
    skip_bits(&s->gb, 1);

    /* get matrix */
    if (get_bits1(&s->gb)) {
        load_matrix(s, s->chroma_intra_matrix, s->intra_matrix, 1);
    } else {
        for (i = 0; i < 64; i++) {
2162
            j = s->idsp.idct_permutation[i];
2163 2164 2165 2166 2167 2168 2169 2170 2171
            v = ff_mpeg1_default_intra_matrix[i];
            s->intra_matrix[j]        = v;
            s->chroma_intra_matrix[j] = v;
        }
    }
    if (get_bits1(&s->gb)) {
        load_matrix(s, s->chroma_inter_matrix, s->inter_matrix, 0);
    } else {
        for (i = 0; i < 64; i++) {
2172
            int j = s->idsp.idct_permutation[i];
2173 2174 2175 2176 2177 2178 2179 2180
            v = ff_mpeg1_default_non_intra_matrix[i];
            s->inter_matrix[j]        = v;
            s->chroma_inter_matrix[j] = v;
        }
    }

    if (show_bits(&s->gb, 23) != 0) {
        av_log(s->avctx, AV_LOG_ERROR, "sequence header damaged\n");
2181
        return AVERROR_INVALIDDATA;
2182 2183
    }

2184
    /* We set MPEG-2 parameters so that it emulates MPEG-1. */
2185 2186 2187
    s->progressive_sequence = 1;
    s->progressive_frame    = 1;
    s->picture_structure    = PICT_FRAME;
2188
    s->first_field          = 0;
2189 2190
    s->frame_pred_frame_dct = 1;
    s->chroma_format        = 1;
2191 2192
    s->codec_id             =
    s->avctx->codec_id      = AV_CODEC_ID_MPEG1VIDEO;
2193 2194
    s->out_format           = FMT_MPEG1;
    s->swap_uv              = 0; // AFAIK VCR2 does not have SEQ_HEADER
2195
    if (s->avctx->flags & CODEC_FLAG_LOW_DELAY)
2196 2197 2198
        s->low_delay = 1;

    if (s->avctx->debug & FF_DEBUG_PICT_INFO)
2199 2200
        av_log(s->avctx, AV_LOG_DEBUG, "vbv buffer: %d, bitrate:%d, aspect_ratio_info: %d \n",
               s->avctx->rc_buffer_size, s->bit_rate, s->aspect_ratio_info);
2201 2202 2203 2204 2205 2206

    return 0;
}

static int vcr2_init_sequence(AVCodecContext *avctx)
{
2207
    Mpeg1Context *s1  = avctx->priv_data;
2208
    MpegEncContext *s = &s1->mpeg_enc_ctx;
2209
    int i, v, ret;
2210 2211 2212 2213

    /* start new MPEG-1 context decoding */
    s->out_format = FMT_MPEG1;
    if (s1->mpeg_enc_ctx_allocated) {
2214
        ff_mpv_common_end(s);
2215
        s1->mpeg_enc_ctx_allocated = 0;
2216
    }
2217 2218
    s->width            = avctx->coded_width;
    s->height           = avctx->coded_height;
2219
    avctx->has_b_frames = 0; // true?
2220
    s->low_delay        = 1;
2221 2222

    avctx->pix_fmt = mpeg_get_pixelformat(avctx);
2223
    setup_hwaccel_for_pixfmt(avctx);
2224

2225
    ff_mpv_idct_init(s);
2226 2227
    if ((ret = ff_mpv_common_init(s)) < 0)
        return ret;
2228 2229 2230
    s1->mpeg_enc_ctx_allocated = 1;

    for (i = 0; i < 64; i++) {
2231
        int j = s->idsp.idct_permutation[i];
2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243
        v = ff_mpeg1_default_intra_matrix[i];
        s->intra_matrix[j]        = v;
        s->chroma_intra_matrix[j] = v;

        v = ff_mpeg1_default_non_intra_matrix[i];
        s->inter_matrix[j]        = v;
        s->chroma_inter_matrix[j] = v;
    }

    s->progressive_sequence  = 1;
    s->progressive_frame     = 1;
    s->picture_structure     = PICT_FRAME;
2244
    s->first_field           = 0;
2245 2246
    s->frame_pred_frame_dct  = 1;
    s->chroma_format         = 1;
2247 2248 2249 2250 2251 2252
    if (s->codec_tag == AV_RL32("BW10")) {
        s->codec_id              = s->avctx->codec_id = AV_CODEC_ID_MPEG1VIDEO;
    } else {
        s->swap_uv = 1; // in case of xvmc we need to swap uv for each MB
        s->codec_id              = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
    }
2253 2254 2255 2256 2257 2258
    s1->save_width           = s->width;
    s1->save_height          = s->height;
    s1->save_progressive_seq = s->progressive_sequence;
    return 0;
}

2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271
static int mpeg_decode_a53_cc(AVCodecContext *avctx,
                              const uint8_t *p, int buf_size)
{
    Mpeg1Context *s1 = avctx->priv_data;

    if (buf_size >= 6 &&
        p[0] == 'G' && p[1] == 'A' && p[2] == '9' && p[3] == '4' &&
        p[4] == 3 && (p[5] & 0x40)) {
        /* extract A53 Part 4 CC data */
        int cc_count = p[5] & 0x1f;
        if (cc_count > 0 && buf_size >= 7 + cc_count * 3) {
            av_freep(&s1->a53_caption);
            s1->a53_caption_size = cc_count * 3;
2272 2273
            s1->a53_caption      = av_malloc(s1->a53_caption_size);
            if (s1->a53_caption)
2274 2275 2276 2277
                memcpy(s1->a53_caption, p + 7, s1->a53_caption_size);
        }
        return 1;
    } else if (buf_size >= 11 &&
2278
               p[0] == 'C' && p[1] == 'C' && p[2] == 0x01 && p[3] == 0xf8) {
2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289
        /* extract DVD CC data */
        int cc_count = 0;
        int i;
        // There is a caption count field in the data, but it is often
        // incorect.  So count the number of captions present.
        for (i = 5; i + 6 <= buf_size && ((p[i] & 0xfe) == 0xfe); i += 6)
            cc_count++;
        // Transform the DVD format into A53 Part 4 format
        if (cc_count > 0) {
            av_freep(&s1->a53_caption);
            s1->a53_caption_size = cc_count * 6;
2290
            s1->a53_caption      = av_malloc(s1->a53_caption_size);
2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311
            if (s1->a53_caption) {
                uint8_t field1 = !!(p[4] & 0x80);
                uint8_t *cap = s1->a53_caption;
                p += 5;
                for (i = 0; i < cc_count; i++) {
                    cap[0] = (p[0] == 0xff && field1) ? 0xfc : 0xfd;
                    cap[1] = p[1];
                    cap[2] = p[2];
                    cap[3] = (p[3] == 0xff && !field1) ? 0xfc : 0xfd;
                    cap[4] = p[4];
                    cap[5] = p[5];
                    cap += 6;
                    p += 6;
                }
            }
        }
        return 1;
    }
    return 0;
}

2312 2313 2314
static void mpeg_decode_user_data(AVCodecContext *avctx,
                                  const uint8_t *p, int buf_size)
{
2315
    Mpeg1Context *s = avctx->priv_data;
2316
    const uint8_t *buf_end = p + buf_size;
2317
    Mpeg1Context *s1 = avctx->priv_data;
2318

2319 2320 2321 2322 2323 2324 2325 2326
#if 0
    int i;
    for(i=0; !(!p[i-2] && !p[i-1] && p[i]==1) && i<buf_size; i++){
        av_log(avctx, AV_LOG_ERROR, "%c", p[i]);
    }
    av_log(avctx, AV_LOG_ERROR, "\n");
#endif

2327
    if (buf_size > 29){
2328 2329
        int i;
        for(i=0; i<20; i++)
2330
            if (!memcmp(p+i, "\0TMPGEXS\0", 9)){
2331 2332 2333
                s->tmpgexs= 1;
            }
    }
2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345
    /* we parse the DTG active format information */
    if (buf_end - p >= 5 &&
        p[0] == 'D' && p[1] == 'T' && p[2] == 'G' && p[3] == '1') {
        int flags = p[4];
        p += 5;
        if (flags & 0x80) {
            /* skip event id */
            p += 2;
        }
        if (flags & 0x40) {
            if (buf_end - p < 1)
                return;
2346
#if FF_API_AFD
2347
FF_DISABLE_DEPRECATION_WARNINGS
2348
            avctx->dtg_active_format = p[0] & 0x0f;
2349
FF_ENABLE_DEPRECATION_WARNINGS
2350
#endif /* FF_API_AFD */
2351
            s1->has_afd = 1;
2352
            s1->afd     = p[0] & 0x0f;
2353
        }
2354 2355 2356 2357 2358 2359 2360 2361 2362 2363
    } else if (buf_end - p >= 6 &&
               p[0] == 'J' && p[1] == 'P' && p[2] == '3' && p[3] == 'D' &&
               p[4] == 0x03) { // S3D_video_format_length
        // the 0x7F mask ignores the reserved_bit value
        const uint8_t S3D_video_format_type = p[5] & 0x7F;

        if (S3D_video_format_type == 0x03 ||
            S3D_video_format_type == 0x04 ||
            S3D_video_format_type == 0x08 ||
            S3D_video_format_type == 0x23) {
2364

2365
            s1->has_stereo3d = 1;
2366 2367 2368

            switch (S3D_video_format_type) {
            case 0x03:
2369
                s1->stereo3d.type = AV_STEREO3D_SIDEBYSIDE;
2370 2371
                break;
            case 0x04:
2372
                s1->stereo3d.type = AV_STEREO3D_TOPBOTTOM;
2373 2374
                break;
            case 0x08:
2375
                s1->stereo3d.type = AV_STEREO3D_2D;
2376 2377
                break;
            case 0x23:
2378
                s1->stereo3d.type = AV_STEREO3D_SIDEBYSIDE_QUINCUNX;
2379 2380 2381
                break;
            }
        }
2382 2383
    } else if (mpeg_decode_a53_cc(avctx, p, buf_size)) {
        return;
2384 2385 2386 2387 2388 2389 2390 2391 2392
    }
}

static void mpeg_decode_gop(AVCodecContext *avctx,
                            const uint8_t *buf, int buf_size)
{
    Mpeg1Context *s1  = avctx->priv_data;
    MpegEncContext *s = &s1->mpeg_enc_ctx;
    int broken_link;
2393
    int64_t tc;
2394

2395
    init_get_bits(&s->gb, buf, buf_size * 8);
2396

2397
    tc = avctx->timecode_frame_start = get_bits(&s->gb, 25);
2398

2399
    s->closed_gop = get_bits1(&s->gb);
2400 2401 2402
    /* broken_link indicate that after editing the
     * reference frames of the first B-Frames after GOP I-Frame
     * are missing (open gop) */
2403 2404
    broken_link = get_bits1(&s->gb);

2405 2406 2407 2408 2409 2410 2411
    if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
        char tcbuf[AV_TIMECODE_STR_SIZE];
        av_timecode_make_mpeg_tc_string(tcbuf, tc);
        av_log(s->avctx, AV_LOG_DEBUG,
               "GOP (%s) closed_gop=%d broken_link=%d\n",
               tcbuf, s->closed_gop, broken_link);
    }
2412 2413
}

2414 2415
static int decode_chunks(AVCodecContext *avctx, AVFrame *picture,
                         int *got_output, const uint8_t *buf, int buf_size)
2416 2417 2418 2419 2420 2421
{
    Mpeg1Context *s = avctx->priv_data;
    MpegEncContext *s2 = &s->mpeg_enc_ctx;
    const uint8_t *buf_ptr = buf;
    const uint8_t *buf_end = buf + buf_size;
    int ret, input_size;
2422
    int last_code = 0, skip_frame = 0;
2423
    int picture_start_code_seen = 0;
2424 2425 2426 2427 2428 2429

    for (;;) {
        /* find next start code */
        uint32_t start_code = -1;
        buf_ptr = avpriv_find_start_code(buf_ptr, buf_end, &start_code);
        if (start_code > 0x1ff) {
2430
            if (!skip_frame) {
2431 2432
                if (HAVE_THREADS &&
                    (avctx->active_thread_type & FF_THREAD_SLICE) &&
2433
                    !avctx->hwaccel) {
2434
                    int i;
2435
                    av_assert0(avctx->thread_count > 1);
2436

2437 2438 2439
                    avctx->execute(avctx, slice_decode_thread,
                                   &s2->thread_context[0], NULL,
                                   s->slice_count, sizeof(void *));
2440 2441 2442 2443
                    for (i = 0; i < s->slice_count; i++)
                        s2->er.error_count += s2->thread_context[i]->er.error_count;
                }

2444 2445 2446 2447
                if ((CONFIG_MPEG_VDPAU_DECODER || CONFIG_MPEG1_VDPAU_DECODER)
                    && uses_vdpau(avctx))
                    ff_vdpau_mpeg_picture_complete(s2, buf, buf_size, s->slice_count);

2448 2449 2450 2451
                ret = slice_end(avctx, picture);
                if (ret < 0)
                    return ret;
                else if (ret) {
2452 2453
                    // FIXME: merge with the stuff in mpeg_decode_slice
                    if (s2->last_picture_ptr || s2->low_delay)
2454 2455 2456 2457
                        *got_output = 1;
                }
            }
            s2->pict_type = 0;
2458 2459 2460 2461

            if (avctx->err_recognition & AV_EF_EXPLODE && s2->er.error_count)
                return AVERROR_INVALIDDATA;

2462 2463 2464 2465 2466
            return FFMAX(0, buf_ptr - buf - s2->parse_context.last_index);
        }

        input_size = buf_end - buf_ptr;

2467
        if (avctx->debug & FF_DEBUG_STARTCODE)
2468
            av_log(avctx, AV_LOG_DEBUG, "%3"PRIX32" at %"PTRDIFF_SPECIFIER" left %d\n",
2469
                   start_code, buf_ptr - buf, input_size);
2470 2471 2472 2473 2474 2475

        /* prepare data for next start code */
        switch (start_code) {
        case SEQ_START_CODE:
            if (last_code == 0) {
                mpeg1_decode_sequence(avctx, buf_ptr, input_size);
2476 2477
                if (buf != avctx->extradata)
                    s->sync = 1;
2478
            } else {
2479 2480
                av_log(avctx, AV_LOG_ERROR,
                       "ignoring SEQ_START_CODE after %X\n", last_code);
2481 2482 2483 2484 2485 2486
                if (avctx->err_recognition & AV_EF_EXPLODE)
                    return AVERROR_INVALIDDATA;
            }
            break;

        case PICTURE_START_CODE:
2487 2488 2489 2490 2491 2492 2493 2494
            if (picture_start_code_seen && s2->picture_structure == PICT_FRAME) {
               /* If it's a frame picture, there can't be more than one picture header.
                  Yet, it does happen and we need to handle it. */
               av_log(avctx, AV_LOG_WARNING, "ignoring extra picture following a frame-picture\n");
               break;
            }
            picture_start_code_seen = 1;

2495 2496 2497 2498 2499 2500
            if (s2->width <= 0 || s2->height <= 0) {
                av_log(avctx, AV_LOG_ERROR, "Invalid frame dimensions %dx%d.\n",
                       s2->width, s2->height);
                return AVERROR_INVALIDDATA;
            }

2501
            if (s->tmpgexs){
2502 2503 2504
                s2->intra_dc_precision= 3;
                s2->intra_matrix[0]= 1;
            }
2505 2506
            if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE) &&
                !avctx->hwaccel && s->slice_count) {
2507 2508 2509 2510
                int i;

                avctx->execute(avctx, slice_decode_thread,
                               s2->thread_context, NULL,
2511
                               s->slice_count, sizeof(void *));
2512 2513 2514 2515 2516 2517 2518
                for (i = 0; i < s->slice_count; i++)
                    s2->er.error_count += s2->thread_context[i]->er.error_count;
                s->slice_count = 0;
            }
            if (last_code == 0 || last_code == SLICE_MIN_START_CODE) {
                ret = mpeg_decode_postinit(avctx);
                if (ret < 0) {
2519 2520
                    av_log(avctx, AV_LOG_ERROR,
                           "mpeg_decode_postinit() failure\n");
2521 2522 2523
                    return ret;
                }

2524
                /* We have a complete image: we try to decompress it. */
2525 2526
                if (mpeg1_decode_picture(avctx, buf_ptr, input_size) < 0)
                    s2->pict_type = 0;
2527
                s->first_slice = 1;
2528
                last_code      = PICTURE_START_CODE;
2529
            } else {
2530 2531
                av_log(avctx, AV_LOG_ERROR,
                       "ignoring pic after %X\n", last_code);
2532 2533 2534 2535 2536
                if (avctx->err_recognition & AV_EF_EXPLODE)
                    return AVERROR_INVALIDDATA;
            }
            break;
        case EXT_START_CODE:
2537
            init_get_bits(&s2->gb, buf_ptr, input_size * 8);
2538 2539 2540 2541

            switch (get_bits(&s2->gb, 4)) {
            case 0x1:
                if (last_code == 0) {
2542
                    mpeg_decode_sequence_extension(s);
2543
                } else {
2544 2545
                    av_log(avctx, AV_LOG_ERROR,
                           "ignoring seq ext after %X\n", last_code);
2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562
                    if (avctx->err_recognition & AV_EF_EXPLODE)
                        return AVERROR_INVALIDDATA;
                }
                break;
            case 0x2:
                mpeg_decode_sequence_display_extension(s);
                break;
            case 0x3:
                mpeg_decode_quant_matrix_extension(s2);
                break;
            case 0x7:
                mpeg_decode_picture_display_extension(s);
                break;
            case 0x8:
                if (last_code == PICTURE_START_CODE) {
                    mpeg_decode_picture_coding_extension(s);
                } else {
2563 2564
                    av_log(avctx, AV_LOG_ERROR,
                           "ignoring pic cod ext after %X\n", last_code);
2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575
                    if (avctx->err_recognition & AV_EF_EXPLODE)
                        return AVERROR_INVALIDDATA;
                }
                break;
            }
            break;
        case USER_START_CODE:
            mpeg_decode_user_data(avctx, buf_ptr, input_size);
            break;
        case GOP_START_CODE:
            if (last_code == 0) {
2576
                s2->first_field = 0;
2577
                mpeg_decode_gop(avctx, buf_ptr, input_size);
2578
                s->sync = 1;
2579
            } else {
2580 2581
                av_log(avctx, AV_LOG_ERROR,
                       "ignoring GOP_START_CODE after %X\n", last_code);
2582 2583 2584 2585 2586
                if (avctx->err_recognition & AV_EF_EXPLODE)
                    return AVERROR_INVALIDDATA;
            }
            break;
        default:
2587 2588 2589 2590
            if (start_code >= SLICE_MIN_START_CODE &&
                start_code <= SLICE_MAX_START_CODE && last_code == PICTURE_START_CODE) {
                if (s2->progressive_sequence && !s2->progressive_frame) {
                    s2->progressive_frame = 1;
2591 2592
                    av_log(s2->avctx, AV_LOG_ERROR,
                           "interlaced frame in progressive sequence, ignoring\n");
2593 2594
                }

2595 2596 2597 2598 2599
                if (s2->picture_structure == 0 ||
                    (s2->progressive_frame && s2->picture_structure != PICT_FRAME)) {
                    av_log(s2->avctx, AV_LOG_ERROR,
                           "picture_structure %d invalid, ignoring\n",
                           s2->picture_structure);
2600 2601 2602
                    s2->picture_structure = PICT_FRAME;
                }

2603
                if (s2->progressive_sequence && !s2->frame_pred_frame_dct)
2604 2605 2606 2607 2608 2609 2610 2611 2612 2613 2614
                    av_log(s2->avctx, AV_LOG_WARNING, "invalid frame_pred_frame_dct\n");

                if (s2->picture_structure == PICT_FRAME) {
                    s2->first_field = 0;
                    s2->v_edge_pos  = 16 * s2->mb_height;
                } else {
                    s2->first_field ^= 1;
                    s2->v_edge_pos   = 8 * s2->mb_height;
                    memset(s2->mbskip_table, 0, s2->mb_stride * s2->mb_height);
                }
            }
2615 2616 2617
            if (start_code >= SLICE_MIN_START_CODE &&
                start_code <= SLICE_MAX_START_CODE && last_code != 0) {
                const int field_pic = s2->picture_structure != PICT_FRAME;
2618
                int mb_y = start_code - SLICE_MIN_START_CODE;
2619
                last_code = SLICE_MIN_START_CODE;
2620
                if (s2->codec_id != AV_CODEC_ID_MPEG1VIDEO && s2->mb_height > 2800/16)
2621
                    mb_y += (*buf_ptr&0xE0)<<2;
2622

2623
                mb_y <<= field_pic;
2624 2625 2626
                if (s2->picture_structure == PICT_BOTTOM_FIELD)
                    mb_y++;

2627 2628 2629 2630 2631
                if (buf_end - buf_ptr < 2) {
                    av_log(s2->avctx, AV_LOG_ERROR, "slice too small\n");
                    return AVERROR_INVALIDDATA;
                }

2632
                if (mb_y >= s2->mb_height) {
2633 2634
                    av_log(s2->avctx, AV_LOG_ERROR,
                           "slice below image (%d >= %d)\n", mb_y, s2->mb_height);
2635
                    return AVERROR_INVALIDDATA;
2636 2637
                }

2638
                if (!s2->last_picture_ptr) {
2639 2640
                    /* Skip B-frames if we do not have reference frames and
                     * GOP is not closed. */
2641
                    if (s2->pict_type == AV_PICTURE_TYPE_B) {
2642
                        if (!s2->closed_gop) {
2643
                            skip_frame = 1;
2644
                            break;
2645
                        }
2646 2647
                    }
                }
2648
                if (s2->pict_type == AV_PICTURE_TYPE_I || (s2->avctx->flags2 & CODEC_FLAG2_SHOW_ALL))
2649
                    s->sync = 1;
2650
                if (!s2->next_picture_ptr) {
2651 2652
                    /* Skip P-frames if we do not have a reference frame or
                     * we have an invalid header. */
2653 2654 2655 2656
                    if (s2->pict_type == AV_PICTURE_TYPE_P && !s->sync) {
                        skip_frame = 1;
                        break;
                    }
2657
                }
2658 2659 2660 2661 2662
                if ((avctx->skip_frame >= AVDISCARD_NONREF &&
                     s2->pict_type == AV_PICTURE_TYPE_B) ||
                    (avctx->skip_frame >= AVDISCARD_NONKEY &&
                     s2->pict_type != AV_PICTURE_TYPE_I) ||
                    avctx->skip_frame >= AVDISCARD_ALL) {
2663
                    skip_frame = 1;
2664
                    break;
2665
                }
2666 2667 2668 2669 2670

                if (!s->mpeg_enc_ctx_allocated)
                    break;

                if (s2->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
2671 2672
                    if (mb_y < avctx->skip_top ||
                        mb_y >= s2->mb_height - avctx->skip_bottom)
2673 2674 2675 2676 2677 2678 2679 2680 2681 2682
                        break;
                }

                if (!s2->pict_type) {
                    av_log(avctx, AV_LOG_ERROR, "Missing picture start code\n");
                    if (avctx->err_recognition & AV_EF_EXPLODE)
                        return AVERROR_INVALIDDATA;
                    break;
                }

2683
                if (s->first_slice) {
2684
                    skip_frame     = 0;
2685
                    s->first_slice = 0;
2686 2687
                    if ((ret = mpeg_field_start(s2, buf, buf_size)) < 0)
                        return ret;
2688 2689
                }
                if (!s2->current_picture_ptr) {
2690 2691
                    av_log(avctx, AV_LOG_ERROR,
                           "current_picture not initialized\n");
2692 2693 2694
                    return AVERROR_INVALIDDATA;
                }

2695 2696 2697 2698 2699
                if (uses_vdpau(avctx)) {
                    s->slice_count++;
                    break;
                }

2700 2701
                if (HAVE_THREADS &&
                    (avctx->active_thread_type & FF_THREAD_SLICE) &&
2702
                    !avctx->hwaccel) {
2703 2704 2705
                    int threshold = (s2->mb_height * s->slice_count +
                                     s2->slice_context_count / 2) /
                                    s2->slice_context_count;
2706
                    av_assert0(avctx->thread_count > 1);
2707 2708 2709 2710 2711 2712
                    if (threshold <= mb_y) {
                        MpegEncContext *thread_context = s2->thread_context[s->slice_count];

                        thread_context->start_mb_y = mb_y;
                        thread_context->end_mb_y   = s2->mb_height;
                        if (s->slice_count) {
2713 2714
                            s2->thread_context[s->slice_count - 1]->end_mb_y = mb_y;
                            ret = ff_update_duplicate_context(thread_context, s2);
2715 2716 2717
                            if (ret < 0)
                                return ret;
                        }
2718
                        init_get_bits(&thread_context->gb, buf_ptr, input_size * 8);
2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729
                        s->slice_count++;
                    }
                    buf_ptr += 2; // FIXME add minimum number of bytes per slice
                } else {
                    ret = mpeg_decode_slice(s2, mb_y, &buf_ptr, input_size);
                    emms_c();

                    if (ret < 0) {
                        if (avctx->err_recognition & AV_EF_EXPLODE)
                            return ret;
                        if (s2->resync_mb_x >= 0 && s2->resync_mb_y >= 0)
2730 2731 2732
                            ff_er_add_slice(&s2->er, s2->resync_mb_x,
                                            s2->resync_mb_y, s2->mb_x, s2->mb_y,
                                            ER_AC_ERROR | ER_DC_ERROR | ER_MV_ERROR);
2733
                    } else {
2734 2735 2736
                        ff_er_add_slice(&s2->er, s2->resync_mb_x,
                                        s2->resync_mb_y, s2->mb_x - 1, s2->mb_y,
                                        ER_AC_END | ER_DC_END | ER_MV_END);
2737 2738 2739 2740 2741 2742 2743 2744
                    }
                }
            }
            break;
        }
    }
}

2745 2746
static int mpeg_decode_frame(AVCodecContext *avctx, void *data,
                             int *got_output, AVPacket *avpkt)
2747 2748
{
    const uint8_t *buf = avpkt->data;
2749
    int ret;
2750 2751 2752 2753 2754 2755 2756 2757
    int buf_size = avpkt->size;
    Mpeg1Context *s = avctx->priv_data;
    AVFrame *picture = data;
    MpegEncContext *s2 = &s->mpeg_enc_ctx;

    if (buf_size == 0 || (buf_size == 4 && AV_RB32(buf) == SEQ_END_CODE)) {
        /* special case for last picture */
        if (s2->low_delay == 0 && s2->next_picture_ptr) {
2758
            int ret = av_frame_ref(picture, s2->next_picture_ptr->f);
2759 2760 2761 2762 2763 2764 2765 2766 2767 2768
            if (ret < 0)
                return ret;

            s2->next_picture_ptr = NULL;

            *got_output = 1;
        }
        return buf_size;
    }

2769
    if (s2->avctx->flags & CODEC_FLAG_TRUNCATED) {
2770 2771
        int next = ff_mpeg1_find_frame_end(&s2->parse_context, buf,
                                           buf_size, NULL);
2772

2773 2774
        if (ff_combine_frame(&s2->parse_context, next,
                             (const uint8_t **) &buf, &buf_size) < 0)
2775 2776 2777
            return buf_size;
    }

2778 2779 2780 2781
    s2->codec_tag = avpriv_toupper4(avctx->codec_tag);
    if (s->mpeg_enc_ctx_allocated == 0 && (   s2->codec_tag == AV_RL32("VCR2")
                                           || s2->codec_tag == AV_RL32("BW10")
                                          ))
2782 2783 2784 2785 2786
        vcr2_init_sequence(avctx);

    s->slice_count = 0;

    if (avctx->extradata && !s->extradata_decoded) {
2787 2788 2789
        ret = decode_chunks(avctx, picture, got_output,
                            avctx->extradata, avctx->extradata_size);
        if (*got_output) {
2790 2791 2792
            av_log(avctx, AV_LOG_ERROR, "picture in extradata\n");
            *got_output = 0;
        }
2793
        s->extradata_decoded = 1;
2794 2795
        if (ret < 0 && (avctx->err_recognition & AV_EF_EXPLODE)) {
            s2->current_picture_ptr = NULL;
2796
            return ret;
2797
        }
2798 2799
    }

2800 2801 2802 2803 2804
    ret = decode_chunks(avctx, picture, got_output, buf, buf_size);
    if (ret<0 || *got_output)
        s2->current_picture_ptr = NULL;

    return ret;
2805 2806 2807 2808 2809 2810
}

static void flush(AVCodecContext *avctx)
{
    Mpeg1Context *s = avctx->priv_data;

2811
    s->sync       = 0;
2812 2813 2814 2815

    ff_mpeg_flush(avctx);
}

2816
static av_cold int mpeg_decode_end(AVCodecContext *avctx)
2817 2818 2819 2820
{
    Mpeg1Context *s = avctx->priv_data;

    if (s->mpeg_enc_ctx_allocated)
2821
        ff_mpv_common_end(&s->mpeg_enc_ctx);
2822
    av_freep(&s->a53_caption);
2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834
    return 0;
}

static const AVProfile mpeg2_video_profiles[] = {
    { FF_PROFILE_MPEG2_422,          "4:2:2"              },
    { FF_PROFILE_MPEG2_HIGH,         "High"               },
    { FF_PROFILE_MPEG2_SS,           "Spatially Scalable" },
    { FF_PROFILE_MPEG2_SNR_SCALABLE, "SNR Scalable"       },
    { FF_PROFILE_MPEG2_MAIN,         "Main"               },
    { FF_PROFILE_MPEG2_SIMPLE,       "Simple"             },
    { FF_PROFILE_RESERVED,           "Reserved"           },
    { FF_PROFILE_RESERVED,           "Reserved"           },
2835
    { FF_PROFILE_UNKNOWN                                  },
2836 2837 2838 2839
};

AVCodec ff_mpeg1video_decoder = {
    .name                  = "mpeg1video",
2840
    .long_name             = NULL_IF_CONFIG_SMALL("MPEG-1 video"),
2841 2842 2843 2844 2845 2846 2847 2848 2849 2850
    .type                  = AVMEDIA_TYPE_VIDEO,
    .id                    = AV_CODEC_ID_MPEG1VIDEO,
    .priv_data_size        = sizeof(Mpeg1Context),
    .init                  = mpeg_decode_init,
    .close                 = mpeg_decode_end,
    .decode                = mpeg_decode_frame,
    .capabilities          = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
                             CODEC_CAP_TRUNCATED | CODEC_CAP_DELAY |
                             CODEC_CAP_SLICE_THREADS,
    .flush                 = flush,
2851
    .max_lowres            = 3,
2852 2853 2854 2855 2856
    .update_thread_context = ONLY_IF_THREADS_ENABLED(mpeg_decode_update_thread_context)
};

AVCodec ff_mpeg2video_decoder = {
    .name           = "mpeg2video",
2857
    .long_name      = NULL_IF_CONFIG_SMALL("MPEG-2 video"),
2858 2859 2860 2861 2862 2863 2864 2865 2866 2867
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_MPEG2VIDEO,
    .priv_data_size = sizeof(Mpeg1Context),
    .init           = mpeg_decode_init,
    .close          = mpeg_decode_end,
    .decode         = mpeg_decode_frame,
    .capabilities   = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
                      CODEC_CAP_TRUNCATED | CODEC_CAP_DELAY |
                      CODEC_CAP_SLICE_THREADS,
    .flush          = flush,
2868
    .max_lowres     = 3,
2869 2870 2871
    .profiles       = NULL_IF_CONFIG_SMALL(mpeg2_video_profiles),
};

2872 2873 2874
//legacy decoder
AVCodec ff_mpegvideo_decoder = {
    .name           = "mpegvideo",
2875
    .long_name      = NULL_IF_CONFIG_SMALL("MPEG-1 video"),
2876 2877 2878 2879 2880 2881 2882 2883 2884 2885 2886
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_MPEG2VIDEO,
    .priv_data_size = sizeof(Mpeg1Context),
    .init           = mpeg_decode_init,
    .close          = mpeg_decode_end,
    .decode         = mpeg_decode_frame,
    .capabilities   = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED | CODEC_CAP_DELAY | CODEC_CAP_SLICE_THREADS,
    .flush          = flush,
    .max_lowres     = 3,
};

2887
#if FF_API_XVMC
2888 2889 2890 2891 2892 2893 2894 2895
#if CONFIG_MPEG_XVMC_DECODER
static av_cold int mpeg_mc_decode_init(AVCodecContext *avctx)
{
    if (avctx->active_thread_type & FF_THREAD_SLICE)
        return -1;
    if (!(avctx->slice_flags & SLICE_FLAG_CODED_ORDER))
        return -1;
    if (!(avctx->slice_flags & SLICE_FLAG_ALLOW_FIELD)) {
2896
        ff_dlog(avctx, "mpeg12.c: XvMC decoder will work better if SLICE_FLAG_ALLOW_FIELD is set\n");
2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907
    }
    mpeg_decode_init(avctx);

    avctx->pix_fmt           = AV_PIX_FMT_XVMC_MPEG2_IDCT;
    avctx->xvmc_acceleration = 2; // 2 - the blocks are packed!

    return 0;
}

AVCodec ff_mpeg_xvmc_decoder = {
    .name           = "mpegvideo_xvmc",
2908
    .long_name      = NULL_IF_CONFIG_SMALL("MPEG-1/2 video XvMC (X-Video Motion Compensation)"),
2909 2910 2911 2912 2913 2914 2915
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_MPEG2VIDEO_XVMC,
    .priv_data_size = sizeof(Mpeg1Context),
    .init           = mpeg_mc_decode_init,
    .close          = mpeg_decode_end,
    .decode         = mpeg_decode_frame,
    .capabilities   = CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 |
2916
                      CODEC_CAP_TRUNCATED | CODEC_CAP_HWACCEL | CODEC_CAP_DELAY,
2917 2918 2919 2920
    .flush          = flush,
};

#endif
2921
#endif /* FF_API_XVMC */
2922 2923 2924 2925

#if CONFIG_MPEG_VDPAU_DECODER
AVCodec ff_mpeg_vdpau_decoder = {
    .name           = "mpegvideo_vdpau",
2926
    .long_name      = NULL_IF_CONFIG_SMALL("MPEG-1/2 video (VDPAU acceleration)"),
2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_MPEG2VIDEO,
    .priv_data_size = sizeof(Mpeg1Context),
    .init           = mpeg_decode_init,
    .close          = mpeg_decode_end,
    .decode         = mpeg_decode_frame,
    .capabilities   = CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED |
                      CODEC_CAP_HWACCEL_VDPAU | CODEC_CAP_DELAY,
    .flush          = flush,
};
#endif

#if CONFIG_MPEG1_VDPAU_DECODER
AVCodec ff_mpeg1_vdpau_decoder = {
    .name           = "mpeg1video_vdpau",
2942
    .long_name      = NULL_IF_CONFIG_SMALL("MPEG-1 video (VDPAU acceleration)"),
2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_MPEG1VIDEO,
    .priv_data_size = sizeof(Mpeg1Context),
    .init           = mpeg_decode_init,
    .close          = mpeg_decode_end,
    .decode         = mpeg_decode_frame,
    .capabilities   = CODEC_CAP_DR1 | CODEC_CAP_TRUNCATED |
                      CODEC_CAP_HWACCEL_VDPAU | CODEC_CAP_DELAY,
    .flush          = flush,
};
#endif