blob: c268300116a68e61a0b87a957a83e3721b92a969 [file] [log] [blame]
/******************************************************************************
* *
* 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 <limits.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include "ixheaac_type_def.h"
#include "ixheaac_constants.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 "ixheaace_aac_constants.h"
#include "ixheaac_error_standards.h"
#include "ixheaace_error_codes.h"
#include "ixheaac_basic_ops32.h"
#include "ixheaac_basic_ops16.h"
#include "ixheaac_basic_ops40.h"
#include "ixheaac_basic_ops.h"
#include "ixheaace_psy_const.h"
#include "ixheaace_tns.h"
#include "ixheaace_tns_params.h"
#include "ixheaace_rom.h"
#include "ixheaace_common_rom.h"
#include "ixheaace_bitbuffer.h"
#include "ixheaace_bits_count.h"
#include "ixheaace_dynamic_bits.h"
#include "ixheaace_common_utils.h"
static WORD32 ia_enhaacplus_enc_calc_side_info_bits(WORD32 sfb_cnt, WORD32 block_type) {
WORD32 seg_len_bits = (block_type == SHORT_WINDOW ? 3 : 5);
WORD32 escape_val = (block_type == SHORT_WINDOW ? 7 : 31);
WORD32 side_info_bits, tmp;
side_info_bits = CODE_BCK_BITS;
tmp = sfb_cnt;
while (tmp >= 0) {
side_info_bits += seg_len_bits;
tmp -= escape_val;
}
return (side_info_bits);
}
static VOID ia_enhaacplus_enc_build_bit_look_up(
const WORD16 *ptr_quant_spec, const WORD32 max_sfb, const WORD32 *ptr_sfb_offset,
const UWORD16 *sfb_max,
WORD32 bit_look_up[MAXIMUM_SCALE_FACTOR_BAND_LONG][CODE_BCK_ESC_NDX + 1],
ixheaace_section_info *pstr_section_info, ixheaace_huffman_tables *pstr_huffman_tbl,
WORD32 aot) {
WORD8 i;
for (i = 0; i < max_sfb; i++) {
WORD32 sfb_width, max_val;
pstr_section_info[i].sfb_cnt = 1;
pstr_section_info[i].sfb_start = i;
switch (aot) {
case AOT_AAC_LC:
case AOT_SBR:
case AOT_PS:
pstr_section_info[i].section_bits = INVALID_BITCOUNT_LC;
break;
case AOT_AAC_LD:
case AOT_AAC_ELD:
pstr_section_info[i].section_bits = INVALID_BITCOUNT_LD;
break;
}
pstr_section_info[i].code_book = -1;
sfb_width = ptr_sfb_offset[i + 1] - ptr_sfb_offset[i];
max_val = sfb_max[i];
ia_enhaacplus_enc_bitcount(ptr_quant_spec + ptr_sfb_offset[i], sfb_width, max_val,
bit_look_up[i], pstr_huffman_tbl, aot);
}
}
static WORD32 ia_enhaacplus_enc_find_best_book(const WORD32 *ptr_bit_cnt, WORD8 *ptr_codebook,
WORD32 aot) {
WORD32 min_bits = 0, temp1, temp2;
WORD8 temp_book = CODE_BCK_ESC_NDX;
WORD8 j;
switch (aot) {
case AOT_AAC_LC:
case AOT_SBR:
case AOT_PS:
min_bits = INVALID_BITCOUNT_LC;
break;
case AOT_AAC_LD:
case AOT_AAC_ELD:
min_bits = INVALID_BITCOUNT_LD;
break;
}
for (j = 0; j <= CODE_BCK_ESC_NDX; j += 2) {
temp1 = *ptr_bit_cnt++;
temp2 = *ptr_bit_cnt++;
if (temp1 < min_bits) {
min_bits = temp1;
temp_book = j;
}
if (temp2 < min_bits) {
min_bits = temp2;
temp_book = j + 1;
}
}
*ptr_codebook = temp_book;
return min_bits;
}
static WORD32 ia_enhaacplus_enc_find_min_merge_bits(const WORD32 *ptr_bit_cnt1,
const WORD32 *ptr_bit_cnt2, WORD32 aot) {
WORD32 min_bits = 0, j, temp1, temp2, temp3, temp4;
switch (aot) {
case AOT_AAC_LC:
case AOT_SBR:
case AOT_PS:
min_bits = INVALID_BITCOUNT_LC;
break;
case AOT_AAC_LD:
case AOT_AAC_ELD:
min_bits = INVALID_BITCOUNT_LD;
break;
}
for (j = CODE_BCK_ESC_NDX; j >= 0; j -= 2) {
temp1 = *ptr_bit_cnt1++;
temp2 = *ptr_bit_cnt2++;
temp3 = *ptr_bit_cnt1++;
temp1 = temp1 + temp2;
min_bits = MIN(temp1, min_bits);
temp4 = *ptr_bit_cnt2++;
temp1 = temp3 + temp4;
min_bits = MIN(temp1, min_bits);
}
return min_bits;
}
static VOID ia_enhaacplus_enc_merge_bit_look_up(WORD32 *ptr_bit_cnt1, const WORD32 *ptr_bit_cnt2,
WORD32 aot) {
WORD32 j, temp1, temp2, temp3, temp4;
WORD32 invalid_bitcnt = 0;
switch (aot) {
case AOT_AAC_LC:
case AOT_SBR:
case AOT_PS:
invalid_bitcnt = INVALID_BITCOUNT_LC;
break;
case AOT_AAC_LD:
case AOT_AAC_ELD:
invalid_bitcnt = INVALID_BITCOUNT_LD;
break;
}
for (j = 0; j <= CODE_BCK_ESC_NDX; j += 2) {
temp1 = ptr_bit_cnt1[j];
temp2 = ptr_bit_cnt2[j];
temp3 = ptr_bit_cnt1[j + 1];
ptr_bit_cnt1[j] = MIN(temp1 + temp2, invalid_bitcnt);
temp4 = ptr_bit_cnt2[j + 1];
ptr_bit_cnt1[j + 1] = MIN(temp3 + temp4, invalid_bitcnt);
}
}
static WORD32 ia_enhaacplus_enc_find_max_merge(
const WORD32 merge_gain_look_up[MAXIMUM_SCALE_FACTOR_BAND_LONG],
const ixheaace_section_info *section, const WORD32 max_sfb, WORD32 *max_ndx) {
WORD32 i, max_merge_gain = 0;
for (i = 0; i + section[i].sfb_cnt < max_sfb; i += section[i].sfb_cnt) {
if (merge_gain_look_up[i] > max_merge_gain) {
max_merge_gain = merge_gain_look_up[i];
*max_ndx = i;
}
}
return (max_merge_gain);
}
static WORD32 ia_enhaacplus_enc_calc_merge_gain(
const ixheaace_section_info *pstr_section_info,
WORD32 bit_look_up[MAXIMUM_SCALE_FACTOR_BAND_LONG][CODE_BCK_ESC_NDX + 1],
const WORD32 *pstr_side_info_tab, const WORD32 idx1, const WORD32 idx2, WORD32 aot) {
WORD32 split_bits;
WORD32 merge_bits;
WORD32 merge_gain;
split_bits = pstr_section_info[idx1].section_bits + pstr_section_info[idx2].section_bits;
merge_bits =
pstr_side_info_tab[pstr_section_info[idx1].sfb_cnt + pstr_section_info[idx2].sfb_cnt] +
ia_enhaacplus_enc_find_min_merge_bits(bit_look_up[idx1], bit_look_up[idx2], aot);
merge_gain = split_bits - merge_bits;
return merge_gain;
}
static VOID ia_enhaacplus_enc_gm_stage0(
ixheaace_section_info *pstr_section,
WORD32 bit_look_up[MAXIMUM_SCALE_FACTOR_BAND_LONG][CODE_BCK_ESC_NDX + 1],
const WORD32 max_sfb, WORD32 aot) {
WORD32 i = 0;
WORD32 invalid_bitcnt = 0;
switch (aot) {
case AOT_AAC_LC:
case AOT_SBR:
case AOT_PS:
invalid_bitcnt = INVALID_BITCOUNT_LC;
break;
case AOT_AAC_LD:
case AOT_AAC_ELD:
invalid_bitcnt = INVALID_BITCOUNT_LD;
break;
}
while (i < max_sfb) {
if (pstr_section[i].section_bits == invalid_bitcnt) {
pstr_section[i].section_bits = (WORD16)ia_enhaacplus_enc_find_best_book(
bit_look_up[i], &(pstr_section[i].code_book), aot);
}
i++;
}
}
static VOID ia_enhaacplus_enc_gm_stage1(
ixheaace_section_info *pstr_section_info,
WORD32 bit_look_up[MAXIMUM_SCALE_FACTOR_BAND_LONG][CODE_BCK_ESC_NDX + 1],
const WORD32 max_sfb, const WORD32 *ptr_side_info_tab, WORD32 aot) {
WORD32 merge_start = 0, merge_end;
do {
for (merge_end = merge_start + 1; merge_end < max_sfb; merge_end++) {
if (pstr_section_info[merge_start].code_book != pstr_section_info[merge_end].code_book) {
break;
}
pstr_section_info[merge_start].sfb_cnt++;
pstr_section_info[merge_start].section_bits += pstr_section_info[merge_end].section_bits;
ia_enhaacplus_enc_merge_bit_look_up(bit_look_up[merge_start], bit_look_up[merge_end], aot);
}
pstr_section_info[merge_start].section_bits +=
(WORD16)ptr_side_info_tab[pstr_section_info[merge_start].sfb_cnt];
pstr_section_info[merge_end - 1].sfb_start = pstr_section_info[merge_start].sfb_start;
merge_start = merge_end;
} while (merge_start < max_sfb);
}
static VOID ia_enhaacplus_enc_gm_stage2(
ixheaace_section_info *pstr_section_info,
WORD32 merge_gain_look_up[MAXIMUM_SCALE_FACTOR_BAND_LONG],
WORD32 bit_look_up[MAXIMUM_SCALE_FACTOR_BAND_LONG][CODE_BCK_ESC_NDX + 1],
const WORD32 max_sfb, const WORD32 *ptr_side_info_tab, WORD32 aot) {
WORD32 i;
for (i = 0; i + pstr_section_info[i].sfb_cnt < max_sfb; i += pstr_section_info[i].sfb_cnt) {
merge_gain_look_up[i] =
ia_enhaacplus_enc_calc_merge_gain(pstr_section_info, bit_look_up, ptr_side_info_tab, i,
i + pstr_section_info[i].sfb_cnt, aot);
}
while (TRUE) {
WORD32 max_merge_gain = 0, max_idx = 0, max_idx_next = 0, max_idx_last = 0;
max_merge_gain = ia_enhaacplus_enc_find_max_merge(merge_gain_look_up, pstr_section_info,
max_sfb, &max_idx);
if (max_merge_gain <= 0) {
break;
}
max_idx_next = max_idx + pstr_section_info[max_idx].sfb_cnt;
pstr_section_info[max_idx].sfb_cnt += pstr_section_info[max_idx_next].sfb_cnt;
pstr_section_info[max_idx].section_bits +=
pstr_section_info[max_idx_next].section_bits - (WORD16)max_merge_gain;
ia_enhaacplus_enc_merge_bit_look_up(bit_look_up[max_idx], bit_look_up[max_idx_next], aot);
if (max_idx != 0) {
max_idx_last = pstr_section_info[max_idx - 1].sfb_start;
merge_gain_look_up[max_idx_last] = ia_enhaacplus_enc_calc_merge_gain(
pstr_section_info, bit_look_up, ptr_side_info_tab, max_idx_last, max_idx, aot);
}
max_idx_next = max_idx + pstr_section_info[max_idx].sfb_cnt;
pstr_section_info[max_idx_next - 1].sfb_start = pstr_section_info[max_idx].sfb_start;
if (max_idx_next < max_sfb) {
merge_gain_look_up[max_idx] = ia_enhaacplus_enc_calc_merge_gain(
pstr_section_info, bit_look_up, ptr_side_info_tab, max_idx, max_idx_next, aot);
}
}
}
static IA_ERRORCODE ia_enhaacplus_enc_noiseless_counter(
ixheaace_section_data *pstr_section_data,
WORD32 merge_gain_look_up[MAXIMUM_SCALE_FACTOR_BAND_LONG],
WORD32 bit_look_up[MAXIMUM_SCALE_FACTOR_BAND_LONG][CODE_BCK_ESC_NDX + 1],
const WORD16 *ptr_quant_spec, const UWORD16 *ptr_max_val_in_sfb, const WORD32 *ptr_sfb_offset,
const WORD32 block_type, WORD32 *ptr_side_info_tab_long, WORD32 *ptr_side_info_tab_short,
ixheaace_huffman_tables *pstr_huffman_tbl, WORD32 aot) {
WORD32 grp_idx;
WORD32 *ptr_side_info_tab = 0;
ixheaace_section_info *pstr_section_info;
WORD32 i;
for (i = 0; i < MAXIMUM_SCALE_FACTOR_BAND_LONG; i++) {
memset(bit_look_up[i], 0, (CODE_BCK_ESC_NDX + 1) * sizeof(bit_look_up[0][0]));
}
/* counting previous operations */
switch (block_type) {
case LONG_WINDOW:
case START_WINDOW:
case STOP_WINDOW:
ptr_side_info_tab = ptr_side_info_tab_long;
break;
case SHORT_WINDOW:
ptr_side_info_tab = ptr_side_info_tab_short;
break;
default:
return IA_EXHEAACE_EXE_FATAL_INVALID_BLOCK_TYPE;
}
pstr_section_data->num_of_sections = 0;
pstr_section_data->huffman_bits = 0;
pstr_section_data->side_info_bits = 0;
if (pstr_section_data->max_sfb_per_grp == 0) {
return IA_NO_ERROR;
}
for (grp_idx = 0; grp_idx < pstr_section_data->sfb_cnt;
grp_idx += pstr_section_data->sfb_per_group) {
pstr_section_info = pstr_section_data->section + pstr_section_data->num_of_sections;
ia_enhaacplus_enc_build_bit_look_up(ptr_quant_spec, pstr_section_data->max_sfb_per_grp,
ptr_sfb_offset + grp_idx, ptr_max_val_in_sfb + grp_idx,
bit_look_up, pstr_section_info, pstr_huffman_tbl, aot);
ia_enhaacplus_enc_gm_stage0(pstr_section_info, bit_look_up,
pstr_section_data->max_sfb_per_grp, aot);
ia_enhaacplus_enc_gm_stage1(pstr_section_info, bit_look_up,
pstr_section_data->max_sfb_per_grp, ptr_side_info_tab, aot);
ia_enhaacplus_enc_gm_stage2(pstr_section_info, merge_gain_look_up, bit_look_up,
pstr_section_data->max_sfb_per_grp, ptr_side_info_tab, aot);
for (i = 0; i < pstr_section_data->max_sfb_per_grp; i += pstr_section_info[i].sfb_cnt) {
ia_enhaacplus_enc_find_best_book(bit_look_up[i], &(pstr_section_info[i].code_book), aot);
pstr_section_info[i].sfb_start += (WORD8)grp_idx;
pstr_section_data->huffman_bits +=
pstr_section_info[i].section_bits - ptr_side_info_tab[pstr_section_info[i].sfb_cnt];
pstr_section_data->side_info_bits += ptr_side_info_tab[pstr_section_info[i].sfb_cnt];
pstr_section_data->section[pstr_section_data->num_of_sections++] = pstr_section_info[i];
}
}
return IA_NO_ERROR;
}
static WORD32 ia_enhaacplus_enc_bit_count_scalefactor_delta(
WORD32 delta, ixheaace_huffman_tables *pstr_huffman_tbl) {
return pstr_huffman_tbl->huff_ltabscf[delta + CODE_BCK_SCF_LAV];
}
static IA_ERRORCODE ia_enhaacplus_enc_scf_count(const WORD16 *ptr_scale_fac,
const UWORD16 *ptr_max_val_in_sfb,
ixheaace_section_data *pstr_section_data,
ixheaace_huffman_tables *pstr_huffman_tbl) {
WORD32 sect_idx1 = 0;
WORD32 sfb_idx1 = 0;
WORD32 scf_idx = 0;
WORD32 sect_idx2 = 0;
WORD32 sfb_idx2 = 0;
WORD32 last_val_scf = 0;
WORD32 delta_scf = 0;
WORD32 found = 0;
WORD32 scf_skip_counter = 0;
pstr_section_data->scale_fac_bits = 0;
if (ptr_scale_fac == 0) {
return IA_EXHEAACE_EXE_FATAL_INVALID_SCALE_FACTOR_GAIN;
}
last_val_scf = 0;
pstr_section_data->first_scf = 0;
for (sect_idx1 = 0; sect_idx1 < pstr_section_data->num_of_sections; sect_idx1++) {
if (pstr_section_data->section[sect_idx1].code_book != CODE_BCK_ZERO_NO) {
pstr_section_data->first_scf = pstr_section_data->section[sect_idx1].sfb_start;
last_val_scf = ptr_scale_fac[pstr_section_data->first_scf];
break;
}
}
for (sect_idx1 = 0; sect_idx1 < pstr_section_data->num_of_sections; sect_idx1++) {
if ((pstr_section_data->section[sect_idx1].code_book != CODE_BCK_ZERO_NO) &&
(pstr_section_data->section[sect_idx1].code_book != CODE_BCK_PNS_NO)) {
for (sfb_idx1 = pstr_section_data->section[sect_idx1].sfb_start;
sfb_idx1 < pstr_section_data->section[sect_idx1].sfb_start +
pstr_section_data->section[sect_idx1].sfb_cnt;
sfb_idx1++) {
if (ptr_max_val_in_sfb[sfb_idx1] == 0) {
found = 0;
if (scf_skip_counter == 0) {
if (sfb_idx1 == (pstr_section_data->section[sect_idx1].sfb_start +
pstr_section_data->section[sect_idx1].sfb_cnt - 1)) {
found = 0;
} else {
for (scf_idx = (sfb_idx1 + 1);
scf_idx < pstr_section_data->section[sect_idx1].sfb_start +
pstr_section_data->section[sect_idx1].sfb_cnt;
scf_idx++) {
if (ptr_max_val_in_sfb[scf_idx] != 0) {
found = 1;
if ((abs32(ptr_scale_fac[scf_idx] - last_val_scf)) < CODE_BCK_SCF_LAV) {
delta_scf = 0;
} else {
delta_scf = -(ptr_scale_fac[sfb_idx1] - last_val_scf);
last_val_scf = ptr_scale_fac[sfb_idx1];
scf_skip_counter = 0;
}
break;
}
/* count scalefactor skip */
scf_skip_counter = scf_skip_counter + 1;
}
}
/* search for the next ptr_max_val_in_sfb[] != 0 in all other sections */
for (sect_idx2 = (sect_idx1 + 1);
(sect_idx2 < pstr_section_data->num_of_sections) && (found == 0); sect_idx2++) {
if ((pstr_section_data->section[sect_idx2].code_book != CODE_BCK_ZERO_NO) &&
(pstr_section_data->section[sect_idx2].code_book != CODE_BCK_PNS_NO)) {
for (sfb_idx2 = pstr_section_data->section[sect_idx2].sfb_start;
sfb_idx2 < pstr_section_data->section[sect_idx2].sfb_start +
pstr_section_data->section[sect_idx2].sfb_cnt;
sfb_idx2++) {
if (ptr_max_val_in_sfb[sfb_idx2] != 0) {
found = 1;
if ((abs32(ptr_scale_fac[sfb_idx2] - last_val_scf)) < CODE_BCK_SCF_LAV) {
delta_scf = 0;
} else {
delta_scf = -(ptr_scale_fac[sfb_idx1] - last_val_scf);
last_val_scf = ptr_scale_fac[sfb_idx1];
scf_skip_counter = 0;
}
break;
}
scf_skip_counter = scf_skip_counter + 1;
}
}
}
if (found == 0) {
delta_scf = 0;
scf_skip_counter = 0;
}
} else {
delta_scf = 0;
scf_skip_counter = scf_skip_counter - 1;
}
} else {
delta_scf = -(ptr_scale_fac[sfb_idx1] - last_val_scf);
last_val_scf = ptr_scale_fac[sfb_idx1];
}
pstr_section_data->scale_fac_bits +=
ia_enhaacplus_enc_bit_count_scalefactor_delta(delta_scf, pstr_huffman_tbl);
}
}
}
return IA_NO_ERROR;
}
IA_ERRORCODE ia_enhaacplus_enc_dyn_bitcount(
const WORD16 *ptr_quant_spec, const UWORD16 *ptr_max_val_in_sfb, const WORD16 *ptr_scale_fac,
const WORD32 block_type, const WORD32 sfb_cnt, const WORD32 max_sfb_per_grp,
const WORD32 sfb_per_grp, const WORD32 *ptr_sfb_offset,
ixheaace_section_data *pstr_section_data, WORD32 *ptr_side_info_tab_long,
WORD32 *ptr_side_info_tab_short, ixheaace_huffman_tables *ptr_huffman_tbl,
WORD32 *ptr_scratch_buf, WORD32 aot, WORD32 *bit_cnt) {
IA_ERRORCODE err_code;
WORD32(*ptr_bit_look_up)
[CODE_BCK_ESC_NDX + 1] = (WORD32(*)[CODE_BCK_ESC_NDX + 1]) ptr_scratch_buf;
WORD32 *ptr_merge_gain_look_up =
ptr_scratch_buf + MAXIMUM_SCALE_FACTOR_BAND_LONG * (CODE_BCK_ESC_NDX + 1);
*bit_cnt = 0;
pstr_section_data->block_type = block_type;
pstr_section_data->sfb_cnt = sfb_cnt;
pstr_section_data->sfb_per_group = sfb_per_grp;
pstr_section_data->total_groups_cnt = sfb_cnt / sfb_per_grp;
pstr_section_data->max_sfb_per_grp = max_sfb_per_grp;
err_code = ia_enhaacplus_enc_noiseless_counter(
pstr_section_data, ptr_merge_gain_look_up, ptr_bit_look_up, ptr_quant_spec,
ptr_max_val_in_sfb, ptr_sfb_offset, block_type, ptr_side_info_tab_long,
ptr_side_info_tab_short, ptr_huffman_tbl, aot);
if (err_code != IA_NO_ERROR) {
return err_code;
}
err_code = ia_enhaacplus_enc_scf_count(ptr_scale_fac, ptr_max_val_in_sfb, pstr_section_data,
ptr_huffman_tbl);
if (err_code != IA_NO_ERROR) {
return err_code;
}
*bit_cnt = (pstr_section_data->huffman_bits + pstr_section_data->side_info_bits +
pstr_section_data->scale_fac_bits);
return IA_NO_ERROR;
}
VOID ia_enhaacplus_enc_bitcount_init(WORD32 *side_info_tab_long, WORD32 *side_info_tab_short) {
WORD32 i;
/* side_info_tab_long[] */
for (i = 0; i <= MAXIMUM_SCALE_FACTOR_BAND_LONG; i++) {
side_info_tab_long[i] = ia_enhaacplus_enc_calc_side_info_bits(i, LONG_WINDOW);
}
/* side_info_tab_short[] */
for (i = 0; i <= MAXIMUM_SCALE_FACTOR_BAND_SHORT; i++) {
side_info_tab_short[i] = ia_enhaacplus_enc_calc_side_info_bits(i, SHORT_WINDOW);
}
}