/******************************************************************************
 *                                                                            *
 * 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;
}