]> git.sesse.net Git - ffmpeg/blob - libavcodec/avfft.c
x86/jpeg2000dsp: add ff_ict_float_{fma3,fma4}
[ffmpeg] / libavcodec / avfft.c
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
28 FFTContext *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
38 void av_fft_permute(FFTContext *s, FFTComplex *z)
39 {
40     s->fft_permute(s, z);
41 }
42
43 void av_fft_calc(FFTContext *s, FFTComplex *z)
44 {
45     s->fft_calc(s, z);
46 }
47
48 av_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
58 FFTContext *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
68 void av_imdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input)
69 {
70     s->imdct_calc(s, output, input);
71 }
72
73 void av_imdct_half(FFTContext *s, FFTSample *output, const FFTSample *input)
74 {
75     s->imdct_half(s, output, input);
76 }
77
78 void av_mdct_calc(FFTContext *s, FFTSample *output, const FFTSample *input)
79 {
80     s->mdct_calc(s, output, input);
81 }
82
83 av_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
95 RDFTContext *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
105 void av_rdft_calc(RDFTContext *s, FFTSample *data)
106 {
107     s->rdft_calc(s, data);
108 }
109
110 av_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
122 DCTContext *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
132 void av_dct_calc(DCTContext *s, FFTSample *data)
133 {
134     s->dct_calc(s, data);
135 }
136
137 av_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 */