Imported Debian version 2.5.3~trusty1
[deb_ffmpeg.git] / ffmpeg / libavcodec / atrac3plusdsp.c
CommitLineData
2ba45a60
DM
1/*
2 * ATRAC3+ compatible decoder
3 *
4 * Copyright (c) 2010-2013 Maxim Poliakovski
5 *
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23/**
24 * @file
25 * DSP functions for ATRAC3+ decoder.
26 */
27
28#include <math.h>
29
30#include "libavutil/float_dsp.h"
31#include "avcodec.h"
32#include "sinewin.h"
33#include "fft.h"
34#include "atrac3plus.h"
35
36/**
37 * Map quant unit number to its position in the spectrum.
38 * To get the number of spectral lines in each quant unit do the following:
39 * num_specs = qu_to_spec_pos[i+1] - qu_to_spec_pos[i]
40 */
41const uint16_t ff_atrac3p_qu_to_spec_pos[33] = {
42 0, 16, 32, 48, 64, 80, 96, 112,
43 128, 160, 192, 224, 256, 288, 320, 352,
44 384, 448, 512, 576, 640, 704, 768, 896,
45 1024, 1152, 1280, 1408, 1536, 1664, 1792, 1920,
46 2048
47};
48
49/* Scalefactors table. */
50/* Approx. Equ: pow(2.0, (i - 16.0 + 0.501783948) / 3.0) */
51const float ff_atrac3p_sf_tab[64] = {
52 0.027852058, 0.0350914, 0.044212341, 0.055704117, 0.0701828,
53 0.088424683, 0.11140823, 0.1403656, 0.17684937, 0.22281647, 0.2807312, 0.35369873,
54 0.44563293, 0.5614624, 0.70739746, 0.89126587, 1.1229248, 1.4147949, 1.7825317,
55 2.2458496, 2.8295898, 3.5650635, 4.4916992, 5.6591797, 7.130127, 8.9833984,
56 11.318359, 14.260254, 17.966797, 22.636719, 28.520508, 35.933594, 45.273438,
57 57.041016, 71.867188, 90.546875, 114.08203, 143.73438, 181.09375, 228.16406,
58 287.46875, 362.1875, 456.32812, 574.9375, 724.375, 912.65625, 1149.875,
59 1448.75, 1825.3125, 2299.75, 2897.5, 3650.625, 4599.5, 5795.0,
60 7301.25, 9199.0, 11590.0, 14602.5, 18398.0, 23180.0, 29205.0,
61 36796.0, 46360.0, 58410.0
62};
63
64/* Mantissa table. */
65/* pow(10, x * log10(2) + 0.05) / 2 / ([1,2,3,5,7,15,31] + 0.5) */
66const float ff_atrac3p_mant_tab[8] = {
67 0.0,
68 0.74801636,
69 0.44882202,
70 0.32058716,
71 0.20400238,
72 0.1496048,
73 0.07239151,
74 0.035619736
75};
76
77#define ATRAC3P_MDCT_SIZE (ATRAC3P_SUBBAND_SAMPLES * 2)
78
79av_cold void ff_atrac3p_init_imdct(AVCodecContext *avctx, FFTContext *mdct_ctx)
80{
81 ff_init_ff_sine_windows(7);
82 ff_init_ff_sine_windows(6);
83
84 /* Initialize the MDCT transform. */
85 ff_mdct_init(mdct_ctx, 8, 1, -1.0);
86}
87
88#define TWOPI (2 * M_PI)
89
90#define DEQUANT_PHASE(ph) (((ph) & 0x1F) << 6)
91
92static DECLARE_ALIGNED(32, float, sine_table)[2048]; ///< wave table
93static DECLARE_ALIGNED(32, float, hann_window)[256]; ///< Hann windowing function
94static float amp_sf_tab[64]; ///< scalefactors for quantized amplitudes
95
96av_cold void ff_atrac3p_init_wave_synth(void)
97{
98 int i;
99
100 /* generate sine wave table */
101 for (i = 0; i < 2048; i++)
102 sine_table[i] = sin(TWOPI * i / 2048);
103
104 /* generate Hann window */
105 for (i = 0; i < 256; i++)
106 hann_window[i] = (1.0f - cos(TWOPI * i / 256.0f)) * 0.5f;
107
108 /* generate amplitude scalefactors table */
109 for (i = 0; i < 64; i++)
110 amp_sf_tab[i] = pow(2.0f, ((double)i - 3) / 4.0f);
111}
112
113/**
114 * Synthesize sine waves according to given parameters.
115 *
116 * @param[in] synth_param ptr to common synthesis parameters
117 * @param[in] waves_info parameters for each sine wave
118 * @param[in] envelope envelope data for all waves in a group
119 * @param[in] phase_shift flag indicates 180° phase shift
120 * @param[in] reg_offset region offset for trimming envelope data
121 * @param[out] out receives sythesized data
122 */
123static void waves_synth(Atrac3pWaveSynthParams *synth_param,
124 Atrac3pWavesData *waves_info,
125 Atrac3pWaveEnvelope *envelope,
126 int phase_shift, int reg_offset, float *out)
127{
128 int i, wn, inc, pos;
129 double amp;
130 Atrac3pWaveParam *wave_param = &synth_param->waves[waves_info->start_index];
131
132 for (wn = 0; wn < waves_info->num_wavs; wn++, wave_param++) {
133 /* amplitude dequantization */
134 amp = amp_sf_tab[wave_param->amp_sf] *
135 (!synth_param->amplitude_mode
136 ? (wave_param->amp_index + 1) / 15.13f
137 : 1.0f);
138
139 inc = wave_param->freq_index;
140 pos = DEQUANT_PHASE(wave_param->phase_index) - (reg_offset ^ 128) * inc & 2047;
141
142 /* waveform generation */
143 for (i = 0; i < 128; i++) {
144 out[i] += sine_table[pos] * amp;
145 pos = (pos + inc) & 2047;
146 }
147 }
148
149 /* fade in with steep Hann window if requested */
150 if (envelope->has_start_point) {
151 pos = (envelope->start_pos << 2) - reg_offset;
152 if (pos > 0 && pos <= 128) {
153 memset(out, 0, pos * sizeof(*out));
154 if (!envelope->has_stop_point ||
155 envelope->start_pos != envelope->stop_pos) {
156 out[pos + 0] *= hann_window[0];
157 out[pos + 1] *= hann_window[32];
158 out[pos + 2] *= hann_window[64];
159 out[pos + 3] *= hann_window[96];
160 }
161 }
162 }
163
164 /* fade out with steep Hann window if requested */
165 if (envelope->has_stop_point) {
166 pos = (envelope->stop_pos + 1 << 2) - reg_offset;
167 if (pos > 0 && pos <= 128) {
168 out[pos - 4] *= hann_window[96];
169 out[pos - 3] *= hann_window[64];
170 out[pos - 2] *= hann_window[32];
171 out[pos - 1] *= hann_window[0];
172 memset(&out[pos], 0, (128 - pos) * sizeof(out[pos]));
173 }
174 }
175}
176
177void ff_atrac3p_generate_tones(Atrac3pChanUnitCtx *ch_unit, AVFloatDSPContext *fdsp,
178 int ch_num, int sb, float *out)
179{
180 DECLARE_ALIGNED(32, float, wavreg1)[128] = { 0 };
181 DECLARE_ALIGNED(32, float, wavreg2)[128] = { 0 };
182 int i, reg1_env_nonzero, reg2_env_nonzero;
183 Atrac3pWavesData *tones_now = &ch_unit->channels[ch_num].tones_info_prev[sb];
184 Atrac3pWavesData *tones_next = &ch_unit->channels[ch_num].tones_info[sb];
185
186 /* reconstruct full envelopes for both overlapping regions
187 * from truncated bitstream data */
188 if (tones_next->pend_env.has_start_point &&
189 tones_next->pend_env.start_pos < tones_next->pend_env.stop_pos) {
190 tones_next->curr_env.has_start_point = 1;
191 tones_next->curr_env.start_pos = tones_next->pend_env.start_pos + 32;
192 } else if (tones_now->pend_env.has_start_point) {
193 tones_next->curr_env.has_start_point = 1;
194 tones_next->curr_env.start_pos = tones_now->pend_env.start_pos;
195 } else {
196 tones_next->curr_env.has_start_point = 0;
197 tones_next->curr_env.start_pos = 0;
198 }
199
200 if (tones_now->pend_env.has_stop_point &&
201 tones_now->pend_env.stop_pos >= tones_next->curr_env.start_pos) {
202 tones_next->curr_env.has_stop_point = 1;
203 tones_next->curr_env.stop_pos = tones_now->pend_env.stop_pos;
204 } else if (tones_next->pend_env.has_stop_point) {
205 tones_next->curr_env.has_stop_point = 1;
206 tones_next->curr_env.stop_pos = tones_next->pend_env.stop_pos + 32;
207 } else {
208 tones_next->curr_env.has_stop_point = 0;
209 tones_next->curr_env.stop_pos = 64;
210 }
211
212 /* is the visible part of the envelope non-zero? */
213 reg1_env_nonzero = (tones_now->curr_env.stop_pos < 32) ? 0 : 1;
214 reg2_env_nonzero = (tones_next->curr_env.start_pos >= 32) ? 0 : 1;
215
216 /* synthesize waves for both overlapping regions */
217 if (tones_now->num_wavs && reg1_env_nonzero)
218 waves_synth(ch_unit->waves_info_prev, tones_now, &tones_now->curr_env,
219 ch_unit->waves_info_prev->phase_shift[sb] & ch_num,
220 128, wavreg1);
221
222 if (tones_next->num_wavs && reg2_env_nonzero)
223 waves_synth(ch_unit->waves_info, tones_next, &tones_next->curr_env,
224 ch_unit->waves_info->phase_shift[sb] & ch_num, 0, wavreg2);
225
226 /* Hann windowing for non-faded wave signals */
227 if (tones_now->num_wavs && tones_next->num_wavs &&
228 reg1_env_nonzero && reg2_env_nonzero) {
229 fdsp->vector_fmul(wavreg1, wavreg1, &hann_window[128], 128);
230 fdsp->vector_fmul(wavreg2, wavreg2, hann_window, 128);
231 } else {
232 if (tones_now->num_wavs && !tones_now->curr_env.has_stop_point)
233 fdsp->vector_fmul(wavreg1, wavreg1, &hann_window[128], 128);
234
235 if (tones_next->num_wavs && !tones_next->curr_env.has_start_point)
236 fdsp->vector_fmul(wavreg2, wavreg2, hann_window, 128);
237 }
238
239 /* Overlap and add to residual */
240 for (i = 0; i < 128; i++)
241 out[i] += wavreg1[i] + wavreg2[i];
242}
243
244static const int subband_to_powgrp[ATRAC3P_SUBBANDS] = {
245 0, 1, 1, 2, 2, 2, 3, 3, 3, 3, 3, 4, 4, 4, 4, 4
246};
247
248/* noise table for power compensation */
249static const float noise_tab[1024] = {
250 -0.01358032, -0.05593872, 0.01696777, -0.14871216, -0.26412964, -0.09893799, 0.25723267,
251 0.02008057, -0.72235107, -0.44351196, -0.22985840, 0.16833496, 0.46902466, 0.05917358,
252 -0.15179443, 0.41299438, -0.01287842, 0.13360596, 0.43557739, -0.09530640, -0.58422852,
253 0.39266968, -0.08343506, -0.25604248, 0.22848511, 0.26013184, -0.65588379, 0.17288208,
254 -0.08673096, -0.05203247, 0.07299805, -0.28665161, -0.35806274, 0.06552124, -0.09387207,
255 0.21099854, -0.28347778, -0.72402954, 0.05050659, -0.10635376, -0.18853760, 0.29724121,
256 0.20703125, -0.29791260, -0.37634277, 0.47970581, -0.09976196, 0.32641602, -0.29248047,
257 -0.28237915, 0.26028442, -0.36157227, 0.22042847, -0.03222656, -0.37268066, -0.03759766,
258 0.09909058, 0.23284912, 0.19320679, 0.14453125, -0.02139282, -0.19702148, 0.31533813,
259 -0.16741943, 0.35031128, -0.35656738, -0.66128540, -0.00701904, 0.20898438, 0.26837158,
260 -0.33706665, -0.04568481, 0.12600708, 0.10284424, 0.07321167, -0.18280029, 0.38101196,
261 0.21301270, 0.04541016, 0.01156616, -0.26391602, -0.02346802, -0.22125244, 0.29760742,
262 -0.36233521, -0.31314087, -0.13967896, -0.11276245, -0.19433594, 0.34490967, 0.02343750,
263 0.21963501, -0.02777100, -0.67678833, -0.08999634, 0.14233398, -0.27697754, 0.51422119,
264 -0.05047607, 0.48327637, 0.37167358, -0.60806274, 0.18728638, -0.15191650, 0.00637817,
265 0.02832031, -0.15618896, 0.60644531, 0.21826172, 0.06384277, -0.31863403, 0.08816528,
266 0.15447998, -0.07015991, -0.08154297, -0.40966797, -0.39785767, -0.11709595, 0.22052002,
267 0.18466187, -0.17257690, 0.03759766, -0.06195068, 0.00433350, 0.12176514, 0.34011841,
268 0.25610352, -0.05294800, 0.41033936, 0.16854858, -0.76187134, 0.13845825, -0.19418335,
269 -0.21524048, -0.44412231, -0.08160400, -0.28195190, -0.01873779, 0.15524292, -0.37438965,
270 -0.44860840, 0.43096924, -0.24746704, 0.49856567, 0.14859009, 0.38159180, 0.20541382,
271 -0.39175415, -0.65850830, -0.43716431, 0.13037109, -0.05111694, 0.39956665, 0.21447754,
272 -0.04861450, 0.33654785, 0.10589600, -0.88085938, -0.30822754, 0.38577271, 0.30047607,
273 0.38836670, 0.09118652, -0.36477661, -0.01641846, -0.23031616, 0.26058960, 0.18859863,
274 -0.21868896, -0.17861938, -0.29754639, 0.09777832, 0.10806274, -0.51605225, 0.00076294,
275 0.13259888, 0.11090088, -0.24084473, 0.24957275, 0.01379395, -0.04141235, -0.04937744,
276 0.57394409, 0.27410889, 0.27587891, 0.45013428, -0.32592773, 0.11160278, -0.00970459,
277 0.29092407, 0.03356934, -0.70925903, 0.04882812, 0.43499756, 0.07720947, -0.27554321,
278 -0.01742554, -0.08413696, -0.04028320, -0.52850342, -0.07330322, 0.05181885, 0.21362305,
279 -0.18765259, 0.07058716, -0.03009033, 0.32662964, 0.27023315, -0.28002930, 0.17568970,
280 0.03338623, 0.30242920, -0.03921509, 0.32174683, -0.23733521, 0.08575439, -0.38269043,
281 0.09194946, -0.07238770, 0.17941284, -0.51278687, -0.25146484, 0.19790649, -0.19195557,
282 0.16549683, 0.42456055, 0.39129639, -0.02868652, 0.17980957, 0.24902344, -0.76583862,
283 -0.20959473, 0.61013794, 0.37011719, 0.36859131, -0.04486084, 0.10678101, -0.15994263,
284 -0.05328369, 0.28463745, -0.06420898, -0.36987305, -0.28009033, -0.11764526, 0.04312134,
285 -0.08038330, 0.04885864, -0.03067017, -0.00042725, 0.34289551, -0.00988770, 0.34838867,
286 0.32516479, -0.16271973, 0.38269043, 0.03240967, 0.12417603, -0.14331055, -0.34902954,
287 -0.18325806, 0.29421997, 0.44284058, 0.75170898, -0.67245483, -0.12176514, 0.27914429,
288 -0.29806519, 0.19863892, 0.30087280, 0.22680664, -0.36633301, -0.32534790, -0.57553101,
289 -0.16641235, 0.43811035, 0.08331299, 0.15942383, 0.26516724, -0.24240112, -0.11761475,
290 -0.16827393, -0.14260864, 0.46343994, 0.11804199, -0.55514526, -0.02520752, -0.14309692,
291 0.00448608, 0.02749634, -0.30545044, 0.70965576, 0.45108032, 0.66439819, -0.68255615,
292 -0.12496948, 0.09146118, -0.21109009, -0.23791504, 0.79943848, -0.35205078, -0.24963379,
293 0.18719482, -0.19079590, 0.07458496, 0.07623291, -0.28781128, -0.37121582, -0.19580078,
294 -0.01773071, -0.16717529, 0.13040161, 0.14672852, 0.42379761, 0.03582764, 0.11431885,
295 0.05145264, 0.44702148, 0.08963013, 0.01367188, -0.54519653, -0.12692261, 0.21176147,
296 0.04925537, 0.30670166, -0.11029053, 0.19555664, -0.27740479, 0.23043823, 0.15554810,
297 -0.19299316, -0.25729370, 0.17800903, -0.03579712, -0.05065918, -0.06933594, -0.09500122,
298 -0.07821655, 0.23889160, -0.31900024, 0.03073120, -0.00415039, 0.61315918, 0.37176514,
299 -0.13442993, -0.15536499, -0.19216919, -0.37899780, 0.19992065, 0.02630615, -0.12573242,
300 0.25927734, -0.02447510, 0.29629517, -0.40731812, -0.17333984, 0.24310303, -0.10607910,
301 0.14828491, 0.08792114, -0.18743896, -0.05572510, -0.04833984, 0.10473633, -0.29028320,
302 -0.67687988, -0.28170776, -0.41687012, 0.05413818, -0.23284912, 0.09555054, -0.08969116,
303 -0.15112305, 0.12738037, 0.35986328, 0.28948975, 0.30691528, 0.23956299, 0.06973267,
304 -0.31198120, -0.18450928, 0.22280884, -0.21600342, 0.23522949, -0.61840820, -0.13012695,
305 0.26412964, 0.47320557, -0.26440430, 0.38757324, 0.17352295, -0.26104736, -0.25866699,
306 -0.12274170, -0.29733276, 0.07687378, 0.18588257, -0.08880615, 0.31185913, 0.05313110,
307 -0.10885620, -0.14901733, -0.22323608, -0.08538818, 0.19812012, 0.19732666, -0.18927002,
308 0.29058838, 0.25555420, -0.48599243, 0.18768311, 0.01345825, 0.34887695, 0.21530151,
309 0.19857788, 0.18661499, -0.01394653, -0.09063721, -0.38781738, 0.27160645, -0.20379639,
310 -0.32119751, -0.23889160, 0.27096558, 0.24951172, 0.07922363, 0.07479858, -0.50946045,
311 0.10220337, 0.58364868, -0.19503784, -0.18560791, -0.01165771, 0.47195435, 0.22430420,
312 -0.38635254, -0.03732300, -0.09179688, 0.06991577, 0.15106201, 0.20605469, -0.05969238,
313 -0.41821289, 0.12231445, -0.04672241, -0.05117798, -0.11523438, -0.51849365, -0.04077148,
314 0.44284058, -0.64086914, 0.17019653, 0.02236938, 0.22848511, -0.23214722, -0.32354736,
315 -0.14068604, -0.29690552, -0.19891357, 0.02774048, -0.20965576, -0.52191162, -0.19299316,
316 -0.07290649, 0.49053955, -0.22302246, 0.05642700, 0.13122559, -0.20819092, -0.83590698,
317 -0.08181763, 0.26797485, -0.00091553, -0.09457397, 0.17089844, -0.27020264, 0.30270386,
318 0.05496216, 0.09564209, -0.08590698, 0.02130127, 0.35931396, 0.21728516, -0.15396118,
319 -0.05053711, 0.02719116, 0.16302490, 0.43212891, 0.10229492, -0.40820312, 0.21646118,
320 0.08435059, -0.11145020, -0.39962769, -0.05618286, -0.10223389, -0.60839844, 0.33724976,
321 -0.06341553, -0.47369385, -0.32852173, 0.05242920, 0.19635010, -0.19137573, -0.67901611,
322 0.16180420, 0.05133057, -0.22283936, 0.09646606, 0.24288940, -0.45007324, 0.08804321,
323 0.14053345, 0.22619629, -0.01000977, 0.36355591, -0.19863892, -0.30364990, -0.24118042,
324 -0.57461548, 0.26498413, 0.04345703, -0.09796143, -0.47714233, -0.23739624, 0.18737793,
325 0.08926392, -0.02795410, 0.00305176, -0.08700562, -0.38711548, 0.03222656, 0.10940552,
326 -0.41906738, -0.01620483, -0.47061157, 0.37985229, -0.21624756, 0.47976685, -0.20046997,
327 -0.62533569, -0.26907349, -0.02877808, 0.00671387, -0.29071045, -0.24685669, -0.15722656,
328 -0.26055908, 0.29968262, 0.28225708, -0.08990479, -0.16748047, -0.46759033, -0.25067139,
329 -0.25183105, -0.45932007, 0.05828857, 0.29006958, 0.23840332, -0.17974854, 0.26931763,
330 0.10696411, -0.06848145, -0.17126465, -0.10522461, -0.55386353, -0.42306519, -0.07608032,
331 0.24380493, 0.38586426, 0.16882324, 0.26751709, 0.17303467, 0.35809326, -0.22094727,
332 -0.30703735, -0.28497314, -0.04321289, 0.15219116, -0.17071533, -0.39334106, 0.03439331,
333 -0.10809326, -0.30590820, 0.26449585, -0.07412720, 0.13638306, -0.01062012, 0.27996826,
334 0.04397583, -0.05557251, -0.56933594, 0.03363037, -0.00949097, 0.52642822, -0.44329834,
335 0.28308105, -0.05499268, -0.23312378, -0.29870605, -0.05123901, 0.26831055, -0.35238647,
336 -0.30993652, 0.34646606, -0.19775391, 0.44595337, 0.13769531, 0.45358276, 0.19961548,
337 0.42681885, 0.15722656, 0.00128174, 0.23757935, 0.40988159, 0.25164795, -0.00732422,
338 -0.12405396, -0.43420410, -0.00402832, 0.34243774, 0.36264038, 0.18807983, -0.09301758,
339 -0.10296631, 0.05532837, -0.31652832, 0.14337158, 0.35040283, 0.32540894, 0.05728149,
340 -0.12030029, -0.25942993, -0.20312500, -0.16491699, -0.46051025, -0.08004761, 0.50772095,
341 0.16168213, 0.28439331, 0.08105469, -0.19104004, 0.38589478, -0.16400146, -0.25454712,
342 0.20281982, -0.20730591, -0.06311035, 0.32937622, 0.15032959, -0.05340576, 0.30487061,
343 -0.11648560, 0.38009644, -0.20062256, 0.43466187, 0.01150513, 0.35754395, -0.13146973,
344 0.67489624, 0.05212402, 0.27914429, -0.39431763, 0.75308228, -0.13366699, 0.24453735,
345 0.42248535, -0.65905762, -0.00546265, -0.03491211, -0.13659668, -0.08294678, -0.45666504,
346 0.27188110, 0.12731934, 0.61148071, 0.10449219, -0.28836060, 0.00091553, 0.24618530,
347 0.13119507, 0.05685425, 0.17355347, 0.42034912, 0.08514404, 0.24536133, 0.18951416,
348 -0.19107056, -0.15036011, 0.02334595, 0.54986572, 0.32321167, -0.16104126, -0.03054810,
349 0.43594360, 0.17309570, 0.61053467, 0.24731445, 0.33334351, 0.15240479, 0.15588379,
350 0.36425781, -0.30407715, -0.13302612, 0.00427246, 0.04171753, -0.33178711, 0.34216309,
351 -0.12463379, -0.02764893, 0.05905151, -0.31436157, 0.16531372, 0.34542847, -0.03292847,
352 0.12527466, -0.12313843, -0.13171387, 0.04757690, -0.45095825, -0.19085693, 0.35342407,
353 -0.23239136, -0.34387207, 0.11264038, -0.15740967, 0.05273438, 0.74942017, 0.21505737,
354 0.08514404, -0.42391968, -0.19531250, 0.35293579, 0.25305176, 0.15731812, -0.70324707,
355 -0.21591187, 0.35604858, 0.14132690, 0.11724854, 0.15853882, -0.24597168, 0.07019043,
356 0.02127075, 0.12658691, 0.06390381, -0.12292480, 0.15441895, -0.47640991, 0.06195068,
357 0.58981323, -0.15151978, -0.03604126, -0.45059204, -0.01672363, -0.46997070, 0.25750732,
358 0.18084717, 0.06661987, 0.13253784, 0.67828369, 0.11370850, 0.11325073, -0.04611206,
359 -0.07791138, -0.36544800, -0.06747437, -0.31594849, 0.16131592, 0.41983032, 0.11071777,
360 -0.36889648, 0.30963135, -0.37875366, 0.58508301, 0.00393677, 0.12338257, 0.03424072,
361 -0.21728516, -0.12838745, -0.46981812, 0.05868530, -0.25015259, 0.27407837, 0.65240479,
362 -0.34429932, -0.15179443, 0.14056396, 0.33505249, 0.28826904, 0.09921265, 0.34390259,
363 0.13656616, -0.23608398, 0.00863647, 0.02627563, -0.19119263, 0.19775391, -0.07214355,
364 0.07809448, 0.03454590, -0.03417969, 0.00033569, -0.23095703, 0.18673706, 0.05798340,
365 0.03814697, -0.04318237, 0.05487061, 0.08633423, 0.55950928, -0.06347656, 0.10333252,
366 0.25305176, 0.05853271, 0.12246704, -0.25543213, -0.34262085, -0.36437988, -0.21304321,
367 -0.05093384, 0.02777100, 0.07620239, -0.21215820, -0.09326172, 0.19021606, -0.40579224,
368 -0.01193237, 0.19845581, -0.35336304, -0.07397461, 0.20104980, 0.08615112, -0.44375610,
369 0.11419678, 0.24453735, -0.16555786, -0.05081177, -0.01406860, 0.27893066, -0.18692017,
370 0.07473755, 0.03451538, -0.39733887, 0.21548462, -0.22534180, -0.39651489, -0.04989624,
371 -0.57662964, 0.06390381, 0.62020874, -0.13470459, 0.04345703, -0.21862793, -0.02789307,
372 0.51696777, -0.27587891, 0.39004517, 0.09857178, -0.00738525, 0.31317139, 0.00048828,
373 -0.46572876, 0.29531860, -0.10009766, -0.27856445, 0.03594971, 0.25048828, -0.74584961,
374 -0.25350952, -0.03302002, 0.31188965, 0.01571655, 0.46710205, 0.21591187, 0.07260132,
375 -0.42132568, -0.53900146, -0.13674927, -0.16571045, -0.34454346, 0.12359619, -0.11184692,
376 0.00967407, 0.34576416, -0.05761719, 0.34848022, 0.17645264, -0.39395142, 0.10339355,
377 0.18215942, 0.20697021, 0.59109497, -0.11560059, -0.07385254, 0.10397339, 0.35437012,
378 -0.22863770, 0.01794434, 0.17559814, -0.17495728, 0.12142944, 0.10928345, -1.00000000,
379 -0.01379395, 0.21237183, -0.27035522, 0.27319336, -0.37066650, 0.41354370, -0.40054321,
380 0.00689697, 0.26321411, 0.39266968, 0.65298462, 0.41625977, -0.13909912, 0.78375244,
381 -0.30941772, 0.20169067, -0.39367676, 0.94021606, -0.24066162, 0.05557251, -0.24533081,
382 -0.05444336, -0.76754761, -0.19375610, -0.11041260, -0.17532349, 0.16006470, 0.02188110,
383 0.17465210, -0.04342651, -0.56777954, -0.40988159, 0.26687622, 0.11700439, -0.00344849,
384 -0.05395508, 0.37426758, -0.40719604, -0.15032959, -0.01660156, 0.04196167, -0.04559326,
385 -0.12969971, 0.12011719, 0.08419800, -0.11199951, 0.35174561, 0.10275269, -0.25686646,
386 0.48446655, 0.03225708, 0.28408813, -0.18701172, 0.36282349, -0.03280640, 0.32302856,
387 0.17233276, 0.48269653, 0.31112671, -0.04946899, 0.12774658, 0.52685547, 0.10211182,
388 0.05953979, 0.05999756, 0.20144653, 0.00744629, 0.27316284, 0.24377441, 0.39672852,
389 0.01702881, -0.35513306, 0.11364746, -0.13555908, 0.48880005, -0.15417480, -0.09149170,
390 -0.02615356, 0.46246338, -0.72250366, 0.22332764, 0.23849487, -0.25686646, -0.08514404,
391 -0.02062988, -0.34494019, -0.02297974, -0.80386353, -0.08074951, -0.12689209, -0.06896973,
392 0.24099731, -0.35650635, -0.09558105, 0.29254150, 0.23132324, -0.16726685, 0.00000000,
393 -0.24237061, 0.30899048, 0.29504395, -0.20898438, 0.17059326, -0.07672119, -0.14395142,
394 0.05572510, 0.20602417, -0.51550293, -0.03167725, -0.48840332, -0.20425415, 0.14144897,
395 0.07275391, -0.76669312, -0.22488403, 0.20651245, 0.03259277, 0.00085449, 0.03039551,
396 0.47555542, 0.38351440
397};
398
399/** Noise level table for power compensation.
400 * Equ: pow(2.0f, (double)(6 - i) / 3.0f) where i = 0...15 */
401static const float pwc_levs[16] = {
402 3.96875, 3.15625, 2.5, 2.0, 1.59375, 1.25, 1.0, 0.78125,
403 0.625, 0.5, 0.40625, 0.3125, 0.25, 0.1875, 0.15625, 0.0
404};
405
406/** Map subband number to quant unit number. */
407static const int subband_to_qu[17] = {
408 0, 8, 12, 16, 18, 20, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32
409};
410
411void ff_atrac3p_power_compensation(Atrac3pChanUnitCtx *ctx, int ch_index,
412 float *sp, int rng_index, int sb)
413{
414 AtracGainInfo *g1, *g2;
415 float pwcsp[ATRAC3P_SUBBAND_SAMPLES], *dst, grp_lev, qu_lev;
416 int i, gain_lev, gcv = 0, qu, nsp;
417 int swap_ch = (ctx->unit_type == CH_UNIT_STEREO && ctx->swap_channels[sb]) ? 1 : 0;
418
419 if (ctx->channels[ch_index ^ swap_ch].power_levs[subband_to_powgrp[sb]] == ATRAC3P_POWER_COMP_OFF)
420 return;
421
422 /* generate initial noise spectrum */
423 for (i = 0; i < ATRAC3P_SUBBAND_SAMPLES; i++, rng_index++)
424 pwcsp[i] = noise_tab[rng_index & 0x3FF];
425
426 /* check gain control information */
427 g1 = &ctx->channels[ch_index ^ swap_ch].gain_data[sb];
428 g2 = &ctx->channels[ch_index ^ swap_ch].gain_data_prev[sb];
429
430 gain_lev = (g1->num_points > 0) ? (6 - g1->lev_code[0]) : 0;
431
432 for (i = 0; i < g2->num_points; i++)
433 gcv = FFMAX(gcv, gain_lev - (g2->lev_code[i] - 6));
434
435 for (i = 0; i < g1->num_points; i++)
436 gcv = FFMAX(gcv, 6 - g1->lev_code[i]);
437
438 grp_lev = pwc_levs[ctx->channels[ch_index ^ swap_ch].power_levs[subband_to_powgrp[sb]]] / (1 << gcv);
439
440 /* skip the lowest two quant units (frequencies 0...351 Hz) for subband 0 */
441 for (qu = subband_to_qu[sb] + (!sb ? 2 : 0); qu < subband_to_qu[sb + 1]; qu++) {
442 if (ctx->channels[ch_index].qu_wordlen[qu] <= 0)
443 continue;
444
445 qu_lev = ff_atrac3p_sf_tab[ctx->channels[ch_index].qu_sf_idx[qu]] *
446 ff_atrac3p_mant_tab[ctx->channels[ch_index].qu_wordlen[qu]] /
447 (1 << ctx->channels[ch_index].qu_wordlen[qu]) * grp_lev;
448
449 dst = &sp[ff_atrac3p_qu_to_spec_pos[qu]];
450 nsp = ff_atrac3p_qu_to_spec_pos[qu + 1] - ff_atrac3p_qu_to_spec_pos[qu];
451
452 for (i = 0; i < nsp; i++)
453 dst[i] += pwcsp[i] * qu_lev;
454 }
455}
456
457void ff_atrac3p_imdct(AVFloatDSPContext *fdsp, FFTContext *mdct_ctx, float *pIn,
458 float *pOut, int wind_id, int sb)
459{
460 int i;
461
462 if (sb & 1)
463 for (i = 0; i < ATRAC3P_SUBBAND_SAMPLES / 2; i++)
464 FFSWAP(float, pIn[i], pIn[ATRAC3P_SUBBAND_SAMPLES - 1 - i]);
465
466 mdct_ctx->imdct_calc(mdct_ctx, pOut, pIn);
467
468 /* Perform windowing on the output.
469 * ATRAC3+ uses two different MDCT windows:
470 * - The first one is just the plain sine window of size 256
471 * - The 2nd one is the plain sine window of size 128
472 * wrapped into zero (at the start) and one (at the end) regions.
473 * Both regions are 32 samples long. */
474 if (wind_id & 2) { /* 1st half: steep window */
475 memset(pOut, 0, sizeof(float) * 32);
476 fdsp->vector_fmul(&pOut[32], &pOut[32], ff_sine_64, 64);
477 } else /* 1st half: simple sine window */
478 fdsp->vector_fmul(pOut, pOut, ff_sine_128, ATRAC3P_MDCT_SIZE / 2);
479
480 if (wind_id & 1) { /* 2nd half: steep window */
481 fdsp->vector_fmul_reverse(&pOut[160], &pOut[160], ff_sine_64, 64);
482 memset(&pOut[224], 0, sizeof(float) * 32);
483 } else /* 2nd half: simple sine window */
484 fdsp->vector_fmul_reverse(&pOut[128], &pOut[128], ff_sine_128,
485 ATRAC3P_MDCT_SIZE / 2);
486}
487
488/* lookup table for fast modulo 23 op required for cyclic buffers of the IPQF */
489static const int mod23_lut[26] = {
490 23, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11,
491 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 0
492};
493
494/* First half of the 384-tap IPQF filtering coefficients. */
495static const float ipqf_coeffs1[ATRAC3P_PQF_FIR_LEN][16] = {
496 { -5.8336207e-7, -8.0604229e-7, -4.2005411e-7, -4.4400572e-8,
497 3.226247e-8, 3.530856e-8, 1.2660377e-8, 0.000010516783,
498 -0.000011838618, 6.005389e-7, 0.0000014333754, 0.0000023108685,
499 0.0000032569742, 0.0000046192422, 0.0000063894258, 0.0000070302972 },
500 { -0.0000091622824, -0.000010502935, -0.0000079212787, -0.0000041712024,
501 -0.0000026336629, -0.0000015432918, -5.7168614e-7, 0.0000018111954,
502 0.000023530851, 0.00002780562, 0.000032302323, 0.000036968919,
503 0.000041575615, 0.000045337845, 0.000046043948, 0.000048585582 },
504 { -0.000064464548, -0.000068306952, -0.000073081472, -0.00007612785,
505 -0.000074850752, -0.000070208509, -0.000062285151, -0.000058270442,
506 -0.000056296329, -0.000049888811, -0.000035615325, -0.000018532943,
507 0.0000016657353, 0.00002610587, 0.000053397067, 0.00008079566 },
508 { -0.00054488552, -0.00052537228, -0.00049731287, -0.00045778,
509 -0.00040612387, -0.00034301577, -0.00026866337, -0.00018248901,
510 -0.000084307925, 0.000025081157, 0.00014135583, 0.00026649953,
511 0.00039945057, 0.00053928449, 0.00068422867, 0.00083093712 },
512 { -0.0014771431, -0.001283227, -0.0010566821, -0.00079780724,
513 -0.00050782406, -0.00018855913, 0.00015771533, 0.00052769453,
514 0.00091862219, 0.001326357, 0.0017469483, 0.0021754825,
515 0.0026067684, 0.0030352892, 0.0034549395, 0.0038591374 },
516 { -0.0022995141, -0.001443546, -0.00049266568, 0.00055068987,
517 0.001682895, 0.0028992873, 0.0041943151, 0.0055614738,
518 0.0069935122, 0.0084823566, 0.010018963, 0.011593862,
519 0.013196872, 0.014817309, 0.016444042, 0.018065533 },
520 { -0.034426283, -0.034281436, -0.033992987, -0.033563249,
521 -0.032995768, -0.032295227, -0.031467363, -0.030518902,
522 -0.02945766, -0.028291954, -0.027031265, -0.025685543,
523 -0.024265358, -0.022781773, -0.021246184, -0.019670162 },
524 { -0.0030586775, -0.0037203205, -0.0042847847, -0.0047529764,
525 -0.0051268316, -0.0054091476, -0.0056034233, -0.005714261,
526 -0.0057445862, -0.0057025906, -0.0055920109, -0.0054194843,
527 -0.0051914565, -0.0049146507, -0.0045959447, -0.0042418269 },
528 { -0.0016376863, -0.0017651899, -0.0018608454, -0.0019252141,
529 -0.0019593791, -0.0019653172, -0.0019450618, -0.0018990048,
530 -0.00183808, -0.0017501717, -0.0016481078, -0.0015320742,
531 -0.0014046903, -0.0012685474, -0.001125814, -0.00097943726 },
532 { -0.00055432378, -0.00055472925, -0.00054783461, -0.00053276919,
533 -0.00051135791, -0.00048466062, -0.00045358928, -0.00042499689,
534 -0.00036942671, -0.0003392619, -0.00030001783, -0.00025986304,
535 -0.0002197204, -0.00018116167, -0.00014691355, -0.00011279432 },
536 { -0.000064147389, -0.00006174868, -0.000054267788, -0.000047133824,
537 -0.000042927582, -0.000039477309, -0.000036340745, -0.000029687517,
538 -0.000049787737, -0.000041577889, -0.000033864744, -0.000026534748,
539 -0.000019841305, -0.000014789486, -0.000013131184, -0.0000099198869 },
540 { -0.0000062990207, -0.0000072701259, -0.000011984052, -0.000017348082,
541 -0.000019907106, -0.000021348773, -0.000021961965, -0.000012203576,
542 -0.000010840992, 4.6299544e-7, 5.2588763e-7, 2.7792686e-7,
543 -2.3649704e-7, -0.0000010897784, -9.171448e-7, -5.22682e-7 }
544};
545
546/* Second half of the 384-tap IPQF filtering coefficients. */
547static const float ipqf_coeffs2[ATRAC3P_PQF_FIR_LEN][16] = {
548 { 5.22682e-7, 9.171448e-7, 0.0000010897784, 2.3649704e-7,
549 -2.7792686e-7, -5.2588763e-7, -4.6299544e-7, 0.000010840992,
550 -0.000012203576, -0.000021961965, -0.000021348773, -0.000019907106,
551 -0.000017348082, -0.000011984052, -0.0000072701259, -0.0000062990207 },
552 { 0.0000099198869, 0.000013131184, 0.000014789486, 0.000019841305,
553 0.000026534748, 0.000033864744, 0.000041577889, 0.000049787737,
554 -0.000029687517, -0.000036340745, -0.000039477309, -0.000042927582,
555 -0.000047133824, -0.000054267788, -0.00006174868, -0.000064147389 },
556 { 0.00011279432, 0.00014691355, 0.00018116167, 0.0002197204,
557 0.00025986304, 0.00030001783, 0.0003392619, 0.00036942671,
558 -0.00042499689, -0.00045358928, -0.00048466062, -0.00051135791,
559 -0.00053276919, -0.00054783461, -0.00055472925, -0.00055432378 },
560 { 0.00097943726, 0.001125814, 0.0012685474, 0.0014046903,
561 0.0015320742, 0.0016481078, 0.0017501717, 0.00183808,
562 -0.0018990048, -0.0019450618, -0.0019653172, -0.0019593791,
563 -0.0019252141, -0.0018608454, -0.0017651899, -0.0016376863 },
564 { 0.0042418269, 0.0045959447, 0.0049146507, 0.0051914565,
565 0.0054194843, 0.0055920109, 0.0057025906, 0.0057445862,
566 -0.005714261, -0.0056034233, -0.0054091476, -0.0051268316,
567 -0.0047529764, -0.0042847847, -0.0037203205, -0.0030586775 },
568 { 0.019670162, 0.021246184, 0.022781773, 0.024265358,
569 0.025685543, 0.027031265, 0.028291954, 0.02945766,
570 -0.030518902, -0.031467363, -0.032295227, -0.032995768,
571 -0.033563249, -0.033992987, -0.034281436, -0.034426283 },
572 { -0.018065533, -0.016444042, -0.014817309, -0.013196872,
573 -0.011593862, -0.010018963, -0.0084823566, -0.0069935122,
574 0.0055614738, 0.0041943151, 0.0028992873, 0.001682895,
575 0.00055068987, -0.00049266568, -0.001443546, -0.0022995141 },
576 { -0.0038591374, -0.0034549395, -0.0030352892, -0.0026067684,
577 -0.0021754825, -0.0017469483, -0.001326357, -0.00091862219,
578 0.00052769453, 0.00015771533, -0.00018855913, -0.00050782406,
579 -0.00079780724, -0.0010566821, -0.001283227, -0.0014771431 },
580 { -0.00083093712, -0.00068422867, -0.00053928449, -0.00039945057,
581 -0.00026649953, -0.00014135583, -0.000025081157, 0.000084307925,
582 -0.00018248901, -0.00026866337, -0.00034301577, -0.00040612387,
583 -0.00045778, -0.00049731287, -0.00052537228, -0.00054488552 },
584 { -0.00008079566, -0.000053397067, -0.00002610587, -0.0000016657353,
585 0.000018532943, 0.000035615325, 0.000049888811, 0.000056296329,
586 -0.000058270442, -0.000062285151, -0.000070208509, -0.000074850752,
587 -0.00007612785, -0.000073081472, -0.000068306952, -0.000064464548 },
588 { -0.000048585582, -0.000046043948, -0.000045337845, -0.000041575615,
589 -0.000036968919, -0.000032302323, -0.00002780562, -0.000023530851,
590 0.0000018111954, -5.7168614e-7, -0.0000015432918, -0.0000026336629,
591 -0.0000041712024, -0.0000079212787, -0.000010502935, -0.0000091622824 },
592 { -0.0000070302972, -0.0000063894258, -0.0000046192422, -0.0000032569742,
593 -0.0000023108685, -0.0000014333754, -6.005389e-7, 0.000011838618,
594 0.000010516783, 1.2660377e-8, 3.530856e-8, 3.226247e-8,
595 -4.4400572e-8, -4.2005411e-7, -8.0604229e-7, -5.8336207e-7 }
596};
597
598void ff_atrac3p_ipqf(FFTContext *dct_ctx, Atrac3pIPQFChannelCtx *hist,
599 const float *in, float *out)
600{
601 int i, s, sb, t, pos_now, pos_next;
602 DECLARE_ALIGNED(32, float, idct_in)[ATRAC3P_SUBBANDS];
603 DECLARE_ALIGNED(32, float, idct_out)[ATRAC3P_SUBBANDS];
604
605 memset(out, 0, ATRAC3P_FRAME_SAMPLES * sizeof(*out));
606
607 for (s = 0; s < ATRAC3P_SUBBAND_SAMPLES; s++) {
608 /* pick up one sample from each subband */
609 for (sb = 0; sb < ATRAC3P_SUBBANDS; sb++)
610 idct_in[sb] = in[sb * ATRAC3P_SUBBAND_SAMPLES + s];
611
612 /* Calculate the sine and cosine part of the PQF using IDCT-IV */
613 dct_ctx->imdct_half(dct_ctx, idct_out, idct_in);
614
615 /* append the result to the history */
616 for (i = 0; i < 8; i++) {
617 hist->buf1[hist->pos][i] = idct_out[i + 8];
618 hist->buf2[hist->pos][i] = idct_out[7 - i];
619 }
620
621 pos_now = hist->pos;
622 pos_next = mod23_lut[pos_now + 2]; // pos_next = (pos_now + 1) % 23;
623
624 for (t = 0; t < ATRAC3P_PQF_FIR_LEN; t++) {
625 for (i = 0; i < 8; i++) {
626 out[s * 16 + i + 0] += hist->buf1[pos_now][i] * ipqf_coeffs1[t][i] +
627 hist->buf2[pos_next][i] * ipqf_coeffs2[t][i];
628 out[s * 16 + i + 8] += hist->buf1[pos_now][7 - i] * ipqf_coeffs1[t][i + 8] +
629 hist->buf2[pos_next][7 - i] * ipqf_coeffs2[t][i + 8];
630 }
631
632 pos_now = mod23_lut[pos_next + 2]; // pos_now = (pos_now + 2) % 23;
633 pos_next = mod23_lut[pos_now + 2]; // pos_next = (pos_next + 2) % 23;
634 }
635
636 hist->pos = mod23_lut[hist->pos]; // hist->pos = (hist->pos - 1) % 23;
637 }
638}