unplugged-system/external/libxaac/decoder/ixheaacd_aac_ec.c

458 lines
18 KiB
C
Raw Normal View History

/******************************************************************************
* *
* 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 "ixheaacd_type_def.h"
#include "ixheaacd_constants.h"
#include "ixheaacd_basic_ops32.h"
#include "ixheaacd_basic_ops16.h"
#include "ixheaacd_basic_ops40.h"
#include "ixheaacd_basic_ops.h"
#include "ixheaacd_sbr_common.h"
#include "ixheaacd_error_standards.h"
#include <ixheaacd_cnst.h>
#include "ixheaacd_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 "ixheaacd_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] = ixheaacd_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) - ixheaacd_norm32(ptr_sfb_offset[sfb] - 0);
for (; line < ptr_sfb_offset[sfb]; line++) {
accu += ixheaacd_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg;
}
ptr_sfb_energy[sfb] = ixheaacd_norm32(accu);
}
q_nrg = (sizeof(accu) << 3) -
ixheaacd_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]);
for (; line < ptr_sfb_offset[sfb + 1]; line++) {
accu += ixheaacd_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg;
}
ptr_sfb_energy[sfb] = ixheaacd_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) - ixheaacd_norm32(ptr_sfb_offset[sfb] - 0);
for (; line < ptr_sfb_offset[sfb]; line++) {
accu +=
ixheaacd_mult32(ptr_spec_coeff[line >> 3], ptr_spec_coeff[line >> 3]) >> q_nrg;
}
ptr_sfb_energy[sfb] = ixheaacd_norm32(accu);
}
q_nrg = (sizeof(accu) << 3) -
ixheaacd_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]);
for (; line < ptr_sfb_offset[sfb + 1]; line++) {
accu +=
ixheaacd_mult32(ptr_spec_coeff[line >> 3], ptr_spec_coeff[line >> 3]) >> q_nrg;
}
ptr_sfb_energy[sfb] = ixheaacd_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) - ixheaacd_norm32(ptr_sfb_offset[sfb] - 0);
for (; line < ptr_sfb_offset[sfb]; line++) {
accu += ixheaacd_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg;
}
ptr_sfb_energy[sfb] = ixheaacd_norm32(accu);
}
q_nrg = (sizeof(accu) << 3) -
ixheaacd_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]);
for (; line < ptr_sfb_offset[sfb + 1]; line++) {
accu += ixheaacd_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg;
}
ptr_sfb_energy[sfb] = ixheaacd_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) - ixheaacd_norm32(ptr_sfb_offset[sfb] - 0);
for (; line < ptr_sfb_offset[sfb] << 3; line++) {
accu += (accu +
(ixheaacd_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg)) >>
3;
}
ptr_sfb_energy[sfb] = ixheaacd_norm32(accu);
}
q_nrg = (sizeof(accu) << 3) -
ixheaacd_norm32(ptr_sfb_offset[sfb + 1] - ptr_sfb_offset[sfb]);
for (; line < ptr_sfb_offset[sfb + 1] << 3; line++) {
accu +=
(accu + (ixheaacd_mult32(ptr_spec_coeff[line], ptr_spec_coeff[line]) >> q_nrg)) >>
3;
}
ptr_sfb_energy[sfb] = ixheaacd_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 =
ixheaacd_mult32x16in32_shl(ptr_spec_coeff[line], ia_ec_interpolation_fac[fac_mod]);
ptr_spec_coeff[line] = ixheaacd_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;
}