/******************************************************************************
 *
 * 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_type_def.h"
#include "ixheaacd_mps_struct_def.h"
#include "ixheaacd_mps_res_rom.h"
#include "ixheaacd_mps_aac_struct.h"
#include "ixheaac_constants.h"
#include "ixheaacd_bitbuffer.h"
#include "ixheaacd_common_rom.h"
#include "ixheaacd_sbrdecsettings.h"
#include "ixheaacd_sbr_scale.h"
#include "ixheaacd_env_extr_part.h"
#include "ixheaacd_sbr_rom.h"
#include "ixheaacd_hybrid.h"
#include "ixheaacd_ps_dec.h"
#include "ixheaacd_mps_polyphase.h"
#include "ixheaacd_config.h"
#include "ixheaacd_qmf_dec.h"
#include "ixheaacd_mps_dec.h"
#include "ixheaacd_mps_macro_def.h"

VOID ixheaacd_dec_interp_umx(WORD32 m[MAX_PARAMETER_SETS][MAX_PARAMETER_BANDS], WORD32 *ptr_r,
                             WORD32 *ptr_m_prev, ia_heaac_mps_state_struct *pstr_mps_state) {
  WORD32 ts, ps = 0, pb;
  WORD32 *r_out;
  const WORD32 *reciprocal_tab = pstr_mps_state->ia_mps_dec_mps_table.m1_m2_table_ptr->reciprocal;

  WORD32 num_parameter_bands = pstr_mps_state->num_parameter_bands;
  WORD32 num_parameter_sets = pstr_mps_state->num_parameter_sets;

  WORD32 prev_slot = -1;
  WORD32 *prm_slot = pstr_mps_state->aux_struct->param_slot;
  WORD32 curr_slot = *prm_slot;

  WORD32 temp = reciprocal_tab[curr_slot];
  r_out = ptr_r;

  for (ts = prev_slot + 1; ts <= curr_slot; ts++) {
    for (pb = 0; pb < num_parameter_bands; pb++) {
      WORD32 alpha = temp * (ts + 1);
      WORD32 one_minus_alpha;
      WORD64 result;

      one_minus_alpha = (ONE_IN_Q28 - alpha);

      result = ((WORD64)(*ptr_m_prev) * (WORD64)one_minus_alpha);
      result += ((WORD64)(m[ps][pb]) * (WORD64)alpha);

      *r_out++ = (WORD32)((WORD64)result >> 28);
      ptr_m_prev++;
    }
    ptr_m_prev -= num_parameter_bands;
  }

  for (ps = 1; ps < num_parameter_sets; ps++) {
    WORD32 prev_slot = prm_slot[ps - 1];
    WORD32 curr_slot = prm_slot[ps];

    temp = reciprocal_tab[curr_slot - prev_slot - 1];

    for (ts = (prev_slot) + 1; ts <= (curr_slot); ts++) {
      for (pb = 0; pb < num_parameter_bands; pb++) {
        WORD32 alpha = (ts - prev_slot) * temp;
        WORD64 result;
        WORD32 one_minus_alpha;
        one_minus_alpha = (ONE_IN_Q28 - alpha);

        result = ((WORD64)(m[ps - 1][pb]) * (WORD64)one_minus_alpha);
        result += ((WORD64)(m[ps][pb]) * (WORD64)alpha);

        *r_out++ = (WORD32)((WORD64)result >> 28);
      }
    }
  }
  return;
}

VOID ixheaacd_apply_abs_kernels(WORD32 *ptr_r_in, WORD32 *ptr_r_out, SIZE_T *ptr_params) {
  WORD32 ts, qb;
  SIZE_T *idx_ptr = (SIZE_T *)ptr_params[0];
  WORD32 time_slots = (WORD32)ptr_params[1];
  WORD32 num_parameter_bands = (WORD32)ptr_params[2];
  WORD32 hybrid_bands = (WORD32)ptr_params[3];

  for (ts = 0; ts < time_slots; ts++) {
    SIZE_T *idx = idx_ptr;
    for (qb = 0; qb < hybrid_bands; qb++) {
      WORD32 idx_v = (WORD32)(*idx);
      *ptr_r_out++ = *(ptr_r_in + idx_v);
      idx++;
    }
    ptr_r_in += num_parameter_bands;
  }
  return;
}