libxaac/encoder/ixheaace_adjust_threshold.c
akshayragir833 1ce177ea0e
Encoder enhancements and minor bug fixes (#73)
Test done: Smoke test
2023-12-28 17:47:33 +05:30

991 lines
40 KiB
C

/******************************************************************************
* *
* Copyright (C) 2023 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 <float.h>
#include <math.h>
#include <stdlib.h>
#include <string.h>
#include "ixheaac_type_def.h"
#include "ixheaac_constants.h"
#include "impd_drc_common_enc.h"
#include "impd_drc_uni_drc.h"
#include "impd_drc_tables.h"
#include "impd_drc_api.h"
#include "ixheaace_api.h"
#include "ixheaace_aac_constants.h"
#include "ixheaace_psy_const.h"
#include "ixheaace_tns.h"
#include "ixheaace_tns_params.h"
#include "ixheaace_rom.h"
#include "ixheaac_basic_ops32.h"
#include "ixheaac_basic_ops16.h"
#include "ixheaac_basic_ops40.h"
#include "ixheaac_basic_ops.h"
#include "ixheaace_common_rom.h"
#include "ixheaace_bitbuffer.h"
#include "ixheaace_adjust_threshold_data.h"
#include "ixheaace_dynamic_bits.h"
#include "ixheaace_qc_data.h"
#include "ixheaace_block_switch.h"
#include "ixheaace_psy_data.h"
#include "ixheaace_interface.h"
#include "ixheaace_adjust_threshold.h"
#include "ixheaace_common_utils.h"
VOID iaace_adj_thr_init(ia_adj_thr_state_struct *pstr_adj_thr_state, const FLOAT32 mean_pe,
WORD32 ch_bitrate, WORD32 aot) {
ia_adj_thr_elem_struct *pstr_adj_thr_ele = &pstr_adj_thr_state->str_adj_thr_ele;
ia_min_snr_adapt_param_struct *pstr_min_snr_params =
&pstr_adj_thr_ele->str_min_snr_adapt_params;
pstr_adj_thr_state->str_bitres_params_long.clip_save_low = CLIP_SAVE_LO_LONG;
pstr_adj_thr_state->str_bitres_params_long.clip_save_high = CLIP_SAVE_HI_LONG;
pstr_adj_thr_state->str_bitres_params_long.min_bit_save = MIN_BITS_SAVE_LONG;
pstr_adj_thr_state->str_bitres_params_long.max_bit_save = MAX_BITS_SAVE_LONG;
pstr_adj_thr_state->str_bitres_params_long.clip_spend_low = CLIP_SPEND_LO_LONG;
pstr_adj_thr_state->str_bitres_params_long.clip_spend_high = CLIP_SPEND_HI_LONG;
pstr_adj_thr_state->str_bitres_params_long.min_bits_spend = MIN_BITS_SPEND_LONG;
pstr_adj_thr_state->str_bitres_params_long.max_bits_spend = MAX_BITS_SPEND_LONG;
pstr_adj_thr_state->str_bitres_params_short.clip_save_low = CLIP_SAVE_LO_SHORT;
pstr_adj_thr_state->str_bitres_params_short.clip_save_high = CLIP_SAVE_HI_SHORT;
pstr_adj_thr_state->str_bitres_params_short.min_bit_save = MIN_BITS_SAVE_SHORT;
pstr_adj_thr_state->str_bitres_params_short.max_bit_save = MAX_BITS_SAVE_SHORT;
pstr_adj_thr_state->str_bitres_params_short.clip_spend_low = CLIP_SPEND_LO_SHORT;
pstr_adj_thr_state->str_bitres_params_short.clip_spend_high = CLIP_SPEND_HI_SHORT;
pstr_adj_thr_state->str_bitres_params_short.min_bits_spend = MIN_BITS_SPEND_SHORT;
pstr_adj_thr_state->str_bitres_params_short.max_bits_spend = MAX_BITS_SPEND_SHORT;
pstr_adj_thr_ele->pe_min = (FLOAT32)0.8f * mean_pe;
pstr_adj_thr_ele->pe_max = (FLOAT32)1.2f * mean_pe;
pstr_adj_thr_ele->pe_offset = 0.0f;
switch (aot) {
case AOT_AAC_LC:
case AOT_SBR:
case AOT_PS:
if (ch_bitrate < 32000) {
pstr_adj_thr_ele->pe_offset = MAX(
(FLOAT32)50.0f, (FLOAT32)(100.0f) - (FLOAT32)(100.0f / 32000) * (FLOAT32)ch_bitrate);
}
break;
case AOT_AAC_LD:
case AOT_AAC_ELD:
if (ch_bitrate <= 32000) {
pstr_adj_thr_ele->pe_offset = MAX(
(FLOAT32)50.0f, (FLOAT32)(100.0f) - (FLOAT32)(100.0f / 32000) * (FLOAT32)ch_bitrate);
}
break;
}
if (ch_bitrate > 20000) {
pstr_adj_thr_ele->str_ah_param.modify_min_snr = TRUE;
pstr_adj_thr_ele->str_ah_param.start_sfb_long = 15;
pstr_adj_thr_ele->str_ah_param.start_sfb_short = 3;
} else {
pstr_adj_thr_ele->str_ah_param.modify_min_snr = FALSE;
pstr_adj_thr_ele->str_ah_param.start_sfb_long = 0;
pstr_adj_thr_ele->str_ah_param.start_sfb_short = 0;
}
pstr_min_snr_params->max_red = (FLOAT32)0.25f;
pstr_min_snr_params->start_ratio = (FLOAT32)1.e1f;
pstr_min_snr_params->max_ratio = (FLOAT32)1.e3f;
pstr_min_snr_params->red_ratio_fac =
(1.0f - pstr_min_snr_params->max_red) /
(10.0f * (FLOAT32)log10(pstr_min_snr_params->start_ratio / pstr_min_snr_params->max_ratio));
pstr_min_snr_params->red_offs = 1.0f - pstr_min_snr_params->red_ratio_fac * 10.0f *
(FLOAT32)log10(pstr_min_snr_params->start_ratio);
pstr_adj_thr_ele->pe_last = (FLOAT32)0.0f;
pstr_adj_thr_ele->dyn_bits_last = 0;
pstr_adj_thr_ele->pe_correction_fac = (FLOAT32)1.0f;
}
FLOAT32 iaace_bits_to_pe(const FLOAT32 bits) { return (bits * 1.18f); }
static VOID iaace_calc_sfb_pe_data(ia_qc_pe_data_struct *pstr_qc_pe_data,
ixheaace_psy_out_channel **pstr_psy_out, WORD32 num_channels,
WORD32 chn) {
WORD32 ch;
WORD32 scf_band_grp;
FLOAT32 num_lines;
FLOAT32 ld_thr, ld_ratio;
WORD32 i = 0, scf;
WORD32 sfb_count;
WORD32 scf_band_per_grp;
WORD32 max_sfb_per_grp;
FLOAT32 *ptr_sfb_energy;
FLOAT32 *ptr_sfb_thr;
ia_qc_pe_chan_data_struct *pstr_qc_pe_chan_data;
pstr_qc_pe_data->pe = pstr_qc_pe_data->offset;
pstr_qc_pe_data->const_part = 0.0f;
pstr_qc_pe_data->num_active_lines = 0.0f;
for (ch = chn; ch < chn + num_channels; ch++) {
sfb_count = pstr_psy_out[ch]->sfb_count;
scf_band_per_grp = pstr_psy_out[ch]->sfb_per_group;
max_sfb_per_grp = pstr_psy_out[ch]->max_sfb_per_grp;
ptr_sfb_energy = pstr_psy_out[ch]->ptr_sfb_energy;
ptr_sfb_thr = pstr_psy_out[ch]->ptr_sfb_thr;
pstr_qc_pe_chan_data = &pstr_qc_pe_data->pe_ch_data[ch];
pstr_qc_pe_chan_data->pe = 0;
pstr_qc_pe_chan_data->num_active_lines = 0;
pstr_qc_pe_chan_data->const_part = 0;
for (scf_band_grp = 0; scf_band_grp < sfb_count; scf_band_grp += scf_band_per_grp) {
i = scf_band_grp;
for (scf = max_sfb_per_grp - 1; scf >= 0; scf--, i++) {
if (ptr_sfb_energy[i] > ptr_sfb_thr[i]) {
ld_thr = (FLOAT32)log(ptr_sfb_thr[i]) * LOG2_1;
ld_ratio = pstr_qc_pe_chan_data->sfb_ld_energy[i] - ld_thr;
num_lines = pstr_qc_pe_chan_data->sfb_lines[i];
if (ld_ratio >= PE_C1) {
pstr_qc_pe_chan_data->sfb_pe[i] = num_lines * ld_ratio;
pstr_qc_pe_chan_data->sfb_const_part[i] =
num_lines * pstr_qc_pe_chan_data->sfb_ld_energy[i];
} else {
pstr_qc_pe_chan_data->sfb_pe[i] = num_lines * (PE_C2 + PE_C3 * ld_ratio);
pstr_qc_pe_chan_data->sfb_const_part[i] =
num_lines * (PE_C2 + PE_C3 * pstr_qc_pe_chan_data->sfb_ld_energy[i]);
num_lines = num_lines * PE_C3;
}
pstr_qc_pe_chan_data->num_sfb_active_lines[i] = num_lines;
} else {
pstr_qc_pe_chan_data->sfb_pe[i] = 0.0f;
pstr_qc_pe_chan_data->sfb_const_part[i] = 0.0f;
pstr_qc_pe_chan_data->num_sfb_active_lines[i] = 0.0;
}
pstr_qc_pe_chan_data->pe += pstr_qc_pe_chan_data->sfb_pe[i];
pstr_qc_pe_chan_data->const_part += pstr_qc_pe_chan_data->sfb_const_part[i];
pstr_qc_pe_chan_data->num_active_lines += pstr_qc_pe_chan_data->num_sfb_active_lines[i];
}
}
pstr_qc_pe_data->pe += pstr_qc_pe_chan_data->pe;
pstr_qc_pe_data->const_part += pstr_qc_pe_chan_data->const_part;
pstr_qc_pe_data->num_active_lines += pstr_qc_pe_chan_data->num_active_lines;
pstr_psy_out[ch]->pe = pstr_qc_pe_data->pe;
}
}
static FLOAT32 iaace_calc_bit_save(ia_bitres_param_struct *pstr_bitres_params, FLOAT32 fill_lvl) {
FLOAT32 bit_save;
const FLOAT32 clip_low = pstr_bitres_params->clip_save_low;
const FLOAT32 clip_high = pstr_bitres_params->clip_save_high;
const FLOAT32 min_bit_save = pstr_bitres_params->min_bit_save;
const FLOAT32 max_bit_save = pstr_bitres_params->max_bit_save;
fill_lvl = MAX(fill_lvl, clip_low);
fill_lvl = MIN(fill_lvl, clip_high);
bit_save = max_bit_save -
((max_bit_save - min_bit_save) / (clip_high - clip_low)) * (fill_lvl - clip_low);
return bit_save;
}
static FLOAT32 iaace_calc_bit_spend(ia_bitres_param_struct *pstr_bitres_params,
FLOAT32 fill_lvl) {
FLOAT32 bit_spend;
const FLOAT32 clip_low = pstr_bitres_params->clip_spend_low;
const FLOAT32 clip_high = pstr_bitres_params->clip_spend_high;
const FLOAT32 min_bits_spend = pstr_bitres_params->min_bits_spend;
const FLOAT32 max_bits_spend = pstr_bitres_params->max_bits_spend;
fill_lvl = MAX(fill_lvl, clip_low);
fill_lvl = MIN(fill_lvl, clip_high);
bit_spend = min_bits_spend + ((max_bits_spend - min_bits_spend) / (clip_high - clip_low)) *
(fill_lvl - clip_low);
return bit_spend;
}
static VOID iaace_adjust_pe_minmax(const FLOAT32 curr_pe, FLOAT32 *ptr_pe_min,
FLOAT32 *ptr_pe_max) {
FLOAT32 min_hi_fac = 0.3f, max_hi_fac = 1.0f, min_low_fac = 0.14f, max_low_fac = 0.07f;
FLOAT32 diff;
FLOAT32 min_diff = curr_pe * (FLOAT32)0.1666666667f;
if (curr_pe > *ptr_pe_max) {
diff = (curr_pe - *ptr_pe_max);
*ptr_pe_min += diff * min_hi_fac;
*ptr_pe_max += diff * max_hi_fac;
} else {
if (curr_pe < *ptr_pe_min) {
diff = (*ptr_pe_min - curr_pe);
*ptr_pe_min -= diff * min_low_fac;
*ptr_pe_max -= diff * max_low_fac;
} else {
*ptr_pe_min += (curr_pe - *ptr_pe_min) * min_hi_fac;
*ptr_pe_max -= (*ptr_pe_max - curr_pe) * max_low_fac;
}
}
if ((*ptr_pe_max - *ptr_pe_min) < min_diff) {
FLOAT32 low_part, high_part;
low_part = MAX((FLOAT32)0.0f, curr_pe - *ptr_pe_min);
high_part = MAX((FLOAT32)0.0f, *ptr_pe_max - curr_pe);
*ptr_pe_max = curr_pe + high_part / (low_part + high_part) * min_diff;
*ptr_pe_min = curr_pe - low_part / (low_part + high_part) * min_diff;
*ptr_pe_min = MAX((FLOAT32)0.0f, *ptr_pe_min);
}
}
static FLOAT32 iaace_bitres_calc_bitfac(const WORD32 bitres_bits, const WORD32 max_bitres_bits,
const FLOAT32 pe, const WORD32 win_seq,
const WORD32 avg_bits, const FLOAT32 max_bit_fac,
ia_adj_thr_state_struct *pstr_adj_thr_state,
ia_adj_thr_elem_struct *pstr_adj_the_elem) {
ia_bitres_param_struct *pstr_bitres_params;
FLOAT32 pex;
FLOAT32 fill_lvl = 0.0f;
FLOAT32 bit_save, bit_spend, bitres_factor;
if (max_bitres_bits) {
fill_lvl = (FLOAT32)bitres_bits / max_bitres_bits;
}
if (win_seq != SHORT_WINDOW) {
pstr_bitres_params = &(pstr_adj_thr_state->str_bitres_params_long);
} else {
pstr_bitres_params = &(pstr_adj_thr_state->str_bitres_params_short);
}
pex = MAX(pe, pstr_adj_the_elem->pe_min);
pex = MIN(pex, pstr_adj_the_elem->pe_max);
bit_save = iaace_calc_bit_save(pstr_bitres_params, fill_lvl);
bit_spend = iaace_calc_bit_spend(pstr_bitres_params, fill_lvl);
bitres_factor =
(FLOAT32)1.0f - bit_save +
((bit_spend + bit_save) / (pstr_adj_the_elem->pe_max - pstr_adj_the_elem->pe_min)) *
(pex - pstr_adj_the_elem->pe_min);
if (avg_bits)
{
bitres_factor = MIN(bitres_factor,
(FLOAT32)1.0f - (FLOAT32)0.3f + (FLOAT32)bitres_bits / (FLOAT32)avg_bits);
}
bitres_factor = MIN(bitres_factor, max_bit_fac);
iaace_adjust_pe_minmax(pe, &pstr_adj_the_elem->pe_min, &pstr_adj_the_elem->pe_max);
return bitres_factor;
}
static VOID iaace_calc_pe_correction(FLOAT32 *ptr_correction_fac, const FLOAT32 pe_act,
const FLOAT32 pe_last, const WORD32 bits_prev) {
if ((bits_prev > 0) && (pe_act < (FLOAT32)1.5f * pe_last) &&
(pe_act > (FLOAT32)0.7f * pe_last) &&
((FLOAT32)1.2f * iaace_bits_to_pe((FLOAT32)bits_prev) > pe_last) &&
((FLOAT32)0.65f * iaace_bits_to_pe((FLOAT32)bits_prev) < pe_last)) {
FLOAT32 new_fac = pe_last / iaace_bits_to_pe((FLOAT32)bits_prev);
if (new_fac < (FLOAT32)1.0f) {
new_fac = MIN((FLOAT32)1.1f * new_fac, (FLOAT32)1.0f);
new_fac = MAX(new_fac, (FLOAT32)0.85f);
} else {
new_fac = MAX((FLOAT32)0.9f * new_fac, (FLOAT32)1.0f);
new_fac = MIN(new_fac, (FLOAT32)1.15f);
}
if (((new_fac > (FLOAT32)1.0f) && (*ptr_correction_fac < (FLOAT32)1.0f)) ||
((new_fac < (FLOAT32)1.0f) && (*ptr_correction_fac > (FLOAT32)1.0f))) {
*ptr_correction_fac = (FLOAT32)1.0f;
}
if ((*ptr_correction_fac < (FLOAT32)1.0f && new_fac < *ptr_correction_fac) ||
(*ptr_correction_fac > (FLOAT32)1.0f && new_fac > *ptr_correction_fac)) {
*ptr_correction_fac = (FLOAT32)0.85f * (*ptr_correction_fac) + (FLOAT32)0.15f * new_fac;
} else {
*ptr_correction_fac = (FLOAT32)0.7f * (*ptr_correction_fac) + (FLOAT32)0.3f * new_fac;
}
*ptr_correction_fac = MIN(*ptr_correction_fac, (FLOAT32)1.15f);
*ptr_correction_fac = MAX(*ptr_correction_fac, (FLOAT32)0.85f);
} else {
*ptr_correction_fac = (FLOAT32)1.0f;
}
}
static VOID iaace_calc_thr_exp(
FLOAT32 thr_exp[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
ixheaace_psy_out_channel **pstr_psy_out, WORD32 num_chans, WORD32 chn) {
WORD32 sfb, ch, scf_band_grp;
ixheaace_psy_out_channel *pstr_psy_chan_out;
FLOAT32 *ptr_scf_band_thr;
FLOAT32 *ptr_thr_exp;
for (ch = chn; ch < chn + num_chans; ch++) {
pstr_psy_chan_out = pstr_psy_out[ch];
ptr_thr_exp = thr_exp[ch];
for (scf_band_grp = 0; scf_band_grp < pstr_psy_chan_out->sfb_count;
scf_band_grp += pstr_psy_chan_out->sfb_per_group) {
FLOAT32 *ptr_thr_exp1 = &ptr_thr_exp[scf_band_grp];
ptr_scf_band_thr = &pstr_psy_chan_out->ptr_sfb_thr[scf_band_grp];
for (sfb = 0; sfb < pstr_psy_chan_out->max_sfb_per_grp; sfb++) {
ptr_thr_exp1[sfb] = (FLOAT32)pow(*ptr_scf_band_thr++, RED_EXP_VAL);
}
}
}
}
static VOID iaace_adapt_min_snr(ixheaace_psy_out_channel **pstr_psy_out,
ia_min_snr_adapt_param_struct *pstr_min_snr_params,
WORD32 num_chans, WORD32 chn) {
WORD32 num_scf_band = 0, ch, scf_band_cnt, scf_band_offs, sfb;
FLOAT32 avg_energy = 0.0f, db_ratio, min_snr_red;
WORD32 i;
for (ch = chn; ch < chn + num_chans; ch++) {
ixheaace_psy_out_channel *pstr_psy_chan_out = pstr_psy_out[ch];
num_scf_band = 0;
avg_energy = 0;
scf_band_cnt = pstr_psy_chan_out->max_sfb_per_grp;
for (scf_band_offs = 0; scf_band_offs < pstr_psy_chan_out->sfb_count;
scf_band_offs += pstr_psy_chan_out->sfb_per_group) {
FLOAT32 *ptr_sfb_energy = &pstr_psy_chan_out->ptr_sfb_energy[scf_band_offs];
for (sfb = scf_band_cnt - 1; sfb >= 0; sfb--) {
avg_energy += ptr_sfb_energy[sfb];
}
num_scf_band += scf_band_cnt;
}
if (num_scf_band > 0) {
avg_energy /= num_scf_band;
}
for (scf_band_offs = 0; scf_band_offs < pstr_psy_chan_out->sfb_count;
scf_band_offs += pstr_psy_chan_out->sfb_per_group) {
i = scf_band_offs;
for (sfb = scf_band_cnt - 1; sfb >= 0; sfb--, i++) {
if (pstr_min_snr_params->start_ratio * pstr_psy_chan_out->ptr_sfb_energy[i] <
avg_energy) {
db_ratio =
(FLOAT32)(10.0f * log10((MIN_FLT_VAL + avg_energy) /
(MIN_FLT_VAL + pstr_psy_chan_out->ptr_sfb_energy[i])));
min_snr_red =
pstr_min_snr_params->red_offs + pstr_min_snr_params->red_ratio_fac * db_ratio;
min_snr_red = MAX(min_snr_red, pstr_min_snr_params->max_red);
pstr_psy_chan_out->sfb_min_snr[i] =
(FLOAT32)pow(pstr_psy_out[ch]->sfb_min_snr[i], min_snr_red);
pstr_psy_chan_out->sfb_min_snr[i] =
MIN(MIN_SNR_LIMIT, pstr_psy_out[ch]->sfb_min_snr[i]);
}
}
}
}
}
static VOID iaace_init_avoid_hole_flag(
WORD32 ah_flag[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
ixheaace_psy_out_channel **pstr_psy_out, ia_ah_param_struct *pstr_ah_param, WORD32 num_chans,
WORD32 chn, WORD32 aot) {
WORD32 ch;
FLOAT32 sfb_energy;
FLOAT32 scale_spread_energy = 0.0f;
WORD32 scf_band_grp, scf_band_cnt, scf_band;
FLOAT32 *ptr_scf_band_spread_energy, *ptr_scf_band_energy, *ptr_scf_band_min_snr;
for (ch = chn; ch < chn + num_chans; ch++) {
ixheaace_psy_out_channel *pstr_psy_chan_out = pstr_psy_out[ch];
if (pstr_psy_chan_out->window_sequence != SHORT_WINDOW) {
switch (aot) {
case AOT_AAC_LC:
case AOT_SBR:
case AOT_PS:
scale_spread_energy = 0.5f;
break;
case AOT_AAC_LD:
case AOT_AAC_ELD:
scale_spread_energy = 0.56f;
break;
}
} else {
scale_spread_energy = 0.63f;
}
for (scf_band_grp = 0; scf_band_grp < pstr_psy_chan_out->sfb_count;
scf_band_grp += pstr_psy_chan_out->sfb_per_group) {
ptr_scf_band_spread_energy = &pstr_psy_chan_out->ptr_sfb_spread_energy[scf_band_grp];
scf_band_cnt = pstr_psy_chan_out->max_sfb_per_grp;
for (scf_band = scf_band_cnt - 1; scf_band >= 0; scf_band--) {
*ptr_scf_band_spread_energy = *ptr_scf_band_spread_energy * scale_spread_energy;
ptr_scf_band_spread_energy++;
}
}
}
if (pstr_ah_param->modify_min_snr) {
for (ch = chn; ch < chn + num_chans; ch++) {
ixheaace_psy_out_channel *pstr_psy_chan_out = pstr_psy_out[ch];
ptr_scf_band_energy = pstr_psy_chan_out->ptr_sfb_energy;
ptr_scf_band_min_snr = pstr_psy_chan_out->sfb_min_snr;
for (scf_band_grp = 0; scf_band_grp < pstr_psy_chan_out->sfb_count;
scf_band_grp += pstr_psy_chan_out->sfb_per_group) {
for (scf_band = 0; scf_band < pstr_psy_chan_out->max_sfb_per_grp; scf_band++) {
FLOAT32 sfb_en_m1, sfb_en_p1, avg_energy;
if (scf_band > 0) {
sfb_en_m1 = ptr_scf_band_energy[scf_band_grp + scf_band - 1];
} else {
sfb_en_m1 = ptr_scf_band_energy[scf_band_grp];
}
if (scf_band < pstr_psy_chan_out->max_sfb_per_grp - 1) {
sfb_en_p1 = ptr_scf_band_energy[scf_band_grp + scf_band + 1];
} else {
sfb_en_p1 = ptr_scf_band_energy[scf_band_grp + scf_band];
}
avg_energy = (sfb_en_m1 + sfb_en_p1) / (FLOAT32)2.0f;
sfb_energy = ptr_scf_band_energy[scf_band_grp + scf_band];
FLOAT32 temp_min_snr = 0.0f;
if (sfb_energy > avg_energy) {
switch (aot) {
case AOT_AAC_LC:
case AOT_SBR:
case AOT_PS:
temp_min_snr = MAX((FLOAT32)0.8f * avg_energy / sfb_energy, (FLOAT32)0.316f);
break;
case AOT_AAC_LD:
case AOT_AAC_ELD:
temp_min_snr = MAX((FLOAT32)0.8f * avg_energy / sfb_energy, (FLOAT32)0.408f);
break;
}
if (pstr_psy_chan_out->window_sequence != SHORT_WINDOW) {
switch (aot) {
case AOT_AAC_LC:
case AOT_PS:
temp_min_snr = MAX(temp_min_snr, (FLOAT32)0.316f);
break;
case AOT_AAC_LD:
temp_min_snr = MAX(temp_min_snr, (FLOAT32)0.408f);
break;
}
} else {
temp_min_snr = MAX(temp_min_snr, (FLOAT32)0.5f);
}
ptr_scf_band_min_snr[scf_band_grp + scf_band] =
MIN(ptr_scf_band_min_snr[scf_band_grp + scf_band], temp_min_snr);
}
if (((FLOAT32)2.0f * sfb_energy < avg_energy) && (sfb_energy > (FLOAT32)0.0f)) {
temp_min_snr = avg_energy / ((FLOAT32)2.0f * sfb_energy) *
ptr_scf_band_min_snr[scf_band_grp + scf_band];
temp_min_snr = MIN((FLOAT32)0.8f, temp_min_snr);
ptr_scf_band_min_snr[scf_band_grp + scf_band] =
MIN(temp_min_snr, ptr_scf_band_min_snr[scf_band_grp + scf_band] * (FLOAT32)3.16f);
}
}
}
}
}
if (num_chans == 2) {
ixheaace_psy_out_channel *pstr_psy_out_mid = pstr_psy_out[chn];
ixheaace_psy_out_channel *pstr_psy_out_side = pstr_psy_out[chn + 1];
WORD32 sfb;
for (sfb = 0; sfb < pstr_psy_out_mid->sfb_count; sfb++) {
if (pstr_psy_out[chn]->ms_used[sfb]) {
FLOAT32 sfb_en_mid = pstr_psy_out_mid->ptr_sfb_energy[sfb];
FLOAT32 sfb_en_side = pstr_psy_out_side->ptr_sfb_energy[sfb];
FLOAT32 max_sfb_en = MAX(sfb_en_mid, sfb_en_side);
FLOAT32 max_thr = 0.25f * pstr_psy_out_mid->sfb_min_snr[sfb] * max_sfb_en;
pstr_psy_out_mid->sfb_min_snr[sfb] = (FLOAT32)MAX(
pstr_psy_out_mid->sfb_min_snr[sfb],
MIN(MAX_FLT_VAL, ((FLOAT32)max_thr / (MIN_FLT_VAL + (FLOAT32)sfb_en_mid))));
if (pstr_psy_out_mid->ptr_sfb_energy[sfb] <= 1.0f) {
pstr_psy_out_mid->ptr_sfb_energy[sfb] =
MIN(pstr_psy_out_mid->ptr_sfb_energy[sfb], 0.8f);
}
pstr_psy_out_side->sfb_min_snr[sfb] = (FLOAT32)MAX(
pstr_psy_out_side->sfb_min_snr[sfb],
MIN(MAX_FLT_VAL, ((FLOAT32)max_thr / (MIN_FLT_VAL + (FLOAT32)sfb_en_side))));
if (pstr_psy_out_side->sfb_min_snr[sfb] <= 1.0f) {
pstr_psy_out_side->sfb_min_snr[sfb] = MIN(pstr_psy_out_side->sfb_min_snr[sfb], 0.8f);
}
}
}
}
for (ch = chn; ch < chn + num_chans; ch++) {
ixheaace_psy_out_channel *pstr_psy_chan_out = pstr_psy_out[ch];
for (scf_band_grp = 0; scf_band_grp < pstr_psy_chan_out->sfb_count;
scf_band_grp += pstr_psy_chan_out->sfb_per_group) {
for (scf_band = 0; scf_band < pstr_psy_chan_out->max_sfb_per_grp; scf_band++) {
if (pstr_psy_chan_out->ptr_sfb_spread_energy[scf_band_grp + scf_band] >
pstr_psy_chan_out->ptr_sfb_energy[scf_band_grp + scf_band] ||
pstr_psy_chan_out->sfb_min_snr[scf_band_grp + scf_band] > 1.0f) {
ah_flag[ch][scf_band_grp + scf_band] = NO_AH;
} else {
ah_flag[ch][scf_band_grp + scf_band] = AH_INACTIVE;
}
}
for (scf_band = pstr_psy_chan_out->max_sfb_per_grp;
scf_band < pstr_psy_chan_out->sfb_per_group; scf_band++) {
ah_flag[ch][scf_band_grp + scf_band] = NO_AH;
}
}
}
}
static VOID iaace_reduce_thr(
ixheaace_psy_out_channel **pstr_psy_out,
WORD32 ah_flag[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
FLOAT32 thr_exp[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
const FLOAT32 red_value, WORD32 num_channels, WORD32 chn) {
WORD32 ch, sfb_group, sfb;
FLOAT32 sfb_energy, sfb_threshold, sfb_thr_reduced;
FLOAT32 *ptr_sfb_energy_fix, *ptr_sfb_threshold_fix, *ptr_sfb_min_snr_fix, *ptr_thr_exp_fix;
for (ch = chn; ch < chn + num_channels; ch++) {
ixheaace_psy_out_channel *pstr_psy_chan_out = pstr_psy_out[ch];
ptr_sfb_energy_fix = pstr_psy_chan_out->ptr_sfb_energy;
ptr_sfb_threshold_fix = pstr_psy_chan_out->ptr_sfb_thr;
ptr_sfb_min_snr_fix = pstr_psy_chan_out->sfb_min_snr;
ptr_thr_exp_fix = &thr_exp[ch][0];
for (sfb_group = 0; sfb_group < pstr_psy_chan_out->sfb_count;
sfb_group += pstr_psy_chan_out->sfb_per_group) {
for (sfb = 0; sfb < pstr_psy_chan_out->max_sfb_per_grp; sfb++) {
sfb_energy = ptr_sfb_energy_fix[sfb_group + sfb];
sfb_threshold = ptr_sfb_threshold_fix[sfb_group + sfb];
if (sfb_energy > sfb_threshold) {
sfb_thr_reduced =
(FLOAT32)pow((ptr_thr_exp_fix[sfb_group + sfb] + red_value), INV_RED_EXP_VAL);
if ((sfb_thr_reduced > ptr_sfb_min_snr_fix[sfb_group + sfb] * sfb_energy) &&
(ah_flag[ch][sfb_group + sfb] != NO_AH)) {
sfb_thr_reduced =
MAX(ptr_sfb_min_snr_fix[sfb_group + sfb] * sfb_energy, sfb_threshold);
ah_flag[ch][sfb_group + sfb] = AH_ACTIVE;
}
ptr_sfb_threshold_fix[sfb_group + sfb] = sfb_thr_reduced;
}
}
}
}
}
static VOID iaace_calc_pe_no_active_holes(
FLOAT32 *ptr_pe, FLOAT32 *ptr_const_part, FLOAT32 *ptr_num_active_lines,
ia_qc_pe_data_struct *pstr_qs_pe_data,
WORD32 ah_flag[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
ixheaace_psy_out_channel **pstr_psy_out, WORD32 num_channels, WORD32 chn) {
WORD32 ch, sfb_group, sfb;
*ptr_pe = 0.0f;
*ptr_const_part = 0.0f;
*ptr_num_active_lines = 0;
for (ch = chn; ch < chn + num_channels; ch++) {
ixheaace_psy_out_channel *pstr_psy_chan_out = pstr_psy_out[ch];
ia_qc_pe_chan_data_struct *ptr_pe_chan_data = &pstr_qs_pe_data->pe_ch_data[ch];
for (sfb_group = 0; sfb_group < pstr_psy_chan_out->sfb_count;
sfb_group += pstr_psy_chan_out->sfb_per_group) {
for (sfb = 0; sfb < pstr_psy_chan_out->max_sfb_per_grp; sfb++) {
if (ah_flag[ch][sfb_group + sfb] < AH_ACTIVE) {
*ptr_pe += ptr_pe_chan_data->sfb_pe[sfb_group + sfb];
*ptr_const_part += ptr_pe_chan_data->sfb_const_part[sfb_group + sfb];
*ptr_num_active_lines += ptr_pe_chan_data->num_sfb_active_lines[sfb_group + sfb];
}
}
}
}
}
static VOID iaace_correct_thr(
ixheaace_psy_out_channel **pstr_psy_out,
WORD32 ah_flag[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
ia_qc_pe_data_struct *pstr_qs_pe_data,
FLOAT32 thr_exp[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
const FLOAT32 red_value, const FLOAT32 delta_pe, WORD32 num_channels, WORD32 chn) {
WORD32 i, ch, sfb_group, sfb;
FLOAT32 delta_sfb_pe;
FLOAT32 thr_factor;
FLOAT32 sfb_pe_factors[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
norm_factor[IXHEAACE_MAX_CH_IN_BS_ELE] = {MIN_FLT_VAL};
FLOAT32 sfb_en, sfb_thr, sfb_thr_reduced;
FLOAT32 *ptr_thr_exp;
FLOAT32 *ptr_sfb_energy, *ptr_sfb_thr, *ptr_sfb_min_snr;
ixheaace_psy_out_channel *pstr_psy_chan_out = NULL;
ia_qc_pe_chan_data_struct *pstr_pe_chan_data = NULL;
for (ch = chn; ch < chn + num_channels; ch++) {
pstr_psy_chan_out = pstr_psy_out[ch];
pstr_pe_chan_data = &pstr_qs_pe_data->pe_ch_data[ch];
norm_factor[ch] = MIN_FLT_VAL;
ptr_thr_exp = thr_exp[ch];
for (sfb_group = 0; sfb_group < pstr_psy_chan_out->sfb_count;
sfb_group += pstr_psy_chan_out->sfb_per_group) {
for (sfb = 0; sfb < pstr_psy_chan_out->max_sfb_per_grp; sfb++) {
if ((ah_flag[ch][sfb_group + sfb] < AH_ACTIVE) || (delta_pe > 0)) {
sfb_pe_factors[ch][sfb_group + sfb] =
pstr_pe_chan_data->num_sfb_active_lines[sfb_group + sfb] /
(ptr_thr_exp[sfb_group + sfb] + red_value);
norm_factor[ch] += sfb_pe_factors[ch][sfb_group + sfb];
} else {
sfb_pe_factors[ch][sfb_group + sfb] = 0.0f;
}
}
}
}
if (num_channels > 1) {
norm_factor[chn] = norm_factor[chn] + norm_factor[chn + 1];
}
norm_factor[chn] = 1.0f / norm_factor[chn];
for (ch = chn; ch < chn + num_channels; ch++) {
pstr_psy_chan_out = pstr_psy_out[ch];
pstr_pe_chan_data = &pstr_qs_pe_data->pe_ch_data[ch];
ptr_sfb_energy = pstr_psy_chan_out->ptr_sfb_energy;
ptr_sfb_thr = pstr_psy_chan_out->ptr_sfb_thr;
ptr_sfb_min_snr = pstr_psy_chan_out->sfb_min_snr;
for (sfb_group = 0; sfb_group < pstr_psy_chan_out->sfb_count;
sfb_group += pstr_psy_chan_out->sfb_per_group) {
i = sfb_group;
for (sfb = pstr_psy_chan_out->max_sfb_per_grp - 1; sfb >= 0; sfb--, i++) {
delta_sfb_pe = sfb_pe_factors[ch][i] * norm_factor[chn] * delta_pe;
if (pstr_pe_chan_data->num_sfb_active_lines[i] > (FLOAT32)0.5f) {
sfb_en = ptr_sfb_energy[i];
sfb_thr = ptr_sfb_thr[i];
thr_factor = MIN(-delta_sfb_pe / pstr_pe_chan_data->num_sfb_active_lines[i], 20.f);
thr_factor = (FLOAT32)pow(2.0f, thr_factor);
sfb_thr_reduced = sfb_thr * thr_factor;
if ((sfb_thr_reduced > ptr_sfb_min_snr[i] * sfb_en) &&
(ah_flag[ch][i] == AH_INACTIVE)) {
sfb_thr_reduced = MAX(ptr_sfb_min_snr[i] * sfb_en, sfb_thr);
ah_flag[ch][i] = AH_ACTIVE;
}
ptr_sfb_thr[i] = sfb_thr_reduced;
}
}
}
}
}
static VOID iaace_reduce_min_snr(
ixheaace_psy_out_channel **pstr_psy_out, ia_qc_pe_data_struct *pstr_qs_pe_data,
WORD32 ah_flag[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
const FLOAT32 desired_pe, WORD32 num_channels, WORD32 chn) {
WORD32 sfb, sfb_sub_win, ch;
FLOAT32 delta_pe;
sfb_sub_win = pstr_psy_out[chn]->max_sfb_per_grp;
while (pstr_qs_pe_data->pe > desired_pe && sfb_sub_win > 0) {
sfb_sub_win--;
for (sfb = sfb_sub_win; sfb < pstr_psy_out[chn]->sfb_count;
sfb += pstr_psy_out[chn]->sfb_per_group) {
for (ch = chn; ch < chn + num_channels; ch++) {
if (ah_flag[ch][sfb] != NO_AH && pstr_psy_out[ch]->sfb_min_snr[sfb] < MIN_SNR_LIMIT) {
pstr_psy_out[ch]->sfb_min_snr[sfb] = MIN_SNR_LIMIT;
pstr_psy_out[ch]->ptr_sfb_thr[sfb] =
pstr_psy_out[ch]->ptr_sfb_energy[sfb] * pstr_psy_out[ch]->sfb_min_snr[sfb];
delta_pe = pstr_qs_pe_data->pe_ch_data[ch].sfb_lines[sfb] * 1.5f -
pstr_qs_pe_data->pe_ch_data[ch].sfb_pe[sfb];
pstr_qs_pe_data->pe += delta_pe;
pstr_qs_pe_data->pe_ch_data[ch].pe += delta_pe;
}
}
if (pstr_qs_pe_data->pe <= desired_pe) {
break;
}
}
}
}
static VOID iaace_allow_more_holes(
ixheaace_psy_out_channel **pstr_psy_out, ia_qc_pe_data_struct *pstr_qs_pe_data,
WORD32 ah_flag[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND],
const ia_ah_param_struct *pstr_str_ah_param, const FLOAT32 desired_pe, WORD32 num_channels,
WORD32 chn) {
WORD32 sfb, ch;
FLOAT32 act_pe = pstr_qs_pe_data->pe;
if (num_channels == 2 &&
pstr_psy_out[chn]->window_sequence == pstr_psy_out[chn + 1]->window_sequence) {
ixheaace_psy_out_channel *pstr_psy_out_left = pstr_psy_out[chn];
ixheaace_psy_out_channel *pstr_psy_out_right = pstr_psy_out[chn + 1];
for (sfb = 0; sfb < pstr_psy_out_left->sfb_count; sfb++) {
if (pstr_psy_out[chn]->ms_used[sfb]) {
if (ah_flag[chn + 1][sfb] != NO_AH &&
0.4f * pstr_psy_out_left->sfb_min_snr[sfb] * pstr_psy_out_left->ptr_sfb_energy[sfb] >
pstr_psy_out_right->ptr_sfb_energy[sfb]) {
ah_flag[chn + 1][sfb] = NO_AH;
pstr_psy_out_right->ptr_sfb_thr[sfb] = 2.0f * pstr_psy_out_right->ptr_sfb_energy[sfb];
act_pe -= pstr_qs_pe_data->pe_ch_data[chn + 1].sfb_pe[sfb];
} else {
if (ah_flag[chn][sfb] != NO_AH && 0.4f * pstr_psy_out_right->sfb_min_snr[sfb] *
pstr_psy_out_right->ptr_sfb_energy[sfb] >
pstr_psy_out_left->ptr_sfb_energy[sfb]) {
ah_flag[chn][sfb] = NO_AH;
pstr_psy_out_left->ptr_sfb_thr[sfb] = 2.0f * pstr_psy_out_left->ptr_sfb_energy[sfb];
act_pe -= pstr_qs_pe_data->pe_ch_data[chn].sfb_pe[sfb];
}
}
if (act_pe < desired_pe) {
break;
}
}
}
}
if (act_pe > desired_pe) {
WORD32 start_sfb[IXHEAACE_MAX_CH_IN_BS_ELE] = {0};
FLOAT32 average_energy, min_energy;
WORD32 ah_cnt;
WORD32 en_idx;
FLOAT32 energy[4];
WORD32 min_sfb, max_sfb;
WORD32 done;
for (ch = chn; ch < chn + num_channels; ch++) {
if (pstr_psy_out[ch]->window_sequence != SHORT_WINDOW) {
start_sfb[ch] = pstr_str_ah_param->start_sfb_long;
} else {
start_sfb[ch] = pstr_str_ah_param->start_sfb_short;
}
}
average_energy = 0.0f;
min_energy = MAX_FLT_VAL;
ah_cnt = 0;
for (ch = chn; ch < chn + num_channels; ch++) {
ixheaace_psy_out_channel *pstr_psy_chan_out = pstr_psy_out[ch];
for (sfb = start_sfb[ch]; sfb < pstr_psy_chan_out->sfb_count; sfb++) {
if ((ah_flag[ch][sfb] != NO_AH) &&
(pstr_psy_chan_out->ptr_sfb_energy[sfb] > pstr_psy_chan_out->ptr_sfb_thr[sfb])) {
min_energy = MIN(min_energy, pstr_psy_chan_out->ptr_sfb_energy[sfb]);
average_energy += pstr_psy_chan_out->ptr_sfb_energy[sfb];
ah_cnt++;
}
}
}
average_energy = MIN(MAX_FLT_VAL, average_energy / (ah_cnt + MIN_FLT_VAL));
for (en_idx = 0; en_idx < 4; en_idx++) {
energy[en_idx] = min_energy * (FLOAT32)pow(average_energy / (min_energy + MIN_FLT_VAL),
(2 * en_idx + 1) / 7.0f);
}
max_sfb = pstr_psy_out[chn]->sfb_count - 1;
min_sfb = start_sfb[chn];
if (num_channels == 2) {
max_sfb = MAX(max_sfb, pstr_psy_out[chn + 1]->sfb_count - 1);
min_sfb = MIN(min_sfb, start_sfb[chn + 1]);
}
sfb = max_sfb;
en_idx = 0;
done = 0;
while (!done) {
for (ch = chn; ch < chn + num_channels; ch++) {
ixheaace_psy_out_channel *pstr_psy_chan_out = pstr_psy_out[ch];
if (sfb >= start_sfb[ch] && sfb < pstr_psy_chan_out->sfb_count) {
if (ah_flag[ch][sfb] != NO_AH &&
pstr_psy_chan_out->ptr_sfb_energy[sfb] < energy[en_idx]) {
ah_flag[ch][sfb] = NO_AH;
pstr_psy_chan_out->ptr_sfb_thr[sfb] = 2.0f * pstr_psy_chan_out->ptr_sfb_energy[sfb];
act_pe -= pstr_qs_pe_data->pe_ch_data[ch].sfb_pe[sfb];
}
if (act_pe < desired_pe) {
done = 1;
break;
}
}
}
sfb--;
if (sfb < min_sfb) {
sfb = max_sfb;
en_idx++;
if (en_idx >= 4) {
done = 1;
}
}
}
}
}
static VOID iaace_adapt_thr_to_pe(
ixheaace_psy_out_channel **pstr_psy_out,
ia_qc_pe_data_struct *pstr_qs_pe_data, const FLOAT32 desired_pe,
ia_ah_param_struct *pstr_ah_param, ia_min_snr_adapt_param_struct *pstr_msa_param,
WORD32 num_channels, WORD32 chn, WORD32 aot) {
FLOAT32 no_red_pe, red_pe, red_pe_no_ah;
FLOAT32 const_part, const_part_no_ah;
FLOAT32 num_active_lines, num_active_lines_no_ah;
FLOAT32 desired_pe_no_ah;
FLOAT32 redval = 0.0f;
WORD32 ah_flag[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND];
FLOAT32 thr_exp[IXHEAACE_MAX_CH_IN_BS_ELE][MAXIMUM_GROUPED_SCALE_FACTOR_BAND];
WORD32 iteration;
iaace_calc_thr_exp(thr_exp, pstr_psy_out, num_channels, chn);
iaace_adapt_min_snr(pstr_psy_out, pstr_msa_param, num_channels, chn);
iaace_init_avoid_hole_flag(ah_flag, pstr_psy_out, pstr_ah_param, num_channels, chn, aot);
no_red_pe = pstr_qs_pe_data->pe;
const_part = pstr_qs_pe_data->const_part;
num_active_lines = pstr_qs_pe_data->num_active_lines;
if (num_active_lines > FLT_EPSILON) {
FLOAT32 avg_thr_exp =
(FLOAT32)pow(2.0f, (const_part - no_red_pe) / (INV_RED_EXP_VAL * num_active_lines));
redval = (FLOAT32)pow(2.0f, (const_part - desired_pe) / (INV_RED_EXP_VAL * num_active_lines))
- avg_thr_exp;
redval = MAX(0.0f, redval);
iaace_reduce_thr(pstr_psy_out, ah_flag, thr_exp, redval, num_channels, chn);
}
iaace_calc_sfb_pe_data(pstr_qs_pe_data, pstr_psy_out, num_channels, chn);
red_pe = pstr_qs_pe_data->pe;
iteration = 0;
do {
iaace_calc_pe_no_active_holes(&red_pe_no_ah, &const_part_no_ah, &num_active_lines_no_ah,
pstr_qs_pe_data, ah_flag, pstr_psy_out, num_channels, chn);
desired_pe_no_ah = MAX(desired_pe - (red_pe - red_pe_no_ah), 0);
if (num_active_lines_no_ah > FLT_EPSILON) {
FLOAT32 avg_thr_exp = (FLOAT32)pow(
2.0f, (const_part_no_ah - red_pe_no_ah) / (INV_RED_EXP_VAL * num_active_lines_no_ah));
redval += (FLOAT32)pow(2.0f, (const_part_no_ah - desired_pe_no_ah) /
(INV_RED_EXP_VAL * num_active_lines_no_ah)) -
avg_thr_exp;
redval = MAX(0.0f, redval);
iaace_reduce_thr(pstr_psy_out, ah_flag, thr_exp, redval, num_channels, chn);
}
iaace_calc_sfb_pe_data(pstr_qs_pe_data, pstr_psy_out, num_channels, chn);
red_pe = pstr_qs_pe_data->pe;
iteration++;
} while ((fabs(red_pe - desired_pe) > (0.05f) * desired_pe) && (iteration < 2));
if (red_pe < 1.15f * desired_pe) {
iaace_correct_thr(pstr_psy_out, ah_flag, pstr_qs_pe_data, thr_exp, redval,
desired_pe - red_pe, num_channels, chn);
} else {
iaace_reduce_min_snr(pstr_psy_out, pstr_qs_pe_data, ah_flag, 1.05f * desired_pe, num_channels,
chn);
iaace_allow_more_holes(pstr_psy_out, pstr_qs_pe_data, ah_flag, pstr_ah_param,
1.05f * desired_pe, num_channels, chn);
}
}
VOID iaace_adjust_threshold(ia_adj_thr_state_struct *pstr_adj_thr_state,
ia_adj_thr_elem_struct *pstr_adj_thr_elem,
ixheaace_psy_out_channel **pstr_psy_out,
FLOAT32 *ptr_ch_bit_dist, ixheaace_qc_out_element *pstr_qc_out_el,
const WORD32 avg_bits, const WORD32 bitres_bits,
const WORD32 max_bitres_bits, const WORD32 side_info_bits,
FLOAT32 *max_bit_fac, FLOAT32 *ptr_sfb_n_relevant_lines,
FLOAT32 *ptr_sfb_ld_energy, WORD32 num_channels, WORD32 chn,
WORD32 aot, WORD8 *ptr_scratch) {
FLOAT32 no_red_pe, granted_pe, granted_pe_corr;
WORD32 curr_win_sequence;
ia_qc_pe_data_struct *pstr_qc_pe_data = (ia_qc_pe_data_struct *)ptr_scratch;
FLOAT32 bit_factor;
WORD32 ch;
for (ch = chn; ch < chn + num_channels; ch++) {
pstr_qc_pe_data->pe_ch_data[ch].sfb_lines =
ptr_sfb_n_relevant_lines + MAXIMUM_GROUPED_SCALE_FACTOR_BAND * ch;
pstr_qc_pe_data->pe_ch_data[ch].sfb_ld_energy =
ptr_sfb_ld_energy + MAXIMUM_GROUPED_SCALE_FACTOR_BAND * ch;
}
pstr_qc_pe_data->offset = pstr_adj_thr_elem->pe_offset;
iaace_calc_sfb_pe_data(pstr_qc_pe_data, pstr_psy_out, num_channels, chn);
no_red_pe = pstr_qc_pe_data->pe;
curr_win_sequence = LONG_WINDOW;
if (num_channels == 2) {
if ((pstr_psy_out[chn]->window_sequence == SHORT_WINDOW) ||
(pstr_psy_out[chn + 1]->window_sequence == SHORT_WINDOW)) {
curr_win_sequence = SHORT_WINDOW;
}
} else {
curr_win_sequence = pstr_psy_out[chn]->window_sequence;
}
bit_factor = iaace_bitres_calc_bitfac(
bitres_bits, max_bitres_bits, no_red_pe + 5.0f * side_info_bits, curr_win_sequence,
avg_bits, *max_bit_fac, pstr_adj_thr_state, pstr_adj_thr_elem);
granted_pe = bit_factor * iaace_bits_to_pe((FLOAT32)avg_bits);
iaace_calc_pe_correction(&(pstr_adj_thr_elem->pe_correction_fac), MIN(granted_pe, no_red_pe),
pstr_adj_thr_elem->pe_last, pstr_adj_thr_elem->dyn_bits_last);
granted_pe_corr = granted_pe * pstr_adj_thr_elem->pe_correction_fac;
if (granted_pe_corr < no_red_pe) {
iaace_adapt_thr_to_pe(pstr_psy_out, pstr_qc_pe_data, granted_pe_corr,
&pstr_adj_thr_elem->str_ah_param,
&pstr_adj_thr_elem->str_min_snr_adapt_params, num_channels, chn, aot);
}
for (ch = chn; ch < chn + num_channels; ch++) {
FLOAT32 tmp_var, temp1;
if (pstr_qc_pe_data->pe) {
tmp_var = 1.0f - num_channels * 0.2f;
temp1 = pstr_qc_pe_data->pe_ch_data[ch].pe / pstr_qc_pe_data->pe;
temp1 = temp1 * tmp_var;
ptr_ch_bit_dist[ch] = temp1 + 0.2f;
if (ptr_ch_bit_dist[ch] < 0.2f) {
ptr_ch_bit_dist[ch] = 0.2f;
}
} else {
ptr_ch_bit_dist[ch] = 0.2f;
}
}
pstr_qc_out_el->pe = no_red_pe;
pstr_adj_thr_elem->pe_last = granted_pe;
}