| /****************************************************************************** |
| * * |
| * 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 <stdio.h> |
| #include <string.h> |
| #include <math.h> |
| #include "ixheaac_type_def.h" |
| #include "ixheaac_constants.h" |
| #include "ixheaac_basic_ops32.h" |
| #include "ixheaac_basic_ops16.h" |
| #include "ixheaac_basic_ops40.h" |
| #include "ixheaac_basic_ops.h" |
| #include "ixheaacd_sbr_common.h" |
| #include "ixheaac_error_standards.h" |
| |
| #include <ixheaacd_cnst.h> |
| #include "ixheaac_constants.h" |
| #include "ixheaacd_intrinsics.h" |
| #include "ixheaacd_common_rom.h" |
| #include "ixheaacd_sbrdecsettings.h" |
| #include "ixheaacd_bitbuffer.h" |
| #include "ixheaacd_defines.h" |
| |
| #include "ixheaacd_pns.h" |
| |
| #include <ixheaacd_aac_rom.h> |
| #include "ixheaacd_pulsedata.h" |
| |
| #include "ixheaacd_drc_data_struct.h" |
| |
| #include "ixheaacd_interface.h" |
| #include "ixheaacd_info.h" |
| #include "ixheaacd_lt_predict.h" |
| #include "ixheaacd_ec_defines.h" |
| #include "ixheaacd_ec_rom.h" |
| #include "ixheaacd_ec_struct_def.h" |
| #include "ixheaacd_channelinfo.h" |
| #include "ixheaac_error_standards.h" |
| |
| #include "ixheaacd_aac_rom.h" |
| |
| static WORD32 ixheaacd_aac_ec_get_win_seq(WORD32 prev_win_seq) { |
| WORD32 new_win_seq = ONLY_LONG_SEQUENCE; |
| |
| if (prev_win_seq == LONG_START_SEQUENCE || prev_win_seq == EIGHT_SHORT_SEQUENCE) { |
| new_win_seq = LONG_STOP_SEQUENCE; |
| } |
| |
| return new_win_seq; |
| } |
| |
| static VOID ixheaacd_aac_ec_flip_spec_sign(WORD32 *ptr_spec_coeff, WORD32 num_samples) { |
| WORD32 idx; |
| WORD32 random_value; |
| |
| for (idx = 0; idx < num_samples; idx++) { |
| random_value = ptr_spec_coeff[idx] ^ idx; |
| if ((random_value & 1) == 0) { |
| ptr_spec_coeff[idx] = ixheaac_negate32_sat(ptr_spec_coeff[idx]); |
| } |
| } |
| } |
| |
| static VOID ixheaacd_aac_ec_store(ia_ec_state_str *pstr_ec_state, |
| ia_aac_dec_channel_info_struct *pstr_aac_dec_channel_info, |
| ia_ics_info_struct *pstr_ics_info) { |
| WORD32 *ptr_spec_coeff = pstr_aac_dec_channel_info->ptr_spec_coeff; |
| WORD16 *ptr_spec_scale = pstr_aac_dec_channel_info->ptr_scale_factor; |
| UWORD8 win_shape = pstr_ec_state->win_shape; |
| WORD32 win_seq = pstr_ec_state->win_seq; |
| WORD16 q_spec_scale[MAX_SPEC_SCALE_LEN]; |
| WORD32 *ptr_temp_spec_coeff = &pstr_ec_state->str_ec_scratch.spec_coeff[0]; |
| |
| memcpy(q_spec_scale, pstr_ec_state->q_spec_coeff, sizeof(q_spec_scale)); |
| |
| pstr_ec_state->win_seq = pstr_ics_info->window_sequence; |
| pstr_ec_state->win_shape = (UWORD8)pstr_ics_info->window_shape; |
| pstr_ec_state->prev_win_group_len = |
| *(pstr_ics_info->window_group_length + pstr_ics_info->num_window_groups - 1); |
| |
| memcpy(pstr_ec_state->q_spec_coeff, ptr_spec_scale, sizeof(pstr_ec_state->q_spec_coeff)); |
| |
| memcpy(ptr_temp_spec_coeff, ptr_spec_coeff, LEN_SUPERFRAME * sizeof(ptr_temp_spec_coeff[0])); |
| memcpy(ptr_spec_coeff, pstr_ec_state->spectral_coeff, |
| LEN_SUPERFRAME * sizeof(ptr_spec_coeff[0])); |
| memcpy(pstr_ec_state->spectral_coeff, ptr_temp_spec_coeff, |
| sizeof(pstr_ec_state->spectral_coeff)); |
| pstr_ics_info->window_sequence = win_seq; |
| pstr_ics_info->window_shape = win_shape; |
| |
| memcpy(ptr_spec_scale, q_spec_scale, MAX_SPEC_SCALE_LEN * sizeof(ptr_spec_scale[0])); |
| } |
| |
| static VOID ixheaacd_aac_ec_calc_sfb_nrg(WORD32 *ptr_spec_coeff, |
| const ia_usac_samp_rate_info *pstr_samp_rate_info, |
| const WORD32 win_seq, WORD32 win_trans, |
| WORD32 *ptr_sfb_energy) { |
| const WORD16 *ptr_sfb_offset; |
| WORD32 line = 0, sfb, total_scale_factor_bands = 0; |
| |
| switch (win_seq) { |
| case EIGHT_SHORT_SEQUENCE: |
| |
| if (win_trans == NO_TRANSITION) { |
| total_scale_factor_bands = pstr_samp_rate_info->num_sfb_128 - 1; |
| ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_128; |
| |
| for (sfb = 0; sfb < total_scale_factor_bands; sfb++) { |
| WORD32 accu = 1; |
| WORD32 q_nrg; |
| if (sfb == 0) { |
| q_nrg = (sizeof(accu) << 3) - ixheaac_norm32(ptr_sfb_offset[sfb] - 0); |
| for (; line < ptr_sfb_offset[sfb]; line++) { |
| accu += ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg; |
| } |
| ptr_sfb_energy[sfb] = ixheaac_norm32(accu); |
| } |
| q_nrg = (sizeof(accu) << 3) - |
| ixheaac_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]); |
| for (; line < ptr_sfb_offset[sfb + 1]; line++) { |
| accu += ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg; |
| } |
| ptr_sfb_energy[sfb] = ixheaac_norm32(accu); |
| } |
| } else { |
| total_scale_factor_bands = pstr_samp_rate_info->num_sfb_1024 - 1; |
| ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_1024; |
| |
| for (sfb = 0; sfb < total_scale_factor_bands; sfb++) { |
| WORD32 accu = 1; |
| WORD32 q_nrg; |
| if (sfb == 0) { |
| q_nrg = (sizeof(accu) << 3) - ixheaac_norm32(ptr_sfb_offset[sfb] - 0); |
| for (; line < ptr_sfb_offset[sfb]; line++) { |
| accu += |
| ixheaac_mult32(ptr_spec_coeff[line >> 3], ptr_spec_coeff[line >> 3]) >> q_nrg; |
| } |
| ptr_sfb_energy[sfb] = ixheaac_norm32(accu); |
| } |
| q_nrg = (sizeof(accu) << 3) - |
| ixheaac_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]); |
| for (; line < ptr_sfb_offset[sfb + 1]; line++) { |
| accu += |
| ixheaac_mult32(ptr_spec_coeff[line >> 3], ptr_spec_coeff[line >> 3]) >> q_nrg; |
| } |
| ptr_sfb_energy[sfb] = ixheaac_norm32(accu); |
| } |
| } |
| break; |
| |
| case ONLY_LONG_SEQUENCE: |
| case LONG_START_SEQUENCE: |
| case LONG_STOP_SEQUENCE: |
| |
| if (win_trans == NO_TRANSITION) { |
| total_scale_factor_bands = pstr_samp_rate_info->num_sfb_1024 - 1; |
| ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_1024; |
| |
| for (sfb = 0; sfb < total_scale_factor_bands; sfb++) { |
| WORD32 accu = 1; |
| WORD32 q_nrg; |
| if (sfb == 0) { |
| q_nrg = (sizeof(accu) << 3) - ixheaac_norm32(ptr_sfb_offset[sfb] - 0); |
| for (; line < ptr_sfb_offset[sfb]; line++) { |
| accu += ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg; |
| } |
| ptr_sfb_energy[sfb] = ixheaac_norm32(accu); |
| } |
| q_nrg = (sizeof(accu) << 3) - |
| ixheaac_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]); |
| for (; line < ptr_sfb_offset[sfb + 1]; line++) { |
| accu += ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg; |
| } |
| ptr_sfb_energy[sfb] = ixheaac_norm32(accu); |
| } |
| } else { |
| total_scale_factor_bands = pstr_samp_rate_info->num_sfb_128 - 1; |
| ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_128; |
| |
| for (sfb = 0; sfb < total_scale_factor_bands; sfb++) { |
| WORD32 accu = 1; |
| WORD32 q_nrg; |
| if (sfb == 0) { |
| q_nrg = (sizeof(accu) << 3) - ixheaac_norm32(ptr_sfb_offset[sfb] - 0); |
| for (; line < ptr_sfb_offset[sfb] << 3; line++) { |
| accu += (accu + |
| (ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg)) >> |
| 3; |
| } |
| ptr_sfb_energy[sfb] = ixheaac_norm32(accu); |
| } |
| q_nrg = (sizeof(accu) << 3) - |
| ixheaac_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]); |
| for (; line < ptr_sfb_offset[sfb + 1] << 3; line++) { |
| accu += |
| (accu + (ixheaac_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg)) >> |
| 3; |
| } |
| ptr_sfb_energy[sfb] = ixheaac_norm32(accu); |
| } |
| } |
| break; |
| } |
| } |
| |
| static VOID ixheaacd_aac_ec_interpolate(WORD32 *ptr_spec_coeff, WORD16 *ptr_spec_scale_prev, |
| WORD16 *ptr_spec_scale_act, WORD16 *ptr_spec_scale_out, |
| WORD32 *nrg_prev, WORD32 *nrg_act, WORD32 num_sfb, |
| const WORD16 *ptr_sfb_offset) { |
| WORD32 sfb, line = 0; |
| WORD32 fac_shift; |
| WORD32 fac_mod; |
| |
| for (sfb = 0; sfb < num_sfb; sfb++) { |
| fac_shift = |
| nrg_prev[sfb] - nrg_act[sfb] + ((*ptr_spec_scale_act - *ptr_spec_scale_prev) << 1); |
| fac_mod = fac_shift & 3; |
| fac_shift = (fac_shift >> 2) + 1; |
| fac_shift += *ptr_spec_scale_prev - max(*ptr_spec_scale_prev, *ptr_spec_scale_act); |
| fac_shift = max(min(fac_shift, INT_BITS - 1), -(INT_BITS - 1)); |
| |
| for (; line < ptr_sfb_offset[sfb]; line++) { |
| WORD32 accu = |
| ixheaac_mult32x16in32_shl(ptr_spec_coeff[line], ia_ec_interpolation_fac[fac_mod]); |
| ptr_spec_coeff[line] = ixheaac_shl32_dir_sat(accu, fac_shift); |
| } |
| } |
| *ptr_spec_scale_out = max(*ptr_spec_scale_prev, *ptr_spec_scale_act); |
| } |
| |
| static VOID ixheaacd_aac_ec_state(ia_ec_state_str *pstr_ec_state, WORD32 frame_status) { |
| WORD32 ec_state_val = (pstr_ec_state->prev_frame_ok[0] << 2) + |
| (pstr_ec_state->prev_frame_ok[1] << 1) + (frame_status); |
| |
| switch (ec_state_val) { |
| case 0: |
| case 4: |
| if (pstr_ec_state->fade_idx < MAX_FADE_FRAMES) { |
| pstr_ec_state->fade_idx++; |
| } |
| pstr_ec_state->conceal_state = FRAME_CONCEAL_SINGLE; |
| break; |
| case 1: |
| case 2: |
| if (pstr_ec_state->fade_idx > 0) { |
| pstr_ec_state->fade_idx--; |
| } |
| pstr_ec_state->conceal_state = FRAME_FADE; |
| break; |
| case 5: |
| if (pstr_ec_state->fade_idx > 0) { |
| pstr_ec_state->fade_idx--; |
| } |
| pstr_ec_state->conceal_state = FRAME_OKAY; |
| break; |
| break; |
| case 3: |
| case 6: |
| case 7: |
| if (pstr_ec_state->fade_idx > 0) { |
| pstr_ec_state->fade_idx--; |
| } |
| pstr_ec_state->conceal_state = FRAME_OKAY; |
| break; |
| default: |
| pstr_ec_state->conceal_state = FRAME_OKAY; |
| } |
| if (pstr_ec_state->fade_idx > MAX_FADE_FRAMES) { |
| pstr_ec_state->fade_idx = MAX_FADE_FRAMES; |
| } |
| if (pstr_ec_state->fade_idx == MAX_FADE_FRAMES) { |
| pstr_ec_state->conceal_state = FRAME_MUTE; |
| } |
| if (pstr_ec_state->fade_idx < 0) { |
| pstr_ec_state->fade_idx = 0; |
| } |
| } |
| |
| static VOID ixheaacd_aac_ec_interpolate_frame( |
| ia_ec_state_str *pstr_ec_state, ia_aac_dec_channel_info_struct *pstr_aac_dec_channel_info, |
| const ia_usac_samp_rate_info *pstr_samp_rate_info, const WORD32 num_samples, |
| const WORD32 frame_status, ia_ics_info_struct *pstr_ics_info) { |
| WORD32 *ptr_spec_coeff = pstr_aac_dec_channel_info->ptr_spec_coeff; |
| WORD16 *ptr_spec_scale = pstr_aac_dec_channel_info->ptr_scale_factor; |
| |
| WORD32 sfb_nrg_prev[WIN_LEN_64]; |
| WORD32 sfb_nrg_act[WIN_LEN_64]; |
| |
| WORD32 idx; |
| |
| memset(sfb_nrg_prev, 0, sizeof(sfb_nrg_prev)); |
| memset(sfb_nrg_act, 0, sizeof(sfb_nrg_act)); |
| |
| if (!frame_status) { |
| pstr_ics_info->window_shape = pstr_ec_state->win_shape; |
| pstr_ics_info->window_sequence = pstr_ec_state->win_seq; |
| |
| for (idx = 0; idx < num_samples; idx++) { |
| ptr_spec_coeff[idx] = pstr_ec_state->spectral_coeff[idx]; |
| } |
| |
| memcpy(ptr_spec_scale, pstr_ec_state->q_spec_coeff, 8 * sizeof(ptr_spec_scale[0])); |
| } |
| |
| if (!pstr_ec_state->prev_frame_ok[1]) { |
| if (frame_status && pstr_ec_state->prev_frame_ok[0]) { |
| if (pstr_ics_info->window_sequence == EIGHT_SHORT_SEQUENCE) { |
| WORD32 window; |
| if (pstr_ec_state->win_seq == EIGHT_SHORT_SEQUENCE) { |
| WORD32 total_scale_factor_bands = pstr_samp_rate_info->num_sfb_128 - 1; |
| const WORD16 *ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_128; |
| pstr_ics_info->window_shape = 1; |
| pstr_ics_info->window_sequence = EIGHT_SHORT_SEQUENCE; |
| |
| for (window = 0; window < 8; window++) { |
| ixheaacd_aac_ec_calc_sfb_nrg(&ptr_spec_coeff[window * (num_samples / 8)], |
| pstr_samp_rate_info, EIGHT_SHORT_SEQUENCE, NO_TRANSITION, |
| sfb_nrg_prev); |
| |
| ixheaacd_aac_ec_calc_sfb_nrg( |
| &pstr_ec_state->spectral_coeff[window * (num_samples / 8)], pstr_samp_rate_info, |
| EIGHT_SHORT_SEQUENCE, NO_TRANSITION, sfb_nrg_act); |
| |
| ixheaacd_aac_ec_interpolate( |
| &ptr_spec_coeff[window * (num_samples / 8)], &ptr_spec_scale[window], |
| &pstr_ec_state->q_spec_coeff[window], &ptr_spec_scale[window], sfb_nrg_prev, |
| sfb_nrg_act, total_scale_factor_bands, ptr_sfb_offset); |
| } |
| } else { |
| WORD32 total_scale_factor_bands = pstr_samp_rate_info->num_sfb_1024 - 1; |
| const WORD16 *ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_1024; |
| WORD16 spec_scale_out; |
| |
| ixheaacd_aac_ec_calc_sfb_nrg(&ptr_spec_coeff[num_samples - (num_samples / 8)], |
| pstr_samp_rate_info, EIGHT_SHORT_SEQUENCE, |
| TRANS_SHORT_LONG, sfb_nrg_act); |
| |
| ixheaacd_aac_ec_calc_sfb_nrg(pstr_ec_state->spectral_coeff, pstr_samp_rate_info, |
| ONLY_LONG_SEQUENCE, NO_TRANSITION, sfb_nrg_prev); |
| |
| pstr_ics_info->window_shape = 0; |
| pstr_ics_info->window_sequence = LONG_STOP_SEQUENCE; |
| |
| for (idx = 0; idx < num_samples; idx++) { |
| ptr_spec_coeff[idx] = pstr_ec_state->spectral_coeff[idx]; |
| } |
| |
| for (idx = 0; idx < 8; idx++) { |
| if (ptr_spec_scale[idx] > ptr_spec_scale[0]) { |
| ptr_spec_scale[0] = ptr_spec_scale[idx]; |
| } |
| } |
| |
| ixheaacd_aac_ec_interpolate(ptr_spec_coeff, &pstr_ec_state->q_spec_coeff[0], |
| &ptr_spec_scale[0], &spec_scale_out, sfb_nrg_prev, |
| sfb_nrg_act, total_scale_factor_bands, ptr_sfb_offset); |
| |
| ptr_spec_scale[0] = spec_scale_out; |
| } |
| } else { |
| WORD32 total_scale_factor_bands = pstr_samp_rate_info->num_sfb_1024 - 1; |
| const WORD16 *ptr_sfb_offset = pstr_samp_rate_info->ptr_sfb_1024; |
| WORD16 spec_scale_act = pstr_ec_state->q_spec_coeff[0]; |
| |
| ixheaacd_aac_ec_calc_sfb_nrg(ptr_spec_coeff, pstr_samp_rate_info, ONLY_LONG_SEQUENCE, |
| NO_TRANSITION, sfb_nrg_prev); |
| |
| if (pstr_ec_state->win_seq == EIGHT_SHORT_SEQUENCE) { |
| pstr_ics_info->window_shape = 1; |
| pstr_ics_info->window_sequence = LONG_START_SEQUENCE; |
| |
| for (idx = 1; idx < 8; idx++) { |
| if (pstr_ec_state->q_spec_coeff[idx] > spec_scale_act) { |
| spec_scale_act = pstr_ec_state->q_spec_coeff[idx]; |
| } |
| } |
| |
| ixheaacd_aac_ec_calc_sfb_nrg(pstr_ec_state->spectral_coeff, pstr_samp_rate_info, |
| EIGHT_SHORT_SEQUENCE, TRANS_SHORT_LONG, sfb_nrg_act); |
| } else { |
| pstr_ics_info->window_shape = 0; |
| pstr_ics_info->window_sequence = ONLY_LONG_SEQUENCE; |
| |
| ixheaacd_aac_ec_calc_sfb_nrg(pstr_ec_state->spectral_coeff, pstr_samp_rate_info, |
| ONLY_LONG_SEQUENCE, NO_TRANSITION, sfb_nrg_act); |
| } |
| |
| ixheaacd_aac_ec_interpolate(ptr_spec_coeff, &ptr_spec_scale[0], &spec_scale_act, |
| &ptr_spec_scale[0], sfb_nrg_prev, sfb_nrg_act, |
| total_scale_factor_bands, ptr_sfb_offset); |
| } |
| } |
| |
| ixheaacd_aac_ec_flip_spec_sign(ptr_spec_coeff, num_samples); |
| } |
| |
| if (FRAME_MUTE == pstr_ec_state->conceal_state) { |
| pstr_ics_info->window_shape = pstr_ec_state->win_shape; |
| pstr_ics_info->window_sequence = ixheaacd_aac_ec_get_win_seq(pstr_ec_state->win_seq); |
| pstr_ec_state->win_seq = pstr_ics_info->window_sequence; |
| memset(ptr_spec_coeff, 0, num_samples * sizeof(ptr_spec_coeff[0])); |
| } |
| } |
| |
| VOID ixheaacd_aac_ec_init(ia_ec_state_str *pstr_ec_state) { |
| pstr_ec_state->win_shape = CONCEAL_NOT_DEFINED; |
| pstr_ec_state->win_seq = ONLY_LONG_SEQUENCE; |
| pstr_ec_state->prev_win_group_len = 1; |
| |
| pstr_ec_state->conceal_state = FRAME_OKAY; |
| |
| memset(pstr_ec_state->spectral_coeff, 0, sizeof(pstr_ec_state->spectral_coeff)); |
| memset(pstr_ec_state->q_spec_coeff, 0, sizeof(pstr_ec_state->q_spec_coeff)); |
| |
| pstr_ec_state->prev_frame_ok[0] = 1; |
| pstr_ec_state->prev_frame_ok[1] = 1; |
| pstr_ec_state->fade_idx = 0; |
| } |
| |
| VOID ixheaacd_aac_apply_ec(ia_ec_state_str *pstr_ec_state, |
| ia_aac_dec_channel_info_struct *pstr_aac_dec_channel_info, |
| const ia_usac_samp_rate_info *pstr_samp_rate_info, |
| const WORD32 num_samples, ia_ics_info_struct *pstr_ics_info, |
| const WORD32 frame_status) { |
| if (pstr_ec_state->win_shape == CONCEAL_NOT_DEFINED) { |
| pstr_ec_state->win_shape = (UWORD8)pstr_ics_info->window_shape; |
| } |
| |
| if (frame_status && pstr_ec_state->prev_frame_ok[1]) { |
| ixheaacd_aac_ec_store(pstr_ec_state, pstr_aac_dec_channel_info, pstr_ics_info); |
| } |
| |
| ixheaacd_aac_ec_state(pstr_ec_state, frame_status); |
| |
| ixheaacd_aac_ec_interpolate_frame(pstr_ec_state, pstr_aac_dec_channel_info, pstr_samp_rate_info, |
| num_samples, frame_status, pstr_ics_info); |
| |
| pstr_ec_state->prev_frame_ok[0] = pstr_ec_state->prev_frame_ok[1]; |
| pstr_ec_state->prev_frame_ok[1] = frame_status; |
| } |
