/******************************************************************************
 *                                                                            *
 * 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 <string.h>
#include "ixheaac_error_standards.h"
#include "ixheaac_type_def.h"
#include "ixheaace_sbr_def.h"
#include "ixheaace_resampler.h"
#include "ixheaace_sbr_rom.h"
#include "ixheaace_common_rom.h"
#include "ixheaace_bitbuffer.h"
#include "ixheaace_sbr_hybrid.h"
#include "ixheaace_sbr_ps_enc.h"
#include "ixheaac_constants.h"
#include "ixheaace_aac_constants.h"
#include "ixheaace_error_codes.h"
#include "ixheaace_bitbuffer.h"

WORD32 ixheaace_get_ps_mode(WORD32 bitrate) {
  WORD32 ps_mode = 0;

  if (bitrate < 21000) {
    ps_mode = PS_MODE_LOW_FREQ_RES_IID_ICC;
  }

  return ps_mode;
}

IA_ERRORCODE
ixheaace_create_ps_enc(ixheaace_pstr_ps_enc pstr_ps_enc, WORD32 ps_mode,
                       FLOAT32 *ptr_common_buffer, FLOAT32 *ptr_common_buffer2,
                       FLOAT32 *ptr_ps_buf3) {
  WORD32 i;
  IA_ERRORCODE err = IA_NO_ERROR;

  FLOAT32 *ptr1, *ptr2, *ptr3, *ptr4;

  if (pstr_ps_enc == NULL) {
    return IA_EXHEAACE_INIT_FATAL_PS_INIT_FAILED;
  }
  ptr1 = &ptr_common_buffer2[IXHEAACE_QMF_TIME_SLOTS * IXHEAACE_QMF_CHANNELS];
  ptr2 = pstr_ps_enc->ps_buf2;
  ptr3 = ptr_ps_buf3;
  ptr4 = &ptr_common_buffer[5 * NO_OF_ESTIMATES * MAXIMUM_FREQ_COEFFS];

  pstr_ps_enc->ps_mode = ps_mode;
  pstr_ps_enc->b_prev_zero_iid = 0;
  pstr_ps_enc->b_prev_zero_icc = 0;
  pstr_ps_enc->b_hi_freq_res_iid_icc = ((ps_mode & PS_MODE_LOW_FREQ_RES_IID_ICC) != 0) ? 0 : 1;
  pstr_ps_enc->iid_icc_bins =
      (pstr_ps_enc->b_hi_freq_res_iid_icc) ? NUMBER_OF_IID_BINS : NUMBER_OF_LOW_RES_IID_BINS;

  pstr_ps_enc->aaa_ICC_data_buf = (FLOAT32 **)ptr1;
  ptr1 += NUMBER_OF_BINS * sizeof(FLOAT32 *) / sizeof(FLOAT32);

  pstr_ps_enc->aaa_IID_data_buf = (FLOAT32 **)ptr1;
  ptr1 += NUMBER_OF_BINS * sizeof(FLOAT32 *) / sizeof(FLOAT32);

  for (i = 0; i < NUMBER_OF_BINS; i++) {
    pstr_ps_enc->aaa_ICC_data_buf[i] = ptr1;
    ptr1 += SYSTEMLOOKAHEAD + 1;

    memset(pstr_ps_enc->aaa_ICC_data_buf[i], 0,
           (SYSTEMLOOKAHEAD + 1) * sizeof(pstr_ps_enc->aaa_ICC_data_buf[0]));

    pstr_ps_enc->aaa_IID_data_buf[i] = ptr1;
    ptr1 += SYSTEMLOOKAHEAD + 1;

    memset(pstr_ps_enc->aaa_IID_data_buf[i], 0,
           (SYSTEMLOOKAHEAD + 1) * sizeof(pstr_ps_enc->aaa_IID_data_buf[0]));
  }

  pstr_ps_enc->ptr_hybrid_left = &pstr_ps_enc->hybrid_left;
  pstr_ps_enc->ptr_hybrid_right = &pstr_ps_enc->hybrid_right;

  err = ixheaace_create_hybrid_filter_bank(pstr_ps_enc->ptr_hybrid_left, &ptr4);

  if (err) {
    return err;
  }

  err = ixheaace_create_hybrid_filter_bank(pstr_ps_enc->ptr_hybrid_right, &ptr4);

  if (err) {
    return err;
  }

  for (i = 0; i < NUMBER_OF_SUBSAMPLES; i++) {
    pstr_ps_enc->m_hybrid_real_left[i] = ptr3;
    ptr3 += IXHEAACE_NUM_HYBRID_BANDS;

    memset(pstr_ps_enc->m_hybrid_real_left[i], 0,
           IXHEAACE_NUM_HYBRID_BANDS * sizeof(pstr_ps_enc->m_hybrid_real_left[0]));

    pstr_ps_enc->m_hybrid_imag_left[i] = ptr3;
    ptr3 += IXHEAACE_NUM_HYBRID_BANDS;

    memset(pstr_ps_enc->m_hybrid_imag_left[i], 0,
           IXHEAACE_NUM_HYBRID_BANDS * sizeof(pstr_ps_enc->m_hybrid_imag_left[0]));

    pstr_ps_enc->m_hybrid_real_right[i] = ptr1;
    ptr1 += IXHEAACE_NUM_HYBRID_BANDS;

    memset(pstr_ps_enc->m_hybrid_real_right[i], 0,
           IXHEAACE_NUM_HYBRID_BANDS * sizeof(pstr_ps_enc->m_hybrid_real_right[0]));

    pstr_ps_enc->m_hybrid_imag_right[i] = ptr1;
    ptr1 += IXHEAACE_NUM_HYBRID_BANDS;

    memset(pstr_ps_enc->m_hybrid_imag_right[i], 0,
           IXHEAACE_NUM_HYBRID_BANDS * sizeof(pstr_ps_enc->m_hybrid_imag_right[0]));
  }

  pstr_ps_enc->temp_qmf_left_real = (FLOAT32 **)ptr1;
  ptr1 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);

  pstr_ps_enc->temp_qmf_left_imag = (FLOAT32 **)ptr1;
  ptr1 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(WORD32 *) / sizeof(WORD32);

  pstr_ps_enc->hist_qmf_left_real = (FLOAT32 **)ptr2;
  ptr2 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);

  pstr_ps_enc->hist_qmf_left_imag = (FLOAT32 **)ptr2;
  ptr2 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);

  pstr_ps_enc->hist_qmf_right_real = (FLOAT32 **)ptr2;
  ptr2 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);

  pstr_ps_enc->hist_qmf_right_imag = (FLOAT32 **)ptr2;
  ptr2 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);

  for (i = 0; i < IXHEAACE_HYBRID_FILTER_DELAY; i++) {
    pstr_ps_enc->temp_qmf_left_real[i] = ptr1;
    ptr1 += NUMBER_OF_QMF_BANDS;
    memset(pstr_ps_enc->temp_qmf_left_real[i], 0,
           NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->temp_qmf_left_real[0]));

    pstr_ps_enc->temp_qmf_left_imag[i] = ptr1;
    ptr1 += NUMBER_OF_QMF_BANDS;

    memset(pstr_ps_enc->temp_qmf_left_imag[i], 0,
           NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->temp_qmf_left_imag[0]));

    pstr_ps_enc->hist_qmf_left_real[i] = ptr2;
    ptr2 += NUMBER_OF_QMF_BANDS;

    memset(pstr_ps_enc->hist_qmf_left_real[i], 0,
           NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->hist_qmf_left_real[0]));

    pstr_ps_enc->hist_qmf_left_imag[i] = ptr2;
    ptr2 += NUMBER_OF_QMF_BANDS;

    memset(pstr_ps_enc->hist_qmf_left_imag[i], 0,
           NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->hist_qmf_left_imag[0]));

    pstr_ps_enc->hist_qmf_right_real[i] = ptr2;
    ptr2 += NUMBER_OF_QMF_BANDS;

    memset(pstr_ps_enc->hist_qmf_right_real[i], 0,
           NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->hist_qmf_right_real[0]));

    pstr_ps_enc->hist_qmf_right_imag[i] = ptr2;
    ptr2 += NUMBER_OF_QMF_BANDS;

    memset(pstr_ps_enc->hist_qmf_right_imag[i], 0,
           NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->hist_qmf_right_imag[0]));
  }

  memset(pstr_ps_enc->pow_left_right, 0, sizeof(pstr_ps_enc->pow_left_right));
  memset(pstr_ps_enc->pow_corr_real_imag, 0, sizeof(pstr_ps_enc->pow_corr_real_imag));

  if ((pstr_ps_enc->hist_qmf_left_real == NULL) || (pstr_ps_enc->hist_qmf_left_imag == NULL) ||
      (pstr_ps_enc->hist_qmf_right_real == NULL) || (pstr_ps_enc->hist_qmf_right_imag == NULL)) {
    return IA_EXHEAACE_INIT_FATAL_PS_INIT_FAILED;
  }

  for (i = 0; i < pstr_ps_enc->iid_icc_bins; i++) {
    pstr_ps_enc->aaa_IID_data_buf[i][0] = 0;
    pstr_ps_enc->aaa_ICC_data_buf[i][0] = -1.0f;
  }

  pstr_ps_enc->bit_buf_read_offset = 0;
  pstr_ps_enc->bit_buf_write_offset = 0;

  ia_enhaacplus_enc_create_bitbuffer(&pstr_ps_enc->ps_bit_buf, (UWORD8 *)ptr1, 255 + 15);

  pstr_ps_enc->ps_bit_buf.ptr_read_next =
      pstr_ps_enc->ps_bit_buf.ptr_bit_buf_base + pstr_ps_enc->bit_buf_read_offset;
  pstr_ps_enc->ps_bit_buf.ptr_write_next =
      pstr_ps_enc->ps_bit_buf.ptr_bit_buf_base + pstr_ps_enc->bit_buf_write_offset;

  return err;
}