/******************************************************************************
 *                                                                            *
 * 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 <math.h>
#include <stdlib.h>

#include "ixheaac_type_def.h"
#include "ixheaac_constants.h"
#include "ixheaace_aac_constants.h"
#include "ixheaac_error_standards.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_psy_const.h"
#include "ixheaace_tns.h"
#include "ixheaace_tns_params.h"
#include "ixheaace_rom.h"
#include "ixheaace_bitbuffer.h"
#include "ixheaace_psy_configuration.h"
#include "ixheaace_tns_func.h"
#include "ixheaace_common_utils.h"

static WORD32 ia_enhaacplus_enc_freq_to_band_with_rounding(WORD32 freq, WORD32 fs,
                                                           WORD32 num_bands,
                                                           const WORD32 *ptr_band_start_offset) {
  WORD32 line_num, band;

  line_num = (freq * ptr_band_start_offset[num_bands] * 4 / fs + 1) / 2;

  /* freq > fs/2 */
  if (line_num >= ptr_band_start_offset[num_bands]) {
    return num_bands;
  }

  /* find band the line number lies in */
  /* bandStartOffset[] */
  for (band = 0; band < num_bands; band++) {
    if (ptr_band_start_offset[band + 1] > line_num) break;
  }

  /* round to nearest band border */
  if (line_num - ptr_band_start_offset[band] > ptr_band_start_offset[band + 1] - line_num) {
    band++;
  }

  return band;
}

static VOID ia_enhaacplus_enc_calc_gauss_window(FLOAT32 *ptr_win, const WORD16 win_size,
                                                const WORD32 sampling_rate,
                                                const WORD16 block_type,
                                                const FLOAT32 time_resolution) {
  WORD16 i;

  FLOAT32 accu_gauss_exp;
  accu_gauss_exp = (sampling_rate * time_resolution) * PI_BY_1000;

  if (block_type != SHORT_WINDOW) {
    accu_gauss_exp = (FLOAT32)(accu_gauss_exp / FRAME_LEN_1024);
  } else {
    accu_gauss_exp = (FLOAT32)(accu_gauss_exp / FRAME_LEN_SHORT_128);
  }

  accu_gauss_exp = -(accu_gauss_exp * 0.5f * accu_gauss_exp);

  for (i = 0; i < win_size; i++) {
    ptr_win[i] = (FLOAT32)exp(accu_gauss_exp * (i + 0.5) * (i + 0.5));
  }
}

IA_ERRORCODE
ia_enhaacplus_enc_init_tns_configuration(WORD32 bit_rate, WORD32 sample_rate, WORD32 channels,
                                         ixheaace_temporal_noise_shaping_config *pstr_tns_config,
                                         ixheaace_psy_configuration_long *pstr_psy_config,
                                         WORD32 active,
                                         ixheaace_temporal_noise_shaping_tables *pstr_tns_tab,
                                         WORD32 frame_length, WORD32 aot)

{
  IA_ERRORCODE error;
  pstr_tns_config->max_order = TEMPORAL_NOISE_SHAPING_MAX_ORDER;
  pstr_tns_config->tns_start_freq = TEMPORAL_NOISE_SHAPING_START_FREQ;
  pstr_tns_config->coef_res = TEMPORAL_NOISE_SHAPING_COEF_RES;

  error = ia_enhaacplus_enc_get_tns_param(&pstr_tns_config->conf_tab, bit_rate / channels,
                                          channels, LONG_WINDOW, pstr_tns_tab->tns_info_tab,
                                          sizeof(pstr_tns_tab->tns_info_tab));

  if (error != IA_NO_ERROR) {
    return error;
  }

  ia_enhaacplus_enc_calc_gauss_window(pstr_tns_config->acf_window_float,
                                      (const WORD16)(pstr_tns_config->max_order + 1), sample_rate,
                                      LONG_WINDOW, pstr_tns_config->conf_tab.tns_time_resolution

  );

  ia_enhaacplus_enc_get_tns_max_bands(
      sample_rate, LONG_WINDOW, &pstr_tns_config->tns_max_sfb, pstr_tns_tab->tns_max_bands_table,
      sizeof(pstr_tns_tab->tns_max_bands_table), aot, frame_length);

  pstr_tns_config->tns_active = 1;

  if (active == 0) {
    pstr_tns_config->tns_active = 0;
  }

  /*now calc band and line borders */
  pstr_tns_config->tns_stop_band = MIN(pstr_psy_config->sfb_cnt, pstr_tns_config->tns_max_sfb);

  pstr_tns_config->tns_stop_line = pstr_psy_config->sfb_offsets[pstr_tns_config->tns_stop_band];

  pstr_tns_config->tns_start_band = ia_enhaacplus_enc_freq_to_band_with_rounding(
      pstr_tns_config->tns_start_freq, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->tns_modify_begin_cb = ia_enhaacplus_enc_freq_to_band_with_rounding(
      TEMPORAL_NOISE_SHAPING_MODIFY_BEGIN, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->tns_ratio_patch_lowest_cb = ia_enhaacplus_enc_freq_to_band_with_rounding(
      RATIO_PATCH_LOWER_BORDER, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->tns_start_line = pstr_psy_config->sfb_offsets[pstr_tns_config->tns_start_band];

  pstr_tns_config->lpc_stop_band = ia_enhaacplus_enc_freq_to_band_with_rounding(
      pstr_tns_config->conf_tab.lpc_stop_freq, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->lpc_stop_band =
      MIN(pstr_tns_config->lpc_stop_band, pstr_psy_config->sfb_active);

  pstr_tns_config->lpc_stop_line = pstr_psy_config->sfb_offsets[pstr_tns_config->lpc_stop_band];
  pstr_tns_config->lpc_start_band = ia_enhaacplus_enc_freq_to_band_with_rounding(
      pstr_tns_config->conf_tab.lpc_start_freq, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->lpc_start_line = pstr_psy_config->sfb_offsets[pstr_tns_config->lpc_start_band];
  pstr_tns_config->threshold = pstr_tns_config->conf_tab.thresh_on;

  return IA_NO_ERROR;
}

IA_ERRORCODE ia_enhaacplus_enc_init_tns_configuration_short(
    WORD32 bit_rate, WORD32 sample_rate, WORD32 channels,
    ixheaace_temporal_noise_shaping_config *pstr_tns_config,
    ixheaace_psy_configuration_short *pstr_psy_config, WORD32 active,
    ixheaace_temporal_noise_shaping_tables *pstr_tns_tab, WORD32 frame_length, WORD32 aot) {
  IA_ERRORCODE error = IA_NO_ERROR;
  pstr_tns_config->max_order = TEMPORAL_NOISE_SHAPING_MAX_ORDER_SHORT;
  pstr_tns_config->tns_start_freq = TEMPORAL_NOISE_SHAPING_START_FREQ_SHORT;
  pstr_tns_config->coef_res = TEMPORAL_NOISE_SHAPING_COEF_RES_SHORT;

  error = ia_enhaacplus_enc_get_tns_param(&pstr_tns_config->conf_tab, bit_rate / channels,
                                          channels, SHORT_WINDOW, pstr_tns_tab->tns_info_tab,
                                          sizeof(pstr_tns_tab->tns_info_tab));
  if (error != IA_NO_ERROR) {
    return error;
  }
  ia_enhaacplus_enc_calc_gauss_window(
      pstr_tns_config->acf_window_float, (const WORD16)(pstr_tns_config->max_order + 1),
      sample_rate, SHORT_WINDOW, pstr_tns_config->conf_tab.tns_time_resolution);

  ia_enhaacplus_enc_get_tns_max_bands(
      sample_rate, SHORT_WINDOW, &pstr_tns_config->tns_max_sfb, pstr_tns_tab->tns_max_bands_table,
      sizeof(pstr_tns_tab->tns_max_bands_table), aot, frame_length);

  pstr_tns_config->tns_active = 1;

  if (active == 0) {
    pstr_tns_config->tns_active = 0;
  }

  /*now calc band and line borders */

  pstr_tns_config->tns_stop_band = MIN(pstr_psy_config->sfb_cnt, pstr_tns_config->tns_max_sfb);

  pstr_tns_config->tns_stop_line = pstr_psy_config->sfb_offsets[pstr_tns_config->tns_stop_band];

  pstr_tns_config->tns_start_band = ia_enhaacplus_enc_freq_to_band_with_rounding(
      pstr_tns_config->tns_start_freq, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->tns_modify_begin_cb = ia_enhaacplus_enc_freq_to_band_with_rounding(
      TEMPORAL_NOISE_SHAPING_MODIFY_BEGIN, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->tns_ratio_patch_lowest_cb = ia_enhaacplus_enc_freq_to_band_with_rounding(
      RATIO_PATCH_LOWER_BORDER, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->tns_start_line = pstr_psy_config->sfb_offsets[pstr_tns_config->tns_start_band];

  pstr_tns_config->lpc_stop_band = ia_enhaacplus_enc_freq_to_band_with_rounding(
      pstr_tns_config->conf_tab.lpc_stop_freq, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->lpc_stop_band =
      MIN(pstr_tns_config->lpc_stop_band, pstr_psy_config->sfb_active);

  pstr_tns_config->lpc_stop_line = pstr_psy_config->sfb_offsets[pstr_tns_config->lpc_stop_band];

  pstr_tns_config->lpc_start_band = ia_enhaacplus_enc_freq_to_band_with_rounding(
      pstr_tns_config->conf_tab.lpc_start_freq, sample_rate, pstr_psy_config->sfb_cnt,
      pstr_psy_config->sfb_offsets);

  pstr_tns_config->lpc_start_line = pstr_psy_config->sfb_offsets[pstr_tns_config->lpc_start_band];
  pstr_tns_config->threshold = pstr_tns_config->conf_tab.thresh_on;

  return IA_NO_ERROR;
}

const WORD32 ia_enhaacplus_enc_m_log2_table[INT_BITS] = {
    0x00000000, 0x4ae00d00, 0x2934f080, 0x15c01a3f, 0x0b31fb80, 0x05aeb4e0, 0x02dcf2d0,
    0x016fe50c, 0x00b84e23, 0x005c3e10, 0x002e24ca, 0x001713d6, 0x000b8a47, 0x0005c53b,
    0x0002e2a3, 0x00017153, 0x0000b8aa, 0x00005c55, 0x00002e2b, 0x00001715, 0x00000b8b,
    0x000005c5, 0x000002e3, 0x00000171, 0x000000b9, 0x0000005c, 0x0000002e, 0x00000017,
    0x0000000c, 0x00000006, 0x00000003, 0x00000001};