It is only used by the MPEG-4 decoder, yet it is used in the common H.263-code (i.e. in libavcodec/h263dec.c) and can therefore not be put in Mpeg4DecContext. Yet moving it to H263DecContext is possible.
Given that this is nevertheless an MPEG-4 only field, it is kept in sync in the MPEG-4-decoder's update_thread_context and not in a (currently inexisting) update_thread_context for H.263 decoders. Signed-off-by: Andreas Rheinhardt <andreas.rheinha...@outlook.com> --- libavcodec/h263dec.c | 27 ++++++++------- libavcodec/h263dec.h | 6 ++++ libavcodec/mpeg4videodec.c | 70 +++++++++++++++++++++++++++----------- libavcodec/mpegvideo.c | 5 --- libavcodec/mpegvideo.h | 6 ---- libavcodec/mpegvideo_dec.c | 24 ------------- libavcodec/nvdec_mpeg4.c | 2 +- 7 files changed, 73 insertions(+), 67 deletions(-) diff --git a/libavcodec/h263dec.c b/libavcodec/h263dec.c index db03e04143..3b0ebbdf9b 100644 --- a/libavcodec/h263dec.c +++ b/libavcodec/h263dec.c @@ -160,9 +160,12 @@ av_cold int ff_h263_decode_init(AVCodecContext *avctx) av_cold int ff_h263_decode_end(AVCodecContext *avctx) { - MPVMainDecContext *const s = avctx->priv_data; + H263DecContext *const h263 = avctx->priv_data; - ff_mpv_common_end(s); + av_freep(&h263->bitstream_buffer); + h263->allocated_bitstream_buffer_size = 0; + + ff_mpv_common_end(&h263->m); return 0; } @@ -174,7 +177,7 @@ static int get_consumed_bytes(H263DecContext *h, int buf_size) MPVDecContext *const s = &h->m.s; int pos = (get_bits_count(&s->gb) + 7) >> 3; - if (s->divx_packed || s->avctx->hwaccel) { + if (h->divx_packed || s->avctx->hwaccel) { /* We would have to scan through the whole buf to handle the weird * reordering ... */ return buf_size; @@ -481,26 +484,26 @@ int ff_h263_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, #endif retry: - if (s->divx_packed && s->bitstream_buffer_size) { + if (h->divx_packed && h->bitstream_buffer_size) { int i; for(i=0; i < buf_size-3; i++) { if (buf[i]==0 && buf[i+1]==0 && buf[i+2]==1) { if (buf[i+3]==0xB0) { av_log(s->avctx, AV_LOG_WARNING, "Discarding excessive bitstream in packed xvid\n"); - s->bitstream_buffer_size = 0; + h->bitstream_buffer_size = 0; } break; } } } - if (s->bitstream_buffer_size && (s->divx_packed || buf_size <= MAX_NVOP_SIZE)) // divx 5.01+/xvid frame reorder - ret = init_get_bits8(&s->gb, s->bitstream_buffer, - s->bitstream_buffer_size); + if (h->bitstream_buffer_size && (h->divx_packed || buf_size <= MAX_NVOP_SIZE)) // divx 5.01+/xvid frame reorder + ret = init_get_bits8(&s->gb, h->bitstream_buffer, + h->bitstream_buffer_size); else ret = init_get_bits8(&s->gb, buf, buf_size); - s->bitstream_buffer_size = 0; + h->bitstream_buffer_size = 0; if (ret < 0) return ret; @@ -625,7 +628,7 @@ retry: if ((ret = ff_mpv_frame_start(m, avctx)) < 0) return ret; - if (!s->divx_packed && !avctx->hwaccel) + if (!h->divx_packed && !avctx->hwaccel) ff_thread_finish_setup(avctx); if (avctx->hwaccel) { @@ -679,7 +682,7 @@ retry: ff_msmpeg4_decode_ext_header(h, buf_size) < 0) s->er.error_status_table[s->mb_num - 1] = ER_MB_ERROR; - av_assert1(s->bitstream_buffer_size == 0); + av_assert1(h->bitstream_buffer_size == 0); frame_end: if (!s->studio_profile) ff_er_frame_end(&s->er); @@ -695,7 +698,7 @@ frame_end: if (CONFIG_MPEG4_DECODER && avctx->codec_id == AV_CODEC_ID_MPEG4) ff_mpeg4_frame_end(avctx, buf, buf_size); - if (!s->divx_packed && avctx->hwaccel) + if (!h->divx_packed && avctx->hwaccel) ff_thread_finish_setup(avctx); av_assert1(s->current_picture.f->pict_type == s->current_picture_ptr->f->pict_type); diff --git a/libavcodec/h263dec.h b/libavcodec/h263dec.h index 7eebffc91c..04c238d90a 100644 --- a/libavcodec/h263dec.h +++ b/libavcodec/h263dec.h @@ -47,6 +47,12 @@ typedef struct H263DecContext { int pb_frame; ///< PB-frame mode (0 = none, 1 = base, 2 = improved) int long_vectors; ///< use horrible H.263v1 long vector mode int ehc_mode; + + /* divx specific, used to workaround (many) bugs in divx5 */ + int divx_packed; + uint8_t *bitstream_buffer; //Divx 5.01 puts several frames in a single one, this is used to reorder them + int bitstream_buffer_size; + unsigned int allocated_bitstream_buffer_size; } H263DecContext; int ff_h263_decode_motion(MPVDecContext *s, int pred, int f_code); diff --git a/libavcodec/mpeg4videodec.c b/libavcodec/mpeg4videodec.c index 7e3b672312..2d1463d564 100644 --- a/libavcodec/mpeg4videodec.c +++ b/libavcodec/mpeg4videodec.c @@ -2774,7 +2774,7 @@ static int decode_user_data(Mpeg4DecContext *ctx, GetBitContext *gb) if (e >= 2) { ctx->divx_version = ver; ctx->divx_build = build; - s->divx_packed = e == 3 && last == 'p'; + h->divx_packed = e == 3 && last == 'p'; } /* libavcodec detection */ @@ -2811,7 +2811,8 @@ static int decode_user_data(Mpeg4DecContext *ctx, GetBitContext *gb) int ff_mpeg4_workaround_bugs(AVCodecContext *avctx) { Mpeg4DecContext *ctx = avctx->priv_data; - MPVMainDecContext *const m = &ctx->h.m; + H263DecContext *const h = &ctx->h; + MPVMainDecContext *const m = &h->m; MPVDecContext *const s = &m->s; if (ctx->xvid_build == -1 && ctx->divx_version == -1 && ctx->lavc_build == -1) { @@ -2914,7 +2915,7 @@ int ff_mpeg4_workaround_bugs(AVCodecContext *avctx) av_log(s->avctx, AV_LOG_DEBUG, "bugs: %X lavc_build:%d xvid_build:%d divx_version:%d divx_build:%d %s\n", s->workaround_bugs, ctx->lavc_build, ctx->xvid_build, - ctx->divx_version, ctx->divx_build, s->divx_packed ? "p" : ""); + ctx->divx_version, ctx->divx_build, h->divx_packed ? "p" : ""); if (CONFIG_MPEG4_DECODER && ctx->xvid_build >= 0 && s->codec_id == AV_CODEC_ID_MPEG4 && @@ -3486,8 +3487,8 @@ int ff_mpeg4_frame_end(AVCodecContext *avctx, const uint8_t *buf, int buf_size) /* divx 5.01+ bitstream reorder stuff */ /* Since this clobbers the input buffer and hwaccel codecs still need the * data during hwaccel->end_frame we should not do this any earlier */ - if (s->divx_packed) { - int current_pos = s->gb.buffer == s->bitstream_buffer ? 0 : (get_bits_count(&s->gb) >> 3); + if (h->divx_packed) { + int current_pos = s->gb.buffer == h->bitstream_buffer ? 0 : (get_bits_count(&s->gb) >> 3); int startcode_found = 0; if (buf_size - current_pos > 7) { @@ -3511,16 +3512,16 @@ int ff_mpeg4_frame_end(AVCodecContext *avctx, const uint8_t *buf, int buf_size) "Consider using the mpeg4_unpack_bframes bitstream filter without encoding but stream copy to fix it.\n"); ctx->showed_packed_warning = 1; } - av_fast_padded_malloc(&s->bitstream_buffer, - &s->allocated_bitstream_buffer_size, + av_fast_padded_malloc(&h->bitstream_buffer, + &h->allocated_bitstream_buffer_size, buf_size - current_pos); - if (!s->bitstream_buffer) { - s->bitstream_buffer_size = 0; + if (!h->bitstream_buffer) { + h->bitstream_buffer_size = 0; return AVERROR(ENOMEM); } - memcpy(s->bitstream_buffer, buf + current_pos, + memcpy(h->bitstream_buffer, buf + current_pos, buf_size - current_pos); - s->bitstream_buffer_size = buf_size - current_pos; + h->bitstream_buffer_size = buf_size - current_pos; } } @@ -3532,7 +3533,9 @@ static int mpeg4_update_thread_context(AVCodecContext *dst, const AVCodecContext *src) { Mpeg4DecContext *s = dst->priv_data; + H263DecContext *const h = &s->h; const Mpeg4DecContext *s1 = src->priv_data; + const H263DecContext *const h1 = &s1->h; int init = s->h.m.context_initialized; int ret = ff_mpeg_update_thread_context(dst, src); @@ -3565,6 +3568,26 @@ static int mpeg4_update_thread_context(AVCodecContext *dst, s->cplx_estimation_trash_b = s1->cplx_estimation_trash_b; s->rgb = s1->rgb; + // DivX handling (doesn't work) + h->divx_packed = h1->divx_packed; + if (h1->bitstream_buffer) { + if (h1->bitstream_buffer_size + + AV_INPUT_BUFFER_PADDING_SIZE > h->allocated_bitstream_buffer_size) { + av_fast_malloc(&h->bitstream_buffer, + &h->allocated_bitstream_buffer_size, + h1->allocated_bitstream_buffer_size); + if (!h->bitstream_buffer) { + h->bitstream_buffer_size = 0; + return AVERROR(ENOMEM); + } + } + h->bitstream_buffer_size = h1->bitstream_buffer_size; + memcpy(h->bitstream_buffer, h1->bitstream_buffer, + h1->bitstream_buffer_size); + memset(h->bitstream_buffer + h->bitstream_buffer_size, 0, + AV_INPUT_BUFFER_PADDING_SIZE); + } + memcpy(s->sprite_shift, s1->sprite_shift, sizeof(s1->sprite_shift)); memcpy(s->sprite_traj, s1->sprite_traj, sizeof(s1->sprite_traj)); @@ -3577,16 +3600,25 @@ static int mpeg4_update_thread_context(AVCodecContext *dst, static int mpeg4_update_thread_context_for_user(AVCodecContext *dst, const AVCodecContext *src) { - MPVMainDecContext *const m = dst->priv_data; - const MPVMainDecContext *const m1 = src->priv_data; + Mpeg4DecContext *const ctx = dst->priv_data; + const Mpeg4DecContext *const ctx1 = src->priv_data; - m->s.quarter_sample = m1->s.quarter_sample; - m->s.divx_packed = m1->s.divx_packed; + ctx->h.m.s.quarter_sample = ctx1->h.m.s.quarter_sample; + ctx->h.divx_packed = ctx1->h.divx_packed; return 0; } #endif +static void mpeg4_flush(AVCodecContext *avctx) +{ + Mpeg4DecContext *const ctx = avctx->priv_data; + H263DecContext *const h = &ctx->h; + + h->bitstream_buffer_size = 0; + ff_mpeg_flush(avctx); +} + static av_cold void mpeg4_init_static(void) { static uint8_t mpeg4_rvlc_rl_tables[2][2][2 * MAX_RUN + MAX_LEVEL + 3]; @@ -3663,11 +3695,11 @@ static av_cold int decode_init(AVCodecContext *avctx) return 0; } -#define OFFSET(x) offsetof(MPVMainDecContext, s.x) +#define OFFSET(x) offsetof(Mpeg4DecContext, x) #define FLAGS AV_OPT_FLAG_EXPORT | AV_OPT_FLAG_READONLY static const AVOption mpeg4_options[] = { - {"quarter_sample", "1/4 subpel MC", OFFSET(quarter_sample), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, FLAGS}, - {"divx_packed", "divx style packed b frames", OFFSET(divx_packed), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, FLAGS}, + {"quarter_sample", "1/4 subpel MC", OFFSET(h.m.s.quarter_sample), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, FLAGS}, + {"divx_packed", "divx style packed b frames", OFFSET(h.divx_packed), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, FLAGS}, {NULL} }; @@ -3694,7 +3726,7 @@ const AVCodec ff_mpeg4_decoder = { AV_CODEC_CAP_DELAY | AV_CODEC_CAP_FRAME_THREADS, .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM | FF_CODEC_CAP_ALLOCATE_PROGRESS, - .flush = ff_mpeg_flush, + .flush = mpeg4_flush, .max_lowres = 3, .pix_fmts = ff_h263_hwaccel_pixfmt_list_420, .profiles = NULL_IF_CONFIG_SMALL(ff_mpeg4_video_profiles), diff --git a/libavcodec/mpegvideo.c b/libavcodec/mpegvideo.c index d0fbdaf1b4..1ee8e7458b 100644 --- a/libavcodec/mpegvideo.c +++ b/libavcodec/mpegvideo.c @@ -643,8 +643,6 @@ static void clear_context(MPVMainContext *m) s->sc.obmc_scratchpad = NULL; - s->bitstream_buffer = NULL; - s->allocated_bitstream_buffer_size = 0; for (int i = 0; i < 2; i++) for (int j = 0; j < 2; j++) s->p_field_mv_table[i][j] = NULL; @@ -792,9 +790,6 @@ void ff_mpv_common_end(MPVMainContext *m) s->parse_context.buffer_size = 0; #endif - av_freep(&s->bitstream_buffer); - s->allocated_bitstream_buffer_size = 0; - if (!s->avctx) return; diff --git a/libavcodec/mpegvideo.h b/libavcodec/mpegvideo.h index 8697f0108c..2b5267f1c3 100644 --- a/libavcodec/mpegvideo.h +++ b/libavcodec/mpegvideo.h @@ -339,12 +339,6 @@ typedef struct MPVContext { int mpeg_quant; int padding_bug_score; ///< used to detect the VERY common padding bug in MPEG-4 - /* divx specific, used to workaround (many) bugs in divx5 */ - int divx_packed; - uint8_t *bitstream_buffer; //Divx 5.01 puts several frames in a single one, this is used to reorder them - int bitstream_buffer_size; - unsigned int allocated_bitstream_buffer_size; - /* RV10 specific */ int rv10_version; ///< RV10 version: 0 or 3 int rv10_first_dc_coded[3]; diff --git a/libavcodec/mpegvideo_dec.c b/libavcodec/mpegvideo_dec.c index 8f3c401aff..7bbae2fa47 100644 --- a/libavcodec/mpegvideo_dec.c +++ b/libavcodec/mpegvideo_dec.c @@ -77,8 +77,6 @@ int ff_mpeg_update_thread_context(AVCodecContext *dst, s->current_picture_ptr = NULL; s->next_picture_ptr = NULL; s->last_picture_ptr = NULL; - s->bitstream_buffer = NULL; - s->bitstream_buffer_size = s->allocated_bitstream_buffer_size = 0; if (m1->context_initialized) { // s->picture_range_start += MAX_PICTURE_COUNT; @@ -154,27 +152,6 @@ do {\ s->low_delay = s1->low_delay; s->droppable = s1->droppable; - // DivX handling (doesn't work) - s->divx_packed = s1->divx_packed; - - if (s1->bitstream_buffer) { - if (s1->bitstream_buffer_size + - AV_INPUT_BUFFER_PADDING_SIZE > s->allocated_bitstream_buffer_size) { - av_fast_malloc(&s->bitstream_buffer, - &s->allocated_bitstream_buffer_size, - s1->allocated_bitstream_buffer_size); - if (!s->bitstream_buffer) { - s->bitstream_buffer_size = 0; - return AVERROR(ENOMEM); - } - } - s->bitstream_buffer_size = s1->bitstream_buffer_size; - memcpy(s->bitstream_buffer, s1->bitstream_buffer, - s1->bitstream_buffer_size); - memset(s->bitstream_buffer + s->bitstream_buffer_size, 0, - AV_INPUT_BUFFER_PADDING_SIZE); - } - // linesize-dependent scratch buffer allocation if (!s->sc.edge_emu_buffer) if (s1->linesize) { @@ -585,7 +562,6 @@ void ff_mpeg_flush(AVCodecContext *avctx) s->parse_context.index = 0; s->parse_context.last_index = 0; #endif - s->bitstream_buffer_size = 0; s->pp_time = 0; } diff --git a/libavcodec/nvdec_mpeg4.c b/libavcodec/nvdec_mpeg4.c index 6d5ac8cd07..61e9900c32 100644 --- a/libavcodec/nvdec_mpeg4.c +++ b/libavcodec/nvdec_mpeg4.c @@ -71,7 +71,7 @@ static int nvdec_mpeg4_start_frame(AVCodecContext *avctx, const uint8_t *buffer, .quant_type = s->mpeg_quant, .quarter_sample = s->quarter_sample, .short_video_header = avctx->codec->id == AV_CODEC_ID_H263, - .divx_flags = s->divx_packed ? 5 : 0, + .divx_flags = m->h.divx_packed ? 5 : 0, .vop_coding_type = s->pict_type - AV_PICTURE_TYPE_I, .vop_coded = 1, -- 2.32.0 _______________________________________________ ffmpeg-devel mailing list ffmpeg-devel@ffmpeg.org https://ffmpeg.org/mailman/listinfo/ffmpeg-devel To unsubscribe, visit link above, or email ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".