opus_rs/
modes.rs

1use once_cell::sync::Lazy;
2
3use crate::mdct::MdctLookup;
4
5pub const BAND_ALLOCATION: [u8; 11 * 21] = [
6    /* 0 */  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
7    /* 1 */ 90, 80, 75, 69, 63, 56, 49, 40, 34, 29, 20, 18, 10,  0,  0,  0,  0,  0,  0,  0,  0,
8    /* 2 */110,100, 90, 84, 78, 71, 65, 58, 51, 45, 39, 32, 26, 20, 12,  0,  0,  0,  0,  0,  0,
9    /* 3 */118,110,103, 93, 86, 80, 75, 70, 65, 59, 53, 47, 40, 31, 23, 15,  4,  0,  0,  0,  0,
10    /* 4 */126,119,112,104, 95, 89, 83, 78, 72, 66, 60, 54, 47, 39, 32, 25, 17, 12,  1,  0,  0,
11    /* 5 */134,127,120,114,103, 97, 91, 85, 78, 72, 66, 60, 54, 47, 41, 35, 29, 23, 16, 10,  1,
12    /* 6 */144,137,130,124,113,107,101, 95, 88, 82, 76, 70, 64, 57, 51, 45, 39, 33, 26, 15,  1,
13    /* 7 */152,145,138,132,123,117,111,105, 98, 92, 86, 80, 74, 67, 61, 55, 49, 43, 36, 20,  1,
14    /* 8 */162,155,148,142,133,127,121,115,108,102, 96, 90, 84, 77, 71, 65, 59, 53, 46, 30,  1,
15    /* 9 */172,165,158,152,143,137,131,125,118,112,106,100, 94, 87, 81, 75, 69, 63, 56, 45, 20,
16    /*10 */200,200,200,200,200,200,200,198,193,188,183,178,173,168,163,158,153,148,129,104,104,
17];
18
19pub const CACHE_INDEX50: [i16; 105] = [
20    -1, -1, -1, -1, -1, -1, -1, -1, 0, 0, 0, 0, 41, 41, 41, 82, 82, 123, 164, 200, 222, 0, 0, 0, 0,
21    0, 0, 0, 0, 41, 41, 41, 41, 123, 123, 123, 164, 164, 240, 266, 283, 295, 41, 41, 41, 41, 41,
22    41, 41, 41, 123, 123, 123, 123, 240, 240, 240, 266, 266, 305, 318, 328, 336, 123, 123, 123,
23    123, 123, 123, 123, 123, 240, 240, 240, 240, 305, 305, 305, 318, 318, 343, 351, 358, 364, 240,
24    240, 240, 240, 240, 240, 240, 240, 305, 305, 305, 305, 343, 343, 343, 351, 351, 370, 376, 382,
25    387,
26];
27
28pub const CACHE_BITS50: [u8; 392] = [
29    40, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
30    7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 40, 15, 23, 28, 31, 34, 36, 38, 39, 41, 42, 43, 44, 45, 46, 47,
31    47, 49, 50, 51, 52, 53, 54, 55, 55, 57, 58, 59, 60, 61, 62, 63, 63, 65, 66, 67, 68, 69, 70, 71,
32    71, 40, 20, 33, 41, 48, 53, 57, 61, 64, 66, 69, 71, 73, 75, 76, 78, 80, 82, 85, 87, 89, 91, 92,
33    94, 96, 98, 101, 103, 105, 107, 108, 110, 112, 114, 117, 119, 121, 123, 124, 126, 128, 40, 23,
34    39, 51, 60, 67, 73, 79, 83, 87, 91, 94, 97, 100, 102, 105, 107, 111, 115, 118, 121, 124, 126,
35    129, 131, 135, 139, 142, 145, 148, 150, 153, 155, 159, 163, 166, 169, 172, 174, 177, 179, 35,
36    28, 49, 65, 78, 89, 99, 107, 114, 120, 126, 132, 136, 141, 145, 149, 153, 159, 165, 171, 176,
37    180, 185, 189, 192, 199, 205, 211, 216, 220, 225, 229, 232, 239, 245, 251, 21, 33, 58, 79, 97,
38    112, 125, 137, 148, 157, 166, 174, 182, 189, 195, 201, 207, 217, 227, 235, 243, 251, 17, 35,
39    63, 86, 106, 123, 139, 152, 165, 177, 187, 197, 206, 214, 222, 230, 237, 250, 25, 31, 55, 75,
40    91, 105, 117, 128, 138, 146, 154, 161, 168, 174, 180, 185, 190, 200, 208, 215, 222, 229, 235,
41    240, 245, 255, 16, 36, 65, 89, 110, 128, 144, 159, 173, 185, 196, 207, 217, 226, 234, 242, 250,
42    11, 41, 74, 103, 128, 151, 172, 191, 209, 225, 241, 255, 9, 43, 79, 110, 138, 163, 186, 207,
43    227, 246, 12, 39, 71, 99, 123, 144, 164, 182, 198, 214, 228, 241, 253, 9, 44, 81, 113, 142,
44    168, 192, 214, 235, 255, 7, 49, 90, 127, 160, 191, 220, 247, 6, 51, 95, 134, 170, 203, 234, 7,
45    47, 87, 123, 155, 184, 212, 237, 6, 52, 97, 137, 174, 208, 240, 5, 57, 106, 151, 192, 231, 5,
46    59, 111, 158, 202, 243, 5, 55, 103, 147, 187, 224, 5, 60, 113, 161, 206, 248, 4, 65, 122, 175,
47    224, 4, 67, 127, 182, 234,
48];
49
50pub const CACHE_CAPS50: [u8; 168] = [
51    224, 224, 224, 224, 224, 224, 224, 224, 160, 160, 160, 160, 185, 185, 185, 178, 178, 168, 134,
52    61, 37, 224, 224, 224, 224, 224, 224, 224, 224, 240, 240, 240, 240, 207, 207, 207, 198, 198,
53    183, 144, 66, 40, 160, 160, 160, 160, 160, 160, 160, 160, 185, 185, 185, 185, 193, 193, 193,
54    183, 183, 172, 138, 64, 38, 240, 240, 240, 240, 240, 240, 240, 240, 207, 207, 207, 207, 204,
55    204, 204, 193, 193, 180, 143, 66, 40, 185, 185, 185, 185, 185, 185, 185, 185, 193, 193, 193,
56    193, 193, 193, 193, 183, 183, 172, 138, 65, 39, 207, 207, 207, 207, 207, 207, 207, 207, 204,
57    204, 204, 204, 201, 201, 201, 188, 188, 176, 141, 66, 40, 193, 193, 193, 193, 193, 193, 193,
58    193, 193, 193, 193, 193, 194, 194, 194, 184, 184, 173, 139, 65, 39, 204, 204, 204, 204, 204,
59    204, 204, 204, 201, 201, 201, 201, 198, 198, 198, 187, 187, 175, 140, 66, 40,
60];
61
62pub struct PulseCache {
63    pub size: usize,
64    pub index: &'static [i16],
65    pub bits: &'static [u8],
66    pub caps: &'static [u8],
67}
68
69pub struct CeltMode {
70    pub fs: i32,
71    pub overlap: usize,
72    pub nb_ebands: usize,
73    pub eff_ebands: usize,
74    pub preemph: [f32; 4],
75    pub e_bands: &'static [i16],
76    pub max_lm: usize,
77    pub nb_short_mdcts: usize,
78    pub short_mdct_size: usize,
79    pub nb_alloc_vectors: usize,
80    pub alloc_vectors: &'static [u8],
81    pub alloc_stride: usize,
82    pub log_n: &'static [i16],
83    pub window: &'static [f32],
84    pub mdct: MdctLookup,
85    pub cache: PulseCache,
86    pub e_means: &'static [f32],
87}
88
89pub const EBAND_5MS: [i16; 22] = [
90    0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 12, 14, 16, 20, 24, 28, 34, 40, 48, 60, 78, 100,
91];
92
93impl CeltMode {
94    pub fn new_48000_960_120() -> Self {
95        let short_mdct_size = 120;
96        let nb_short_mdcts = 8; // = 1 << max_lm; matches C's nbShortMdcts
97
98        let max_lm = 3usize; // log2(nb_short_mdcts) = log2(8) = 3, matches C's maxLM
99        let mdct = MdctLookup::new(2 * short_mdct_size * nb_short_mdcts, max_lm);
100        Self {
101            fs: 48000,
102            overlap: 120,
103            nb_ebands: 21,
104            eff_ebands: 21,
105            preemph: [0.850_006_1, 0.100_006_1, 1.512_834_7, 0.661_010_7],
106            e_bands: &EBAND_5MS,
107            max_lm,
108            nb_short_mdcts,
109            short_mdct_size,
110            nb_alloc_vectors: 11,
111            alloc_vectors: &BAND_ALLOCATION,
112            alloc_stride: 21,
113            log_n: &LOG_N_400,
114            window: &WINDOW_120,
115            mdct,
116            cache: PulseCache {
117                size: 392,
118                index: &CACHE_INDEX50,
119                bits: &CACHE_BITS50,
120                caps: &CACHE_CAPS50,
121            },
122            e_means: &E_MEANS,
123        }
124    }
125}
126
127pub const E_MEANS: [f32; 25] = [
128    6.4375, 6.25, 5.75, 5.3125, 5.0625, 4.8125, 4.5, 4.375, 4.875, 4.6875, 4.5625, 4.4375, 4.875,
129    4.625, 4.3125, 4.5, 4.375, 4.625, 4.75, 4.4375, 3.75, 3.75, 3.75, 3.75, 3.75,
130];
131
132pub fn default_mode() -> &'static CeltMode {
133    &MODE_48000_960_120
134}
135
136static MODE_48000_960_120: Lazy<CeltMode> = Lazy::new(CeltMode::new_48000_960_120);
137
138const LOG_N_400: [i16; 21] = [
139    0, 0, 0, 0, 0, 0, 0, 0, 8, 8, 8, 8, 16, 16, 16, 21, 21, 24, 29, 34, 36,
140];
141
142const WINDOW_120: [f32; 120] = [
143    6.728_696_6e-05,
144    0.000_605_513_5,
145    0.001_681_597,
146    0.003_294_796_2,
147    0.005_443_994_3,
148    0.008_127_692,
149    0.011_344_001,
150    0.015_090_633,
151    0.019_364_886,
152    0.024_163_635,
153    0.029_483_315,
154    0.035_319_905,
155    0.041_668_91,
156    0.048_525_35,
157    0.055_883_718,
158    0.063_737_999,
159    0.072_081_62,
160    0.080_907_43,
161    0.090_207_7,
162    0.099_974_11,
163    0.110_197_69,
164    0.120_868_83,
165    0.131_977_29,
166    0.143_512_14,
167    0.155_461_77,
168    0.167_813_9,
169    0.180_555_5,
170    0.193_672_9,
171    0.207_151_71,
172    0.220_976_82,
173    0.235_132_43,
174    0.249_602_08,
175    0.264_368_6,
176    0.279_414_19,
177    0.294_720_4,
178    0.310_268_2,
179    0.326_037_88,
180    0.342_009_3,
181    0.358_161_77,
182    0.374_474_07,
183    0.390_924_62,
184    0.407_491_42,
185    0.424_152_15,
186    0.440_884_23,
187    0.457_664_84,
188    0.474_471_04,
189    0.491_279_78,
190    0.508_067_98,
191    0.524_812_61,
192    0.541_490_8,
193    0.558_079_7,
194    0.574_557,
195    0.590_900_5,
196    0.607_088_4,
197    0.623_099_5,
198    0.638_913_06,
199    0.654_508_96,
200    0.669_867_76,
201    0.684_970_8,
202    0.699_800_1,
203    0.714_338_7,
204    0.728_570_5,
205    0.742_480_43,
206    0.756_054_2,
207    0.769_278_95,
208    0.782_142_6,
209    0.794_634_3,
210    0.806_744_45,
211    0.818_464_6,
212    0.829_787_3,
213    0.840_706_7,
214    0.851_217_8,
215    0.861_317,
216    0.871_001_83,
217    0.880_271_11,
218    0.889_124_8,
219    0.897_564,
220    0.905_590_94,
221    0.913_209,
222    0.920_422_7,
223    0.927_237_4,
224    0.933_659_55,
225    0.939_696_56,
226    0.945_356_7,
227    0.950_649_1,
228    0.955_583_5,
229    0.960_170_7,
230    0.964_421_7,
231    0.968_348_5,
232    0.971_963_34,
233    0.975_279_06,
234    0.978_308_83,
235    0.981_066_16,
236    0.983_564_8,
237    0.985_818_7,
238    0.987_841_9,
239    0.989_648_6,
240    0.991_252_7,
241    0.992_668_5,
242    0.993_909_7,
243    0.994_990_04,
244    0.995_923,
245    0.996_721_6,
246    0.997_398_74,
247    0.997_966_67,
248    0.998_437_3,
249    0.998_822,
250    0.999_131_47,
251    0.999_376_06,
252    0.999_565_27,
253    0.999_708,
254    0.999_812_5,
255    0.999_886_13,
256    0.999_935_6,
257    0.999_967,
258    0.999_985_18,
259    0.999_994_6,
260    0.999_998_59,
261    0.999_999_8,
262    1.0,
263];
264
265pub const SPREAD_ICDF: [u8; 4] = [25, 23, 2, 0];
266pub const TRIM_ICDF: [u8; 11] = [126, 124, 119, 109, 87, 41, 19, 9, 4, 2, 0];
267pub const TAPSET_ICDF: [u8; 2] = [2, 0];
268
269pub const TF_SELECT_TABLE: [[i8; 8]; 4] = [
270    [0, -1, 0, -1, 0, -1, 0, -1],
271    [0, -1, 0, -2, 1, 0, 1, -1],
272    [0, -2, 0, -3, 2, 0, 1, -1],
273    [0, -2, 0, -3, 3, 0, 1, -1],
274];
opus_rs/modes.rs

opus_rs/
modes.rs
