344 lines
14 KiB
C
344 lines
14 KiB
C
/******************************************************************************
|
|
* *
|
|
* 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 <string.h>
|
|
#include <math.h>
|
|
#include "ixheaac_type_def.h"
|
|
#include "impd_drc_common_enc.h"
|
|
#include "impd_drc_uni_drc.h"
|
|
#include "impd_drc_tables.h"
|
|
#include "impd_drc_api.h"
|
|
#include "ixheaace_api.h"
|
|
#include "ixheaac_error_standards.h"
|
|
#include "ixheaace_error_codes.h"
|
|
#include "ixheaace_mps_common_fix.h"
|
|
#include "ixheaace_mps_defines.h"
|
|
#include "ixheaace_bitbuffer.h"
|
|
#include "ixheaace_mps_common_define.h"
|
|
#include "ixheaace_mps_param_extract.h"
|
|
#include "ixheaace_mps_buf.h"
|
|
#include "ixheaace_mps_lib.h"
|
|
#include "ixheaace_mps_bitstream.h"
|
|
#include "ixheaace_mps_struct_def.h"
|
|
#include "ixheaace_mps_sac_polyphase.h"
|
|
#include "ixheaace_mps_sac_nlc_enc.h"
|
|
#include "ixheaace_mps_sac_hybfilter.h"
|
|
#include "ixheaace_mps_spatial_bitstream.h"
|
|
#include "ixheaace_mps_tree.h"
|
|
#include "ixheaace_mps_rom.h"
|
|
#include "ixheaace_mps_vector_functions.h"
|
|
#include "ixheaac_constants.h"
|
|
#include "ixheaace_aac_constants.h"
|
|
#include "ixheaac_basic_ops32.h"
|
|
#include "ixheaac_basic_ops40.h"
|
|
#include "ixheaac_basic_ops.h"
|
|
|
|
static UWORD8 ixheaace_mps_212_get_icc_correlation_coherence_border(
|
|
const WORD32 aot, const WORD32 use_coherence_only) {
|
|
ixheaace_mps_box_subband_setup pstr_box_subband_setup;
|
|
|
|
if (aot == AOT_AAC_ELD) {
|
|
pstr_box_subband_setup.p_subband_2_parameter_index_ld = (UWORD8 *)subband_2_parameter_ld;
|
|
pstr_box_subband_setup.icc_correlation_coherence_border = 5;
|
|
} else {
|
|
pstr_box_subband_setup.p_subband_2_parameter_index_ld = (UWORD8 *)subband_2_parameter_usac;
|
|
pstr_box_subband_setup.icc_correlation_coherence_border = 8;
|
|
}
|
|
|
|
return ((use_coherence_only) ? 0 : pstr_box_subband_setup.icc_correlation_coherence_border);
|
|
}
|
|
|
|
static VOID ixheaace_mps_212_calc_correlation_vec(FLOAT32 *const data,
|
|
const FLOAT32 *const data_real,
|
|
const FLOAT32 *const power_data_1,
|
|
const FLOAT32 *const power_data_2,
|
|
const WORD32 icc_correlation_coherence_border) {
|
|
WORD32 idx;
|
|
FLOAT32 p_12;
|
|
|
|
for (idx = 0; idx < icc_correlation_coherence_border; idx++) {
|
|
p_12 = power_data_1[idx] * power_data_2[idx];
|
|
if (p_12 > 0.0f) {
|
|
p_12 = 1.0f / ((FLOAT32)sqrt(p_12));
|
|
data[idx] = data_real[idx] * p_12;
|
|
} else {
|
|
data[idx] = 0.9995f;
|
|
}
|
|
}
|
|
}
|
|
|
|
static VOID ixheaace_mps_212_calc_coherence_vec(FLOAT32 *const data,
|
|
const FLOAT32 *const data_real,
|
|
const FLOAT32 *const data_imag,
|
|
const FLOAT32 *const power_data_1,
|
|
const FLOAT32 *const power_data_2,
|
|
const WORD32 icc_correlation_coherence_border) {
|
|
WORD32 idx;
|
|
FLOAT32 coh, p_12, p_12_ri;
|
|
|
|
for (idx = 0; idx < icc_correlation_coherence_border; idx++) {
|
|
p_12_ri = (FLOAT32)(sqrt(data_real[idx] * data_real[idx] + data_imag[idx] * data_imag[idx]));
|
|
p_12 = power_data_1[idx] * power_data_2[idx];
|
|
|
|
if (p_12 > 0.0f) {
|
|
p_12 = 1.0f / ((FLOAT32)sqrt(p_12));
|
|
coh = p_12_ri * p_12;
|
|
data[idx] = coh;
|
|
} else {
|
|
data[idx] = 0.9995f;
|
|
}
|
|
}
|
|
}
|
|
|
|
static VOID ixheaace_mps_212_quantize_coef(const FLOAT32 *const input, const WORD32 num_bands,
|
|
const FLOAT32 *const quant_table,
|
|
const WORD32 idx_offset, const WORD32 num_quant_steps,
|
|
WORD8 *const quant_out) {
|
|
WORD32 band;
|
|
WORD32 forward = (quant_table[1] >= quant_table[0]);
|
|
|
|
for (band = 0; band < num_bands; band++) {
|
|
FLOAT32 q_val;
|
|
FLOAT32 cur_val = input[band];
|
|
WORD32 upper = num_quant_steps - 1;
|
|
WORD32 lower = 0;
|
|
if (forward) {
|
|
while (upper - lower > 1) {
|
|
WORD32 idx = (lower + upper) >> 1;
|
|
q_val = quant_table[idx];
|
|
if (cur_val <= q_val) {
|
|
upper = idx;
|
|
} else {
|
|
lower = idx;
|
|
}
|
|
}
|
|
|
|
if ((cur_val - quant_table[lower]) > (quant_table[upper] - cur_val)) {
|
|
quant_out[band] = (WORD8)(upper - idx_offset);
|
|
} else {
|
|
quant_out[band] = (WORD8)(lower - idx_offset);
|
|
}
|
|
} else {
|
|
while (upper - lower > 1) {
|
|
WORD32 idx = (lower + upper) >> 1;
|
|
q_val = quant_table[idx];
|
|
if (cur_val >= q_val) {
|
|
upper = idx;
|
|
} else {
|
|
lower = idx;
|
|
}
|
|
}
|
|
|
|
if ((cur_val - quant_table[lower]) < (quant_table[upper] - cur_val)) {
|
|
quant_out[band] = (WORD8)(upper - idx_offset);
|
|
} else {
|
|
quant_out[band] = (WORD8)(lower - idx_offset);
|
|
}
|
|
}
|
|
if (quant_out[band] != 0) {
|
|
quant_out[band] = quant_out[band];
|
|
}
|
|
}
|
|
}
|
|
|
|
VOID ixheaace_mps_212_calc_parameter_band_to_hybrid_band_offset(
|
|
const WORD32 num_hybrid_bands, UWORD8 *ptr_parameter_band_2_hybrid_band_offset, WORD32 aot) {
|
|
ixheaace_mps_box_subband_setup pstr_box_subband_setup;
|
|
if (aot == AOT_AAC_ELD) {
|
|
pstr_box_subband_setup.p_subband_2_parameter_index_ld = (UWORD8 *)subband_2_parameter_ld;
|
|
pstr_box_subband_setup.icc_correlation_coherence_border = 5;
|
|
} else {
|
|
pstr_box_subband_setup.p_subband_2_parameter_index_ld = (UWORD8 *)subband_2_parameter_usac;
|
|
pstr_box_subband_setup.icc_correlation_coherence_border = 8;
|
|
}
|
|
|
|
const UWORD8 *p_subband_2_parameter_index;
|
|
|
|
UWORD8 idx;
|
|
WORD32 band;
|
|
|
|
p_subband_2_parameter_index = pstr_box_subband_setup.p_subband_2_parameter_index_ld;
|
|
|
|
for (band = 0, idx = NUM_QMF_BANDS - 1; idx > NUM_QMF_BANDS - num_hybrid_bands; idx--) {
|
|
if (p_subband_2_parameter_index[idx - 1] - p_subband_2_parameter_index[idx]) {
|
|
ptr_parameter_band_2_hybrid_band_offset[band++] = (NUM_QMF_BANDS - idx);
|
|
}
|
|
}
|
|
ptr_parameter_band_2_hybrid_band_offset[band++] = (NUM_QMF_BANDS - idx);
|
|
}
|
|
|
|
IA_ERRORCODE
|
|
ixheaace_mps_212_init_tto_box(ixheaace_mps_pstr_tto_box pstr_tto_box,
|
|
const ixheaace_mps_tto_box_config *const pstr_tto_box_config,
|
|
UWORD8 *ptr_parameter_band_2_hybrid_band_offset, WORD32 aot) {
|
|
IA_ERRORCODE error = IA_NO_ERROR;
|
|
|
|
{
|
|
memset(pstr_tto_box, 0, sizeof(ixheaace_mps_tto_box));
|
|
pstr_tto_box->use_coarse_quant_cld_flag = pstr_tto_box_config->use_coarse_quant_cld_flag;
|
|
pstr_tto_box->use_coarse_quant_icc_flag = pstr_tto_box_config->use_coarse_quant_icc_flag;
|
|
pstr_tto_box->box_quant_mode = pstr_tto_box_config->box_quant_mode;
|
|
pstr_tto_box->icc_correlation_coherence_border =
|
|
ixheaace_mps_212_get_icc_correlation_coherence_border(
|
|
aot, pstr_tto_box_config->b_use_coherence_icc_only);
|
|
pstr_tto_box->num_hybrid_bands_max = pstr_tto_box_config->num_hybrid_bands_max;
|
|
if (aot == 39) {
|
|
pstr_tto_box->num_parameter_bands = IXHEAACE_MPS_SAC_BANDS_ld;
|
|
} else {
|
|
pstr_tto_box->num_parameter_bands = IXHEAACE_MPS_SAC_BANDS_usac;
|
|
}
|
|
pstr_tto_box->ptr_parameter_band_2_hybrid_band_offset =
|
|
ptr_parameter_band_2_hybrid_band_offset;
|
|
pstr_tto_box->frame_keep_flag = pstr_tto_box_config->frame_keep_flag;
|
|
pstr_tto_box->n_cld_quant_steps = pstr_tto_box->use_coarse_quant_cld_flag
|
|
? IXHEAACE_MPS_MAX_CLD_QUANT_COARSE
|
|
: IXHEAACE_MPS_MAX_CLD_QUANT_FINE;
|
|
pstr_tto_box->n_cld_quant_offset = pstr_tto_box->use_coarse_quant_cld_flag
|
|
? IXHEAACE_MPS_OFFSET_CLD_QUANT_COARSE
|
|
: IXHEAACE_MPS_OFFSET_CLD_QUANT_FINE;
|
|
pstr_tto_box->n_icc_quant_steps = pstr_tto_box->use_coarse_quant_icc_flag
|
|
? IXHEAACE_MPS_MAX_ICC_QUANT_COARSE
|
|
: IXHEAACE_MPS_MAX_ICC_QUANT_FINE;
|
|
pstr_tto_box->n_icc_quant_offset = pstr_tto_box->use_coarse_quant_icc_flag
|
|
? IXHEAACE_MPS_OFFSET_ICC_QUANT_COARSE
|
|
: IXHEAACE_MPS_OFFSET_ICC_QUANT_FINE;
|
|
pstr_tto_box->p_icc_quant_table =
|
|
pstr_tto_box->use_coarse_quant_icc_flag ? icc_quant_table_coarse : icc_quant_table_fine;
|
|
pstr_tto_box->p_cld_quant_table_enc = pstr_tto_box->use_coarse_quant_cld_flag
|
|
? cld_quant_table_coarse_enc
|
|
: cld_quant_table_fine_enc;
|
|
|
|
if ((pstr_tto_box->box_quant_mode != IXHEAACE_MPS_QUANTMODE_FINE) &&
|
|
(pstr_tto_box->box_quant_mode != IXHEAACE_MPS_QUANTMODE_EBQ1) &&
|
|
(pstr_tto_box->box_quant_mode != IXHEAACE_MPS_QUANTMODE_EBQ2)) {
|
|
return IA_EXHEAACE_INIT_FATAL_MPS_INIT_FAILED;
|
|
}
|
|
}
|
|
return error;
|
|
}
|
|
|
|
IA_ERRORCODE ixheaace_mps_212_apply_tto_box(
|
|
ixheaace_mps_pstr_tto_box pstr_tto_box, const WORD32 num_time_slots,
|
|
const WORD32 start_time_slot, const WORD32 num_hybrid_bands,
|
|
ixheaace_cmplx_str pp_hybrid_data_1[MAX_ANA_TIME_SLOT][MAX_QMF_BANDS],
|
|
ixheaace_cmplx_str pp_hybrid_data_2[MAX_ANA_TIME_SLOT][MAX_QMF_BANDS], WORD8 *const ptr_icc,
|
|
UWORD8 *const pb_icc_quant_coarse, WORD8 *const ptr_cld, UWORD8 *const pb_cld_quant_coarse,
|
|
const WORD32 b_use_bb_cues) {
|
|
IA_ERRORCODE error = IA_NO_ERROR;
|
|
|
|
WORD32 j, band;
|
|
FLOAT32 power_hybrid_data_1[MAX_NUM_PARAM_BANDS] = {0};
|
|
FLOAT32 power_hybrid_data_2[MAX_NUM_PARAM_BANDS] = {0};
|
|
FLOAT32 prod_hybrid_data_real[MAX_NUM_PARAM_BANDS] = {0};
|
|
FLOAT32 prod_hybrid_data_imag[MAX_NUM_PARAM_BANDS] = {0};
|
|
const WORD32 num_param_bands = pstr_tto_box->num_parameter_bands;
|
|
const WORD32 use_box_quant_mode =
|
|
(pstr_tto_box->box_quant_mode == IXHEAACE_MPS_QUANTMODE_EBQ1) ||
|
|
(pstr_tto_box->box_quant_mode == IXHEAACE_MPS_QUANTMODE_EBQ2);
|
|
|
|
if ((num_hybrid_bands < 0) || (num_hybrid_bands > pstr_tto_box->num_hybrid_bands_max)) {
|
|
return IA_EXHEAACE_CONFIG_NONFATAL_MPS_INVALID_CONFIG;
|
|
}
|
|
|
|
for (j = 0, band = 0; band < num_param_bands; band++) {
|
|
FLOAT32 data_1, data_2;
|
|
data_1 = 0;
|
|
data_2 = 0;
|
|
for (; j < pstr_tto_box->ptr_parameter_band_2_hybrid_band_offset[band]; j++) {
|
|
data_1 += ixheaace_mps_212_sum_up_cplx_pow_2_dim_2(pp_hybrid_data_1, start_time_slot,
|
|
num_time_slots, j, j + 1);
|
|
data_2 += ixheaace_mps_212_sum_up_cplx_pow_2_dim_2(pp_hybrid_data_2, start_time_slot,
|
|
num_time_slots, j, j + 1);
|
|
}
|
|
|
|
power_hybrid_data_1[band] = data_1;
|
|
power_hybrid_data_2[band] = data_2;
|
|
}
|
|
for (j = 0, band = 0; band < num_param_bands; band++) {
|
|
FLOAT32 data_real, data_imag;
|
|
data_real = data_imag = 0;
|
|
for (; j < pstr_tto_box->ptr_parameter_band_2_hybrid_band_offset[band]; j++) {
|
|
ixheaace_cmplx_str scalar_prod;
|
|
ixheaace_mps_212_cplx_scalar_product(&scalar_prod, pp_hybrid_data_1, pp_hybrid_data_2,
|
|
start_time_slot, num_time_slots, j, j + 1);
|
|
data_real += scalar_prod.re;
|
|
data_imag += scalar_prod.im;
|
|
}
|
|
prod_hybrid_data_real[band] = data_real;
|
|
prod_hybrid_data_imag[band] = data_imag;
|
|
}
|
|
|
|
ixheaace_mps_212_calc_correlation_vec(pstr_tto_box->icc, prod_hybrid_data_real,
|
|
power_hybrid_data_1, power_hybrid_data_2,
|
|
pstr_tto_box->icc_correlation_coherence_border);
|
|
|
|
ixheaace_mps_212_calc_coherence_vec(
|
|
&pstr_tto_box->icc[pstr_tto_box->icc_correlation_coherence_border],
|
|
&prod_hybrid_data_real[pstr_tto_box->icc_correlation_coherence_border],
|
|
&prod_hybrid_data_imag[pstr_tto_box->icc_correlation_coherence_border],
|
|
&power_hybrid_data_1[pstr_tto_box->icc_correlation_coherence_border],
|
|
&power_hybrid_data_2[pstr_tto_box->icc_correlation_coherence_border],
|
|
num_param_bands - pstr_tto_box->icc_correlation_coherence_border);
|
|
if (error) {
|
|
return error;
|
|
}
|
|
if (!use_box_quant_mode) {
|
|
FLOAT32 power_1, power_2, cld;
|
|
FLOAT32 max_pow = 30.0f;
|
|
|
|
for (band = 0; band < num_param_bands; band++) {
|
|
power_1 = (FLOAT32)log(power_hybrid_data_1[band] / 2.0f);
|
|
power_2 = (FLOAT32)log(power_hybrid_data_2[band] / 2.0f);
|
|
power_1 = MAX(MIN(power_1, max_pow), -max_pow);
|
|
power_2 = MAX(MIN(power_2, max_pow), -max_pow);
|
|
cld = (INV_LN_10_10 * (power_1 - power_2));
|
|
pstr_tto_box->cld[band] = cld;
|
|
}
|
|
}
|
|
|
|
if (b_use_bb_cues) {
|
|
FLOAT32 temp;
|
|
temp = 0;
|
|
for (band = 0; band < num_param_bands; band++) {
|
|
temp += pstr_tto_box->cld[band];
|
|
}
|
|
temp /= num_param_bands;
|
|
for (band = 0; band < num_param_bands; band++) {
|
|
pstr_tto_box->cld[band] = temp;
|
|
}
|
|
}
|
|
|
|
ixheaace_mps_212_quantize_coef(
|
|
pstr_tto_box->icc, num_param_bands, pstr_tto_box->p_icc_quant_table,
|
|
pstr_tto_box->n_icc_quant_offset, pstr_tto_box->n_icc_quant_steps, ptr_icc);
|
|
|
|
*pb_icc_quant_coarse = pstr_tto_box->use_coarse_quant_icc_flag;
|
|
|
|
if (!use_box_quant_mode) {
|
|
ixheaace_mps_212_quantize_coef(
|
|
pstr_tto_box->cld, num_param_bands, pstr_tto_box->p_cld_quant_table_enc,
|
|
pstr_tto_box->n_cld_quant_offset, pstr_tto_box->n_cld_quant_steps, ptr_cld);
|
|
} else {
|
|
memset(ptr_cld, 0, num_param_bands * sizeof(WORD8));
|
|
}
|
|
*pb_cld_quant_coarse = pstr_tto_box->use_coarse_quant_cld_flag;
|
|
|
|
return error;
|
|
}
|