blob: 4a42fd6c4aa4a6c0d414f7bb47ab42801da14c4c [file] [log] [blame]
/******************************************************************************
*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*****************************************************************************
* Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
*/
#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include "impd_type_def.h"
#include "impd_drc_extr_delta_coded_info.h"
#include "impd_drc_common.h"
#include "impd_drc_struct.h"
#include "impd_parametric_drc_dec.h"
#include "impd_drc_filter_bank.h"
#include "impd_drc_rom.h"
#define PI 3.14159265f
#ifndef max
#define max(a, b) (((a) > (b)) ? (a) : (b))
#endif
#ifndef min
#define min(a, b) (((a) < (b)) ? (a) : (b))
#endif
WORD32 impd_init_parametric_drc(
WORD32 drc_frame_size, WORD32 sampling_rate, WORD32 sub_band_domain_mode,
ia_parametric_drc_params_struct* p_parametricdrc_params) {
static const WORD32 sub_band_count_tbl[4] = {0, 64, 71, 256};
p_parametricdrc_params->drc_frame_size = drc_frame_size;
p_parametricdrc_params->sampling_rate = sampling_rate;
p_parametricdrc_params->sub_band_domain_mode = sub_band_domain_mode;
p_parametricdrc_params->sub_band_count =
sub_band_count_tbl[sub_band_domain_mode];
return 0;
}
WORD32 impd_init_parametric_drc_feed_fwd(
ia_drc_config* pstr_drc_config, WORD32 instance_idx,
WORD32 ch_count_from_dwnmix_id,
ia_parametric_drc_params_struct* p_parametricdrc_params) {
WORD32 err = 0, i = 0;
WORD32 parametric_drc_idx =
p_parametricdrc_params->parametric_drc_idx[instance_idx];
WORD32 gain_set_index = p_parametricdrc_params->gain_set_index[instance_idx];
WORD32* channel_map = p_parametricdrc_params->channel_map[instance_idx];
ia_drc_coeff_parametric_drc_struct* hDrcCoefficientsParametricDrcBs =
&(pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc);
ia_parametric_drc_type_feed_forward_struct* hParametricDrcTypeFeedForwardBs =
&(pstr_drc_config->str_drc_config_ext
.str_parametric_drc_instructions[parametric_drc_idx]
.str_parametric_drc_type_feed_forward);
ia_parametric_drc_type_ff_params_struct*
pstr_parametric_ffwd_type_drc_params =
&(p_parametricdrc_params
->str_parametric_drc_instance_params[instance_idx]
.str_parametric_drc_type_ff_params);
/* level estimation */
pstr_parametric_ffwd_type_drc_params->frame_size =
p_parametricdrc_params->parametric_drc_frame_size;
pstr_parametric_ffwd_type_drc_params->sub_band_domain_mode =
p_parametricdrc_params->sub_band_domain_mode;
pstr_parametric_ffwd_type_drc_params->sub_band_count =
p_parametricdrc_params->sub_band_count;
pstr_parametric_ffwd_type_drc_params->sub_band_compensation_type = 0;
if (pstr_parametric_ffwd_type_drc_params->sub_band_domain_mode ==
SUBBAND_DOMAIN_MODE_QMF64) {
if (p_parametricdrc_params->sampling_rate == 48000) {
pstr_parametric_ffwd_type_drc_params->sub_band_compensation_type = 1;
} else {
/* support of other sampling rates than 48000 might be missing */
return UNEXPECTED_ERROR;
}
}
pstr_parametric_ffwd_type_drc_params->audio_num_chan =
p_parametricdrc_params->audio_num_chan;
pstr_parametric_ffwd_type_drc_params->level_estim_k_weighting_type =
hParametricDrcTypeFeedForwardBs->level_estim_k_weighting_type;
pstr_parametric_ffwd_type_drc_params->level_estim_integration_time =
hParametricDrcTypeFeedForwardBs->level_estim_integration_time;
pstr_parametric_ffwd_type_drc_params->level_estim_frame_index = 0;
pstr_parametric_ffwd_type_drc_params->level_estim_frame_count =
hParametricDrcTypeFeedForwardBs->level_estim_integration_time /
pstr_parametric_ffwd_type_drc_params->frame_size;
memset(pstr_parametric_ffwd_type_drc_params->level, 0,
PARAM_DRC_TYPE_FF_LEVEL_ESTIM_FRAME_COUNT_MAX * sizeof(FLOAT32));
if (ch_count_from_dwnmix_id != 0) {
memcpy(pstr_parametric_ffwd_type_drc_params->level_estim_ch_weight,
hDrcCoefficientsParametricDrcBs
->str_parametric_drc_gain_set_params[gain_set_index]
.level_estim_ch_weight,
ch_count_from_dwnmix_id * sizeof(FLOAT32));
} else {
for (i = 0; i < pstr_parametric_ffwd_type_drc_params->audio_num_chan; i++) {
pstr_parametric_ffwd_type_drc_params->level_estim_ch_weight[i] =
(FLOAT32)channel_map[i];
}
}
if (pstr_parametric_ffwd_type_drc_params->sub_band_domain_mode ==
SUBBAND_DOMAIN_MODE_OFF) {
err = impd_init_lvl_est_filt_time(
pstr_parametric_ffwd_type_drc_params->level_estim_k_weighting_type,
p_parametricdrc_params->sampling_rate,
&pstr_parametric_ffwd_type_drc_params->pre_filt_coeff,
&pstr_parametric_ffwd_type_drc_params->rlb_filt_coeff);
if (err) return (err);
} else {
err = impd_init_lvl_est_filt_subband(
pstr_parametric_ffwd_type_drc_params->level_estim_k_weighting_type,
p_parametricdrc_params->sampling_rate,
p_parametricdrc_params->sub_band_domain_mode,
p_parametricdrc_params->sub_band_count,
pstr_parametric_ffwd_type_drc_params->sub_band_compensation_type,
pstr_parametric_ffwd_type_drc_params->weighting_filt,
&pstr_parametric_ffwd_type_drc_params->filt_coeff_subband);
if (err) return (err);
}
pstr_parametric_ffwd_type_drc_params->node_count =
hParametricDrcTypeFeedForwardBs->node_count;
memcpy(pstr_parametric_ffwd_type_drc_params->node_level,
hParametricDrcTypeFeedForwardBs->node_level,
pstr_parametric_ffwd_type_drc_params->node_count * sizeof(WORD32));
memcpy(pstr_parametric_ffwd_type_drc_params->node_gain,
hParametricDrcTypeFeedForwardBs->node_gain,
pstr_parametric_ffwd_type_drc_params->node_count * sizeof(WORD32));
pstr_parametric_ffwd_type_drc_params->ref_level_parametric_drc =
hDrcCoefficientsParametricDrcBs
->str_parametric_drc_gain_set_params[gain_set_index]
.drc_input_loudness;
{
WORD32 gain_smooth_attack_time_fast =
hParametricDrcTypeFeedForwardBs->gain_smooth_attack_time_fast;
WORD32 gain_smooth_release_time_fast =
hParametricDrcTypeFeedForwardBs->gain_smooth_release_time_fast;
WORD32 gain_smooth_attack_time_slow =
hParametricDrcTypeFeedForwardBs->gain_smooth_attack_time_slow;
WORD32 gain_smooth_release_time_slow =
hParametricDrcTypeFeedForwardBs->gain_smooth_release_time_slow;
WORD32 gain_smooth_hold_off =
hParametricDrcTypeFeedForwardBs->gain_smooth_hold_off;
WORD32 sampling_rate = p_parametricdrc_params->sampling_rate;
WORD32 parametric_drc_frame_size =
p_parametricdrc_params->parametric_drc_frame_size;
pstr_parametric_ffwd_type_drc_params->gain_smooth_attack_alpha_fast =
1 -
(FLOAT32)exp(-1.0 * parametric_drc_frame_size /
(gain_smooth_attack_time_fast * sampling_rate * 0.001));
pstr_parametric_ffwd_type_drc_params->gain_smooth_rel_alpha_fast =
1 -
(FLOAT32)exp(-1.0 * parametric_drc_frame_size /
(gain_smooth_release_time_fast * sampling_rate * 0.001));
pstr_parametric_ffwd_type_drc_params->gain_smooth_attack_alpha_slow =
1 -
(FLOAT32)exp(-1.0 * parametric_drc_frame_size /
(gain_smooth_attack_time_slow * sampling_rate * 0.001));
pstr_parametric_ffwd_type_drc_params->gain_smooth_rel_alpha_slow =
1 -
(FLOAT32)exp(-1.0 * parametric_drc_frame_size /
(gain_smooth_release_time_slow * sampling_rate * 0.001));
pstr_parametric_ffwd_type_drc_params->gain_smooth_hold_off_count =
gain_smooth_hold_off * 256 * sampling_rate /
(parametric_drc_frame_size * 48000);
pstr_parametric_ffwd_type_drc_params->gain_smooth_attack_threshold =
hParametricDrcTypeFeedForwardBs->gain_smooth_attack_threshold;
pstr_parametric_ffwd_type_drc_params->gain_smooth_rel_threshold =
hParametricDrcTypeFeedForwardBs->gain_smooth_rel_threshold;
}
err =
impd_parametric_ffwd_type_drc_reset(pstr_parametric_ffwd_type_drc_params);
if (err) return (err);
return 0;
}
VOID impd_init_parametric_drc_lim(
ia_drc_config* pstr_drc_config, WORD32 instance_idx,
WORD32 ch_count_from_dwnmix_id,
ia_parametric_drc_params_struct* p_parametricdrc_params, pVOID* mem_ptr) {
WORD32 i = 0;
UWORD32 j;
UWORD32 attack, sec_len;
WORD32 parametric_drc_idx =
p_parametricdrc_params->parametric_drc_idx[instance_idx];
WORD32 gain_set_index = p_parametricdrc_params->gain_set_index[instance_idx];
WORD32* channel_map = p_parametricdrc_params->channel_map[instance_idx];
ia_drc_coeff_parametric_drc_struct* hDrcCoefficientsParametricDrcBs =
&(pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc);
ia_parametric_drc_lim_struct* hParametricDrcTypeLimBs =
&(pstr_drc_config->str_drc_config_ext
.str_parametric_drc_instructions[parametric_drc_idx]
.parametric_drc_lim);
ia_parametric_drc_type_lim_params_struct*
pstr_parametric_lim_type_drc_params =
&(p_parametricdrc_params
->str_parametric_drc_instance_params[instance_idx]
.str_parametric_drc_type_lim_params);
pstr_parametric_lim_type_drc_params->frame_size =
p_parametricdrc_params->drc_frame_size;
pstr_parametric_lim_type_drc_params->audio_num_chan =
p_parametricdrc_params->audio_num_chan;
if (ch_count_from_dwnmix_id != 0) {
memcpy(pstr_parametric_lim_type_drc_params->level_estim_ch_weight,
hDrcCoefficientsParametricDrcBs
->str_parametric_drc_gain_set_params[gain_set_index]
.level_estim_ch_weight,
ch_count_from_dwnmix_id * sizeof(FLOAT32));
} else {
for (i = 0; i < pstr_parametric_lim_type_drc_params->audio_num_chan; i++) {
pstr_parametric_lim_type_drc_params->level_estim_ch_weight[i] =
(FLOAT32)channel_map[i];
}
}
attack = (UWORD32)(hParametricDrcTypeLimBs->parametric_lim_attack *
p_parametricdrc_params->sampling_rate / 1000);
sec_len = (UWORD32)sqrt(attack + 1);
pstr_parametric_lim_type_drc_params->sec_len = sec_len;
pstr_parametric_lim_type_drc_params->num_max_buf_sec = (attack + 1) / sec_len;
if (pstr_parametric_lim_type_drc_params->num_max_buf_sec * sec_len <
(attack + 1))
pstr_parametric_lim_type_drc_params->num_max_buf_sec++;
pstr_parametric_lim_type_drc_params->max_buf = (FLOAT32*)(*mem_ptr);
*mem_ptr = (pVOID)((SIZE_T)(*mem_ptr) +
pstr_parametric_lim_type_drc_params->num_max_buf_sec *
sec_len * sizeof(FLOAT32));
pstr_parametric_lim_type_drc_params->attack_ms =
(FLOAT32)hParametricDrcTypeLimBs->parametric_lim_attack;
pstr_parametric_lim_type_drc_params->release_ms =
(FLOAT32)hParametricDrcTypeLimBs->parametric_lim_release;
pstr_parametric_lim_type_drc_params->attack = attack;
pstr_parametric_lim_type_drc_params->attack_constant =
(FLOAT32)pow(0.1, 1.0 / (attack + 1));
pstr_parametric_lim_type_drc_params->release_constant = (FLOAT32)pow(
0.1, 1.0 / (hParametricDrcTypeLimBs->parametric_lim_release *
p_parametricdrc_params->sampling_rate / 1000 +
1));
pstr_parametric_lim_type_drc_params->threshold = (FLOAT32)pow(
10.0f, 0.05f * hParametricDrcTypeLimBs->parametric_lim_threshold);
pstr_parametric_lim_type_drc_params->channels =
pstr_parametric_lim_type_drc_params->audio_num_chan;
pstr_parametric_lim_type_drc_params->sampling_rate =
p_parametricdrc_params->sampling_rate;
pstr_parametric_lim_type_drc_params->cor = 1.0f;
pstr_parametric_lim_type_drc_params->smooth_state_0 = 1.0;
for (j = 0; j < pstr_parametric_lim_type_drc_params->num_max_buf_sec *
pstr_parametric_lim_type_drc_params->sec_len;
j++) {
pstr_parametric_lim_type_drc_params->max_buf[j] = 0.f;
}
}
WORD32 impd_init_parametric_drcInstance(
ia_drc_config* pstr_drc_config, WORD32 instance_idx,
WORD32 ch_count_from_dwnmix_id,
ia_parametric_drc_params_struct* p_parametricdrc_params, pVOID* mem_ptr) {
WORD32 err = 0;
WORD32 parametric_drc_idx =
p_parametricdrc_params->parametric_drc_idx[instance_idx];
ia_parametric_drc_instructions_struct* hParametricDrcInstructions =
&(pstr_drc_config->str_drc_config_ext
.str_parametric_drc_instructions[parametric_drc_idx]);
p_parametricdrc_params->str_parametric_drc_instance_params[instance_idx]
.disable_paramteric_drc =
hParametricDrcInstructions->disable_paramteric_drc;
p_parametricdrc_params->str_parametric_drc_instance_params[instance_idx]
.parametric_drc_type = hParametricDrcInstructions->parametric_drc_type;
p_parametricdrc_params->str_parametric_drc_instance_params[instance_idx]
.str_spline_nodes.num_nodes = p_parametricdrc_params->num_nodes;
if (p_parametricdrc_params->str_parametric_drc_instance_params[instance_idx]
.disable_paramteric_drc == 0) {
if (p_parametricdrc_params->str_parametric_drc_instance_params[instance_idx]
.parametric_drc_type == PARAM_DRC_TYPE_FF) {
err = impd_init_parametric_drc_feed_fwd(pstr_drc_config, instance_idx,
ch_count_from_dwnmix_id,
p_parametricdrc_params);
if (err) return (err);
} else if (p_parametricdrc_params
->str_parametric_drc_instance_params[instance_idx]
.parametric_drc_type == PARAM_DRC_TYPE_LIM) {
p_parametricdrc_params->str_parametric_drc_instance_params[instance_idx]
.str_spline_nodes.num_nodes = p_parametricdrc_params->drc_frame_size;
impd_init_parametric_drc_lim(pstr_drc_config, instance_idx,
ch_count_from_dwnmix_id,
p_parametricdrc_params, mem_ptr);
} else {
return (UNEXPECTED_ERROR);
}
}
return 0;
}
WORD32 impd_init_parametric_drc_after_config(
ia_drc_config* pstr_drc_config,
ia_drc_loudness_info_set_struct* pstr_loudness_info,
ia_parametric_drc_params_struct* p_parametricdrc_params, pVOID* mem_ptr) {
WORD32 err = 0, instance_idx = 0, gain_set_index = 0,
side_chain_config_type = 0, downmix_id = 0,
ch_count_from_dwnmix_id = 0, L = 0;
p_parametricdrc_params->parametric_drc_frame_size =
pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
.parametric_drc_frame_size;
p_parametricdrc_params->reset_parametric_drc =
pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
.reset_parametric_drc;
p_parametricdrc_params->num_nodes =
p_parametricdrc_params->drc_frame_size /
p_parametricdrc_params->parametric_drc_frame_size;
switch (p_parametricdrc_params->sub_band_domain_mode) {
case SUBBAND_DOMAIN_MODE_QMF64:
L = AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF64;
break;
case SUBBAND_DOMAIN_MODE_QMF71:
L = AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF71;
break;
case SUBBAND_DOMAIN_MODE_STFT256:
L = AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_STFT256;
break;
case SUBBAND_DOMAIN_MODE_OFF:
default:
L = 0;
break;
}
if (p_parametricdrc_params->sub_band_domain_mode != SUBBAND_DOMAIN_MODE_OFF &&
p_parametricdrc_params->parametric_drc_frame_size != L) {
return (EXTERNAL_ERROR);
}
for (instance_idx = 0;
instance_idx < p_parametricdrc_params->parametric_drc_instance_count;
instance_idx++) {
gain_set_index = p_parametricdrc_params->gain_set_index[instance_idx];
side_chain_config_type =
pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
.str_parametric_drc_gain_set_params[gain_set_index]
.side_chain_config_type;
downmix_id = pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
.str_parametric_drc_gain_set_params[gain_set_index]
.downmix_id;
if (side_chain_config_type == 1 &&
downmix_id ==
p_parametricdrc_params
->dwnmix_id_from_drc_instructions[instance_idx]) {
ch_count_from_dwnmix_id =
pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
.str_parametric_drc_gain_set_params[gain_set_index]
.ch_count_from_dwnmix_id;
} else {
ch_count_from_dwnmix_id = 0;
}
if (pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
.str_parametric_drc_gain_set_params[gain_set_index]
.drc_input_loudness_present == 0) {
WORD32 n = 0, m = 0, drcInputLoudnessFound = 0;
FLOAT32 drc_input_loudness = 0.f;
for (n = 0; n < pstr_loudness_info->loudness_info_count; n++) {
ia_loudness_info_struct* loudness_info =
&pstr_loudness_info->loudness_info[n];
if (p_parametricdrc_params
->dwnmix_id_from_drc_instructions[instance_idx] ==
loudness_info->downmix_id) {
if (0 == loudness_info->drc_set_id) {
for (m = 0; m < loudness_info->measurement_count; m++) {
if (loudness_info->loudness_measure[m].method_def ==
METHOD_DEFINITION_PROGRAM_LOUDNESS) {
drc_input_loudness =
loudness_info->loudness_measure[m].method_val;
drcInputLoudnessFound = 1;
break;
}
}
if (drcInputLoudnessFound == 0) {
for (m = 0; m < loudness_info->measurement_count; m++) {
if (loudness_info->loudness_measure[m].method_def ==
METHOD_DEFINITION_ANCHOR_LOUDNESS) {
drc_input_loudness =
loudness_info->loudness_measure[m].method_val;
drcInputLoudnessFound = 1;
break;
}
}
}
}
}
}
if (drcInputLoudnessFound == 0) {
for (n = 0; n < pstr_loudness_info->loudness_info_count; n++) {
ia_loudness_info_struct* loudness_info =
&pstr_loudness_info->loudness_info[n];
if (0 == loudness_info->downmix_id) {
if (0 == loudness_info->drc_set_id) {
for (m = 0; m < loudness_info->measurement_count; m++) {
if (loudness_info->loudness_measure[m].method_def ==
METHOD_DEFINITION_PROGRAM_LOUDNESS) {
drc_input_loudness =
loudness_info->loudness_measure[m].method_val;
drcInputLoudnessFound = 1;
break;
}
}
if (drcInputLoudnessFound == 0) {
for (m = 0; m < loudness_info->measurement_count; m++) {
if (loudness_info->loudness_measure[m].method_def ==
METHOD_DEFINITION_ANCHOR_LOUDNESS) {
drc_input_loudness =
loudness_info->loudness_measure[m].method_val;
drcInputLoudnessFound = 1;
break;
}
}
}
}
}
}
}
if (drcInputLoudnessFound == 0) {
return (UNEXPECTED_ERROR);
} else {
pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
.str_parametric_drc_gain_set_params[gain_set_index]
.drc_input_loudness = drc_input_loudness;
}
}
err = impd_init_parametric_drcInstance(pstr_drc_config, instance_idx,
ch_count_from_dwnmix_id,
p_parametricdrc_params, mem_ptr);
if (err) return (err);
}
return 0;
}
WORD32 impd_init_lvl_est_filt_time(
WORD32 level_estim_k_weighting_type, WORD32 sampling_rate,
ia_2nd_order_filt_coeff_struct* pre_filt_coeff,
ia_2nd_order_filt_coeff_struct* rlb_filt_coeff) {
WORD32 i;
const FLOAT32* ptr_samp_tbl;
switch (sampling_rate) {
case 96000:
i = 0;
break;
case 88200:
i = 1;
break;
case 64000:
i = 2;
break;
case 48000:
i = 3;
break;
case 44100:
i = 4;
break;
case 32000:
i = 5;
break;
case 24000:
i = 6;
break;
case 22050:
i = 7;
break;
case 16000:
i = 8;
break;
case 12000:
i = 9;
break;
case 11025:
i = 10;
break;
case 8000:
i = 11;
break;
case 7350:
i = 12;
break;
default:
i = 3;
break;
}
ptr_samp_tbl = samp_rate_tbl[i];
if (level_estim_k_weighting_type == 2) {
pre_filt_coeff->b0 = ptr_samp_tbl[0];
pre_filt_coeff->b1 = ptr_samp_tbl[1];
pre_filt_coeff->b2 = ptr_samp_tbl[2];
pre_filt_coeff->a1 = ptr_samp_tbl[3];
pre_filt_coeff->a2 = ptr_samp_tbl[4];
}
if (level_estim_k_weighting_type == 1 || level_estim_k_weighting_type == 2) {
rlb_filt_coeff->b0 = ptr_samp_tbl[5];
rlb_filt_coeff->b1 = ptr_samp_tbl[6];
rlb_filt_coeff->b2 = ptr_samp_tbl[7];
rlb_filt_coeff->a1 = ptr_samp_tbl[8];
rlb_filt_coeff->a2 = ptr_samp_tbl[9];
}
return 0;
}
WORD32 impd_init_lvl_est_filt_subband(
WORD32 level_estim_k_weighting_type, WORD32 sampling_rate,
WORD32 sub_band_domain_mode, WORD32 sub_band_count,
WORD32 sub_band_compensation_type, FLOAT32* weighting_filt,
ia_2nd_order_filt_coeff_struct* filt_coeff_subband) {
FLOAT32 w0, alpha, sinw0, cosw0;
FLOAT32 b0, b1, b2, a0, a1, a2;
FLOAT32 num_real, num_imag, den_real, den_imag;
const FLOAT32* f_bands_nrm;
WORD32 b;
WORD32 i;
const FLOAT32* ptr_samp_tbl;
switch (sampling_rate) {
case 96000:
i = 0;
break;
case 88200:
i = 1;
break;
case 64000:
i = 2;
break;
case 48000:
i = 3;
break;
case 44100:
i = 4;
break;
case 32000:
i = 5;
break;
case 24000:
i = 6;
break;
case 22050:
i = 7;
break;
case 16000:
i = 8;
break;
case 12000:
i = 9;
break;
case 11025:
i = 10;
break;
case 8000:
i = 11;
break;
case 7350:
i = 12;
break;
default:
i = 3;
break;
}
ptr_samp_tbl = samp_rate_tbl[i];
switch (sub_band_domain_mode) {
case SUBBAND_DOMAIN_MODE_QMF64:
f_bands_nrm = f_bands_nrm_QMF64;
break;
case SUBBAND_DOMAIN_MODE_QMF71:
f_bands_nrm = f_bands_nrm_QMF71;
break;
case SUBBAND_DOMAIN_MODE_STFT256:
f_bands_nrm = f_bands_nrm_STFT256;
break;
default:
return UNEXPECTED_ERROR;
break;
}
for (b = 0; b < sub_band_count; b++) {
weighting_filt[b] = 1.f;
}
if (level_estim_k_weighting_type == 2) {
b0 = ptr_samp_tbl[0];
b1 = ptr_samp_tbl[1];
b2 = ptr_samp_tbl[2];
a1 = ptr_samp_tbl[3];
a2 = ptr_samp_tbl[4];
a0 = 1.f;
for (b = 0; b < sub_band_count; b++) {
num_real = b0 + b1 * (FLOAT32)cos(PI * f_bands_nrm[b]) +
b2 * (FLOAT32)cos(PI * 2 * f_bands_nrm[b]);
num_imag = -b1 * (FLOAT32)sin(PI * f_bands_nrm[b]) -
b2 * (FLOAT32)sin(PI * 2 * f_bands_nrm[b]);
den_real = a0 + a1 * (FLOAT32)cos(PI * f_bands_nrm[b]) +
a2 * (FLOAT32)cos(PI * 2 * f_bands_nrm[b]);
den_imag = -a1 * (FLOAT32)sin(PI * f_bands_nrm[b]) -
a2 * (FLOAT32)sin(PI * 2 * f_bands_nrm[b]);
weighting_filt[b] *=
(FLOAT32)(sqrt((num_real * num_real + num_imag * num_imag) /
(den_real * den_real + den_imag * den_imag)));
}
}
if (level_estim_k_weighting_type == 1 || level_estim_k_weighting_type == 2) {
b0 = ptr_samp_tbl[5];
b1 = ptr_samp_tbl[6];
b2 = ptr_samp_tbl[7];
a1 = ptr_samp_tbl[8];
a2 = ptr_samp_tbl[9];
a0 = 1.f;
for (b = 0; b < sub_band_count; b++) {
if (!(sub_band_compensation_type == 1 && b == 0)) {
num_real = (FLOAT32)(b0 + b1 * cos(PI * f_bands_nrm[b]) +
b2 * cos(PI * 2 * f_bands_nrm[b]));
num_imag = (FLOAT32)(-b1 * sin(PI * f_bands_nrm[b]) -
b2 * sin(PI * 2 * f_bands_nrm[b]));
den_real = (FLOAT32)(a0 + a1 * cos(PI * f_bands_nrm[b]) +
a2 * cos(PI * 2 * f_bands_nrm[b]));
den_imag = (FLOAT32)(-a1 * sin(PI * f_bands_nrm[b]) -
a2 * sin(PI * 2 * f_bands_nrm[b]));
weighting_filt[b] *=
(FLOAT32)(sqrt((num_real * num_real + num_imag * num_imag) /
(den_real * den_real + den_imag * den_imag)));
}
}
if (sub_band_compensation_type == 1) {
w0 = 2.0f * PI * 38.0f / (FLOAT32)sampling_rate *
AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF64;
sinw0 = (FLOAT32)sin(w0);
cosw0 = (FLOAT32)cos(w0);
alpha = sinw0;
b0 = (1 + cosw0) / 2;
b1 = -(1 + cosw0);
b2 = (1 + cosw0) / 2;
a0 = 1 + alpha;
a1 = -2 * cosw0;
a2 = 1 - alpha;
filt_coeff_subband->b0 = b0 / a0;
filt_coeff_subband->b1 = b1 / a0;
filt_coeff_subband->b2 = b2 / a0;
filt_coeff_subband->a1 = a1 / a0;
filt_coeff_subband->a2 = a2 / a0;
}
}
return 0;
}
WORD32 impd_parametric_ffwd_type_drc_reset(
ia_parametric_drc_type_ff_params_struct*
pstr_parametric_ffwd_type_drc_params) {
WORD32 i = 0;
pstr_parametric_ffwd_type_drc_params->level_estim_frame_index = 0;
pstr_parametric_ffwd_type_drc_params->start_up_phase = 1;
for (i = 0; i < PARAM_DRC_TYPE_FF_LEVEL_ESTIM_FRAME_COUNT_MAX; i++) {
pstr_parametric_ffwd_type_drc_params->level[i] = 0.f;
}
for (i = 0; i < MAX_CHANNEL_COUNT; i++) {
pstr_parametric_ffwd_type_drc_params->pre_filt_state[i].z1 = 0.f;
pstr_parametric_ffwd_type_drc_params->pre_filt_state[i].z2 = 0.f;
pstr_parametric_ffwd_type_drc_params->rlb_filt_state[i].z1 = 0.f;
pstr_parametric_ffwd_type_drc_params->rlb_filt_state[i].z2 = 0.f;
pstr_parametric_ffwd_type_drc_params->filt_state_subband_real[i].z1 = 0.f;
pstr_parametric_ffwd_type_drc_params->filt_state_subband_real[i].z2 = 0.f;
pstr_parametric_ffwd_type_drc_params->filt_state_subband_imag[i].z1 = 0.f;
pstr_parametric_ffwd_type_drc_params->filt_state_subband_imag[i].z2 = 0.f;
}
pstr_parametric_ffwd_type_drc_params->db_level_smooth = -135.f;
pstr_parametric_ffwd_type_drc_params->db_gain_smooth = 0.f;
pstr_parametric_ffwd_type_drc_params->hold_counter = 0;
return 0;
}
WORD32 impd_parametric_drc_instance_process(
FLOAT32* audio_in_out_buf[], FLOAT32* audio_real_buff[],
FLOAT32* audio_imag_buff[],
ia_parametric_drc_params_struct* p_parametricdrc_params,
ia_parametric_drc_instance_params_struct*
pstr_parametric_drc_instance_params) {
WORD32 err = 0, i = 0;
if (pstr_parametric_drc_instance_params->disable_paramteric_drc) {
for (i = 0; i < p_parametricdrc_params->num_nodes; i++) {
pstr_parametric_drc_instance_params->str_spline_nodes.str_node[i]
.loc_db_gain = 0.f;
pstr_parametric_drc_instance_params->str_spline_nodes.str_node[i].slope =
0.f;
pstr_parametric_drc_instance_params->str_spline_nodes.str_node[i].time =
(i + 1) * p_parametricdrc_params->parametric_drc_frame_size - 1;
}
} else {
if (pstr_parametric_drc_instance_params->parametric_drc_type ==
PARAM_DRC_TYPE_FF) {
ia_parametric_drc_type_ff_params_struct*
pstr_parametric_ffwd_type_drc_params =
&(pstr_parametric_drc_instance_params
->str_parametric_drc_type_ff_params);
for (i = 0; i < p_parametricdrc_params->num_nodes; i++) {
err = impd_parametric_ffwd_type_drc_process(
audio_in_out_buf, audio_real_buff, audio_imag_buff, i,
pstr_parametric_ffwd_type_drc_params,
&pstr_parametric_drc_instance_params->str_spline_nodes);
if (err) return (err);
}
} else if (pstr_parametric_drc_instance_params->parametric_drc_type ==
PARAM_DRC_TYPE_LIM) {
return (UNEXPECTED_ERROR);
} else {
return (UNEXPECTED_ERROR);
}
}
return 0;
}
VOID iir_second_order_filter(ia_2nd_order_filt_coeff_struct* coeff,
ia_2nd_order_filt_state_struct* state,
WORD32 frame_len, FLOAT32* input,
FLOAT32* output) {
FLOAT32 z2 = state->z2;
FLOAT32 z1 = state->z1;
FLOAT32 z0;
WORD32 i;
for (i = 0; i < frame_len; i++) {
z0 = input[i] - coeff->a1 * z1 - coeff->a2 * z2;
output[i] = coeff->b0 * z0 + coeff->b1 * z1 + coeff->b2 * z2;
z2 = z1;
z1 = z0;
}
state->z1 = z1;
state->z2 = z2;
}
WORD32 impd_parametric_ffwd_type_drc_process(
FLOAT32* audio_in_out_buf[], FLOAT32* audio_real_buff[],
FLOAT32* audio_imag_buff[], WORD32 nodeIdx,
ia_parametric_drc_type_ff_params_struct*
pstr_parametric_ffwd_type_drc_params,
ia_spline_nodes_struct* str_spline_nodes) {
WORD32 c, t, b, n, i, offset;
FLOAT32 x, y, channelLevel, level, levelDb, loc_db_gain, levelDelta, alpha;
WORD32 frame_size = pstr_parametric_ffwd_type_drc_params->frame_size;
WORD32 sub_band_count = pstr_parametric_ffwd_type_drc_params->sub_band_count;
FLOAT32* level_estim_ch_weight =
pstr_parametric_ffwd_type_drc_params->level_estim_ch_weight;
WORD32 level_estim_k_weighting_type =
pstr_parametric_ffwd_type_drc_params->level_estim_k_weighting_type;
ia_2nd_order_filt_coeff_struct preC =
pstr_parametric_ffwd_type_drc_params->pre_filt_coeff;
ia_2nd_order_filt_coeff_struct rlbC =
pstr_parametric_ffwd_type_drc_params->rlb_filt_coeff;
ia_2nd_order_filt_state_struct* preS =
pstr_parametric_ffwd_type_drc_params->pre_filt_state;
ia_2nd_order_filt_state_struct* rlbS =
pstr_parametric_ffwd_type_drc_params->rlb_filt_state;
ia_2nd_order_filt_coeff_struct rlbC_sb =
pstr_parametric_ffwd_type_drc_params->filt_coeff_subband;
ia_2nd_order_filt_state_struct* rlbS_sbReal =
pstr_parametric_ffwd_type_drc_params->filt_state_subband_real;
ia_2nd_order_filt_state_struct* rlbS_sbImag =
pstr_parametric_ffwd_type_drc_params->filt_state_subband_imag;
FLOAT32* weighting_filt =
pstr_parametric_ffwd_type_drc_params->weighting_filt;
WORD32 sub_band_compensation_type =
pstr_parametric_ffwd_type_drc_params->sub_band_compensation_type;
if (audio_in_out_buf != NULL) {
level = 0;
offset = nodeIdx * pstr_parametric_ffwd_type_drc_params->frame_size;
for (c = 0; c < pstr_parametric_ffwd_type_drc_params->audio_num_chan; c++) {
channelLevel = 0.f;
if (!level_estim_ch_weight[c]) continue;
if (level_estim_k_weighting_type == 0) {
for (t = 0; t < frame_size; t++) {
x = audio_in_out_buf[c][offset + t];
channelLevel += x * x;
}
} else if (level_estim_k_weighting_type == 1) {
for (t = 0; t < frame_size; t++) {
x = audio_in_out_buf[c][offset + t];
iir_second_order_filter(&rlbC, &rlbS[c], 1, &x, &x);
channelLevel += x * x;
}
} else if (level_estim_k_weighting_type == 2) {
for (t = 0; t < frame_size; t++) {
x = audio_in_out_buf[c][offset + t];
iir_second_order_filter(&preC, &preS[c], 1, &x, &x);
iir_second_order_filter(&rlbC, &rlbS[c], 1, &x, &x);
channelLevel += x * x;
}
} else {
return (UNEXPECTED_ERROR);
}
level += level_estim_ch_weight[c] * channelLevel;
}
} else {
level = 0;
offset = nodeIdx * pstr_parametric_ffwd_type_drc_params->sub_band_count;
for (c = 0; c < pstr_parametric_ffwd_type_drc_params->audio_num_chan; c++) {
channelLevel = 0.f;
if (!level_estim_ch_weight[c]) continue;
if (level_estim_k_weighting_type == 0) {
for (b = 0; b < sub_band_count; b++) {
x = audio_real_buff[c][offset + b];
y = audio_imag_buff[c][offset + b];
channelLevel += x * x + y * y;
}
} else if (level_estim_k_weighting_type == 1 ||
level_estim_k_weighting_type == 2) {
for (b = 0; b < sub_band_count; b++) {
x = audio_real_buff[c][offset + b] * weighting_filt[b];
y = audio_imag_buff[c][offset + b] * weighting_filt[b];
if (b == 0 && sub_band_compensation_type == 1) {
iir_second_order_filter(&rlbC_sb, &rlbS_sbReal[c], 1, &x, &x);
iir_second_order_filter(&rlbC_sb, &rlbS_sbImag[c], 1, &y, &y);
}
channelLevel += x * x + y * y;
}
} else {
return (UNEXPECTED_ERROR);
}
level += level_estim_ch_weight[c] * channelLevel;
}
level /= sub_band_count;
}
pstr_parametric_ffwd_type_drc_params
->level[pstr_parametric_ffwd_type_drc_params->level_estim_frame_index] =
level;
pstr_parametric_ffwd_type_drc_params->level_estim_frame_index++;
level = 0.f;
if (pstr_parametric_ffwd_type_drc_params->start_up_phase) {
for (i = 0;
i < pstr_parametric_ffwd_type_drc_params->level_estim_frame_index;
i++) {
level += pstr_parametric_ffwd_type_drc_params->level[i];
}
level /= pstr_parametric_ffwd_type_drc_params->level_estim_frame_index *
pstr_parametric_ffwd_type_drc_params->frame_size;
} else {
for (i = 0;
i < pstr_parametric_ffwd_type_drc_params->level_estim_frame_count;
i++) {
level += pstr_parametric_ffwd_type_drc_params->level[i];
}
level /= pstr_parametric_ffwd_type_drc_params->level_estim_integration_time;
}
if (pstr_parametric_ffwd_type_drc_params->level_estim_frame_index ==
pstr_parametric_ffwd_type_drc_params->level_estim_frame_count) {
pstr_parametric_ffwd_type_drc_params->level_estim_frame_index = 0;
pstr_parametric_ffwd_type_drc_params->start_up_phase = 0;
}
if (level < 1e-10f) level = 1e-10f;
if (level_estim_k_weighting_type == 2) {
levelDb = -0.691f + 10 * (FLOAT32)log10(level) + 3;
} else {
levelDb = 10 * (FLOAT32)log10(level) + 3;
}
levelDb -= pstr_parametric_ffwd_type_drc_params->ref_level_parametric_drc;
for (n = 0; n < pstr_parametric_ffwd_type_drc_params->node_count; n++) {
if (levelDb <=
(FLOAT32)pstr_parametric_ffwd_type_drc_params->node_level[n]) {
break;
}
}
if (n == 0) {
loc_db_gain = (FLOAT32)pstr_parametric_ffwd_type_drc_params->node_gain[n];
} else if (n == pstr_parametric_ffwd_type_drc_params->node_count) {
loc_db_gain =
(FLOAT32)pstr_parametric_ffwd_type_drc_params->node_gain[n - 1] -
levelDb +
(FLOAT32)pstr_parametric_ffwd_type_drc_params->node_level[n - 1];
} else {
loc_db_gain =
(FLOAT32)pstr_parametric_ffwd_type_drc_params->node_gain[n] +
(levelDb -
(FLOAT32)pstr_parametric_ffwd_type_drc_params->node_level[n]) /
(FLOAT32)(pstr_parametric_ffwd_type_drc_params->node_level[n - 1] -
pstr_parametric_ffwd_type_drc_params->node_level[n]) *
(FLOAT32)(pstr_parametric_ffwd_type_drc_params->node_gain[n - 1] -
pstr_parametric_ffwd_type_drc_params->node_gain[n]);
}
levelDelta = levelDb - pstr_parametric_ffwd_type_drc_params->db_level_smooth;
if (loc_db_gain < pstr_parametric_ffwd_type_drc_params->db_gain_smooth) {
if (levelDelta >
pstr_parametric_ffwd_type_drc_params->gain_smooth_attack_threshold) {
alpha =
pstr_parametric_ffwd_type_drc_params->gain_smooth_attack_alpha_fast;
} else {
alpha =
pstr_parametric_ffwd_type_drc_params->gain_smooth_attack_alpha_slow;
}
} else {
if (levelDelta <
-pstr_parametric_ffwd_type_drc_params->gain_smooth_rel_threshold) {
alpha = pstr_parametric_ffwd_type_drc_params->gain_smooth_rel_alpha_fast;
} else {
alpha = pstr_parametric_ffwd_type_drc_params->gain_smooth_rel_alpha_slow;
}
}
if (loc_db_gain < pstr_parametric_ffwd_type_drc_params->db_gain_smooth ||
pstr_parametric_ffwd_type_drc_params->hold_counter == 0) {
pstr_parametric_ffwd_type_drc_params->db_level_smooth =
(1 - alpha) * pstr_parametric_ffwd_type_drc_params->db_level_smooth +
alpha * levelDb;
pstr_parametric_ffwd_type_drc_params->db_gain_smooth =
(1 - alpha) * pstr_parametric_ffwd_type_drc_params->db_gain_smooth +
alpha * loc_db_gain;
}
if (pstr_parametric_ffwd_type_drc_params->hold_counter) {
pstr_parametric_ffwd_type_drc_params->hold_counter -= 1;
}
if (loc_db_gain < pstr_parametric_ffwd_type_drc_params->db_gain_smooth) {
pstr_parametric_ffwd_type_drc_params->hold_counter =
pstr_parametric_ffwd_type_drc_params->gain_smooth_hold_off_count;
}
str_spline_nodes->str_node[nodeIdx].loc_db_gain =
pstr_parametric_ffwd_type_drc_params->db_gain_smooth;
str_spline_nodes->str_node[nodeIdx].slope = 0.f;
str_spline_nodes->str_node[nodeIdx].time =
pstr_parametric_ffwd_type_drc_params->frame_size + offset - 1;
return 0;
}
VOID impd_parametric_lim_type_drc_process(
FLOAT32* samples[], FLOAT32 loudness_normalization_gain_db,
ia_parametric_drc_type_lim_params_struct*
pstr_parametric_lim_type_drc_params,
FLOAT32* lpcm_gains) {
WORD32 i, j;
FLOAT32 tmp, gain;
// FLOAT32 min_gain = 1;
FLOAT32 maximum, sectionMaximum;
FLOAT32 loudness_normalization_gain =
(FLOAT32)pow(10.0f, 0.05f * loudness_normalization_gain_db);
FLOAT32* level_estim_ch_weight =
pstr_parametric_lim_type_drc_params->level_estim_ch_weight;
WORD32 num_channels = pstr_parametric_lim_type_drc_params->channels;
WORD32 attack_time_samples = pstr_parametric_lim_type_drc_params->attack;
FLOAT32 attack_constant =
pstr_parametric_lim_type_drc_params->attack_constant;
FLOAT32 release_constant =
pstr_parametric_lim_type_drc_params->release_constant;
FLOAT32 limit_threshold = pstr_parametric_lim_type_drc_params->threshold;
FLOAT32* max_buf = pstr_parametric_lim_type_drc_params->max_buf;
FLOAT32 gain_modified = pstr_parametric_lim_type_drc_params->cor;
FLOAT64 pre_smoothed_gain =
pstr_parametric_lim_type_drc_params->smooth_state_0;
for (i = 0; i < pstr_parametric_lim_type_drc_params->frame_size; i++) {
tmp = 0.0f;
for (j = 0; j < num_channels; j++) {
if (!level_estim_ch_weight[j]) continue;
tmp =
max(tmp, (FLOAT32)fabs(loudness_normalization_gain *
(level_estim_ch_weight[j]) * (samples[j][i])));
}
for (j = attack_time_samples; j > 0; j--) {
max_buf[j] = max_buf[j - 1];
}
max_buf[0] = tmp;
sectionMaximum = tmp;
for (j = 1; j < (attack_time_samples + 1); j++) {
if (max_buf[j] > sectionMaximum) sectionMaximum = max_buf[j];
}
maximum = sectionMaximum;
if (maximum > limit_threshold) {
gain = limit_threshold / maximum;
} else {
gain = 1;
}
if (gain < pre_smoothed_gain) {
gain_modified =
min(gain_modified,
(gain - 0.1f * (FLOAT32)pre_smoothed_gain) * 1.11111111f);
} else {
gain_modified = gain;
}
if (gain_modified < pre_smoothed_gain) {
pre_smoothed_gain =
attack_constant * (pre_smoothed_gain - gain_modified) + gain_modified;
pre_smoothed_gain = max(pre_smoothed_gain, gain);
} else {
pre_smoothed_gain =
release_constant * (pre_smoothed_gain - gain_modified) +
gain_modified;
}
gain = (FLOAT32)pre_smoothed_gain;
lpcm_gains[i] = gain;
}
pstr_parametric_lim_type_drc_params->cor = gain_modified;
pstr_parametric_lim_type_drc_params->smooth_state_0 = pre_smoothed_gain;
return;
}