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#ifndef IMPD_PARAMETRIC_DRC_DEC_H
21#define IMPD_PARAMETRIC_DRC_DEC_H
22
23#ifdef __cplusplus
24extern "C" {
25#endif
26
27typedef struct ia_2nd_order_filt_coeff_struct_t {
28  FLOAT32 b0, b1, b2;
29  FLOAT32 a1, a2;
30} ia_2nd_order_filt_coeff_struct;
31
32typedef struct ia_2nd_order_filt_state_struct_t {
33  FLOAT32 z1, z2;
34} ia_2nd_order_filt_state_struct;
35
36typedef struct ia_parametric_drc_type_ff_params_struct_t {
37  WORD32 audio_num_chan;
38  WORD32 frame_size;
39  WORD32 sub_band_domain_mode;
40  WORD32 sub_band_count;
41  WORD32 level_estim_integration_time;
42  WORD32 level_estim_frame_index;
43  WORD32 level_estim_frame_count;
44  FLOAT32 level[PARAM_DRC_TYPE_FF_LEVEL_ESTIM_FRAME_COUNT_MAX];
45  WORD32 start_up_phase;
46  FLOAT32 level_estim_ch_weight[MAX_CHANNEL_COUNT];
47  WORD32 level_estim_k_weighting_type;
48  ia_2nd_order_filt_coeff_struct pre_filt_coeff;
49  ia_2nd_order_filt_coeff_struct rlb_filt_coeff;
50  ia_2nd_order_filt_state_struct pre_filt_state[MAX_CHANNEL_COUNT];
51  ia_2nd_order_filt_state_struct rlb_filt_state[MAX_CHANNEL_COUNT];
52  FLOAT32 weighting_filt[AUDIO_CODEC_SUBBAND_COUNT_MAX];
53  WORD32 sub_band_compensation_type;
54  ia_2nd_order_filt_coeff_struct filt_coeff_subband;
55  ia_2nd_order_filt_state_struct filt_state_subband_real[MAX_CHANNEL_COUNT];
56  ia_2nd_order_filt_state_struct filt_state_subband_imag[MAX_CHANNEL_COUNT];
57  FLOAT32 ref_level_parametric_drc;
58
59  WORD32 node_count;
60  WORD32 node_level[PARAM_DRC_TYPE_FF_NODE_COUNT_MAX];
61  WORD32 node_gain[PARAM_DRC_TYPE_FF_NODE_COUNT_MAX];
62
63  FLOAT32 gain_smooth_attack_alpha_slow;
64  FLOAT32 gain_smooth_rel_alpha_slow;
65  FLOAT32 gain_smooth_attack_alpha_fast;
66  FLOAT32 gain_smooth_rel_alpha_fast;
67  WORD32 gain_smooth_attack_threshold;
68  WORD32 gain_smooth_rel_threshold;
69  WORD32 gain_smooth_hold_off_count;
70  FLOAT32 db_level_smooth;
71  FLOAT32 db_gain_smooth;
72  WORD32 hold_counter;
73
74} ia_parametric_drc_type_ff_params_struct;
75
76typedef struct ia_parametric_drc_type_lim_params_struct_t {
77  WORD32 audio_num_chan;
78  WORD32 frame_size;
79  FLOAT32 level_estim_ch_weight[MAX_CHANNEL_COUNT];
80
81  UWORD32 attack;
82  FLOAT32 attack_constant;
83  FLOAT32 release_constant;
84  FLOAT32 attack_ms;
85  FLOAT32 release_ms;
86  FLOAT32 threshold;
87  UWORD32 channels;
88  UWORD32 sampling_rate;
89  FLOAT32 cor;
90  FLOAT32* max_buf;
91  FLOAT32* max_buf_slow;
92  UWORD32 max_buf_idx;
93  UWORD32 max_buf_slow_idx;
94  UWORD32 sec_len;
95  UWORD32 num_max_buf_sec;
96  UWORD32 max_buf_sec_idx;
97  UWORD32 max_buf_sec_ctr;
98  FLOAT64 smooth_state_0;
99
100} ia_parametric_drc_type_lim_params_struct;
101
102typedef struct ia_parametric_drc_instance_params_struct_t {
103  WORD32 disable_paramteric_drc;
104  WORD32 parametric_drc_type;
105  ia_spline_nodes_struct str_spline_nodes;
106  ia_parametric_drc_type_ff_params_struct str_parametric_drc_type_ff_params;
107  ia_parametric_drc_type_lim_params_struct str_parametric_drc_type_lim_params;
108} ia_parametric_drc_instance_params_struct;
109
110typedef struct ia_parametric_drc_params_struct_t {
111  WORD32 sampling_rate;
112  WORD32 audio_num_chan;
113  WORD32 sub_band_domain_mode;
114  WORD32 sub_band_count;
115
116  WORD32 num_nodes;
117  WORD32 drc_frame_size;
118  WORD32 parametric_drc_frame_size;
119  WORD32 parametric_drc_look_ahead_samples_max;
120  WORD32 reset_parametric_drc;
121
122  WORD32 parametric_drc_instance_count;
123  WORD32 parametric_drc_idx[PARAM_DRC_INSTANCE_COUNT_MAX];
124  WORD32 gain_set_index[PARAM_DRC_INSTANCE_COUNT_MAX];
125  WORD32 dwnmix_id_from_drc_instructions[PARAM_DRC_INSTANCE_COUNT_MAX];
126  WORD32 channel_map[PARAM_DRC_INSTANCE_COUNT_MAX][MAX_CHANNEL_COUNT];
127
128  ia_parametric_drc_instance_params_struct
129      str_parametric_drc_instance_params[PARAM_DRC_INSTANCE_COUNT_MAX];
130
131} ia_parametric_drc_params_struct;
132
133WORD32 impd_init_parametric_drc(
134    WORD32 drc_frame_size, WORD32 sampling_rate, WORD32 sub_band_domain_mode,
135    ia_parametric_drc_params_struct* pstr_parametric_drc_params);
136
137WORD32 impd_init_parametric_drc_after_config(
138    ia_drc_config* pstr_drc_config,
139    ia_drc_loudness_info_set_struct* pstr_loudness_info,
140    ia_parametric_drc_params_struct* pstr_parametric_drc_params,
141    pVOID* mem_ptr);
142
143WORD32
144impd_init_lvl_est_filt_time(WORD32 level_estim_k_weighting_type,
145                            WORD32 sampling_rate,
146                            ia_2nd_order_filt_coeff_struct* pre_filt_coeff,
147                            ia_2nd_order_filt_coeff_struct* rlb_filt_coeff);
148
149WORD32
150impd_init_lvl_est_filt_subband(
151    WORD32 level_estim_k_weighting_type, WORD32 sampling_rate,
152    WORD32 sub_band_domain_mode, WORD32 sub_band_count,
153    WORD32 sub_band_compensation_type, FLOAT32* weighting_filt,
154    ia_2nd_order_filt_coeff_struct* filt_coeff_subband);
155
156WORD32
157impd_parametric_ffwd_type_drc_reset(ia_parametric_drc_type_ff_params_struct*
158                                        pstr_parametric_ffwd_type_drc_params);
159
160WORD32
161impd_parametric_lim_type_drc_reset(WORD32 instance_idx,
162                                   ia_parametric_drc_type_lim_params_struct*
163                                       pstr_parametric_lim_type_drc_params);
164
165WORD32
166impd_parametric_drc_instance_process(
167    FLOAT32* audio_in_out_buf[], FLOAT32* audio_real_buff[],
168    FLOAT32* audio_imag_buff[],
169    ia_parametric_drc_params_struct* pstr_parametric_drc_params,
170    ia_parametric_drc_instance_params_struct*
171        pstr_parametric_drc_instance_params);
172
173WORD32
174impd_parametric_ffwd_type_drc_process(FLOAT32* audio_in_out_buf[],
175                                      FLOAT32* audio_real_buff[],
176                                      FLOAT32* audio_imag_buff[],
177                                      WORD32 nodeIdx,
178                                      ia_parametric_drc_type_ff_params_struct*
179                                          pstr_parametric_ffwd_type_drc_params,
180                                      ia_spline_nodes_struct* str_spline_nodes);
181
182WORD32
183impd_parametric_lim_type_drc_process(FLOAT32* audio_in_out_buf[],
184                                     FLOAT32 loudness_normalization_gain_db,
185                                     ia_parametric_drc_type_lim_params_struct*
186                                         pstr_parametric_lim_type_drc_params,
187                                     FLOAT32* lpcm_gains);
188
189#ifdef __cplusplus
190}
191#endif
192#endif
193