2013-03-26 21:50:40 +03:00
|
|
|
/*
|
|
|
|
* MPEG-1/2 decoder
|
|
|
|
* Copyright (c) 2000, 2001 Fabrice Bellard
|
2013-03-28 13:45:04 +03:00
|
|
|
* Copyright (c) 2002-2013 Michael Niedermayer <michaelni@gmx.at>
|
2013-03-26 21:50:40 +03:00
|
|
|
*
|
2013-03-28 13:45:04 +03:00
|
|
|
* This file is part of FFmpeg.
|
2013-03-26 21:50:40 +03:00
|
|
|
*
|
2013-03-28 13:45:04 +03:00
|
|
|
* FFmpeg is free software; you can redistribute it and/or
|
2013-03-26 21:50:40 +03:00
|
|
|
* 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.
|
|
|
|
*
|
2013-03-28 13:45:04 +03:00
|
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
2013-03-26 21:50:40 +03:00
|
|
|
* 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
|
2013-03-28 13:45:04 +03:00
|
|
|
* License along with FFmpeg; if not, write to the Free Software
|
2013-03-26 21:50:40 +03:00
|
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
*/
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @file
|
|
|
|
* MPEG-1/2 decoder
|
|
|
|
*/
|
|
|
|
|
2022-02-23 14:56:49 +02:00
|
|
|
#include "config_components.h"
|
|
|
|
|
2014-01-24 19:33:31 +03:00
|
|
|
#define UNCHECKED_BITSTREAM_READER 1
|
2014-03-13 14:13:33 +03:00
|
|
|
#include <inttypes.h>
|
|
|
|
|
2013-05-02 11:20:12 +03:00
|
|
|
#include "libavutil/attributes.h"
|
2023-09-01 00:23:35 +02:00
|
|
|
#include "libavutil/emms.h"
|
2015-05-05 04:15:31 +02:00
|
|
|
#include "libavutil/imgutils.h"
|
2013-03-26 21:50:40 +03:00
|
|
|
#include "libavutil/internal.h"
|
2020-05-27 14:54:38 +02:00
|
|
|
#include "libavutil/mem_internal.h"
|
2022-03-17 11:39:42 +02:00
|
|
|
#include "libavutil/reverse.h"
|
2013-07-31 20:04:50 +03:00
|
|
|
#include "libavutil/stereo3d.h"
|
2021-12-22 21:25:13 +02:00
|
|
|
#include "libavutil/timecode.h"
|
2014-02-04 16:57:54 +03:00
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
#include "avcodec.h"
|
2022-03-16 19:18:28 +02:00
|
|
|
#include "codec_internal.h"
|
2022-08-24 21:28:16 +02:00
|
|
|
#include "decode.h"
|
2013-03-26 21:50:40 +03:00
|
|
|
#include "error_resilience.h"
|
2023-08-01 19:44:22 +02:00
|
|
|
#include "hwaccel_internal.h"
|
2020-04-13 17:33:14 +02:00
|
|
|
#include "hwconfig.h"
|
2014-01-24 13:55:16 +03:00
|
|
|
#include "idctdsp.h"
|
2022-01-28 02:14:42 +02:00
|
|
|
#include "internal.h"
|
2014-06-20 16:05:45 +03:00
|
|
|
#include "mpeg_er.h"
|
2013-03-26 21:50:40 +03:00
|
|
|
#include "mpeg12.h"
|
2022-09-30 01:33:12 +02:00
|
|
|
#include "mpeg12codecs.h"
|
2013-03-26 21:50:40 +03:00
|
|
|
#include "mpeg12data.h"
|
2022-02-01 09:00:27 +02:00
|
|
|
#include "mpeg12dec.h"
|
2014-02-28 15:40:54 +03:00
|
|
|
#include "mpegutils.h"
|
2014-02-04 16:57:54 +03:00
|
|
|
#include "mpegvideo.h"
|
2015-05-29 20:44:03 +02:00
|
|
|
#include "mpegvideodata.h"
|
2022-02-01 09:42:25 +02:00
|
|
|
#include "mpegvideodec.h"
|
2014-12-31 08:57:59 +02:00
|
|
|
#include "profiles.h"
|
2022-02-04 16:43:14 +02:00
|
|
|
#include "startcode.h"
|
2013-03-26 21:50:40 +03:00
|
|
|
#include "thread.h"
|
|
|
|
|
2020-09-21 22:40:48 +02:00
|
|
|
#define A53_MAX_CC_COUNT 2000
|
|
|
|
|
2024-03-12 17:43:11 +02:00
|
|
|
enum Mpeg2ClosedCaptionsFormat {
|
|
|
|
CC_FORMAT_AUTO,
|
|
|
|
CC_FORMAT_A53_PART4,
|
|
|
|
CC_FORMAT_SCTE20,
|
|
|
|
CC_FORMAT_DVD
|
|
|
|
};
|
|
|
|
|
2013-03-26 21:52:21 +03:00
|
|
|
typedef struct Mpeg1Context {
|
|
|
|
MpegEncContext mpeg_enc_ctx;
|
2014-02-04 16:57:54 +03:00
|
|
|
int repeat_field; /* true if we must repeat the field */
|
|
|
|
AVPanScan pan_scan; /* some temporary storage for the panscan */
|
2024-06-23 03:44:52 +02:00
|
|
|
enum AVStereo3DType stereo3d_type;
|
2014-02-13 12:39:08 +03:00
|
|
|
int has_stereo3d;
|
2020-06-21 14:57:22 +02:00
|
|
|
AVBufferRef *a53_buf_ref;
|
2024-03-12 17:43:11 +02:00
|
|
|
enum Mpeg2ClosedCaptionsFormat cc_format;
|
2014-08-03 21:24:56 +03:00
|
|
|
uint8_t afd;
|
|
|
|
int has_afd;
|
2013-03-26 21:52:21 +03:00
|
|
|
int slice_count;
|
2022-01-25 14:29:49 +02:00
|
|
|
unsigned aspect_ratio_info;
|
2015-01-02 04:18:54 +02:00
|
|
|
AVRational save_aspect;
|
2013-03-26 21:52:21 +03:00
|
|
|
int save_width, save_height, save_progressive_seq;
|
2014-02-04 16:57:54 +03:00
|
|
|
AVRational frame_rate_ext; /* MPEG-2 specific framerate modificator */
|
2022-01-25 18:39:19 +02:00
|
|
|
unsigned frame_rate_index;
|
2014-02-04 16:57:54 +03:00
|
|
|
int sync; /* Did we reach a sync point like a GOP/SEQ/KEYFrame? */
|
2021-12-12 21:36:05 +02:00
|
|
|
int closed_gop;
|
2013-03-28 13:56:05 +03:00
|
|
|
int tmpgexs;
|
2013-11-30 00:26:15 +03:00
|
|
|
int first_slice;
|
2013-03-26 21:52:21 +03:00
|
|
|
int extradata_decoded;
|
2022-01-25 15:18:01 +02:00
|
|
|
int64_t timecode_frame_start; /*< GOP timecode frame start number, in non drop frame format */
|
2013-03-26 21:52:21 +03:00
|
|
|
} Mpeg1Context;
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
/* as H.263, but only 17 codes */
|
|
|
|
static int mpeg_decode_motion(MpegEncContext *s, int fcode, int pred)
|
|
|
|
{
|
|
|
|
int code, sign, val, shift;
|
|
|
|
|
2023-09-26 19:51:45 +02:00
|
|
|
code = get_vlc2(&s->gb, ff_mv_vlc, MV_VLC_BITS, 2);
|
2014-02-04 16:57:54 +03:00
|
|
|
if (code == 0)
|
2013-03-26 21:50:40 +03:00
|
|
|
return pred;
|
2014-02-04 16:57:54 +03:00
|
|
|
if (code < 0)
|
2013-03-26 21:50:40 +03:00
|
|
|
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);
|
|
|
|
}
|
|
|
|
|
2016-01-25 22:11:43 +02:00
|
|
|
#define MAX_INDEX (64 - 1)
|
2014-02-04 16:57:54 +03:00
|
|
|
#define check_scantable_index(ctx, x) \
|
|
|
|
do { \
|
2016-01-25 22:11:43 +02:00
|
|
|
if ((x) > MAX_INDEX) { \
|
2014-02-04 16:57:54 +03:00
|
|
|
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_inter(MpegEncContext *s,
|
|
|
|
int16_t *block, int n)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
|
|
|
int level, i, j, run;
|
2023-10-04 21:33:18 +02:00
|
|
|
const uint8_t *const scantable = s->intra_scantable.permutated;
|
2013-03-26 21:50:40 +03:00
|
|
|
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);
|
2014-02-04 16:57:54 +03:00
|
|
|
if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
|
2013-03-26 21:50:40 +03:00
|
|
|
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);
|
2014-02-04 16:57:54 +03:00
|
|
|
if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
|
2013-03-26 21:50:40 +03:00
|
|
|
goto end;
|
|
|
|
}
|
|
|
|
/* now quantify & encode AC coefficients */
|
|
|
|
for (;;) {
|
2022-10-21 23:17:07 +02:00
|
|
|
GET_RL_VLC(level, run, re, &s->gb, ff_mpeg1_rl_vlc,
|
2014-02-04 16:57:54 +03:00
|
|
|
TEX_VLC_BITS, 2, 0);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
if (level != 0) {
|
|
|
|
i += run;
|
2016-01-25 22:11:43 +02:00
|
|
|
if (i > MAX_INDEX)
|
|
|
|
break;
|
2013-03-26 21:50:40 +03:00
|
|
|
j = scantable[i];
|
|
|
|
level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
|
|
|
|
level = (level - 1) | 1;
|
2014-02-04 16:57:54 +03:00
|
|
|
level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
|
|
|
|
SHOW_SBITS(re, &s->gb, 1);
|
2013-03-26 21:50:40 +03:00
|
|
|
SKIP_BITS(re, &s->gb, 1);
|
|
|
|
} else {
|
|
|
|
/* escape */
|
2014-02-04 16:57:54 +03:00
|
|
|
run = SHOW_UBITS(re, &s->gb, 6) + 1;
|
|
|
|
LAST_SKIP_BITS(re, &s->gb, 6);
|
2013-03-26 21:50:40 +03:00
|
|
|
UPDATE_CACHE(re, &s->gb);
|
2014-02-04 16:57:54 +03:00
|
|
|
level = SHOW_SBITS(re, &s->gb, 8);
|
|
|
|
SKIP_BITS(re, &s->gb, 8);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (level == -128) {
|
2014-02-04 16:57:54 +03:00
|
|
|
level = SHOW_UBITS(re, &s->gb, 8) - 256;
|
|
|
|
SKIP_BITS(re, &s->gb, 8);
|
2013-03-26 21:50:40 +03:00
|
|
|
} else if (level == 0) {
|
2014-02-04 16:57:54 +03:00
|
|
|
level = SHOW_UBITS(re, &s->gb, 8);
|
|
|
|
SKIP_BITS(re, &s->gb, 8);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
i += run;
|
2016-01-25 22:11:43 +02:00
|
|
|
if (i > MAX_INDEX)
|
|
|
|
break;
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
2014-02-04 16:57:54 +03:00
|
|
|
if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
|
2013-03-26 21:50:40 +03:00
|
|
|
break;
|
|
|
|
UPDATE_CACHE(re, &s->gb);
|
|
|
|
}
|
|
|
|
end:
|
|
|
|
LAST_SKIP_BITS(re, &s->gb, 2);
|
|
|
|
CLOSE_READER(re, &s->gb);
|
|
|
|
}
|
2016-01-25 22:11:43 +02:00
|
|
|
|
|
|
|
check_scantable_index(s, i);
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
s->block_last_index[n] = i;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
static inline int mpeg2_decode_block_non_intra(MpegEncContext *s,
|
|
|
|
int16_t *block, int n)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
|
|
|
int level, i, j, run;
|
2023-10-04 21:33:18 +02:00
|
|
|
const uint8_t *const scantable = s->intra_scantable.permutated;
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
// Special case for first coefficient, no need to add second VLC table.
|
2013-03-26 21:50:40 +03:00
|
|
|
UPDATE_CACHE(re, &s->gb);
|
2014-02-04 16:57:54 +03:00
|
|
|
if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
|
|
|
|
level = (3 * qscale * quant_matrix[0]) >> 5;
|
2013-03-26 21:50:40 +03:00
|
|
|
if (GET_CACHE(re, &s->gb) & 0x40000000)
|
|
|
|
level = -level;
|
|
|
|
block[0] = level;
|
|
|
|
mismatch ^= level;
|
|
|
|
i++;
|
|
|
|
SKIP_BITS(re, &s->gb, 2);
|
2014-02-04 16:57:54 +03:00
|
|
|
if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
|
2013-03-26 21:50:40 +03:00
|
|
|
goto end;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* now quantify & encode AC coefficients */
|
|
|
|
for (;;) {
|
2022-10-21 23:17:07 +02:00
|
|
|
GET_RL_VLC(level, run, re, &s->gb, ff_mpeg1_rl_vlc,
|
2014-02-04 16:57:54 +03:00
|
|
|
TEX_VLC_BITS, 2, 0);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
if (level != 0) {
|
|
|
|
i += run;
|
2016-01-25 22:11:43 +02:00
|
|
|
if (i > MAX_INDEX)
|
|
|
|
break;
|
2013-03-26 21:50:40 +03:00
|
|
|
j = scantable[i];
|
|
|
|
level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
|
2014-02-04 16:57:54 +03:00
|
|
|
level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
|
|
|
|
SHOW_SBITS(re, &s->gb, 1);
|
2013-03-26 21:50:40 +03:00
|
|
|
SKIP_BITS(re, &s->gb, 1);
|
|
|
|
} else {
|
|
|
|
/* escape */
|
2014-02-04 16:57:54 +03:00
|
|
|
run = SHOW_UBITS(re, &s->gb, 6) + 1;
|
|
|
|
LAST_SKIP_BITS(re, &s->gb, 6);
|
2013-03-26 21:50:40 +03:00
|
|
|
UPDATE_CACHE(re, &s->gb);
|
2014-02-04 16:57:54 +03:00
|
|
|
level = SHOW_SBITS(re, &s->gb, 12);
|
|
|
|
SKIP_BITS(re, &s->gb, 12);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
i += run;
|
2016-01-25 22:11:43 +02:00
|
|
|
if (i > MAX_INDEX)
|
|
|
|
break;
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
2014-02-04 16:57:54 +03:00
|
|
|
if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
|
2013-03-26 21:50:40 +03:00
|
|
|
break;
|
|
|
|
UPDATE_CACHE(re, &s->gb);
|
|
|
|
}
|
|
|
|
end:
|
|
|
|
LAST_SKIP_BITS(re, &s->gb, 2);
|
|
|
|
CLOSE_READER(re, &s->gb);
|
|
|
|
}
|
|
|
|
block[63] ^= (mismatch & 1);
|
|
|
|
|
2016-01-25 22:11:43 +02:00
|
|
|
check_scantable_index(s, i);
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
s->block_last_index[n] = i;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
static inline int mpeg2_decode_block_intra(MpegEncContext *s,
|
|
|
|
int16_t *block, int n)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
|
|
|
int level, dc, diff, i, j, run;
|
|
|
|
int component;
|
2022-10-21 23:17:07 +02:00
|
|
|
const RL_VLC_ELEM *rl_vlc;
|
2023-10-04 21:33:18 +02:00
|
|
|
const uint8_t *const scantable = s->intra_scantable.permutated;
|
2013-03-26 21:50:40 +03:00
|
|
|
const uint16_t *quant_matrix;
|
|
|
|
const int qscale = s->qscale;
|
|
|
|
int mismatch;
|
|
|
|
|
|
|
|
/* DC coefficient */
|
|
|
|
if (n < 4) {
|
|
|
|
quant_matrix = s->intra_matrix;
|
2014-02-04 16:57:54 +03:00
|
|
|
component = 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
|
|
|
quant_matrix = s->chroma_intra_matrix;
|
2014-02-04 16:57:54 +03:00
|
|
|
component = (n & 1) + 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
diff = decode_dc(&s->gb, component);
|
|
|
|
dc = s->last_dc[component];
|
|
|
|
dc += diff;
|
|
|
|
s->last_dc[component] = dc;
|
2017-03-03 05:39:05 +02:00
|
|
|
block[0] = dc * (1 << (3 - s->intra_dc_precision));
|
2015-04-25 19:42:49 +02:00
|
|
|
ff_tlog(s->avctx, "dc=%d\n", block[0]);
|
2013-03-26 21:50:40 +03:00
|
|
|
mismatch = block[0] ^ 1;
|
|
|
|
i = 0;
|
|
|
|
if (s->intra_vlc_format)
|
2022-10-21 23:17:07 +02:00
|
|
|
rl_vlc = ff_mpeg2_rl_vlc;
|
2013-03-26 21:50:40 +03:00
|
|
|
else
|
2022-10-21 23:17:07 +02:00
|
|
|
rl_vlc = ff_mpeg1_rl_vlc;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
{
|
|
|
|
OPEN_READER(re, &s->gb);
|
|
|
|
/* now quantify & encode AC coefficients */
|
|
|
|
for (;;) {
|
|
|
|
UPDATE_CACHE(re, &s->gb);
|
2022-10-21 23:17:07 +02:00
|
|
|
GET_RL_VLC(level, run, re, &s->gb, rl_vlc,
|
2014-02-04 16:57:54 +03:00
|
|
|
TEX_VLC_BITS, 2, 0);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
if (level == 127) {
|
|
|
|
break;
|
|
|
|
} else if (level != 0) {
|
|
|
|
i += run;
|
2016-01-25 22:11:43 +02:00
|
|
|
if (i > MAX_INDEX)
|
|
|
|
break;
|
2014-02-04 16:57:54 +03:00
|
|
|
j = scantable[i];
|
2013-03-26 21:50:40 +03:00
|
|
|
level = (level * qscale * quant_matrix[j]) >> 4;
|
2014-02-04 16:57:54 +03:00
|
|
|
level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
|
|
|
|
SHOW_SBITS(re, &s->gb, 1);
|
2013-03-26 21:50:40 +03:00
|
|
|
LAST_SKIP_BITS(re, &s->gb, 1);
|
|
|
|
} else {
|
|
|
|
/* escape */
|
2014-02-04 16:57:54 +03:00
|
|
|
run = SHOW_UBITS(re, &s->gb, 6) + 1;
|
2020-10-08 21:45:21 +02:00
|
|
|
SKIP_BITS(re, &s->gb, 6);
|
2014-02-04 16:57:54 +03:00
|
|
|
level = SHOW_SBITS(re, &s->gb, 12);
|
2020-10-08 21:45:21 +02:00
|
|
|
LAST_SKIP_BITS(re, &s->gb, 12);
|
2013-03-26 21:50:40 +03:00
|
|
|
i += run;
|
2016-01-25 22:11:43 +02:00
|
|
|
if (i > MAX_INDEX)
|
|
|
|
break;
|
2014-02-04 16:57:54 +03:00
|
|
|
j = scantable[i];
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
|
|
|
|
2016-01-25 22:11:43 +02:00
|
|
|
check_scantable_index(s, i);
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
s->block_last_index[n] = i;
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* 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);
|
2014-11-20 17:31:27 +02:00
|
|
|
int ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2015-04-25 19:42:49 +02:00
|
|
|
ff_tlog(s->avctx, "decode_mb: x=%d y=%d\n", s->mb_x, s->mb_y);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
av_assert2(s->mb_skipped == 0);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
if (s->mb_skip_run-- != 0) {
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_P) {
|
|
|
|
s->mb_skipped = 1;
|
2023-10-04 03:34:28 +02:00
|
|
|
s->cur_pic.mb_type[s->mb_x + s->mb_y * s->mb_stride] =
|
2024-06-03 00:07:04 +02:00
|
|
|
MB_TYPE_SKIP | MB_TYPE_FORWARD_MV | MB_TYPE_16x16;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
|
|
|
int mb_type;
|
|
|
|
|
|
|
|
if (s->mb_x)
|
2023-10-04 03:34:28 +02:00
|
|
|
mb_type = s->cur_pic.mb_type[s->mb_x + s->mb_y * s->mb_stride - 1];
|
2013-03-26 21:50:40 +03:00
|
|
|
else
|
2014-02-04 16:57:54 +03:00
|
|
|
// FIXME not sure if this is allowed in MPEG at all
|
2023-10-04 03:34:28 +02:00
|
|
|
mb_type = s->cur_pic.mb_type[s->mb_width + (s->mb_y - 1) * s->mb_stride - 1];
|
2013-03-28 13:45:04 +03:00
|
|
|
if (IS_INTRA(mb_type)) {
|
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "skip with previntra\n");
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-28 13:45:04 +03:00
|
|
|
}
|
2023-10-04 03:34:28 +02:00
|
|
|
s->cur_pic.mb_type[s->mb_x + s->mb_y * s->mb_stride] =
|
2013-03-26 21:50:40 +03:00
|
|
|
mb_type | MB_TYPE_SKIP;
|
2013-03-28 13:45:04 +03:00
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
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) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR,
|
2016-04-27 19:45:23 +02:00
|
|
|
"Invalid mb type in I-frame at %d %d\n",
|
2014-02-04 16:57:54 +03:00
|
|
|
s->mb_x, s->mb_y);
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
mb_type = MB_TYPE_QUANT | MB_TYPE_INTRA;
|
|
|
|
} else {
|
|
|
|
mb_type = MB_TYPE_INTRA;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case AV_PICTURE_TYPE_P:
|
2023-09-26 19:51:45 +02:00
|
|
|
mb_type = get_vlc2(&s->gb, ff_mb_ptype_vlc, MB_PTYPE_VLC_BITS, 1);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (mb_type < 0) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR,
|
2016-04-27 19:45:23 +02:00
|
|
|
"Invalid mb type in P-frame at %d %d\n", s->mb_x, s->mb_y);
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
case AV_PICTURE_TYPE_B:
|
2023-09-26 19:51:45 +02:00
|
|
|
mb_type = get_vlc2(&s->gb, ff_mb_btype_vlc, MB_BTYPE_VLC_BITS, 1);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (mb_type < 0) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR,
|
2016-04-27 19:45:23 +02:00
|
|
|
"Invalid mb type in B-frame at %d %d\n", s->mb_x, s->mb_y);
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
2015-04-25 19:42:49 +02:00
|
|
|
ff_tlog(s->avctx, "mb_type=%x\n", mb_type);
|
2013-03-26 21:50:40 +03:00
|
|
|
// motion_type = 0; /* avoid warning */
|
|
|
|
if (IS_INTRA(mb_type)) {
|
2014-01-14 12:33:47 +03:00
|
|
|
s->bdsp.clear_blocks(s->block[0]);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
if (!s->chroma_y_shift)
|
2014-01-14 12:33:47 +03:00
|
|
|
s->bdsp.clear_blocks(s->block[6]);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
/* compute DCT type */
|
2014-02-04 16:57:54 +03:00
|
|
|
// FIXME: add an interlaced_dct coded var?
|
|
|
|
if (s->picture_structure == PICT_FRAME &&
|
|
|
|
!s->frame_pred_frame_dct)
|
2013-03-26 21:50:40 +03:00
|
|
|
s->interlaced_dct = get_bits1(&s->gb);
|
|
|
|
|
|
|
|
if (IS_QUANT(mb_type))
|
2017-12-29 17:42:14 +02:00
|
|
|
s->qscale = mpeg_get_qscale(s);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
if (s->concealment_motion_vectors) {
|
|
|
|
/* just parse them */
|
|
|
|
if (s->picture_structure != PICT_FRAME)
|
2014-02-04 16:57:54 +03:00
|
|
|
skip_bits1(&s->gb); /* field select */
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
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]);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2016-06-22 20:07:46 +02:00
|
|
|
check_marker(s->avctx, &s->gb, "after concealment_motion_vectors");
|
2014-02-04 16:57:54 +03:00
|
|
|
} else {
|
|
|
|
/* reset mv prediction */
|
|
|
|
memset(s->last_mv, 0, sizeof(s->last_mv));
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
s->mb_intra = 1;
|
|
|
|
|
|
|
|
if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
|
2023-10-29 23:53:07 +02:00
|
|
|
for (i = 0; i < mb_block_count; i++)
|
2024-04-28 21:34:30 +02:00
|
|
|
if ((ret = mpeg2_decode_block_intra(s, s->block[i], i)) < 0)
|
2023-10-29 23:53:07 +02:00
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2016-01-25 19:56:11 +02:00
|
|
|
for (i = 0; i < 6; i++) {
|
|
|
|
ret = ff_mpeg1_decode_block_intra(&s->gb,
|
|
|
|
s->intra_matrix,
|
|
|
|
s->intra_scantable.permutated,
|
2024-04-28 21:34:30 +02:00
|
|
|
s->last_dc, s->block[i],
|
2016-02-16 22:36:29 +02:00
|
|
|
i, s->qscale);
|
2016-01-25 19:56:11 +02:00
|
|
|
if (ret < 0) {
|
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n",
|
|
|
|
s->mb_x, s->mb_y);
|
2014-11-20 17:31:27 +02:00
|
|
|
return ret;
|
2016-01-25 19:56:11 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
s->block_last_index[i] = ret;
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
if (mb_type & MB_TYPE_ZERO_MV) {
|
2013-03-28 13:45:04 +03:00
|
|
|
av_assert2(mb_type & MB_TYPE_CBP);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
s->mv_dir = MV_DIR_FORWARD;
|
|
|
|
if (s->picture_structure == PICT_FRAME) {
|
2013-03-28 13:45:04 +03:00
|
|
|
if (s->picture_structure == PICT_FRAME
|
|
|
|
&& !s->frame_pred_frame_dct)
|
2013-03-26 21:50:40 +03:00
|
|
|
s->interlaced_dct = get_bits1(&s->gb);
|
|
|
|
s->mv_type = MV_TYPE_16X16;
|
|
|
|
} else {
|
2014-02-04 16:57:54 +03:00
|
|
|
s->mv_type = MV_TYPE_FIELD;
|
|
|
|
mb_type |= MB_TYPE_INTERLACED;
|
2013-03-26 21:50:40 +03:00
|
|
|
s->field_select[0][0] = s->picture_structure - 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (IS_QUANT(mb_type))
|
2017-12-29 17:42:14 +02:00
|
|
|
s->qscale = mpeg_get_qscale(s);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
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;
|
2014-02-04 16:57:54 +03:00
|
|
|
s->mv[0][0][0] = 0;
|
|
|
|
s->mv[0][0][1] = 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2024-06-03 00:07:04 +02:00
|
|
|
av_assert2(mb_type & MB_TYPE_BIDIR_MV);
|
2013-03-26 21:50:40 +03:00
|
|
|
// FIXME decide if MBs in field pictures are MB_TYPE_INTERLACED
|
|
|
|
/* get additional motion vector type */
|
2014-02-05 05:59:37 +03:00
|
|
|
if (s->picture_structure == PICT_FRAME && s->frame_pred_frame_dct) {
|
2013-03-26 21:50:40 +03:00
|
|
|
motion_type = MT_FRAME;
|
2014-02-04 16:57:54 +03:00
|
|
|
} else {
|
2013-03-26 21:50:40 +03:00
|
|
|
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))
|
2017-12-29 17:42:14 +02:00
|
|
|
s->qscale = mpeg_get_qscale(s);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
/* motion vectors */
|
2024-06-03 00:07:04 +02:00
|
|
|
s->mv_dir = MB_TYPE_MV_2_MV_DIR(mb_type);
|
2015-04-25 19:42:49 +02:00
|
|
|
ff_tlog(s->avctx, "motion_type=%d\n", motion_type);
|
2013-03-26 21:50:40 +03:00
|
|
|
switch (motion_type) {
|
|
|
|
case MT_FRAME: /* or MT_16X8 */
|
|
|
|
if (s->picture_structure == PICT_FRAME) {
|
2014-02-04 16:57:54 +03:00
|
|
|
mb_type |= MB_TYPE_16x16;
|
2013-03-26 21:50:40 +03:00
|
|
|
s->mv_type = MV_TYPE_16X16;
|
|
|
|
for (i = 0; i < 2; i++) {
|
2024-06-03 00:07:04 +02:00
|
|
|
if (HAS_MV(mb_type, i)) {
|
2013-03-26 21:50:40 +03:00
|
|
|
/* MT_FRAME */
|
2014-02-04 16:57:54 +03:00
|
|
|
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]);
|
2013-03-26 21:50:40 +03:00
|
|
|
/* full_pel: only for MPEG-1 */
|
|
|
|
if (s->full_pel[i]) {
|
2017-02-19 19:47:13 +02:00
|
|
|
s->mv[i][0][0] *= 2;
|
|
|
|
s->mv[i][0][1] *= 2;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
2014-02-04 16:57:54 +03:00
|
|
|
mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
|
2013-03-26 21:50:40 +03:00
|
|
|
s->mv_type = MV_TYPE_16X8;
|
|
|
|
for (i = 0; i < 2; i++) {
|
2024-06-03 00:07:04 +02:00
|
|
|
if (HAS_MV(mb_type, i)) {
|
2013-03-26 21:50:40 +03:00
|
|
|
/* 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++) {
|
2024-06-03 00:07:04 +02:00
|
|
|
if (HAS_MV(mb_type, i)) {
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
2015-04-25 19:42:49 +02:00
|
|
|
ff_tlog(s->avctx, "fmx=%d\n", val);
|
2013-03-26 21:50:40 +03:00
|
|
|
val = mpeg_decode_motion(s, s->mpeg_f_code[i][1],
|
|
|
|
s->last_mv[i][j][1] >> 1);
|
2015-03-11 05:24:57 +02:00
|
|
|
s->last_mv[i][j][1] = 2 * val;
|
2013-03-26 21:50:40 +03:00
|
|
|
s->mv[i][j][1] = val;
|
2015-04-25 19:42:49 +02:00
|
|
|
ff_tlog(s->avctx, "fmy=%d\n", val);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
2013-03-28 13:45:04 +03:00
|
|
|
av_assert0(!s->progressive_sequence);
|
2013-03-26 21:50:40 +03:00
|
|
|
mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
|
|
|
|
for (i = 0; i < 2; i++) {
|
2024-06-03 00:07:04 +02:00
|
|
|
if (HAS_MV(mb_type, i)) {
|
2013-03-26 21:50:40 +03:00
|
|
|
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:
|
2014-02-05 05:59:37 +03:00
|
|
|
if (s->progressive_sequence){
|
2013-03-28 13:45:04 +03:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "MT_DMV in progressive_sequence\n");
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-28 13:45:04 +03:00
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
s->mv_type = MV_TYPE_DMV;
|
|
|
|
for (i = 0; i < 2; i++) {
|
2024-06-03 00:07:04 +02:00
|
|
|
if (HAS_MV(mb_type, i)) {
|
2013-03-26 21:50:40 +03:00
|
|
|
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);
|
|
|
|
|
|
|
|
|
2017-02-25 22:07:24 +02:00
|
|
|
s->last_mv[i][0][1] = my * (1 << my_shift);
|
|
|
|
s->last_mv[i][1][1] = my * (1 << my_shift);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
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:
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR,
|
|
|
|
"00 motion_type at %d %d\n", s->mb_x, s->mb_y);
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
s->mb_intra = 0;
|
|
|
|
if (HAS_CBP(mb_type)) {
|
2014-01-14 12:33:47 +03:00
|
|
|
s->bdsp.clear_blocks(s->block[0]);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2023-09-26 19:51:45 +02:00
|
|
|
cbp = get_vlc2(&s->gb, ff_mb_pat_vlc, MB_PAT_VLC_BITS, 1);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (mb_block_count > 6) {
|
2017-03-12 04:04:04 +02:00
|
|
|
cbp *= 1 << mb_block_count - 6;
|
2014-02-04 16:57:54 +03:00
|
|
|
cbp |= get_bits(&s->gb, mb_block_count - 6);
|
2014-01-14 12:33:47 +03:00
|
|
|
s->bdsp.clear_blocks(s->block[6]);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
if (cbp <= 0) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR,
|
2014-02-05 05:59:37 +03:00
|
|
|
"invalid cbp %d at %d %d\n", cbp, s->mb_x, s->mb_y);
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if (s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
|
2023-10-29 23:53:07 +02:00
|
|
|
cbp <<= 12 - mb_block_count;
|
|
|
|
|
|
|
|
for (i = 0; i < mb_block_count; i++) {
|
|
|
|
if (cbp & (1 << 11)) {
|
2024-04-28 21:34:30 +02:00
|
|
|
if ((ret = mpeg2_decode_block_non_intra(s, s->block[i], i)) < 0)
|
2023-10-29 23:53:07 +02:00
|
|
|
return ret;
|
|
|
|
} else {
|
|
|
|
s->block_last_index[i] = -1;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
2023-10-29 23:53:07 +02:00
|
|
|
cbp += cbp;
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2023-10-29 23:53:07 +02:00
|
|
|
for (i = 0; i < 6; i++) {
|
|
|
|
if (cbp & 32) {
|
2024-04-28 21:34:30 +02:00
|
|
|
if ((ret = mpeg1_decode_block_inter(s, s->block[i], i)) < 0)
|
2023-10-29 23:53:07 +02:00
|
|
|
return ret;
|
|
|
|
} else {
|
|
|
|
s->block_last_index[i] = -1;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
2023-10-29 23:53:07 +02:00
|
|
|
cbp += cbp;
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
for (i = 0; i < 12; i++)
|
|
|
|
s->block_last_index[i] = -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-10-04 03:34:28 +02:00
|
|
|
s->cur_pic.mb_type[s->mb_x + s->mb_y * s->mb_stride] = mb_type;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static av_cold int mpeg_decode_init(AVCodecContext *avctx)
|
|
|
|
{
|
2014-02-04 16:57:54 +03:00
|
|
|
Mpeg1Context *s = avctx->priv_data;
|
2013-03-26 21:50:40 +03:00
|
|
|
MpegEncContext *s2 = &s->mpeg_enc_ctx;
|
avcodec/mpegpicture: Make MPVPicture refcounted
Up until now, an initialized MpegEncContext had an array of
MPVPictures (way more than were ever needed) and the MPVPicture*
contained in the MPVWorkPictures as well as the input_picture
and reordered_input_picture arrays (for the encoder) pointed
into this array. Several of the pointers could point to the
same slot and because there was no reference counting involved,
one had to check for aliasing before unreferencing.
Furthermore, given that these pointers were not ownership pointers
the pointers were often simply reset without unreferencing
the slot (happened e.g. for the RV30 and RV40 decoders) or
there were moved without resetting the src pointer (happened
for the encoders where the entries in the input_picture
and reordered_input_picture arrays were not reset).
Instead actually releasing these pictures was performed by looping
over the whole array and checking which one of the entries needed
to be kept. Given that the array had way too many slots (36),
this meant that more than 30 MPVPictures have been unnecessarily
unreferenced in every ff_mpv_frame_start(); something similar
happened for the encoder.
This commit changes this by making the MPVPictures refcounted
via the RefStruct API. The MPVPictures itself are part of a pool
so that this does not entail constant allocations; instead,
the amount of allocations actually goes down, because the
earlier code used such a large array of MPVPictures (36 entries) and
allocated an AVFrame for every one of these on every
ff_mpv_common_init(). In fact, the pool is only freed when closing
the codec, so that reinitializations don't lead to new allocations
(this avoids having to sync the pool in update_thread_context).
Making MPVPictures refcounted also has another key benefit:
It makes it possible to directly share them across threads
(when using frame-threaded decoding), eliminating ugly code
with underlying av_frame_ref()'s; sharing these pictures
can't fail any more.
The pool is allocated in ff_mpv_decode_init() for decoders,
which therefore can fail now. This and the fact that the pool
is not unreferenced in ff_mpv_common_end() also necessitated
to mark several mpegvideo-decoders with the FF_CODEC_CAP_INIT_CLEANUP
flag.
*: This also means that there is no good reason any more for
ff_mpv_common_frame_size_change() to exist.
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
2023-10-15 21:34:43 +02:00
|
|
|
int ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2024-04-22 02:11:15 +02:00
|
|
|
s2->out_format = FMT_MPEG1;
|
|
|
|
|
2014-11-05 02:21:59 +02:00
|
|
|
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
|
avcodec/mpegpicture: Make MPVPicture refcounted
Up until now, an initialized MpegEncContext had an array of
MPVPictures (way more than were ever needed) and the MPVPicture*
contained in the MPVWorkPictures as well as the input_picture
and reordered_input_picture arrays (for the encoder) pointed
into this array. Several of the pointers could point to the
same slot and because there was no reference counting involved,
one had to check for aliasing before unreferencing.
Furthermore, given that these pointers were not ownership pointers
the pointers were often simply reset without unreferencing
the slot (happened e.g. for the RV30 and RV40 decoders) or
there were moved without resetting the src pointer (happened
for the encoders where the entries in the input_picture
and reordered_input_picture arrays were not reset).
Instead actually releasing these pictures was performed by looping
over the whole array and checking which one of the entries needed
to be kept. Given that the array had way too many slots (36),
this meant that more than 30 MPVPictures have been unnecessarily
unreferenced in every ff_mpv_frame_start(); something similar
happened for the encoder.
This commit changes this by making the MPVPictures refcounted
via the RefStruct API. The MPVPictures itself are part of a pool
so that this does not entail constant allocations; instead,
the amount of allocations actually goes down, because the
earlier code used such a large array of MPVPictures (36 entries) and
allocated an AVFrame for every one of these on every
ff_mpv_common_init(). In fact, the pool is only freed when closing
the codec, so that reinitializations don't lead to new allocations
(this avoids having to sync the pool in update_thread_context).
Making MPVPictures refcounted also has another key benefit:
It makes it possible to directly share them across threads
(when using frame-threaded decoding), eliminating ugly code
with underlying av_frame_ref()'s; sharing these pictures
can't fail any more.
The pool is allocated in ff_mpv_decode_init() for decoders,
which therefore can fail now. This and the fact that the pool
is not unreferenced in ff_mpv_common_end() also necessitated
to mark several mpegvideo-decoders with the FF_CODEC_CAP_INIT_CLEANUP
flag.
*: This also means that there is no good reason any more for
ff_mpv_common_frame_size_change() to exist.
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
2023-10-15 21:34:43 +02:00
|
|
|
ret = ff_mpv_decode_init(s2, avctx);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
ff_mpeg12_init_vlcs();
|
|
|
|
|
2015-09-30 21:16:29 +02:00
|
|
|
s2->chroma_format = 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
s->repeat_field = 0;
|
2014-02-04 16:57:54 +03:00
|
|
|
avctx->color_range = AVCOL_RANGE_MPEG;
|
2013-03-26 21:50:40 +03:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2015-09-17 04:22:27 +02:00
|
|
|
#if HAVE_THREADS
|
2014-02-04 16:57:54 +03:00
|
|
|
static int mpeg_decode_update_thread_context(AVCodecContext *avctx,
|
|
|
|
const AVCodecContext *avctx_from)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
|
|
|
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;
|
|
|
|
|
2024-04-05 11:29:04 +02:00
|
|
|
if (avctx == avctx_from || !s1->context_initialized)
|
2013-03-26 21:50:40 +03:00
|
|
|
return 0;
|
|
|
|
|
|
|
|
err = ff_mpeg_update_thread_context(avctx, avctx_from);
|
2014-02-04 16:57:54 +03:00
|
|
|
if (err)
|
|
|
|
return err;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2024-04-05 11:29:04 +02:00
|
|
|
if (!s->context_initialized)
|
2013-03-26 21:50:40 +03:00
|
|
|
memcpy(s + 1, s1 + 1, sizeof(Mpeg1Context) - sizeof(MpegEncContext));
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
2015-09-17 04:22:27 +02:00
|
|
|
#endif
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
static const enum AVPixelFormat mpeg1_hwaccel_pixfmt_list_420[] = {
|
2017-11-16 17:38:10 +02:00
|
|
|
#if CONFIG_MPEG1_NVDEC_HWACCEL
|
|
|
|
AV_PIX_FMT_CUDA,
|
|
|
|
#endif
|
2015-08-17 01:35:59 +02:00
|
|
|
#if CONFIG_MPEG1_VDPAU_HWACCEL
|
2013-05-05 17:15:41 +03:00
|
|
|
AV_PIX_FMT_VDPAU,
|
2013-03-28 13:45:04 +03:00
|
|
|
#endif
|
|
|
|
AV_PIX_FMT_YUV420P,
|
|
|
|
AV_PIX_FMT_NONE
|
|
|
|
};
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
static const enum AVPixelFormat mpeg2_hwaccel_pixfmt_list_420[] = {
|
2017-11-16 17:35:17 +02:00
|
|
|
#if CONFIG_MPEG2_NVDEC_HWACCEL
|
|
|
|
AV_PIX_FMT_CUDA,
|
|
|
|
#endif
|
2015-08-17 01:35:59 +02:00
|
|
|
#if CONFIG_MPEG2_VDPAU_HWACCEL
|
2013-05-05 17:15:41 +03:00
|
|
|
AV_PIX_FMT_VDPAU,
|
2013-03-28 13:45:04 +03:00
|
|
|
#endif
|
2013-03-26 21:50:40 +03:00
|
|
|
#if CONFIG_MPEG2_DXVA2_HWACCEL
|
|
|
|
AV_PIX_FMT_DXVA2_VLD,
|
|
|
|
#endif
|
2015-05-11 10:52:47 +02:00
|
|
|
#if CONFIG_MPEG2_D3D11VA_HWACCEL
|
|
|
|
AV_PIX_FMT_D3D11VA_VLD,
|
2017-06-07 17:11:17 +02:00
|
|
|
AV_PIX_FMT_D3D11,
|
2015-05-11 10:52:47 +02:00
|
|
|
#endif
|
2023-12-05 08:46:48 +02:00
|
|
|
#if CONFIG_MPEG2_D3D12VA_HWACCEL
|
|
|
|
AV_PIX_FMT_D3D12,
|
|
|
|
#endif
|
2013-03-26 21:50:40 +03:00
|
|
|
#if CONFIG_MPEG2_VAAPI_HWACCEL
|
2015-07-28 10:16:59 +02:00
|
|
|
AV_PIX_FMT_VAAPI,
|
2015-07-11 13:23:21 +02:00
|
|
|
#endif
|
|
|
|
#if CONFIG_MPEG2_VIDEOTOOLBOX_HWACCEL
|
|
|
|
AV_PIX_FMT_VIDEOTOOLBOX,
|
2013-03-26 21:50:40 +03:00
|
|
|
#endif
|
|
|
|
AV_PIX_FMT_YUV420P,
|
|
|
|
AV_PIX_FMT_NONE
|
|
|
|
};
|
|
|
|
|
2014-09-25 11:59:58 +03:00
|
|
|
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
|
|
|
|
};
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
static enum AVPixelFormat mpeg_get_pixelformat(AVCodecContext *avctx)
|
|
|
|
{
|
2014-02-04 16:57:54 +03:00
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
2013-03-26 21:50:40 +03:00
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
2014-09-25 11:59:58 +03:00
|
|
|
const enum AVPixelFormat *pix_fmts;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2015-07-27 21:14:31 +02:00
|
|
|
if (CONFIG_GRAY && (avctx->flags & AV_CODEC_FLAG_GRAY))
|
2015-05-09 15:58:02 +02:00
|
|
|
return AV_PIX_FMT_GRAY8;
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
if (s->chroma_format < 2)
|
2014-09-27 04:38:12 +03:00
|
|
|
pix_fmts = avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO ?
|
2013-03-28 13:45:04 +03:00
|
|
|
mpeg1_hwaccel_pixfmt_list_420 :
|
2014-09-27 04:38:12 +03:00
|
|
|
mpeg2_hwaccel_pixfmt_list_420;
|
2013-11-05 10:16:31 +03:00
|
|
|
else if (s->chroma_format == 2)
|
2014-09-25 11:59:58 +03:00
|
|
|
pix_fmts = mpeg12_pixfmt_list_422;
|
2013-03-28 13:45:04 +03:00
|
|
|
else
|
2014-09-25 11:59:58 +03:00
|
|
|
pix_fmts = mpeg12_pixfmt_list_444;
|
|
|
|
|
2023-09-10 14:12:16 +02:00
|
|
|
return ff_get_format(avctx, pix_fmts);
|
2013-03-28 13:45:04 +03:00
|
|
|
}
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
/* 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)
|
|
|
|
{
|
2014-02-04 16:57:54 +03:00
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
2013-03-26 21:50:40 +03:00
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
2013-10-27 12:02:26 +03:00
|
|
|
int ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2015-01-02 03:33:34 +02:00
|
|
|
if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
|
|
|
|
// MPEG-1 aspect
|
2022-01-25 14:29:49 +02:00
|
|
|
AVRational aspect_inv = av_d2q(ff_mpeg1_aspect[s1->aspect_ratio_info], 255);
|
2017-05-11 00:49:31 +02:00
|
|
|
avctx->sample_aspect_ratio = (AVRational) { aspect_inv.den, aspect_inv.num };
|
2015-01-02 03:33:34 +02:00
|
|
|
} else { // MPEG-2
|
|
|
|
// MPEG-2 aspect
|
2022-01-25 14:29:49 +02:00
|
|
|
if (s1->aspect_ratio_info > 1) {
|
2015-01-02 03:33:34 +02:00
|
|
|
AVRational dar =
|
2022-01-25 14:29:49 +02:00
|
|
|
av_mul_q(av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
|
2015-01-02 03:33:34 +02:00
|
|
|
(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 =
|
2022-01-25 14:29:49 +02:00
|
|
|
av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
|
2015-01-02 03:33:34 +02:00
|
|
|
(AVRational) { s->width, s->height });
|
|
|
|
} else {
|
|
|
|
s->avctx->sample_aspect_ratio =
|
2022-01-25 14:29:49 +02:00
|
|
|
av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
|
2015-01-02 03:33:34 +02:00
|
|
|
(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});
|
2015-04-25 20:23:28 +02:00
|
|
|
ff_dlog(avctx, "aspect A %d/%d\n",
|
2022-01-25 14:29:49 +02:00
|
|
|
ff_mpeg2_aspect[s1->aspect_ratio_info].num,
|
|
|
|
ff_mpeg2_aspect[s1->aspect_ratio_info].den);
|
2015-04-25 20:23:28 +02:00
|
|
|
ff_dlog(avctx, "aspect B %d/%d\n", s->avctx->sample_aspect_ratio.num,
|
2015-01-02 03:33:34 +02:00
|
|
|
s->avctx->sample_aspect_ratio.den);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
s->avctx->sample_aspect_ratio =
|
2022-01-25 14:29:49 +02:00
|
|
|
ff_mpeg2_aspect[s1->aspect_ratio_info];
|
2015-01-02 03:33:34 +02:00
|
|
|
}
|
|
|
|
} // MPEG-2
|
|
|
|
|
2015-05-05 04:15:31 +02:00
|
|
|
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 };
|
|
|
|
}
|
2015-01-02 03:33:34 +02:00
|
|
|
|
2024-04-05 11:29:04 +02:00
|
|
|
if (!s->context_initialized ||
|
2014-02-04 16:57:54 +03:00
|
|
|
avctx->coded_width != s->width ||
|
|
|
|
avctx->coded_height != s->height ||
|
2013-03-26 21:50:40 +03:00
|
|
|
s1->save_width != s->width ||
|
|
|
|
s1->save_height != s->height ||
|
2015-01-02 04:18:54 +02:00
|
|
|
av_cmp_q(s1->save_aspect, s->avctx->sample_aspect_ratio) ||
|
2014-06-28 01:16:09 +03:00
|
|
|
(s1->save_progressive_seq != s->progressive_sequence && FFALIGN(s->height, 16) != FFALIGN(s->height, 32)) ||
|
2014-02-04 16:57:54 +03:00
|
|
|
0) {
|
2024-04-05 11:29:04 +02:00
|
|
|
if (s->context_initialized)
|
2014-08-10 18:25:12 +03:00
|
|
|
ff_mpv_common_end(s);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2013-10-27 12:02:26 +03:00
|
|
|
ret = ff_set_dimensions(avctx, s->width, s->height);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
|
2023-09-01 18:39:23 +02:00
|
|
|
if (avctx->codec_id == AV_CODEC_ID_MPEG2VIDEO && s->bit_rate &&
|
|
|
|
(s->bit_rate != 0x3FFFF*400)) {
|
2013-03-28 13:45:04 +03:00
|
|
|
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;
|
|
|
|
}
|
2015-01-02 04:18:54 +02:00
|
|
|
s1->save_aspect = s->avctx->sample_aspect_ratio;
|
2013-03-26 21:50:40 +03:00
|
|
|
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) {
|
2014-02-04 16:57:54 +03:00
|
|
|
// MPEG-1 fps
|
2022-01-25 18:39:19 +02:00
|
|
|
avctx->framerate = ff_mpeg12_frame_rate_tab[s1->frame_rate_index];
|
2023-05-04 11:50:48 +02:00
|
|
|
#if FF_API_TICKS_PER_FRAME
|
|
|
|
FF_DISABLE_DEPRECATION_WARNINGS
|
2014-02-04 16:57:54 +03:00
|
|
|
avctx->ticks_per_frame = 1;
|
2023-05-04 11:50:48 +02:00
|
|
|
FF_ENABLE_DEPRECATION_WARNINGS
|
|
|
|
#endif
|
2015-05-22 15:08:58 +02:00
|
|
|
|
|
|
|
avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
|
2014-02-04 16:57:54 +03:00
|
|
|
} else { // MPEG-2
|
|
|
|
// MPEG-2 fps
|
2014-04-04 13:47:44 +03:00
|
|
|
av_reduce(&s->avctx->framerate.num,
|
|
|
|
&s->avctx->framerate.den,
|
2022-01-25 18:39:19 +02:00
|
|
|
ff_mpeg12_frame_rate_tab[s1->frame_rate_index].num * s1->frame_rate_ext.num,
|
|
|
|
ff_mpeg12_frame_rate_tab[s1->frame_rate_index].den * s1->frame_rate_ext.den,
|
2013-03-26 21:50:40 +03:00
|
|
|
1 << 30);
|
2023-05-04 11:50:48 +02:00
|
|
|
#if FF_API_TICKS_PER_FRAME
|
|
|
|
FF_DISABLE_DEPRECATION_WARNINGS
|
2013-03-26 21:50:40 +03:00
|
|
|
avctx->ticks_per_frame = 2;
|
2023-05-04 11:50:48 +02:00
|
|
|
FF_ENABLE_DEPRECATION_WARNINGS
|
|
|
|
#endif
|
2015-05-22 15:08:58 +02:00
|
|
|
|
|
|
|
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;
|
2015-08-12 22:26:41 +02:00
|
|
|
default: av_assert0(0);
|
2015-05-22 15:08:58 +02:00
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
} // MPEG-2
|
|
|
|
|
|
|
|
avctx->pix_fmt = mpeg_get_pixelformat(avctx);
|
|
|
|
|
2014-11-24 19:05:00 +02:00
|
|
|
if ((ret = ff_mpv_common_init(s)) < 0)
|
|
|
|
return ret;
|
2024-06-04 11:33:04 +02:00
|
|
|
if (!s->avctx->lowres)
|
|
|
|
ff_mpv_framesize_disable(&s->sc);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
static int mpeg1_decode_picture(AVCodecContext *avctx, const uint8_t *buf,
|
|
|
|
int buf_size)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
2014-02-04 16:57:54 +03:00
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
2013-03-26 21:50:40 +03:00
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
2021-07-27 20:25:04 +02:00
|
|
|
int ref, f_code, vbv_delay, ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2021-07-27 20:25:04 +02:00
|
|
|
ret = init_get_bits8(&s->gb, buf, buf_size);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
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)
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
vbv_delay = get_bits(&s->gb, 16);
|
2013-03-28 13:45:04 +03:00
|
|
|
s->vbv_delay = vbv_delay;
|
2014-02-04 16:57:54 +03:00
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_P ||
|
|
|
|
s->pict_type == AV_PICTURE_TYPE_B) {
|
2013-03-26 21:50:40 +03:00
|
|
|
s->full_pel[0] = get_bits1(&s->gb);
|
|
|
|
f_code = get_bits(&s->gb, 3);
|
2013-03-28 13:45:04 +03:00
|
|
|
if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-28 13:45:04 +03:00
|
|
|
f_code += !f_code;
|
2013-03-26 21:50:40 +03:00
|
|
|
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);
|
2013-03-28 13:45:04 +03:00
|
|
|
if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-28 13:45:04 +03:00
|
|
|
f_code += !f_code;
|
2013-03-26 21:50:40 +03:00
|
|
|
s->mpeg_f_code[1][0] = f_code;
|
|
|
|
s->mpeg_f_code[1][1] = f_code;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (avctx->debug & FF_DEBUG_PICT_INFO)
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_DEBUG,
|
|
|
|
"vbv_delay %d, ref %d type:%d\n", vbv_delay, ref, s->pict_type);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
s->y_dc_scale = 8;
|
|
|
|
s->c_dc_scale = 8;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void mpeg_decode_sequence_extension(Mpeg1Context *s1)
|
|
|
|
{
|
2014-02-04 16:57:54 +03:00
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
2013-03-26 21:50:40 +03:00
|
|
|
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);
|
2014-02-04 16:57:54 +03:00
|
|
|
s->progressive_sequence = get_bits1(&s->gb); /* progressive_sequence */
|
2013-03-26 21:50:40 +03:00
|
|
|
s->chroma_format = get_bits(&s->gb, 2); /* chroma_format 1=420, 2=422, 3=444 */
|
2015-08-12 22:26:41 +02:00
|
|
|
|
|
|
|
if (!s->chroma_format) {
|
|
|
|
s->chroma_format = 1;
|
|
|
|
av_log(s->avctx, AV_LOG_WARNING, "Chroma format invalid\n");
|
|
|
|
}
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
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 */
|
2015-09-02 21:11:07 +02:00
|
|
|
s->bit_rate += (bit_rate_ext << 18) * 400LL;
|
2016-06-22 20:07:46 +02:00
|
|
|
check_marker(s->avctx, &s->gb, "after bit rate extension");
|
2023-09-01 18:20:55 +02:00
|
|
|
s->avctx->rc_buffer_size += get_bits(&s->gb, 8) * 1024 * 16 << 10;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
s->low_delay = get_bits1(&s->gb);
|
2015-06-29 21:59:37 +02:00
|
|
|
if (s->avctx->flags & AV_CODEC_FLAG_LOW_DELAY)
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
|
|
|
|
2015-03-16 10:57:36 +02:00
|
|
|
ff_dlog(s->avctx, "sequence extension\n");
|
2014-02-04 16:57:54 +03:00
|
|
|
s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
if (s->avctx->debug & FF_DEBUG_PICT_INFO)
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(s->avctx, AV_LOG_DEBUG,
|
2015-09-02 21:02:31 +02:00
|
|
|
"profile: %d, level: %d ps: %d cf:%d vbv buffer: %d, bitrate:%"PRId64"\n",
|
2014-02-05 05:59:37 +03:00
|
|
|
s->avctx->profile, s->avctx->level, s->progressive_sequence, s->chroma_format,
|
2023-09-01 18:20:55 +02:00
|
|
|
s->avctx->rc_buffer_size, s->bit_rate);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
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);
|
2014-02-04 16:57:54 +03:00
|
|
|
skip_bits(&s->gb, 1); // marker
|
2013-03-26 21:50:40 +03:00
|
|
|
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)
|
2014-03-13 14:13:33 +03:00
|
|
|
av_log(s->avctx, AV_LOG_DEBUG,
|
|
|
|
"pde (%"PRId16",%"PRId16") (%"PRId16",%"PRId16") (%"PRId16",%"PRId16")\n",
|
2013-03-26 21:50:40 +03:00
|
|
|
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]);
|
|
|
|
}
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
static int load_matrix(MpegEncContext *s, uint16_t matrix0[64],
|
|
|
|
uint16_t matrix1[64], int intra)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i = 0; i < 64; i++) {
|
2014-01-24 13:55:16 +03:00
|
|
|
int j = s->idsp.idct_permutation[ff_zigzag_direct[i]];
|
2013-03-26 21:50:40 +03:00
|
|
|
int v = get_bits(&s->gb, 8);
|
|
|
|
if (v == 0) {
|
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "matrix damaged\n");
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
if (intra && i == 0 && v != 8) {
|
2013-03-28 13:45:04 +03:00
|
|
|
av_log(s->avctx, AV_LOG_DEBUG, "intra matrix specifies invalid DC quantizer %d, ignoring\n", v);
|
2013-03-26 21:50:40 +03:00
|
|
|
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)
|
|
|
|
{
|
2015-03-16 10:57:36 +02:00
|
|
|
ff_dlog(s->avctx, "matrix extension\n");
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
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);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
2021-08-01 03:51:58 +02:00
|
|
|
static int mpeg_decode_picture_coding_extension(Mpeg1Context *s1)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
s->full_pel[0] = s->full_pel[1] = 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
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);
|
2021-08-31 20:15:09 +02:00
|
|
|
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];
|
2024-04-05 11:29:04 +02:00
|
|
|
if (!s->pict_type && s->context_initialized) {
|
2021-08-01 03:51:58 +02:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "Missing picture start code\n");
|
|
|
|
if (s->avctx->err_recognition & AV_EF_EXPLODE)
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
av_log(s->avctx, AV_LOG_WARNING, "Guessing pict_type from mpeg_f_code\n");
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
|
|
|
}
|
2013-03-28 13:45:04 +03:00
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
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);
|
|
|
|
|
2024-04-28 20:56:45 +02:00
|
|
|
// We only initialize intra_scantable, as both scantables always coincide
|
|
|
|
// and all code therefore only uses the intra one.
|
|
|
|
ff_init_scantable(s->idsp.idct_permutation, &s->intra_scantable,
|
|
|
|
s->alternate_scan ? ff_alternate_vertical_scan : ff_zigzag_direct);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
/* composite display not parsed */
|
2015-03-16 10:57:36 +02:00
|
|
|
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);
|
2021-08-01 03:51:58 +02:00
|
|
|
|
|
|
|
return 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
2024-04-28 20:42:41 +02:00
|
|
|
static int mpeg_field_start(Mpeg1Context *s1, const uint8_t *buf, int buf_size)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
2024-04-28 20:42:41 +02:00
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
2013-03-26 21:50:40 +03:00
|
|
|
AVCodecContext *avctx = s->avctx;
|
2024-04-28 21:34:30 +02:00
|
|
|
int second_field = 0;
|
2014-11-20 17:31:27 +02:00
|
|
|
int ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2019-09-04 15:09:09 +02:00
|
|
|
if (!(avctx->flags2 & AV_CODEC_FLAG2_CHUNKS)) {
|
|
|
|
if (s->mb_width * s->mb_height * 11LL / (33 * 2 * 8) > buf_size)
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
/* start frame decoding */
|
|
|
|
if (s->first_field || s->picture_structure == PICT_FRAME) {
|
|
|
|
AVFrameSideData *pan_scan;
|
|
|
|
|
2014-11-20 17:31:27 +02:00
|
|
|
if ((ret = ff_mpv_frame_start(s, avctx)) < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2023-10-07 23:34:08 +02:00
|
|
|
if (s->picture_structure != PICT_FRAME) {
|
2023-10-08 12:25:07 +02:00
|
|
|
s->cur_pic.ptr->f->flags |= AV_FRAME_FLAG_TOP_FIELD_FIRST *
|
2023-10-04 03:34:28 +02:00
|
|
|
(s->picture_structure == PICT_TOP_FIELD);
|
2023-10-07 23:34:08 +02:00
|
|
|
|
2024-04-05 06:06:17 +02:00
|
|
|
for (int i = 0; i < 3; i++) {
|
2023-10-07 23:34:08 +02:00
|
|
|
if (s->picture_structure == PICT_BOTTOM_FIELD) {
|
2023-10-04 17:27:24 +02:00
|
|
|
s->cur_pic.data[i] = FF_PTR_ADD(s->cur_pic.data[i],
|
|
|
|
s->cur_pic.linesize[i]);
|
2023-10-07 23:34:08 +02:00
|
|
|
}
|
2023-10-04 17:27:24 +02:00
|
|
|
s->cur_pic.linesize[i] *= 2;
|
2023-10-07 23:34:08 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
ff_mpeg_er_frame_start(s);
|
|
|
|
|
|
|
|
/* first check if we must repeat the frame */
|
2023-10-08 12:25:07 +02:00
|
|
|
s->cur_pic.ptr->f->repeat_pict = 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
if (s->repeat_first_field) {
|
|
|
|
if (s->progressive_sequence) {
|
|
|
|
if (s->top_field_first)
|
2023-10-08 12:25:07 +02:00
|
|
|
s->cur_pic.ptr->f->repeat_pict = 4;
|
2013-03-26 21:50:40 +03:00
|
|
|
else
|
2023-10-08 12:25:07 +02:00
|
|
|
s->cur_pic.ptr->f->repeat_pict = 2;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else if (s->progressive_frame) {
|
2023-10-08 12:25:07 +02:00
|
|
|
s->cur_pic.ptr->f->repeat_pict = 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-10-08 12:25:07 +02:00
|
|
|
ret = ff_frame_new_side_data(s->avctx, s->cur_pic.ptr->f,
|
2024-02-17 22:52:23 +02:00
|
|
|
AV_FRAME_DATA_PANSCAN, sizeof(s1->pan_scan),
|
|
|
|
&pan_scan);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
if (pan_scan)
|
|
|
|
memcpy(pan_scan->data, &s1->pan_scan, sizeof(s1->pan_scan));
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2020-06-21 14:57:22 +02:00
|
|
|
if (s1->a53_buf_ref) {
|
2024-02-17 22:52:23 +02:00
|
|
|
ret = ff_frame_new_side_data_from_buf(
|
2023-10-08 12:25:07 +02:00
|
|
|
s->avctx, s->cur_pic.ptr->f, AV_FRAME_DATA_A53_CC,
|
2024-07-29 18:59:35 +02:00
|
|
|
&s1->a53_buf_ref);
|
2024-02-17 22:52:23 +02:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2013-11-26 02:57:28 +03:00
|
|
|
}
|
2014-02-13 12:39:08 +03:00
|
|
|
|
|
|
|
if (s1->has_stereo3d) {
|
2023-10-08 12:25:07 +02:00
|
|
|
AVStereo3D *stereo = av_stereo3d_create_side_data(s->cur_pic.ptr->f);
|
2014-02-13 12:39:08 +03:00
|
|
|
if (!stereo)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
2024-06-23 03:44:52 +02:00
|
|
|
stereo->type = s1->stereo3d_type;
|
2014-02-13 12:39:08 +03:00
|
|
|
s1->has_stereo3d = 0;
|
|
|
|
}
|
2014-08-03 21:24:56 +03:00
|
|
|
|
|
|
|
if (s1->has_afd) {
|
2024-02-17 22:52:23 +02:00
|
|
|
AVFrameSideData *sd;
|
2023-10-08 12:25:07 +02:00
|
|
|
ret = ff_frame_new_side_data(s->avctx, s->cur_pic.ptr->f,
|
2024-02-17 22:52:23 +02:00
|
|
|
AV_FRAME_DATA_AFD, 1, &sd);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
if (sd)
|
|
|
|
*sd->data = s1->afd;
|
2014-08-03 21:24:56 +03:00
|
|
|
s1->has_afd = 0;
|
|
|
|
}
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_FRAME))
|
|
|
|
ff_thread_finish_setup(avctx);
|
|
|
|
} else { // second field
|
2024-04-28 21:34:30 +02:00
|
|
|
second_field = 1;
|
2023-10-08 12:25:07 +02:00
|
|
|
if (!s->cur_pic.ptr) {
|
2013-03-26 21:50:40 +03:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "first field missing\n");
|
2014-11-24 19:05:00 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
2017-10-23 09:43:30 +02:00
|
|
|
if (s->avctx->hwaccel) {
|
2023-08-01 19:44:22 +02:00
|
|
|
if ((ret = FF_HW_SIMPLE_CALL(s->avctx, end_frame)) < 0) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR,
|
|
|
|
"hardware accelerator failed to decode first field\n");
|
2015-08-05 19:54:41 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
2024-04-28 01:45:47 +02:00
|
|
|
ret = ff_mpv_alloc_dummy_frames(s);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2024-04-05 06:06:17 +02:00
|
|
|
for (int i = 0; i < 3; i++) {
|
2023-10-08 12:25:07 +02:00
|
|
|
s->cur_pic.data[i] = s->cur_pic.ptr->f->data[i];
|
2014-02-04 16:57:54 +03:00
|
|
|
if (s->picture_structure == PICT_BOTTOM_FIELD)
|
2023-10-04 17:27:24 +02:00
|
|
|
s->cur_pic.data[i] +=
|
2023-10-08 12:25:07 +02:00
|
|
|
s->cur_pic.ptr->f->linesize[i];
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (avctx->hwaccel) {
|
2023-08-01 19:44:22 +02:00
|
|
|
if ((ret = FF_HW_CALL(avctx, start_frame, buf, buf_size)) < 0)
|
2014-11-20 17:31:27 +02:00
|
|
|
return ret;
|
2024-04-28 21:34:30 +02:00
|
|
|
} else if (s->codec_tag == MKTAG('V', 'C', 'R', '2')) {
|
|
|
|
// Exchange UV
|
|
|
|
FFSWAP(uint8_t*, s->cur_pic.data[1], s->cur_pic.data[2]);
|
|
|
|
FFSWAP(ptrdiff_t, s->cur_pic.linesize[1], s->cur_pic.linesize[2]);
|
|
|
|
if (!second_field) {
|
|
|
|
FFSWAP(uint8_t*, s->next_pic.data[1], s->next_pic.data[2]);
|
|
|
|
FFSWAP(ptrdiff_t, s->next_pic.linesize[1], s->next_pic.linesize[2]);
|
|
|
|
FFSWAP(uint8_t*, s->last_pic.data[1], s->last_pic.data[2]);
|
|
|
|
FFSWAP(ptrdiff_t, s->last_pic.linesize[1], s->last_pic.linesize[2]);
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
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;
|
2013-03-28 13:45:04 +03:00
|
|
|
const int lowres = s->avctx->lowres;
|
2013-03-26 21:50:40 +03:00
|
|
|
const int field_pic = s->picture_structure != PICT_FRAME;
|
2014-11-20 17:31:27 +02:00
|
|
|
int ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
s->resync_mb_x =
|
|
|
|
s->resync_mb_y = -1;
|
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
av_assert0(mb_y < s->mb_height);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2023-01-05 01:47:14 +02:00
|
|
|
ret = init_get_bits8(&s->gb, *buf, buf_size);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
|
2014-02-05 05:59:37 +03:00
|
|
|
if (s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
|
2013-03-28 13:45:04 +03:00
|
|
|
skip_bits(&s->gb, 3);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
ff_mpeg1_clean_buffers(s);
|
|
|
|
s->interlaced_dct = 0;
|
|
|
|
|
2017-12-29 17:42:14 +02:00
|
|
|
s->qscale = mpeg_get_qscale(s);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
if (s->qscale == 0) {
|
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "qscale == 0\n");
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
/* extra slice info */
|
2013-10-30 20:28:00 +03:00
|
|
|
if (skip_1stop_8data_bits(&s->gb) < 0)
|
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
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) {
|
2023-09-26 19:51:45 +02:00
|
|
|
int code = get_vlc2(&s->gb, ff_mbincr_vlc,
|
2013-03-26 21:50:40 +03:00
|
|
|
MBINCR_VLC_BITS, 2);
|
|
|
|
if (code < 0) {
|
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "first mb_incr damaged\n");
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
if (code >= 33) {
|
2014-02-04 16:57:54 +03:00
|
|
|
if (code == 33)
|
2013-03-26 21:50:40 +03:00
|
|
|
s->mb_x += 33;
|
|
|
|
/* otherwise, stuffing, nothing to do */
|
|
|
|
} else {
|
|
|
|
s->mb_x += code;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
if (s->mb_x >= (unsigned) s->mb_width) {
|
2013-03-26 21:50:40 +03:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "initial skip overflow\n");
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
2023-08-01 17:21:23 +02:00
|
|
|
if (avctx->hwaccel) {
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
2023-08-01 19:44:22 +02:00
|
|
|
if (FF_HW_CALL(avctx, decode_slice, buf_start, buf_end - buf_start) < 0)
|
2013-03-26 21:50:40 +03:00
|
|
|
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) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(s->avctx, AV_LOG_DEBUG,
|
2021-12-23 02:27:59 +02:00
|
|
|
"qp:%d fc:%2d%2d%2d%2d %c %s %s %s %s dc:%d pstruct:%d fdct:%d cmv:%d qtype:%d ivlc:%d rff:%d %s\n",
|
2014-02-04 16:57:54 +03:00
|
|
|
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],
|
2021-12-23 02:27:59 +02:00
|
|
|
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')),
|
2014-02-04 16:57:54 +03:00
|
|
|
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" : "");
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for (;;) {
|
2014-11-20 17:31:27 +02:00
|
|
|
if ((ret = mpeg_decode_mb(s, s->block)) < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
// Note motion_val is normally NULL unless we want to extract the MVs.
|
2023-10-04 03:34:28 +02:00
|
|
|
if (s->cur_pic.motion_val[0]) {
|
2013-03-26 21:50:40 +03:00
|
|
|
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++) {
|
2014-02-04 16:57:54 +03:00
|
|
|
if (s->mb_intra ||
|
|
|
|
(dir == 1 && s->pict_type != AV_PICTURE_TYPE_B)) {
|
2013-03-26 21:50:40 +03:00
|
|
|
motion_x = motion_y = 0;
|
2014-02-04 16:57:54 +03:00
|
|
|
} else if (s->mv_type == MV_TYPE_16X16 ||
|
|
|
|
(s->mv_type == MV_TYPE_FIELD && field_pic)) {
|
2013-03-26 21:50:40 +03:00
|
|
|
motion_x = s->mv[dir][0][0];
|
|
|
|
motion_y = s->mv[dir][0][1];
|
2014-02-04 16:57:54 +03:00
|
|
|
} else { /* if ((s->mv_type == MV_TYPE_FIELD) || (s->mv_type == MV_TYPE_16X8)) */
|
2013-03-26 21:50:40 +03:00
|
|
|
motion_x = s->mv[dir][i][0];
|
|
|
|
motion_y = s->mv[dir][i][1];
|
|
|
|
}
|
|
|
|
|
2023-10-04 03:34:28 +02:00
|
|
|
s->cur_pic.motion_val[dir][xy][0] = motion_x;
|
|
|
|
s->cur_pic.motion_val[dir][xy][1] = motion_y;
|
|
|
|
s->cur_pic.motion_val[dir][xy + 1][0] = motion_x;
|
|
|
|
s->cur_pic.motion_val[dir][xy + 1][1] = motion_y;
|
|
|
|
s->cur_pic.ref_index [dir][b8_xy] =
|
|
|
|
s->cur_pic.ref_index [dir][b8_xy + 1] = s->field_select[dir][i];
|
2014-02-05 05:59:37 +03:00
|
|
|
av_assert2(s->field_select[dir][i] == 0 ||
|
|
|
|
s->field_select[dir][i] == 1);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
2014-02-04 16:57:54 +03:00
|
|
|
xy += wrap;
|
|
|
|
b8_xy += 2;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
s->dest[0] += 16 >> lowres;
|
|
|
|
s->dest[1] +=(16 >> lowres) >> s->chroma_x_shift;
|
|
|
|
s->dest[2] +=(16 >> lowres) >> s->chroma_x_shift;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2017-06-18 20:15:05 +02:00
|
|
|
ff_mpv_reconstruct_mb(s, s->block);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
if (++s->mb_x >= s->mb_width) {
|
2013-03-28 13:45:04 +03:00
|
|
|
const int mb_size = 16 >> s->avctx->lowres;
|
2016-02-24 02:44:27 +02:00
|
|
|
int left;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
ff_mpeg_draw_horiz_band(s, mb_size * (s->mb_y >> field_pic), mb_size);
|
2014-08-10 18:25:12 +03:00
|
|
|
ff_mpv_report_decode_progress(s);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
s->mb_x = 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
s->mb_y += 1 << field_pic;
|
|
|
|
|
|
|
|
if (s->mb_y >= s->mb_height) {
|
|
|
|
int left = get_bits_left(&s->gb);
|
2014-02-04 16:57:54 +03:00
|
|
|
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 */;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
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;
|
|
|
|
}
|
2016-05-01 01:25:20 +02:00
|
|
|
if (left > 32 && show_bits_long(&gb, 32) == 0x201) {
|
|
|
|
av_log(avctx, AV_LOG_DEBUG, "skipping m704 alpha (unsupported)\n");
|
|
|
|
goto eos;
|
|
|
|
}
|
2013-03-28 13:45:04 +03:00
|
|
|
}
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
if (left < 0 ||
|
|
|
|
(left && show_bits(&s->gb, FFMIN(left, 23)) && !is_d10) ||
|
2014-02-05 05:59:37 +03:00
|
|
|
((avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_AGGRESSIVE)) && left > 8)) {
|
2016-05-01 01:26:03 +02:00
|
|
|
av_log(avctx, AV_LOG_ERROR, "end mismatch left=%d %0X at %d %d\n",
|
|
|
|
left, left>0 ? show_bits(&s->gb, FFMIN(left, 23)) : 0, s->mb_x, s->mb_y);
|
2014-11-24 19:05:00 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else
|
|
|
|
goto eos;
|
|
|
|
}
|
2014-08-02 23:57:34 +03:00
|
|
|
// There are some files out there which are missing the last slice
|
2014-08-03 01:02:23 +03:00
|
|
|
// in cases where the slice is completely outside the visible
|
2014-08-02 23:57:34 +03:00
|
|
|
// area, we detect this here instead of running into the end expecting
|
|
|
|
// more data
|
2016-02-24 02:44:27 +02:00
|
|
|
left = get_bits_left(&s->gb);
|
2014-08-01 22:10:43 +03:00
|
|
|
if (s->mb_y >= ((s->height + 15) >> 4) &&
|
|
|
|
!s->progressive_sequence &&
|
2016-02-24 02:44:27 +02:00
|
|
|
left <= 25 &&
|
|
|
|
left >= 0 &&
|
2014-08-01 22:10:43 +03:00
|
|
|
s->mb_skip_run == -1 &&
|
2016-02-24 02:44:27 +02:00
|
|
|
(!left || show_bits(&s->gb, left) == 0))
|
2014-08-01 22:10:43 +03:00
|
|
|
goto eos;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
ff_init_block_index(s);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* skip mb handling */
|
|
|
|
if (s->mb_skip_run == -1) {
|
|
|
|
/* read increment again */
|
|
|
|
s->mb_skip_run = 0;
|
|
|
|
for (;;) {
|
2023-09-26 19:51:45 +02:00
|
|
|
int code = get_vlc2(&s->gb, ff_mbincr_vlc,
|
2013-03-26 21:50:40 +03:00
|
|
|
MBINCR_VLC_BITS, 2);
|
|
|
|
if (code < 0) {
|
|
|
|
av_log(s->avctx, AV_LOG_ERROR, "mb incr damaged\n");
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
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");
|
2014-11-24 19:05:00 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
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) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(s->avctx, AV_LOG_ERROR,
|
2016-04-27 19:45:23 +02:00
|
|
|
"skipped MB in I-frame at %d %d\n", s->mb_x, s->mb_y);
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
/* 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];
|
2018-02-10 11:36:32 +02:00
|
|
|
s->field_select[0][0] = (s->picture_structure - 1) & 1;
|
|
|
|
s->field_select[1][0] = (s->picture_structure - 1) & 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
eos: // end of slice
|
2014-08-01 22:06:13 +03:00
|
|
|
if (get_bits_left(&s->gb) < 0) {
|
|
|
|
av_log(s, AV_LOG_ERROR, "overread %d\n", -get_bits_left(&s->gb));
|
2014-01-24 19:45:26 +03:00
|
|
|
return AVERROR_INVALIDDATA;
|
2014-08-01 22:06:13 +03:00
|
|
|
}
|
2014-02-04 16:57:54 +03:00
|
|
|
*buf += (get_bits_count(&s->gb) - 1) / 8;
|
2015-04-25 20:23:28 +02:00
|
|
|
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);
|
2013-03-26 21:50:40 +03:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int slice_decode_thread(AVCodecContext *c, void *arg)
|
|
|
|
{
|
2014-02-04 16:57:54 +03:00
|
|
|
MpegEncContext *s = *(void **) arg;
|
2013-03-26 21:50:40 +03:00
|
|
|
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();
|
2015-03-16 10:57:36 +02:00
|
|
|
ff_dlog(c, "ret:%d resync:%d/%d mb:%d/%d ts:%d/%d ec:%d\n",
|
2013-03-26 21:50:40 +03:00
|
|
|
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)
|
2014-02-04 16:57:54 +03:00
|
|
|
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);
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2014-02-04 16:57:54 +03:00
|
|
|
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);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if (s->mb_y == s->end_mb_y)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
start_code = -1;
|
2014-02-04 16:57:54 +03:00
|
|
|
buf = avpriv_find_start_code(buf, s->gb.buffer_end, &start_code);
|
2019-09-20 00:16:57 +02:00
|
|
|
if (start_code < SLICE_MIN_START_CODE || start_code > SLICE_MAX_START_CODE)
|
|
|
|
return AVERROR_INVALIDDATA;
|
2014-02-05 05:59:37 +03:00
|
|
|
mb_y = start_code - SLICE_MIN_START_CODE;
|
|
|
|
if (s->codec_id != AV_CODEC_ID_MPEG1VIDEO && s->mb_height > 2800/16)
|
2013-03-28 13:45:04 +03:00
|
|
|
mb_y += (*buf&0xE0)<<2;
|
|
|
|
mb_y <<= field_pic;
|
2013-03-26 21:50:40 +03:00
|
|
|
if (s->picture_structure == PICT_BOTTOM_FIELD)
|
|
|
|
mb_y++;
|
2019-09-20 00:16:57 +02:00
|
|
|
if (mb_y >= s->end_mb_y)
|
2014-11-24 19:05:00 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Handle slice ends.
|
|
|
|
* @return 1 if it seems to be the last slice
|
|
|
|
*/
|
2024-04-28 01:45:47 +02:00
|
|
|
static int slice_end(AVCodecContext *avctx, AVFrame *pict, int *got_output)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
2014-02-04 16:57:54 +03:00
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
2013-03-26 21:50:40 +03:00
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
|
|
|
|
2023-10-08 12:25:07 +02:00
|
|
|
if (!s->context_initialized || !s->cur_pic.ptr)
|
2013-03-26 21:50:40 +03:00
|
|
|
return 0;
|
|
|
|
|
|
|
|
if (s->avctx->hwaccel) {
|
2023-08-01 19:44:22 +02:00
|
|
|
int ret = FF_HW_SIMPLE_CALL(s->avctx, end_frame);
|
2015-08-05 19:54:41 +02:00
|
|
|
if (ret < 0) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR,
|
|
|
|
"hardware accelerator failed to decode picture\n");
|
2015-08-05 19:54:41 +02:00
|
|
|
return ret;
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
/* end of slice reached */
|
2014-02-05 05:59:37 +03:00
|
|
|
if (/* s->mb_y << field_pic == s->mb_height && */ !s->first_field && !s1->first_slice) {
|
2013-03-26 21:50:40 +03:00
|
|
|
/* end of image */
|
|
|
|
|
2023-09-12 15:47:42 +02:00
|
|
|
ff_er_frame_end(&s->er, NULL);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2014-08-10 18:25:12 +03:00
|
|
|
ff_mpv_frame_end(s);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {
|
2023-10-08 12:25:07 +02:00
|
|
|
int ret = av_frame_ref(pict, s->cur_pic.ptr->f);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2023-10-08 12:25:07 +02:00
|
|
|
ff_print_debug_info(s, s->cur_pic.ptr, pict);
|
|
|
|
ff_mpv_export_qp_table(s, pict, s->cur_pic.ptr, FF_MPV_QSCALE_TYPE_MPEG2);
|
2024-04-28 01:45:47 +02:00
|
|
|
*got_output = 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
|
|
|
/* latency of 1 frame for I- and P-frames */
|
2023-10-08 12:25:07 +02:00
|
|
|
if (s->last_pic.ptr && !s->last_pic.ptr->dummy) {
|
|
|
|
int ret = av_frame_ref(pict, s->last_pic.ptr->f);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2023-10-08 12:25:07 +02:00
|
|
|
ff_print_debug_info(s, s->last_pic.ptr, pict);
|
|
|
|
ff_mpv_export_qp_table(s, pict, s->last_pic.ptr, FF_MPV_QSCALE_TYPE_MPEG2);
|
2024-04-28 01:45:47 +02:00
|
|
|
*got_output = 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 1;
|
|
|
|
} else {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int mpeg1_decode_sequence(AVCodecContext *avctx,
|
|
|
|
const uint8_t *buf, int buf_size)
|
|
|
|
{
|
2014-02-04 16:57:54 +03:00
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
2013-03-26 21:50:40 +03:00
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
|
|
|
int width, height;
|
|
|
|
int i, v, j;
|
|
|
|
|
2023-01-05 01:47:14 +02:00
|
|
|
int ret = init_get_bits8(&s->gb, buf, buf_size);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
width = get_bits(&s->gb, 12);
|
|
|
|
height = get_bits(&s->gb, 12);
|
|
|
|
if (width == 0 || height == 0) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_WARNING,
|
|
|
|
"Invalid horizontal or vertical size value.\n");
|
2013-03-28 13:45:04 +03:00
|
|
|
if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
|
2013-03-26 21:50:40 +03:00
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
2022-01-25 14:29:49 +02:00
|
|
|
s1->aspect_ratio_info = get_bits(&s->gb, 4);
|
|
|
|
if (s1->aspect_ratio_info == 0) {
|
2013-03-26 21:50:40 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR, "aspect ratio has forbidden 0 value\n");
|
2013-03-28 13:45:04 +03:00
|
|
|
if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
2022-01-25 18:39:19 +02:00
|
|
|
s1->frame_rate_index = get_bits(&s->gb, 4);
|
|
|
|
if (s1->frame_rate_index == 0 || s1->frame_rate_index > 13) {
|
2014-11-24 19:04:58 +02:00
|
|
|
av_log(avctx, AV_LOG_WARNING,
|
2022-01-25 18:39:19 +02:00
|
|
|
"frame_rate_index %d is invalid\n", s1->frame_rate_index);
|
|
|
|
s1->frame_rate_index = 1;
|
2014-11-20 16:08:18 +02:00
|
|
|
}
|
2015-09-02 21:11:07 +02:00
|
|
|
s->bit_rate = get_bits(&s->gb, 18) * 400LL;
|
2016-06-22 20:07:46 +02:00
|
|
|
if (check_marker(s->avctx, &s->gb, "in sequence header") == 0) {
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2014-11-20 16:08:18 +02:00
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2023-09-01 18:20:55 +02:00
|
|
|
s->avctx->rc_buffer_size = get_bits(&s->gb, 10) * 1024 * 16;
|
2013-03-26 21:50:40 +03:00
|
|
|
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++) {
|
2014-01-24 13:55:16 +03:00
|
|
|
j = s->idsp.idct_permutation[i];
|
2013-03-26 21:50:40 +03:00
|
|
|
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++) {
|
2014-01-24 13:55:16 +03:00
|
|
|
int j = s->idsp.idct_permutation[i];
|
2013-03-26 21:50:40 +03:00
|
|
|
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");
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
2015-09-01 02:45:10 +02:00
|
|
|
s->width = width;
|
|
|
|
s->height = height;
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
/* We set MPEG-2 parameters so that it emulates MPEG-1. */
|
2013-03-26 21:50:40 +03:00
|
|
|
s->progressive_sequence = 1;
|
|
|
|
s->progressive_frame = 1;
|
|
|
|
s->picture_structure = PICT_FRAME;
|
2013-03-28 13:45:04 +03:00
|
|
|
s->first_field = 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
s->frame_pred_frame_dct = 1;
|
|
|
|
s->chroma_format = 1;
|
2014-02-04 16:57:54 +03:00
|
|
|
s->codec_id =
|
|
|
|
s->avctx->codec_id = AV_CODEC_ID_MPEG1VIDEO;
|
2015-06-29 21:59:37 +02:00
|
|
|
if (s->avctx->flags & AV_CODEC_FLAG_LOW_DELAY)
|
2013-03-26 21:50:40 +03:00
|
|
|
s->low_delay = 1;
|
|
|
|
|
|
|
|
if (s->avctx->debug & FF_DEBUG_PICT_INFO)
|
2015-09-02 21:02:31 +02:00
|
|
|
av_log(s->avctx, AV_LOG_DEBUG, "vbv buffer: %d, bitrate:%"PRId64", aspect_ratio_info: %d \n",
|
2023-09-01 18:20:55 +02:00
|
|
|
s->avctx->rc_buffer_size, s->bit_rate, s1->aspect_ratio_info);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int vcr2_init_sequence(AVCodecContext *avctx)
|
|
|
|
{
|
2014-02-04 16:57:54 +03:00
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
2013-03-26 21:50:40 +03:00
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
2014-11-24 19:05:00 +02:00
|
|
|
int i, v, ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
/* start new MPEG-1 context decoding */
|
2024-04-05 11:29:04 +02:00
|
|
|
if (s->context_initialized)
|
2014-08-10 18:25:12 +03:00
|
|
|
ff_mpv_common_end(s);
|
2024-04-05 11:29:04 +02:00
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
s->width = avctx->coded_width;
|
|
|
|
s->height = avctx->coded_height;
|
2013-03-26 21:50:40 +03:00
|
|
|
avctx->has_b_frames = 0; // true?
|
2014-02-04 16:57:54 +03:00
|
|
|
s->low_delay = 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
avctx->pix_fmt = mpeg_get_pixelformat(avctx);
|
|
|
|
|
2014-11-20 17:31:27 +02:00
|
|
|
if ((ret = ff_mpv_common_init(s)) < 0)
|
|
|
|
return ret;
|
2024-06-04 11:33:04 +02:00
|
|
|
if (!s->avctx->lowres)
|
|
|
|
ff_mpv_framesize_disable(&s->sc);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
for (i = 0; i < 64; i++) {
|
2014-01-24 13:55:16 +03:00
|
|
|
int j = s->idsp.idct_permutation[i];
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
2013-03-28 13:45:04 +03:00
|
|
|
s->first_field = 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
s->frame_pred_frame_dct = 1;
|
|
|
|
s->chroma_format = 1;
|
2013-03-28 13:45:04 +03:00
|
|
|
if (s->codec_tag == AV_RL32("BW10")) {
|
|
|
|
s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG1VIDEO;
|
|
|
|
} else {
|
|
|
|
s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
s1->save_width = s->width;
|
|
|
|
s1->save_height = s->height;
|
|
|
|
s1->save_progressive_seq = s->progressive_sequence;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2024-03-12 17:43:11 +02:00
|
|
|
static void mpeg_set_cc_format(AVCodecContext *avctx, enum Mpeg2ClosedCaptionsFormat format,
|
|
|
|
const char *label)
|
|
|
|
{
|
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
|
|
|
|
|
|
|
av_assert2(format != CC_FORMAT_AUTO);
|
|
|
|
|
|
|
|
if (!s1->cc_format) {
|
|
|
|
s1->cc_format = format;
|
|
|
|
|
|
|
|
av_log(avctx, AV_LOG_DEBUG, "CC: first seen substream is %s format\n", label);
|
|
|
|
}
|
2024-10-12 20:04:50 +02:00
|
|
|
|
|
|
|
avctx->properties |= FF_CODEC_PROPERTY_CLOSED_CAPTIONS;
|
2024-03-12 17:43:11 +02:00
|
|
|
}
|
|
|
|
|
2013-11-26 02:57:28 +03:00
|
|
|
static int mpeg_decode_a53_cc(AVCodecContext *avctx,
|
|
|
|
const uint8_t *p, int buf_size)
|
|
|
|
{
|
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
|
|
|
|
2024-03-12 17:43:11 +02:00
|
|
|
if ((!s1->cc_format || s1->cc_format == CC_FORMAT_A53_PART4) &&
|
|
|
|
buf_size >= 6 &&
|
2013-11-26 02:57:28 +03:00
|
|
|
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) {
|
2020-06-21 14:57:22 +02:00
|
|
|
int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
|
|
|
|
const uint64_t new_size = (old_size + cc_count
|
|
|
|
* UINT64_C(3));
|
|
|
|
int ret;
|
|
|
|
|
2020-09-21 22:40:48 +02:00
|
|
|
if (new_size > 3*A53_MAX_CC_COUNT)
|
2020-06-21 14:57:22 +02:00
|
|
|
return AVERROR(EINVAL);
|
|
|
|
|
|
|
|
ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
|
|
|
|
if (ret >= 0)
|
|
|
|
memcpy(s1->a53_buf_ref->data + old_size, p + 7, cc_count * UINT64_C(3));
|
|
|
|
|
2024-03-12 17:43:11 +02:00
|
|
|
mpeg_set_cc_format(avctx, CC_FORMAT_A53_PART4, "A/53 Part 4");
|
2013-11-26 02:57:28 +03:00
|
|
|
}
|
|
|
|
return 1;
|
2024-03-12 17:43:11 +02:00
|
|
|
} else if ((!s1->cc_format || s1->cc_format == CC_FORMAT_SCTE20) &&
|
|
|
|
buf_size >= 2 &&
|
2017-03-15 21:53:03 +02:00
|
|
|
p[0] == 0x03 && (p[1]&0x7f) == 0x01) {
|
|
|
|
/* extract SCTE-20 CC data */
|
|
|
|
GetBitContext gb;
|
|
|
|
int cc_count = 0;
|
2020-06-21 14:57:22 +02:00
|
|
|
int i, ret;
|
2017-03-15 21:53:03 +02:00
|
|
|
|
2023-01-05 01:47:14 +02:00
|
|
|
ret = init_get_bits8(&gb, p + 2, buf_size - 2);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2017-03-15 21:53:03 +02:00
|
|
|
cc_count = get_bits(&gb, 5);
|
|
|
|
if (cc_count > 0) {
|
2020-06-21 14:57:22 +02:00
|
|
|
int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
|
|
|
|
const uint64_t new_size = (old_size + cc_count
|
|
|
|
* UINT64_C(3));
|
2020-09-21 22:40:48 +02:00
|
|
|
if (new_size > 3*A53_MAX_CC_COUNT)
|
2020-06-21 14:57:22 +02:00
|
|
|
return AVERROR(EINVAL);
|
|
|
|
|
|
|
|
ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
|
|
|
|
if (ret >= 0) {
|
2017-03-15 21:53:03 +02:00
|
|
|
uint8_t field, cc1, cc2;
|
2020-06-21 14:57:22 +02:00
|
|
|
uint8_t *cap = s1->a53_buf_ref->data;
|
2020-06-28 14:46:19 +02:00
|
|
|
|
|
|
|
memset(s1->a53_buf_ref->data + old_size, 0, cc_count * 3);
|
2017-03-15 21:53:03 +02:00
|
|
|
for (i = 0; i < cc_count && get_bits_left(&gb) >= 26; i++) {
|
|
|
|
skip_bits(&gb, 2); // priority
|
|
|
|
field = get_bits(&gb, 2);
|
|
|
|
skip_bits(&gb, 5); // line_offset
|
|
|
|
cc1 = get_bits(&gb, 8);
|
|
|
|
cc2 = get_bits(&gb, 8);
|
|
|
|
skip_bits(&gb, 1); // marker
|
|
|
|
|
|
|
|
if (!field) { // forbidden
|
|
|
|
cap[0] = cap[1] = cap[2] = 0x00;
|
|
|
|
} else {
|
|
|
|
field = (field == 2 ? 1 : 0);
|
|
|
|
if (!s1->mpeg_enc_ctx.top_field_first) field = !field;
|
|
|
|
cap[0] = 0x04 | field;
|
|
|
|
cap[1] = ff_reverse[cc1];
|
|
|
|
cap[2] = ff_reverse[cc2];
|
|
|
|
}
|
|
|
|
cap += 3;
|
|
|
|
}
|
|
|
|
}
|
2024-03-12 17:43:11 +02:00
|
|
|
|
|
|
|
mpeg_set_cc_format(avctx, CC_FORMAT_SCTE20, "SCTE-20");
|
2017-03-15 21:53:03 +02:00
|
|
|
}
|
|
|
|
return 1;
|
2024-03-12 17:43:11 +02:00
|
|
|
} else if ((!s1->cc_format || s1->cc_format == CC_FORMAT_DVD) &&
|
|
|
|
buf_size >= 11 &&
|
2014-02-04 16:57:54 +03:00
|
|
|
p[0] == 'C' && p[1] == 'C' && p[2] == 0x01 && p[3] == 0xf8) {
|
2016-09-12 21:34:48 +02:00
|
|
|
/* extract DVD CC data
|
|
|
|
*
|
|
|
|
* uint32_t user_data_start_code 0x000001B2 (big endian)
|
|
|
|
* uint16_t user_identifier 0x4343 "CC"
|
|
|
|
* uint8_t user_data_type_code 0x01
|
|
|
|
* uint8_t caption_block_size 0xF8
|
|
|
|
* uint8_t
|
|
|
|
* bit 7 caption_odd_field_first 1=odd field (CC1/CC2) first 0=even field (CC3/CC4) first
|
|
|
|
* bit 6 caption_filler 0
|
|
|
|
* bit 5:1 caption_block_count number of caption blocks (pairs of caption words = frames). Most DVDs use 15 per start of GOP.
|
|
|
|
* bit 0 caption_extra_field_added 1=one additional caption word
|
|
|
|
*
|
|
|
|
* struct caption_field_block {
|
|
|
|
* uint8_t
|
|
|
|
* bit 7:1 caption_filler 0x7F (all 1s)
|
|
|
|
* bit 0 caption_field_odd 1=odd field (this is CC1/CC2) 0=even field (this is CC3/CC4)
|
|
|
|
* uint8_t caption_first_byte
|
|
|
|
* uint8_t caption_second_byte
|
|
|
|
* } caption_block[(caption_block_count * 2) + caption_extra_field_added];
|
|
|
|
*
|
|
|
|
* Some DVDs encode caption data for both fields with caption_field_odd=1. The only way to decode the fields
|
|
|
|
* correctly is to start on the field indicated by caption_odd_field_first and count between odd/even fields.
|
|
|
|
* Don't assume that the first caption word is the odd field. There do exist MPEG files in the wild that start
|
|
|
|
* on the even field. There also exist DVDs in the wild that encode an odd field count and the
|
|
|
|
* caption_extra_field_added/caption_odd_field_first bits change per packet to allow that. */
|
2013-11-26 02:57:28 +03:00
|
|
|
int cc_count = 0;
|
2020-06-21 14:57:22 +02:00
|
|
|
int i, ret;
|
2013-11-26 02:57:28 +03:00
|
|
|
// There is a caption count field in the data, but it is often
|
2016-04-27 19:45:23 +02:00
|
|
|
// incorrect. So count the number of captions present.
|
2013-11-26 02:57:28 +03:00
|
|
|
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) {
|
2020-06-21 14:57:22 +02:00
|
|
|
int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
|
|
|
|
const uint64_t new_size = (old_size + cc_count
|
|
|
|
* UINT64_C(6));
|
2020-09-21 22:40:48 +02:00
|
|
|
if (new_size > 3*A53_MAX_CC_COUNT)
|
2020-06-21 14:57:22 +02:00
|
|
|
return AVERROR(EINVAL);
|
|
|
|
|
|
|
|
ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
|
|
|
|
if (ret >= 0) {
|
2013-11-26 02:57:28 +03:00
|
|
|
uint8_t field1 = !!(p[4] & 0x80);
|
2020-06-21 14:57:22 +02:00
|
|
|
uint8_t *cap = s1->a53_buf_ref->data;
|
2013-11-26 02:57:28 +03:00
|
|
|
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;
|
|
|
|
}
|
|
|
|
}
|
2024-03-12 17:43:11 +02:00
|
|
|
|
|
|
|
mpeg_set_cc_format(avctx, CC_FORMAT_DVD, "DVD");
|
2013-11-26 02:57:28 +03:00
|
|
|
}
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
static void mpeg_decode_user_data(AVCodecContext *avctx,
|
|
|
|
const uint8_t *p, int buf_size)
|
|
|
|
{
|
2013-03-28 13:45:04 +03:00
|
|
|
Mpeg1Context *s = avctx->priv_data;
|
2013-03-26 21:50:40 +03:00
|
|
|
const uint8_t *buf_end = p + buf_size;
|
2014-08-03 21:24:56 +03:00
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2015-01-02 03:27:18 +02:00
|
|
|
#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
|
|
|
|
|
2014-02-05 05:59:37 +03:00
|
|
|
if (buf_size > 29){
|
2013-03-28 13:45:04 +03:00
|
|
|
int i;
|
|
|
|
for(i=0; i<20; i++)
|
2014-02-05 05:59:37 +03:00
|
|
|
if (!memcmp(p+i, "\0TMPGEXS\0", 9)){
|
2013-03-28 13:45:04 +03:00
|
|
|
s->tmpgexs= 1;
|
|
|
|
}
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
/* 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;
|
2014-08-03 21:24:56 +03:00
|
|
|
s1->has_afd = 1;
|
2014-08-03 21:24:56 +03:00
|
|
|
s1->afd = p[0] & 0x0f;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
2013-07-31 20:04:50 +03:00
|
|
|
} 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) {
|
2013-12-15 11:18:05 +03:00
|
|
|
|
2014-02-13 12:39:08 +03:00
|
|
|
s1->has_stereo3d = 1;
|
2013-07-31 20:04:50 +03:00
|
|
|
|
|
|
|
switch (S3D_video_format_type) {
|
|
|
|
case 0x03:
|
2024-06-23 03:44:52 +02:00
|
|
|
s1->stereo3d_type = AV_STEREO3D_SIDEBYSIDE;
|
2013-07-31 20:04:50 +03:00
|
|
|
break;
|
|
|
|
case 0x04:
|
2024-06-23 03:44:52 +02:00
|
|
|
s1->stereo3d_type = AV_STEREO3D_TOPBOTTOM;
|
2013-07-31 20:04:50 +03:00
|
|
|
break;
|
|
|
|
case 0x08:
|
2024-06-23 03:44:52 +02:00
|
|
|
s1->stereo3d_type = AV_STEREO3D_2D;
|
2013-07-31 20:04:50 +03:00
|
|
|
break;
|
|
|
|
case 0x23:
|
2024-06-23 03:44:52 +02:00
|
|
|
s1->stereo3d_type = AV_STEREO3D_SIDEBYSIDE_QUINCUNX;
|
2013-07-31 20:04:50 +03:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2013-11-26 02:57:28 +03:00
|
|
|
} else if (mpeg_decode_a53_cc(avctx, p, buf_size)) {
|
|
|
|
return;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-01-05 01:47:14 +02:00
|
|
|
static int mpeg_decode_gop(AVCodecContext *avctx,
|
2013-03-26 21:50:40 +03:00
|
|
|
const uint8_t *buf, int buf_size)
|
|
|
|
{
|
|
|
|
Mpeg1Context *s1 = avctx->priv_data;
|
|
|
|
MpegEncContext *s = &s1->mpeg_enc_ctx;
|
|
|
|
int broken_link;
|
2013-03-28 13:45:04 +03:00
|
|
|
int64_t tc;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2023-01-05 01:47:14 +02:00
|
|
|
int ret = init_get_bits8(&s->gb, buf, buf_size);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2022-01-25 15:18:01 +02:00
|
|
|
tc = s1->timecode_frame_start = get_bits(&s->gb, 25);
|
2016-01-31 16:32:39 +02:00
|
|
|
|
2021-12-12 21:36:05 +02:00
|
|
|
s1->closed_gop = get_bits1(&s->gb);
|
2016-10-09 12:56:59 +02:00
|
|
|
/* broken_link indicates that after editing the
|
2014-02-04 16:57:54 +03:00
|
|
|
* reference frames of the first B-Frames after GOP I-Frame
|
|
|
|
* are missing (open gop) */
|
2013-03-26 21:50:40 +03:00
|
|
|
broken_link = get_bits1(&s->gb);
|
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
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",
|
2021-12-12 21:36:05 +02:00
|
|
|
tcbuf, s1->closed_gop, broken_link);
|
2013-03-28 13:45:04 +03:00
|
|
|
}
|
2023-01-05 01:47:14 +02:00
|
|
|
|
|
|
|
return 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
static int decode_chunks(AVCodecContext *avctx, AVFrame *picture,
|
|
|
|
int *got_output, const uint8_t *buf, int buf_size)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
|
|
|
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;
|
2013-05-21 11:40:01 +03:00
|
|
|
int last_code = 0, skip_frame = 0;
|
2013-03-28 13:45:04 +03:00
|
|
|
int picture_start_code_seen = 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
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) {
|
2013-05-21 11:40:01 +03:00
|
|
|
if (!skip_frame) {
|
2014-02-04 16:57:54 +03:00
|
|
|
if (HAVE_THREADS &&
|
|
|
|
(avctx->active_thread_type & FF_THREAD_SLICE) &&
|
2013-07-24 19:50:43 +03:00
|
|
|
!avctx->hwaccel) {
|
2013-03-26 21:50:40 +03:00
|
|
|
int i;
|
2013-03-28 13:45:04 +03:00
|
|
|
av_assert0(avctx->thread_count > 1);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
avctx->execute(avctx, slice_decode_thread,
|
|
|
|
&s2->thread_context[0], NULL,
|
|
|
|
s->slice_count, sizeof(void *));
|
2013-03-26 21:50:40 +03:00
|
|
|
for (i = 0; i < s->slice_count; i++)
|
|
|
|
s2->er.error_count += s2->thread_context[i]->er.error_count;
|
|
|
|
}
|
|
|
|
|
2024-04-28 01:45:47 +02:00
|
|
|
ret = slice_end(avctx, picture, got_output);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
s2->pict_type = 0;
|
2013-10-30 04:06:32 +03:00
|
|
|
|
|
|
|
if (avctx->err_recognition & AV_EF_EXPLODE && s2->er.error_count)
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
|
2021-03-08 12:26:32 +02:00
|
|
|
return FFMAX(0, buf_ptr - buf);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
input_size = buf_end - buf_ptr;
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
if (avctx->debug & FF_DEBUG_STARTCODE)
|
2014-04-24 19:01:30 +03:00
|
|
|
av_log(avctx, AV_LOG_DEBUG, "%3"PRIX32" at %"PTRDIFF_SPECIFIER" left %d\n",
|
2014-02-04 16:57:54 +03:00
|
|
|
start_code, buf_ptr - buf, input_size);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
/* 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);
|
2014-02-05 05:59:37 +03:00
|
|
|
if (buf != avctx->extradata)
|
|
|
|
s->sync = 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR,
|
|
|
|
"ignoring SEQ_START_CODE after %X\n", last_code);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (avctx->err_recognition & AV_EF_EXPLODE)
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case PICTURE_START_CODE:
|
2013-03-28 13:45:04 +03:00
|
|
|
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;
|
|
|
|
|
2023-06-25 20:38:32 +02:00
|
|
|
if (buf == avctx->extradata && avctx->codec_tag == AV_RL32("AVmp")) {
|
|
|
|
av_log(avctx, AV_LOG_WARNING, "ignoring picture start code in AVmp extradata\n");
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2014-02-05 05:59:37 +03:00
|
|
|
if (s->tmpgexs){
|
2013-03-28 13:45:04 +03:00
|
|
|
s2->intra_dc_precision= 3;
|
|
|
|
s2->intra_matrix[0]= 1;
|
|
|
|
}
|
2013-07-24 19:50:43 +03:00
|
|
|
if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE) &&
|
|
|
|
!avctx->hwaccel && s->slice_count) {
|
2013-03-26 21:50:40 +03:00
|
|
|
int i;
|
|
|
|
|
|
|
|
avctx->execute(avctx, slice_decode_thread,
|
|
|
|
s2->thread_context, NULL,
|
2014-02-04 16:57:54 +03:00
|
|
|
s->slice_count, sizeof(void *));
|
2013-03-26 21:50:40 +03:00
|
|
|
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) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR,
|
|
|
|
"mpeg_decode_postinit() failure\n");
|
2013-03-26 21:50:40 +03:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
/* We have a complete image: we try to decompress it. */
|
2013-03-26 21:50:40 +03:00
|
|
|
if (mpeg1_decode_picture(avctx, buf_ptr, input_size) < 0)
|
|
|
|
s2->pict_type = 0;
|
2013-11-30 00:26:15 +03:00
|
|
|
s->first_slice = 1;
|
2014-02-04 16:57:54 +03:00
|
|
|
last_code = PICTURE_START_CODE;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR,
|
|
|
|
"ignoring pic after %X\n", last_code);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (avctx->err_recognition & AV_EF_EXPLODE)
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case EXT_START_CODE:
|
2023-01-04 20:54:24 +02:00
|
|
|
ret = init_get_bits8(&s2->gb, buf_ptr, input_size);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
switch (get_bits(&s2->gb, 4)) {
|
|
|
|
case 0x1:
|
|
|
|
if (last_code == 0) {
|
2014-02-04 16:57:54 +03:00
|
|
|
mpeg_decode_sequence_extension(s);
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR,
|
|
|
|
"ignoring seq ext after %X\n", last_code);
|
2013-03-26 21:50:40 +03:00
|
|
|
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) {
|
2021-08-01 03:51:58 +02:00
|
|
|
int ret = mpeg_decode_picture_coding_extension(s);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR,
|
|
|
|
"ignoring pic cod ext after %X\n", last_code);
|
2013-03-26 21:50:40 +03:00
|
|
|
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) {
|
2014-02-04 16:57:54 +03:00
|
|
|
s2->first_field = 0;
|
2023-01-05 01:47:14 +02:00
|
|
|
ret = mpeg_decode_gop(avctx, buf_ptr, input_size);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2014-02-04 16:57:54 +03:00
|
|
|
s->sync = 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR,
|
|
|
|
"ignoring GOP_START_CODE after %X\n", last_code);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (avctx->err_recognition & AV_EF_EXPLODE)
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
2013-03-28 13:45:04 +03:00
|
|
|
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;
|
2014-02-05 05:59:37 +03:00
|
|
|
av_log(s2->avctx, AV_LOG_ERROR,
|
|
|
|
"interlaced frame in progressive sequence, ignoring\n");
|
2013-03-28 13:45:04 +03:00
|
|
|
}
|
|
|
|
|
2014-02-05 05:59:37 +03:00
|
|
|
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);
|
2013-03-28 13:45:04 +03:00
|
|
|
s2->picture_structure = PICT_FRAME;
|
|
|
|
}
|
|
|
|
|
2014-02-05 05:59:37 +03:00
|
|
|
if (s2->progressive_sequence && !s2->frame_pred_frame_dct)
|
2013-03-28 13:45:04 +03:00
|
|
|
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);
|
|
|
|
}
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
2013-03-28 13:45:04 +03:00
|
|
|
int mb_y = start_code - SLICE_MIN_START_CODE;
|
2013-03-26 21:50:40 +03:00
|
|
|
last_code = SLICE_MIN_START_CODE;
|
2014-02-05 05:59:37 +03:00
|
|
|
if (s2->codec_id != AV_CODEC_ID_MPEG1VIDEO && s2->mb_height > 2800/16)
|
2013-03-28 13:45:04 +03:00
|
|
|
mb_y += (*buf_ptr&0xE0)<<2;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
mb_y <<= field_pic;
|
2013-03-26 21:50:40 +03:00
|
|
|
if (s2->picture_structure == PICT_BOTTOM_FIELD)
|
|
|
|
mb_y++;
|
|
|
|
|
2013-08-13 20:36:50 +03:00
|
|
|
if (buf_end - buf_ptr < 2) {
|
|
|
|
av_log(s2->avctx, AV_LOG_ERROR, "slice too small\n");
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
|
2013-03-26 21:50:40 +03:00
|
|
|
if (mb_y >= s2->mb_height) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(s2->avctx, AV_LOG_ERROR,
|
|
|
|
"slice below image (%d >= %d)\n", mb_y, s2->mb_height);
|
2014-11-20 17:25:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
2023-10-08 12:25:07 +02:00
|
|
|
if (!s2->last_pic.ptr) {
|
2014-02-04 16:57:54 +03:00
|
|
|
/* Skip B-frames if we do not have reference frames and
|
|
|
|
* GOP is not closed. */
|
2013-03-26 21:50:40 +03:00
|
|
|
if (s2->pict_type == AV_PICTURE_TYPE_B) {
|
2021-12-12 21:36:05 +02:00
|
|
|
if (!s->closed_gop) {
|
2013-05-21 11:40:01 +03:00
|
|
|
skip_frame = 1;
|
2017-02-18 20:03:39 +02:00
|
|
|
av_log(s2->avctx, AV_LOG_DEBUG,
|
|
|
|
"Skipping B slice due to open GOP\n");
|
2013-03-26 21:50:40 +03:00
|
|
|
break;
|
2013-05-21 11:40:01 +03:00
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
2015-07-27 21:14:31 +02:00
|
|
|
if (s2->pict_type == AV_PICTURE_TYPE_I || (s2->avctx->flags2 & AV_CODEC_FLAG2_SHOW_ALL))
|
2014-02-04 16:57:54 +03:00
|
|
|
s->sync = 1;
|
2023-10-08 12:25:07 +02:00
|
|
|
if (!s2->next_pic.ptr) {
|
2014-02-04 16:57:54 +03:00
|
|
|
/* Skip P-frames if we do not have a reference frame or
|
|
|
|
* we have an invalid header. */
|
2013-05-21 11:40:01 +03:00
|
|
|
if (s2->pict_type == AV_PICTURE_TYPE_P && !s->sync) {
|
|
|
|
skip_frame = 1;
|
2017-02-18 20:03:39 +02:00
|
|
|
av_log(s2->avctx, AV_LOG_DEBUG,
|
|
|
|
"Skipping P slice due to !sync\n");
|
2013-05-21 11:40:01 +03:00
|
|
|
break;
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
2014-02-04 16:57:54 +03:00
|
|
|
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) {
|
2013-05-21 11:40:01 +03:00
|
|
|
skip_frame = 1;
|
2013-03-26 21:50:40 +03:00
|
|
|
break;
|
2013-05-21 11:40:01 +03:00
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2024-04-05 11:29:04 +02:00
|
|
|
if (!s2->context_initialized)
|
2013-03-26 21:50:40 +03:00
|
|
|
break;
|
|
|
|
|
|
|
|
if (s2->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
|
2014-02-04 16:57:54 +03:00
|
|
|
if (mb_y < avctx->skip_top ||
|
|
|
|
mb_y >= s2->mb_height - avctx->skip_bottom)
|
2013-03-26 21:50:40 +03:00
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2013-11-30 00:26:15 +03:00
|
|
|
if (s->first_slice) {
|
2014-02-04 16:57:54 +03:00
|
|
|
skip_frame = 0;
|
2013-11-30 00:26:15 +03:00
|
|
|
s->first_slice = 0;
|
2024-04-28 20:42:41 +02:00
|
|
|
if ((ret = mpeg_field_start(s, buf, buf_size)) < 0)
|
2014-11-20 17:31:27 +02:00
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
2023-10-08 12:25:07 +02:00
|
|
|
if (!s2->cur_pic.ptr) {
|
2014-02-04 16:57:54 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR,
|
|
|
|
"current_picture not initialized\n");
|
2013-03-26 21:50:40 +03:00
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
if (HAVE_THREADS &&
|
|
|
|
(avctx->active_thread_type & FF_THREAD_SLICE) &&
|
2013-07-24 19:50:43 +03:00
|
|
|
!avctx->hwaccel) {
|
2013-03-26 21:50:40 +03:00
|
|
|
int threshold = (s2->mb_height * s->slice_count +
|
|
|
|
s2->slice_context_count / 2) /
|
|
|
|
s2->slice_context_count;
|
2013-03-28 13:45:04 +03:00
|
|
|
av_assert0(avctx->thread_count > 1);
|
2013-03-26 21:50:40 +03:00
|
|
|
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) {
|
2014-02-04 16:57:54 +03:00
|
|
|
s2->thread_context[s->slice_count - 1]->end_mb_y = mb_y;
|
|
|
|
ret = ff_update_duplicate_context(thread_context, s2);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
}
|
2023-01-05 01:47:14 +02:00
|
|
|
ret = init_get_bits8(&thread_context->gb, buf_ptr, input_size);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
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)
|
2014-02-04 16:57:54 +03:00
|
|
|
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);
|
2013-03-26 21:50:40 +03:00
|
|
|
} else {
|
2014-02-04 16:57:54 +03:00
|
|
|
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);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-03-30 21:33:24 +02:00
|
|
|
static int mpeg_decode_frame(AVCodecContext *avctx, AVFrame *picture,
|
2014-02-04 16:57:54 +03:00
|
|
|
int *got_output, AVPacket *avpkt)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
|
|
|
const uint8_t *buf = avpkt->data;
|
2013-03-28 13:45:04 +03:00
|
|
|
int ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
int buf_size = avpkt->size;
|
|
|
|
Mpeg1Context *s = avctx->priv_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 */
|
2023-10-08 12:25:07 +02:00
|
|
|
if (s2->low_delay == 0 && s2->next_pic.ptr) {
|
|
|
|
int ret = av_frame_ref(picture, s2->next_pic.ptr->f);
|
2013-03-26 21:50:40 +03:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
|
avcodec/mpegpicture: Make MPVPicture refcounted
Up until now, an initialized MpegEncContext had an array of
MPVPictures (way more than were ever needed) and the MPVPicture*
contained in the MPVWorkPictures as well as the input_picture
and reordered_input_picture arrays (for the encoder) pointed
into this array. Several of the pointers could point to the
same slot and because there was no reference counting involved,
one had to check for aliasing before unreferencing.
Furthermore, given that these pointers were not ownership pointers
the pointers were often simply reset without unreferencing
the slot (happened e.g. for the RV30 and RV40 decoders) or
there were moved without resetting the src pointer (happened
for the encoders where the entries in the input_picture
and reordered_input_picture arrays were not reset).
Instead actually releasing these pictures was performed by looping
over the whole array and checking which one of the entries needed
to be kept. Given that the array had way too many slots (36),
this meant that more than 30 MPVPictures have been unnecessarily
unreferenced in every ff_mpv_frame_start(); something similar
happened for the encoder.
This commit changes this by making the MPVPictures refcounted
via the RefStruct API. The MPVPictures itself are part of a pool
so that this does not entail constant allocations; instead,
the amount of allocations actually goes down, because the
earlier code used such a large array of MPVPictures (36 entries) and
allocated an AVFrame for every one of these on every
ff_mpv_common_init(). In fact, the pool is only freed when closing
the codec, so that reinitializations don't lead to new allocations
(this avoids having to sync the pool in update_thread_context).
Making MPVPictures refcounted also has another key benefit:
It makes it possible to directly share them across threads
(when using frame-threaded decoding), eliminating ugly code
with underlying av_frame_ref()'s; sharing these pictures
can't fail any more.
The pool is allocated in ff_mpv_decode_init() for decoders,
which therefore can fail now. This and the fact that the pool
is not unreferenced in ff_mpv_common_end() also necessitated
to mark several mpegvideo-decoders with the FF_CODEC_CAP_INIT_CLEANUP
flag.
*: This also means that there is no good reason any more for
ff_mpv_common_frame_size_change() to exist.
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
2023-10-15 21:34:43 +02:00
|
|
|
ff_mpv_unref_picture(&s2->next_pic);
|
2013-03-26 21:50:40 +03:00
|
|
|
|
|
|
|
*got_output = 1;
|
|
|
|
}
|
|
|
|
return buf_size;
|
|
|
|
}
|
|
|
|
|
2024-04-05 11:29:04 +02:00
|
|
|
if (!s2->context_initialized &&
|
|
|
|
(s2->codec_tag == AV_RL32("VCR2") || s2->codec_tag == AV_RL32("BW10")))
|
2013-03-26 21:50:40 +03:00
|
|
|
vcr2_init_sequence(avctx);
|
|
|
|
|
|
|
|
s->slice_count = 0;
|
|
|
|
|
|
|
|
if (avctx->extradata && !s->extradata_decoded) {
|
2014-02-05 05:59:37 +03:00
|
|
|
ret = decode_chunks(avctx, picture, got_output,
|
|
|
|
avctx->extradata, avctx->extradata_size);
|
|
|
|
if (*got_output) {
|
2013-03-28 13:45:04 +03:00
|
|
|
av_log(avctx, AV_LOG_ERROR, "picture in extradata\n");
|
2016-10-20 22:51:55 +02:00
|
|
|
av_frame_unref(picture);
|
2013-03-28 13:45:04 +03:00
|
|
|
*got_output = 0;
|
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
s->extradata_decoded = 1;
|
2013-03-28 13:45:04 +03:00
|
|
|
if (ret < 0 && (avctx->err_recognition & AV_EF_EXPLODE)) {
|
avcodec/mpegpicture: Make MPVPicture refcounted
Up until now, an initialized MpegEncContext had an array of
MPVPictures (way more than were ever needed) and the MPVPicture*
contained in the MPVWorkPictures as well as the input_picture
and reordered_input_picture arrays (for the encoder) pointed
into this array. Several of the pointers could point to the
same slot and because there was no reference counting involved,
one had to check for aliasing before unreferencing.
Furthermore, given that these pointers were not ownership pointers
the pointers were often simply reset without unreferencing
the slot (happened e.g. for the RV30 and RV40 decoders) or
there were moved without resetting the src pointer (happened
for the encoders where the entries in the input_picture
and reordered_input_picture arrays were not reset).
Instead actually releasing these pictures was performed by looping
over the whole array and checking which one of the entries needed
to be kept. Given that the array had way too many slots (36),
this meant that more than 30 MPVPictures have been unnecessarily
unreferenced in every ff_mpv_frame_start(); something similar
happened for the encoder.
This commit changes this by making the MPVPictures refcounted
via the RefStruct API. The MPVPictures itself are part of a pool
so that this does not entail constant allocations; instead,
the amount of allocations actually goes down, because the
earlier code used such a large array of MPVPictures (36 entries) and
allocated an AVFrame for every one of these on every
ff_mpv_common_init(). In fact, the pool is only freed when closing
the codec, so that reinitializations don't lead to new allocations
(this avoids having to sync the pool in update_thread_context).
Making MPVPictures refcounted also has another key benefit:
It makes it possible to directly share them across threads
(when using frame-threaded decoding), eliminating ugly code
with underlying av_frame_ref()'s; sharing these pictures
can't fail any more.
The pool is allocated in ff_mpv_decode_init() for decoders,
which therefore can fail now. This and the fact that the pool
is not unreferenced in ff_mpv_common_end() also necessitated
to mark several mpegvideo-decoders with the FF_CODEC_CAP_INIT_CLEANUP
flag.
*: This also means that there is no good reason any more for
ff_mpv_common_frame_size_change() to exist.
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
2023-10-15 21:34:43 +02:00
|
|
|
ff_mpv_unref_picture(&s2->cur_pic);
|
2013-03-26 21:50:40 +03:00
|
|
|
return ret;
|
2013-03-28 13:45:04 +03:00
|
|
|
}
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
ret = decode_chunks(avctx, picture, got_output, buf, buf_size);
|
2016-01-31 16:32:39 +02:00
|
|
|
if (ret<0 || *got_output) {
|
avcodec/mpegpicture: Make MPVPicture refcounted
Up until now, an initialized MpegEncContext had an array of
MPVPictures (way more than were ever needed) and the MPVPicture*
contained in the MPVWorkPictures as well as the input_picture
and reordered_input_picture arrays (for the encoder) pointed
into this array. Several of the pointers could point to the
same slot and because there was no reference counting involved,
one had to check for aliasing before unreferencing.
Furthermore, given that these pointers were not ownership pointers
the pointers were often simply reset without unreferencing
the slot (happened e.g. for the RV30 and RV40 decoders) or
there were moved without resetting the src pointer (happened
for the encoders where the entries in the input_picture
and reordered_input_picture arrays were not reset).
Instead actually releasing these pictures was performed by looping
over the whole array and checking which one of the entries needed
to be kept. Given that the array had way too many slots (36),
this meant that more than 30 MPVPictures have been unnecessarily
unreferenced in every ff_mpv_frame_start(); something similar
happened for the encoder.
This commit changes this by making the MPVPictures refcounted
via the RefStruct API. The MPVPictures itself are part of a pool
so that this does not entail constant allocations; instead,
the amount of allocations actually goes down, because the
earlier code used such a large array of MPVPictures (36 entries) and
allocated an AVFrame for every one of these on every
ff_mpv_common_init(). In fact, the pool is only freed when closing
the codec, so that reinitializations don't lead to new allocations
(this avoids having to sync the pool in update_thread_context).
Making MPVPictures refcounted also has another key benefit:
It makes it possible to directly share them across threads
(when using frame-threaded decoding), eliminating ugly code
with underlying av_frame_ref()'s; sharing these pictures
can't fail any more.
The pool is allocated in ff_mpv_decode_init() for decoders,
which therefore can fail now. This and the fact that the pool
is not unreferenced in ff_mpv_common_end() also necessitated
to mark several mpegvideo-decoders with the FF_CODEC_CAP_INIT_CLEANUP
flag.
*: This also means that there is no good reason any more for
ff_mpv_common_frame_size_change() to exist.
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
2023-10-15 21:34:43 +02:00
|
|
|
ff_mpv_unref_picture(&s2->cur_pic);
|
2013-03-28 13:45:04 +03:00
|
|
|
|
2022-01-25 15:18:01 +02:00
|
|
|
if (s->timecode_frame_start != -1 && *got_output) {
|
2020-06-20 17:09:09 +02:00
|
|
|
char tcbuf[AV_TIMECODE_STR_SIZE];
|
2016-01-31 16:32:39 +02:00
|
|
|
AVFrameSideData *tcside = av_frame_new_side_data(picture,
|
|
|
|
AV_FRAME_DATA_GOP_TIMECODE,
|
|
|
|
sizeof(int64_t));
|
|
|
|
if (!tcside)
|
|
|
|
return AVERROR(ENOMEM);
|
2022-01-25 15:18:01 +02:00
|
|
|
memcpy(tcside->data, &s->timecode_frame_start, sizeof(int64_t));
|
2016-01-31 16:32:39 +02:00
|
|
|
|
2022-01-25 15:18:01 +02:00
|
|
|
av_timecode_make_mpeg_tc_string(tcbuf, s->timecode_frame_start);
|
2020-06-20 17:09:09 +02:00
|
|
|
av_dict_set(&picture->metadata, "timecode", tcbuf, 0);
|
|
|
|
|
2022-01-25 15:18:01 +02:00
|
|
|
s->timecode_frame_start = -1;
|
2016-01-31 16:32:39 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
return ret;
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
static void flush(AVCodecContext *avctx)
|
|
|
|
{
|
|
|
|
Mpeg1Context *s = avctx->priv_data;
|
|
|
|
|
2014-02-04 16:57:54 +03:00
|
|
|
s->sync = 0;
|
2021-12-12 21:36:05 +02:00
|
|
|
s->closed_gop = 0;
|
2013-03-26 21:50:40 +03:00
|
|
|
|
2023-01-02 21:18:36 +02:00
|
|
|
av_buffer_unref(&s->a53_buf_ref);
|
2013-03-26 21:50:40 +03:00
|
|
|
ff_mpeg_flush(avctx);
|
|
|
|
}
|
|
|
|
|
2013-05-02 11:20:12 +03:00
|
|
|
static av_cold int mpeg_decode_end(AVCodecContext *avctx)
|
2013-03-26 21:50:40 +03:00
|
|
|
{
|
|
|
|
Mpeg1Context *s = avctx->priv_data;
|
|
|
|
|
2020-06-21 14:57:22 +02:00
|
|
|
av_buffer_unref(&s->a53_buf_ref);
|
2024-04-28 17:40:58 +02:00
|
|
|
return ff_mpv_decode_close(avctx);
|
2013-03-26 21:50:40 +03:00
|
|
|
}
|
|
|
|
|
2022-03-16 22:09:54 +02:00
|
|
|
const FFCodec ff_mpeg1video_decoder = {
|
|
|
|
.p.name = "mpeg1video",
|
2022-08-29 13:38:02 +02:00
|
|
|
CODEC_LONG_NAME("MPEG-1 video"),
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.type = AVMEDIA_TYPE_VIDEO,
|
|
|
|
.p.id = AV_CODEC_ID_MPEG1VIDEO,
|
2013-03-26 21:50:40 +03:00
|
|
|
.priv_data_size = sizeof(Mpeg1Context),
|
|
|
|
.init = mpeg_decode_init,
|
|
|
|
.close = mpeg_decode_end,
|
2022-03-30 23:28:24 +02:00
|
|
|
FF_CODEC_DECODE_CB(mpeg_decode_frame),
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
|
2021-03-08 12:26:32 +02:00
|
|
|
AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
|
2022-07-10 00:05:45 +02:00
|
|
|
.caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
|
2013-03-26 21:50:40 +03:00
|
|
|
.flush = flush,
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.max_lowres = 3,
|
2022-08-29 15:03:21 +02:00
|
|
|
UPDATE_THREAD_CONTEXT(mpeg_decode_update_thread_context),
|
2020-12-27 18:36:59 +02:00
|
|
|
.hw_configs = (const AVCodecHWConfigInternal *const []) {
|
2018-01-22 03:57:41 +02:00
|
|
|
#if CONFIG_MPEG1_NVDEC_HWACCEL
|
2017-10-26 01:18:40 +02:00
|
|
|
HWACCEL_NVDEC(mpeg1),
|
|
|
|
#endif
|
|
|
|
#if CONFIG_MPEG1_VDPAU_HWACCEL
|
|
|
|
HWACCEL_VDPAU(mpeg1),
|
|
|
|
#endif
|
|
|
|
#if CONFIG_MPEG1_VIDEOTOOLBOX_HWACCEL
|
|
|
|
HWACCEL_VIDEOTOOLBOX(mpeg1),
|
|
|
|
#endif
|
|
|
|
NULL
|
|
|
|
},
|
2013-03-26 21:50:40 +03:00
|
|
|
};
|
|
|
|
|
2024-03-12 17:43:11 +02:00
|
|
|
#define M2V_OFFSET(x) offsetof(Mpeg1Context, x)
|
|
|
|
#define M2V_PARAM AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
|
|
|
|
|
|
|
|
static const AVOption mpeg2video_options[] = {
|
|
|
|
{ "cc_format", "extract a specific Closed Captions format",
|
|
|
|
M2V_OFFSET(cc_format), AV_OPT_TYPE_INT, { .i64 = CC_FORMAT_AUTO },
|
|
|
|
CC_FORMAT_AUTO, CC_FORMAT_DVD, M2V_PARAM, .unit = "cc_format" },
|
|
|
|
|
|
|
|
{ "auto", "pick first seen CC substream", 0, AV_OPT_TYPE_CONST,
|
|
|
|
{ .i64 = CC_FORMAT_AUTO }, .flags = M2V_PARAM, .unit = "cc_format" },
|
|
|
|
{ "a53", "pick A/53 Part 4 CC substream", 0, AV_OPT_TYPE_CONST,
|
|
|
|
{ .i64 = CC_FORMAT_A53_PART4 }, .flags = M2V_PARAM, .unit = "cc_format" },
|
|
|
|
{ "scte20", "pick SCTE-20 CC substream", 0, AV_OPT_TYPE_CONST,
|
|
|
|
{ .i64 = CC_FORMAT_SCTE20 }, .flags = M2V_PARAM, .unit = "cc_format" },
|
|
|
|
{ "dvd", "pick DVD CC substream", 0, AV_OPT_TYPE_CONST,
|
|
|
|
{ .i64 = CC_FORMAT_DVD }, .flags = M2V_PARAM, .unit = "cc_format" },
|
|
|
|
{ NULL }
|
|
|
|
};
|
|
|
|
|
|
|
|
static const AVClass mpeg2video_class = {
|
|
|
|
.class_name = "MPEG-2 video",
|
|
|
|
.item_name = av_default_item_name,
|
|
|
|
.option = mpeg2video_options,
|
|
|
|
.version = LIBAVUTIL_VERSION_INT,
|
|
|
|
.category = AV_CLASS_CATEGORY_DECODER,
|
|
|
|
};
|
|
|
|
|
2022-03-16 22:09:54 +02:00
|
|
|
const FFCodec ff_mpeg2video_decoder = {
|
|
|
|
.p.name = "mpeg2video",
|
2022-08-29 13:38:02 +02:00
|
|
|
CODEC_LONG_NAME("MPEG-2 video"),
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.type = AVMEDIA_TYPE_VIDEO,
|
|
|
|
.p.id = AV_CODEC_ID_MPEG2VIDEO,
|
2024-03-12 17:43:11 +02:00
|
|
|
.p.priv_class = &mpeg2video_class,
|
2013-03-26 21:50:40 +03:00
|
|
|
.priv_data_size = sizeof(Mpeg1Context),
|
|
|
|
.init = mpeg_decode_init,
|
|
|
|
.close = mpeg_decode_end,
|
2022-03-30 23:28:24 +02:00
|
|
|
FF_CODEC_DECODE_CB(mpeg_decode_frame),
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
|
2021-03-08 12:26:32 +02:00
|
|
|
AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
|
2022-07-10 00:05:45 +02:00
|
|
|
.caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
|
2013-03-26 21:50:40 +03:00
|
|
|
.flush = flush,
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.max_lowres = 3,
|
|
|
|
.p.profiles = NULL_IF_CONFIG_SMALL(ff_mpeg2_video_profiles),
|
2020-12-27 18:36:59 +02:00
|
|
|
.hw_configs = (const AVCodecHWConfigInternal *const []) {
|
2017-10-26 01:18:40 +02:00
|
|
|
#if CONFIG_MPEG2_DXVA2_HWACCEL
|
|
|
|
HWACCEL_DXVA2(mpeg2),
|
|
|
|
#endif
|
|
|
|
#if CONFIG_MPEG2_D3D11VA_HWACCEL
|
|
|
|
HWACCEL_D3D11VA(mpeg2),
|
|
|
|
#endif
|
|
|
|
#if CONFIG_MPEG2_D3D11VA2_HWACCEL
|
|
|
|
HWACCEL_D3D11VA2(mpeg2),
|
|
|
|
#endif
|
2023-12-05 08:46:48 +02:00
|
|
|
#if CONFIG_MPEG2_D3D12VA_HWACCEL
|
|
|
|
HWACCEL_D3D12VA(mpeg2),
|
|
|
|
#endif
|
2017-10-26 01:18:40 +02:00
|
|
|
#if CONFIG_MPEG2_NVDEC_HWACCEL
|
|
|
|
HWACCEL_NVDEC(mpeg2),
|
|
|
|
#endif
|
|
|
|
#if CONFIG_MPEG2_VAAPI_HWACCEL
|
|
|
|
HWACCEL_VAAPI(mpeg2),
|
|
|
|
#endif
|
|
|
|
#if CONFIG_MPEG2_VDPAU_HWACCEL
|
|
|
|
HWACCEL_VDPAU(mpeg2),
|
|
|
|
#endif
|
|
|
|
#if CONFIG_MPEG2_VIDEOTOOLBOX_HWACCEL
|
|
|
|
HWACCEL_VIDEOTOOLBOX(mpeg2),
|
|
|
|
#endif
|
|
|
|
NULL
|
|
|
|
},
|
2013-03-26 21:50:40 +03:00
|
|
|
};
|
|
|
|
|
2013-03-28 13:45:04 +03:00
|
|
|
//legacy decoder
|
2022-03-16 22:09:54 +02:00
|
|
|
const FFCodec ff_mpegvideo_decoder = {
|
|
|
|
.p.name = "mpegvideo",
|
2022-08-29 13:38:02 +02:00
|
|
|
CODEC_LONG_NAME("MPEG-1 video"),
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.type = AVMEDIA_TYPE_VIDEO,
|
|
|
|
.p.id = AV_CODEC_ID_MPEG2VIDEO,
|
2013-03-28 13:45:04 +03:00
|
|
|
.priv_data_size = sizeof(Mpeg1Context),
|
|
|
|
.init = mpeg_decode_init,
|
|
|
|
.close = mpeg_decode_end,
|
2022-03-30 23:28:24 +02:00
|
|
|
FF_CODEC_DECODE_CB(mpeg_decode_frame),
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
|
2021-03-08 12:26:32 +02:00
|
|
|
AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
|
2022-07-10 00:05:45 +02:00
|
|
|
.caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
|
2013-03-28 13:45:04 +03:00
|
|
|
.flush = flush,
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.max_lowres = 3,
|
2013-03-28 13:45:04 +03:00
|
|
|
};
|
2020-09-14 11:35:26 +02:00
|
|
|
|
|
|
|
typedef struct IPUContext {
|
|
|
|
MpegEncContext m;
|
|
|
|
|
|
|
|
int flags;
|
|
|
|
DECLARE_ALIGNED(32, int16_t, block)[6][64];
|
|
|
|
} IPUContext;
|
|
|
|
|
2022-03-30 21:33:24 +02:00
|
|
|
static int ipu_decode_frame(AVCodecContext *avctx, AVFrame *frame,
|
2020-09-14 11:35:26 +02:00
|
|
|
int *got_frame, AVPacket *avpkt)
|
|
|
|
{
|
|
|
|
IPUContext *s = avctx->priv_data;
|
|
|
|
MpegEncContext *m = &s->m;
|
|
|
|
GetBitContext *gb = &m->gb;
|
|
|
|
int ret;
|
|
|
|
|
2022-11-28 00:34:33 +02:00
|
|
|
// Check for minimal intra MB size (considering mb header, luma & chroma dc VLC, ac EOB VLC)
|
2024-05-11 21:04:00 +02:00
|
|
|
if (avpkt->size*8LL < (avctx->width+15)/16 * ((avctx->height+15)/16) * (2LL + 3*4 + 2*2 + 2*6))
|
2022-11-28 00:34:33 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
|
2020-09-14 11:35:26 +02:00
|
|
|
ret = ff_get_buffer(avctx, frame, 0);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
|
|
|
|
ret = init_get_bits8(gb, avpkt->data, avpkt->size);
|
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
|
|
|
|
s->flags = get_bits(gb, 8);
|
2020-09-25 22:05:03 +02:00
|
|
|
m->intra_dc_precision = s->flags & 3;
|
|
|
|
m->q_scale_type = !!(s->flags & 0x40);
|
|
|
|
m->intra_vlc_format = !!(s->flags & 0x20);
|
|
|
|
m->alternate_scan = !!(s->flags & 0x10);
|
|
|
|
|
2024-04-22 01:44:52 +02:00
|
|
|
ff_init_scantable(m->idsp.idct_permutation, &m->intra_scantable,
|
|
|
|
s->flags & 0x10 ? ff_alternate_vertical_scan : ff_zigzag_direct);
|
2020-09-25 22:05:03 +02:00
|
|
|
|
|
|
|
m->last_dc[0] = m->last_dc[1] = m->last_dc[2] = 1 << (7 + (s->flags & 3));
|
2020-09-14 11:35:26 +02:00
|
|
|
m->qscale = 1;
|
|
|
|
|
|
|
|
for (int y = 0; y < avctx->height; y += 16) {
|
|
|
|
int intraquant;
|
|
|
|
|
|
|
|
for (int x = 0; x < avctx->width; x += 16) {
|
|
|
|
if (x || y) {
|
|
|
|
if (!get_bits1(gb))
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
if (get_bits1(gb)) {
|
|
|
|
intraquant = 0;
|
|
|
|
} else {
|
|
|
|
if (!get_bits1(gb))
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
intraquant = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (s->flags & 4)
|
|
|
|
skip_bits1(gb);
|
|
|
|
|
|
|
|
if (intraquant)
|
|
|
|
m->qscale = mpeg_get_qscale(m);
|
|
|
|
|
|
|
|
memset(s->block, 0, sizeof(s->block));
|
|
|
|
|
|
|
|
for (int n = 0; n < 6; n++) {
|
2020-09-25 22:05:03 +02:00
|
|
|
if (s->flags & 0x80) {
|
2020-09-25 23:32:34 +02:00
|
|
|
ret = ff_mpeg1_decode_block_intra(&m->gb,
|
|
|
|
m->intra_matrix,
|
|
|
|
m->intra_scantable.permutated,
|
|
|
|
m->last_dc, s->block[n],
|
|
|
|
n, m->qscale);
|
2020-09-25 22:05:03 +02:00
|
|
|
} else {
|
2020-09-25 23:32:34 +02:00
|
|
|
ret = mpeg2_decode_block_intra(m, s->block[n], n);
|
2020-09-25 22:05:03 +02:00
|
|
|
}
|
|
|
|
|
2020-09-14 11:35:26 +02:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
m->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x,
|
|
|
|
frame->linesize[0], s->block[0]);
|
|
|
|
m->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x + 8,
|
|
|
|
frame->linesize[0], s->block[1]);
|
|
|
|
m->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x,
|
|
|
|
frame->linesize[0], s->block[2]);
|
|
|
|
m->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x + 8,
|
|
|
|
frame->linesize[0], s->block[3]);
|
|
|
|
m->idsp.idct_put(frame->data[1] + (y >> 1) * frame->linesize[1] + (x >> 1),
|
|
|
|
frame->linesize[1], s->block[4]);
|
|
|
|
m->idsp.idct_put(frame->data[2] + (y >> 1) * frame->linesize[2] + (x >> 1),
|
|
|
|
frame->linesize[2], s->block[5]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
align_get_bits(gb);
|
|
|
|
if (get_bits_left(gb) != 32)
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
|
|
|
|
*got_frame = 1;
|
|
|
|
|
|
|
|
return avpkt->size;
|
|
|
|
}
|
|
|
|
|
|
|
|
static av_cold int ipu_decode_init(AVCodecContext *avctx)
|
|
|
|
{
|
|
|
|
IPUContext *s = avctx->priv_data;
|
|
|
|
MpegEncContext *m = &s->m;
|
|
|
|
|
|
|
|
avctx->pix_fmt = AV_PIX_FMT_YUV420P;
|
2024-04-22 01:57:56 +02:00
|
|
|
m->avctx = avctx;
|
2020-09-14 11:35:26 +02:00
|
|
|
|
2024-04-22 01:57:56 +02:00
|
|
|
ff_idctdsp_init(&m->idsp, avctx);
|
2020-09-14 11:35:26 +02:00
|
|
|
ff_mpeg12_init_vlcs();
|
|
|
|
|
|
|
|
for (int i = 0; i < 64; i++) {
|
|
|
|
int j = m->idsp.idct_permutation[i];
|
|
|
|
int v = ff_mpeg1_default_intra_matrix[i];
|
|
|
|
m->intra_matrix[j] = v;
|
|
|
|
m->chroma_intra_matrix[j] = v;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2022-03-16 22:09:54 +02:00
|
|
|
const FFCodec ff_ipu_decoder = {
|
|
|
|
.p.name = "ipu",
|
2022-08-29 13:38:02 +02:00
|
|
|
CODEC_LONG_NAME("IPU Video"),
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.type = AVMEDIA_TYPE_VIDEO,
|
|
|
|
.p.id = AV_CODEC_ID_IPU,
|
2020-09-14 11:35:26 +02:00
|
|
|
.priv_data_size = sizeof(IPUContext),
|
|
|
|
.init = ipu_decode_init,
|
2022-03-30 23:28:24 +02:00
|
|
|
FF_CODEC_DECODE_CB(ipu_decode_frame),
|
2022-03-16 22:09:54 +02:00
|
|
|
.p.capabilities = AV_CODEC_CAP_DR1,
|
2020-09-14 11:35:26 +02:00
|
|
|
};
|