| /****************************************************************************** |
| * * |
| * 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_type_def.h> |
| #include <ixheaacd_type_def.h> |
| #include "ixheaacd_memory_standards.h" |
| #include "ixheaacd_sbrdecsettings.h" |
| #include "ixheaacd_env_extr_part.h" |
| #include "ixheaacd_defines.h" |
| #include <ixheaacd_aac_rom.h> |
| #include "ixheaacd_common_rom.h" |
| #include <ixheaacd_sbr_rom.h> |
| #include "ixheaacd_bitbuffer.h" |
| #include "ixheaacd_pulsedata.h" |
| #include "ixheaacd_pns.h" |
| |
| #include "ixheaacd_sbr_common.h" |
| #include "ixheaacd_drc_data_struct.h" |
| #include "ixheaacd_drc_dec.h" |
| |
| #include "ixheaacd_lt_predict.h" |
| #include "ixheaacd_channelinfo.h" |
| #include "ixheaacd_channel.h" |
| #include "ixheaacd_sbrdecoder.h" |
| #include "ixheaacd_audioobjtypes.h" |
| #include "ixheaacd_latmdemux.h" |
| #include "ixheaacd_aacdec.h" |
| #include "ixheaacd_sbr_common.h" |
| |
| #include "ixheaacd_mps_polyphase.h" |
| #include "ixheaacd_config.h" |
| #include "ixheaacd_mps_dec.h" |
| #include "ixheaacd_mps_interface.h" |
| #include "ixheaacd_struct_def.h" |
| #include "ixheaacd_config.h" |
| #include "ixheaacd_bitbuffer.h" |
| #include "ixheaacd_interface.h" |
| |
| #include "ixheaacd_tns_usac.h" |
| #include "ixheaacd_cnst.h" |
| #include "ixheaacd_acelp_info.h" |
| #include "ixheaacd_sbrdecsettings.h" |
| #include "ixheaacd_info.h" |
| #include "ixheaacd_struct.h" |
| |
| #include "ixheaacd_sbrdecoder.h" |
| #include "ixheaacd_mps_polyphase.h" |
| #include "ixheaacd_sbr_const.h" |
| #include "ixheaacd_main.h" |
| #include "ixheaacd_arith_dec.h" |
| #include "ixheaacd_bit_extract.h" |
| #include "ixheaacd_create.h" |
| |
| #include "ixheaacd_func_def.h" |
| #include "ixheaacd_mps_interface.h" |
| #include "ixheaacd_mps_polyphase.h" |
| |
| #include "ixheaacd_defines.h" |
| |
| #include "ixheaacd_sbr_scale.h" |
| #include "ixheaacd_lpp_tran.h" |
| #include "ixheaacd_hybrid.h" |
| #include "ixheaacd_ps_dec.h" |
| #include "ixheaacd_env_extr.h" |
| #include "ixheaacd_env_calc.h" |
| #include "ixheaacd_qmf_dec.h" |
| |
| #include "ixheaacd_pvc_dec.h" |
| #include "ixheaacd_sbr_dec.h" |
| |
| #include "ixheaacd_error_codes.h" |
| |
| #define MAXNRSBRELEMENTS 6 |
| |
| VOID ixheaacd_allocate_sbr_scr(ia_sbr_scr_struct *sbr_scratch_struct, |
| VOID *base_scratch_ptr, VOID *output_ptr, |
| WORD total_elements, WORD ch_fac, |
| WORD32 object_type); |
| |
| WORD16 ixheaacd_applysbr( |
| ia_handle_sbr_dec_inst_struct self, |
| ia_aac_dec_sbr_bitstream_struct *p_sbr_bit_stream, WORD16 *core_sample_buf, |
| WORD16 *codec_num_channels, FLAG frame_status, FLAG down_samp_flag, |
| FLAG down_mix_flag, ia_sbr_scr_struct *sbr_scratch_struct, WORD32 ps_enable, |
| WORD32 ch_fac, WORD32 slot_element, ia_bit_buf_struct *it_bit_buff, |
| ia_drc_dec_struct *pstr_drc_dec, WORD eld_sbr_flag, WORD32 object_type); |
| |
| WORD16 ixheaacd_esbr_process(ia_usac_data_struct *usac_data, |
| ia_bit_buf_struct *it_bit_buff, |
| WORD32 stereo_config_idx, WORD16 num_channels, |
| WORD32 audio_object_type) { |
| WORD16 err_code = 0; |
| ia_aac_dec_sbr_bitstream_struct *esbr_bit_str = &usac_data->esbr_bit_str[0]; |
| ia_handle_sbr_dec_inst_struct self = usac_data->pstr_esbr_dec; |
| |
| ia_sbr_scr_struct sbr_scratch_struct; |
| ixheaacd_allocate_sbr_scr(&sbr_scratch_struct, |
| usac_data->sbr_scratch_mem_base, NULL, 2, 1, |
| audio_object_type); |
| |
| self->usac_independency_flag = usac_data->usac_independency_flg; |
| |
| self->time_sample_buf[0] = usac_data->time_sample_vector[0]; |
| self->time_sample_buf[1] = usac_data->time_sample_vector[1]; |
| self->stereo_config_idx = stereo_config_idx; |
| |
| self->sbr_mode = usac_data->sbr_mode; |
| self->aot_usac_flag = usac_data->usac_flag; |
| |
| err_code = ixheaacd_applysbr(self, esbr_bit_str, NULL, &num_channels, 1, 0, 0, |
| &sbr_scratch_struct, 0, 1, 0, it_bit_buff, NULL, |
| 0, audio_object_type); |
| |
| usac_data->sbr_mode = self->sbr_mode; |
| |
| return err_code; |
| } |
| |
| static VOID ixheaacd_sbr_ext_data_read( |
| ia_bit_buf_struct *it_bit_buff, |
| ia_aac_dec_sbr_bitstream_struct *esbr_bit_str) { |
| WORD32 count = 0; |
| WORD32 read_bits = 0; |
| WORD32 unaligned_bits = 0; |
| WORD32 cnt_bits_in; |
| |
| cnt_bits_in = it_bit_buff->cnt_bits; |
| count = (it_bit_buff->cnt_bits) >> 3; |
| if (count > 0) { |
| if ((esbr_bit_str->no_elements < MAXNRSBRELEMENTS)) { |
| esbr_bit_str->str_sbr_ele[esbr_bit_str->no_elements].extension_type = |
| EXT_SBR_DATA; |
| esbr_bit_str->str_sbr_ele[esbr_bit_str->no_elements].size_payload = count; |
| |
| read_bits = count << 3; |
| |
| unaligned_bits = (cnt_bits_in - read_bits); |
| if (unaligned_bits > 0 && unaligned_bits < 8) { |
| count++; |
| esbr_bit_str->str_sbr_ele[esbr_bit_str->no_elements].size_payload = |
| count; |
| } |
| esbr_bit_str->no_elements += 1; |
| } |
| } |
| } |
| |
| static WORD32 ixheaacd_read_ext_element( |
| UWORD32 usac_ext_element_default_length, |
| UWORD32 usac_ext_element_payload_frag, ia_bit_buf_struct *it_bit_buff, |
| ia_usac_decoder_config_struct *pstr_usac_dec_config, WORD32 elem_idx) { |
| UWORD32 usac_ext_element_present; |
| UWORD32 usac_ext_element_use_dft_length; |
| UWORD32 pay_load_length, tmp; |
| UWORD32 i; |
| usac_ext_element_present = ixheaacd_read_bits_buf(it_bit_buff, 1); |
| |
| if (usac_ext_element_present) { |
| usac_ext_element_use_dft_length = ixheaacd_read_bits_buf(it_bit_buff, 1); |
| |
| if (usac_ext_element_use_dft_length) { |
| pay_load_length = usac_ext_element_default_length; |
| } else { |
| pay_load_length = ixheaacd_read_bits_buf(it_bit_buff, 8); |
| |
| if (pay_load_length == 255) { |
| WORD32 val_add = 0; |
| val_add = ixheaacd_read_bits_buf(it_bit_buff, 16); |
| pay_load_length = (UWORD32)((WORD32)pay_load_length + val_add - 2); |
| } |
| } |
| if ((it_bit_buff->cnt_bits >> 3) < (WORD32)pay_load_length) |
| return IA_ENHAACPLUS_DEC_EXE_NONFATAL_INSUFFICIENT_INPUT_BYTES; |
| if (pay_load_length > 0) { |
| if (usac_ext_element_payload_frag) |
| tmp = ixheaacd_read_bits_buf(it_bit_buff, 2); |
| |
| if (pstr_usac_dec_config->usac_ext_ele_payload_present[elem_idx]) { |
| for (i = 0; i < pay_load_length; i++) { |
| pstr_usac_dec_config->usac_ext_gain_payload_buf |
| [i + pstr_usac_dec_config->usac_ext_gain_payload_len] = |
| ixheaacd_read_bits_buf(it_bit_buff, 8); |
| } |
| pstr_usac_dec_config->usac_ext_gain_payload_len += pay_load_length; |
| } else { |
| it_bit_buff->ptr_read_next = |
| it_bit_buff->ptr_read_next + pay_load_length; |
| it_bit_buff->cnt_bits = |
| it_bit_buff->cnt_bits - (WORD32)(pay_load_length * 8); |
| } |
| } |
| } |
| return 0; |
| } |
| |
| static VOID ixheaacd_sbr_ele_type_set( |
| ia_aac_dec_sbr_bitstream_struct *esbr_bit_str0, |
| ia_aac_dec_sbr_bitstream_struct *esbr_bit_str1, WORD32 ele_id, |
| WORD32 st_config_idx) { |
| if (ele_id == ID_USAC_SCE) { |
| esbr_bit_str0->str_sbr_ele[esbr_bit_str0->no_elements].sbr_ele_id = |
| SBR_ID_SCE; |
| esbr_bit_str1->str_sbr_ele[esbr_bit_str1->no_elements].sbr_ele_id = |
| SBR_ID_SCE; |
| } |
| if (ele_id == ID_USAC_CPE) { |
| if ((st_config_idx == 0) || (st_config_idx == 3)) { |
| esbr_bit_str0->str_sbr_ele[esbr_bit_str0->no_elements].sbr_ele_id = |
| SBR_ID_CPE; |
| esbr_bit_str1->str_sbr_ele[esbr_bit_str1->no_elements].sbr_ele_id = |
| SBR_ID_CPE; |
| } else { |
| esbr_bit_str0->str_sbr_ele[esbr_bit_str0->no_elements].sbr_ele_id = |
| SBR_ID_SCE; |
| esbr_bit_str1->str_sbr_ele[esbr_bit_str1->no_elements].sbr_ele_id = |
| SBR_ID_SCE; |
| } |
| } |
| if (ele_id == ID_USAC_LFE) { |
| esbr_bit_str0->str_sbr_ele[esbr_bit_str0->no_elements].sbr_ele_id = |
| SBR_ID_SCE; |
| esbr_bit_str1->str_sbr_ele[esbr_bit_str1->no_elements].sbr_ele_id = |
| SBR_ID_SCE; |
| |
| esbr_bit_str0->no_elements++; |
| esbr_bit_str0->str_sbr_ele[0].size_payload = 0; |
| } |
| } |
| |
| VOID ixheaacd_ms_processing(ia_usac_data_struct *pstr_usac_data) { |
| WORD32 i; |
| |
| FLOAT32 tmp, tmp1; |
| FLOAT32 ms_factor = (FLOAT32)0.7071067812; |
| for (i = 0; i < pstr_usac_data->ccfl; i++) { |
| tmp = (FLOAT32)((pstr_usac_data->time_sample_vector[0][i] + |
| pstr_usac_data->time_sample_vector[1][i]) * |
| ms_factor); |
| tmp1 = (FLOAT32)((pstr_usac_data->time_sample_vector[0][i] - |
| pstr_usac_data->time_sample_vector[1][i]) * |
| ms_factor); |
| pstr_usac_data->time_sample_vector[1][i] = tmp1; |
| pstr_usac_data->time_sample_vector[0][i] = tmp; |
| } |
| } |
| |
| WORD32 ixheaacd_usac_process(ia_dec_data_struct *pstr_dec_data, |
| WORD32 *num_out_channels, VOID *codec_handle) { |
| WORD32 ele_id = 0; |
| WORD32 err_code = 0; |
| |
| ia_aac_dec_state_struct *p_state_aac_dec = |
| (ia_aac_dec_state_struct *)codec_handle; |
| |
| ia_usac_data_struct *pstr_usac_data = &(pstr_dec_data->str_usac_data); |
| ia_bit_buf_struct *it_bit_buff = &pstr_dec_data->dec_bit_buf; |
| |
| ia_frame_data_struct *fd = &(pstr_dec_data->str_frame_data); |
| |
| ia_usac_config_struct *pstr_usac_config = |
| &(fd->str_audio_specific_config.str_usac_config); |
| ia_usac_decoder_config_struct *pstr_usac_dec_config = |
| &(fd->str_audio_specific_config.str_usac_config.str_usac_dec_config); |
| |
| WORD32 err = 0; |
| WORD16 nr_core_coder_channels = 0; |
| WORD32 ch_offset = 0; |
| |
| WORD32 elem_idx = 0; |
| WORD32 num_elements = pstr_usac_dec_config->num_elements; |
| |
| pstr_usac_dec_config->usac_ext_gain_payload_len = 0; |
| pstr_usac_data->usac_independency_flg = |
| ixheaacd_read_bits_buf(it_bit_buff, 1); |
| |
| for (elem_idx = 0; elem_idx < num_elements; elem_idx++) { |
| WORD32 stereo_config_index = |
| pstr_usac_config->str_usac_dec_config.str_usac_element_config[elem_idx] |
| .stereo_config_index; |
| |
| pstr_usac_data->esbr_bit_str[0].no_elements = 0; |
| pstr_usac_data->esbr_bit_str[1].no_elements = 0; |
| |
| switch (ele_id = pstr_usac_dec_config->usac_element_type[elem_idx]) { |
| case ID_USAC_SCE: |
| nr_core_coder_channels = 1; |
| goto core_data_extracting; |
| |
| case ID_USAC_CPE: |
| nr_core_coder_channels = (stereo_config_index == 1) ? 1 : 2; |
| if ((stereo_config_index > 1) && |
| (p_state_aac_dec->num_of_output_ch < 2)) |
| return -1; |
| goto core_data_extracting; |
| case ID_USAC_LFE: |
| nr_core_coder_channels = 1; |
| |
| core_data_extracting: |
| if (ch_offset >= MAX_NUM_CHANNELS) return -1; |
| err = ixheaacd_core_coder_data(ele_id, pstr_usac_data, elem_idx, |
| &ch_offset, it_bit_buff, |
| nr_core_coder_channels); |
| if (err != 0) return -1; |
| |
| ixheaacd_sbr_ele_type_set(&pstr_usac_data->esbr_bit_str[0], |
| &pstr_usac_data->esbr_bit_str[1], ele_id, |
| stereo_config_index); |
| |
| if (pstr_usac_data->mps_pseudo_lr[elem_idx]) |
| ixheaacd_ms_processing(pstr_usac_data); |
| |
| if (ele_id != ID_USAC_LFE) { |
| if (pstr_usac_data->sbr_ratio_idx > 0) |
| ixheaacd_sbr_ext_data_read(it_bit_buff, |
| &pstr_usac_data->esbr_bit_str[0]); |
| } |
| |
| if ((pstr_usac_data->sbr_ratio_idx > 0) && |
| (pstr_usac_data->esbr_bit_str[0].no_elements != 0)) { |
| err_code = ixheaacd_esbr_process( |
| pstr_usac_data, it_bit_buff, stereo_config_index, |
| nr_core_coder_channels, |
| pstr_dec_data->str_usac_data.audio_object_type); |
| if (err_code < 0) return err_code; |
| } |
| |
| if (stereo_config_index > 0) { |
| FLOAT32 **ptr_inp[2 * 2]; |
| WORD32 ch; |
| |
| *num_out_channels = p_state_aac_dec->mps_dec_handle.out_ch_count; |
| |
| ixheaacd_mps_frame_parsing(&p_state_aac_dec->mps_dec_handle, |
| pstr_usac_data->usac_independency_flg, |
| it_bit_buff); |
| |
| for (ch = 0; ch < nr_core_coder_channels; ch++) { |
| ptr_inp[2 * ch] = |
| pstr_usac_data->pstr_esbr_dec->pstr_sbr_channel[ch] |
| ->str_sbr_dec.pp_qmf_buf_real; |
| ptr_inp[2 * ch + 1] = |
| pstr_usac_data->pstr_esbr_dec->pstr_sbr_channel[ch] |
| ->str_sbr_dec.pp_qmf_buf_imag; |
| p_state_aac_dec->mps_dec_handle.p_sbr_dec[ch] = |
| (VOID *)(&pstr_usac_data->pstr_esbr_dec->pstr_sbr_channel[ch] |
| ->str_sbr_dec); |
| p_state_aac_dec->mps_dec_handle.p_sbr_header[ch] = |
| (VOID *)(pstr_usac_data->pstr_esbr_dec->pstr_sbr_header[ch]); |
| p_state_aac_dec->mps_dec_handle.p_sbr_frame[ch] = |
| (VOID *)(pstr_usac_data->pstr_esbr_dec->frame_buffer[ch]); |
| } |
| if (nr_core_coder_channels == 1) { |
| p_state_aac_dec->mps_dec_handle.p_sbr_dec[1] = |
| (VOID *)(&pstr_usac_data->pstr_esbr_dec->pstr_sbr_channel[1] |
| ->str_sbr_dec); |
| p_state_aac_dec->mps_dec_handle.p_sbr_header[1] = |
| (VOID *)(pstr_usac_data->pstr_esbr_dec->pstr_sbr_header[1]); |
| p_state_aac_dec->mps_dec_handle.p_sbr_frame[1] = |
| (VOID *)(pstr_usac_data->pstr_esbr_dec->frame_buffer[1]); |
| } |
| |
| err = ixheaacd_mps_apply(&p_state_aac_dec->mps_dec_handle, ptr_inp, |
| pstr_usac_data->time_sample_vector); |
| if (err) return err; |
| } |
| |
| ch_offset += nr_core_coder_channels; |
| break; |
| |
| case ID_USAC_EXT: { |
| ia_usac_dec_element_config_struct *pusac_element_config = |
| &pstr_usac_dec_config->str_usac_element_config[elem_idx]; |
| err = ixheaacd_read_ext_element( |
| pusac_element_config->usac_ext_eleme_def_len, |
| pusac_element_config->usac_ext_elem_pld_frag, it_bit_buff, |
| pstr_usac_dec_config, elem_idx); |
| if (err != 0) return err; |
| |
| break; |
| } |
| |
| default: |
| |
| return -1; |
| |
| break; |
| } |
| } |
| |
| return 0; |
| } |
