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