/******************************************************************************
*                                                                            *
* Copyright (C) 2018 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 <stdio.h>
#include <string.h>

#include <math.h>

#include "ixheaacd_sbr_common.h"
#include "ixheaacd_type_def.h"

#include "ixheaacd_constants.h"
#include "ixheaacd_basic_ops32.h"
#include "ixheaacd_basic_ops16.h"
#include "ixheaacd_basic_ops40.h"
#include "ixheaacd_basic_ops.h"
#include "ixheaacd_bitbuffer.h"

#include "ixheaacd_basic_op.h"
#include "ixheaacd_intrinsics.h"
#include "ixheaacd_sbr_const.h"

#include "ixheaacd_defines.h"

#include "ixheaacd_aac_rom.h"

#include "ixheaacd_definitions.h"

#include "ixheaacd_error_codes.h"

#include "ixheaacd_pulsedata.h"

#include "ixheaacd_pns.h"
#include "ixheaacd_drc_data_struct.h"

#include "ixheaacd_lt_predict.h"

#include "ixheaacd_channelinfo.h"
#include "ixheaacd_cnst.h"
#include "ixheaacd_drc_dec.h"
#include "ixheaacd_sbrdecoder.h"
#include "ixheaacd_block.h"
#include "ixheaacd_channel.h"

#include "ixheaacd_sbr_payload.h"
#include "ixheaacd_common_rom.h"

#include "ixheaacd_type_def.h"

#include "ixheaacd_sbrdecsettings.h"
#include "ixheaacd_sbr_scale.h"
#include "ixheaacd_env_extr_part.h"
#include "ixheaacd_sbr_rom.h"
#include "ixheaacd_stereo.h"
#include "ixheaacd_lpp_tran.h"
#include "ixheaacd_hybrid.h"
#include "ixheaacd_ps_dec.h"
#include "ixheaacd_env_extr.h"
#include "ixheaacd_adts.h"
#
#include "ixheaacd_audioobjtypes.h"
#include "ixheaacd_memory_standards.h"
#include "ixheaacd_latmdemux.h"
#include "ixheaacd_aacdec.h"
#include "ixheaacd_mps_polyphase.h"
#include "ixheaacd_config.h"
#include "ixheaacd_qmf_dec.h"
#include "ixheaacd_env_calc.h"

#include "ixheaacd_pvc_dec.h"
#include "ixheaacd_sbr_dec.h"
#include "ixheaacd_mps_dec.h"
#include "ixheaacd_struct_def.h"

#include "ixheaacd_mps_process.h"

#include "ixheaacd_ld_mps_dec.h"
#include "ixheaacd_interface.h"
#include "ixheaacd_info.h"
#include "ixheaacd_tns_usac.h"
#include "ixheaacd_acelp_info.h"
#include "ixheaacd_main.h"
#include "ixheaacd_struct.h"
#include "ixheaacd_create.h"

WORD32 ixheaacd_ld_qmf_analysis(ia_mps_dec_state_struct *self,
                                WORD16 *output_buf) {
  WORD32 *p_arr_qmf_buf_real[MAX_ENV_COLS + MAX_ENV_COLS];
  WORD32 **p_arr_qmf_buf_imag = &p_arr_qmf_buf_real[MAX_ENV_COLS];
  WORD32 temp_arr[64 * 64];
  WORD32 *ptr = &temp_arr[0];
  float in_gain_n_q8_fac = self->input_gain * 0.00390625f;

  int i;
  WORD32 ts;

  for (ts = 2; ts < self->time_slots + 2; ts++) {
    p_arr_qmf_buf_real[ts] = ptr;
    ptr += NO_SYNTHESIS_CHANNELS;

    p_arr_qmf_buf_imag[ts] = ptr;
    ptr += NO_SYNTHESIS_CHANNELS;
  }

  self->str_mps_qmf_bank.num_time_slots = self->time_slots;
  self->str_mps_qmf_bank.no_channels = self->qmf_band_count;

  ixheaacd_cplx_anal_qmffilt_32((WORD32 *)output_buf, self->str_sbr_scale_fact,
                                &p_arr_qmf_buf_real[0], &p_arr_qmf_buf_imag[0],
                                &self->str_mps_qmf_bank,
                                self->qmf_dec_tables_ptr, 2, 0);

  for (ts = 0; ts < self->time_slots; ts++) {
    for (i = 0; i < self->qmf_band_count; i++) {
      self->qmf_in[0][ts][i].re =
          (p_arr_qmf_buf_real[ts + 2][i] * in_gain_n_q8_fac);
      self->qmf_in[0][ts][i].im =
          (p_arr_qmf_buf_imag[ts + 2][i] * in_gain_n_q8_fac);
    }
  }

  return 0;
}

WORD32 ixheaacd_ld_mps_apply(ia_exhaacplus_dec_api_struct *p_obj_exhaacplus_dec,
                             WORD16 *output_buf) {
  ia_aac_dec_state_struct *aac_handle = p_obj_exhaacplus_dec->p_state_aac;
  ia_mps_dec_state_struct *mps_handle = &(aac_handle->mps_dec_handle);
  WORD32 err = 0;
  WORD32 ts, i;
  ia_dec_data_struct *pstr_dec_data =
      (ia_dec_data_struct *)aac_handle->pstr_dec_data;
  ia_usac_data_struct *pstr_usac_data = &(pstr_dec_data->str_usac_data);

  mps_handle->hyb_band_count[0] = mps_handle->hyb_band_count_max;
  mps_handle->hyb_band_count[1] = mps_handle->hyb_band_count_max;

  err = ixheaacd_mps_frame_decode(mps_handle);
  if (err != IA_NO_ERROR) return err;

  if (mps_handle->ldmps_config.bs_tree_config == 7) {
    ixheaacd_pre_and_mix_matrix_calculation(&(aac_handle->mps_dec_handle));
    ixheaacd_mps_pre_matrix_mix_matrix_smoothing(&(aac_handle->mps_dec_handle));
  }
  mps_handle->output_buffer = pstr_usac_data->time_sample_vector;
  if (mps_handle->ldmps_config.no_ldsbr_present)
    ixheaacd_ld_qmf_analysis(mps_handle, output_buf);
  else {
    for (ts = 0; ts < mps_handle->time_slots; ts++) {
      FLOAT32 *qmf_re = p_obj_exhaacplus_dec->p_state_aac->str_sbr_dec_info[0]
                          ->pstr_sbr_channel[0]
                          ->str_sbr_dec.mps_qmf_buf_real[ts + 2];
      FLOAT32 *qmf_im = p_obj_exhaacplus_dec->p_state_aac->str_sbr_dec_info[0]
                          ->pstr_sbr_channel[0]
                          ->str_sbr_dec.mps_qmf_buf_imag[ts + 2];
      for (i = 0; i < mps_handle->qmf_band_count; i++) {
        mps_handle->qmf_in[0][ts][i].re = qmf_re[i] * mps_handle->input_gain;
        mps_handle->qmf_in[0][ts][i].im = qmf_im[i] * mps_handle->input_gain;
      }
    }
  }
  ixheaacd_mps_qmf_hyb_analysis(mps_handle);

  err = ixheaacd_mps_apply_pre_matrix(mps_handle);
  if (err < 0) return err;

  ixheaacd_mps_create_w(mps_handle);

  if ((!(mps_handle->res_bands | mps_handle->pre_mix_req)) &&
      (mps_handle->config->bs_phase_coding == 0)) {
    err = ixheaacd_mps_apply_mix_matrix_type1(mps_handle);

  } else if (mps_handle->pre_mix_req) {
    err = ixheaacd_mps_apply_mix_matrix_type2(mps_handle);

  } else {
    err = ixheaacd_mps_apply_mix_matrix_type3(mps_handle);
  }
  if (err < 0) return err;

  if (mps_handle->ldmps_config.bs_temp_shape_config == 2) {
    ixheaacd_mps_time_env_shaping(mps_handle);
  }

  err = ixheaacd_mps_temp_process(mps_handle);
  mps_handle->pre_mix_req = 0;
  if (err)
    return err;

  return err;
}