xref: /aosp_15_r20/external/libxaac/encoder/iusace_esbr_inter_tes.h (revision 15dc779a375ca8b5125643b829a8aa4b70d7f451)
1 /******************************************************************************
2  *                                                                            *
3  * Copyright (C) 2023 The Android Open Source Project
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at:
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *****************************************************************************
18  * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19  */
20 
21 #pragma once
22 
23 #define IXHEAACE_ESBR_TES_ENERGY_MAX_THR (1.0e6f)
24 #define IXHEAACE_TIMESLOT_BUFFER_SIZE (78)
25 #define IXHEAACE_MAX_ENVELOPES (8)
26 #define IXHEAACE_MAX_NOISE_ENVELOPES (2)
27 #define IXHEAACE_MAX_NOISE_COEFFS (5)
28 #define IXHEAACE_MAX_NUM_NOISE_VALUES (IXHEAACE_MAX_NOISE_ENVELOPES * IXHEAACE_MAX_NOISE_COEFFS)
29 #define IXHEAACE_MAX_FREQ_COEFFS (56)
30 #define IXHEAACE_MAX_NUM_PATCHES (6)
31 #define IXHEAACE_MAX_NUM_LIMITERS (12)
32 #define IXHEAACE_MAXDEG (3)
33 #define IXHEAACE_SBR_HF_RELAXATION_PARAM (0.999999f)
34 #define IXHEAACE_ESBR_NUM_GAMMA_IDXS (4)
35 #define IXHEAACE_ESBR_HBE_DELAY_OFFSET (32)
36 #define IXHEAACE_SBR_HF_ADJ_OFFSET (2)
37 #define IXHEAACE_SBR_TES_SHAPE_BITS (1)
38 #define IXHEAACE_SBR_TES_SHAPE_MODE_BITS (2)
39 
40 typedef struct {
41   FLOAT32 phi_0_1_real;
42   FLOAT32 phi_0_1_imag;
43   FLOAT32 phi_0_2_real;
44   FLOAT32 phi_0_2_imag;
45   FLOAT32 phi_1_1;
46   FLOAT32 phi_1_2_real;
47   FLOAT32 phi_1_2_imag;
48   FLOAT32 phi_2_2;
49   FLOAT32 det;
50 } ixheaace_str_auto_corr_ele;
51 
52 typedef struct {
53   WORD16 num_sf_bands[2];
54   WORD16 num_nf_bands;
55   WORD16 num_mf_bands;
56   WORD16 sub_band_start;
57   WORD16 sub_band_end;
58   WORD16 freq_band_tbl_lim[IXHEAACE_MAX_NUM_LIMITERS + 1];
59   WORD16 num_lf_bands;
60   WORD16 num_if_bands;
61   WORD16 *ptr_freq_band_tab[2];
62   WORD16 freq_band_tbl_lo[IXHEAACE_MAX_FREQ_COEFFS / 2 + 1];
63   WORD16 freq_band_tbl_hi[IXHEAACE_MAX_FREQ_COEFFS + 1];
64   WORD16 freq_band_tbl_noise[IXHEAACE_MAX_NOISE_COEFFS + 1];
65   WORD16 f_master_tbl[IXHEAACE_MAX_FREQ_COEFFS + 1];
66   WORD16 qmf_sb_prev;
67 } ia_str_freq_band_data;
68 
69 typedef struct {
70   WORD16 frame_class;
71   WORD16 num_env;
72   WORD16 transient_env;
73   WORD16 num_noise_env;
74   WORD16 border_vec[IXHEAACE_MAX_ENVELOPES + 1];
75   WORD16 freq_res[IXHEAACE_MAX_ENVELOPES];
76   WORD16 noise_border_vec[IXHEAACE_MAX_NOISE_ENVELOPES + 1];
77 } ia_str_frame_info;
78 
79 typedef struct {
80   WORD32 num_if_bands;
81   WORD32 sub_band_start;
82   WORD32 sub_band_end;
83   WORD32 num_mf_bands;
84   WORD32 sbr_patching_mode;
85   WORD32 pre_proc_flag;
86   WORD32 is_usf_4;
87   WORD32 hbe_flag;
88   WORD32 out_fs;
89   WORD32 num_env;
90   WORD32 op_delay;
91   WORD32 codec_delay;
92   WORD32 sbr_ratio_index;
93   WORD16 invf_band_tbl[MAXIMUM_NUM_NOISE_VALUES + 1];
94   WORD16 f_master_tbl[MAXIMUM_FREQ_COEFFS + 1];
95   WORD32 inv_filt_mode[MAXIMUM_NUM_NOISE_VALUES];
96   WORD32 inv_filt_mode_prev[IXHEAACE_MAX_NUM_NOISE_VALUES];
97   FLOAT32 bw_array_prev[IXHEAACE_MAX_NUM_PATCHES];
98   WORD16 border_vec[IXHEAACE_MAX_ENV + 1];
99   WORD32 bs_tes_shape[IXHEAACE_MAX_ENV + 1];
100   WORD32 bs_tes_shape_mode[IXHEAACE_MAX_ENV + 1];
101   FLOAT32 qmf_buf_real[IXHEAACE_TIMESLOT_BUFFER_SIZE + 2 * IXHEAACE_ESBR_HBE_DELAY_OFFSET]
102                       [IXHEAACE_QMF_CHANNELS];
103   FLOAT32 qmf_buf_imag[IXHEAACE_TIMESLOT_BUFFER_SIZE + 2 * IXHEAACE_ESBR_HBE_DELAY_OFFSET]
104                       [IXHEAACE_QMF_CHANNELS];
105 } ixheaace_str_inter_tes_params;
106 
107 typedef struct {
108   FLOAT32 dst_qmf_r[IXHEAACE_TIMESLOT_BUFFER_SIZE][IXHEAACE_QMF_CHANNELS];
109   FLOAT32 dst_qmf_i[IXHEAACE_TIMESLOT_BUFFER_SIZE][IXHEAACE_QMF_CHANNELS];
110 } ixheaace_str_inter_tes_scr;
111 
112 VOID ixheaace_init_esbr_inter_tes(ixheaace_str_inter_tes_params *pstr_tes_enc,
113                                   WORD32 sbr_ratio_index);
114 IA_ERRORCODE ixheaace_process_inter_tes(ixheaace_str_inter_tes_params *pstr_tes_enc,
115                                         WORD8 *ptr_scr);
116