wrapper/libxaac
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
libxaac/encoder/iusace_lpd_enc.c
akshayragir833 63d6a41982
USAC encoder enhancements and internal bug fixes (#76) 
Tests done: Smoke test
2024-01-12 13:05:51 +05:30

848 lines
33 KiB
C
Raw Permalink Blame 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 <string.h>
#include <math.h>
#include "ixheaac_type_def.h"
#include "ixheaace_adjust_threshold_data.h"
#include "iusace_bitbuffer.h"
/* DRC */
#include "impd_drc_common_enc.h"
#include "impd_drc_uni_drc.h"
#include "impd_drc_tables.h"
#include "impd_drc_api.h"
#include "impd_drc_uni_drc_eq.h"
#include "impd_drc_uni_drc_filter_bank.h"
#include "impd_drc_gain_enc.h"
#include "impd_drc_struct_def.h"
#include "iusace_cnst.h"
#include "iusace_tns_usac.h"
#include "iusace_psy_mod.h"
#include "iusace_psy_utils.h"
#include "iusace_tns_usac.h"
#include "iusace_config.h"
#include "iusace_arith_enc.h"
#include "iusace_fd_qc_util.h"
#include "iusace_fd_quant.h"
#include "iusace_block_switch_const.h"
#include "iusace_block_switch_struct_def.h"
#include "iusace_ms.h"
#include "iusace_signal_classifier.h"
#include "ixheaace_sbr_header.h"
#include "ixheaace_config.h"
#include "ixheaace_asc_write.h"
#include "iusace_main.h"
#include "iusace_lpd_rom.h"
#include "iusace_lpd.h"
#include "iusace_func_prototypes.h"
#include "ixheaac_error_standards.h"
VOID iusace_init_td_data(ia_usac_td_encoder_struct *st, WORD32 len_frame) {
WORD32 len_window;
WORD32 num_frames = NUM_FRAMES;
st->len_subfrm = len_frame / num_frames;
st->len_frame = len_frame;
st->num_subfrm = (MAX_NUM_SUBFR * len_frame) / LEN_SUPERFRAME;
iusace_reset_td_enc(st);
st->prev_mode = -1;
st->arith_reset_flag = 1;
if (st->fscale <= FSCALE_DENOM) {
len_window = (LEN_LP_WINDOW * len_frame) / LEN_SUPERFRAME;
} else {
len_window = (LEN_LP_WINDOW_HIGH_RATE * len_frame) / LEN_SUPERFRAME;
}
switch (len_window) {
case 512:
st->lp_analysis_window = iusace_cos_window_512;
break;
case 448:
st->lp_analysis_window = iusace_cos_window_448;
break;
case 384:
st->lp_analysis_window = iexheaac_cos_window_384;
break;
default:
st->lp_analysis_window = iusace_cos_window_512;
break;
}
return;
}
VOID iusace_config_acelp_core_mode(ia_usac_td_encoder_struct *st, WORD32 sampling_rate,
WORD32 bitrate) {
WORD32 max_bits, coder_bits;
const WORD32 *p_acelp_core_numbits_table;
p_acelp_core_numbits_table = (WORD32 *)iusace_acelp_core_numbits_1024;
max_bits = (WORD32)((FLOAT32)(bitrate * LEN_SUPERFRAME) / (FLOAT32)sampling_rate);
for (st->acelp_core_mode = 5; st->acelp_core_mode >= 0; st->acelp_core_mode--) {
coder_bits = p_acelp_core_numbits_table[st->acelp_core_mode];
if (coder_bits <= max_bits) {
return;
}
}
if (st->acelp_core_mode == -1) {
st->acelp_core_mode = 0;
}
return;
}
VOID iusace_reset_td_enc(ia_usac_td_encoder_struct *st) {
WORD32 i;
memset(st->old_speech_pe, 0, (ORDER + LEN_NEXT_HIGH_RATE) * sizeof(FLOAT32));
memset(st->prev_exc, 0, (MAX_PITCH + LEN_INTERPOL) * sizeof(FLOAT32));
memset(st->prev_wsp, 0, (MAX_PITCH / OPL_DECIM) * sizeof(FLOAT32));
memset(st->mem_lp_decim2, 0, 3 * sizeof(FLOAT32));
memset(st->weighted_sig, 0, 128 * sizeof(FLOAT32));
st->lpd_state.mode = -1;
st->lpd_state.num_bits = 0;
memset(st->lpd_state.lpc_coeffs_quant, 0, (2 * (ORDER + 1)) * sizeof(FLOAT32));
memset(st->lpd_state.lpc_coeffs, 0, (2 * (ORDER + 1)) * sizeof(FLOAT32));
memset(st->lpd_state.synth, 0, (ORDER + 128) * sizeof(FLOAT32));
memset(st->lpd_state.wsynth, 0, (1 + 128) * sizeof(FLOAT32));
memset(st->lpd_state.acelp_exc, 0, (2 * LEN_FRAME) * sizeof(FLOAT32));
memset(st->lpd_state.tcx_mem, 0, 128 * sizeof(FLOAT32));
memset(st->lpd_state.tcx_quant, 0, (1 + 256) * sizeof(FLOAT32));
st->lpd_state.tcx_fac = 0.0f;
memset(st->prev_hp_wsp, 0,
(LEN_SUPERFRAME / OPL_DECIM + (MAX_PITCH / OPL_DECIM)) * sizeof(FLOAT32));
memset(st->hp_ol_ltp_mem, 0, (3 * 2 + 1) * sizeof(FLOAT32));
for (i = 0; i < 5; i++) st->prev_ol_lags[i] = 40;
st->prev_wsyn_mem = 0.0;
st->prev_wsp_mem = 0.0;
st->prev_xnq_mem = 0.0;
st->mem_wsp = 0.0;
st->prev_ovlp_size = 0;
st->prev_pitch_med = 40;
st->ol_wght_flg = 0;
st->ada_w = 0.0;
memcpy(st->isf_old, iusace_lsf_init, ORDER * sizeof(FLOAT32));
memcpy(st->isp_old, iusace_ispold_init, ORDER * sizeof(FLOAT32));
memcpy(st->isp_old_q, st->isp_old, ORDER * sizeof(FLOAT32));
st->mem_preemph = 0.0;
memset(st->mem_sig_in, 0, 4 * sizeof(FLOAT32));
memset(st->xn_buffer, 0, 128 * sizeof(FLOAT32));
return;
}
VOID iusace_highpass_prev_wsp(ia_usac_td_encoder_struct *st, FLOAT32 *decim_sig,
WORD32 pitch_max) {
WORD32 i, k;
WORD32 wsp_offset = pitch_max / OPL_DECIM;
WORD32 num_frames = (2 * LEN_SUBFR) / OPL_DECIM;
FLOAT32 *hp_wsp_mem = st->hp_ol_ltp_mem;
FLOAT32 *prev_hp_wsp = st->prev_hp_wsp;
for (i = 0; i < 2 * st->len_subfrm / OPL_DECIM; i += num_frames) {
FLOAT32 *data_a, *data_b, *hp_wsp, o;
FLOAT32 *wsp = decim_sig + ORDER + i;
data_a = hp_wsp_mem;
data_b = hp_wsp_mem + HP_ORDER;
hp_wsp = prev_hp_wsp + wsp_offset;
for (k = 0; k < num_frames; k++) {
data_b[0] = data_b[1];
data_b[1] = data_b[2];
data_b[2] = data_b[3];
data_b[HP_ORDER] = wsp[k];
o = data_b[0] * 0.83787057505665F;
o += data_b[1] * -2.50975570071058F;
o += data_b[2] * 2.50975570071058F;
o += data_b[3] * -0.83787057505665F;
o -= data_a[0] * -2.64436711600664F;
o -= data_a[1] * 2.35087386625360F;
o -= data_a[2] * -0.70001156927424F;
data_a[2] = data_a[1];
data_a[1] = data_a[0];
data_a[0] = o;
hp_wsp[k] = o;
}
memmove(prev_hp_wsp, &prev_hp_wsp[num_frames], wsp_offset * sizeof(FLOAT32));
}
}
VOID iusace_find_weighted_speech(FLOAT32 *filter_coef, FLOAT32 *speech, FLOAT32 *wsp,
FLOAT32 *mem_wsp, WORD32 length) {
WORD32 i_subfr;
FLOAT32 weighted_lpc[ORDER + 1];
for (i_subfr = 0; i_subfr < length; i_subfr += LEN_SUBFR) {
iusace_get_weighted_lpc(filter_coef, weighted_lpc);
iusace_compute_lp_residual(weighted_lpc, &speech[i_subfr], &wsp[i_subfr], LEN_SUBFR);
filter_coef += (ORDER + 1);
}
iusace_apply_deemph(wsp, TILT_FAC, length, mem_wsp);
return;
}
VOID iusace_get_interpolated_lpc(FLOAT32 *lsp_old, FLOAT32 *lsp_new, FLOAT32 *lpc,
WORD32 num_subfrm) {
FLOAT32 lsp[ORDER], *p_lpc, inc, fnew, fold;
WORD32 i, k;
inc = 1.0f / (FLOAT32)num_subfrm;
p_lpc = lpc;
fnew = 0.0f;
for (k = 0; k < num_subfrm; k++) {
fold = 1.0f - fnew;
for (i = 0; i < ORDER; i++) {
lsp[i] = (FLOAT32)(lsp_old[i] * fold + lsp_new[i] * fnew);
}
fnew += inc;
iusace_lsp_to_lp_conversion(lsp, p_lpc);
p_lpc += (ORDER + 1);
}
iusace_lsp_to_lp_conversion(lsp_new, p_lpc);
}
VOID iusace_core_lpd_encode(ia_usac_data_struct *usac_data, FLOAT32 *speech, WORD32 *mode,
WORD32 *num_tcx_param, WORD32 ch_idx) {
WORD32 first_lpd_flag = (usac_data->core_mode_prev[ch_idx] == CORE_MODE_FD);
iusace_scratch_mem *pstr_scratch = &usac_data->str_scratch;
WORD32 pit_adj = usac_data->td_encoder[ch_idx]->fscale;
WORD32 *num_fac_bits = &usac_data->num_td_fac_bits[ch_idx];
WORD16 *serial_fac_out = usac_data->fac_out_stream[ch_idx];
WORD32 *lpc_params = usac_data->param_buf + (NUM_FRAMES * MAX_NUM_TCX_PRM_PER_DIV);
ia_usac_td_encoder_struct *st = usac_data->td_encoder[ch_idx];
WORD32 *acelp_tcx_params = usac_data->param_buf;
WORD16 *codec_mode = &st->acelp_core_mode;
const FLOAT32 *lp_analysis_window = st->lp_analysis_window;
FLOAT32 *lp_filter_coeff = pstr_scratch->p_lp_filter_coeff;
FLOAT32 *lp_filter_coeff_q = pstr_scratch->p_lp_filter_coeff_q;
FLOAT32 *ptr_stack_mem = (FLOAT32 *)pstr_scratch->ptr_stack_mem;
FLOAT32 *auto_corr_vector = ptr_stack_mem;
ptr_stack_mem += (ORDER + 1);
memset(auto_corr_vector, 0, sizeof(*auto_corr_vector) * (ORDER + 1));
FLOAT32 *isp_new = ptr_stack_mem;
ptr_stack_mem += ORDER;
memset(isp_new, 0, sizeof(*isp_new) * (ORDER));
FLOAT32 *isp_curr = ptr_stack_mem;
ptr_stack_mem += ((NUM_FRAMES + 1) * ORDER);
memset(isp_curr, 0, sizeof(*isp_curr) * ((NUM_FRAMES + 1) * ORDER));
FLOAT32 *isp_curr_q = ptr_stack_mem;
ptr_stack_mem += ((NUM_FRAMES + 1) * ORDER);
memset(isp_curr_q, 0, sizeof(*isp_curr_q) * ((NUM_FRAMES + 1) * ORDER));
FLOAT32 *isf_curr = ptr_stack_mem;
ptr_stack_mem += ((NUM_FRAMES + 1) * ORDER);
memset(isf_curr, 0, sizeof(*isf_curr) * ((NUM_FRAMES + 1) * ORDER));
FLOAT32 *isf_curr_q = ptr_stack_mem;
ptr_stack_mem += ((NUM_FRAMES + 1) * ORDER);
memset(isf_curr_q, 0, sizeof(*isf_curr_q) * ((NUM_FRAMES + 1) * ORDER));
FLOAT32 *auto_corr_lp_filter_coeff = ptr_stack_mem;
ptr_stack_mem += (ORDER + 1);
memset(auto_corr_lp_filter_coeff, 0, sizeof(*auto_corr_lp_filter_coeff) * (ORDER + 1));
WORD32 num_indices = 0, num_bits = 0;
WORD32 *prm_tcx = pstr_scratch->p_prm_tcx;
FLOAT32 *p_wsp_prev_buf;
FLOAT32 *wsp_prev_buf = pstr_scratch->p_wsp_prev_buf;
memset(lp_filter_coeff, 0, ((NUM_SUBFR_SUPERFRAME + 1) * (ORDER + 1)) * sizeof(FLOAT32));
memset(lp_filter_coeff_q, 0, ((NUM_SUBFR_SUPERFRAME + 1) * (ORDER + 1)) * sizeof(FLOAT32));
memset(wsp_prev_buf, 0, ((MAX_PITCH1 / OPL_DECIM) + LEN_FRAME) * sizeof(FLOAT32));
WORD32 i, j, k, i1, i2;
WORD32 *p_params;
FLOAT32 energy = 0, max_corr = 0, t0 = 0, *p, *p1;
WORD32 ol_pitch_lag[2 * NUM_FRAMES] = {0};
FLOAT32 norm_corr[2 * NUM_FRAMES] = {0};
WORD32 num_params, pitch_min, pitch_max;
ia_usac_lpd_state_struct *lpd_state[6], *lpd_state_temp;
lpd_state_temp = (ia_usac_lpd_state_struct *)ptr_stack_mem;
ptr_stack_mem +=
(sizeof(ia_usac_lpd_state_struct) + sizeof(*ptr_stack_mem)) / (sizeof(*ptr_stack_mem));
memset(lpd_state_temp, 0, sizeof(*lpd_state_temp));
for (j = 0; j < 6; j++) {
lpd_state[j] = (ia_usac_lpd_state_struct *)ptr_stack_mem;
ptr_stack_mem +=
(sizeof(ia_usac_lpd_state_struct) + sizeof(*ptr_stack_mem)) / (sizeof(*ptr_stack_mem));
memset(lpd_state[j], 0, sizeof(*lpd_state[0]));
}
WORD32 num_bits_acelp = 0, num_bits_tcx = 0;
WORD32 range_pitch_search = 0;
WORD32 len_subfrm = 0;
WORD32 num_subfrm = 0;
WORD32 num_sbfrm_per_supfrm = 0;
WORD32 window_len = 0;
WORD32 len = (MAX_PITCH / OPL_DECIM);
FLOAT32 mem_wsyn;
FLOAT32 ssnr_256 = 0.0f, ssnr_512 = 0.0f, ssnr_1024 = 0.0f;
FLOAT32 tmp_ssnr = 0.0f;
len_subfrm = st->len_subfrm;
num_subfrm = st->num_subfrm;
num_sbfrm_per_supfrm = NUM_FRAMES * num_subfrm;
if (pit_adj <= FSCALE_DENOM) {
window_len = (LEN_LP_WINDOW * len_subfrm) / LEN_FRAME;
} else {
window_len = (LEN_LP_WINDOW_HIGH_RATE * len_subfrm) / LEN_FRAME;
}
memcpy(pstr_scratch->p_wsig_buf, st->weighted_sig, 128 * sizeof(FLOAT32));
num_bits_acelp = (iusace_acelp_core_numbits_1024[*codec_mode] - NBITS_MODE) >> 2;
num_bits_tcx = (WORD32)(0.85f * num_bits_acelp - NBITS_LPC);
if (pit_adj == 0) {
pitch_min = TMIN;
pitch_max = TMAX;
} else {
i = (((pit_adj * TMIN) + (FSCALE_DENOM / 2)) / FSCALE_DENOM) - TMIN;
pitch_min = TMIN + i;
pitch_max = TMAX + (6 * i);
}
p_wsp_prev_buf = wsp_prev_buf + MAX_PITCH1 / OPL_DECIM;
memcpy(wsp_prev_buf, st->prev_wsp, (WORD32)((MAX_PITCH / OPL_DECIM) * sizeof(FLOAT32)));
if (first_lpd_flag) {
memcpy(st->isp_old, iusace_ispold_init, ORDER * sizeof(FLOAT32));
iusace_autocorr_plus(&speech[-(window_len / 2)], auto_corr_vector, window_len,
(FLOAT32 *)lp_analysis_window, pstr_scratch->p_buf_aut_corr);
for (j = 0; j <= ORDER; j++) {
auto_corr_vector[j] *= (FLOAT32)iusace_lag_window[j];
}
iusace_levinson_durbin_algo(auto_corr_vector, auto_corr_lp_filter_coeff);
iusace_lpc_2_lsp_conversion(auto_corr_lp_filter_coeff, isp_new, st->isp_old);
memcpy(st->isp_old, isp_new, ORDER * sizeof(FLOAT32));
iusace_lsp_2_lsf_conversion(isp_new, isf_curr);
memcpy(st->isf_old, isf_curr, ORDER * sizeof(FLOAT32));
}
memcpy(isp_curr, st->isp_old, ORDER * sizeof(FLOAT32));
for (i = 0; i < NUM_FRAMES; i++) {
iusace_autocorr_plus(&speech[((i + 1) * len_subfrm) - (window_len / 2)], auto_corr_vector,
window_len, (FLOAT32 *)lp_analysis_window, pstr_scratch->p_buf_aut_corr);
for (j = 0; j <= ORDER; j++) {
auto_corr_vector[j] *= (FLOAT32)iusace_lag_window[j];
}
iusace_levinson_durbin_algo(auto_corr_vector, auto_corr_lp_filter_coeff);
iusace_lpc_2_lsp_conversion(auto_corr_lp_filter_coeff, isp_new, st->isp_old);
memcpy(&isp_curr[(i + 1) * ORDER], isp_new, ORDER * sizeof(FLOAT32));
iusace_interpolation_lsp_params(st->isp_old, isp_new,
&lp_filter_coeff[i * num_subfrm * (ORDER + 1)], num_subfrm);
iusace_lsp_2_lsf_conversion(&isp_curr[(i + 1) * ORDER], &isf_curr[(i + 1) * ORDER]);
memcpy(st->isp_old, isp_new, ORDER * sizeof(FLOAT32));
}
memcpy(isf_curr, st->isf_old, ORDER * sizeof(FLOAT32));
memcpy(st->isf_old, &isf_curr[NUM_FRAMES * ORDER], ORDER * sizeof(FLOAT32));
if (!first_lpd_flag) {
iusace_lsp_2_lsf_conversion(st->isp_old_q, isf_curr_q);
}
iusace_quantize_lpc_avq(&isf_curr[ORDER], &isf_curr_q[ORDER], first_lpd_flag, &lpc_params[0],
&num_indices, &num_bits);
for (i = 0; i < NUM_FRAMES; i++) {
iusace_lsf_2_lsp_conversion(&isf_curr_q[(i + 1) * ORDER], &isp_curr_q[(i + 1) * ORDER]);
}
if (first_lpd_flag) {
iusace_lsf_2_lsp_conversion(isf_curr_q, isp_curr_q);
memcpy(st->isp_old_q, isp_curr_q, ORDER * sizeof(FLOAT32));
}
*num_fac_bits = 0;
if (first_lpd_flag) {
FLOAT32 *temp_speech = pstr_scratch->p_buf_speech;
FLOAT32 *temp_res = pstr_scratch->p_buf_res;
FLOAT32 lpc[9 * (ORDER + 1)];
FLOAT32 hp_mem[4];
FLOAT32 premph_mem = 0.0f;
WORD32 fac_length;
WORD32 num_bits_fac = (WORD32)((FLOAT32)num_bits_tcx / 2.f);
FLOAT32 *temp_signal = pstr_scratch->p_buf_signal;
if (st->last_was_short) {
fac_length = (st->len_frame) / 16;
} else {
fac_length = len_subfrm / 2;
}
iusace_get_interpolated_lpc(st->isp_old_q, st->isp_old_q, lpc, (2 * len_subfrm) / LEN_SUBFR);
memset(temp_speech, 0, (ORDER + 2 * len_subfrm) * sizeof(FLOAT32));
memset(temp_res, 0, (2 * len_subfrm) * sizeof(FLOAT32));
iusace_fac_apply(&st->fd_orig[1 + ORDER], len_subfrm, fac_length, st->low_pass_line,
num_bits_fac, &st->fd_synth[1 + ORDER], lpc, serial_fac_out, num_fac_bits,
pstr_scratch);
memset(hp_mem, 0, 4 * sizeof(FLOAT32));
iusace_highpass_50hz_12k8(st->fd_orig, 2 * len_subfrm + 1 + ORDER, hp_mem, pit_adj);
premph_mem = 0.0f;
iusace_apply_preemph(st->fd_orig, PREEMPH_FILT_FAC, 2 * len_subfrm + 1 + ORDER, &premph_mem);
memcpy(temp_signal, st->fd_orig + len_subfrm + 1, (len_subfrm + ORDER) * sizeof(FLOAT32));
premph_mem = temp_signal[0];
iusace_apply_deemph(temp_signal, PREEMPH_FILT_FAC, len_subfrm + ORDER, &premph_mem);
memcpy(st->lpd_state.tcx_mem, &temp_signal[len_subfrm + ORDER - 128], 128 * sizeof(FLOAT32));
premph_mem = 0.0f;
iusace_apply_preemph(st->fd_synth, PREEMPH_FILT_FAC, 2 * len_subfrm + 1 + ORDER, &premph_mem);
memcpy(st->lpd_state.synth, st->fd_synth + 2 * len_subfrm - ORDER - 128 + 1 + ORDER,
(ORDER + 128) * sizeof(FLOAT32));
memcpy(temp_speech + ORDER, st->fd_synth + 1 + ORDER, 2 * len_subfrm * sizeof(FLOAT32));
premph_mem = 0.0f;
iusace_find_weighted_speech(lpc, temp_speech + ORDER, temp_res, &premph_mem, 2 * len_subfrm);
st->prev_wsyn_mem = premph_mem;
memcpy(st->lpd_state.wsynth, temp_res + 2 * len_subfrm - ORDER - 128,
(ORDER + 128) * sizeof(FLOAT32));
memcpy(temp_speech + ORDER, st->fd_synth + 1 + ORDER, 2 * len_subfrm * sizeof(FLOAT32));
memset(temp_res, 0, 2 * len_subfrm * sizeof(FLOAT32));
for (i = 0; i < 2 * len_subfrm; i += LEN_SUBFR) {
iusace_compute_lp_residual(lpc, &temp_speech[ORDER + i], &temp_res[i], LEN_SUBFR);
}
memcpy(st->lpd_state.acelp_exc, temp_res, 2 * len_subfrm * sizeof(FLOAT32));
premph_mem = 0.0f;
iusace_find_weighted_speech(lp_filter_coeff, st->fd_orig + 1 + ORDER, temp_speech + ORDER,
&(st->mem_wsp), 2 * len_subfrm);
memcpy(st->weighted_sig, temp_speech + ORDER + 2 * len_subfrm - 128, 128 * sizeof(FLOAT32));
memcpy(pstr_scratch->p_wsig_buf, st->weighted_sig, 128 * sizeof(FLOAT32));
for (i = 0; i < 2 * len_subfrm; i += len_subfrm) {
iusace_decim2_fir_filter(&temp_speech[i + ORDER], len_subfrm, st->mem_lp_decim2,
pstr_scratch->p_fir_sig_buf);
memcpy(temp_speech + ORDER + i / OPL_DECIM, temp_speech + ORDER + i,
(len_subfrm / OPL_DECIM) * sizeof(FLOAT32));
}
memcpy(wsp_prev_buf, temp_speech + ORDER + 2 * len_subfrm / OPL_DECIM - MAX_PITCH / OPL_DECIM,
(WORD32)((MAX_PITCH / OPL_DECIM) * sizeof(FLOAT32)));
iusace_highpass_prev_wsp(st, temp_speech, pitch_max);
}
memcpy(isp_curr_q, st->isp_old_q, ORDER * sizeof(FLOAT32));
memcpy(st->isp_old_q, &isp_curr_q[NUM_FRAMES * ORDER], ORDER * sizeof(FLOAT32));
for (i = 0; i < NUM_FRAMES; i++) {
iusace_find_weighted_speech(
&lp_filter_coeff[i * (num_sbfrm_per_supfrm / NUM_FRAMES) * (ORDER + 1)],
&speech[i * len_subfrm], &pstr_scratch->p_wsig_buf[i * len_subfrm], &(st->mem_wsp),
len_subfrm);
memcpy(p_wsp_prev_buf, &pstr_scratch->p_wsig_buf[i * len_subfrm],
len_subfrm * sizeof(FLOAT32));
iusace_decim2_fir_filter(p_wsp_prev_buf, len_subfrm, st->mem_lp_decim2,
pstr_scratch->p_fir_sig_buf);
range_pitch_search = 2 * LEN_SUBFR;
if (num_subfrm < 4) {
range_pitch_search = 3 * LEN_SUBFR;
}
iusace_open_loop_search(p_wsp_prev_buf, (pitch_min / OPL_DECIM) + 1, pitch_max / OPL_DECIM,
range_pitch_search / OPL_DECIM, &ol_pitch_lag[i * 2], st);
if (st->ol_gain > 0.6) {
st->prev_pitch_med = iusace_get_ol_lag_median(ol_pitch_lag[i * 2], st->prev_ol_lags);
st->ada_w = 1.0;
} else {
st->ada_w = st->ada_w * 0.9f;
}
if (st->ada_w < 0.8) {
st->ol_wght_flg = 0;
} else {
st->ol_wght_flg = 1;
}
max_corr = 0.0f;
p = &p_wsp_prev_buf[0];
p1 = p_wsp_prev_buf - ol_pitch_lag[i * 2];
for (j = 0; j < range_pitch_search / OPL_DECIM; j++) {
max_corr += *p++ * *p1++;
}
t0 = 0.01f;
p = p_wsp_prev_buf - ol_pitch_lag[i * 2];
for (j = 0; j < range_pitch_search / OPL_DECIM; j++, p++) {
t0 += *p * *p;
}
t0 = (FLOAT32)(1.0 / sqrt(t0));
norm_corr[i * 2] = max_corr * t0;
energy = 0.01f;
for (j = 0; j < range_pitch_search / OPL_DECIM; j++) {
energy += p_wsp_prev_buf[j] * p_wsp_prev_buf[j];
}
energy = (FLOAT32)(1.0 / sqrt(energy));
norm_corr[i * 2] *= energy;
if (num_subfrm < 4) {
ol_pitch_lag[(i * 2) + 1] = ol_pitch_lag[i * 2];
norm_corr[(i * 2) + 1] = norm_corr[i * 2];
} else {
iusace_open_loop_search(p_wsp_prev_buf + ((2 * LEN_SUBFR) / OPL_DECIM),
(pitch_min / OPL_DECIM) + 1, pitch_max / OPL_DECIM,
(2 * LEN_SUBFR) / OPL_DECIM, &ol_pitch_lag[(i * 2) + 1], st);
if (st->ol_gain > 0.6) {
st->prev_pitch_med =
iusace_get_ol_lag_median(ol_pitch_lag[(i * 2) + 1], st->prev_ol_lags);
st->ada_w = 1.0;
} else {
st->ada_w = st->ada_w * 0.9f;
}
if (st->ada_w < 0.8) {
st->ol_wght_flg = 0;
} else {
st->ol_wght_flg = 1;
}
max_corr = 0.0f;
p = p_wsp_prev_buf + (2 * LEN_SUBFR) / OPL_DECIM;
p1 = p_wsp_prev_buf + ((2 * LEN_SUBFR) / OPL_DECIM) - ol_pitch_lag[(i * 2) + 1];
for (j = 0; j < (2 * LEN_SUBFR) / OPL_DECIM; j++) {
max_corr += *p++ * *p1++;
}
t0 = 0.01f;
p = p_wsp_prev_buf + ((2 * LEN_SUBFR) / OPL_DECIM) - ol_pitch_lag[(i * 2) + 1];
for (j = 0; j < (2 * LEN_SUBFR) / OPL_DECIM; j++, p++) {
t0 += *p * *p;
}
t0 = (FLOAT32)(1.0 / sqrt(t0));
norm_corr[(i * 2) + 1] = max_corr * t0;
energy = 0.01f;
for (j = 0; j < (2 * LEN_SUBFR) / OPL_DECIM; j++) {
energy += p_wsp_prev_buf[((2 * LEN_SUBFR) / OPL_DECIM) + j] *
p_wsp_prev_buf[((2 * LEN_SUBFR) / OPL_DECIM) + j];
}
energy = (FLOAT32)(1.0 / sqrt(energy));
norm_corr[(i * 2) + 1] *= energy;
}
memmove(wsp_prev_buf, &wsp_prev_buf[len_subfrm / OPL_DECIM],
(WORD32)((MAX_PITCH / OPL_DECIM) * sizeof(FLOAT32)));
}
memcpy(lpd_state[0], &st->lpd_state, sizeof(*lpd_state[0]));
ssnr_1024 = 0;
if (usac_data->use_acelp_only) {
for (i1 = 0; i1 < 2; i1++) {
for (i2 = 0; i2 < 2; i2++) {
k = (i1 * 2) + i2;
p_params = acelp_tcx_params + (k * MAX_NUM_TCX_PRM_PER_DIV);
iusace_interpolation_lsp_params(&isp_curr_q[k * ORDER], &isp_curr_q[(k + 1) * ORDER],
lp_filter_coeff_q, st->num_subfrm);
memcpy(lpd_state[k + 1], lpd_state[k], sizeof(*lpd_state[0]));
iusace_acelp_encode(
&lp_filter_coeff[k * (num_sbfrm_per_supfrm / 4) * (ORDER + 1)], lp_filter_coeff_q,
&speech[k * st->len_subfrm], &pstr_scratch->p_wsig_buf[k * st->len_subfrm],
&pstr_scratch->p_synth_buf[k * st->len_subfrm],
&pstr_scratch->p_wsyn_buf[k * st->len_subfrm], *codec_mode, lpd_state[k + 1],
st->len_subfrm, norm_corr[k * 2], norm_corr[(k * 2) + 1], ol_pitch_lag[k * 2],
ol_pitch_lag[(k * 2) + 1], pit_adj, p_params, pstr_scratch);
mem_wsyn = lpd_state[k]->mem_wsyn;
iusace_find_weighted_speech(&lp_filter_coeff[k * (num_sbfrm_per_supfrm / 4) *
(ORDER + 1)], &pstr_scratch->p_synth_buf[k * LEN_FRAME],
pstr_scratch->p_temp_wsyn_buf, &mem_wsyn, LEN_FRAME);
lpd_state[k + 1]->mem_wsyn = mem_wsyn;
mode[k] = 0;
num_tcx_param[k] = 0;
}
}
memcpy(&st->lpd_state, lpd_state[4], sizeof(*lpd_state[4]));
memcpy(st->weighted_sig, pstr_scratch->p_wsig_buf + (st->len_frame), 128 * sizeof(FLOAT32));
memcpy(st->prev_wsp, wsp_prev_buf, (len * sizeof(FLOAT32)));
return;
}
for (i1 = 0; i1 < 2; i1++) {
ssnr_512 = 0;
for (i2 = 0; i2 < 2; i2++) {
k = (i1 * 2) + i2;
p_params = acelp_tcx_params + (k * MAX_NUM_TCX_PRM_PER_DIV);
iusace_interpolation_lsp_params(&isp_curr_q[k * ORDER], &isp_curr_q[(k + 1) * ORDER],
lp_filter_coeff_q, st->num_subfrm);
memcpy(lpd_state[k + 1], lpd_state[k], sizeof(*lpd_state[0]));
iusace_acelp_encode(
&lp_filter_coeff[k * (num_sbfrm_per_supfrm / 4) * (ORDER + 1)], lp_filter_coeff_q,
&speech[k * st->len_subfrm], &pstr_scratch->p_wsig_buf[k * st->len_subfrm],
&pstr_scratch->p_synth_buf[k * st->len_subfrm],
&pstr_scratch->p_wsyn_buf[k * st->len_subfrm], *codec_mode, lpd_state[k + 1],
st->len_subfrm, norm_corr[k * 2], norm_corr[(k * 2) + 1], ol_pitch_lag[k * 2],
ol_pitch_lag[(k * 2) + 1], pit_adj, p_params, pstr_scratch);
mem_wsyn = lpd_state[k]->mem_wsyn;
iusace_find_weighted_speech(&lp_filter_coeff[k * (num_sbfrm_per_supfrm / 4) * (ORDER + 1)],
&pstr_scratch->p_synth_buf[k * LEN_FRAME],
pstr_scratch->p_temp_wsyn_buf, &mem_wsyn, LEN_FRAME);
lpd_state[k + 1]->mem_wsyn = mem_wsyn;
ssnr_256 = iusace_cal_segsnr(&pstr_scratch->p_wsig_buf[k * LEN_FRAME],
pstr_scratch->p_temp_wsyn_buf, LEN_FRAME, LEN_SUBFR);
mode[k] = 0;
num_tcx_param[k] = 0;
iusace_lpc_coef_gen(&isp_curr_q[k * ORDER], &isp_curr_q[(k + 1) * ORDER], lp_filter_coeff_q,
st->num_subfrm, ORDER);
memcpy(lpd_state_temp, lpd_state[k], sizeof(*lpd_state[0]));
iusace_tcx_fac_encode(usac_data,
&lp_filter_coeff[k * (num_sbfrm_per_supfrm / 4) * (ORDER + 1)],
lp_filter_coeff_q, &speech[k * st->len_subfrm], st->len_subfrm,
num_bits_tcx, lpd_state_temp, prm_tcx, &num_params, ch_idx, k);
mem_wsyn = lpd_state[k]->mem_wsyn;
iusace_find_weighted_speech(&lp_filter_coeff[k * (num_sbfrm_per_supfrm / 4) * (ORDER + 1)],
pstr_scratch->p_synth_tcx_buf, pstr_scratch->p_temp_wsyn_buf,
&mem_wsyn, LEN_FRAME);
lpd_state_temp->mem_wsyn = mem_wsyn;
tmp_ssnr = iusace_cal_segsnr(&pstr_scratch->p_wsig_buf[k * LEN_FRAME],
pstr_scratch->p_temp_wsyn_buf, LEN_FRAME, LEN_SUBFR);
if (tmp_ssnr > ssnr_256) {
ssnr_256 = tmp_ssnr;
mode[k] = 1;
num_tcx_param[k] = num_params;
memcpy(lpd_state[k + 1], lpd_state_temp, sizeof(*lpd_state[0]));
memcpy(&pstr_scratch->p_synth_buf[(k * st->len_subfrm) - 128],
pstr_scratch->p_synth_tcx_buf - 128, (st->len_subfrm + 128) * sizeof(FLOAT32));
memcpy(&pstr_scratch->p_wsyn_buf[(k * st->len_subfrm) - 128],
pstr_scratch->p_wsyn_tcx_buf - 128, (st->len_subfrm + 128) * sizeof(FLOAT32));
memcpy(p_params, prm_tcx, NUM_TCX20_PRM * sizeof(WORD32));
}
ssnr_512 += 0.50f * ssnr_256;
}
k = i1 * 2;
p_params = acelp_tcx_params + (k * MAX_NUM_TCX_PRM_PER_DIV);
iusace_lpc_coef_gen(&isp_curr_q[2 * i1 * ORDER], &isp_curr_q[(2 * i1 + 2) * ORDER],
lp_filter_coeff_q, (num_sbfrm_per_supfrm / 2), ORDER);
memcpy(lpd_state_temp, lpd_state[2 * i1], sizeof(*lpd_state[0]));
iusace_tcx_fac_encode(usac_data,
&lp_filter_coeff[2 * i1 * (num_sbfrm_per_supfrm / 4) * (ORDER + 1)],
lp_filter_coeff_q, &speech[2 * i1 * st->len_subfrm], 2 * st->len_subfrm,
2 * num_bits_tcx, lpd_state_temp, prm_tcx, &num_params, ch_idx, 2 * i1);
mem_wsyn = lpd_state[2 * i1]->mem_wsyn;
iusace_find_weighted_speech(
&lp_filter_coeff[2 * i1 * (num_sbfrm_per_supfrm / 4) * (ORDER + 1)],
pstr_scratch->p_synth_tcx_buf, pstr_scratch->p_temp_wsyn_buf, &mem_wsyn, LEN_FRAME * 2);
lpd_state_temp->mem_wsyn = mem_wsyn;
tmp_ssnr = iusace_cal_segsnr(&pstr_scratch->p_wsig_buf[2 * i1 * LEN_FRAME],
pstr_scratch->p_temp_wsyn_buf, LEN_FRAME * 2, LEN_SUBFR);
if (tmp_ssnr > ssnr_512) {
ssnr_512 = tmp_ssnr;
for (i = 0; i < 2; i++) {
mode[k + i] = 2;
num_tcx_param[k + i] = num_params;
}
memcpy(lpd_state[k + 2], lpd_state_temp, sizeof(*lpd_state[0]));
memcpy(&pstr_scratch->p_synth_buf[(2 * i1 * st->len_subfrm) - 128],
pstr_scratch->p_synth_tcx_buf - 128, ((2 * st->len_subfrm) + 128) * sizeof(FLOAT32));
memcpy(&pstr_scratch->p_wsyn_buf[(2 * i1 * st->len_subfrm) - 128],
pstr_scratch->p_wsyn_tcx_buf - 128, ((2 * st->len_subfrm) + 128) * sizeof(FLOAT32));
memcpy(p_params, prm_tcx, NUM_TCX40_PRM * sizeof(WORD32));
}
ssnr_1024 += 0.50f * ssnr_512;
}
k = 0;
p_params = acelp_tcx_params + (k * MAX_NUM_TCX_PRM_PER_DIV);
iusace_lpc_coef_gen(&isp_curr_q[k * ORDER], &isp_curr_q[(k + 4) * ORDER], lp_filter_coeff_q,
num_sbfrm_per_supfrm, ORDER);
memcpy(lpd_state_temp, lpd_state[k], sizeof(*lpd_state[0]));
iusace_tcx_fac_encode(usac_data, &lp_filter_coeff[k * (num_sbfrm_per_supfrm / 4) * (ORDER + 1)],
lp_filter_coeff_q, &speech[k * st->len_subfrm], 4 * st->len_subfrm,
4 * num_bits_tcx, lpd_state_temp, prm_tcx, &num_params, ch_idx, k);
mem_wsyn = lpd_state[k]->mem_wsyn;
iusace_find_weighted_speech(&lp_filter_coeff[k * (num_sbfrm_per_supfrm / 4) * (ORDER + 1)],
pstr_scratch->p_synth_tcx_buf, pstr_scratch->p_temp_wsyn_buf,
&mem_wsyn, LEN_FRAME * 4);
lpd_state_temp->mem_wsyn = mem_wsyn;
tmp_ssnr = iusace_cal_segsnr(&pstr_scratch->p_wsig_buf[k * LEN_FRAME],
pstr_scratch->p_temp_wsyn_buf, LEN_FRAME * 4, LEN_SUBFR);
if (tmp_ssnr > ssnr_1024) {
for (i = 0; i < 4; i++) {
mode[k + i] = 3;
num_tcx_param[k + i] = num_params;
}
memcpy(lpd_state[k + 4], lpd_state_temp, sizeof(*lpd_state[0]));
memcpy(&pstr_scratch->p_synth_buf[(k * st->len_subfrm) - 128],
pstr_scratch->p_synth_tcx_buf - 128, ((4 * st->len_subfrm) + 128) * sizeof(FLOAT32));
memcpy(&pstr_scratch->p_wsyn_buf[(k * st->len_subfrm) - 128],
pstr_scratch->p_wsyn_tcx_buf - 128, ((4 * st->len_subfrm) + 128) * sizeof(FLOAT32));
memcpy(p_params, prm_tcx, NUM_TCX80_PRM * sizeof(WORD32));
}
memcpy(&st->lpd_state, lpd_state[4], sizeof(*lpd_state[4]));
memcpy(st->weighted_sig, pstr_scratch->p_wsig_buf + (st->len_frame), 128 * sizeof(FLOAT32));
memcpy(st->prev_wsp, wsp_prev_buf, (len * sizeof(FLOAT32)));
return;
}
IA_ERRORCODE iusace_lpd_frm_enc(ia_usac_data_struct *usac_data, WORD32 *mod_out,
WORD32 const usac_independency_flg, WORD32 len_frame, WORD32 i_ch,
ia_bit_buf_struct *pstr_it_bit_buff) {
WORD32 i;
WORD32 len_next_high_rate = (LEN_NEXT_HIGH_RATE * len_frame) / LEN_SUPERFRAME;
WORD32 len_lpc0 = (LEN_LPC0 * len_frame) / LEN_SUPERFRAME;
FLOAT32 *input_data = &usac_data->td_in_buf[i_ch][len_next_high_rate];
ia_usac_td_encoder_struct *td_encoder = usac_data->td_encoder[i_ch];
WORD32 fscale = usac_data->td_encoder[i_ch]->fscale;
WORD32 first_lpd_flag = (usac_data->core_mode_prev[i_ch] == CORE_MODE_FD);
FLOAT32 *speech_buf = usac_data->speech_buf;
FLOAT32 *ptr_scratch_buf = usac_data->str_scratch.p_lpd_frm_enc_scratch;
FLOAT32 *speech, *new_speech;
WORD32 mode_buf[1 + NUM_FRAMES] = {0}, *mode;
WORD32 num_tcx_params[NUM_FRAMES] = {0};
WORD32 len_subfrm;
len_subfrm = td_encoder->len_subfrm;
if (usac_data->core_mode_prev[i_ch] == CORE_MODE_FD) {
iusace_reset_td_enc(usac_data->td_encoder[i_ch]);
FLOAT32 *in_data = usac_data->td_in_prev_buf[i_ch];
FLOAT32 *ptr_speech = usac_data->speech_buf;
WORD32 length = len_next_high_rate + len_lpc0;
ia_usac_td_encoder_struct *st = usac_data->td_encoder[i_ch];
memcpy(ptr_speech, in_data, length * sizeof(FLOAT32));
iusace_highpass_50hz_12k8(ptr_speech, length, st->mem_sig_in, st->fscale);
iusace_apply_preemph(ptr_speech, PREEMPH_FILT_FAC, length, &(st->mem_preemph));
memcpy(st->old_speech_pe + ORDER, ptr_speech, length * sizeof(FLOAT32));
}
if (first_lpd_flag) {
td_encoder->prev_mode = -1;
}
mode = mode_buf + 1;
mode[-1] = td_encoder->prev_mode;
fscale = (fscale * len_subfrm) / LEN_FRAME;
new_speech = speech_buf + ORDER + (LEN_NEXT_HIGH_RATE * len_subfrm) / LEN_FRAME;
speech = speech_buf + ORDER;
if (first_lpd_flag) {
new_speech += (LEN_LPC0 * len_subfrm) / LEN_FRAME;
speech += (LEN_LPC0 * len_subfrm) / LEN_FRAME;
}
memcpy(new_speech, input_data, td_encoder->len_frame * sizeof(FLOAT32));
iusace_highpass_50hz_12k8(new_speech, td_encoder->len_frame, td_encoder->mem_sig_in, fscale);
iusace_apply_preemph(new_speech, PREEMPH_FILT_FAC, td_encoder->len_frame,
&(td_encoder->mem_preemph));
if (first_lpd_flag) {
memcpy(speech_buf, td_encoder->old_speech_pe,
((ORDER + (((LEN_NEXT_HIGH_RATE + LEN_LPC0) * len_subfrm) / LEN_FRAME))) *
sizeof(FLOAT32));
for (i = 0; i < (len_subfrm + 1); i++) {
ptr_scratch_buf[i] = speech[-len_subfrm - 1 + i];
}
iusace_apply_deemph(ptr_scratch_buf, PREEMPH_FILT_FAC, len_subfrm + 1, &ptr_scratch_buf[0]);
memcpy(td_encoder->lpd_state.tcx_mem, &ptr_scratch_buf[len_subfrm - 128 + 1],
128 * sizeof(FLOAT32));
} else {
memcpy(speech_buf, td_encoder->old_speech_pe,
((ORDER + ((LEN_NEXT_HIGH_RATE * len_subfrm) / LEN_FRAME))) * sizeof(FLOAT32));
}
iusace_core_lpd_encode(usac_data, speech, mode, num_tcx_params, i_ch);
if (first_lpd_flag) {
memcpy(td_encoder->old_speech_pe,
&speech_buf[(td_encoder->len_frame) + (LEN_LPC0 * len_subfrm) / LEN_FRAME],
(ORDER + ((LEN_NEXT_HIGH_RATE * len_subfrm) / LEN_FRAME)) * sizeof(FLOAT32));
} else {
memcpy(td_encoder->old_speech_pe, &speech_buf[(td_encoder->len_frame)],
(ORDER + ((LEN_NEXT_HIGH_RATE * len_subfrm) / LEN_FRAME)) * sizeof(FLOAT32));
}
iusace_encode_fac_params(mode, num_tcx_params, usac_data, usac_independency_flg,
pstr_it_bit_buff, i_ch);
td_encoder->prev_mode = (WORD16)mode[3];
memcpy(mod_out, mode, 4 * sizeof(WORD32));
return IA_NO_ERROR;
}
Reference in a new issue View git blame Copy permalink