xref: /aosp_15_r20/external/libxaac/encoder/iusace_esbr_pvc.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 #define IXHEAACE_ESBR_PVC_MOD_NUM_TAB (2)
23 #define IXHEAACE_ESBR_PVC_NUM_TAB_IDX_1 (3)
24 #define IXHEAACE_ESBR_PVC_NUM_TAB_IDX_2 (128)
25 #define IXHEAACE_ESBR_PVC_NUM_PVCID (128)
26 #define IXHEAACE_ESBR_PVC_NUM_TS (16)
27 #define IXHEAACE_ESBR_PVC_NUM_QMF_BANDS (64)
28 #define IXHEAACE_ESBR_PVC_NUM_QMF_BANDS_CORE (64)
29 #define IXHEAACE_ESBR_PVC_NUM_BANDS_CORE (3)
30 #define IXHEAACE_ESBR_PVC_NUM_BANDS_SBR_MODE1 (8)
31 #define IXHEAACE_ESBR_PVC_NUM_BANDS_SBR_MODE2 (6)
32 
33 #define IXHEAACE_ESBR_PVC_POW_THRS (0.1f)
34 #define IXHEAACE_ESBR_PVC_NS_MODE_PRD_THRS (2)
35 
36 #define IXHEAACE_ESBR_PVC_RESIDUAL_VAL (1000000.0f)
37 #define IXHEAACE_ESBR_PVC_NTS_GRP_ID (8)
38 
39 #define IXHEAACE_ESBR_PVC_DIV_MODE_BITS (3)
40 #define IXHEAACE_ESBR_PVC_NS_MODE_BITS (1)
41 #define IXHEAACE_ESBR_PVC_ID_BITS (7)
42 #define IXHEAACE_ESBR_PVC_GRID_INFO_BITS (1)
43 #define IXHEAACE_ESBR_PVC_REUSE_BITS (1)
44 
45 #define IXHEAACE_ESBR_PVC_FLAG_PREV_DFLT (0)
46 #define IXHEAACE_ESBR_PVC_ID_PREV_DFLT (0xFF)
47 #define IXHEAACE_ESBR_PVC_RATE_PREV_DFLT (0xFF)
48 #define IXHEAACE_ESBR_PVC_STRT_BAND_PREV_DFLT (0xFF)
49 
50 #define IXHEAACE_ESBR_PVC_MODE_1 (1)
51 #define IXHEAACE_ESBR_PVC_MODE_2 (2)
52 
53 typedef struct {
54   const FLOAT32 pvc_smth_win_ns_16[16];
55   const FLOAT32 pvc_smth_win_ns_12[12];
56   const FLOAT32 pvc_smth_win_ns_4[4];
57   const FLOAT32 pvc_smth_win_ns_3[3];
58   const UWORD8 pvc_idx_mode_1[IXHEAACE_ESBR_PVC_MOD_NUM_TAB];
59   const UWORD8 pvc_prd_coef_kb_3_mode_1[IXHEAACE_ESBR_PVC_NUM_TAB_IDX_2]
60                                        [IXHEAACE_ESBR_PVC_NUM_BANDS_SBR_MODE1];
61   const UWORD8 pvc_prd_coef_kb_012_mode_1[IXHEAACE_ESBR_PVC_NUM_TAB_IDX_1]
62                                          [IXHEAACE_ESBR_PVC_NUM_BANDS_CORE]
63                                          [IXHEAACE_ESBR_PVC_NUM_BANDS_SBR_MODE1];
64   const FLOAT32 pvc_scaling_coef_mode_1[IXHEAACE_ESBR_PVC_NUM_BANDS_CORE + 1];
65   const UWORD8 pvc_idx_mode_2[IXHEAACE_ESBR_PVC_MOD_NUM_TAB];
66   const UWORD8 pvc_prd_coef_kb_3_mode_2[IXHEAACE_ESBR_PVC_NUM_TAB_IDX_2]
67                                        [IXHEAACE_ESBR_PVC_NUM_BANDS_SBR_MODE2];
68   const UWORD8 pvc_prd_coef_kb_012_mode_2[IXHEAACE_ESBR_PVC_NUM_TAB_IDX_1]
69                                          [IXHEAACE_ESBR_PVC_NUM_BANDS_CORE]
70                                          [IXHEAACE_ESBR_PVC_NUM_BANDS_SBR_MODE2];
71   const FLOAT32 pvc_scaling_coef_mode_2[IXHEAACE_ESBR_PVC_NUM_BANDS_CORE + 1];
72 } ixheaace_pvc_tabs_struct;
73 
74 extern const ixheaace_pvc_tabs_struct ixheaace_pvc_tabs;
75 
76 typedef struct {
77   UWORD8 pvc_mode;
78   UWORD8 div_mode;
79   UWORD8 ns_mode;
80   UWORD16 pvc_id[IXHEAACE_ESBR_PVC_NUM_TS];
81   UWORD8 time_smth_ts;
82   UWORD8 num_grp_core;
83   UWORD8 num_grp_sbr;
84   UWORD8 hbw;
85   UWORD8 num_pvc_id;
86   UWORD8 pvc_rate;
87   WORD32 usac_indep_flag;
88 } ixheaace_pvc_params;
89 
90 typedef struct {
91   UWORD16 pvc_id;
92   UWORD8 start_band;
93   UWORD8 pvc_flag;
94   UWORD8 pvc_mode;
95   UWORD8 pvc_rate;
96 } ixheaace_pvc_prv_frm_params;
97 
98 typedef struct {
99   const FLOAT32 *smoothing_coef;
100   const FLOAT32 *scaling_coef;
101   const UWORD8 *pvc_pred_coef_kb_3;
102   const UWORD8 *pvc_pred_coef_kb_012;
103   const UWORD8 *pvc_idx_tab;
104 } ixheaace_pvc_coef_tabs;
105 
106 typedef struct {
107   UWORD8 div_mode;
108   UWORD8 grid_info[IXHEAACE_ESBR_PVC_NUM_TS];
109   UWORD8 ns_mode;
110   WORD32 num_grid_info;
111   UWORD16 pvc_id_bs[IXHEAACE_ESBR_PVC_NUM_TS];
112 } ixheaace_pvc_bs_info;
113 
114 typedef struct {
115   ixheaace_pvc_bs_info pvc_bs_info;
116   ixheaace_pvc_params pvc_param;
117   ixheaace_pvc_prv_frm_params pvc_prv_param;
118   ixheaace_pvc_coef_tabs pvc_tabs;
119   FLOAT32 sb_grp_energy[IXHEAACE_ESBR_PVC_NUM_TS + 16 - 1][3];
120 } ixheaace_pvc_enc;
121 
122 typedef struct {
123   FLOAT32 pvc_qmf_low[IXHEAACE_ESBR_PVC_NUM_TS * IXHEAACE_ESBR_PVC_NUM_QMF_BANDS_CORE];
124   FLOAT32 pvc_qmf_high[IXHEAACE_ESBR_PVC_NUM_TS * IXHEAACE_ESBR_PVC_NUM_QMF_BANDS];
125 } ixheaace_pvc_scratch;
126 
127 IA_ERRORCODE ixheaace_pvc_enc_init(ixheaace_pvc_enc *pstr_pvc_enc, WORD32 sbr_pvc_rate);
128 
129 IA_ERRORCODE ixheaace_pvc_encode_frame(ixheaace_pvc_enc *pstr_pvc_enc, UWORD8 pvc_mode,
130                                        FLOAT32 *ptr_qmf_low, FLOAT32 *ptr_qmf_high,
131                                        UWORD8 start_band, UWORD8 stop_band);
132