avcodec/mpegvideo: Only allocate coded_block when needed

It is only needed for msmpeg4v3, wmv1, wmv2 and VC-1.

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
Andreas Rheinhardt 2024-04-20 22:56:44 +02:00
parent 1a5c21daee
commit ba033acb56
1 changed files with 8 additions and 3 deletions

View File

@ -596,11 +596,16 @@ int ff_mpv_init_context_frame(MpegEncContext *s)
}
if (s->out_format == FMT_H263) {
/* cbp values, cbp, ac_pred, pred_dir */
if (!(s->coded_block_base = av_mallocz(y_size + (s->mb_height&1)*2*s->b8_stride)) ||
!(s->cbp_table = av_mallocz(mb_array_size)) ||
/* cbp, ac_pred, pred_dir */
if (!(s->cbp_table = av_mallocz(mb_array_size)) ||
!(s->pred_dir_table = av_mallocz(mb_array_size)))
return AVERROR(ENOMEM);
}
if (s->msmpeg4_version >= 3) {
s->coded_block_base = av_mallocz(y_size + (s->mb_height&1)*2*s->b8_stride);
if (!s->coded_block_base)
return AVERROR(ENOMEM);
s->coded_block = s->coded_block_base + s->b8_stride + 1;
}