1 /******************************************************************************
2  *
3  * Copyright (C) 2018 The Android Open Source Project
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at:
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *****************************************************************************
18  * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19 */
20 #include <stdlib.h>
21 #include <stdio.h>
22 #include <math.h>
23 #include <string.h>
24 
25 #include "impd_type_def.h"
26 #include "impd_error_standards.h"
27 #include "impd_drc_extr_delta_coded_info.h"
28 #include "impd_drc_common.h"
29 #include "impd_drc_struct.h"
30 #include "impd_parametric_drc_dec.h"
31 #include "impd_drc_gain_dec.h"
32 #include "impd_drc_filter_bank.h"
33 #include "impd_drc_multi_band.h"
34 #include "impd_drc_interface.h"
35 #include "impd_drc_gain_dec.h"
36 #include "impd_drc_eq.h"
37 #include "impd_drc_process_audio.h"
38 #include "impd_drc_gain_decoder.h"
39 #include "impd_drc_dec.h"
40 
impd_init_drc_decode(WORD32 frame_size,WORD32 sample_rate,WORD32 gain_delay_samples,WORD32 delay_mode,WORD32 sub_band_domain_mode,ia_drc_gain_dec_struct * p_drc_gain_dec_structs)41 IA_ERRORCODE impd_init_drc_decode(
42     WORD32 frame_size, WORD32 sample_rate, WORD32 gain_delay_samples,
43     WORD32 delay_mode, WORD32 sub_band_domain_mode,
44     ia_drc_gain_dec_struct* p_drc_gain_dec_structs) {
45   IA_ERRORCODE err_code = IA_NO_ERROR;
46 
47   err_code = impd_init_drc_params(
48       frame_size, sample_rate, gain_delay_samples, delay_mode,
49       sub_band_domain_mode, &p_drc_gain_dec_structs->ia_drc_params_struct);
50 
51   if (err_code != IA_NO_ERROR) return (err_code);
52 
53   err_code = impd_init_parametric_drc(
54       p_drc_gain_dec_structs->ia_drc_params_struct.drc_frame_size, sample_rate,
55       sub_band_domain_mode, &p_drc_gain_dec_structs->parametricdrc_params);
56 
57   if (err_code != IA_NO_ERROR) return (err_code);
58 
59   return err_code;
60 }
61 
impd_init_drc_decode_post_config(WORD32 audio_num_chan,WORD32 * drc_set_id_processed,WORD32 * downmix_id_processed,WORD32 num_sets_processed,WORD32 eq_set_id_processed,ia_drc_gain_dec_struct * p_drc_gain_dec_structs,ia_drc_config * pstr_drc_config,ia_drc_loudness_info_set_struct * pstr_loudness_info,pVOID * mem_ptr)62 IA_ERRORCODE impd_init_drc_decode_post_config(
63     WORD32 audio_num_chan, WORD32* drc_set_id_processed,
64     WORD32* downmix_id_processed, WORD32 num_sets_processed,
65     WORD32 eq_set_id_processed,
66 
67     ia_drc_gain_dec_struct* p_drc_gain_dec_structs,
68     ia_drc_config* pstr_drc_config,
69     ia_drc_loudness_info_set_struct* pstr_loudness_info, pVOID* mem_ptr) {
70   IA_ERRORCODE err_code = 0;
71   WORD32 i, j, k, maxMultibandAudioSignalCount = 0;
72   ia_drc_params_struct* p_drc_params_struct =
73       &p_drc_gain_dec_structs->ia_drc_params_struct;
74   ia_audio_in_out_buf* p_audio_in_out_buf =
75       &p_drc_gain_dec_structs->audio_in_out_buf;
76 
77   for (i = 0; i < num_sets_processed; i++) {
78     err_code = impd_init_selected_drc_set(
79         pstr_drc_config, p_drc_params_struct,
80         &p_drc_gain_dec_structs->parametricdrc_params, audio_num_chan,
81         drc_set_id_processed[i], downmix_id_processed[i],
82         &p_drc_gain_dec_structs->ia_filter_banks_struct,
83         &p_drc_gain_dec_structs->str_overlap_params,
84         p_drc_gain_dec_structs->shape_filter_block);
85     if (err_code) return (err_code);
86   }
87 
88   p_drc_gain_dec_structs->audio_num_chan = audio_num_chan;
89   p_drc_gain_dec_structs->ia_drc_params_struct.audio_delay_samples =
90       p_drc_gain_dec_structs->ia_drc_params_struct.parametric_drc_delay;
91   if (pstr_drc_config->str_drc_config_ext.parametric_drc_present) {
92     err_code = impd_init_parametric_drc_after_config(
93         pstr_drc_config, pstr_loudness_info,
94         &p_drc_gain_dec_structs->parametricdrc_params, mem_ptr);
95     if (err_code) return (err_code);
96   }
97 
98   p_audio_in_out_buf->audio_num_chan = audio_num_chan;
99   p_audio_in_out_buf->audio_delay_samples =
100       p_drc_params_struct->audio_delay_samples;
101   p_audio_in_out_buf->frame_size = p_drc_params_struct->drc_frame_size;
102 
103   if (p_drc_params_struct->sub_band_domain_mode == SUBBAND_DOMAIN_MODE_QMF64) {
104     p_audio_in_out_buf->audio_delay_sub_band_samples =
105         p_drc_params_struct->audio_delay_samples /
106         AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF64;
107     p_audio_in_out_buf->audio_sub_band_frame_size =
108         p_drc_params_struct->drc_frame_size /
109         AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF64;
110     p_audio_in_out_buf->audio_sub_band_count = AUDIO_CODEC_SUBBAND_COUNT_QMF64;
111   } else if (p_drc_params_struct->sub_band_domain_mode ==
112              SUBBAND_DOMAIN_MODE_QMF71) {
113     p_audio_in_out_buf->audio_delay_sub_band_samples =
114         p_drc_params_struct->audio_delay_samples /
115         AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF71;
116     p_audio_in_out_buf->audio_sub_band_frame_size =
117         p_drc_params_struct->drc_frame_size /
118         AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF71;
119     p_audio_in_out_buf->audio_sub_band_count = AUDIO_CODEC_SUBBAND_COUNT_QMF71;
120   } else if (p_drc_params_struct->sub_band_domain_mode ==
121              SUBBAND_DOMAIN_MODE_STFT256) {
122     p_audio_in_out_buf->audio_delay_sub_band_samples =
123         p_drc_params_struct->audio_delay_samples /
124         AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_STFT256;
125     p_audio_in_out_buf->audio_sub_band_frame_size =
126         p_drc_params_struct->drc_frame_size /
127         AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_STFT256;
128     p_audio_in_out_buf->audio_sub_band_count =
129         AUDIO_CODEC_SUBBAND_COUNT_STFT256;
130   } else {
131     p_audio_in_out_buf->audio_delay_sub_band_samples = 0;
132     p_audio_in_out_buf->audio_sub_band_frame_size = 0;
133     p_audio_in_out_buf->audio_sub_band_count = 0;
134   }
135 
136   for (k = 0; k < SEL_DRC_COUNT; k++) {
137     if (p_drc_params_struct->sel_drc_array[k].drc_instructions_index >= 0) {
138       ia_drc_instructions_struct* drc_instruction_str =
139           &(pstr_drc_config->str_drc_instruction_str
140                 [p_drc_params_struct->sel_drc_array[k].drc_instructions_index]);
141       if (drc_instruction_str->gain_element_count > 0) {
142         p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
143             .buf_interpolation = (ia_interp_buf_struct*)*mem_ptr;
144         *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
145                            drc_instruction_str->gain_element_count *
146                                sizeof(ia_interp_buf_struct) +
147                            32);
148         p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
149             .buf_interpolation_count = drc_instruction_str->gain_element_count;
150         for (i = 0;
151              i < p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
152                      .buf_interpolation_count;
153              i++) {
154           p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
155               .buf_interpolation[i]
156               .str_node.time = 0;
157           p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
158               .buf_interpolation[i]
159               .prev_node.time = -1;
160           p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
161               .buf_interpolation[i]
162               .str_node.loc_db_gain = 0.0f;
163           p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
164               .buf_interpolation[i]
165               .str_node.slope = 0.0f;
166 
167           for (j = 0; j < 2 * AUDIO_CODEC_FRAME_SIZE_MAX + MAX_SIGNAL_DELAY;
168                j++) {
169             p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
170                 .buf_interpolation[i]
171                 .lpcm_gains[j] = 1.f;
172           }
173         }
174       }
175     }
176   }
177 
178   if (eq_set_id_processed > 0) {
179     for (i = 0; i < pstr_drc_config->str_drc_config_ext.eq_instructions_count;
180          i++) {
181       if (pstr_drc_config->str_drc_config_ext.str_eq_instructions[i]
182               .eq_set_id == eq_set_id_processed)
183         break;
184     }
185     if (i == pstr_drc_config->str_drc_config_ext.eq_instructions_count) {
186       return -1;
187     }
188 
189     p_drc_gain_dec_structs->eq_set = (ia_eq_set_struct*)*mem_ptr;
190     *mem_ptr = (pVOID)((SIZE_T)*mem_ptr + sizeof(ia_eq_set_struct));
191 
192     if (err_code) return (err_code);
193 
194     err_code = impd_derive_eq_set(
195         &pstr_drc_config->str_drc_config_ext.str_eq_coeff,
196         &(pstr_drc_config->str_drc_config_ext.str_eq_instructions[i]),
197         (FLOAT32)p_drc_gain_dec_structs->ia_drc_params_struct.sample_rate,
198         p_drc_gain_dec_structs->ia_drc_params_struct.drc_frame_size,
199         p_drc_gain_dec_structs->ia_drc_params_struct.sub_band_domain_mode,
200         p_drc_gain_dec_structs->eq_set);
201     if (err_code) return (err_code);
202 
203     impd_get_eq_set_delay(
204         p_drc_gain_dec_structs->eq_set,
205         &p_drc_gain_dec_structs->ia_drc_params_struct.eq_delay);
206   }
207 
208   for (i = 0; i < p_drc_params_struct->drc_set_counter; i++) {
209     ia_drc_instructions_struct* drc_instruction_str;
210     drc_instruction_str =
211         &(pstr_drc_config->str_drc_instruction_str
212               [p_drc_params_struct->sel_drc_array[i].drc_instructions_index]);
213     maxMultibandAudioSignalCount =
214         max(maxMultibandAudioSignalCount,
215             drc_instruction_str->multiband_audio_sig_count);
216   }
217 
218   p_drc_gain_dec_structs->audio_band_buffer.non_interleaved_audio = *mem_ptr;
219   *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
220                      (maxMultibandAudioSignalCount * sizeof(FLOAT32*)));
221 
222   for (i = 0; i < maxMultibandAudioSignalCount; i++) {
223     p_drc_gain_dec_structs->audio_band_buffer.non_interleaved_audio[i] =
224         *mem_ptr;
225     *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
226                        (p_drc_params_struct->drc_frame_size * sizeof(FLOAT32)));
227   }
228   p_drc_gain_dec_structs->audio_band_buffer.multiband_audio_sig_count =
229       maxMultibandAudioSignalCount;
230   p_drc_gain_dec_structs->audio_band_buffer.frame_size =
231       p_drc_params_struct->drc_frame_size;
232   ;
233 
234   if (p_drc_params_struct->sub_band_domain_mode == SUBBAND_DOMAIN_MODE_OFF &&
235       p_audio_in_out_buf->audio_delay_samples) {
236     p_audio_in_out_buf->audio_io_buffer_delayed = *mem_ptr;
237     *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
238                        (p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
239     p_audio_in_out_buf->audio_in_out_buf = *mem_ptr;
240     *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
241                        (p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
242 
243     for (i = 0; i < p_audio_in_out_buf->audio_num_chan; i++) {
244       p_audio_in_out_buf->audio_io_buffer_delayed[i] = *mem_ptr;
245       *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
246                          ((p_audio_in_out_buf->frame_size +
247                            p_audio_in_out_buf->audio_delay_samples) *
248                           sizeof(FLOAT32*)));
249       p_audio_in_out_buf->audio_in_out_buf[i] =
250           &p_audio_in_out_buf->audio_io_buffer_delayed
251                [i][p_audio_in_out_buf->audio_delay_samples];
252     }
253   }
254   if (p_drc_params_struct->sub_band_domain_mode != SUBBAND_DOMAIN_MODE_OFF &&
255       p_audio_in_out_buf->audio_delay_sub_band_samples) {
256     p_audio_in_out_buf->audio_buffer_delayed_real = *mem_ptr;
257     *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
258                        (p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
259     p_audio_in_out_buf->audio_buffer_delayed_imag = *mem_ptr;
260     *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
261                        (p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
262     p_audio_in_out_buf->audio_real_buff = *mem_ptr;
263     *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
264                        (p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
265     p_audio_in_out_buf->audio_imag_buff = *mem_ptr;
266     *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
267                        (p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
268 
269     for (i = 0; i < p_audio_in_out_buf->audio_num_chan; i++) {
270       p_audio_in_out_buf->audio_buffer_delayed_real[i] = *mem_ptr;
271       *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
272                          ((p_audio_in_out_buf->audio_sub_band_frame_size +
273                            p_audio_in_out_buf->audio_delay_sub_band_samples) *
274                           sizeof(FLOAT32*)));
275       p_audio_in_out_buf->audio_buffer_delayed_imag[i] = *mem_ptr;
276       *mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
277                          ((p_audio_in_out_buf->audio_sub_band_frame_size +
278                            p_audio_in_out_buf->audio_delay_sub_band_samples) *
279                           sizeof(FLOAT32*)));
280 
281       p_audio_in_out_buf->audio_real_buff[i] =
282           &p_audio_in_out_buf->audio_buffer_delayed_real
283                [i][p_audio_in_out_buf->audio_delay_sub_band_samples *
284                    p_audio_in_out_buf->audio_sub_band_count];
285       p_audio_in_out_buf->audio_imag_buff[i] =
286           &p_audio_in_out_buf->audio_buffer_delayed_imag
287                [i][p_audio_in_out_buf->audio_delay_sub_band_samples *
288                    p_audio_in_out_buf->audio_sub_band_count];
289     }
290   }
291 
292   return err_code;
293 }
294 
impd_drc_process_time_domain(ia_drc_gain_dec_struct * p_drc_gain_dec_structs,ia_drc_config * pstr_drc_config,ia_drc_gain_struct * pstr_drc_gain,FLOAT32 * audio_in_out_buf[],FLOAT32 loudness_normalization_gain_db,FLOAT32 boost,FLOAT32 compress,WORD32 drc_characteristic_target)295 IA_ERRORCODE impd_drc_process_time_domain(
296     ia_drc_gain_dec_struct* p_drc_gain_dec_structs,
297     ia_drc_config* pstr_drc_config, ia_drc_gain_struct* pstr_drc_gain,
298     FLOAT32* audio_in_out_buf[], FLOAT32 loudness_normalization_gain_db,
299     FLOAT32 boost, FLOAT32 compress, WORD32 drc_characteristic_target) {
300   WORD32 sel_drc_index;
301   IA_ERRORCODE err_code = 0;
302   WORD32 passThru;
303   ia_drc_instructions_struct* str_drc_instruction_str =
304       pstr_drc_config->str_drc_instruction_str;
305 
306   if (p_drc_gain_dec_structs->eq_set) {
307     WORD32 ch;
308     FLOAT32* audio_channel;
309     for (ch = 0; ch < p_drc_gain_dec_structs->eq_set->audio_num_chan; ch++) {
310       audio_channel = audio_in_out_buf[ch];
311 
312       err_code = impd_process_eq_set_time_domain(
313           p_drc_gain_dec_structs->eq_set, ch, audio_channel, audio_channel,
314           p_drc_gain_dec_structs->ia_drc_params_struct.drc_frame_size);
315       if (err_code) return (err_code);
316     }
317   }
318 
319   impd_store_audio_io_buffer_time(audio_in_out_buf,
320                                   &p_drc_gain_dec_structs->audio_in_out_buf);
321 
322   if (pstr_drc_config->apply_drc) {
323     for (sel_drc_index = 0;
324          sel_drc_index <
325          p_drc_gain_dec_structs->ia_drc_params_struct.drc_set_counter;
326          sel_drc_index++) {
327       err_code = impd_get_drc_gain(
328           p_drc_gain_dec_structs, pstr_drc_config, pstr_drc_gain, compress,
329           boost, drc_characteristic_target, loudness_normalization_gain_db,
330           sel_drc_index, &p_drc_gain_dec_structs->drc_gain_buffers);
331       if (err_code != IA_NO_ERROR) return (err_code);
332     }
333 
334     if (p_drc_gain_dec_structs->ia_drc_params_struct.drc_set_counter == 0) {
335       impd_retrieve_audio_io_buffer_time(
336           audio_in_out_buf, &p_drc_gain_dec_structs->audio_in_out_buf);
337 
338     } else {
339       for (sel_drc_index = 0;
340            sel_drc_index <
341            p_drc_gain_dec_structs->ia_drc_params_struct.drc_set_counter;
342            sel_drc_index++) {
343         if (p_drc_gain_dec_structs->ia_drc_params_struct
344                 .multiband_sel_drc_idx == sel_drc_index) {
345           passThru = 0;
346         } else {
347           passThru = 1;
348         }
349         err_code = impd_filter_banks_process(
350             str_drc_instruction_str,
351             p_drc_gain_dec_structs->ia_drc_params_struct
352                 .sel_drc_array[sel_drc_index]
353                 .drc_instructions_index,
354             &p_drc_gain_dec_structs->ia_drc_params_struct,
355             p_drc_gain_dec_structs->audio_in_out_buf.audio_io_buffer_delayed,
356             &p_drc_gain_dec_structs->audio_band_buffer,
357             &p_drc_gain_dec_structs->ia_filter_banks_struct, passThru);
358         if (err_code != IA_NO_ERROR) return (err_code);
359 
360         impd_apply_gains_and_add(
361             str_drc_instruction_str,
362             p_drc_gain_dec_structs->ia_drc_params_struct
363                 .sel_drc_array[sel_drc_index]
364                 .drc_instructions_index,
365             &p_drc_gain_dec_structs->ia_drc_params_struct,
366             &(p_drc_gain_dec_structs->drc_gain_buffers
367                   .pstr_gain_buf[sel_drc_index]),
368             p_drc_gain_dec_structs->shape_filter_block,
369             p_drc_gain_dec_structs->audio_band_buffer.non_interleaved_audio,
370             audio_in_out_buf, 1);
371       }
372     }
373   }
374 
375   impd_advance_audio_io_buffer_time(&p_drc_gain_dec_structs->audio_in_out_buf);
376 
377   return err_code;
378 }
379 
380 
impd_get_parametric_drc_delay(ia_drc_gain_dec_struct * p_drc_gain_dec_structs,ia_drc_config * pstr_drc_config,WORD32 * parametric_drc_delay,WORD32 * parametric_drc_delay_max)381 VOID impd_get_parametric_drc_delay(
382     ia_drc_gain_dec_struct* p_drc_gain_dec_structs,
383     ia_drc_config* pstr_drc_config, WORD32* parametric_drc_delay,
384     WORD32* parametric_drc_delay_max) {
385   *parametric_drc_delay =
386       p_drc_gain_dec_structs->ia_drc_params_struct.parametric_drc_delay;
387 
388   if (pstr_drc_config->str_drc_config_ext.parametric_drc_present &&
389       pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
390           .parametric_drc_delay_max_present) {
391     *parametric_drc_delay_max =
392         pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
393             .parametric_drc_delay_max;
394   } else if (pstr_drc_config->str_drc_config_ext.parametric_drc_present == 0) {
395     *parametric_drc_delay_max = 0;
396   } else {
397     *parametric_drc_delay_max = -1;
398   }
399 
400   return;
401 }
402 
impd_get_eq_delay(ia_drc_gain_dec_struct * p_drc_gain_dec_structs,ia_drc_config * pstr_drc_config,WORD32 * eq_delay,WORD32 * eq_delay_max)403 VOID impd_get_eq_delay(ia_drc_gain_dec_struct* p_drc_gain_dec_structs,
404                        ia_drc_config* pstr_drc_config, WORD32* eq_delay,
405                        WORD32* eq_delay_max) {
406   *eq_delay = p_drc_gain_dec_structs->ia_drc_params_struct.eq_delay;
407 
408   if (pstr_drc_config->str_drc_config_ext.eq_flag &&
409       pstr_drc_config->str_drc_config_ext.str_eq_coeff.eq_delay_max_present) {
410     *eq_delay_max =
411         pstr_drc_config->str_drc_config_ext.str_eq_coeff.eq_delay_max;
412   } else if (pstr_drc_config->str_drc_config_ext.eq_flag == 0) {
413     *eq_delay_max = 0;
414   } else {
415     *eq_delay_max = -1;
416   }
417 
418   return;
419 }
420