From 4ad73f29065051c68991eb96aeae7f771039209a Mon Sep 17 00:00:00 2001
From: Lynne <dev@lynne.ee>
Date: Sat, 24 Sep 2022 01:08:00 +0200
Subject: [PATCH 6/6] wmaprodec: convert to lavu/tx
---
libavcodec/wmaprodec.c | 24 ++++++++++++++----------
1 file changed, 14 insertions(+), 10 deletions(-)
@@ -89,6 +89,7 @@
#include <inttypes.h>
#include "libavutil/audio_fifo.h"
+#include "libavutil/tx.h"
#include "libavutil/ffmath.h"
#include "libavutil/float_dsp.h"
#include "libavutil/intfloat.h"
@@ -185,7 +186,8 @@ typedef struct WMAProDecodeCtx {
uint8_t frame_data[MAX_FRAMESIZE +
AV_INPUT_BUFFER_PADDING_SIZE];///< compressed frame data
PutBitContext pb; ///< context for filling the frame_data buffer
- FFTContext mdct_ctx[WMAPRO_BLOCK_SIZES]; ///< MDCT context per block size
+ AVTXContext *tx[WMAPRO_BLOCK_SIZES]; ///< MDCT context per block size
+ av_tx_fn tx_fn[WMAPRO_BLOCK_SIZES];
DECLARE_ALIGNED(32, float, tmp)[WMAPRO_BLOCK_MAX_SIZE]; ///< IMDCT output buffer
const float* windows[WMAPRO_BLOCK_SIZES]; ///< windows for the different block sizes
@@ -287,7 +289,7 @@ static av_cold int decode_end(WMAProDecodeCtx *s)
av_freep(&s->fdsp);
for (i = 0; i < WMAPRO_BLOCK_SIZES; i++)
- ff_mdct_end(&s->mdct_ctx[i]);
+ av_tx_uninit(&s->tx[i]);
return 0;
}
@@ -552,12 +554,13 @@ static av_cold int decode_init(WMAProDecodeCtx *s, AVCodecContext *avctx, int nu
return AVERROR(ENOMEM);
/** init MDCT, FIXME: only init needed sizes */
- for (int i = 0; i < WMAPRO_BLOCK_SIZES; i++) {
- ret = ff_mdct_init(&s->mdct_ctx[i], WMAPRO_BLOCK_MIN_BITS + 1 + i, 1,
- 1.0 / (1 << (WMAPRO_BLOCK_MIN_BITS + i - 1))
- / (1ll << (s->bits_per_sample - 1)));
- if (ret < 0)
- return ret;
+ for (i = 0; i < WMAPRO_BLOCK_SIZES; i++) {
+ const float scale = 1.0 / (1 << (WMAPRO_BLOCK_MIN_BITS + i - 1))
+ / (1ll << (s->bits_per_sample - 1));
+ int err = av_tx_init(&s->tx[i], &s->tx_fn[i], AV_TX_FLOAT_MDCT, 1,
+ 1 << (WMAPRO_BLOCK_MIN_BITS + i), &scale, 0);
+ if (err < 0)
+ return err;
}
/** init MDCT windows: simple sine window */
@@ -1386,7 +1389,8 @@ static int decode_subframe(WMAProDecodeCtx *s)
get_bits_count(&s->gb) - s->subframe_offset);
if (transmit_coeffs) {
- FFTContext *mdct = &s->mdct_ctx[av_log2(subframe_len) - WMAPRO_BLOCK_MIN_BITS];
+ AVTXContext *tx = s->tx[av_log2(subframe_len) - WMAPRO_BLOCK_MIN_BITS];
+ av_tx_fn tx_fn = s->tx_fn[av_log2(subframe_len) - WMAPRO_BLOCK_MIN_BITS];
/** reconstruct the per channel data */
inverse_channel_transform(s);
for (i = 0; i < s->channels_for_cur_subframe; i++) {
@@ -1412,7 +1416,7 @@ static int decode_subframe(WMAProDecodeCtx *s)
}
/** apply imdct (imdct_half == DCTIV with reverse) */
- mdct->imdct_half(mdct, s->channel[c].coeffs, s->tmp);
+ tx_fn(tx, s->channel[c].coeffs, s->tmp, sizeof(float));
}
}
--
2.37.2.609.g9ff673ca1a