blob: 829f576ea166080aa99f075d4192b7640cbeec77 [file] [log] [blame]
/******************************************************************************
*
* 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 <math.h>
#include <string.h>
#include "impd_type_def.h"
#include "impd_error_standards.h"
#include "impd_drc_extr_delta_coded_info.h"
#include "impd_drc_common.h"
#include "impd_drc_struct.h"
#include "impd_parametric_drc_dec.h"
#include "impd_drc_gain_dec.h"
#include "impd_drc_filter_bank.h"
#include "impd_drc_multi_band.h"
#include "impd_drc_interface.h"
#include "impd_drc_gain_dec.h"
#include "impd_drc_eq.h"
#include "impd_drc_process_audio.h"
#include "impd_drc_gain_decoder.h"
#include "impd_drc_dec.h"
IA_ERRORCODE impd_init_drc_decode(
WORD32 frame_size, WORD32 sample_rate, WORD32 gain_delay_samples,
WORD32 delay_mode, WORD32 sub_band_domain_mode,
ia_drc_gain_dec_struct* p_drc_gain_dec_structs) {
IA_ERRORCODE err_code = IA_NO_ERROR;
err_code = impd_init_drc_params(
frame_size, sample_rate, gain_delay_samples, delay_mode,
sub_band_domain_mode, &p_drc_gain_dec_structs->ia_drc_params_struct);
if (err_code != IA_NO_ERROR) return (err_code);
impd_init_parametric_drc(
p_drc_gain_dec_structs->ia_drc_params_struct.drc_frame_size, sample_rate,
sub_band_domain_mode, &p_drc_gain_dec_structs->parametricdrc_params);
if (err_code != IA_NO_ERROR) return (err_code);
return err_code;
}
IA_ERRORCODE impd_init_drc_decode_post_config(
WORD32 audio_num_chan, WORD32* drc_set_id_processed,
WORD32* downmix_id_processed, WORD32 num_sets_processed,
WORD32 eq_set_id_processed,
ia_drc_gain_dec_struct* p_drc_gain_dec_structs,
ia_drc_config* pstr_drc_config,
ia_drc_loudness_info_set_struct* pstr_loudness_info, pVOID* mem_ptr) {
IA_ERRORCODE err_code = 0;
WORD32 i, j, k, maxMultibandAudioSignalCount = 0;
ia_drc_params_struct* p_drc_params_struct =
&p_drc_gain_dec_structs->ia_drc_params_struct;
ia_audio_in_out_buf* p_audio_in_out_buf =
&p_drc_gain_dec_structs->audio_in_out_buf;
for (i = 0; i < num_sets_processed; i++) {
err_code = impd_init_selected_drc_set(
pstr_drc_config, p_drc_params_struct,
&p_drc_gain_dec_structs->parametricdrc_params, audio_num_chan,
drc_set_id_processed[i], downmix_id_processed[i],
&p_drc_gain_dec_structs->ia_filter_banks_struct,
&p_drc_gain_dec_structs->str_overlap_params,
p_drc_gain_dec_structs->shape_filter_block);
if (err_code) return (err_code);
}
p_drc_gain_dec_structs->audio_num_chan = audio_num_chan;
p_drc_gain_dec_structs->ia_drc_params_struct.audio_delay_samples =
p_drc_gain_dec_structs->ia_drc_params_struct.parametric_drc_delay;
if (pstr_drc_config->str_drc_config_ext.parametric_drc_present) {
err_code = impd_init_parametric_drc_after_config(
pstr_drc_config, pstr_loudness_info,
&p_drc_gain_dec_structs->parametricdrc_params, mem_ptr);
if (err_code) return (err_code);
}
p_audio_in_out_buf->audio_num_chan = audio_num_chan;
p_audio_in_out_buf->audio_delay_samples =
p_drc_params_struct->audio_delay_samples;
p_audio_in_out_buf->frame_size = p_drc_params_struct->drc_frame_size;
if (p_drc_params_struct->sub_band_domain_mode == SUBBAND_DOMAIN_MODE_QMF64) {
p_audio_in_out_buf->audio_delay_sub_band_samples =
p_drc_params_struct->audio_delay_samples /
AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF64;
p_audio_in_out_buf->audio_sub_band_frame_size =
p_drc_params_struct->drc_frame_size /
AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF64;
p_audio_in_out_buf->audio_sub_band_count = AUDIO_CODEC_SUBBAND_COUNT_QMF64;
} else if (p_drc_params_struct->sub_band_domain_mode ==
SUBBAND_DOMAIN_MODE_QMF71) {
p_audio_in_out_buf->audio_delay_sub_band_samples =
p_drc_params_struct->audio_delay_samples /
AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF71;
p_audio_in_out_buf->audio_sub_band_frame_size =
p_drc_params_struct->drc_frame_size /
AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF71;
p_audio_in_out_buf->audio_sub_band_count = AUDIO_CODEC_SUBBAND_COUNT_QMF71;
} else if (p_drc_params_struct->sub_band_domain_mode ==
SUBBAND_DOMAIN_MODE_STFT256) {
p_audio_in_out_buf->audio_delay_sub_band_samples =
p_drc_params_struct->audio_delay_samples /
AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_STFT256;
p_audio_in_out_buf->audio_sub_band_frame_size =
p_drc_params_struct->drc_frame_size /
AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_STFT256;
p_audio_in_out_buf->audio_sub_band_count =
AUDIO_CODEC_SUBBAND_COUNT_STFT256;
} else {
p_audio_in_out_buf->audio_delay_sub_band_samples = 0;
p_audio_in_out_buf->audio_sub_band_frame_size = 0;
p_audio_in_out_buf->audio_sub_band_count = 0;
}
for (k = 0; k < SEL_DRC_COUNT; k++) {
if (p_drc_params_struct->sel_drc_array[k].drc_instructions_index >= 0) {
ia_drc_instructions_struct* drc_instruction_str =
&(pstr_drc_config->str_drc_instruction_str
[p_drc_params_struct->sel_drc_array[k].drc_instructions_index]);
if (drc_instruction_str->gain_element_count > 0) {
p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
.buf_interpolation = (ia_interp_buf_struct*)*mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
drc_instruction_str->gain_element_count *
sizeof(ia_interp_buf_struct) +
32);
p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
.buf_interpolation_count = drc_instruction_str->gain_element_count;
for (i = 0;
i < p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
.buf_interpolation_count;
i++) {
p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
.buf_interpolation[i]
.str_node.time = 0;
p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
.buf_interpolation[i]
.prev_node.time = -1;
p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
.buf_interpolation[i]
.str_node.loc_db_gain = 0.0f;
p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
.buf_interpolation[i]
.str_node.slope = 0.0f;
for (j = 0; j < 2 * AUDIO_CODEC_FRAME_SIZE_MAX + MAX_SIGNAL_DELAY;
j++) {
p_drc_gain_dec_structs->drc_gain_buffers.pstr_gain_buf[k]
.buf_interpolation[i]
.lpcm_gains[j] = 1.f;
}
}
}
}
}
if (eq_set_id_processed > 0) {
for (i = 0; i < pstr_drc_config->str_drc_config_ext.eq_instructions_count;
i++) {
if (pstr_drc_config->str_drc_config_ext.str_eq_instructions[i]
.eq_set_id == eq_set_id_processed)
break;
}
if (i == pstr_drc_config->str_drc_config_ext.eq_instructions_count) {
return -1;
}
p_drc_gain_dec_structs->eq_set = (ia_eq_set_struct*)*mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr + sizeof(ia_eq_set_struct));
if (err_code) return (err_code);
err_code = impd_derive_eq_set(
&pstr_drc_config->str_drc_config_ext.str_eq_coeff,
&(pstr_drc_config->str_drc_config_ext.str_eq_instructions[i]),
(FLOAT32)p_drc_gain_dec_structs->ia_drc_params_struct.sample_rate,
p_drc_gain_dec_structs->ia_drc_params_struct.drc_frame_size,
p_drc_gain_dec_structs->ia_drc_params_struct.sub_band_domain_mode,
p_drc_gain_dec_structs->eq_set);
if (err_code) return (err_code);
impd_get_eq_set_delay(
p_drc_gain_dec_structs->eq_set,
&p_drc_gain_dec_structs->ia_drc_params_struct.eq_delay);
}
for (i = 0; i < p_drc_params_struct->drc_set_counter; i++) {
ia_drc_instructions_struct* drc_instruction_str;
drc_instruction_str =
&(pstr_drc_config->str_drc_instruction_str
[p_drc_params_struct->sel_drc_array[i].drc_instructions_index]);
maxMultibandAudioSignalCount =
max(maxMultibandAudioSignalCount,
drc_instruction_str->multiband_audio_sig_count);
}
p_drc_gain_dec_structs->audio_band_buffer.non_interleaved_audio = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
(maxMultibandAudioSignalCount * sizeof(FLOAT32*)));
for (i = 0; i < maxMultibandAudioSignalCount; i++) {
p_drc_gain_dec_structs->audio_band_buffer.non_interleaved_audio[i] =
*mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
(p_drc_params_struct->drc_frame_size * sizeof(FLOAT32)));
}
p_drc_gain_dec_structs->audio_band_buffer.multiband_audio_sig_count =
maxMultibandAudioSignalCount;
p_drc_gain_dec_structs->audio_band_buffer.frame_size =
p_drc_params_struct->drc_frame_size;
;
if (p_drc_params_struct->sub_band_domain_mode == SUBBAND_DOMAIN_MODE_OFF &&
p_audio_in_out_buf->audio_delay_samples) {
p_audio_in_out_buf->audio_io_buffer_delayed = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
(p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
p_audio_in_out_buf->audio_in_out_buf = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
(p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
for (i = 0; i < p_audio_in_out_buf->audio_num_chan; i++) {
p_audio_in_out_buf->audio_io_buffer_delayed[i] = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
((p_audio_in_out_buf->frame_size +
p_audio_in_out_buf->audio_delay_samples) *
sizeof(FLOAT32*)));
p_audio_in_out_buf->audio_in_out_buf[i] =
&p_audio_in_out_buf->audio_io_buffer_delayed
[i][p_audio_in_out_buf->audio_delay_samples];
}
}
if (p_drc_params_struct->sub_band_domain_mode != SUBBAND_DOMAIN_MODE_OFF &&
p_audio_in_out_buf->audio_delay_sub_band_samples) {
p_audio_in_out_buf->audio_buffer_delayed_real = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
(p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
p_audio_in_out_buf->audio_buffer_delayed_imag = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
(p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
p_audio_in_out_buf->audio_real_buff = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
(p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
p_audio_in_out_buf->audio_imag_buff = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
(p_audio_in_out_buf->audio_num_chan * sizeof(FLOAT32*)));
for (i = 0; i < p_audio_in_out_buf->audio_num_chan; i++) {
p_audio_in_out_buf->audio_buffer_delayed_real[i] = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
((p_audio_in_out_buf->audio_sub_band_frame_size +
p_audio_in_out_buf->audio_delay_sub_band_samples) *
sizeof(FLOAT32*)));
p_audio_in_out_buf->audio_buffer_delayed_imag[i] = *mem_ptr;
*mem_ptr = (pVOID)((SIZE_T)*mem_ptr +
((p_audio_in_out_buf->audio_sub_band_frame_size +
p_audio_in_out_buf->audio_delay_sub_band_samples) *
sizeof(FLOAT32*)));
p_audio_in_out_buf->audio_real_buff[i] =
&p_audio_in_out_buf->audio_buffer_delayed_real
[i][p_audio_in_out_buf->audio_delay_sub_band_samples *
p_audio_in_out_buf->audio_sub_band_count];
p_audio_in_out_buf->audio_imag_buff[i] =
&p_audio_in_out_buf->audio_buffer_delayed_imag
[i][p_audio_in_out_buf->audio_delay_sub_band_samples *
p_audio_in_out_buf->audio_sub_band_count];
}
}
return err_code;
}
IA_ERRORCODE impd_drc_process_time_domain(
ia_drc_gain_dec_struct* p_drc_gain_dec_structs,
ia_drc_config* pstr_drc_config, ia_drc_gain_struct* pstr_drc_gain,
FLOAT32* audio_in_out_buf[], FLOAT32 loudness_normalization_gain_db,
FLOAT32 boost, FLOAT32 compress, WORD32 drc_characteristic_target) {
WORD32 sel_drc_index;
IA_ERRORCODE err_code = 0;
WORD32 passThru;
ia_drc_instructions_struct* str_drc_instruction_str =
pstr_drc_config->str_drc_instruction_str;
if (p_drc_gain_dec_structs->eq_set) {
WORD32 ch;
FLOAT32* audio_channel;
for (ch = 0; ch < p_drc_gain_dec_structs->eq_set->audio_num_chan; ch++) {
audio_channel = audio_in_out_buf[ch];
impd_process_eq_set_time_domain(
p_drc_gain_dec_structs->eq_set, ch, audio_channel, audio_channel,
p_drc_gain_dec_structs->ia_drc_params_struct.drc_frame_size);
}
}
err_code = impd_store_audio_io_buffer_time(
audio_in_out_buf, &p_drc_gain_dec_structs->audio_in_out_buf);
if (err_code != IA_NO_ERROR) return (err_code);
if (pstr_drc_config->apply_drc) {
for (sel_drc_index = 0;
sel_drc_index <
p_drc_gain_dec_structs->ia_drc_params_struct.drc_set_counter;
sel_drc_index++) {
err_code = impd_get_drc_gain(
p_drc_gain_dec_structs, pstr_drc_config, pstr_drc_gain, compress,
boost, drc_characteristic_target, loudness_normalization_gain_db,
sel_drc_index, &p_drc_gain_dec_structs->drc_gain_buffers);
if (err_code != IA_NO_ERROR) return (err_code);
}
if (p_drc_gain_dec_structs->ia_drc_params_struct.drc_set_counter == 0) {
err_code = impd_retrieve_audio_io_buffer_time(
audio_in_out_buf, &p_drc_gain_dec_structs->audio_in_out_buf);
if (err_code) return (err_code);
} else {
for (sel_drc_index = 0;
sel_drc_index <
p_drc_gain_dec_structs->ia_drc_params_struct.drc_set_counter;
sel_drc_index++) {
if (p_drc_gain_dec_structs->ia_drc_params_struct
.multiband_sel_drc_idx == sel_drc_index) {
passThru = 0;
} else {
passThru = 1;
}
err_code = impd_filter_banks_process(
str_drc_instruction_str,
p_drc_gain_dec_structs->ia_drc_params_struct
.sel_drc_array[sel_drc_index]
.drc_instructions_index,
&p_drc_gain_dec_structs->ia_drc_params_struct,
p_drc_gain_dec_structs->audio_in_out_buf.audio_io_buffer_delayed,
&p_drc_gain_dec_structs->audio_band_buffer,
&p_drc_gain_dec_structs->ia_filter_banks_struct, passThru);
if (err_code != IA_NO_ERROR) return (err_code);
err_code = impd_apply_gains_and_add(
str_drc_instruction_str,
p_drc_gain_dec_structs->ia_drc_params_struct
.sel_drc_array[sel_drc_index]
.drc_instructions_index,
&p_drc_gain_dec_structs->ia_drc_params_struct,
&(p_drc_gain_dec_structs->drc_gain_buffers
.pstr_gain_buf[sel_drc_index]),
p_drc_gain_dec_structs->shape_filter_block,
p_drc_gain_dec_structs->audio_band_buffer.non_interleaved_audio,
audio_in_out_buf, 1);
if (err_code != IA_NO_ERROR) return (err_code);
}
}
}
err_code = impd_advance_audio_io_buffer_time(
&p_drc_gain_dec_structs->audio_in_out_buf);
if (err_code != IA_NO_ERROR) return (err_code);
return err_code;
}
IA_ERRORCODE impd_drc_process_freq_domain(
ia_drc_gain_dec_struct* p_drc_gain_dec_structs,
ia_drc_config* pstr_drc_config, ia_drc_gain_struct* pstr_drc_gain,
FLOAT32* audio_real_buff[], FLOAT32* audio_imag_buff[],
FLOAT32 loudness_normalization_gain_db, FLOAT32 boost, FLOAT32 compress,
WORD32 drc_characteristic_target) {
WORD32 sel_drc_index;
IA_ERRORCODE err_code = 0;
ia_drc_instructions_struct* str_drc_instruction_str =
pstr_drc_config->str_drc_instruction_str;
if (p_drc_gain_dec_structs->eq_set) {
WORD32 ch;
for (ch = 0; ch < p_drc_gain_dec_structs->eq_set->audio_num_chan; ch++) {
err_code = impd_process_eq_set_subband_domain(
p_drc_gain_dec_structs->eq_set, ch, audio_real_buff[ch],
audio_imag_buff[ch]);
if (err_code != IA_NO_ERROR) return (err_code);
}
}
err_code = impd_store_audio_io_buffer_freq(
audio_real_buff, audio_imag_buff,
&p_drc_gain_dec_structs->audio_in_out_buf);
if (err_code != IA_NO_ERROR) return (err_code);
if (pstr_drc_config->apply_drc) {
for (sel_drc_index = 0;
sel_drc_index <
p_drc_gain_dec_structs->ia_drc_params_struct.drc_set_counter;
sel_drc_index++) {
err_code = impd_get_drc_gain(
p_drc_gain_dec_structs, pstr_drc_config, pstr_drc_gain, compress,
boost, drc_characteristic_target, loudness_normalization_gain_db,
sel_drc_index, &p_drc_gain_dec_structs->drc_gain_buffers);
if (err_code != IA_NO_ERROR) return (err_code);
}
if (p_drc_gain_dec_structs->ia_drc_params_struct.drc_set_counter == 0) {
err_code = impd_retrieve_audio_buffer_freq(
audio_real_buff, audio_imag_buff,
&p_drc_gain_dec_structs->audio_in_out_buf);
if (err_code != IA_NO_ERROR) return (err_code);
} else {
for (sel_drc_index = 0;
sel_drc_index <
p_drc_gain_dec_structs->ia_drc_params_struct.drc_set_counter;
sel_drc_index++) {
err_code = impd_apply_gains_subband(
str_drc_instruction_str,
p_drc_gain_dec_structs->ia_drc_params_struct
.sel_drc_array[sel_drc_index]
.drc_instructions_index,
&p_drc_gain_dec_structs->ia_drc_params_struct,
&(p_drc_gain_dec_structs->drc_gain_buffers
.pstr_gain_buf[sel_drc_index]),
&p_drc_gain_dec_structs->str_overlap_params,
p_drc_gain_dec_structs->audio_in_out_buf.audio_buffer_delayed_real,
p_drc_gain_dec_structs->audio_in_out_buf.audio_buffer_delayed_imag,
audio_real_buff, audio_imag_buff);
if (err_code != IA_NO_ERROR) return (err_code);
}
}
}
err_code =
impd_advance_audio_buff_freq(&p_drc_gain_dec_structs->audio_in_out_buf);
return err_code;
}
VOID impd_get_parametric_drc_delay(
ia_drc_gain_dec_struct* p_drc_gain_dec_structs,
ia_drc_config* pstr_drc_config, WORD32* parametric_drc_delay,
WORD32* parametric_drc_delay_max) {
*parametric_drc_delay =
p_drc_gain_dec_structs->ia_drc_params_struct.parametric_drc_delay;
if (pstr_drc_config->str_drc_config_ext.parametric_drc_present &&
pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
.parametric_drc_delay_max_present) {
*parametric_drc_delay_max =
pstr_drc_config->str_drc_config_ext.str_drc_coeff_param_drc
.parametric_drc_delay_max;
} else if (pstr_drc_config->str_drc_config_ext.parametric_drc_present == 0) {
*parametric_drc_delay_max = 0;
} else {
*parametric_drc_delay_max = -1;
}
return;
}
VOID impd_get_eq_delay(ia_drc_gain_dec_struct* p_drc_gain_dec_structs,
ia_drc_config* pstr_drc_config, WORD32* eq_delay,
WORD32* eq_delay_max) {
*eq_delay = p_drc_gain_dec_structs->ia_drc_params_struct.eq_delay;
if (pstr_drc_config->str_drc_config_ext.eq_flag &&
pstr_drc_config->str_drc_config_ext.str_eq_coeff.eq_delay_max_present) {
*eq_delay_max =
pstr_drc_config->str_drc_config_ext.str_eq_coeff.eq_delay_max;
} else if (pstr_drc_config->str_drc_config_ext.eq_flag == 0) {
*eq_delay_max = 0;
} else {
*eq_delay_max = -1;
}
return;
}