libxaac-mirror/encoder/ixheaace_ps_enc_init.c
2023-10-12 19:24:54 +05:30

210 lines
7.4 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 "ixheaac_error_standards.h"
#include "ixheaac_type_def.h"
#include "ixheaace_sbr_def.h"
#include "ixheaace_resampler.h"
#include "ixheaace_sbr_rom.h"
#include "ixheaace_common_rom.h"
#include "ixheaace_bitbuffer.h"
#include "ixheaace_sbr_hybrid.h"
#include "ixheaace_sbr_ps_enc.h"
#include "ixheaac_constants.h"
#include "ixheaace_aac_constants.h"
#include "ixheaace_error_codes.h"
#include "ixheaace_bitbuffer.h"
WORD32 ixheaace_get_ps_mode(WORD32 bitrate) {
WORD32 ps_mode = 0;
if (bitrate < 21000) {
ps_mode = PS_MODE_LOW_FREQ_RES_IID_ICC;
}
return ps_mode;
}
IA_ERRORCODE
ixheaace_create_ps_enc(ixheaace_pstr_ps_enc pstr_ps_enc, WORD32 ps_mode,
FLOAT32 *ptr_common_buffer, FLOAT32 *ptr_common_buffer2,
FLOAT32 *ptr_ps_buf3) {
WORD32 i;
IA_ERRORCODE err = IA_NO_ERROR;
FLOAT32 *ptr1, *ptr2, *ptr3, *ptr4;
if (pstr_ps_enc == NULL) {
return IA_EXHEAACE_INIT_FATAL_PS_INIT_FAILED;
}
ptr1 = &ptr_common_buffer2[IXHEAACE_QMF_TIME_SLOTS * IXHEAACE_QMF_CHANNELS];
ptr2 = pstr_ps_enc->ps_buf2;
ptr3 = ptr_ps_buf3;
ptr4 = &ptr_common_buffer[5 * NO_OF_ESTIMATES * MAXIMUM_FREQ_COEFFS];
pstr_ps_enc->ps_mode = ps_mode;
pstr_ps_enc->b_prev_zero_iid = 0;
pstr_ps_enc->b_prev_zero_icc = 0;
pstr_ps_enc->b_hi_freq_res_iid_icc = ((ps_mode & PS_MODE_LOW_FREQ_RES_IID_ICC) != 0) ? 0 : 1;
pstr_ps_enc->iid_icc_bins =
(pstr_ps_enc->b_hi_freq_res_iid_icc) ? NUMBER_OF_IID_BINS : NUMBER_OF_LOW_RES_IID_BINS;
pstr_ps_enc->aaa_ICC_data_buf = (FLOAT32 **)ptr1;
ptr1 += NUMBER_OF_BINS * sizeof(FLOAT32 *) / sizeof(FLOAT32);
pstr_ps_enc->aaa_IID_data_buf = (FLOAT32 **)ptr1;
ptr1 += NUMBER_OF_BINS * sizeof(FLOAT32 *) / sizeof(FLOAT32);
for (i = 0; i < NUMBER_OF_BINS; i++) {
pstr_ps_enc->aaa_ICC_data_buf[i] = ptr1;
ptr1 += SYSTEMLOOKAHEAD + 1;
memset(pstr_ps_enc->aaa_ICC_data_buf[i], 0,
(SYSTEMLOOKAHEAD + 1) * sizeof(pstr_ps_enc->aaa_ICC_data_buf[0]));
pstr_ps_enc->aaa_IID_data_buf[i] = ptr1;
ptr1 += SYSTEMLOOKAHEAD + 1;
memset(pstr_ps_enc->aaa_IID_data_buf[i], 0,
(SYSTEMLOOKAHEAD + 1) * sizeof(pstr_ps_enc->aaa_IID_data_buf[0]));
}
pstr_ps_enc->ptr_hybrid_left = &pstr_ps_enc->hybrid_left;
pstr_ps_enc->ptr_hybrid_right = &pstr_ps_enc->hybrid_right;
err = ixheaace_create_hybrid_filter_bank(pstr_ps_enc->ptr_hybrid_left, &ptr4);
if (err) {
return err;
}
err = ixheaace_create_hybrid_filter_bank(pstr_ps_enc->ptr_hybrid_right, &ptr4);
if (err) {
return err;
}
for (i = 0; i < NUMBER_OF_SUBSAMPLES; i++) {
pstr_ps_enc->m_hybrid_real_left[i] = ptr3;
ptr3 += IXHEAACE_NUM_HYBRID_BANDS;
memset(pstr_ps_enc->m_hybrid_real_left[i], 0,
IXHEAACE_NUM_HYBRID_BANDS * sizeof(pstr_ps_enc->m_hybrid_real_left[0]));
pstr_ps_enc->m_hybrid_imag_left[i] = ptr3;
ptr3 += IXHEAACE_NUM_HYBRID_BANDS;
memset(pstr_ps_enc->m_hybrid_imag_left[i], 0,
IXHEAACE_NUM_HYBRID_BANDS * sizeof(pstr_ps_enc->m_hybrid_imag_left[0]));
pstr_ps_enc->m_hybrid_real_right[i] = ptr1;
ptr1 += IXHEAACE_NUM_HYBRID_BANDS;
memset(pstr_ps_enc->m_hybrid_real_right[i], 0,
IXHEAACE_NUM_HYBRID_BANDS * sizeof(pstr_ps_enc->m_hybrid_real_right[0]));
pstr_ps_enc->m_hybrid_imag_right[i] = ptr1;
ptr1 += IXHEAACE_NUM_HYBRID_BANDS;
memset(pstr_ps_enc->m_hybrid_imag_right[i], 0,
IXHEAACE_NUM_HYBRID_BANDS * sizeof(pstr_ps_enc->m_hybrid_imag_right[0]));
}
pstr_ps_enc->temp_qmf_left_real = (FLOAT32 **)ptr1;
ptr1 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);
pstr_ps_enc->temp_qmf_left_imag = (FLOAT32 **)ptr1;
ptr1 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(WORD32 *) / sizeof(WORD32);
pstr_ps_enc->hist_qmf_left_real = (FLOAT32 **)ptr2;
ptr2 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);
pstr_ps_enc->hist_qmf_left_imag = (FLOAT32 **)ptr2;
ptr2 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);
pstr_ps_enc->hist_qmf_right_real = (FLOAT32 **)ptr2;
ptr2 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);
pstr_ps_enc->hist_qmf_right_imag = (FLOAT32 **)ptr2;
ptr2 += IXHEAACE_HYBRID_FILTER_DELAY * sizeof(FLOAT32 *) / sizeof(FLOAT32);
for (i = 0; i < IXHEAACE_HYBRID_FILTER_DELAY; i++) {
pstr_ps_enc->temp_qmf_left_real[i] = ptr1;
ptr1 += NUMBER_OF_QMF_BANDS;
memset(pstr_ps_enc->temp_qmf_left_real[i], 0,
NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->temp_qmf_left_real[0]));
pstr_ps_enc->temp_qmf_left_imag[i] = ptr1;
ptr1 += NUMBER_OF_QMF_BANDS;
memset(pstr_ps_enc->temp_qmf_left_imag[i], 0,
NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->temp_qmf_left_imag[0]));
pstr_ps_enc->hist_qmf_left_real[i] = ptr2;
ptr2 += NUMBER_OF_QMF_BANDS;
memset(pstr_ps_enc->hist_qmf_left_real[i], 0,
NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->hist_qmf_left_real[0]));
pstr_ps_enc->hist_qmf_left_imag[i] = ptr2;
ptr2 += NUMBER_OF_QMF_BANDS;
memset(pstr_ps_enc->hist_qmf_left_imag[i], 0,
NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->hist_qmf_left_imag[0]));
pstr_ps_enc->hist_qmf_right_real[i] = ptr2;
ptr2 += NUMBER_OF_QMF_BANDS;
memset(pstr_ps_enc->hist_qmf_right_real[i], 0,
NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->hist_qmf_right_real[0]));
pstr_ps_enc->hist_qmf_right_imag[i] = ptr2;
ptr2 += NUMBER_OF_QMF_BANDS;
memset(pstr_ps_enc->hist_qmf_right_imag[i], 0,
NUMBER_OF_QMF_BANDS * sizeof(pstr_ps_enc->hist_qmf_right_imag[0]));
}
memset(pstr_ps_enc->pow_left_right, 0, sizeof(pstr_ps_enc->pow_left_right));
memset(pstr_ps_enc->pow_corr_real_imag, 0, sizeof(pstr_ps_enc->pow_corr_real_imag));
if ((pstr_ps_enc->hist_qmf_left_real == NULL) || (pstr_ps_enc->hist_qmf_left_imag == NULL) ||
(pstr_ps_enc->hist_qmf_right_real == NULL) || (pstr_ps_enc->hist_qmf_right_imag == NULL)) {
return IA_EXHEAACE_INIT_FATAL_PS_INIT_FAILED;
}
for (i = 0; i < pstr_ps_enc->iid_icc_bins; i++) {
pstr_ps_enc->aaa_IID_data_buf[i][0] = 0;
pstr_ps_enc->aaa_ICC_data_buf[i][0] = -1.0f;
}
pstr_ps_enc->bit_buf_read_offset = 0;
pstr_ps_enc->bit_buf_write_offset = 0;
ia_enhaacplus_enc_create_bitbuffer(&pstr_ps_enc->ps_bit_buf, (UWORD8 *)ptr1, 255 + 15);
pstr_ps_enc->ps_bit_buf.ptr_read_next =
pstr_ps_enc->ps_bit_buf.ptr_bit_buf_base + pstr_ps_enc->bit_buf_read_offset;
pstr_ps_enc->ps_bit_buf.ptr_write_next =
pstr_ps_enc->ps_bit_buf.ptr_bit_buf_base + pstr_ps_enc->bit_buf_write_offset;
return err;
}