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 #include "ixheaac_type_def.h"
22 #include "ixheaac_constants.h"
23 #include "iusace_cnst.h"
24 #include "iusace_bitbuffer.h"
25 #include "iusace_tns_usac.h"
26 #include "iusace_psy_mod.h"
27 #include "ixheaac_basic_ops32.h"
28 #include "ixheaac_basic_ops40.h"
29 #include "ixheaac_basic_ops.h"
30
iusace_ms_apply(ia_psy_mod_data_struct * pstr_psy_data,FLOAT64 * ptr_spec_left,FLOAT64 * ptr_spec_right,WORD32 * ms_select,WORD32 ms_used[MAX_SHORT_WINDOWS][MAX_SFB_LONG],const WORD32 sfb_count,const WORD32 sfb_per_group,const WORD32 max_sfb_per_grp,const WORD32 * ptr_sfb_offsets,WORD32 chn,FLOAT64 * ptr_ms_spec)31 VOID iusace_ms_apply(ia_psy_mod_data_struct *pstr_psy_data, FLOAT64 *ptr_spec_left,
32 FLOAT64 *ptr_spec_right, WORD32 *ms_select,
33 WORD32 ms_used[MAX_SHORT_WINDOWS][MAX_SFB_LONG], const WORD32 sfb_count,
34 const WORD32 sfb_per_group, const WORD32 max_sfb_per_grp,
35 const WORD32 *ptr_sfb_offsets, WORD32 chn, FLOAT64 *ptr_ms_spec) {
36 FLOAT32 *ptr_sfb_enegry_left = pstr_psy_data[chn].ptr_sfb_energy_long;
37 FLOAT32 *ptr_sfb_energy_right = pstr_psy_data[chn + 1].ptr_sfb_energy_long;
38 const FLOAT32 *ptr_sfb_energy_mid = pstr_psy_data[chn].ptr_sfb_energy_long_ms;
39 const FLOAT32 *ptr_sfb_energy_side = pstr_psy_data[chn + 1].ptr_sfb_energy_long_ms;
40 FLOAT32 *ptr_sfb_thr_left = pstr_psy_data[chn].ptr_sfb_thr_long;
41 FLOAT32 *ptr_sfb_thr_right = pstr_psy_data[chn + 1].ptr_sfb_thr_long;
42 FLOAT32 *ptr_sfb_spread_energy_left = pstr_psy_data[chn].ptr_sfb_spreaded_energy_long;
43 FLOAT32 *ptr_sfb_spread_energy_right = pstr_psy_data[chn + 1].ptr_sfb_spreaded_energy_long;
44 WORD32 sfb, sfb_offsets, j;
45 WORD32 grp = 0;
46 WORD32 ms_counter = 0;
47 WORD32 lr_counter = 0;
48
49 *ms_select = 0;
50
51 for (sfb = 0; sfb < sfb_count; sfb += sfb_per_group, grp++) {
52 for (sfb_offsets = 0; sfb_offsets < max_sfb_per_grp; sfb_offsets++) {
53 FLOAT32 left_right, mid_side, min_thr;
54 WORD32 use_ms;
55 ms_used[grp][sfb_offsets] = 0;
56
57 min_thr = MIN(ptr_sfb_thr_left[sfb + sfb_offsets], ptr_sfb_thr_right[sfb + sfb_offsets]);
58
59 left_right =
60 (ptr_sfb_thr_left[sfb + sfb_offsets] /
61 MAX(ptr_sfb_enegry_left[sfb + sfb_offsets], ptr_sfb_thr_left[sfb + sfb_offsets])) *
62 (ptr_sfb_thr_right[sfb + sfb_offsets] /
63 max(ptr_sfb_energy_right[sfb + sfb_offsets], ptr_sfb_thr_right[sfb + sfb_offsets]));
64
65 mid_side = (min_thr / max(ptr_sfb_energy_mid[sfb + sfb_offsets], min_thr)) *
66 (min_thr / max(ptr_sfb_energy_side[sfb + sfb_offsets], min_thr));
67
68 use_ms = (mid_side >= left_right);
69
70 if (use_ms) {
71 ms_used[grp][sfb_offsets] = 1;
72
73 for (j = ptr_sfb_offsets[sfb + sfb_offsets]; j < ptr_sfb_offsets[sfb + sfb_offsets + 1];
74 j++) {
75 if (ptr_ms_spec != NULL) {
76 ptr_spec_left[j] = ptr_ms_spec[j];
77 ptr_spec_right[j] = ptr_ms_spec[1024 + j];
78 } else {
79 FLOAT64 tmp = ptr_spec_left[j];
80
81 ptr_spec_left[j] = 0.5f * (ptr_spec_left[j] + ptr_spec_right[j]);
82
83 ptr_spec_right[j] = 0.5f * (tmp - ptr_spec_right[j]);
84 }
85 }
86
87 ptr_sfb_thr_left[sfb + sfb_offsets] = ptr_sfb_thr_right[sfb + sfb_offsets] = min_thr;
88
89 ptr_sfb_enegry_left[sfb + sfb_offsets] = ptr_sfb_energy_mid[sfb + sfb_offsets];
90 ptr_sfb_energy_right[sfb + sfb_offsets] = ptr_sfb_energy_side[sfb + sfb_offsets];
91
92 ptr_sfb_spread_energy_left[sfb + sfb_offsets] =
93 ptr_sfb_spread_energy_right[sfb + sfb_offsets] =
94 min(ptr_sfb_spread_energy_left[sfb + sfb_offsets],
95 ptr_sfb_spread_energy_right[sfb + sfb_offsets]) *
96 0.5f;
97
98 ms_counter++;
99 } else {
100 ms_used[grp][sfb_offsets] = 0;
101 lr_counter++;
102 }
103 }
104 }
105
106 if (ms_counter == 0) {
107 *ms_select = 0;
108 } else {
109 if (lr_counter != 0) {
110 *ms_select = 1;
111 } else {
112 *ms_select = 2;
113 }
114 }
115 return;
116 }
117
iusace_calc_ms_band_energy(const FLOAT64 * ptr_spec_left,const FLOAT64 * ptr_spec_right,const WORD32 * ptr_band_offset,const WORD32 num_bands,FLOAT32 * ptr_band_energy_mid,FLOAT32 * ptr_band_energy_side)118 VOID iusace_calc_ms_band_energy(const FLOAT64 *ptr_spec_left, const FLOAT64 *ptr_spec_right,
119 const WORD32 *ptr_band_offset, const WORD32 num_bands,
120 FLOAT32 *ptr_band_energy_mid, FLOAT32 *ptr_band_energy_side) {
121 WORD32 i, j;
122
123 j = 0;
124 for (i = 0; i < num_bands; i++) {
125 ptr_band_energy_mid[i] = 0.0f;
126 ptr_band_energy_side[i] = 0.0f;
127
128 while (j < ptr_band_offset[i + 1]) {
129 FLOAT32 specm, specs;
130
131 specm = (FLOAT32)(0.5f * (ptr_spec_left[j] + ptr_spec_right[j]));
132 specs = (FLOAT32)(0.5f * (ptr_spec_left[j] - ptr_spec_right[j]));
133
134 ptr_band_energy_mid[i] += specm * specm;
135 ptr_band_energy_side[i] += specs * specs;
136
137 j++;
138 }
139 }
140
141 return;
142 }
143