1/* 2 * This file is part of FFmpeg. 3 * 4 * FFmpeg is free software; you can redistribute it and/or 5 * modify it under the terms of the GNU Lesser General Public 6 * License as published by the Free Software Foundation; either 7 * version 2.1 of the License, or (at your option) any later version. 8 * 9 * FFmpeg is distributed in the hope that it will be useful, 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 12 * Lesser General Public License for more details. 13 * 14 * You should have received a copy of the GNU Lesser General Public 15 * License along with FFmpeg; if not, write to the Free Software 16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 17 */ 18 19#include "libavutil/attributes.h" 20#include "libavutil/mem.h" 21#include "avfft.h" 22#include "fft.h" 23#include "rdft.h" 24#include "dct.h" 25 26/* FFT */ 27 28FFTContext *av_fft_init(int nbits, int inverse) 29{ 30 FFTContext *s = av_mallocz(sizeof(*s)); 31 32 if (s && ff_fft_init(s, nbits, inverse)) 33 av_freep(&s); 34 35 return s; 36} 37 38void av_fft_permute(FFTContext *s, FFTComplex *z) 39{ 40 s->fft_permute(s, z); 41} 42 43void av_fft_calc(FFTContext *s, FFTComplex *z) 44{ 45 s->fft_calc(s, z); 46} 47 48av_cold void av_fft_end(FFTContext *s) 49{ 50 if (s) { 51 ff_fft_end(s); 52 av_free(s); 53 } 54} 55 56#if CONFIG_MDCT 57 58FFTContext *av_mdct_init(int nbits, int inverse, double scale) 59{ 60 FFTContext *s = av_malloc(sizeof(*s)); 61 62 if (s && ff_mdct_init(s, nbits, inverse, scale)) 63 av_freep(&s); 64 65 return s; 66} 67 68void av_imdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input) 69{ 70 s->imdct_calc(s, output, input); 71} 72 73void av_imdct_half(FFTContext *s, FFTSample *output, const FFTSample *input) 74{ 75 s->imdct_half(s, output, input); 76} 77 78void av_mdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input) 79{ 80 s->mdct_calc(s, output, input); 81} 82 83av_cold void av_mdct_end(FFTContext *s) 84{ 85 if (s) { 86 ff_mdct_end(s); 87 av_free(s); 88 } 89} 90 91#endif /* CONFIG_MDCT */ 92 93#if CONFIG_RDFT 94 95RDFTContext *av_rdft_init(int nbits, enum RDFTransformType trans) 96{ 97 RDFTContext *s = av_malloc(sizeof(*s)); 98 99 if (s && ff_rdft_init(s, nbits, trans)) 100 av_freep(&s); 101 102 return s; 103} 104 105void av_rdft_calc(RDFTContext *s, FFTSample *data) 106{ 107 s->rdft_calc(s, data); 108} 109 110av_cold void av_rdft_end(RDFTContext *s) 111{ 112 if (s) { 113 ff_rdft_end(s); 114 av_free(s); 115 } 116} 117 118#endif /* CONFIG_RDFT */ 119 120#if CONFIG_DCT 121 122DCTContext *av_dct_init(int nbits, enum DCTTransformType inverse) 123{ 124 DCTContext *s = av_malloc(sizeof(*s)); 125 126 if (s && ff_dct_init(s, nbits, inverse)) 127 av_freep(&s); 128 129 return s; 130} 131 132void av_dct_calc(DCTContext *s, FFTSample *data) 133{ 134 s->dct_calc(s, data); 135} 136 137av_cold void av_dct_end(DCTContext *s) 138{ 139 if (s) { 140 ff_dct_end(s); 141 av_free(s); 142 } 143} 144 145#endif /* CONFIG_DCT */ 146