decoder/ixheaacd_mps_calc_m1m2_emm.c - platform/external/libxaac - Git at Google

 /******************************************************************************
  *
  * 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 + MAX_PARAMETER_BANDS;
   h21 = h12 + MAX_PARAMETER_BANDS;
   h22 = h21 + MAX_PARAMETER_BANDS;
   dummy1 = h22 + MAX_PARAMETER_BANDS;
   dummy2 = dummy1 + MAX_PARAMETER_BANDS;

   dummy3 = (WORD16 *)pstr_mps_state->mps_scratch_mem_v + PARAMETER_BANDSX12;
   dummy4 = dummy3 + MAX_PARAMETER_BANDS;

   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;
 }
	/******************************************************************************
	*
	* 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 + MAX_PARAMETER_BANDS;
	h21 = h12 + MAX_PARAMETER_BANDS;
	h22 = h21 + MAX_PARAMETER_BANDS;
	dummy1 = h22 + MAX_PARAMETER_BANDS;
	dummy2 = dummy1 + MAX_PARAMETER_BANDS;

	dummy3 = (WORD16 *)pstr_mps_state->mps_scratch_mem_v + PARAMETER_BANDSX12;
	dummy4 = dummy3 + MAX_PARAMETER_BANDS;

	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;
	}