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_fft

Browse Source
This commit is contained in:
Lynne
2022-11-10 11:23:38 +01:00
parent 4acd08be6c
commit dfcd4bbf51
1 changed files with 30 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
libavcodec/avfft.c
View File

@@ -18,38 +18,60 @@
#include "libavutil/attributes.h" #include "libavutil/attributes.h"
#include "libavutil/mem.h" #include "libavutil/mem.h"
#include "libavutil/tx.h"
#include "avfft.h" #include "avfft.h"
#include "fft.h" #include "fft.h"
#include "rdft.h" #include "rdft.h"
#include "dct.h" #include "dct.h"
typedef struct AVTXWrapper {
AVTXContext *ctx;
av_tx_fn fn;
AVTXContext *ctx2;
av_tx_fn fn2;
ptrdiff_t stride;
} AVTXWrapper;
/* FFT */ /* FFT */
FFTContext *av_fft_init(int nbits, int inverse) FFTContext *av_fft_init(int nbits, int inverse)
{ {
FFTContext *s = av_mallocz(sizeof(*s)); int ret;
float scale = 1.0f;
AVTXWrapper *s = av_malloc(sizeof(*s));
if (!s)
return NULL;
if (s && ff_fft_init(s, nbits, inverse)) ret = av_tx_init(&s->ctx, &s->fn, AV_TX_FLOAT_FFT, inverse, 1 << nbits,
av_freep(&s); &scale, AV_TX_INPLACE);
if (ret < 0) {
av_free(s);
return NULL;
}
return s; return (FFTContext *)s;
} }
void av_fft_permute(FFTContext *s, FFTComplex *z) void av_fft_permute(FFTContext *s, FFTComplex *z)
{ {
s->fft_permute(s, z); /* Empty */
} }
void av_fft_calc(FFTContext *s, FFTComplex *z) void av_fft_calc(FFTContext *s, FFTComplex *z)
{ {
s->fft_calc(s, z); AVTXWrapper *w = (AVTXWrapper *)s;
w->fn(w->ctx, z, (void *)z, sizeof(AVComplexFloat));
} }
av_cold void av_fft_end(FFTContext *s) av_cold void av_fft_end(FFTContext *s)
{ {
if (s) { if (s) {
ff_fft_end(s); AVTXWrapper *w = (AVTXWrapper *)s;
av_free(s); av_tx_uninit(&w->ctx);
av_tx_uninit(&w->ctx2);
av_free(w);
} }
} }