blob: 15115a275c8b29a31ff7ee19070f54e12b038738 [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 <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <string.h>
#include "impd_type_def.h"
#include "impd_drc_extr_delta_coded_info.h"
#include "impd_drc_common.h"
#include "impd_drc_struct.h"
#include "impd_drc_filter_bank.h"
#include "impd_drc_multi_band.h"
#include "impd_drc_gain_dec.h"
#include "impd_drc_process_audio.h"
WORD32 impd_apply_gains_and_add(
ia_drc_instructions_struct* pstr_drc_instruction_arr,
const WORD32 drc_instructions_index,
ia_drc_params_struct* ia_drc_params_struct,
ia_gain_buffer_struct* pstr_gain_buf,
shape_filter_block shape_filter_block[], FLOAT32* deinterleaved_audio[],
FLOAT32* channel_audio[], WORD32 impd_apply_gains) {
WORD32 c, b, g, i;
WORD32 offset = 0, signalIndex = 0;
WORD32 gainIndexForGroup[CHANNEL_GROUP_COUNT_MAX];
WORD32 signalIndexForChannel[MAX_CHANNEL_COUNT];
FLOAT32* lpcm_gains;
FLOAT32 sum;
FLOAT32 drc_gain_last, gainThr;
WORD32 iEnd, iStart;
ia_drc_instructions_struct* str_drc_instruction_str =
&(pstr_drc_instruction_arr[drc_instructions_index]);
if (drc_instructions_index >= 0) {
str_drc_instruction_str =
&(pstr_drc_instruction_arr[drc_instructions_index]);
{
if (str_drc_instruction_str->drc_set_id > 0) {
if (ia_drc_params_struct->delay_mode == DELAY_MODE_LOW_DELAY) {
offset = ia_drc_params_struct->drc_frame_size;
}
gainIndexForGroup[0] = 0;
for (g = 0; g < str_drc_instruction_str->num_drc_ch_groups - 1; g++) {
gainIndexForGroup[g + 1] =
gainIndexForGroup[g] +
str_drc_instruction_str->band_count_of_ch_group[g];
}
signalIndexForChannel[0] = 0;
for (c = 0; c < str_drc_instruction_str->audio_num_chan - 1; c++) {
if (str_drc_instruction_str->channel_group_of_ch[c] >= 0) {
signalIndexForChannel[c + 1] =
signalIndexForChannel[c] +
str_drc_instruction_str->band_count_of_ch_group
[str_drc_instruction_str->channel_group_of_ch[c]];
} else {
signalIndexForChannel[c + 1] = signalIndexForChannel[c] + 1;
}
}
for (g = 0; g < str_drc_instruction_str->num_drc_ch_groups; g++) {
for (b = 0; b < str_drc_instruction_str->band_count_of_ch_group[g];
b++) {
if (str_drc_instruction_str->ch_group_parametric_drc_flag[g] == 0) {
lpcm_gains =
pstr_gain_buf->buf_interpolation[gainIndexForGroup[g] + b]
.lpcm_gains +
MAX_SIGNAL_DELAY - ia_drc_params_struct->gain_delay_samples -
ia_drc_params_struct->audio_delay_samples + offset;
} else {
lpcm_gains =
pstr_gain_buf->buf_interpolation[gainIndexForGroup[g] + b]
.lpcm_gains +
MAX_SIGNAL_DELAY +
str_drc_instruction_str
->parametric_drc_look_ahead_samples[g] -
ia_drc_params_struct->audio_delay_samples;
}
iEnd = 0;
iStart = 0;
while (iEnd < ia_drc_params_struct->drc_frame_size) {
if (shape_filter_block[g].shape_flter_block_flag) {
drc_gain_last = shape_filter_block[g].drc_gain_last;
gainThr = 0.0001f * drc_gain_last;
while ((iEnd < ia_drc_params_struct->drc_frame_size) &&
(fabs(lpcm_gains[iEnd] - drc_gain_last) <= gainThr))
iEnd++;
} else {
iEnd = ia_drc_params_struct->drc_frame_size;
}
for (c = 0; c < str_drc_instruction_str->audio_num_chan; c++)
{
if (g == str_drc_instruction_str->channel_group_of_ch[c]) {
signalIndex = signalIndexForChannel[c] + b;
if (impd_apply_gains == 1) {
impd_shape_filt_block_time_process(
&shape_filter_block[g], &lpcm_gains[0], signalIndex,
&deinterleaved_audio[signalIndex][0], iStart, iEnd);
} else {
for (i = iStart; i < iEnd; i++) {
deinterleaved_audio[signalIndex][i] = lpcm_gains[i];
}
}
}
}
if ((iEnd < ia_drc_params_struct->drc_frame_size) &&
(shape_filter_block[g].shape_flter_block_flag)) {
impd_shape_filt_block_adapt(lpcm_gains[iEnd],
&shape_filter_block[g]);
}
iStart = iEnd;
}
}
}
}
}
}
signalIndex = 0;
if (str_drc_instruction_str->drc_set_id > 0) {
for (c = 0; c < str_drc_instruction_str->audio_num_chan; c++)
{
g = str_drc_instruction_str->channel_group_of_ch[c];
if (g >= 0) {
for (i = 0; i < ia_drc_params_struct->drc_frame_size; i++) {
sum = 0.0f;
for (b = 0; b < str_drc_instruction_str->band_count_of_ch_group[g];
b++) {
sum += deinterleaved_audio[signalIndex + b][i];
}
channel_audio[c][i] = sum;
}
signalIndex += str_drc_instruction_str->band_count_of_ch_group[g];
} else {
for (i = 0; i < ia_drc_params_struct->drc_frame_size; i++) {
channel_audio[c][i] = deinterleaved_audio[signalIndex][i];
}
signalIndex++;
}
}
} else {
for (c = 0; c < str_drc_instruction_str->audio_num_chan; c++)
{
for (i = 0; i < ia_drc_params_struct->drc_frame_size; i++) {
channel_audio[c][i] = deinterleaved_audio[c][i];
}
}
}
return (0);
}
/* subband-domain DRC: in-place application of DRC gains to audio frame */
WORD32
impd_apply_gains_subband(ia_drc_instructions_struct* pstr_drc_instruction_arr,
const WORD32 drc_instructions_index,
ia_drc_params_struct* ia_drc_params_struct,
ia_gain_buffer_struct* pstr_gain_buf,
ia_overlap_params_struct* pstr_overlap_params,
FLOAT32* deinterleaved_audio_delayed_re[],
FLOAT32* deinterleaved_audio_delayed_im[],
FLOAT32* deinterleaved_audio_re[],
FLOAT32* deinterleaved_audio_im[]) {
WORD32 c, b, g, m, s;
WORD32 gainIndexForGroup[CHANNEL_GROUP_COUNT_MAX];
FLOAT32* lpcm_gains;
FLOAT32 gainSb, gainLr;
ia_drc_instructions_struct* str_drc_instruction_str;
WORD32 offset = 0, signalIndex = 0;
WORD32 drc_frame_sizeSb = 0;
WORD32 nDecoderSubbands = 0;
WORD32 L = 0; /* L: downsampling factor */
WORD32 analysisDelay = 0;
switch (ia_drc_params_struct->sub_band_domain_mode) {
case SUBBAND_DOMAIN_MODE_QMF64:
nDecoderSubbands = AUDIO_CODEC_SUBBAND_COUNT_QMF64;
L = AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF64;
analysisDelay = AUDIO_CODEC_SUBBAND_ANALYSE_DELAY_QMF64;
break;
case SUBBAND_DOMAIN_MODE_QMF71:
nDecoderSubbands = AUDIO_CODEC_SUBBAND_COUNT_QMF71;
L = AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_QMF71;
analysisDelay = AUDIO_CODEC_SUBBAND_ANALYSE_DELAY_QMF71;
break;
case SUBBAND_DOMAIN_MODE_STFT256:
nDecoderSubbands = AUDIO_CODEC_SUBBAND_COUNT_STFT256;
L = AUDIO_CODEC_SUBBAND_DOWNSAMPLING_FACTOR_STFT256;
analysisDelay = AUDIO_CODEC_SUBBAND_ANALYSE_DELAY_STFT256;
break;
default:
return -1;
break;
}
drc_frame_sizeSb = ia_drc_params_struct->drc_frame_size / L;
if (drc_instructions_index >= 0) {
str_drc_instruction_str =
&(pstr_drc_instruction_arr[drc_instructions_index]);
{
if (str_drc_instruction_str->drc_set_id > 0) {
if (ia_drc_params_struct->delay_mode == DELAY_MODE_LOW_DELAY) {
offset = ia_drc_params_struct->drc_frame_size;
}
gainIndexForGroup[0] = 0;
for (g = 0; g < str_drc_instruction_str->num_drc_ch_groups - 1; g++) {
gainIndexForGroup[g + 1] =
gainIndexForGroup[g] +
str_drc_instruction_str
->band_count_of_ch_group[g]; /* index of first gain sequence
in channel group */
}
for (c = 0; c < str_drc_instruction_str->audio_num_chan; c++)
{
g = str_drc_instruction_str->channel_group_of_ch[c];
if (g >= 0) {
for (m = 0; m < drc_frame_sizeSb; m++) {
if (str_drc_instruction_str->band_count_of_ch_group[g] >
1) { /* multiband DRC */
for (s = 0; s < nDecoderSubbands; s++) {
gainSb = 0.0f;
for (b = 0;
b < str_drc_instruction_str->band_count_of_ch_group[g];
b++) {
if (str_drc_instruction_str
->ch_group_parametric_drc_flag[g] == 0) {
lpcm_gains =
pstr_gain_buf
->buf_interpolation[gainIndexForGroup[g] + b]
.lpcm_gains +
MAX_SIGNAL_DELAY -
ia_drc_params_struct->gain_delay_samples -
ia_drc_params_struct->audio_delay_samples + offset;
} else {
lpcm_gains =
pstr_gain_buf
->buf_interpolation[gainIndexForGroup[g] + b]
.lpcm_gains +
MAX_SIGNAL_DELAY +
str_drc_instruction_str
->parametric_drc_look_ahead_samples[g] -
ia_drc_params_struct->audio_delay_samples +
analysisDelay;
}
/* get gain for this timeslot by downsampling */
gainLr = lpcm_gains[(m * L + (L - 1) / 2)];
gainSb += pstr_overlap_params->str_group_overlap_params[g]
.str_band_overlap_params[b]
.overlap_weight[s] *
gainLr;
}
deinterleaved_audio_re[signalIndex][m * nDecoderSubbands +
s] =
gainSb *
deinterleaved_audio_delayed_re[signalIndex]
[m * nDecoderSubbands + s];
if (ia_drc_params_struct->sub_band_domain_mode ==
SUBBAND_DOMAIN_MODE_STFT256) { /* For STFT filterbank, the
real value of the
nyquist band is stored
at the imag value of the
first band */
if (s != 0)
deinterleaved_audio_im[signalIndex][m * nDecoderSubbands +
s] =
gainSb *
deinterleaved_audio_delayed_im[signalIndex]
[m * nDecoderSubbands +
s];
if (s == (nDecoderSubbands - 1))
deinterleaved_audio_im[signalIndex][m * nDecoderSubbands +
0] =
gainSb *
deinterleaved_audio_delayed_im[signalIndex]
[m * nDecoderSubbands +
0];
} else {
deinterleaved_audio_im[signalIndex][m * nDecoderSubbands +
s] =
gainSb *
deinterleaved_audio_delayed_im[signalIndex]
[m * nDecoderSubbands +
s];
}
}
} else { /* single-band DRC */
if (str_drc_instruction_str->ch_group_parametric_drc_flag[g] ==
0) {
lpcm_gains =
pstr_gain_buf->buf_interpolation[gainIndexForGroup[g]]
.lpcm_gains +
MAX_SIGNAL_DELAY -
ia_drc_params_struct->gain_delay_samples -
ia_drc_params_struct->audio_delay_samples + offset;
} else {
lpcm_gains =
pstr_gain_buf->buf_interpolation[gainIndexForGroup[g]]
.lpcm_gains +
MAX_SIGNAL_DELAY +
str_drc_instruction_str
->parametric_drc_look_ahead_samples[g] -
ia_drc_params_struct->audio_delay_samples + analysisDelay;
}
/* get gain for this timeslot by downsampling */
gainSb = lpcm_gains[(m * L + (L - 1) / 2)];
for (s = 0; s < nDecoderSubbands; s++) {
deinterleaved_audio_re[signalIndex][m * nDecoderSubbands +
s] =
gainSb *
deinterleaved_audio_delayed_re[signalIndex]
[m * nDecoderSubbands + s];
deinterleaved_audio_im[signalIndex][m * nDecoderSubbands +
s] =
gainSb *
deinterleaved_audio_delayed_im[signalIndex]
[m * nDecoderSubbands + s];
}
}
}
}
signalIndex++;
}
}
}
}
return (0);
}
WORD32
impd_filter_banks_process(ia_drc_instructions_struct* pstr_drc_instruction_arr,
const WORD32 drc_instructions_index,
ia_drc_params_struct* ia_drc_params_struct,
FLOAT32* audio_io_buf[],
ia_audio_band_buffer_struct* audio_band_buffer,
ia_filter_banks_struct* ia_filter_banks_struct,
const WORD32 passThru) {
WORD32 c, g, e, i, num_bands;
// WORD32 err = 0;
FLOAT32* audio_in;
FLOAT32** audio_out;
ia_drc_filter_bank_struct* str_drc_filter_bank;
ia_drc_instructions_struct* str_drc_instruction_str;
WORD32 drc_frame_size = ia_drc_params_struct->drc_frame_size;
if (drc_instructions_index >= 0) {
str_drc_instruction_str =
&(pstr_drc_instruction_arr[drc_instructions_index]);
} else {
return -1;
}
e = 0;
for (c = 0; c < str_drc_instruction_str->audio_num_chan; c++)
{
str_drc_filter_bank = NULL;
audio_in = audio_io_buf[c];
audio_out = &(audio_band_buffer->non_interleaved_audio[e]);
if ((passThru == 0) && (drc_instructions_index >= 0)) {
if (str_drc_instruction_str->drc_set_id < 0) {
num_bands = 1;
} else {
g = str_drc_instruction_str->channel_group_of_ch[c];
if (g == -1) {
num_bands = 1;
// if (ia_filter_banks_struct->str_drc_filter_bank != NULL)
//{
str_drc_filter_bank =
&(ia_filter_banks_struct->str_drc_filter_bank
[str_drc_instruction_str->num_drc_ch_groups]);
//}
} else {
num_bands = str_drc_instruction_str->band_count_of_ch_group[g];
// if (ia_filter_banks_struct->str_drc_filter_bank != NULL)
//{
str_drc_filter_bank =
&(ia_filter_banks_struct->str_drc_filter_bank[g]);
//}
}
// if (ia_filter_banks_struct->str_drc_filter_bank != NULL)
//{
// if (&str_drc_filter_bank->str_all_pass_cascade != NULL)
//{
impd_all_pass_cascade_process(
&str_drc_filter_bank->str_all_pass_cascade, c, drc_frame_size,
audio_in);
//}
//}
}
} else {
num_bands = 1;
}
switch (num_bands) {
case 1:
for (i = 0; i < drc_frame_size; i++) {
audio_out[0][i] = audio_in[i];
}
e++;
break;
case 2:
impd_two_band_filter_process(&str_drc_filter_bank->str_two_band_bank, c,
drc_frame_size, audio_in, audio_out);
e += 2;
break;
case 3:
impd_three_band_filter_process(
&str_drc_filter_bank->str_three_band_bank, c, drc_frame_size,
audio_in, audio_out);
e += 3;
break;
case 4:
impd_four_band_filter_process(&str_drc_filter_bank->str_four_band_bank,
c, drc_frame_size, audio_in, audio_out);
e += 4;
break;
default:
return (PARAM_ERROR);
break;
}
}
return (0);
}
WORD32
impd_store_audio_io_buffer_time(FLOAT32* audio_in_out_buf[],
ia_audio_in_out_buf* audio_io_buf_internal) {
WORD32 i, j;
if (audio_io_buf_internal->audio_delay_samples) {
for (i = 0; i < audio_io_buf_internal->audio_num_chan; i++) {
for (j = 0; j < audio_io_buf_internal->frame_size; j++) {
audio_io_buf_internal->audio_io_buffer_delayed
[i][audio_io_buf_internal->audio_delay_samples + j] =
audio_in_out_buf[i][j];
}
}
} else {
audio_io_buf_internal->audio_io_buffer_delayed = audio_in_out_buf;
audio_io_buf_internal->audio_in_out_buf = audio_in_out_buf;
}
return 0;
}
WORD32
impd_store_audio_io_buffer_freq(FLOAT32* audio_real_buff[],
FLOAT32* audio_imag_buff[],
ia_audio_in_out_buf* audio_io_buf_internal) {
WORD32 i, j;
if (audio_io_buf_internal->audio_delay_sub_band_samples) {
for (i = 0; i < audio_io_buf_internal->audio_num_chan; i++) {
for (j = 0; j < audio_io_buf_internal->audio_sub_band_frame_size *
audio_io_buf_internal->audio_sub_band_count;
j++) {
audio_io_buf_internal->audio_buffer_delayed_real
[i][audio_io_buf_internal->audio_delay_sub_band_samples *
audio_io_buf_internal->audio_sub_band_count +
j] = audio_real_buff[i][j];
audio_io_buf_internal->audio_buffer_delayed_imag
[i][audio_io_buf_internal->audio_delay_sub_band_samples *
audio_io_buf_internal->audio_sub_band_count +
j] = audio_imag_buff[i][j];
}
}
} else {
audio_io_buf_internal->audio_buffer_delayed_real = audio_real_buff;
audio_io_buf_internal->audio_buffer_delayed_imag = audio_imag_buff;
audio_io_buf_internal->audio_real_buff = audio_real_buff;
audio_io_buf_internal->audio_imag_buff = audio_imag_buff;
}
return 0;
}
WORD32
impd_retrieve_audio_io_buffer_time(FLOAT32* audio_in_out_buf[],
ia_audio_in_out_buf* audio_io_buf_internal) {
WORD32 i, j;
if (audio_io_buf_internal->audio_delay_samples) {
for (i = 0; i < audio_io_buf_internal->audio_num_chan; i++) {
for (j = 0; j < audio_io_buf_internal->frame_size; j++) {
audio_in_out_buf[i][j] =
audio_io_buf_internal->audio_io_buffer_delayed[i][j];
}
}
}
return 0;
}
WORD32
impd_retrieve_audio_buffer_freq(FLOAT32* audio_real_buff[],
FLOAT32* audio_imag_buff[],
ia_audio_in_out_buf* audio_io_buf_internal) {
WORD32 i, j;
if (audio_io_buf_internal->audio_delay_sub_band_samples) {
for (i = 0; i < audio_io_buf_internal->audio_num_chan; i++) {
for (j = 0; j < audio_io_buf_internal->audio_sub_band_frame_size *
audio_io_buf_internal->audio_sub_band_count;
j++) {
audio_real_buff[i][j] =
audio_io_buf_internal->audio_buffer_delayed_real
[i][audio_io_buf_internal->audio_sub_band_count + j];
audio_imag_buff[i][j] =
audio_io_buf_internal->audio_buffer_delayed_imag
[i][audio_io_buf_internal->audio_sub_band_count + j];
}
}
}
return 0;
}
WORD32
impd_advance_audio_io_buffer_time(ia_audio_in_out_buf* audio_io_buf_internal) {
WORD32 i;
if (audio_io_buf_internal->audio_delay_samples) {
for (i = 0; i < audio_io_buf_internal->audio_num_chan; i++) {
memmove(
audio_io_buf_internal->audio_io_buffer_delayed[i],
&audio_io_buf_internal
->audio_io_buffer_delayed[i][audio_io_buf_internal->frame_size],
sizeof(FLOAT32) * audio_io_buf_internal->audio_delay_samples);
}
}
return 0;
}
WORD32
impd_advance_audio_buff_freq(ia_audio_in_out_buf* audio_io_buf_internal) {
WORD32 i;
if (audio_io_buf_internal->audio_delay_sub_band_samples) {
for (i = 0; i < audio_io_buf_internal->audio_num_chan; i++) {
memmove(audio_io_buf_internal->audio_buffer_delayed_real[i],
&audio_io_buf_internal->audio_buffer_delayed_real
[i][audio_io_buf_internal->audio_sub_band_frame_size *
audio_io_buf_internal->audio_sub_band_count],
sizeof(FLOAT32) *
audio_io_buf_internal->audio_delay_sub_band_samples *
audio_io_buf_internal->audio_sub_band_count);
memmove(audio_io_buf_internal->audio_buffer_delayed_imag[i],
&audio_io_buf_internal->audio_buffer_delayed_imag
[i][audio_io_buf_internal->audio_sub_band_frame_size *
audio_io_buf_internal->audio_sub_band_count],
sizeof(FLOAT32) *
audio_io_buf_internal->audio_delay_sub_band_samples *
audio_io_buf_internal->audio_sub_band_count);
}
}
return 0;
}