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 "impd_type_def.h"
22 #include "impd_drc_extr_delta_coded_info.h"
23 #include "impd_drc_common.h"
24 #include "impd_drc_struct.h"
25 #include "impd_parametric_drc_dec.h"
26 #include "impd_drc_gain_dec.h"
27 #include "impd_drc_filter_bank.h"
28 #include "impd_drc_multi_band.h"
29
impd_init_drc_params(WORD32 frame_size,WORD32 sample_rate,WORD32 gain_delay_samples,WORD32 delay_mode,WORD32 sub_band_domain_mode,ia_drc_params_struct * ia_drc_params_struct)30 WORD32 impd_init_drc_params(WORD32 frame_size, WORD32 sample_rate,
31 WORD32 gain_delay_samples, WORD32 delay_mode,
32 WORD32 sub_band_domain_mode,
33 ia_drc_params_struct* ia_drc_params_struct) {
34 WORD32 k;
35 if (frame_size < 1 || frame_size > AUDIO_CODEC_FRAME_SIZE_MAX) {
36 return -1;
37 }
38
39 if (sample_rate < 1000) {
40 return -1;
41 }
42
43 ia_drc_params_struct->drc_frame_size = frame_size;
44
45 if (ia_drc_params_struct->drc_frame_size < 0.001f * sample_rate) {
46 return -1;
47 }
48
49 ia_drc_params_struct->sample_rate = sample_rate;
50
51 ia_drc_params_struct->delta_tmin_default = impd_get_delta_tmin(sample_rate);
52
53 if (ia_drc_params_struct->delta_tmin_default >
54 ia_drc_params_struct->drc_frame_size) {
55 return -1;
56 }
57
58 if ((delay_mode != DELAY_MODE_REGULAR_DELAY) &&
59 (delay_mode != DELAY_MODE_LOW_DELAY)) {
60 return -1;
61 }
62
63 ia_drc_params_struct->delay_mode = delay_mode;
64
65 ia_drc_params_struct->drc_set_counter = 0;
66 ia_drc_params_struct->multiband_sel_drc_idx = -1;
67
68 for (k = 0; k < SEL_DRC_COUNT; k++) {
69 ia_drc_params_struct->sel_drc_array[k].drc_instructions_index = -1;
70 ia_drc_params_struct->sel_drc_array[k].dwnmix_instructions_index = -1;
71 ia_drc_params_struct->sel_drc_array[k].drc_coeff_idx = -1;
72 }
73
74 if ((gain_delay_samples > MAX_SIGNAL_DELAY) || (gain_delay_samples < 0)) {
75 return -1;
76 } else {
77 ia_drc_params_struct->gain_delay_samples = gain_delay_samples;
78 }
79
80 switch (sub_band_domain_mode) {
81 case SUBBAND_DOMAIN_MODE_OFF:
82 case SUBBAND_DOMAIN_MODE_QMF64:
83 case SUBBAND_DOMAIN_MODE_QMF71:
84 case SUBBAND_DOMAIN_MODE_STFT256:
85 ia_drc_params_struct->sub_band_domain_mode = sub_band_domain_mode;
86 break;
87 default:
88 return -1;
89 break;
90 }
91
92 ia_drc_params_struct->parametric_drc_delay = 0;
93 ia_drc_params_struct->eq_delay = 0;
94
95 return 0;
96 }
97
impd_select_drc_coefficients(ia_drc_config * drc_config,ia_uni_drc_coeffs_struct ** drc_coefficients_drc,WORD32 * drc_coefficients_selected)98 WORD32 impd_select_drc_coefficients(
99 ia_drc_config* drc_config, ia_uni_drc_coeffs_struct** drc_coefficients_drc,
100 WORD32* drc_coefficients_selected) {
101 WORD32 i;
102 WORD32 cof1 = -1;
103 WORD32 cof0 = -1;
104 for (i = 0; i < drc_config->drc_coefficients_drc_count; i++) {
105 if (drc_config->str_p_loc_drc_coefficients_uni_drc[i].drc_location == 1) {
106 if (drc_config->str_p_loc_drc_coefficients_uni_drc[i].version == 0) {
107 cof0 = i;
108 *drc_coefficients_selected = cof0;
109 } else {
110 cof1 = i;
111 *drc_coefficients_selected = cof1;
112 }
113 }
114 }
115
116 if (cof1 >= 0) {
117 *drc_coefficients_drc =
118 &(drc_config->str_p_loc_drc_coefficients_uni_drc[cof1]);
119 } else if (cof0 >= 0) {
120 *drc_coefficients_drc =
121 &(drc_config->str_p_loc_drc_coefficients_uni_drc[cof0]);
122 } else {
123 *drc_coefficients_drc = NULL;
124 }
125
126 return 0;
127 }
128
impd_init_selected_drc_set(ia_drc_config * drc_config,ia_drc_params_struct * ia_drc_params_struct,ia_parametric_drc_params_struct * p_parametric_drc_params,WORD32 audio_num_chan,WORD32 drc_set_id_selected,WORD32 downmix_id_selected,ia_filter_banks_struct * ia_filter_banks_struct,ia_overlap_params_struct * pstr_overlap_params,shape_filter_block * shape_filter_block)129 WORD32 impd_init_selected_drc_set(
130 ia_drc_config* drc_config, ia_drc_params_struct* ia_drc_params_struct,
131 ia_parametric_drc_params_struct* p_parametric_drc_params,
132 WORD32 audio_num_chan, WORD32 drc_set_id_selected,
133 WORD32 downmix_id_selected, ia_filter_banks_struct* ia_filter_banks_struct,
134 ia_overlap_params_struct* pstr_overlap_params
135
136 ,
137 shape_filter_block* shape_filter_block) {
138 WORD32 g, n, c, err = 0;
139 WORD32 channel_count = 0;
140 WORD32 i;
141
142 ia_drc_instructions_struct* drc_instructions_uni_drc = NULL;
143 ia_uni_drc_coeffs_struct* drc_coefficients_uni_drc = NULL;
144 WORD32 selected_drc_is_multiband = 0;
145 WORD32 drc_instructions_selected = -1;
146 WORD32 downmix_instructions_selected = -1;
147 WORD32 drc_coefficients_selected = -1;
148 p_parametric_drc_params->parametric_drc_instance_count = 0;
149
150 if (drc_config->drc_coefficients_drc_count &&
151 drc_config->str_p_loc_drc_coefficients_uni_drc->drc_frame_size_present) {
152 if (drc_config->str_p_loc_drc_coefficients_uni_drc->drc_frame_size !=
153 ia_drc_params_struct->drc_frame_size) {
154 return -1;
155 }
156 }
157
158 for (n = 0; n < drc_config->drc_instructions_count_plus; n++) {
159 if (drc_config->str_drc_instruction_str[n].drc_set_id ==
160 drc_set_id_selected)
161 break;
162 }
163 if (n == drc_config->drc_instructions_count_plus) {
164 return -1;
165 }
166 drc_instructions_selected = n;
167 drc_instructions_uni_drc =
168 &(drc_config->str_drc_instruction_str[drc_instructions_selected]);
169
170 if (downmix_id_selected == ID_FOR_BASE_LAYOUT) {
171 channel_count = drc_config->channel_layout.base_channel_count;
172 } else if (downmix_id_selected == ID_FOR_ANY_DOWNMIX) {
173 channel_count = audio_num_chan;
174 } else {
175 for (n = 0; n < drc_config->dwnmix_instructions_count; n++) {
176 if (drc_config->dwnmix_instructions[n].downmix_id == downmix_id_selected)
177 break;
178 }
179 if (n == drc_config->dwnmix_instructions_count) {
180 return (UNEXPECTED_ERROR);
181 }
182 channel_count = drc_config->dwnmix_instructions[n].target_channel_count;
183
184 downmix_instructions_selected = n;
185 }
186 drc_instructions_uni_drc->audio_num_chan = channel_count;
187
188 if (drc_instructions_uni_drc->drc_set_id <= 0) {
189 drc_coefficients_selected = 0;
190 } else {
191 err = impd_select_drc_coefficients(drc_config, &drc_coefficients_uni_drc,
192 &drc_coefficients_selected);
193 }
194
195 ia_drc_params_struct->sel_drc_array[ia_drc_params_struct->drc_set_counter]
196 .drc_instructions_index = drc_instructions_selected;
197 ia_drc_params_struct->sel_drc_array[ia_drc_params_struct->drc_set_counter]
198 .dwnmix_instructions_index = downmix_instructions_selected;
199 ia_drc_params_struct->sel_drc_array[ia_drc_params_struct->drc_set_counter]
200 .drc_coeff_idx = drc_coefficients_selected;
201
202 if ((drc_instructions_uni_drc->downmix_id[0] == ID_FOR_ANY_DOWNMIX) ||
203 (drc_instructions_uni_drc->dwnmix_id_count > 1)) {
204 WORD32 idx = drc_instructions_uni_drc->gain_set_index[0];
205 for (c = 0; c < drc_instructions_uni_drc->audio_num_chan; c++) {
206 drc_instructions_uni_drc->channel_group_of_ch[c] = (idx >= 0) ? 0 : -1;
207 }
208 }
209
210 for (g = 0; g < drc_instructions_uni_drc->num_drc_ch_groups; g++) {
211 drc_instructions_uni_drc->num_chan_per_ch_group[g] = 0;
212 for (c = 0; c < drc_instructions_uni_drc->audio_num_chan; c++) {
213 if (drc_instructions_uni_drc->channel_group_of_ch[c] == g) {
214 drc_instructions_uni_drc->num_chan_per_ch_group[g]++;
215 }
216 }
217 }
218
219 if (drc_instructions_uni_drc->drc_set_effect &
220 (EFFECT_BIT_DUCK_OTHER | EFFECT_BIT_DUCK_SELF)) {
221 drc_instructions_uni_drc->multiband_audio_sig_count =
222 drc_instructions_uni_drc->audio_num_chan;
223 } else {
224 drc_instructions_uni_drc->multiband_audio_sig_count = 0;
225 for (c = 0; c < drc_instructions_uni_drc->audio_num_chan; c++) {
226 g = drc_instructions_uni_drc->channel_group_of_ch[c];
227 if (g < 0) {
228 drc_instructions_uni_drc->multiband_audio_sig_count++;
229 } else {
230 drc_instructions_uni_drc->multiband_audio_sig_count +=
231 drc_instructions_uni_drc->band_count_of_ch_group[g];
232 }
233 }
234 }
235
236 for (g = 0; g < drc_instructions_uni_drc->num_drc_ch_groups; g++) {
237 if (g == 0) {
238 drc_instructions_uni_drc->parametric_drc_look_ahead_samples_max = 0;
239 }
240 if (drc_instructions_uni_drc->ch_group_parametric_drc_flag[g] == 0) {
241 if (drc_instructions_uni_drc->band_count_of_ch_group[g] > 1) {
242 if (ia_drc_params_struct->multiband_sel_drc_idx != -1) {
243 return (UNEXPECTED_ERROR);
244 }
245 selected_drc_is_multiband = 1;
246 }
247 } else {
248 WORD32 gain_set_index =
249 drc_instructions_uni_drc->gain_set_idx_of_ch_group_parametric_drc[g];
250 WORD32 parametric_drc_id =
251 drc_config->str_drc_config_ext.str_drc_coeff_param_drc
252 .str_parametric_drc_gain_set_params[gain_set_index]
253 .parametric_drc_id;
254 WORD32 parametric_drc_look_ahead_samples = 0;
255 ia_parametric_drc_instructions_struct* str_parametric_drc_instructions;
256
257 for (i = 0;
258 i < drc_config->str_drc_config_ext.parametric_drc_instructions_count;
259 i++) {
260 if (parametric_drc_id ==
261 drc_config->str_drc_config_ext.str_parametric_drc_instructions[i]
262 .parametric_drc_id)
263 break;
264 }
265 if (i ==
266 drc_config->str_drc_config_ext.parametric_drc_instructions_count) {
267 return (UNEXPECTED_ERROR);
268 }
269 str_parametric_drc_instructions =
270 &drc_config->str_drc_config_ext.str_parametric_drc_instructions[i];
271
272 p_parametric_drc_params->parametric_drc_idx
273 [p_parametric_drc_params->parametric_drc_instance_count] = i;
274 p_parametric_drc_params->gain_set_index
275 [p_parametric_drc_params->parametric_drc_instance_count] =
276 gain_set_index;
277 if (drc_instructions_uni_drc->drc_apply_to_dwnmix == 0) {
278 p_parametric_drc_params->dwnmix_id_from_drc_instructions
279 [p_parametric_drc_params->parametric_drc_instance_count] =
280 ID_FOR_BASE_LAYOUT;
281 } else {
282 if (drc_instructions_uni_drc->dwnmix_id_count > 1) {
283 p_parametric_drc_params->dwnmix_id_from_drc_instructions
284 [p_parametric_drc_params->parametric_drc_instance_count] =
285 ID_FOR_ANY_DOWNMIX;
286 } else {
287 p_parametric_drc_params->dwnmix_id_from_drc_instructions
288 [p_parametric_drc_params->parametric_drc_instance_count] =
289 drc_instructions_uni_drc->downmix_id[0];
290 }
291 }
292 p_parametric_drc_params->audio_num_chan = channel_count;
293 for (i = 0; i < p_parametric_drc_params->audio_num_chan; i++) {
294 if (drc_instructions_uni_drc->channel_group_of_ch[i] == g) {
295 p_parametric_drc_params->channel_map
296 [p_parametric_drc_params->parametric_drc_instance_count][i] = 1;
297 } else {
298 p_parametric_drc_params->channel_map
299 [p_parametric_drc_params->parametric_drc_instance_count][i] = 0;
300 }
301 }
302 drc_instructions_uni_drc->parametric_drc_look_ahead_samples[g] = 0;
303 if (str_parametric_drc_instructions->parametric_drc_look_ahead_flag) {
304 parametric_drc_look_ahead_samples = (WORD32)(
305 (FLOAT32)
306 str_parametric_drc_instructions->parametric_drc_look_ahead *
307 (FLOAT32)p_parametric_drc_params->sampling_rate * 0.001f);
308 } else {
309 if (str_parametric_drc_instructions->parametric_drc_type ==
310 PARAM_DRC_TYPE_FF) {
311 parametric_drc_look_ahead_samples = (WORD32)(
312 10.0f * (FLOAT32)p_parametric_drc_params->sampling_rate * 0.001f);
313 } else if (str_parametric_drc_instructions->parametric_drc_type ==
314 PARAM_DRC_TYPE_LIM) {
315 parametric_drc_look_ahead_samples = (WORD32)(
316 5.0f * (FLOAT32)p_parametric_drc_params->sampling_rate * 0.001f);
317 } else {
318 return (UNEXPECTED_ERROR);
319 }
320 }
321 drc_instructions_uni_drc->parametric_drc_look_ahead_samples[g] =
322 parametric_drc_look_ahead_samples;
323 if (drc_instructions_uni_drc->parametric_drc_look_ahead_samples_max <
324 drc_instructions_uni_drc->parametric_drc_look_ahead_samples[g]) {
325 drc_instructions_uni_drc->parametric_drc_look_ahead_samples_max =
326 drc_instructions_uni_drc->parametric_drc_look_ahead_samples[g];
327 }
328 p_parametric_drc_params->parametric_drc_instance_count += 1;
329 selected_drc_is_multiband = 0;
330 }
331 }
332 ia_drc_params_struct->parametric_drc_delay +=
333 drc_instructions_uni_drc->parametric_drc_look_ahead_samples_max;
334
335 if (selected_drc_is_multiband == 1) {
336 ia_drc_params_struct->multiband_sel_drc_idx =
337 ia_drc_params_struct->drc_set_counter;
338 err = impd_init_all_filter_banks(
339 drc_coefficients_uni_drc,
340 &(drc_config->str_drc_instruction_str[drc_instructions_selected]),
341 ia_filter_banks_struct);
342 if (err) return (err);
343
344 err = impd_init_overlap_weight(
345 drc_coefficients_uni_drc,
346 &(drc_config->str_drc_instruction_str[drc_instructions_selected]),
347 ia_drc_params_struct->sub_band_domain_mode, pstr_overlap_params);
348 if (err) return (err);
349 } else {
350 ia_gain_modifiers_struct* gain_modifiers =
351 drc_config->str_drc_instruction_str->str_gain_modifiers_of_ch_group;
352 for (g = 0; g < drc_instructions_uni_drc->num_drc_ch_groups; g++) {
353 if (gain_modifiers[g].shape_filter_flag == 1) {
354 impd_shape_filt_block_init(
355 &drc_coefficients_uni_drc->str_shape_filter_block_params
356 [gain_modifiers[g].shape_filter_idx],
357 &shape_filter_block[g]);
358 } else {
359 shape_filter_block[g].shape_flter_block_flag = 0;
360 }
361 }
362 }
363
364 ia_drc_params_struct->drc_set_counter++;
365
366 return (0);
367 }
368