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
30WORD32 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
98WORD32 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
129WORD32 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