Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
4349 | Serge | 1 | /* |
2 | * MPEG-4 Parametric Stereo decoding functions |
||
3 | * Copyright (c) 2010 Alex Converse |
||
4 | * |
||
5 | * This file is part of FFmpeg. |
||
6 | * |
||
7 | * FFmpeg is free software; you can redistribute it and/or |
||
8 | * modify it under the terms of the GNU Lesser General Public |
||
9 | * License as published by the Free Software Foundation; either |
||
10 | * version 2.1 of the License, or (at your option) any later version. |
||
11 | * |
||
12 | * FFmpeg is distributed in the hope that it will be useful, |
||
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 |
||
18 | * License along with FFmpeg; if not, write to the Free Software |
||
19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||
20 | */ |
||
21 | |||
22 | #include |
||
23 | #include "libavutil/common.h" |
||
24 | #include "libavutil/internal.h" |
||
25 | #include "libavutil/mathematics.h" |
||
26 | #include "avcodec.h" |
||
27 | #include "get_bits.h" |
||
28 | #include "aacps.h" |
||
29 | #include "aacps_tablegen.h" |
||
30 | #include "aacpsdata.c" |
||
31 | |||
32 | #define PS_BASELINE 0 ///< Operate in Baseline PS mode |
||
33 | ///< Baseline implies 10 or 20 stereo bands, |
||
34 | ///< mixing mode A, and no ipd/opd |
||
35 | |||
36 | #define numQMFSlots 32 //numTimeSlots * RATE |
||
37 | |||
38 | static const int8_t num_env_tab[2][4] = { |
||
39 | { 0, 1, 2, 4, }, |
||
40 | { 1, 2, 3, 4, }, |
||
41 | }; |
||
42 | |||
43 | static const int8_t nr_iidicc_par_tab[] = { |
||
44 | 10, 20, 34, 10, 20, 34, |
||
45 | }; |
||
46 | |||
47 | static const int8_t nr_iidopd_par_tab[] = { |
||
48 | 5, 11, 17, 5, 11, 17, |
||
49 | }; |
||
50 | |||
51 | enum { |
||
52 | huff_iid_df1, |
||
53 | huff_iid_dt1, |
||
54 | huff_iid_df0, |
||
55 | huff_iid_dt0, |
||
56 | huff_icc_df, |
||
57 | huff_icc_dt, |
||
58 | huff_ipd_df, |
||
59 | huff_ipd_dt, |
||
60 | huff_opd_df, |
||
61 | huff_opd_dt, |
||
62 | }; |
||
63 | |||
64 | static const int huff_iid[] = { |
||
65 | huff_iid_df0, |
||
66 | huff_iid_df1, |
||
67 | huff_iid_dt0, |
||
68 | huff_iid_dt1, |
||
69 | }; |
||
70 | |||
71 | static VLC vlc_ps[10]; |
||
72 | |||
73 | #define READ_PAR_DATA(PAR, OFFSET, MASK, ERR_CONDITION) \ |
||
74 | /** \ |
||
75 | * Read Inter-channel Intensity Difference/Inter-Channel Coherence/ \ |
||
76 | * Inter-channel Phase Difference/Overall Phase Difference parameters from the \ |
||
77 | * bitstream. \ |
||
78 | * \ |
||
79 | * @param avctx contains the current codec context \ |
||
80 | * @param gb pointer to the input bitstream \ |
||
81 | * @param ps pointer to the Parametric Stereo context \ |
||
82 | * @param PAR pointer to the parameter to be read \ |
||
83 | * @param e envelope to decode \ |
||
84 | * @param dt 1: time delta-coded, 0: frequency delta-coded \ |
||
85 | */ \ |
||
86 | static int read_ ## PAR ## _data(AVCodecContext *avctx, GetBitContext *gb, PSContext *ps, \ |
||
87 | int8_t (*PAR)[PS_MAX_NR_IIDICC], int table_idx, int e, int dt) \ |
||
88 | { \ |
||
89 | int b, num = ps->nr_ ## PAR ## _par; \ |
||
90 | VLC_TYPE (*vlc_table)[2] = vlc_ps[table_idx].table; \ |
||
91 | if (dt) { \ |
||
92 | int e_prev = e ? e - 1 : ps->num_env_old - 1; \ |
||
93 | e_prev = FFMAX(e_prev, 0); \ |
||
94 | for (b = 0; b < num; b++) { \ |
||
95 | int val = PAR[e_prev][b] + get_vlc2(gb, vlc_table, 9, 3) - OFFSET; \ |
||
96 | if (MASK) val &= MASK; \ |
||
97 | PAR[e][b] = val; \ |
||
98 | if (ERR_CONDITION) \ |
||
99 | goto err; \ |
||
100 | } \ |
||
101 | } else { \ |
||
102 | int val = 0; \ |
||
103 | for (b = 0; b < num; b++) { \ |
||
104 | val += get_vlc2(gb, vlc_table, 9, 3) - OFFSET; \ |
||
105 | if (MASK) val &= MASK; \ |
||
106 | PAR[e][b] = val; \ |
||
107 | if (ERR_CONDITION) \ |
||
108 | goto err; \ |
||
109 | } \ |
||
110 | } \ |
||
111 | return 0; \ |
||
112 | err: \ |
||
113 | av_log(avctx, AV_LOG_ERROR, "illegal "#PAR"\n"); \ |
||
114 | return -1; \ |
||
115 | } |
||
116 | |||
117 | READ_PAR_DATA(iid, huff_offset[table_idx], 0, FFABS(ps->iid_par[e][b]) > 7 + 8 * ps->iid_quant) |
||
118 | READ_PAR_DATA(icc, huff_offset[table_idx], 0, ps->icc_par[e][b] > 7U) |
||
119 | READ_PAR_DATA(ipdopd, 0, 0x07, 0) |
||
120 | |||
121 | static int ps_read_extension_data(GetBitContext *gb, PSContext *ps, int ps_extension_id) |
||
122 | { |
||
123 | int e; |
||
124 | int count = get_bits_count(gb); |
||
125 | |||
126 | if (ps_extension_id) |
||
127 | return 0; |
||
128 | |||
129 | ps->enable_ipdopd = get_bits1(gb); |
||
130 | if (ps->enable_ipdopd) { |
||
131 | for (e = 0; e < ps->num_env; e++) { |
||
132 | int dt = get_bits1(gb); |
||
133 | read_ipdopd_data(NULL, gb, ps, ps->ipd_par, dt ? huff_ipd_dt : huff_ipd_df, e, dt); |
||
134 | dt = get_bits1(gb); |
||
135 | read_ipdopd_data(NULL, gb, ps, ps->opd_par, dt ? huff_opd_dt : huff_opd_df, e, dt); |
||
136 | } |
||
137 | } |
||
138 | skip_bits1(gb); //reserved_ps |
||
139 | return get_bits_count(gb) - count; |
||
140 | } |
||
141 | |||
142 | static void ipdopd_reset(int8_t *ipd_hist, int8_t *opd_hist) |
||
143 | { |
||
144 | int i; |
||
145 | for (i = 0; i < PS_MAX_NR_IPDOPD; i++) { |
||
146 | opd_hist[i] = 0; |
||
147 | ipd_hist[i] = 0; |
||
148 | } |
||
149 | } |
||
150 | |||
151 | int ff_ps_read_data(AVCodecContext *avctx, GetBitContext *gb_host, PSContext *ps, int bits_left) |
||
152 | { |
||
153 | int e; |
||
154 | int bit_count_start = get_bits_count(gb_host); |
||
155 | int header; |
||
156 | int bits_consumed; |
||
157 | GetBitContext gbc = *gb_host, *gb = &gbc; |
||
158 | |||
159 | header = get_bits1(gb); |
||
160 | if (header) { //enable_ps_header |
||
161 | ps->enable_iid = get_bits1(gb); |
||
162 | if (ps->enable_iid) { |
||
163 | int iid_mode = get_bits(gb, 3); |
||
164 | if (iid_mode > 5) { |
||
165 | av_log(avctx, AV_LOG_ERROR, "iid_mode %d is reserved.\n", |
||
166 | iid_mode); |
||
167 | goto err; |
||
168 | } |
||
169 | ps->nr_iid_par = nr_iidicc_par_tab[iid_mode]; |
||
170 | ps->iid_quant = iid_mode > 2; |
||
171 | ps->nr_ipdopd_par = nr_iidopd_par_tab[iid_mode]; |
||
172 | } |
||
173 | ps->enable_icc = get_bits1(gb); |
||
174 | if (ps->enable_icc) { |
||
175 | ps->icc_mode = get_bits(gb, 3); |
||
176 | if (ps->icc_mode > 5) { |
||
177 | av_log(avctx, AV_LOG_ERROR, "icc_mode %d is reserved.\n", |
||
178 | ps->icc_mode); |
||
179 | goto err; |
||
180 | } |
||
181 | ps->nr_icc_par = nr_iidicc_par_tab[ps->icc_mode]; |
||
182 | } |
||
183 | ps->enable_ext = get_bits1(gb); |
||
184 | } |
||
185 | |||
186 | ps->frame_class = get_bits1(gb); |
||
187 | ps->num_env_old = ps->num_env; |
||
188 | ps->num_env = num_env_tab[ps->frame_class][get_bits(gb, 2)]; |
||
189 | |||
190 | ps->border_position[0] = -1; |
||
191 | if (ps->frame_class) { |
||
192 | for (e = 1; e <= ps->num_env; e++) |
||
193 | ps->border_position[e] = get_bits(gb, 5); |
||
194 | } else |
||
195 | for (e = 1; e <= ps->num_env; e++) |
||
196 | ps->border_position[e] = (e * numQMFSlots >> ff_log2_tab[ps->num_env]) - 1; |
||
197 | |||
198 | if (ps->enable_iid) { |
||
199 | for (e = 0; e < ps->num_env; e++) { |
||
200 | int dt = get_bits1(gb); |
||
201 | if (read_iid_data(avctx, gb, ps, ps->iid_par, huff_iid[2*dt+ps->iid_quant], e, dt)) |
||
202 | goto err; |
||
203 | } |
||
204 | } else |
||
205 | memset(ps->iid_par, 0, sizeof(ps->iid_par)); |
||
206 | |||
207 | if (ps->enable_icc) |
||
208 | for (e = 0; e < ps->num_env; e++) { |
||
209 | int dt = get_bits1(gb); |
||
210 | if (read_icc_data(avctx, gb, ps, ps->icc_par, dt ? huff_icc_dt : huff_icc_df, e, dt)) |
||
211 | goto err; |
||
212 | } |
||
213 | else |
||
214 | memset(ps->icc_par, 0, sizeof(ps->icc_par)); |
||
215 | |||
216 | if (ps->enable_ext) { |
||
217 | int cnt = get_bits(gb, 4); |
||
218 | if (cnt == 15) { |
||
219 | cnt += get_bits(gb, 8); |
||
220 | } |
||
221 | cnt *= 8; |
||
222 | while (cnt > 7) { |
||
223 | int ps_extension_id = get_bits(gb, 2); |
||
224 | cnt -= 2 + ps_read_extension_data(gb, ps, ps_extension_id); |
||
225 | } |
||
226 | if (cnt < 0) { |
||
227 | av_log(avctx, AV_LOG_ERROR, "ps extension overflow %d\n", cnt); |
||
228 | goto err; |
||
229 | } |
||
230 | skip_bits(gb, cnt); |
||
231 | } |
||
232 | |||
233 | ps->enable_ipdopd &= !PS_BASELINE; |
||
234 | |||
235 | //Fix up envelopes |
||
236 | if (!ps->num_env || ps->border_position[ps->num_env] < numQMFSlots - 1) { |
||
237 | //Create a fake envelope |
||
238 | int source = ps->num_env ? ps->num_env - 1 : ps->num_env_old - 1; |
||
239 | int b; |
||
240 | if (source >= 0 && source != ps->num_env) { |
||
241 | if (ps->enable_iid) { |
||
242 | memcpy(ps->iid_par+ps->num_env, ps->iid_par+source, sizeof(ps->iid_par[0])); |
||
243 | } |
||
244 | if (ps->enable_icc) { |
||
245 | memcpy(ps->icc_par+ps->num_env, ps->icc_par+source, sizeof(ps->icc_par[0])); |
||
246 | } |
||
247 | if (ps->enable_ipdopd) { |
||
248 | memcpy(ps->ipd_par+ps->num_env, ps->ipd_par+source, sizeof(ps->ipd_par[0])); |
||
249 | memcpy(ps->opd_par+ps->num_env, ps->opd_par+source, sizeof(ps->opd_par[0])); |
||
250 | } |
||
251 | } |
||
252 | if (ps->enable_iid){ |
||
253 | for (b = 0; b < ps->nr_iid_par; b++) { |
||
254 | if (FFABS(ps->iid_par[ps->num_env][b]) > 7 + 8 * ps->iid_quant) { |
||
255 | av_log(avctx, AV_LOG_ERROR, "iid_par invalid\n"); |
||
256 | goto err; |
||
257 | } |
||
258 | } |
||
259 | } |
||
260 | if (ps->enable_icc){ |
||
261 | for (b = 0; b < ps->nr_iid_par; b++) { |
||
262 | if (ps->icc_par[ps->num_env][b] > 7U) { |
||
263 | av_log(avctx, AV_LOG_ERROR, "icc_par invalid\n"); |
||
264 | goto err; |
||
265 | } |
||
266 | } |
||
267 | } |
||
268 | ps->num_env++; |
||
269 | ps->border_position[ps->num_env] = numQMFSlots - 1; |
||
270 | } |
||
271 | |||
272 | |||
273 | ps->is34bands_old = ps->is34bands; |
||
274 | if (!PS_BASELINE && (ps->enable_iid || ps->enable_icc)) |
||
275 | ps->is34bands = (ps->enable_iid && ps->nr_iid_par == 34) || |
||
276 | (ps->enable_icc && ps->nr_icc_par == 34); |
||
277 | |||
278 | //Baseline |
||
279 | if (!ps->enable_ipdopd) { |
||
280 | memset(ps->ipd_par, 0, sizeof(ps->ipd_par)); |
||
281 | memset(ps->opd_par, 0, sizeof(ps->opd_par)); |
||
282 | } |
||
283 | |||
284 | if (header) |
||
285 | ps->start = 1; |
||
286 | |||
287 | bits_consumed = get_bits_count(gb) - bit_count_start; |
||
288 | if (bits_consumed <= bits_left) { |
||
289 | skip_bits_long(gb_host, bits_consumed); |
||
290 | return bits_consumed; |
||
291 | } |
||
292 | av_log(avctx, AV_LOG_ERROR, "Expected to read %d PS bits actually read %d.\n", bits_left, bits_consumed); |
||
293 | err: |
||
294 | ps->start = 0; |
||
295 | skip_bits_long(gb_host, bits_left); |
||
296 | memset(ps->iid_par, 0, sizeof(ps->iid_par)); |
||
297 | memset(ps->icc_par, 0, sizeof(ps->icc_par)); |
||
298 | memset(ps->ipd_par, 0, sizeof(ps->ipd_par)); |
||
299 | memset(ps->opd_par, 0, sizeof(ps->opd_par)); |
||
300 | return bits_left; |
||
301 | } |
||
302 | |||
303 | /** Split one subband into 2 subsubbands with a symmetric real filter. |
||
304 | * The filter must have its non-center even coefficients equal to zero. */ |
||
305 | static void hybrid2_re(float (*in)[2], float (*out)[32][2], const float filter[8], int len, int reverse) |
||
306 | { |
||
307 | int i, j; |
||
308 | for (i = 0; i < len; i++, in++) { |
||
309 | float re_in = filter[6] * in[6][0]; //real inphase |
||
310 | float re_op = 0.0f; //real out of phase |
||
311 | float im_in = filter[6] * in[6][1]; //imag inphase |
||
312 | float im_op = 0.0f; //imag out of phase |
||
313 | for (j = 0; j < 6; j += 2) { |
||
314 | re_op += filter[j+1] * (in[j+1][0] + in[12-j-1][0]); |
||
315 | im_op += filter[j+1] * (in[j+1][1] + in[12-j-1][1]); |
||
316 | } |
||
317 | out[ reverse][i][0] = re_in + re_op; |
||
318 | out[ reverse][i][1] = im_in + im_op; |
||
319 | out[!reverse][i][0] = re_in - re_op; |
||
320 | out[!reverse][i][1] = im_in - im_op; |
||
321 | } |
||
322 | } |
||
323 | |||
324 | /** Split one subband into 6 subsubbands with a complex filter */ |
||
325 | static void hybrid6_cx(PSDSPContext *dsp, float (*in)[2], float (*out)[32][2], const float (*filter)[8][2], int len) |
||
326 | { |
||
327 | int i; |
||
328 | int N = 8; |
||
329 | LOCAL_ALIGNED_16(float, temp, [8], [2]); |
||
330 | |||
331 | for (i = 0; i < len; i++, in++) { |
||
332 | dsp->hybrid_analysis(temp, in, filter, 1, N); |
||
333 | out[0][i][0] = temp[6][0]; |
||
334 | out[0][i][1] = temp[6][1]; |
||
335 | out[1][i][0] = temp[7][0]; |
||
336 | out[1][i][1] = temp[7][1]; |
||
337 | out[2][i][0] = temp[0][0]; |
||
338 | out[2][i][1] = temp[0][1]; |
||
339 | out[3][i][0] = temp[1][0]; |
||
340 | out[3][i][1] = temp[1][1]; |
||
341 | out[4][i][0] = temp[2][0] + temp[5][0]; |
||
342 | out[4][i][1] = temp[2][1] + temp[5][1]; |
||
343 | out[5][i][0] = temp[3][0] + temp[4][0]; |
||
344 | out[5][i][1] = temp[3][1] + temp[4][1]; |
||
345 | } |
||
346 | } |
||
347 | |||
348 | static void hybrid4_8_12_cx(PSDSPContext *dsp, float (*in)[2], float (*out)[32][2], const float (*filter)[8][2], int N, int len) |
||
349 | { |
||
350 | int i; |
||
351 | |||
352 | for (i = 0; i < len; i++, in++) { |
||
353 | dsp->hybrid_analysis(out[0] + i, in, filter, 32, N); |
||
354 | } |
||
355 | } |
||
356 | |||
357 | static void hybrid_analysis(PSDSPContext *dsp, float out[91][32][2], |
||
358 | float in[5][44][2], float L[2][38][64], |
||
359 | int is34, int len) |
||
360 | { |
||
361 | int i, j; |
||
362 | for (i = 0; i < 5; i++) { |
||
363 | for (j = 0; j < 38; j++) { |
||
364 | in[i][j+6][0] = L[0][j][i]; |
||
365 | in[i][j+6][1] = L[1][j][i]; |
||
366 | } |
||
367 | } |
||
368 | if (is34) { |
||
369 | hybrid4_8_12_cx(dsp, in[0], out, f34_0_12, 12, len); |
||
370 | hybrid4_8_12_cx(dsp, in[1], out+12, f34_1_8, 8, len); |
||
371 | hybrid4_8_12_cx(dsp, in[2], out+20, f34_2_4, 4, len); |
||
372 | hybrid4_8_12_cx(dsp, in[3], out+24, f34_2_4, 4, len); |
||
373 | hybrid4_8_12_cx(dsp, in[4], out+28, f34_2_4, 4, len); |
||
374 | dsp->hybrid_analysis_ileave(out + 27, L, 5, len); |
||
375 | } else { |
||
376 | hybrid6_cx(dsp, in[0], out, f20_0_8, len); |
||
377 | hybrid2_re(in[1], out+6, g1_Q2, len, 1); |
||
378 | hybrid2_re(in[2], out+8, g1_Q2, len, 0); |
||
379 | dsp->hybrid_analysis_ileave(out + 7, L, 3, len); |
||
380 | } |
||
381 | //update in_buf |
||
382 | for (i = 0; i < 5; i++) { |
||
383 | memcpy(in[i], in[i]+32, 6 * sizeof(in[i][0])); |
||
384 | } |
||
385 | } |
||
386 | |||
387 | static void hybrid_synthesis(PSDSPContext *dsp, float out[2][38][64], |
||
388 | float in[91][32][2], int is34, int len) |
||
389 | { |
||
390 | int i, n; |
||
391 | if (is34) { |
||
392 | for (n = 0; n < len; n++) { |
||
393 | memset(out[0][n], 0, 5*sizeof(out[0][n][0])); |
||
394 | memset(out[1][n], 0, 5*sizeof(out[1][n][0])); |
||
395 | for (i = 0; i < 12; i++) { |
||
396 | out[0][n][0] += in[ i][n][0]; |
||
397 | out[1][n][0] += in[ i][n][1]; |
||
398 | } |
||
399 | for (i = 0; i < 8; i++) { |
||
400 | out[0][n][1] += in[12+i][n][0]; |
||
401 | out[1][n][1] += in[12+i][n][1]; |
||
402 | } |
||
403 | for (i = 0; i < 4; i++) { |
||
404 | out[0][n][2] += in[20+i][n][0]; |
||
405 | out[1][n][2] += in[20+i][n][1]; |
||
406 | out[0][n][3] += in[24+i][n][0]; |
||
407 | out[1][n][3] += in[24+i][n][1]; |
||
408 | out[0][n][4] += in[28+i][n][0]; |
||
409 | out[1][n][4] += in[28+i][n][1]; |
||
410 | } |
||
411 | } |
||
412 | dsp->hybrid_synthesis_deint(out, in + 27, 5, len); |
||
413 | } else { |
||
414 | for (n = 0; n < len; n++) { |
||
415 | out[0][n][0] = in[0][n][0] + in[1][n][0] + in[2][n][0] + |
||
416 | in[3][n][0] + in[4][n][0] + in[5][n][0]; |
||
417 | out[1][n][0] = in[0][n][1] + in[1][n][1] + in[2][n][1] + |
||
418 | in[3][n][1] + in[4][n][1] + in[5][n][1]; |
||
419 | out[0][n][1] = in[6][n][0] + in[7][n][0]; |
||
420 | out[1][n][1] = in[6][n][1] + in[7][n][1]; |
||
421 | out[0][n][2] = in[8][n][0] + in[9][n][0]; |
||
422 | out[1][n][2] = in[8][n][1] + in[9][n][1]; |
||
423 | } |
||
424 | dsp->hybrid_synthesis_deint(out, in + 7, 3, len); |
||
425 | } |
||
426 | } |
||
427 | |||
428 | /// All-pass filter decay slope |
||
429 | #define DECAY_SLOPE 0.05f |
||
430 | /// Number of frequency bands that can be addressed by the parameter index, b(k) |
||
431 | static const int NR_PAR_BANDS[] = { 20, 34 }; |
||
432 | /// Number of frequency bands that can be addressed by the sub subband index, k |
||
433 | static const int NR_BANDS[] = { 71, 91 }; |
||
434 | /// Start frequency band for the all-pass filter decay slope |
||
435 | static const int DECAY_CUTOFF[] = { 10, 32 }; |
||
436 | /// Number of all-pass filer bands |
||
437 | static const int NR_ALLPASS_BANDS[] = { 30, 50 }; |
||
438 | /// First stereo band using the short one sample delay |
||
439 | static const int SHORT_DELAY_BAND[] = { 42, 62 }; |
||
440 | |||
441 | /** Table 8.46 */ |
||
442 | static void map_idx_10_to_20(int8_t *par_mapped, const int8_t *par, int full) |
||
443 | { |
||
444 | int b; |
||
445 | if (full) |
||
446 | b = 9; |
||
447 | else { |
||
448 | b = 4; |
||
449 | par_mapped[10] = 0; |
||
450 | } |
||
451 | for (; b >= 0; b--) { |
||
452 | par_mapped[2*b+1] = par_mapped[2*b] = par[b]; |
||
453 | } |
||
454 | } |
||
455 | |||
456 | static void map_idx_34_to_20(int8_t *par_mapped, const int8_t *par, int full) |
||
457 | { |
||
458 | par_mapped[ 0] = (2*par[ 0] + par[ 1]) / 3; |
||
459 | par_mapped[ 1] = ( par[ 1] + 2*par[ 2]) / 3; |
||
460 | par_mapped[ 2] = (2*par[ 3] + par[ 4]) / 3; |
||
461 | par_mapped[ 3] = ( par[ 4] + 2*par[ 5]) / 3; |
||
462 | par_mapped[ 4] = ( par[ 6] + par[ 7]) / 2; |
||
463 | par_mapped[ 5] = ( par[ 8] + par[ 9]) / 2; |
||
464 | par_mapped[ 6] = par[10]; |
||
465 | par_mapped[ 7] = par[11]; |
||
466 | par_mapped[ 8] = ( par[12] + par[13]) / 2; |
||
467 | par_mapped[ 9] = ( par[14] + par[15]) / 2; |
||
468 | par_mapped[10] = par[16]; |
||
469 | if (full) { |
||
470 | par_mapped[11] = par[17]; |
||
471 | par_mapped[12] = par[18]; |
||
472 | par_mapped[13] = par[19]; |
||
473 | par_mapped[14] = ( par[20] + par[21]) / 2; |
||
474 | par_mapped[15] = ( par[22] + par[23]) / 2; |
||
475 | par_mapped[16] = ( par[24] + par[25]) / 2; |
||
476 | par_mapped[17] = ( par[26] + par[27]) / 2; |
||
477 | par_mapped[18] = ( par[28] + par[29] + par[30] + par[31]) / 4; |
||
478 | par_mapped[19] = ( par[32] + par[33]) / 2; |
||
479 | } |
||
480 | } |
||
481 | |||
482 | static void map_val_34_to_20(float par[PS_MAX_NR_IIDICC]) |
||
483 | { |
||
484 | par[ 0] = (2*par[ 0] + par[ 1]) * 0.33333333f; |
||
485 | par[ 1] = ( par[ 1] + 2*par[ 2]) * 0.33333333f; |
||
486 | par[ 2] = (2*par[ 3] + par[ 4]) * 0.33333333f; |
||
487 | par[ 3] = ( par[ 4] + 2*par[ 5]) * 0.33333333f; |
||
488 | par[ 4] = ( par[ 6] + par[ 7]) * 0.5f; |
||
489 | par[ 5] = ( par[ 8] + par[ 9]) * 0.5f; |
||
490 | par[ 6] = par[10]; |
||
491 | par[ 7] = par[11]; |
||
492 | par[ 8] = ( par[12] + par[13]) * 0.5f; |
||
493 | par[ 9] = ( par[14] + par[15]) * 0.5f; |
||
494 | par[10] = par[16]; |
||
495 | par[11] = par[17]; |
||
496 | par[12] = par[18]; |
||
497 | par[13] = par[19]; |
||
498 | par[14] = ( par[20] + par[21]) * 0.5f; |
||
499 | par[15] = ( par[22] + par[23]) * 0.5f; |
||
500 | par[16] = ( par[24] + par[25]) * 0.5f; |
||
501 | par[17] = ( par[26] + par[27]) * 0.5f; |
||
502 | par[18] = ( par[28] + par[29] + par[30] + par[31]) * 0.25f; |
||
503 | par[19] = ( par[32] + par[33]) * 0.5f; |
||
504 | } |
||
505 | |||
506 | static void map_idx_10_to_34(int8_t *par_mapped, const int8_t *par, int full) |
||
507 | { |
||
508 | if (full) { |
||
509 | par_mapped[33] = par[9]; |
||
510 | par_mapped[32] = par[9]; |
||
511 | par_mapped[31] = par[9]; |
||
512 | par_mapped[30] = par[9]; |
||
513 | par_mapped[29] = par[9]; |
||
514 | par_mapped[28] = par[9]; |
||
515 | par_mapped[27] = par[8]; |
||
516 | par_mapped[26] = par[8]; |
||
517 | par_mapped[25] = par[8]; |
||
518 | par_mapped[24] = par[8]; |
||
519 | par_mapped[23] = par[7]; |
||
520 | par_mapped[22] = par[7]; |
||
521 | par_mapped[21] = par[7]; |
||
522 | par_mapped[20] = par[7]; |
||
523 | par_mapped[19] = par[6]; |
||
524 | par_mapped[18] = par[6]; |
||
525 | par_mapped[17] = par[5]; |
||
526 | par_mapped[16] = par[5]; |
||
527 | } else { |
||
528 | par_mapped[16] = 0; |
||
529 | } |
||
530 | par_mapped[15] = par[4]; |
||
531 | par_mapped[14] = par[4]; |
||
532 | par_mapped[13] = par[4]; |
||
533 | par_mapped[12] = par[4]; |
||
534 | par_mapped[11] = par[3]; |
||
535 | par_mapped[10] = par[3]; |
||
536 | par_mapped[ 9] = par[2]; |
||
537 | par_mapped[ 8] = par[2]; |
||
538 | par_mapped[ 7] = par[2]; |
||
539 | par_mapped[ 6] = par[2]; |
||
540 | par_mapped[ 5] = par[1]; |
||
541 | par_mapped[ 4] = par[1]; |
||
542 | par_mapped[ 3] = par[1]; |
||
543 | par_mapped[ 2] = par[0]; |
||
544 | par_mapped[ 1] = par[0]; |
||
545 | par_mapped[ 0] = par[0]; |
||
546 | } |
||
547 | |||
548 | static void map_idx_20_to_34(int8_t *par_mapped, const int8_t *par, int full) |
||
549 | { |
||
550 | if (full) { |
||
551 | par_mapped[33] = par[19]; |
||
552 | par_mapped[32] = par[19]; |
||
553 | par_mapped[31] = par[18]; |
||
554 | par_mapped[30] = par[18]; |
||
555 | par_mapped[29] = par[18]; |
||
556 | par_mapped[28] = par[18]; |
||
557 | par_mapped[27] = par[17]; |
||
558 | par_mapped[26] = par[17]; |
||
559 | par_mapped[25] = par[16]; |
||
560 | par_mapped[24] = par[16]; |
||
561 | par_mapped[23] = par[15]; |
||
562 | par_mapped[22] = par[15]; |
||
563 | par_mapped[21] = par[14]; |
||
564 | par_mapped[20] = par[14]; |
||
565 | par_mapped[19] = par[13]; |
||
566 | par_mapped[18] = par[12]; |
||
567 | par_mapped[17] = par[11]; |
||
568 | } |
||
569 | par_mapped[16] = par[10]; |
||
570 | par_mapped[15] = par[ 9]; |
||
571 | par_mapped[14] = par[ 9]; |
||
572 | par_mapped[13] = par[ 8]; |
||
573 | par_mapped[12] = par[ 8]; |
||
574 | par_mapped[11] = par[ 7]; |
||
575 | par_mapped[10] = par[ 6]; |
||
576 | par_mapped[ 9] = par[ 5]; |
||
577 | par_mapped[ 8] = par[ 5]; |
||
578 | par_mapped[ 7] = par[ 4]; |
||
579 | par_mapped[ 6] = par[ 4]; |
||
580 | par_mapped[ 5] = par[ 3]; |
||
581 | par_mapped[ 4] = (par[ 2] + par[ 3]) / 2; |
||
582 | par_mapped[ 3] = par[ 2]; |
||
583 | par_mapped[ 2] = par[ 1]; |
||
584 | par_mapped[ 1] = (par[ 0] + par[ 1]) / 2; |
||
585 | par_mapped[ 0] = par[ 0]; |
||
586 | } |
||
587 | |||
588 | static void map_val_20_to_34(float par[PS_MAX_NR_IIDICC]) |
||
589 | { |
||
590 | par[33] = par[19]; |
||
591 | par[32] = par[19]; |
||
592 | par[31] = par[18]; |
||
593 | par[30] = par[18]; |
||
594 | par[29] = par[18]; |
||
595 | par[28] = par[18]; |
||
596 | par[27] = par[17]; |
||
597 | par[26] = par[17]; |
||
598 | par[25] = par[16]; |
||
599 | par[24] = par[16]; |
||
600 | par[23] = par[15]; |
||
601 | par[22] = par[15]; |
||
602 | par[21] = par[14]; |
||
603 | par[20] = par[14]; |
||
604 | par[19] = par[13]; |
||
605 | par[18] = par[12]; |
||
606 | par[17] = par[11]; |
||
607 | par[16] = par[10]; |
||
608 | par[15] = par[ 9]; |
||
609 | par[14] = par[ 9]; |
||
610 | par[13] = par[ 8]; |
||
611 | par[12] = par[ 8]; |
||
612 | par[11] = par[ 7]; |
||
613 | par[10] = par[ 6]; |
||
614 | par[ 9] = par[ 5]; |
||
615 | par[ 8] = par[ 5]; |
||
616 | par[ 7] = par[ 4]; |
||
617 | par[ 6] = par[ 4]; |
||
618 | par[ 5] = par[ 3]; |
||
619 | par[ 4] = (par[ 2] + par[ 3]) * 0.5f; |
||
620 | par[ 3] = par[ 2]; |
||
621 | par[ 2] = par[ 1]; |
||
622 | par[ 1] = (par[ 0] + par[ 1]) * 0.5f; |
||
623 | } |
||
624 | |||
625 | static void decorrelation(PSContext *ps, float (*out)[32][2], const float (*s)[32][2], int is34) |
||
626 | { |
||
627 | LOCAL_ALIGNED_16(float, power, [34], [PS_QMF_TIME_SLOTS]); |
||
628 | LOCAL_ALIGNED_16(float, transient_gain, [34], [PS_QMF_TIME_SLOTS]); |
||
629 | float *peak_decay_nrg = ps->peak_decay_nrg; |
||
630 | float *power_smooth = ps->power_smooth; |
||
631 | float *peak_decay_diff_smooth = ps->peak_decay_diff_smooth; |
||
632 | float (*delay)[PS_QMF_TIME_SLOTS + PS_MAX_DELAY][2] = ps->delay; |
||
633 | float (*ap_delay)[PS_AP_LINKS][PS_QMF_TIME_SLOTS + PS_MAX_AP_DELAY][2] = ps->ap_delay; |
||
634 | const int8_t *k_to_i = is34 ? k_to_i_34 : k_to_i_20; |
||
635 | const float peak_decay_factor = 0.76592833836465f; |
||
636 | const float transient_impact = 1.5f; |
||
637 | const float a_smooth = 0.25f; ///< Smoothing coefficient |
||
638 | int i, k, m, n; |
||
639 | int n0 = 0, nL = 32; |
||
640 | |||
641 | memset(power, 0, 34 * sizeof(*power)); |
||
642 | |||
643 | if (is34 != ps->is34bands_old) { |
||
644 | memset(ps->peak_decay_nrg, 0, sizeof(ps->peak_decay_nrg)); |
||
645 | memset(ps->power_smooth, 0, sizeof(ps->power_smooth)); |
||
646 | memset(ps->peak_decay_diff_smooth, 0, sizeof(ps->peak_decay_diff_smooth)); |
||
647 | memset(ps->delay, 0, sizeof(ps->delay)); |
||
648 | memset(ps->ap_delay, 0, sizeof(ps->ap_delay)); |
||
649 | } |
||
650 | |||
651 | for (k = 0; k < NR_BANDS[is34]; k++) { |
||
652 | int i = k_to_i[k]; |
||
653 | ps->dsp.add_squares(power[i], s[k], nL - n0); |
||
654 | } |
||
655 | |||
656 | //Transient detection |
||
657 | for (i = 0; i < NR_PAR_BANDS[is34]; i++) { |
||
658 | for (n = n0; n < nL; n++) { |
||
659 | float decayed_peak = peak_decay_factor * peak_decay_nrg[i]; |
||
660 | float denom; |
||
661 | peak_decay_nrg[i] = FFMAX(decayed_peak, power[i][n]); |
||
662 | power_smooth[i] += a_smooth * (power[i][n] - power_smooth[i]); |
||
663 | peak_decay_diff_smooth[i] += a_smooth * (peak_decay_nrg[i] - power[i][n] - peak_decay_diff_smooth[i]); |
||
664 | denom = transient_impact * peak_decay_diff_smooth[i]; |
||
665 | transient_gain[i][n] = (denom > power_smooth[i]) ? |
||
666 | power_smooth[i] / denom : 1.0f; |
||
667 | } |
||
668 | } |
||
669 | |||
670 | //Decorrelation and transient reduction |
||
671 | // PS_AP_LINKS - 1 |
||
672 | // ----- |
||
673 | // | | Q_fract_allpass[k][m]*z^-link_delay[m] - a[m]*g_decay_slope[k] |
||
674 | //H[k][z] = z^-2 * phi_fract[k] * | | ---------------------------------------------------------------- |
||
675 | // | | 1 - a[m]*g_decay_slope[k]*Q_fract_allpass[k][m]*z^-link_delay[m] |
||
676 | // m = 0 |
||
677 | //d[k][z] (out) = transient_gain_mapped[k][z] * H[k][z] * s[k][z] |
||
678 | for (k = 0; k < NR_ALLPASS_BANDS[is34]; k++) { |
||
679 | int b = k_to_i[k]; |
||
680 | float g_decay_slope = 1.f - DECAY_SLOPE * (k - DECAY_CUTOFF[is34]); |
||
681 | g_decay_slope = av_clipf(g_decay_slope, 0.f, 1.f); |
||
682 | memcpy(delay[k], delay[k]+nL, PS_MAX_DELAY*sizeof(delay[k][0])); |
||
683 | memcpy(delay[k]+PS_MAX_DELAY, s[k], numQMFSlots*sizeof(delay[k][0])); |
||
684 | for (m = 0; m < PS_AP_LINKS; m++) { |
||
685 | memcpy(ap_delay[k][m], ap_delay[k][m]+numQMFSlots, 5*sizeof(ap_delay[k][m][0])); |
||
686 | } |
||
687 | ps->dsp.decorrelate(out[k], delay[k] + PS_MAX_DELAY - 2, ap_delay[k], |
||
688 | phi_fract[is34][k], Q_fract_allpass[is34][k], |
||
689 | transient_gain[b], g_decay_slope, nL - n0); |
||
690 | } |
||
691 | for (; k < SHORT_DELAY_BAND[is34]; k++) { |
||
692 | int i = k_to_i[k]; |
||
693 | memcpy(delay[k], delay[k]+nL, PS_MAX_DELAY*sizeof(delay[k][0])); |
||
694 | memcpy(delay[k]+PS_MAX_DELAY, s[k], numQMFSlots*sizeof(delay[k][0])); |
||
695 | //H = delay 14 |
||
696 | ps->dsp.mul_pair_single(out[k], delay[k] + PS_MAX_DELAY - 14, |
||
697 | transient_gain[i], nL - n0); |
||
698 | } |
||
699 | for (; k < NR_BANDS[is34]; k++) { |
||
700 | int i = k_to_i[k]; |
||
701 | memcpy(delay[k], delay[k]+nL, PS_MAX_DELAY*sizeof(delay[k][0])); |
||
702 | memcpy(delay[k]+PS_MAX_DELAY, s[k], numQMFSlots*sizeof(delay[k][0])); |
||
703 | //H = delay 1 |
||
704 | ps->dsp.mul_pair_single(out[k], delay[k] + PS_MAX_DELAY - 1, |
||
705 | transient_gain[i], nL - n0); |
||
706 | } |
||
707 | } |
||
708 | |||
709 | static void remap34(int8_t (**p_par_mapped)[PS_MAX_NR_IIDICC], |
||
710 | int8_t (*par)[PS_MAX_NR_IIDICC], |
||
711 | int num_par, int num_env, int full) |
||
712 | { |
||
713 | int8_t (*par_mapped)[PS_MAX_NR_IIDICC] = *p_par_mapped; |
||
714 | int e; |
||
715 | if (num_par == 20 || num_par == 11) { |
||
716 | for (e = 0; e < num_env; e++) { |
||
717 | map_idx_20_to_34(par_mapped[e], par[e], full); |
||
718 | } |
||
719 | } else if (num_par == 10 || num_par == 5) { |
||
720 | for (e = 0; e < num_env; e++) { |
||
721 | map_idx_10_to_34(par_mapped[e], par[e], full); |
||
722 | } |
||
723 | } else { |
||
724 | *p_par_mapped = par; |
||
725 | } |
||
726 | } |
||
727 | |||
728 | static void remap20(int8_t (**p_par_mapped)[PS_MAX_NR_IIDICC], |
||
729 | int8_t (*par)[PS_MAX_NR_IIDICC], |
||
730 | int num_par, int num_env, int full) |
||
731 | { |
||
732 | int8_t (*par_mapped)[PS_MAX_NR_IIDICC] = *p_par_mapped; |
||
733 | int e; |
||
734 | if (num_par == 34 || num_par == 17) { |
||
735 | for (e = 0; e < num_env; e++) { |
||
736 | map_idx_34_to_20(par_mapped[e], par[e], full); |
||
737 | } |
||
738 | } else if (num_par == 10 || num_par == 5) { |
||
739 | for (e = 0; e < num_env; e++) { |
||
740 | map_idx_10_to_20(par_mapped[e], par[e], full); |
||
741 | } |
||
742 | } else { |
||
743 | *p_par_mapped = par; |
||
744 | } |
||
745 | } |
||
746 | |||
747 | static void stereo_processing(PSContext *ps, float (*l)[32][2], float (*r)[32][2], int is34) |
||
748 | { |
||
749 | int e, b, k; |
||
750 | |||
751 | float (*H11)[PS_MAX_NUM_ENV+1][PS_MAX_NR_IIDICC] = ps->H11; |
||
752 | float (*H12)[PS_MAX_NUM_ENV+1][PS_MAX_NR_IIDICC] = ps->H12; |
||
753 | float (*H21)[PS_MAX_NUM_ENV+1][PS_MAX_NR_IIDICC] = ps->H21; |
||
754 | float (*H22)[PS_MAX_NUM_ENV+1][PS_MAX_NR_IIDICC] = ps->H22; |
||
755 | int8_t *opd_hist = ps->opd_hist; |
||
756 | int8_t *ipd_hist = ps->ipd_hist; |
||
757 | int8_t iid_mapped_buf[PS_MAX_NUM_ENV][PS_MAX_NR_IIDICC]; |
||
758 | int8_t icc_mapped_buf[PS_MAX_NUM_ENV][PS_MAX_NR_IIDICC]; |
||
759 | int8_t ipd_mapped_buf[PS_MAX_NUM_ENV][PS_MAX_NR_IIDICC]; |
||
760 | int8_t opd_mapped_buf[PS_MAX_NUM_ENV][PS_MAX_NR_IIDICC]; |
||
761 | int8_t (*iid_mapped)[PS_MAX_NR_IIDICC] = iid_mapped_buf; |
||
762 | int8_t (*icc_mapped)[PS_MAX_NR_IIDICC] = icc_mapped_buf; |
||
763 | int8_t (*ipd_mapped)[PS_MAX_NR_IIDICC] = ipd_mapped_buf; |
||
764 | int8_t (*opd_mapped)[PS_MAX_NR_IIDICC] = opd_mapped_buf; |
||
765 | const int8_t *k_to_i = is34 ? k_to_i_34 : k_to_i_20; |
||
766 | const float (*H_LUT)[8][4] = (PS_BASELINE || ps->icc_mode < 3) ? HA : HB; |
||
767 | |||
768 | //Remapping |
||
769 | if (ps->num_env_old) { |
||
770 | memcpy(H11[0][0], H11[0][ps->num_env_old], PS_MAX_NR_IIDICC*sizeof(H11[0][0][0])); |
||
771 | memcpy(H11[1][0], H11[1][ps->num_env_old], PS_MAX_NR_IIDICC*sizeof(H11[1][0][0])); |
||
772 | memcpy(H12[0][0], H12[0][ps->num_env_old], PS_MAX_NR_IIDICC*sizeof(H12[0][0][0])); |
||
773 | memcpy(H12[1][0], H12[1][ps->num_env_old], PS_MAX_NR_IIDICC*sizeof(H12[1][0][0])); |
||
774 | memcpy(H21[0][0], H21[0][ps->num_env_old], PS_MAX_NR_IIDICC*sizeof(H21[0][0][0])); |
||
775 | memcpy(H21[1][0], H21[1][ps->num_env_old], PS_MAX_NR_IIDICC*sizeof(H21[1][0][0])); |
||
776 | memcpy(H22[0][0], H22[0][ps->num_env_old], PS_MAX_NR_IIDICC*sizeof(H22[0][0][0])); |
||
777 | memcpy(H22[1][0], H22[1][ps->num_env_old], PS_MAX_NR_IIDICC*sizeof(H22[1][0][0])); |
||
778 | } |
||
779 | |||
780 | if (is34) { |
||
781 | remap34(&iid_mapped, ps->iid_par, ps->nr_iid_par, ps->num_env, 1); |
||
782 | remap34(&icc_mapped, ps->icc_par, ps->nr_icc_par, ps->num_env, 1); |
||
783 | if (ps->enable_ipdopd) { |
||
784 | remap34(&ipd_mapped, ps->ipd_par, ps->nr_ipdopd_par, ps->num_env, 0); |
||
785 | remap34(&opd_mapped, ps->opd_par, ps->nr_ipdopd_par, ps->num_env, 0); |
||
786 | } |
||
787 | if (!ps->is34bands_old) { |
||
788 | map_val_20_to_34(H11[0][0]); |
||
789 | map_val_20_to_34(H11[1][0]); |
||
790 | map_val_20_to_34(H12[0][0]); |
||
791 | map_val_20_to_34(H12[1][0]); |
||
792 | map_val_20_to_34(H21[0][0]); |
||
793 | map_val_20_to_34(H21[1][0]); |
||
794 | map_val_20_to_34(H22[0][0]); |
||
795 | map_val_20_to_34(H22[1][0]); |
||
796 | ipdopd_reset(ipd_hist, opd_hist); |
||
797 | } |
||
798 | } else { |
||
799 | remap20(&iid_mapped, ps->iid_par, ps->nr_iid_par, ps->num_env, 1); |
||
800 | remap20(&icc_mapped, ps->icc_par, ps->nr_icc_par, ps->num_env, 1); |
||
801 | if (ps->enable_ipdopd) { |
||
802 | remap20(&ipd_mapped, ps->ipd_par, ps->nr_ipdopd_par, ps->num_env, 0); |
||
803 | remap20(&opd_mapped, ps->opd_par, ps->nr_ipdopd_par, ps->num_env, 0); |
||
804 | } |
||
805 | if (ps->is34bands_old) { |
||
806 | map_val_34_to_20(H11[0][0]); |
||
807 | map_val_34_to_20(H11[1][0]); |
||
808 | map_val_34_to_20(H12[0][0]); |
||
809 | map_val_34_to_20(H12[1][0]); |
||
810 | map_val_34_to_20(H21[0][0]); |
||
811 | map_val_34_to_20(H21[1][0]); |
||
812 | map_val_34_to_20(H22[0][0]); |
||
813 | map_val_34_to_20(H22[1][0]); |
||
814 | ipdopd_reset(ipd_hist, opd_hist); |
||
815 | } |
||
816 | } |
||
817 | |||
818 | //Mixing |
||
819 | for (e = 0; e < ps->num_env; e++) { |
||
820 | for (b = 0; b < NR_PAR_BANDS[is34]; b++) { |
||
821 | float h11, h12, h21, h22; |
||
822 | h11 = H_LUT[iid_mapped[e][b] + 7 + 23 * ps->iid_quant][icc_mapped[e][b]][0]; |
||
823 | h12 = H_LUT[iid_mapped[e][b] + 7 + 23 * ps->iid_quant][icc_mapped[e][b]][1]; |
||
824 | h21 = H_LUT[iid_mapped[e][b] + 7 + 23 * ps->iid_quant][icc_mapped[e][b]][2]; |
||
825 | h22 = H_LUT[iid_mapped[e][b] + 7 + 23 * ps->iid_quant][icc_mapped[e][b]][3]; |
||
826 | |||
827 | if (!PS_BASELINE && ps->enable_ipdopd && 2*b <= NR_PAR_BANDS[is34]) { |
||
828 | //The spec say says to only run this smoother when enable_ipdopd |
||
829 | //is set but the reference decoder appears to run it constantly |
||
830 | float h11i, h12i, h21i, h22i; |
||
831 | float ipd_adj_re, ipd_adj_im; |
||
832 | int opd_idx = opd_hist[b] * 8 + opd_mapped[e][b]; |
||
833 | int ipd_idx = ipd_hist[b] * 8 + ipd_mapped[e][b]; |
||
834 | float opd_re = pd_re_smooth[opd_idx]; |
||
835 | float opd_im = pd_im_smooth[opd_idx]; |
||
836 | float ipd_re = pd_re_smooth[ipd_idx]; |
||
837 | float ipd_im = pd_im_smooth[ipd_idx]; |
||
838 | opd_hist[b] = opd_idx & 0x3F; |
||
839 | ipd_hist[b] = ipd_idx & 0x3F; |
||
840 | |||
841 | ipd_adj_re = opd_re*ipd_re + opd_im*ipd_im; |
||
842 | ipd_adj_im = opd_im*ipd_re - opd_re*ipd_im; |
||
843 | h11i = h11 * opd_im; |
||
844 | h11 = h11 * opd_re; |
||
845 | h12i = h12 * ipd_adj_im; |
||
846 | h12 = h12 * ipd_adj_re; |
||
847 | h21i = h21 * opd_im; |
||
848 | h21 = h21 * opd_re; |
||
849 | h22i = h22 * ipd_adj_im; |
||
850 | h22 = h22 * ipd_adj_re; |
||
851 | H11[1][e+1][b] = h11i; |
||
852 | H12[1][e+1][b] = h12i; |
||
853 | H21[1][e+1][b] = h21i; |
||
854 | H22[1][e+1][b] = h22i; |
||
855 | } |
||
856 | H11[0][e+1][b] = h11; |
||
857 | H12[0][e+1][b] = h12; |
||
858 | H21[0][e+1][b] = h21; |
||
859 | H22[0][e+1][b] = h22; |
||
860 | } |
||
861 | for (k = 0; k < NR_BANDS[is34]; k++) { |
||
862 | float h[2][4]; |
||
863 | float h_step[2][4]; |
||
864 | int start = ps->border_position[e]; |
||
865 | int stop = ps->border_position[e+1]; |
||
866 | float width = 1.f / (stop - start); |
||
867 | b = k_to_i[k]; |
||
868 | h[0][0] = H11[0][e][b]; |
||
869 | h[0][1] = H12[0][e][b]; |
||
870 | h[0][2] = H21[0][e][b]; |
||
871 | h[0][3] = H22[0][e][b]; |
||
872 | if (!PS_BASELINE && ps->enable_ipdopd) { |
||
873 | //Is this necessary? ps_04_new seems unchanged |
||
874 | if ((is34 && k <= 13 && k >= 9) || (!is34 && k <= 1)) { |
||
875 | h[1][0] = -H11[1][e][b]; |
||
876 | h[1][1] = -H12[1][e][b]; |
||
877 | h[1][2] = -H21[1][e][b]; |
||
878 | h[1][3] = -H22[1][e][b]; |
||
879 | } else { |
||
880 | h[1][0] = H11[1][e][b]; |
||
881 | h[1][1] = H12[1][e][b]; |
||
882 | h[1][2] = H21[1][e][b]; |
||
883 | h[1][3] = H22[1][e][b]; |
||
884 | } |
||
885 | } |
||
886 | //Interpolation |
||
887 | h_step[0][0] = (H11[0][e+1][b] - h[0][0]) * width; |
||
888 | h_step[0][1] = (H12[0][e+1][b] - h[0][1]) * width; |
||
889 | h_step[0][2] = (H21[0][e+1][b] - h[0][2]) * width; |
||
890 | h_step[0][3] = (H22[0][e+1][b] - h[0][3]) * width; |
||
891 | if (!PS_BASELINE && ps->enable_ipdopd) { |
||
892 | h_step[1][0] = (H11[1][e+1][b] - h[1][0]) * width; |
||
893 | h_step[1][1] = (H12[1][e+1][b] - h[1][1]) * width; |
||
894 | h_step[1][2] = (H21[1][e+1][b] - h[1][2]) * width; |
||
895 | h_step[1][3] = (H22[1][e+1][b] - h[1][3]) * width; |
||
896 | } |
||
897 | ps->dsp.stereo_interpolate[!PS_BASELINE && ps->enable_ipdopd]( |
||
898 | l[k] + start + 1, r[k] + start + 1, |
||
899 | h, h_step, stop - start); |
||
900 | } |
||
901 | } |
||
902 | } |
||
903 | |||
904 | int ff_ps_apply(AVCodecContext *avctx, PSContext *ps, float L[2][38][64], float R[2][38][64], int top) |
||
905 | { |
||
906 | LOCAL_ALIGNED_16(float, Lbuf, [91], [32][2]); |
||
907 | LOCAL_ALIGNED_16(float, Rbuf, [91], [32][2]); |
||
908 | const int len = 32; |
||
909 | int is34 = ps->is34bands; |
||
910 | |||
911 | top += NR_BANDS[is34] - 64; |
||
912 | memset(ps->delay+top, 0, (NR_BANDS[is34] - top)*sizeof(ps->delay[0])); |
||
913 | if (top < NR_ALLPASS_BANDS[is34]) |
||
914 | memset(ps->ap_delay + top, 0, (NR_ALLPASS_BANDS[is34] - top)*sizeof(ps->ap_delay[0])); |
||
915 | |||
916 | hybrid_analysis(&ps->dsp, Lbuf, ps->in_buf, L, is34, len); |
||
917 | decorrelation(ps, Rbuf, Lbuf, is34); |
||
918 | stereo_processing(ps, Lbuf, Rbuf, is34); |
||
919 | hybrid_synthesis(&ps->dsp, L, Lbuf, is34, len); |
||
920 | hybrid_synthesis(&ps->dsp, R, Rbuf, is34, len); |
||
921 | |||
922 | return 0; |
||
923 | } |
||
924 | |||
925 | #define PS_INIT_VLC_STATIC(num, size) \ |
||
926 | INIT_VLC_STATIC(&vlc_ps[num], 9, ps_tmp[num].table_size / ps_tmp[num].elem_size, \ |
||
927 | ps_tmp[num].ps_bits, 1, 1, \ |
||
928 | ps_tmp[num].ps_codes, ps_tmp[num].elem_size, ps_tmp[num].elem_size, \ |
||
929 | size); |
||
930 | |||
931 | #define PS_VLC_ROW(name) \ |
||
932 | { name ## _codes, name ## _bits, sizeof(name ## _codes), sizeof(name ## _codes[0]) } |
||
933 | |||
934 | av_cold void ff_ps_init(void) { |
||
935 | // Syntax initialization |
||
936 | static const struct { |
||
937 | const void *ps_codes, *ps_bits; |
||
938 | const unsigned int table_size, elem_size; |
||
939 | } ps_tmp[] = { |
||
940 | PS_VLC_ROW(huff_iid_df1), |
||
941 | PS_VLC_ROW(huff_iid_dt1), |
||
942 | PS_VLC_ROW(huff_iid_df0), |
||
943 | PS_VLC_ROW(huff_iid_dt0), |
||
944 | PS_VLC_ROW(huff_icc_df), |
||
945 | PS_VLC_ROW(huff_icc_dt), |
||
946 | PS_VLC_ROW(huff_ipd_df), |
||
947 | PS_VLC_ROW(huff_ipd_dt), |
||
948 | PS_VLC_ROW(huff_opd_df), |
||
949 | PS_VLC_ROW(huff_opd_dt), |
||
950 | }; |
||
951 | |||
952 | PS_INIT_VLC_STATIC(0, 1544); |
||
953 | PS_INIT_VLC_STATIC(1, 832); |
||
954 | PS_INIT_VLC_STATIC(2, 1024); |
||
955 | PS_INIT_VLC_STATIC(3, 1036); |
||
956 | PS_INIT_VLC_STATIC(4, 544); |
||
957 | PS_INIT_VLC_STATIC(5, 544); |
||
958 | PS_INIT_VLC_STATIC(6, 512); |
||
959 | PS_INIT_VLC_STATIC(7, 512); |
||
960 | PS_INIT_VLC_STATIC(8, 512); |
||
961 | PS_INIT_VLC_STATIC(9, 512); |
||
962 | |||
963 | ps_tableinit(); |
||
964 | } |
||
965 | |||
966 | av_cold void ff_ps_ctx_init(PSContext *ps) |
||
967 | { |
||
968 | ff_psdsp_init(&ps->dsp); |
||
969 | }>=>=>>=>>>>>>>>>>>>>>>>>>>>>>>>>>>>=>>>>>>>=>=>>>>>>>> |