virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-08-15 14:13:16 +02:00
Code Issues Releases Activity

avfft: wrap lavu/tx instead of ff_mdct

Browse Source
This commit is contained in:
Lynne
2022-11-10 11:26:33 +01:00
parent dfcd4bbf51
commit 517e4fcca6
1 changed files with 30 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
libavcodec/avfft.c
View File

@@ -75,43 +75,60 @@ av_cold void av_fft_end(FFTContext *s)
} }
} }
#if CONFIG_MDCT
FFTContext *av_mdct_init(int nbits, int inverse, double scale) FFTContext *av_mdct_init(int nbits, int inverse, double scale)
{ {
FFTContext *s = av_malloc(sizeof(*s)); int ret;
float scale_f = scale;
AVTXWrapper *s = av_malloc(sizeof(*s));
if (!s)
return NULL;
if (s && ff_mdct_init(s, nbits, inverse, scale)) ret = av_tx_init(&s->ctx, &s->fn, AV_TX_FLOAT_MDCT, inverse, 1 << (nbits - 1), &scale_f, 0);
av_freep(&s); if (ret < 0) {
av_free(s);
return NULL;
}
return s; if (inverse) {
ret = av_tx_init(&s->ctx2, &s->fn2, AV_TX_FLOAT_MDCT, inverse, 1 << (nbits - 1),
&scale_f, AV_TX_FULL_IMDCT);
if (ret < 0) {
av_tx_uninit(&s->ctx);
av_free(s);
return NULL;
}
}
return (FFTContext *)s;
} }
void av_imdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input) void av_imdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input)
{ {
s->imdct_calc(s, output, input); AVTXWrapper *w = (AVTXWrapper *)s;
w->fn2(w->ctx2, output, (void *)input, sizeof(float));
} }
void av_imdct_half(FFTContext *s, FFTSample *output, const FFTSample *input) void av_imdct_half(FFTContext *s, FFTSample *output, const FFTSample *input)
{ {
s->imdct_half(s, output, input); AVTXWrapper *w = (AVTXWrapper *)s;
w->fn(w->ctx, output, (void *)input, sizeof(float));
} }
void av_mdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input) void av_mdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input)
{ {
s->mdct_calc(s, output, input); AVTXWrapper *w = (AVTXWrapper *)s;
w->fn(w->ctx, output, (void *)input, sizeof(float));
} }
av_cold void av_mdct_end(FFTContext *s) av_cold void av_mdct_end(FFTContext *s)
{ {
if (s) { if (s) {
ff_mdct_end(s); AVTXWrapper *w = (AVTXWrapper *)s;
av_free(s); av_tx_uninit(&w->ctx);
av_free(w);
} }
} }
#endif /* CONFIG_MDCT */
#if CONFIG_RDFT #if CONFIG_RDFT
RDFTContext *av_rdft_init(int nbits, enum RDFTransformType trans) RDFTContext *av_rdft_init(int nbits, enum RDFTransformType trans)