[FFmpeg-cvslog] avcodec/ac3enc: Avoid indirections, allocations of small arrays
Andreas Rheinhardt
git at videolan.org
Thu Apr 18 16:16:14 EEST 2024
ffmpeg | branch: master | Andreas Rheinhardt <andreas.rheinhardt at outlook.com> | Sun Apr 14 18:21:15 2024 +0200| [79d1814b719a0f7efe99850b1adee5d1503e27f5] | committer: Andreas Rheinhardt
avcodec/ac3enc: Avoid indirections, allocations of small arrays
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt at outlook.com>
> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=79d1814b719a0f7efe99850b1adee5d1503e27f5
---
libavcodec/ac3enc.c | 44 ++++----------------------------------------
libavcodec/ac3enc.h | 22 +++++++++++-----------
2 files changed, 15 insertions(+), 51 deletions(-)
diff --git a/libavcodec/ac3enc.c b/libavcodec/ac3enc.c
index a3a05b3ac8..1ef670622a 100644
--- a/libavcodec/ac3enc.c
+++ b/libavcodec/ac3enc.c
@@ -2180,13 +2180,10 @@ static void dprint_options(AC3EncodeContext *s)
*/
av_cold int ff_ac3_encode_close(AVCodecContext *avctx)
{
- int blk, ch;
AC3EncodeContext *s = avctx->priv_data;
- if (s->planar_samples)
- for (ch = 0; ch < s->channels; ch++)
- av_freep(&s->planar_samples[ch]);
- av_freep(&s->planar_samples);
+ for (int ch = 0; ch < s->channels; ch++)
+ av_freep(&s->planar_samples[ch]);
av_freep(&s->bap_buffer);
av_freep(&s->bap1_buffer);
av_freep(&s->mdct_coef_buffer);
@@ -2200,19 +2197,6 @@ av_cold int ff_ac3_encode_close(AVCodecContext *avctx)
av_freep(&s->cpl_coord_exp_buffer);
av_freep(&s->cpl_coord_mant_buffer);
av_freep(&s->fdsp);
- for (blk = 0; blk < s->num_blocks; blk++) {
- AC3Block *block = &s->blocks[blk];
- av_freep(&block->mdct_coef);
- av_freep(&block->fixed_coef);
- av_freep(&block->exp);
- av_freep(&block->grouped_exp);
- av_freep(&block->psd);
- av_freep(&block->band_psd);
- av_freep(&block->mask);
- av_freep(&block->qmant);
- av_freep(&block->cpl_coord_exp);
- av_freep(&block->cpl_coord_mant);
- }
av_tx_uninit(&s->tx);
@@ -2457,9 +2441,6 @@ static av_cold int allocate_buffers(AC3EncodeContext *s)
int total_coefs = AC3_MAX_COEFS * channel_blocks;
const unsigned sampletype_size = SAMPLETYPE_SIZE(s);
- if (!FF_ALLOCZ_TYPED_ARRAY(s->planar_samples, s->channels))
- return AVERROR(ENOMEM);
-
for (int ch = 0; ch < s->channels; ch++) {
s->planar_samples[ch] = av_mallocz((AC3_FRAME_SIZE + AC3_BLOCK_SIZE) *
sampletype_size);
@@ -2486,21 +2467,6 @@ static av_cold int allocate_buffers(AC3EncodeContext *s)
for (blk = 0; blk < s->num_blocks; blk++) {
AC3Block *block = &s->blocks[blk];
- if (!FF_ALLOCZ_TYPED_ARRAY(block->mdct_coef, channels) ||
- !FF_ALLOCZ_TYPED_ARRAY(block->exp, channels) ||
- !FF_ALLOCZ_TYPED_ARRAY(block->grouped_exp, channels) ||
- !FF_ALLOCZ_TYPED_ARRAY(block->psd, channels) ||
- !FF_ALLOCZ_TYPED_ARRAY(block->band_psd, channels) ||
- !FF_ALLOCZ_TYPED_ARRAY(block->mask, channels) ||
- !FF_ALLOCZ_TYPED_ARRAY(block->qmant, channels))
- return AVERROR(ENOMEM);
-
- if (s->cpl_enabled) {
- if (!FF_ALLOCZ_TYPED_ARRAY(block->cpl_coord_exp, channels) ||
- !FF_ALLOCZ_TYPED_ARRAY(block->cpl_coord_mant, channels))
- return AVERROR(ENOMEM);
- }
-
for (ch = 0; ch < channels; ch++) {
/* arrangement: block, channel, coeff */
block->grouped_exp[ch] = &s->grouped_exp_buffer[128 * (blk * channels + ch)];
@@ -2524,16 +2490,14 @@ static av_cold int allocate_buffers(AC3EncodeContext *s)
return AVERROR(ENOMEM);
for (blk = 0; blk < s->num_blocks; blk++) {
AC3Block *block = &s->blocks[blk];
- if (!FF_ALLOCZ_TYPED_ARRAY(block->fixed_coef, channels))
- return AVERROR(ENOMEM);
+
for (ch = 0; ch < channels; ch++)
block->fixed_coef[ch] = &s->fixed_coef_buffer[AC3_MAX_COEFS * (s->num_blocks * ch + blk)];
}
} else {
for (blk = 0; blk < s->num_blocks; blk++) {
AC3Block *block = &s->blocks[blk];
- if (!FF_ALLOCZ_TYPED_ARRAY(block->fixed_coef, channels))
- return AVERROR(ENOMEM);
+
for (ch = 0; ch < channels; ch++)
block->fixed_coef[ch] = (int32_t *)block->mdct_coef[ch];
}
diff --git a/libavcodec/ac3enc.h b/libavcodec/ac3enc.h
index 41b9a3a20b..54e14d43d9 100644
--- a/libavcodec/ac3enc.h
+++ b/libavcodec/ac3enc.h
@@ -127,16 +127,16 @@ typedef struct AC3EncOptions {
* Data for a single audio block.
*/
typedef struct AC3Block {
- CoefType **mdct_coef; ///< MDCT coefficients
- int32_t **fixed_coef; ///< fixed-point MDCT coefficients
- uint8_t **exp; ///< original exponents
- uint8_t **grouped_exp; ///< grouped exponents
- int16_t **psd; ///< psd per frequency bin
- int16_t **band_psd; ///< psd per critical band
- int16_t **mask; ///< masking curve
- uint16_t **qmant; ///< quantized mantissas
- uint8_t **cpl_coord_exp; ///< coupling coord exponents (cplcoexp)
- uint8_t **cpl_coord_mant; ///< coupling coord mantissas (cplcomant)
+ CoefType *mdct_coef[AC3_MAX_CHANNELS]; ///< MDCT coefficients
+ int32_t *fixed_coef[AC3_MAX_CHANNELS]; ///< fixed-point MDCT coefficients
+ uint8_t *exp[AC3_MAX_CHANNELS]; ///< original exponents
+ uint8_t *grouped_exp[AC3_MAX_CHANNELS]; ///< grouped exponents
+ int16_t *psd[AC3_MAX_CHANNELS]; ///< psd per frequency bin
+ int16_t *band_psd[AC3_MAX_CHANNELS]; ///< psd per critical band
+ int16_t *mask[AC3_MAX_CHANNELS]; ///< masking curve
+ uint16_t *qmant[AC3_MAX_CHANNELS]; ///< quantized mantissas
+ uint8_t *cpl_coord_exp[AC3_MAX_CHANNELS]; ///< coupling coord exponents (cplcoexp)
+ uint8_t *cpl_coord_mant[AC3_MAX_CHANNELS]; ///< coupling coord mantissas (cplcomant)
uint8_t new_rematrixing_strategy; ///< send new rematrixing flags in this block
int num_rematrixing_bands; ///< number of rematrixing bands
uint8_t rematrixing_flags[4]; ///< rematrixing flags
@@ -232,7 +232,7 @@ typedef struct AC3EncodeContext {
int frame_bits; ///< all frame bits except exponents and mantissas
int exponent_bits; ///< number of bits used for exponents
- uint8_t **planar_samples;
+ uint8_t *planar_samples[AC3_MAX_CHANNELS - 1];
uint8_t *bap_buffer;
uint8_t *bap1_buffer;
CoefType *mdct_coef_buffer;
More information about the ffmpeg-cvslog
mailing list