aosp12/external/libxaac/decoder/ixheaacd_lt_predict.c

524 lines
18 KiB
C

/******************************************************************************
* *
* 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 <stdlib.h>
#include <stdio.h>
#include <string.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_defines.h"
#include "ixheaacd_aac_rom.h"
#include "ixheaacd_audioobjtypes.h"
#include "ixheaacd_bitbuffer.h"
#include "ixheaacd_pulsedata.h"
#include "ixheaacd_pns.h"
#include "ixheaacd_lt_predict.h"
#include "ixheaacd_channelinfo.h"
#include "ixheaacd_cnst.h"
#include "ixheaacd_tns.h"
#include "ixheaacd_aac_imdct.h"
static const WORD32 ixheaacd_codebook_Q30[8] = {
612922971, 747985734, 872956397, 978505219,
1057528322, 1146642451, 1282693056, 1470524861};
#define SHIFT_VAL 8
#define SHIFT_VAL1 (15 - SHIFT_VAL)
VOID ixheaacd_lt_prediction(
ia_aac_dec_channel_info_struct *ptr_aac_dec_channel_info, ltp_info *ltp,
WORD32 *spec, ia_aac_dec_tables_struct *aac_tables_ptr,
UWORD16 win_shape_prev, UWORD32 sr_index, UWORD32 object_type,
UWORD32 frame_len, WORD32 *in_data, WORD32 *out_data) {
ia_ics_info_struct *ptr_ics_info = &ptr_aac_dec_channel_info->str_ics_info;
WORD16 *lt_pred_stat = ptr_aac_dec_channel_info->ltp_buf;
UWORD16 win_shape = ptr_aac_dec_channel_info->str_ics_info.window_shape;
WORD16 sfb;
WORD16 bin, i, num_samples;
const WORD8 *swb_offset = aac_tables_ptr->scale_factor_bands_long[sr_index];
WORD32 *ptr_spec = &spec[0];
WORD32 *ptr_x_est = &out_data[0];
if (512 == ptr_ics_info->frame_length) {
swb_offset = aac_tables_ptr->scale_fac_bands_512[sr_index];
} else if (480 == ptr_ics_info->frame_length) {
swb_offset = aac_tables_ptr->scale_fac_bands_480[sr_index];
}
if (ptr_ics_info->window_sequence != EIGHT_SHORT_SEQUENCE) {
if (ltp->data_present) {
num_samples = frame_len << 1;
for (i = 0; i < num_samples; i++) {
in_data[i] =
ixheaacd_shr32(ixheaacd_mult32x16in32_shl_sat(
ixheaacd_codebook_Q30[ltp->coef],
lt_pred_stat[num_samples + i - ltp->lag]),
SHIFT_VAL);
}
ixheaacd_filter_bank_ltp(aac_tables_ptr, ptr_ics_info->window_sequence,
win_shape, win_shape_prev, in_data, out_data,
object_type, frame_len);
if (ptr_aac_dec_channel_info->str_tns_info.tns_data_present == 1)
ixheaacd_aac_tns_process(ptr_aac_dec_channel_info, 1, aac_tables_ptr,
object_type, 0, out_data);
for (sfb = 0; sfb < ltp->last_band; sfb++) {
WORD8 sfb_width = swb_offset[sfb];
if (ltp->long_used[sfb]) {
for (bin = sfb_width - 1; bin >= 0; bin--) {
WORD32 temp = *ptr_spec;
temp = ixheaacd_add32_sat(temp,
ixheaacd_shr32(*ptr_x_est++, SHIFT_VAL1));
*ptr_spec++ = temp;
}
} else {
ptr_spec += sfb_width;
ptr_x_est += sfb_width;
}
}
}
}
}
VOID ixheaacd_filter_bank_ltp(ia_aac_dec_tables_struct *aac_tables_ptr,
WORD16 window_sequence, WORD16 window_shape,
WORD16 window_shape_prev, WORD32 *in_data,
WORD32 *out_mdct, UWORD32 object_type,
UWORD32 frame_len) {
WORD32 i;
const WORD16 *window_long = NULL;
const WORD16 *window_long_prev = NULL;
const WORD16 *window_short = NULL;
const WORD16 *window_short_prev = NULL;
UWORD16 nlong = frame_len;
UWORD16 nlong2 = frame_len << 1;
UWORD16 nshort = frame_len / 8;
UWORD16 nflat_ls = (nlong - nshort) / 2;
WORD32 imdct_scale = 0;
WORD32 expo = 0;
if (object_type == AOT_ER_AAC_LD) {
if (!window_shape) {
if (512 == frame_len) {
window_long =
(WORD16 *)aac_tables_ptr->pstr_imdct_tables->window_sine_512;
} else {
window_long =
(WORD16 *)aac_tables_ptr->pstr_imdct_tables->window_sine_480;
}
} else {
if (512 == frame_len) {
window_long =
(WORD16 *)aac_tables_ptr->pstr_imdct_tables->low_overlap_win;
} else {
window_long =
(WORD16 *)aac_tables_ptr->pstr_imdct_tables->low_overlap_win_480;
}
}
if (!window_shape_prev) {
if (512 == frame_len) {
window_long_prev =
(WORD16 *)aac_tables_ptr->pstr_imdct_tables->window_sine_512;
} else {
window_long_prev =
(WORD16 *)aac_tables_ptr->pstr_imdct_tables->window_sine_480;
}
} else {
if (512 == frame_len) {
window_long_prev =
(WORD16 *)aac_tables_ptr->pstr_imdct_tables->low_overlap_win;
} else {
window_long_prev =
(WORD16 *)aac_tables_ptr->pstr_imdct_tables->low_overlap_win_480;
}
}
if (!window_shape)
window_short = aac_tables_ptr->pstr_imdct_tables->only_short_window_sine;
else
window_short = aac_tables_ptr->pstr_imdct_tables->only_short_window_kbd;
if (!window_shape_prev)
window_short_prev =
aac_tables_ptr->pstr_imdct_tables->only_short_window_sine;
else
window_short_prev =
aac_tables_ptr->pstr_imdct_tables->only_short_window_kbd;
} else {
if (!window_shape)
window_long = aac_tables_ptr->pstr_imdct_tables->only_long_window_sine;
else
window_long = aac_tables_ptr->pstr_imdct_tables->only_long_window_kbd;
if (!window_shape_prev)
window_long_prev =
aac_tables_ptr->pstr_imdct_tables->only_long_window_sine;
else
window_long_prev =
aac_tables_ptr->pstr_imdct_tables->only_long_window_kbd;
if (!window_shape)
window_short = aac_tables_ptr->pstr_imdct_tables->only_short_window_sine;
else
window_short = aac_tables_ptr->pstr_imdct_tables->only_short_window_kbd;
if (!window_shape_prev)
window_short_prev =
aac_tables_ptr->pstr_imdct_tables->only_short_window_sine;
else
window_short_prev =
aac_tables_ptr->pstr_imdct_tables->only_short_window_kbd;
}
switch (window_sequence) {
case ONLY_LONG_SEQUENCE:
if ((512 != nlong) && (480 != nlong)) {
for (i = 0; i<nlong>> 1; i++) {
in_data[i] =
ixheaacd_mult32x16in32_shl(in_data[i], window_long_prev[2 * i]);
in_data[i + nlong] = ixheaacd_mult32x16in32_shl(
in_data[i + nlong], window_long[2 * i + 1]);
}
for (i = 0; i<nlong>> 1; i++) {
in_data[i + (nlong >> 1)] = ixheaacd_mult32x16in32_shl(
in_data[i + (nlong >> 1)], window_long_prev[nlong - 1 - 2 * i]);
in_data[i + nlong + (nlong >> 1)] =
ixheaacd_mult32x16in32_shl(in_data[i + nlong + (nlong >> 1)],
window_long[nlong - 1 - 2 * i - 1]);
}
} else {
WORD32 *win1, *win2, *win3;
WORD32 *ptr_in1, *ptr_in2;
win1 = (WORD32 *)window_long_prev;
win2 = (WORD32 *)window_long;
ptr_in1 = &in_data[0];
ptr_in2 = &in_data[nlong];
win3 = win2 + nlong - 1;
for (i = nlong - 1; i >= 0; i--) {
WORD32 temp1 = ixheaacd_mult32_shl(*ptr_in1, *win1++);
WORD32 temp2 = ixheaacd_mult32_shl(*ptr_in2, win2[i]);
*ptr_in1++ = temp1;
*ptr_in2++ = temp2;
}
}
for (i = 0; i < nlong / 2; i++) {
out_mdct[nlong / 2 + i] =
ixheaacd_sub32(in_data[i], in_data[nlong - 1 - i]);
out_mdct[i] = (-ixheaacd_add32(in_data[nlong + i + nlong / 2],
in_data[nlong2 - nlong / 2 - 1 - i]));
}
if (512 == nlong || (480 == nlong)) {
if (512 == nlong)
ixheaacd_inverse_transform_512(
out_mdct, in_data, &imdct_scale,
aac_tables_ptr->pstr_imdct_tables->cosine_array_1024,
aac_tables_ptr->pstr_imdct_tables, object_type);
else
ixheaacd_mdct_480_ld(out_mdct, in_data, &imdct_scale, 1,
aac_tables_ptr->pstr_imdct_tables, object_type);
imdct_scale += 1;
if (imdct_scale > 0) {
WORD32 *ptr_out_mdct = &out_mdct[0];
for (i = nlong - 1; i >= 0; i -= 4) {
*ptr_out_mdct = ixheaacd_shl32(*ptr_out_mdct, imdct_scale);
ptr_out_mdct++;
*ptr_out_mdct = ixheaacd_shl32(*ptr_out_mdct, imdct_scale);
ptr_out_mdct++;
*ptr_out_mdct = ixheaacd_shl32(*ptr_out_mdct, imdct_scale);
ptr_out_mdct++;
*ptr_out_mdct = ixheaacd_shl32(*ptr_out_mdct, imdct_scale);
ptr_out_mdct++;
}
} else if (imdct_scale < 0) {
WORD32 *ptr_out_mdct = &out_mdct[0];
imdct_scale = -imdct_scale;
for (i = nlong - 1; i >= 0; i -= 4) {
*ptr_out_mdct = ixheaacd_shr32(*ptr_out_mdct, imdct_scale);
ptr_out_mdct++;
*ptr_out_mdct = ixheaacd_shr32(*ptr_out_mdct, imdct_scale);
ptr_out_mdct++;
*ptr_out_mdct = ixheaacd_shr32(*ptr_out_mdct, imdct_scale);
ptr_out_mdct++;
*ptr_out_mdct = ixheaacd_shr32(*ptr_out_mdct, imdct_scale);
ptr_out_mdct++;
}
}
}
else if (1024 == nlong) {
expo = ixheaacd_calc_max_spectral_line_dec(out_mdct, 1024) - 1;
expo = 8 - expo;
imdct_scale = ixheaacd_inverse_transform(
out_mdct, in_data, aac_tables_ptr->pstr_imdct_tables, expo, 1024);
ixheaacd_post_twiddle_dec(in_data, out_mdct,
aac_tables_ptr->pstr_imdct_tables, 1024);
imdct_scale += 1;
for (i = 0; i < nlong; i++) {
out_mdct[i] = ixheaacd_shl32_dir(in_data[i], imdct_scale);
}
}
break;
case LONG_START_SEQUENCE:
for (i = 0; i<nlong>> 1; i++)
in_data[i] =
ixheaacd_mult32x16in32_shl(in_data[i], window_long_prev[2 * i]);
for (i = 0; i<nlong>> 1; i++)
in_data[i + (nlong >> 1)] = ixheaacd_mult32x16in32_shl(
in_data[i + (nlong >> 1)], window_long_prev[nlong - 1 - 2 * i - 1]);
for (i = 0; i<nshort>> 1; i++)
in_data[i + nlong + nflat_ls + (nshort >> 1)] =
ixheaacd_mult32x16in32_shl(
in_data[i + nlong + nflat_ls + (nshort >> 1)],
window_short[nshort - 1 - 2 * i - 1]);
for (i = 0; i < nflat_ls; i++) in_data[i + nlong + nflat_ls + nshort] = 0;
for (i = 0; i < nlong / 2; i++) {
out_mdct[nlong / 2 + i] =
ixheaacd_sub32(in_data[i], in_data[nlong - 1 - i]);
out_mdct[nlong / 2 - 1 - i] =
-ixheaacd_add32(in_data[nlong + i], in_data[nlong2 - 1 - i]);
}
{
expo = ixheaacd_calc_max_spectral_line_dec(out_mdct, 1024) - 1;
expo = 8 - expo;
imdct_scale = ixheaacd_inverse_transform(
out_mdct, in_data, aac_tables_ptr->pstr_imdct_tables, expo, 1024);
ixheaacd_post_twiddle_dec(in_data, out_mdct,
aac_tables_ptr->pstr_imdct_tables, 1024);
}
imdct_scale += 1;
for (i = 0; i < nlong; i++) {
out_mdct[i] = ixheaacd_shl32_dir(in_data[i], imdct_scale);
}
break;
case LONG_STOP_SEQUENCE:
for (i = 0; i < nflat_ls; i++) in_data[i] = 0;
for (i = 0; i<nshort>> 1; i++)
in_data[i + nflat_ls] = ixheaacd_mult32x16in32_shl(
in_data[i + nflat_ls], window_short_prev[2 * i]);
for (i = 0; i<nshort>> 1; i++)
in_data[i + nflat_ls + (nshort >> 1)] =
ixheaacd_mult32x16in32_shl(in_data[i + nflat_ls + (nshort >> 1)],
window_short_prev[127 - 2 * i]);
for (i = 0; i<nlong>> 1; i++)
in_data[i + nlong] = ixheaacd_mult32x16in32_shl(in_data[i + nlong],
window_long[2 * i + 1]);
for (i = 0; i<nlong>> 1; i++)
in_data[i + nlong + (nlong >> 1)] =
ixheaacd_mult32x16in32_shl(in_data[i + nlong + (nlong >> 1)],
window_long[nlong - 1 - 2 * i - 1]);
for (i = 0; i < nlong / 2; i++) {
out_mdct[nlong / 2 + i] =
ixheaacd_sub32(in_data[i], in_data[nlong - 1 - i]);
out_mdct[nlong / 2 - 1 - i] =
-ixheaacd_add32(in_data[nlong + i], in_data[nlong2 - 1 - i]);
}
{
expo = ixheaacd_calc_max_spectral_line_dec(out_mdct, 1024) - 1;
expo = 8 - expo;
imdct_scale = ixheaacd_inverse_transform(
out_mdct, in_data, aac_tables_ptr->pstr_imdct_tables, expo, 1024);
ixheaacd_post_twiddle_dec(in_data, out_mdct,
aac_tables_ptr->pstr_imdct_tables, 1024);
}
imdct_scale += 1;
for (i = 0; i < nlong; i++) {
out_mdct[i] = ixheaacd_shl32_dir(in_data[i], imdct_scale);
}
break;
}
}
VOID ixheaacd_lt_update_state(WORD16 *lt_pred_stat, WORD16 *time,
WORD32 *overlap, WORD32 frame_len,
WORD32 object_type, WORD32 stride,
WORD16 window_sequence, WORD16 *p_window_next) {
WORD32 i;
if (object_type == AOT_ER_AAC_LD) {
WORD16 *ptr_ltp_state0 = &lt_pred_stat[0];
WORD16 *ptr_ltp_state_fl = &lt_pred_stat[frame_len + 0];
WORD16 *ptr_ltp_state_2fl = &lt_pred_stat[(frame_len * 2) + 0];
WORD16 *ptr_time_in = &time[0 * stride];
for (i = 0; i < frame_len; i++) {
*ptr_ltp_state0++ = *ptr_ltp_state_fl;
*ptr_ltp_state_fl++ = *ptr_ltp_state_2fl;
*ptr_ltp_state_2fl++ = *ptr_time_in;
ptr_time_in += stride;
}
} else {
WORD16 *ptr_ltp_state0 = &lt_pred_stat[0];
WORD16 *ptr_ltp_state_fl = &lt_pred_stat[frame_len + 0];
WORD16 *ptr_time_in = &time[0 * stride];
for (i = 0; i < frame_len; i++) {
*ptr_ltp_state0++ = *ptr_ltp_state_fl;
*ptr_ltp_state_fl++ = *ptr_time_in;
ptr_time_in += stride;
}
}
if ((window_sequence == ONLY_LONG_SEQUENCE) ||
(window_sequence == LONG_STOP_SEQUENCE)) {
if (512 == frame_len) {
WORD32 *window = (WORD32 *)p_window_next;
for (i = 0; i < 256; i++) {
lt_pred_stat[(frame_len * 3) + i] =
ixheaacd_round16(ixheaacd_mult16x16in32_shl(
(WORD16)ixheaacd_shl16(
(WORD16)-ixheaacd_sat16(overlap[255 - i]), 1),
(WORD16)ixheaacd_shr32(window[511 - i], 15)));
lt_pred_stat[(frame_len * 3) + 256 + i] =
ixheaacd_round16(ixheaacd_mult16x16in32_shl(
(WORD16)ixheaacd_shl16((WORD16)-ixheaacd_sat16(overlap[i]), 1),
(WORD16)ixheaacd_shr32(window[255 - i], 15)));
}
} else if (480 == frame_len) {
WORD32 *window = (WORD32 *)p_window_next;
for (i = 0; i < 240; i++) {
lt_pred_stat[(frame_len * 3) + i] =
ixheaacd_round16(ixheaacd_mult16x16in32_shl(
(WORD16)ixheaacd_shl16(
(WORD16)-ixheaacd_sat16(overlap[239 - i]), 1),
(WORD16)ixheaacd_shr32(window[479 - i], 15)));
lt_pred_stat[(frame_len * 3) + 240 + i] =
ixheaacd_round16(ixheaacd_mult16x16in32_shl(
(WORD16)ixheaacd_shl16((WORD16)-ixheaacd_sat16(overlap[i]), 1),
(WORD16)ixheaacd_shr32(window[239 - i], 15)));
}
} else {
for (i = 0; i < 512; i++) {
lt_pred_stat[(frame_len * 2) + i] = ixheaacd_round16(
ixheaacd_shl32_sat(ixheaacd_mult16x16in32_shl(
(WORD16)-ixheaacd_sat16(overlap[511 - i]),
p_window_next[2 * i + 1]),
1));
lt_pred_stat[(frame_len * 2) + 512 + i] =
ixheaacd_round16(ixheaacd_shl32_sat(
ixheaacd_mult16x16in32_shl((WORD16)-ixheaacd_sat16(overlap[i]),
p_window_next[1023 - 2 * i - 1]),
1));
}
}
} else if (window_sequence == LONG_START_SEQUENCE) {
for (i = 0; i < 448; i++) {
lt_pred_stat[(frame_len * 2) + i] =
ixheaacd_shl16((WORD16)-ixheaacd_sat16(overlap[511 - i]), 1);
}
for (i = 0; i < 64; i++) {
lt_pred_stat[(frame_len * 2) + 448 + i] =
ixheaacd_round16(ixheaacd_shl32_sat(
ixheaacd_mult16x16in32_shl(
(WORD16)-ixheaacd_sat16(overlap[511 - 448 - i]),
p_window_next[2 * i + 1]),
1));
}
for (i = 0; i < 64; i++) {
lt_pred_stat[(frame_len * 2) + 512 + i] =
ixheaacd_round16(ixheaacd_shl32_sat(
ixheaacd_mult16x16in32_shl((WORD16)-ixheaacd_sat16(overlap[i]),
p_window_next[127 - 2 * i - 1]),
1));
}
for (i = 576; i < 1024; i++) {
lt_pred_stat[(frame_len * 2) + i] = 0;
}
} else {
for (i = 0; i < 448; i++) {
lt_pred_stat[(frame_len * 2) + i] =
ixheaacd_shl16(ixheaacd_sat16(overlap[i]), 1);
}
for (i = 0; i < 64; i++) {
lt_pred_stat[(frame_len * 2) + 448 + i] = ixheaacd_round16(
ixheaacd_shl32_sat(ixheaacd_mult16x16in32_shl(
(WORD16)-ixheaacd_sat16(overlap[511 - i]),
p_window_next[2 * i + 1]),
1));
}
for (i = 0; i < 64; i++) {
lt_pred_stat[(frame_len * 2) + 512 + i] = ixheaacd_round16(
ixheaacd_shl32_sat(ixheaacd_mult16x16in32_shl(
(WORD16)-ixheaacd_sat16(overlap[448 + i]),
p_window_next[127 - 2 * i - 1]),
1));
}
for (i = 576; i < 1024; i++) {
lt_pred_stat[(frame_len * 2) + i] = 0;
}
}
}