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

 *

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

#include "ixheaac_constants.h"

#include "ixheaac_basic_ops32.h"

#include "ixheaac_basic_ops40.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 "ixheaac_error_standards.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"

#include "ixheaacd_mps_basic_op.h"

#include "ixheaacd_mps_calc_m1m2_common.h"

VOID ixheaacd_calc_m1m2_emm(ia_heaac_mps_state_struct *pstr_mps_state) {

  ia_mps_dec_auxilary_struct *p_aux_struct = pstr_mps_state->aux_struct;

  ia_mps_dec_m2_param_struct *m2_param = p_aux_struct->m2_param;

  ia_mps_dec_m1_param_struct *m1_param = pstr_mps_state->array_struct->m1_param;

  WORD32 ps;

  WORD32 pb;

  WORD32 col;

  WORD32 row;

  WORD64 acc;

  WORD32 *h11, *h12, *h21, *h22, *dummy1, *dummy2;

  WORD16 *dummy3, *dummy4;

  h11 = pstr_mps_state->mps_scratch_mem_v;

  h12 = h11 + IXHEAAC_GET_SIZE_ALIGNED_TYPE(MAX_PARAMETER_BANDS, sizeof(*h12), BYTE_ALIGN_8);

  h21 = h12 + IXHEAAC_GET_SIZE_ALIGNED_TYPE(MAX_PARAMETER_BANDS, sizeof(*h21), BYTE_ALIGN_8);

  h22 = h21 + IXHEAAC_GET_SIZE_ALIGNED_TYPE(MAX_PARAMETER_BANDS, sizeof(*h22), BYTE_ALIGN_8);

  dummy1 =

      h22 + IXHEAAC_GET_SIZE_ALIGNED_TYPE(MAX_PARAMETER_BANDS, sizeof(*dummy1), BYTE_ALIGN_8);

  dummy2 =

      dummy1 + IXHEAAC_GET_SIZE_ALIGNED_TYPE(MAX_PARAMETER_BANDS, sizeof(*dummy2), BYTE_ALIGN_8);

  dummy3 = (WORD16 *)pstr_mps_state->mps_scratch_mem_v +

           IXHEAAC_GET_SIZE_ALIGNED_TYPE(PARAMETER_BANDSX12, sizeof(*dummy3), BYTE_ALIGN_8);

  dummy4 =

      dummy3 + IXHEAAC_GET_SIZE_ALIGNED_TYPE(MAX_PARAMETER_BANDS, sizeof(*dummy4), BYTE_ALIGN_8);

  for (ps = 0; ps < pstr_mps_state->num_parameter_sets; ps++) {

    ixheaacd_param_2_umx_ps(pstr_mps_state, h11, h12, h21, h22, dummy1, dummy2, dummy3, dummy4, 0,

                            ps, 0);

    for (pb = 0; pb < pstr_mps_state->num_parameter_bands; pb++) {

      WORD32 m11 = p_aux_struct->ttt_cpc_1[0][ps][pb] + ONE_IN_Q16;

      WORD32 m12 = p_aux_struct->ttt_cpc_2[0][ps][pb] - ONE_IN_Q15;

      WORD32 m21 = p_aux_struct->ttt_cpc_1[0][ps][pb] - ONE_IN_Q15;

      WORD32 m22 = p_aux_struct->ttt_cpc_2[0][ps][pb] + ONE_IN_Q16;

      WORD32 m31 = ONE_IN_Q15 - p_aux_struct->ttt_cpc_1[0][ps][pb];

      WORD32 m32 = ONE_IN_Q15 - p_aux_struct->ttt_cpc_2[0][ps][pb];

      WORD32 weight1;

      WORD32 weight2;

      WORD32 h_real[2][2];

      WORD32 h_imag[2][2];

      ia_mps_dec_spatial_bs_frame_struct *p_cur_bs = pstr_mps_state->bs_frame;

      ixheaacd_get_matrix_inversion_weights(

          p_cur_bs->ott_cld_idx[0][ps][pb], p_cur_bs->ott_cld_idx[0][ps][pb], 1,

          p_aux_struct->ttt_cpc_1[0][ps][pb], p_aux_struct->ttt_cpc_2[0][ps][pb], &weight1,

          &weight2, &(pstr_mps_state->ia_mps_dec_mps_table));

      ixheaacd_invert_matrix(weight1, weight2, h_real, h_imag,

                             pstr_mps_state->ia_mps_dec_mps_table.common_table_ptr);

      pstr_mps_state->m1_param_imag_present = 1;

      acc = (WORD64)((WORD64)h_real[0][0] * (WORD64)m11 + (WORD64)h_real[1][0] * (WORD64)m12);

      acc >>= 15;

      m1_param->m1_param_real[0][0][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_real[0][1] * (WORD64)m11 + (WORD64)h_real[1][1] * (WORD64)m12);

      acc >>= 15;

      m1_param->m1_param_real[0][1][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_imag[0][0] * (WORD64)m11 + (WORD64)h_imag[1][0] * (WORD64)m12);

      acc >>= 15;

      m1_param->m1_param_imag[0][0][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_imag[0][1] * (WORD64)m11 + (WORD64)h_imag[1][1] * (WORD64)m12);

      acc >>= 15;

      m1_param->m1_param_imag[0][1][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_real[0][0] * (WORD64)m21 + (WORD64)h_real[1][0] * (WORD64)m22);

      acc >>= 15;

      m1_param->m1_param_real[1][0][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_real[0][1] * (WORD64)m21 + (WORD64)h_real[1][1] * (WORD64)m22);

      acc >>= 15;

      m1_param->m1_param_real[1][1][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_imag[0][0] * (WORD64)m21 + (WORD64)h_imag[1][0] * (WORD64)m22);

      acc >>= 15;

      m1_param->m1_param_imag[1][0][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_imag[0][1] * (WORD64)m21 + (WORD64)h_imag[1][1] * (WORD64)m22);

      acc >>= 15;

      m1_param->m1_param_imag[1][1][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_real[0][0] * (WORD64)m31 + (WORD64)h_real[1][0] * (WORD64)m32);

      acc >>= 15;

      m1_param->m1_param_real[2][0][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_real[0][1] * (WORD64)m31 + (WORD64)h_real[1][1] * (WORD64)m32);

      acc >>= 15;

      m1_param->m1_param_real[2][1][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_imag[0][0] * (WORD64)m31 + (WORD64)h_imag[1][0] * (WORD64)m32);

      acc >>= 15;

      m1_param->m1_param_imag[2][0][ps][pb] = (WORD32)acc;

      acc = (WORD64)((WORD64)h_imag[0][1] * (WORD64)m31 + (WORD64)h_imag[1][1] * (WORD64)m32);

      acc >>= 15;

      m1_param->m1_param_imag[2][1][ps][pb] = (WORD32)acc;

      m1_param->m1_param_real[0][2][ps][pb] = ONE_IN_Q15;

      m1_param->m1_param_real[1][2][ps][pb] = ONE_IN_Q15;

      m1_param->m1_param_real[2][2][ps][pb] = -32768;

      m1_param->m1_param_imag[0][2][ps][pb] = 0;

      m1_param->m1_param_imag[1][2][ps][pb] = 0;

      m1_param->m1_param_imag[2][2][ps][pb] = 0;

      for (row = 0; row < 3; row++) {

        for (col = 0; col < 3; col++) {

          m1_param->m1_param_real[row][col][ps][pb] = ixheaacd_mps_mult32_shr_15(

              m1_param->m1_param_real[row][col][ps][pb], ONE_BY_THREE_Q15);

          m1_param->m1_param_imag[row][col][ps][pb] = ixheaacd_mps_mult32_shr_15(

              m1_param->m1_param_imag[row][col][ps][pb], ONE_BY_THREE_Q15);

        }

        m1_param->m1_param_real[2][col][ps][pb] =

            ixheaacd_mps_mult32_shr_15(m1_param->m1_param_real[2][col][ps][pb], SQRT_TWO_Q15);

        m1_param->m1_param_imag[2][col][ps][pb] =

            ixheaacd_mps_mult32_shr_15(m1_param->m1_param_imag[2][col][ps][pb], SQRT_TWO_Q15);

        m1_param->m1_param_real[3][col][ps][pb] = m1_param->m1_param_real[0][col][ps][pb];

        m1_param->m1_param_imag[3][col][ps][pb] = m1_param->m1_param_imag[0][col][ps][pb];

        m1_param->m1_param_real[4][col][ps][pb] = m1_param->m1_param_real[1][col][ps][pb];

        m1_param->m1_param_imag[4][col][ps][pb] = m1_param->m1_param_imag[1][col][ps][pb];

        m1_param->m1_param_real[5][col][ps][pb] = 0;

        m1_param->m1_param_imag[5][col][ps][pb] = 0;

      }

    }

    for (pb = 0; pb < pstr_mps_state->num_parameter_bands; pb++) {

      m2_param->m2_decor_real[0][ps][pb] = h12[pb];

      m2_param->m2_decor_real[1][ps][pb] = h22[pb];

      m2_param->m2_decor_real[2][ps][pb] = h12[pb];

      m2_param->m2_decor_real[3][ps][pb] = h22[pb];

    }

    for (pb = 0; pb < pstr_mps_state->num_parameter_bands; pb++) {

      m2_param->m2_resid_real[0][ps][pb] = h11[pb];

      m2_param->m2_resid_real[1][ps][pb] = h21[pb];

      m2_param->m2_resid_real[2][ps][pb] = h21[pb];

      m2_param->m2_resid_real[3][ps][pb] = 32768;

    }

  }

  return;

}