Mercurial > libavcodec.hg
annotate mdct.c @ 11557:53822d92c3f7 libavcodec
Make sure the EC code does not attempt to use inter based concealment if there
is no reference frame available. (this can happen because the EC code will attempt
to use reference frames even for I/IDR frames)
author | michael |
---|---|
date | Tue, 30 Mar 2010 20:46:46 +0000 |
parents | 6f1697664bf2 |
children | 7dd2a45249a9 |
rev | line source |
---|---|
781 | 1 /* |
2 * MDCT/IMDCT transforms | |
8629
04423b2f6e0b
cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents:
8567
diff
changeset
|
3 * Copyright (c) 2002 Fabrice Bellard |
781 | 4 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
5 * This file is part of FFmpeg. |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
6 * |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
781 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * License as published by the Free Software Foundation; either | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
781 | 11 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
781 | 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
3036
0b546eab515d
Update licensing information: The FSF changed postal address.
diego
parents:
2967
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
781 | 20 */ |
11370 | 21 |
11444
6f1697664bf2
Replace many includes of libavutil/common.h with what is actually needed
mru
parents:
11370
diff
changeset
|
22 #include <stdlib.h> |
6f1697664bf2
Replace many includes of libavutil/common.h with what is actually needed
mru
parents:
11370
diff
changeset
|
23 #include <string.h> |
6f1697664bf2
Replace many includes of libavutil/common.h with what is actually needed
mru
parents:
11370
diff
changeset
|
24 #include "libavutil/common.h" |
11370 | 25 #include "libavutil/mathematics.h" |
26 #include "fft.h" | |
781 | 27 |
1106 | 28 /** |
8718
e9d9d946f213
Use full internal pathname in doxygen @file directives.
diego
parents:
8629
diff
changeset
|
29 * @file libavcodec/mdct.c |
1106 | 30 * MDCT/IMDCT transforms. |
31 */ | |
32 | |
6139
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
33 // Generate a Kaiser-Bessel Derived Window. |
6142
a35b838ab955
Add variable alpha and size of half window for Kaiser-Bessel Derived window
superdump
parents:
6139
diff
changeset
|
34 #define BESSEL_I0_ITER 50 // default: 50 iterations of Bessel I0 approximation |
8737
eeca2fc122f8
Add av_cold attributes to *_init and *_end functions.
alexc
parents:
8718
diff
changeset
|
35 av_cold void ff_kbd_window_init(float *window, float alpha, int n) |
6139
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
36 { |
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
37 int i, j; |
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
38 double sum = 0.0, bessel, tmp; |
6142
a35b838ab955
Add variable alpha and size of half window for Kaiser-Bessel Derived window
superdump
parents:
6139
diff
changeset
|
39 double local_window[n]; |
a35b838ab955
Add variable alpha and size of half window for Kaiser-Bessel Derived window
superdump
parents:
6139
diff
changeset
|
40 double alpha2 = (alpha * M_PI / n) * (alpha * M_PI / n); |
6139
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
41 |
6142
a35b838ab955
Add variable alpha and size of half window for Kaiser-Bessel Derived window
superdump
parents:
6139
diff
changeset
|
42 for (i = 0; i < n; i++) { |
a35b838ab955
Add variable alpha and size of half window for Kaiser-Bessel Derived window
superdump
parents:
6139
diff
changeset
|
43 tmp = i * (n - i) * alpha2; |
6139
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
44 bessel = 1.0; |
6142
a35b838ab955
Add variable alpha and size of half window for Kaiser-Bessel Derived window
superdump
parents:
6139
diff
changeset
|
45 for (j = BESSEL_I0_ITER; j > 0; j--) |
6139
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
46 bessel = bessel * tmp / (j * j) + 1; |
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
47 sum += bessel; |
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
48 local_window[i] = sum; |
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
49 } |
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
50 |
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
51 sum++; |
6142
a35b838ab955
Add variable alpha and size of half window for Kaiser-Bessel Derived window
superdump
parents:
6139
diff
changeset
|
52 for (i = 0; i < n; i++) |
6139
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
53 window[i] = sqrt(local_window[i] / sum); |
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
54 } |
5077d1562573
Make the Kaiser-Bessel window generator a common function
andoma
parents:
3947
diff
changeset
|
55 |
10827
3d011a01a6a0
Add support for hard-coded MDCT-related ff_sine_windows tables.
reimar
parents:
10204
diff
changeset
|
56 #include "mdct_tablegen.h" |
7094
b0820b8bd4dd
Add generic ff_sine_window_init function and implement in codecs appropriately
superdump
parents:
6498
diff
changeset
|
57 |
1106 | 58 /** |
59 * init MDCT or IMDCT computation. | |
781 | 60 */ |
10199 | 61 av_cold int ff_mdct_init(FFTContext *s, int nbits, int inverse, double scale) |
781 | 62 { |
63 int n, n4, i; | |
9658
67a20f0eb42c
Support for getting (i)MDCT output multiplied by a constant scaling factor.
serge
parents:
9480
diff
changeset
|
64 double alpha, theta; |
10204
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
65 int tstep; |
781 | 66 |
67 memset(s, 0, sizeof(*s)); | |
68 n = 1 << nbits; | |
10199 | 69 s->mdct_bits = nbits; |
70 s->mdct_size = n; | |
781 | 71 n4 = n >> 2; |
10204
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
72 s->permutation = FF_MDCT_PERM_NONE; |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
73 |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
74 if (ff_fft_init(s, s->mdct_bits - 2, inverse) < 0) |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
75 goto fail; |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
76 |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
77 s->tcos = av_malloc(n/2 * sizeof(FFTSample)); |
781 | 78 if (!s->tcos) |
79 goto fail; | |
10204
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
80 |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
81 switch (s->permutation) { |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
82 case FF_MDCT_PERM_NONE: |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
83 s->tsin = s->tcos + n4; |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
84 tstep = 1; |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
85 break; |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
86 case FF_MDCT_PERM_INTERLEAVE: |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
87 s->tsin = s->tcos + 1; |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
88 tstep = 2; |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
89 break; |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
90 default: |
781 | 91 goto fail; |
10204
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
92 } |
781 | 93 |
9658
67a20f0eb42c
Support for getting (i)MDCT output multiplied by a constant scaling factor.
serge
parents:
9480
diff
changeset
|
94 theta = 1.0 / 8.0 + (scale < 0 ? n4 : 0); |
67a20f0eb42c
Support for getting (i)MDCT output multiplied by a constant scaling factor.
serge
parents:
9480
diff
changeset
|
95 scale = sqrt(fabs(scale)); |
781 | 96 for(i=0;i<n4;i++) { |
9658
67a20f0eb42c
Support for getting (i)MDCT output multiplied by a constant scaling factor.
serge
parents:
9480
diff
changeset
|
97 alpha = 2 * M_PI * (i + theta) / n; |
10204
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
98 s->tcos[i*tstep] = -cos(alpha) * scale; |
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
99 s->tsin[i*tstep] = -sin(alpha) * scale; |
781 | 100 } |
101 return 0; | |
102 fail: | |
10204
db033d1fbf44
Allow arch-specific mdct code to request interleaving of cos/sin tables
mru
parents:
10199
diff
changeset
|
103 ff_mdct_end(s); |
781 | 104 return -1; |
105 } | |
106 | |
107 /* complex multiplication: p = a * b */ | |
108 #define CMUL(pre, pim, are, aim, bre, bim) \ | |
109 {\ | |
7545 | 110 FFTSample _are = (are);\ |
111 FFTSample _aim = (aim);\ | |
112 FFTSample _bre = (bre);\ | |
113 FFTSample _bim = (bim);\ | |
781 | 114 (pre) = _are * _bre - _aim * _bim;\ |
115 (pim) = _are * _bim + _aim * _bre;\ | |
116 } | |
117 | |
7544 | 118 /** |
119 * Compute the middle half of the inverse MDCT of size N = 2^nbits, | |
120 * thus excluding the parts that can be derived by symmetry | |
121 * @param output N/2 samples | |
122 * @param input N/2 samples | |
123 */ | |
10199 | 124 void ff_imdct_half_c(FFTContext *s, FFTSample *output, const FFTSample *input) |
781 | 125 { |
7544 | 126 int k, n8, n4, n2, n, j; |
10199 | 127 const uint16_t *revtab = s->revtab; |
781 | 128 const FFTSample *tcos = s->tcos; |
129 const FFTSample *tsin = s->tsin; | |
130 const FFTSample *in1, *in2; | |
7544 | 131 FFTComplex *z = (FFTComplex *)output; |
781 | 132 |
10199 | 133 n = 1 << s->mdct_bits; |
781 | 134 n2 = n >> 1; |
135 n4 = n >> 2; | |
7544 | 136 n8 = n >> 3; |
781 | 137 |
138 /* pre rotation */ | |
139 in1 = input; | |
140 in2 = input + n2 - 1; | |
141 for(k = 0; k < n4; k++) { | |
142 j=revtab[k]; | |
143 CMUL(z[j].re, z[j].im, *in2, *in1, tcos[k], tsin[k]); | |
144 in1 += 2; | |
145 in2 -= 2; | |
146 } | |
10199 | 147 ff_fft_calc(s, z); |
781 | 148 |
149 /* post rotation + reordering */ | |
7544 | 150 for(k = 0; k < n8; k++) { |
151 FFTSample r0, i0, r1, i1; | |
152 CMUL(r0, i1, z[n8-k-1].im, z[n8-k-1].re, tsin[n8-k-1], tcos[n8-k-1]); | |
153 CMUL(r1, i0, z[n8+k ].im, z[n8+k ].re, tsin[n8+k ], tcos[n8+k ]); | |
154 z[n8-k-1].re = r0; | |
155 z[n8-k-1].im = i0; | |
156 z[n8+k ].re = r1; | |
157 z[n8+k ].im = i1; | |
781 | 158 } |
7263 | 159 } |
160 | |
161 /** | |
162 * Compute inverse MDCT of size N = 2^nbits | |
163 * @param output N samples | |
164 * @param input N/2 samples | |
165 */ | |
10199 | 166 void ff_imdct_calc_c(FFTContext *s, FFTSample *output, const FFTSample *input) |
7263 | 167 { |
7544 | 168 int k; |
10199 | 169 int n = 1 << s->mdct_bits; |
7544 | 170 int n2 = n >> 1; |
171 int n4 = n >> 2; | |
781 | 172 |
7547 | 173 ff_imdct_half_c(s, output+n4, input); |
7263 | 174 |
7544 | 175 for(k = 0; k < n4; k++) { |
176 output[k] = -output[n2-k-1]; | |
177 output[n-k-1] = output[n2+k]; | |
7263 | 178 } |
179 } | |
180 | |
181 /** | |
781 | 182 * Compute MDCT of size N = 2^nbits |
183 * @param input N samples | |
184 * @param out N/2 samples | |
185 */ | |
10199 | 186 void ff_mdct_calc_c(FFTContext *s, FFTSample *out, const FFTSample *input) |
781 | 187 { |
188 int i, j, n, n8, n4, n2, n3; | |
7546 | 189 FFTSample re, im; |
10199 | 190 const uint16_t *revtab = s->revtab; |
781 | 191 const FFTSample *tcos = s->tcos; |
192 const FFTSample *tsin = s->tsin; | |
7544 | 193 FFTComplex *x = (FFTComplex *)out; |
781 | 194 |
10199 | 195 n = 1 << s->mdct_bits; |
781 | 196 n2 = n >> 1; |
197 n4 = n >> 2; | |
198 n8 = n >> 3; | |
199 n3 = 3 * n4; | |
200 | |
201 /* pre rotation */ | |
202 for(i=0;i<n8;i++) { | |
203 re = -input[2*i+3*n4] - input[n3-1-2*i]; | |
204 im = -input[n4+2*i] + input[n4-1-2*i]; | |
205 j = revtab[i]; | |
206 CMUL(x[j].re, x[j].im, re, im, -tcos[i], tsin[i]); | |
207 | |
208 re = input[2*i] - input[n2-1-2*i]; | |
209 im = -(input[n2+2*i] + input[n-1-2*i]); | |
210 j = revtab[n8 + i]; | |
211 CMUL(x[j].re, x[j].im, re, im, -tcos[n8 + i], tsin[n8 + i]); | |
212 } | |
213 | |
10199 | 214 ff_fft_calc(s, x); |
2967 | 215 |
781 | 216 /* post rotation */ |
7544 | 217 for(i=0;i<n8;i++) { |
218 FFTSample r0, i0, r1, i1; | |
219 CMUL(i1, r0, x[n8-i-1].re, x[n8-i-1].im, -tsin[n8-i-1], -tcos[n8-i-1]); | |
220 CMUL(i0, r1, x[n8+i ].re, x[n8+i ].im, -tsin[n8+i ], -tcos[n8+i ]); | |
221 x[n8-i-1].re = r0; | |
222 x[n8-i-1].im = i0; | |
223 x[n8+i ].re = r1; | |
224 x[n8+i ].im = i1; | |
781 | 225 } |
226 } | |
227 | |
10199 | 228 av_cold void ff_mdct_end(FFTContext *s) |
781 | 229 { |
230 av_freep(&s->tcos); | |
10199 | 231 ff_fft_end(s); |
781 | 232 } |