/******************************************************************************

 *                                                                            *

 * 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 "ixheaac_error_standards.h"

#include "ixheaace_error_codes.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 "ixheaace_common_rom.h"

#include "ixheaace_bitbuffer.h"

IA_ERRORCODE

ia_enhaacplus_enc_get_tns_param(

    ixheaace_temporal_noise_shaping_config_tabulated *pstr_tns_config_tab, WORD32 bit_rate,

    WORD32 channels, WORD32 block_type,

    const ixheaace_temporal_noise_shaping_info_tab *pstr_tns_info_tab, WORD32 size) {

  UWORD32 i;

  if (pstr_tns_config_tab == NULL) {

    return IA_EXHEAACE_INIT_FATAL_TNS_CONFIG_INIT_FAILED;

  }

  pstr_tns_config_tab->thresh_on = -1;

  for (i = 0; i < size / sizeof(ixheaace_temporal_noise_shaping_info_tab); i++) {

    if ((bit_rate >= pstr_tns_info_tab[i].bit_rate_from) &&

        bit_rate <= pstr_tns_info_tab[i].bit_rate_to) {

      switch (block_type) {

        case LONG_WINDOW:

          switch (channels) {

            case NUM_CHANS_MONO:

              pstr_tns_config_tab->thresh_on = pstr_tns_info_tab[i].param_mono_long.thresh_on;

              pstr_tns_config_tab->lpc_start_freq =

                  pstr_tns_info_tab[i].param_mono_long.lpc_start_freq;

              pstr_tns_config_tab->lpc_stop_freq =

                  pstr_tns_info_tab[i].param_mono_long.lpc_stop_freq;

              pstr_tns_config_tab->tns_time_resolution =

                  pstr_tns_info_tab[i].param_mono_long.tns_time_resolution;

              break;

            case NUM_CHANS_STEREO:

              pstr_tns_config_tab->thresh_on = pstr_tns_info_tab[i].param_stereo_long.thresh_on;

              pstr_tns_config_tab->lpc_start_freq =

                  pstr_tns_info_tab[i].param_stereo_long.lpc_start_freq;

              pstr_tns_config_tab->lpc_stop_freq =

                  pstr_tns_info_tab[i].param_stereo_long.lpc_stop_freq;

              pstr_tns_config_tab->tns_time_resolution =

                  pstr_tns_info_tab[i].param_stereo_long.tns_time_resolution;

              break;

          }

          break;

        case SHORT_WINDOW:

          switch (channels) {

            case NUM_CHANS_MONO:

              pstr_tns_config_tab->thresh_on = pstr_tns_info_tab[i].param_mono_short.thresh_on;

              pstr_tns_config_tab->lpc_start_freq =

                  pstr_tns_info_tab[i].param_mono_short.lpc_start_freq;

              pstr_tns_config_tab->lpc_stop_freq =

                  pstr_tns_info_tab[i].param_mono_short.lpc_stop_freq;

              pstr_tns_config_tab->tns_time_resolution =

                  pstr_tns_info_tab[i].param_mono_short.tns_time_resolution;

              break;

            case NUM_CHANS_STEREO:

              pstr_tns_config_tab->thresh_on = pstr_tns_info_tab[i].param_stereo_short.thresh_on;

              pstr_tns_config_tab->lpc_start_freq =

                  pstr_tns_info_tab[i].param_stereo_short.lpc_start_freq;

              pstr_tns_config_tab->lpc_stop_freq =

                  pstr_tns_info_tab[i].param_stereo_short.lpc_stop_freq;

              pstr_tns_config_tab->tns_time_resolution =

                  pstr_tns_info_tab[i].param_stereo_short.tns_time_resolution;

              break;

          }

          break;

      }

    }

  }

  // This check is not being done now

  if (pstr_tns_config_tab->thresh_on == -1) {

    return IA_EXHEAACE_INIT_FATAL_INVALID_TNS_PARAM;

  }

  return IA_NO_ERROR;

}

VOID ia_enhaacplus_enc_get_tns_max_bands(

    WORD32 sampling_rate, WORD32 block_type, WORD32 *tns_max_sfb,

    const ixheaace_temporal_noise_shaping_max_table *pstr_tns_max_bands_tab, WORD32 size,

    WORD32 aot, WORD32 frame_length) {

  UWORD32 i;

  *tns_max_sfb = -1;

  for (i = 0; i < size / sizeof(ixheaace_temporal_noise_shaping_max_table); i++) {

    if (sampling_rate == pstr_tns_max_bands_tab[i].sampling_rate) {

      if (aot == AOT_AAC_LC || aot == AOT_SBR || aot == AOT_PS) {

        if (block_type == SHORT_WINDOW) {

          *tns_max_sfb = (frame_length == (FRAME_LEN_SHORT_128))

                             ? pstr_tns_max_bands_tab[i].max_band_1024_short_lc

                             : pstr_tns_max_bands_tab[i].max_band_960_short_lc;

        } else {

          *tns_max_sfb = (frame_length == FRAME_LEN_1024)

                             ? pstr_tns_max_bands_tab[i].max_band_1024_long_lc

                             : pstr_tns_max_bands_tab[i].max_band_960_long_lc;

        }

      } else if (aot == AOT_AAC_LD || aot == AOT_AAC_ELD) {

        *tns_max_sfb = (frame_length == FRAME_LEN_512)

                           ? pstr_tns_max_bands_tab[i].max_band_512_ld

                           : pstr_tns_max_bands_tab[i].max_band_480_ld;

      }

      break;

    }

  }

}