]>
Commit | Line | Data |
---|---|---|
5641dccf KS |
1 | use nihav_core::formats::*; |
2 | use nihav_core::frame::*; | |
3 | use nihav_core::codecs::*; | |
5641dccf KS |
4 | use nihav_core::dsp::mdct::IMDCT; |
5 | use nihav_core::io::bitreader::*; | |
6 | use nihav_core::io::byteio::{ByteReader, MemoryReader}; | |
7 | use nihav_core::io::codebook::*; | |
8 | use nihav_core::io::intcode::*; | |
594ca5ca KS |
9 | use std::f32::consts; |
10 | use std::mem::swap; | |
11 | ||
12 | #[derive(Debug,Clone,Copy,PartialEq)] | |
13 | enum Mode { | |
14 | Mono, | |
15 | Stereo, | |
16 | JointStereo, | |
17 | } | |
18 | ||
19 | impl Mode { | |
e07387c7 KS |
20 | fn get_channels(self) -> usize { |
21 | match self { | |
594ca5ca KS |
22 | Mode::Mono => 1, |
23 | _ => 2, | |
24 | } | |
25 | } | |
26 | } | |
27 | ||
28 | struct CookBookReader { | |
29 | bits: &'static [u8], | |
30 | codes: &'static [u16], | |
31 | } | |
32 | impl CodebookDescReader<u16> for CookBookReader { | |
33 | fn bits(&mut self, idx: usize) -> u8 { self.bits[idx] } | |
34 | fn code(&mut self, idx: usize) -> u32 { self.codes[idx] as u32 } | |
35 | fn sym (&mut self, idx: usize) -> u16 { idx as u16 } | |
36 | fn len(&mut self) -> usize { self.bits.len() } | |
37 | } | |
38 | ||
39 | struct Codebooks { | |
40 | cpl_cb: [Codebook<u16>; 5], | |
41 | quant_cb: Vec<Codebook<u16>>, | |
42 | vq_cb: [Codebook<u16>; 7], | |
43 | } | |
44 | ||
45 | impl Codebooks { | |
46 | fn new() -> Self { | |
47 | let mut cpl0 = CookBookReader { codes: COOK_CPL_2BITS_CODES, bits: COOK_CPL_2BITS_BITS }; | |
48 | let mut cpl1 = CookBookReader { codes: COOK_CPL_3BITS_CODES, bits: COOK_CPL_3BITS_BITS }; | |
49 | let mut cpl2 = CookBookReader { codes: COOK_CPL_4BITS_CODES, bits: COOK_CPL_4BITS_BITS }; | |
50 | let mut cpl3 = CookBookReader { codes: COOK_CPL_5BITS_CODES, bits: COOK_CPL_5BITS_BITS }; | |
51 | let mut cpl4 = CookBookReader { codes: COOK_CPL_6BITS_CODES, bits: COOK_CPL_6BITS_BITS }; | |
52 | let cpl_cb = [Codebook::new(&mut cpl0, CodebookMode::MSB).unwrap(), | |
53 | Codebook::new(&mut cpl1, CodebookMode::MSB).unwrap(), | |
54 | Codebook::new(&mut cpl2, CodebookMode::MSB).unwrap(), | |
55 | Codebook::new(&mut cpl3, CodebookMode::MSB).unwrap(), | |
56 | Codebook::new(&mut cpl4, CodebookMode::MSB).unwrap()]; | |
57 | let mut quant_cb: Vec<Codebook<u16>> = Vec::with_capacity(COOK_QUANT_CODES.len()); | |
58 | for i in 0..COOK_QUANT_CODES.len() { | |
59 | let mut quant = CookBookReader { codes: COOK_QUANT_CODES[i], bits: COOK_QUANT_BITS[i] }; | |
60 | quant_cb.push(Codebook::new(&mut quant, CodebookMode::MSB).unwrap()); | |
61 | } | |
62 | let mut vq0 = CookBookReader { codes: COOK_VQ0_CODES, bits: COOK_VQ0_BITS }; | |
63 | let mut vq1 = CookBookReader { codes: COOK_VQ1_CODES, bits: COOK_VQ1_BITS }; | |
64 | let mut vq2 = CookBookReader { codes: COOK_VQ2_CODES, bits: COOK_VQ2_BITS }; | |
65 | let mut vq3 = CookBookReader { codes: COOK_VQ3_CODES, bits: COOK_VQ3_BITS }; | |
66 | let mut vq4 = CookBookReader { codes: COOK_VQ4_CODES, bits: COOK_VQ4_BITS }; | |
67 | let mut vq5 = CookBookReader { codes: COOK_VQ5_CODES, bits: COOK_VQ5_BITS }; | |
68 | let mut vq6 = CookBookReader { codes: COOK_VQ6_CODES, bits: COOK_VQ6_BITS }; | |
69 | let vq_cb = [Codebook::new(&mut vq0, CodebookMode::MSB).unwrap(), | |
70 | Codebook::new(&mut vq1, CodebookMode::MSB).unwrap(), | |
71 | Codebook::new(&mut vq2, CodebookMode::MSB).unwrap(), | |
72 | Codebook::new(&mut vq3, CodebookMode::MSB).unwrap(), | |
73 | Codebook::new(&mut vq4, CodebookMode::MSB).unwrap(), | |
74 | Codebook::new(&mut vq5, CodebookMode::MSB).unwrap(), | |
75 | Codebook::new(&mut vq6, CodebookMode::MSB).unwrap()]; | |
76 | Codebooks { | |
e07387c7 KS |
77 | cpl_cb, |
78 | quant_cb, | |
79 | vq_cb, | |
594ca5ca KS |
80 | } |
81 | } | |
82 | } | |
83 | ||
84 | struct CookDSP { | |
85 | imdct: IMDCT, | |
86 | window: [f32; 1024], | |
87 | out: [f32; 2048], | |
88 | size: usize, | |
89 | pow_tab: [f32; 128], | |
90 | hpow_tab: [f32; 128], | |
91 | gain_tab: [f32; 23], | |
92 | } | |
93 | ||
94 | impl CookDSP { | |
95 | fn new(samples: usize) -> Self { | |
96 | let fsamples = samples as f32; | |
97 | let mut window: [f32; 1024] = [0.0; 1024]; | |
98 | let factor = consts::PI / (2.0 * fsamples); | |
99 | let scale = (2.0 / fsamples).sqrt() / 32768.0; | |
100 | for k in 0..samples { | |
101 | window[k] = (factor * ((k as f32) + 0.5)).sin() * scale; | |
102 | } | |
103 | let mut pow_tab: [f32; 128] = [0.0; 128]; | |
104 | let mut hpow_tab: [f32; 128] = [0.0; 128]; | |
105 | for i in 0..128 { | |
106 | pow_tab[i] = 2.0f32.powf((i as f32) - 64.0); | |
107 | hpow_tab[i] = 2.0f32.powf(((i as f32) - 64.0) * 0.5); | |
108 | } | |
109 | let mut gain_tab: [f32; 23] = [0.0; 23]; | |
110 | for i in 0..23 { | |
111 | gain_tab[i] = pow_tab[i + 53].powf(8.0 / fsamples); | |
112 | } | |
113 | let size = samples; | |
e07387c7 | 114 | CookDSP { imdct: IMDCT::new(samples*2, false), window, out: [0.0; 2048], size, pow_tab, hpow_tab, gain_tab } |
594ca5ca KS |
115 | } |
116 | } | |
117 | ||
118 | trait ClipCat { | |
119 | fn clip_cat(&self) -> usize; | |
120 | } | |
121 | ||
122 | impl ClipCat for i32 { | |
123 | fn clip_cat(&self) -> usize { ((*self).max(0) as usize).min(NUM_CATEGORIES - 1) } | |
124 | } | |
125 | ||
126 | const BAND_SIZE: usize = 20; | |
127 | const MAX_SAMPLES: usize = MAX_SUBBANDS * BAND_SIZE; | |
128 | const MAX_PAIRS: usize = 5; | |
129 | const MAX_SUBBANDS: usize = 52; | |
130 | const NUM_CATEGORIES: usize = 8; | |
131 | ||
132 | #[derive(Clone,Copy)] | |
133 | struct CookChannelPair { | |
134 | start_ch: usize, | |
135 | mode: Mode, | |
136 | samples: usize, | |
137 | subbands: usize, | |
138 | js_start: usize, | |
139 | js_bits: u8, | |
140 | vector_bits: u8, | |
141 | ||
142 | decouple: [u8; BAND_SIZE], | |
143 | category: [u8; MAX_SUBBANDS * 2], | |
144 | ||
69f77596 | 145 | block: [[f32; MAX_SAMPLES * 2]; 2], |
594ca5ca KS |
146 | delay: [[f32; MAX_SAMPLES]; 2], |
147 | gains: [[i32; 9]; 2], | |
148 | prev_gains: [[i32; 9]; 2], | |
149 | qindex: [i8; MAX_SUBBANDS * 2], | |
150 | } | |
151 | ||
152 | impl CookChannelPair { | |
153 | fn new() -> Self { | |
154 | CookChannelPair { | |
155 | start_ch: 0, | |
156 | mode: Mode::Mono, | |
157 | samples: 0, | |
158 | subbands: 0, | |
159 | js_start: 0, | |
160 | js_bits: 0, | |
161 | vector_bits: 0, | |
162 | ||
163 | decouple: [0; BAND_SIZE], | |
164 | category: [0; MAX_SUBBANDS * 2], | |
165 | ||
69f77596 | 166 | block: [[0.0; MAX_SAMPLES * 2]; 2], |
594ca5ca KS |
167 | delay: [[0.0; MAX_SAMPLES]; 2], |
168 | gains: [[0; 9]; 2], | |
169 | prev_gains: [[0; 9]; 2], | |
170 | qindex: [0; MAX_SUBBANDS * 2], | |
171 | } | |
172 | } | |
173 | fn read_hdr_v1(&mut self, br: &mut ByteReader) -> DecoderResult<()> { | |
174 | let ver = br.read_u32be()?; | |
175 | let micro_ver = ver & 0xFF; | |
176 | self.samples = br.read_u16be()? as usize; | |
177 | validate!(self.samples > 0 && ((self.samples & (self.samples - 1)) == 0)); | |
178 | self.subbands = br.read_u16be()? as usize; | |
179 | validate!(self.subbands <= MAX_SUBBANDS); | |
180 | match micro_ver { | |
181 | 1 => { | |
182 | self.mode = Mode::Mono; | |
183 | self.js_start = 0; | |
184 | self.js_bits = 0; | |
185 | }, | |
186 | 2 => { | |
187 | self.mode = Mode::Stereo; | |
188 | self.js_start = 0; | |
189 | self.js_bits = 0; | |
190 | }, | |
191 | 3 => { | |
192 | self.mode = Mode::JointStereo; | |
193 | let _delay = br.read_u32be()?; | |
194 | self.js_start = br.read_u16be()? as usize; | |
195 | self.js_bits = br.read_u16be()? as u8; | |
196 | validate!(self.js_start < MAX_SUBBANDS); | |
197 | validate!((self.js_bits >= 2) && (self.js_bits <= 6)); | |
198 | }, | |
199 | _ => { return Err(DecoderError::InvalidData);} | |
200 | } | |
201 | Ok(()) | |
202 | } | |
203 | fn read_hdr_v2(&mut self, br: &mut ByteReader) -> DecoderResult<u32> { | |
204 | let ver = br.read_u32be()?; | |
205 | validate!((ver >> 24) == 2); | |
206 | self.samples = br.read_u16be()? as usize; | |
207 | self.subbands = br.read_u16be()? as usize; | |
208 | validate!(self.subbands <= MAX_SUBBANDS); | |
209 | let _delay = br.read_u32be()?; | |
210 | self.js_start = br.read_u16be()? as usize; | |
211 | validate!(self.js_start < MAX_SUBBANDS); | |
212 | let js_bits = br.read_u16be()?; | |
213 | let chmap = br.read_u32be()?; | |
214 | if chmap.count_ones() == 1 { | |
215 | self.js_bits = 0; | |
216 | self.mode = Mode::Mono; | |
217 | } else { | |
218 | validate!((js_bits >= 2) && (js_bits <= 6)); | |
219 | self.js_bits = js_bits as u8; | |
220 | self.mode = Mode::JointStereo; | |
221 | } | |
222 | Ok(chmap) | |
223 | } | |
224 | fn bitalloc(&mut self, num_vectors: usize, bits: usize) { | |
225 | let avail_bits = (if bits > self.samples { self.samples + ((bits - self.samples) * 5) / 8 } else { bits }) as i32; | |
226 | let total_subbands = self.subbands + self.js_start; | |
227 | ||
228 | let mut bias: i32 = -32; | |
229 | for i in 0..6 { | |
230 | let mut sum = 0; | |
231 | for j in 0..total_subbands { | |
232 | let idx = ((32 >> i) + bias - (self.qindex[j] as i32)) / 2; | |
233 | sum += COOK_EXP_BITS[idx.clip_cat()]; | |
234 | } | |
235 | if sum >= (avail_bits - 32) { | |
236 | bias += 32 >> i; | |
237 | } | |
238 | } | |
239 | ||
240 | let mut exp_index1: [usize; MAX_SUBBANDS * 2] = [0; MAX_SUBBANDS * 2]; | |
241 | let mut exp_index2: [usize; MAX_SUBBANDS * 2] = [0; MAX_SUBBANDS * 2]; | |
242 | let mut sum = 0; | |
243 | for i in 0..total_subbands { | |
244 | let idx = ((bias - (self.qindex[i] as i32)) / 2).clip_cat(); | |
245 | sum += COOK_EXP_BITS[idx]; | |
246 | exp_index1[i] = idx; | |
247 | exp_index2[i] = idx; | |
248 | } | |
249 | ||
250 | let mut tbias1 = sum; | |
251 | let mut tbias2 = sum; | |
252 | let mut tcat: [usize; 128*2] = [0; 128*2]; | |
253 | let mut tcat_idx1 = 128; | |
254 | let mut tcat_idx2 = 128; | |
255 | for _ in 1..(1 << self.vector_bits) { | |
256 | if tbias1 + tbias2 > avail_bits * 2 { | |
257 | let mut max = -999999; | |
258 | let mut idx = total_subbands + 1; | |
259 | for j in 0..total_subbands { | |
260 | if exp_index1[j] >= (NUM_CATEGORIES - 1) { continue; } | |
261 | let t = -2 * (exp_index1[j] as i32) - (self.qindex[j] as i32) + bias; | |
262 | if t >= max { | |
263 | max = t; | |
264 | idx = j; | |
265 | } | |
266 | } | |
267 | if idx >= total_subbands { break; } | |
268 | tcat[tcat_idx1] = idx; | |
269 | tcat_idx1 += 1; | |
270 | tbias1 -= COOK_EXP_BITS[exp_index1[idx]] - COOK_EXP_BITS[exp_index1[idx] + 1]; | |
271 | exp_index1[idx] += 1; | |
272 | } else { | |
273 | let mut min = 999999; | |
274 | let mut idx = total_subbands + 1; | |
275 | for j in 0..total_subbands { | |
276 | if exp_index2[j] == 0 { continue; } | |
277 | let t = -2 * (exp_index2[j] as i32) - (self.qindex[j] as i32) + bias; | |
278 | if t < min { | |
279 | min = t; | |
280 | idx = j; | |
281 | } | |
282 | } | |
283 | if idx >= total_subbands { break; } | |
284 | tcat_idx2 -= 1; | |
285 | tcat[tcat_idx2] = idx; | |
286 | tbias2 -= COOK_EXP_BITS[exp_index2[idx]] - COOK_EXP_BITS[exp_index2[idx] - 1]; | |
287 | exp_index2[idx] -= 1; | |
288 | } | |
289 | } | |
290 | for i in 0..total_subbands { | |
291 | self.category[i] = exp_index2[i] as u8; | |
292 | } | |
293 | ||
294 | for _ in 0..num_vectors { | |
295 | let idx = tcat[tcat_idx2]; | |
296 | tcat_idx2 += 1; | |
297 | self.category[idx] = (self.category[idx] + 1).min((NUM_CATEGORIES - 1) as u8) as u8; | |
298 | } | |
299 | } | |
300 | fn decode_channel_data(&mut self, dsp: &mut CookDSP, rnd: &mut RND, codebooks: &Codebooks, src: &[u8], buf: &mut [u8], channel: usize) -> DecoderResult<()> { | |
301 | // decrypt | |
302 | for (i, b) in src.iter().enumerate() { | |
303 | buf[i] = b ^ COOK_XOR_KEY[i & 3]; | |
304 | } | |
305 | let mut br = BitReader::new(buf, src.len(), BitReaderMode::BE); | |
306 | ||
307 | let num_gains = br.read_code(UintCodeType::UnaryOnes)? as usize; | |
308 | validate!(num_gains <= 8); | |
309 | ||
310 | swap(&mut self.gains[channel], &mut self.prev_gains[channel]); | |
69f77596 | 311 | self.block[channel] = [0.0; MAX_SAMPLES * 2]; |
594ca5ca KS |
312 | |
313 | // gains | |
314 | let mut ipos = 0; | |
315 | for _ in 0..num_gains { | |
316 | let idx = br.read(3)? as usize; | |
317 | let val; | |
318 | if br.read_bool()? { | |
319 | val = (br.read(4)? as i32) - 7; | |
320 | } else { | |
321 | val = -1; | |
322 | } | |
323 | validate!(idx >= ipos); | |
324 | while ipos <= idx { | |
325 | self.prev_gains[channel][ipos] = val; | |
326 | ipos += 1; | |
327 | } | |
328 | } | |
329 | while ipos <= 8 { | |
330 | self.prev_gains[channel][ipos] = 0; | |
331 | ipos += 1; | |
332 | } | |
333 | ||
334 | // coupling information | |
335 | if self.mode == Mode::JointStereo { | |
336 | let cstart = COOK_CPL_BAND[self.js_start] as usize; | |
337 | let cend = COOK_CPL_BAND[self.subbands - 1] as usize; | |
338 | if br.read_bool()? { | |
339 | let cb = &codebooks.cpl_cb[(self.js_bits - 2) as usize]; | |
e07387c7 | 340 | for i in cstart..=cend { |
594ca5ca KS |
341 | self.decouple[i] = br.read_cb(cb)? as u8; |
342 | } | |
343 | } else { | |
e07387c7 | 344 | for i in cstart..=cend { |
594ca5ca KS |
345 | self.decouple[i] = br.read(self.js_bits)? as u8; |
346 | } | |
347 | } | |
348 | } | |
349 | ||
350 | // envelope | |
351 | let tot_subbands = self.subbands + self.js_start; | |
352 | self.qindex[0] = (br.read(6)? as i8) - 6; | |
353 | for i in 1..tot_subbands { | |
354 | let mut pos = i; | |
355 | if pos >= self.js_start * 2 { | |
356 | pos -= self.js_start; | |
357 | } else { | |
358 | pos >>= 1; | |
359 | } | |
360 | let ipos = ((pos as i8) - 1).max(0).min(12); | |
361 | let cb = &codebooks.quant_cb[ipos as usize]; | |
362 | self.qindex[i] = (br.read_cb(cb)? as i8) + self.qindex[i - 1] - 12; | |
363 | validate!((self.qindex[i] >= -63) && (self.qindex[i] <= 63)); | |
364 | } | |
365 | let num_vectors = br.read(self.vector_bits)? as usize; | |
366 | self.bitalloc(num_vectors, br.left() as usize); | |
367 | ||
368 | // coefficients | |
69f77596 | 369 | self.block[channel] = [0.0; MAX_SAMPLES * 2]; |
594ca5ca KS |
370 | let mut off = 0; |
371 | for sb in 0..tot_subbands { | |
372 | let mut coef_index: [u8; BAND_SIZE] = [0; BAND_SIZE]; | |
373 | let mut coef_sign: [bool; BAND_SIZE] = [false; BAND_SIZE]; | |
374 | let cat = self.category[sb] as usize; | |
375 | if (cat < NUM_CATEGORIES - 1) && br.left() > 0 { | |
376 | unpack_band(&mut br, codebooks, &mut coef_index, &mut coef_sign, cat)?; | |
377 | } | |
378 | for i in 0..BAND_SIZE { | |
379 | let val; | |
380 | if coef_index[i] == 0 { | |
381 | let v = COOK_DITHER_TAB[cat]; | |
382 | val = if !rnd.get_sign() { v } else { -v }; | |
383 | } else { | |
384 | let v = COOK_QUANT_CENTROID[cat][coef_index[i] as usize]; | |
385 | val = if !coef_sign[i] { v } else { -v }; | |
386 | } | |
387 | self.block[channel][off + i] = val * dsp.hpow_tab[(self.qindex[sb] + 64) as usize]; | |
388 | } | |
389 | off += BAND_SIZE; | |
390 | } | |
391 | ||
392 | Ok(()) | |
393 | } | |
394 | fn decode(&mut self, dsp: &mut CookDSP, rnd: &mut RND, codebooks: &Codebooks, src: &[u8], buf: &mut [u8], abuf: &mut NABufferType) -> DecoderResult<()> { | |
395 | if self.mode == Mode::Stereo { | |
396 | let mut schunk = src.chunks(src.len() / 2); | |
397 | self.decode_channel_data(dsp, rnd, codebooks, schunk.next().unwrap(), buf, 0)?; | |
398 | self.decode_channel_data(dsp, rnd, codebooks, schunk.next().unwrap(), buf, 1)?; | |
399 | } else { | |
400 | self.decode_channel_data(dsp, rnd, codebooks, src, buf, 0)?; | |
401 | } | |
402 | // uncouple joint stereo channels | |
403 | if self.mode == Mode::JointStereo { | |
404 | for i in 0..self.js_start { | |
405 | for j in 0..BAND_SIZE { | |
406 | self.block[1][i * BAND_SIZE + j] = self.block[0][(i * 2 + 1) * BAND_SIZE + j]; | |
407 | self.block[0][i * BAND_SIZE + j] = self.block[0][(i * 2) * BAND_SIZE + j]; | |
408 | } | |
409 | } | |
410 | let scale_idx = (self.js_bits as usize) - 2; | |
411 | let scale_off = (1 << self.js_bits) as usize; | |
412 | for i in self.js_start..self.subbands { | |
413 | let idx = self.decouple[COOK_CPL_BAND[i] as usize] as usize; | |
414 | let doff = i * BAND_SIZE; | |
415 | let soff = (i + self.js_start) * BAND_SIZE; | |
416 | let m1 = COOK_CPL_SCALES[scale_idx][ 1 + idx]; | |
417 | let m2 = COOK_CPL_SCALES[scale_idx][scale_off - 1 - idx]; | |
418 | for j in 0..BAND_SIZE { | |
419 | self.block[0][doff + j] = self.block[0][soff + j] * m1; | |
420 | self.block[1][doff + j] = self.block[0][soff + j] * m2; | |
421 | } | |
422 | } | |
423 | for i in (self.subbands * BAND_SIZE)..MAX_SAMPLES { | |
424 | self.block[0][i] = 0.0; | |
425 | self.block[1][i] = 0.0; | |
426 | } | |
427 | self.gains[1] = self.gains[0]; | |
428 | self.prev_gains[1] = self.prev_gains[0]; | |
429 | } | |
430 | for ch in 0..self.mode.get_channels() { | |
431 | let off = abuf.get_offset(ch + self.start_ch); | |
432 | let mut adata = abuf.get_abuf_f32().unwrap(); | |
1a967e6b | 433 | let output = adata.get_data_mut().unwrap(); |
594ca5ca KS |
434 | let dst = &mut output[off..]; |
435 | ||
436 | dsp.imdct.imdct(&self.block[ch], &mut dsp.out); | |
437 | ||
438 | let prev_gain = dsp.pow_tab[(self.prev_gains[ch][0] + 64) as usize]; | |
439 | let mut cur_gain = 0.0; | |
440 | let mut cur_gain2 = 0.0; | |
441 | let mut gain_idx = 0; | |
442 | let eighthmask = (self.samples >> 3) - 1; | |
443 | for (i, out) in dst.iter_mut().take(self.samples).enumerate() { | |
444 | *out = dsp.out[i + self.samples] * prev_gain * dsp.window[i] | |
445 | - self.delay[ch][i] * dsp.window[self.samples - i - 1]; | |
446 | if (i & eighthmask) == 0 { | |
447 | if (self.gains[ch][gain_idx] == 0) && (self.gains[ch][gain_idx + 1] == 0) { | |
448 | cur_gain = 1.0; | |
449 | cur_gain2 = 1.0; | |
450 | } else { | |
451 | cur_gain = dsp.pow_tab[(self.gains[ch][gain_idx] + 64) as usize]; | |
452 | cur_gain2 = dsp.gain_tab[(self.gains[ch][gain_idx + 1] - self.gains[ch][gain_idx] + 11) as usize]; | |
453 | } | |
454 | gain_idx += 1; | |
455 | } | |
456 | *out *= cur_gain; | |
457 | cur_gain *= cur_gain2; | |
458 | } | |
459 | for i in 0..self.samples { self.delay[ch][i] = dsp.out[i]; } | |
460 | } | |
461 | Ok(()) | |
462 | } | |
463 | } | |
464 | ||
465 | const COOK_VQ_GROUP_SIZE: [usize; 7] = [ 2, 2, 2, 4, 4, 5, 5 ]; | |
466 | const COOK_NUM_VQ_GROUPS: [usize; 7] = [ 10, 10, 10, 5, 5, 4, 4 ]; | |
467 | const COOK_VQ_INV_RADIX: [u32; 7] = [ 74899, 104858, 149797, 209716, 262144, 349526, 524288 ]; | |
468 | const COOK_VQ_MULT: [u32; 7] = [ 13, 9, 6, 4, 3, 2, 1 ]; | |
469 | fn unpack_band(br: &mut BitReader, codebooks: &Codebooks, coef_index: &mut [u8; BAND_SIZE], coef_sign: &mut [bool; BAND_SIZE], cat: usize) -> DecoderResult<()> { | |
470 | let cb = &codebooks.vq_cb[cat]; | |
471 | let group_size = COOK_VQ_GROUP_SIZE[cat]; | |
472 | let mult = COOK_VQ_MULT[cat] + 1; | |
473 | for i in 0..COOK_NUM_VQ_GROUPS[cat] { | |
474 | let ret = br.read_cb(cb); | |
475 | let mut val; | |
476 | if let Ok(v) = ret { | |
477 | val = v as u32; | |
478 | } else { | |
479 | let left = br.left() as u32; | |
480 | br.skip(left)?; | |
481 | break; | |
482 | } | |
483 | let mut nnz = 0; | |
484 | for j in (0..group_size).rev() { | |
485 | let t = (val * COOK_VQ_INV_RADIX[cat]) >> 20; | |
486 | coef_index[i * group_size + j] = (val - t * mult) as u8; | |
487 | if coef_index[i * group_size + j] != 0 { | |
488 | nnz += 1; | |
489 | } | |
490 | val = t; | |
491 | } | |
492 | if (br.left() as usize) < nnz { | |
493 | let left = br.left() as u32; | |
494 | br.skip(left)?; | |
495 | break; | |
496 | } | |
497 | for j in 0..group_size { | |
498 | if coef_index[i * group_size + j] != 0 { | |
499 | coef_sign[i * group_size + j] = br.read_bool()?; | |
500 | } else { | |
501 | coef_sign[i * group_size + j] = false; | |
502 | } | |
503 | } | |
504 | } | |
505 | Ok(()) | |
506 | } | |
507 | ||
508 | struct RND { | |
509 | state: u32, | |
510 | } | |
511 | ||
512 | impl RND { | |
513 | fn new() -> Self { | |
514 | Self { state: 0xC0DECC00 } | |
515 | } | |
516 | fn get_sign(&mut self) -> bool { | |
517 | self.state = (self.state & 0xFFFF).wrapping_mul(36969).wrapping_add(self.state >> 16); | |
518 | (self.state & 0x10000) != 0 | |
519 | } | |
520 | } | |
521 | ||
522 | struct CookDecoder { | |
2422d969 | 523 | info: NACodecInfoRef, |
594ca5ca KS |
524 | chmap: NAChannelMap, |
525 | src: [u8; 65536], | |
526 | num_pairs: usize, | |
527 | pairs: [CookChannelPair; MAX_PAIRS], | |
528 | channels: usize, | |
529 | samples: usize, | |
530 | codebooks: Codebooks, | |
531 | rnd: RND, | |
532 | dsp: CookDSP, | |
533 | } | |
534 | ||
535 | impl CookDecoder { | |
536 | fn new() -> Self { | |
537 | CookDecoder { | |
538 | info: NACodecInfo::new_dummy(), | |
539 | chmap: NAChannelMap::new(), | |
540 | src: [0; 65536], | |
541 | num_pairs: 0, | |
542 | channels: 0, | |
543 | samples: 0, | |
544 | pairs: [CookChannelPair::new(); MAX_PAIRS], | |
545 | codebooks: Codebooks::new(), | |
546 | rnd: RND::new(), | |
547 | dsp: CookDSP::new(1024), | |
548 | } | |
549 | } | |
550 | } | |
551 | ||
552 | impl NADecoder for CookDecoder { | |
01613464 | 553 | fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> { |
594ca5ca KS |
554 | if let NACodecTypeInfo::Audio(ainfo) = info.get_properties() { |
555 | let edata = info.get_extradata().unwrap(); | |
556 | validate!(edata.len() >= 4); | |
557 | ||
558 | let mut mr = MemoryReader::new_read(&edata); | |
559 | let mut br = ByteReader::new(&mut mr); | |
560 | let ver = br.peek_u32be()?; | |
561 | ||
562 | let maj_ver = ver >> 24; | |
563 | let mut chmap: u32 = 0; | |
564 | match maj_ver { | |
565 | 1 => { | |
566 | self.num_pairs = 1; | |
567 | self.pairs[0].read_hdr_v1(&mut br)?; | |
568 | self.channels = self.pairs[0].mode.get_channels(); | |
569 | if ainfo.get_channels() == 1 { // forced mono | |
b7eda71d | 570 | self.pairs[0].mode = Mode::Mono; |
594ca5ca KS |
571 | self.channels = 1; |
572 | chmap = 0x4; | |
573 | } else { | |
574 | chmap = 0x3; | |
575 | } | |
576 | }, | |
577 | 2 => { | |
578 | self.num_pairs = (edata.len() - (br.tell() as usize)) / 20; | |
579 | validate!(self.num_pairs <= MAX_PAIRS); | |
580 | let mut start_ch = 0; | |
581 | for i in 0..self.num_pairs { | |
582 | let pair_chmap = self.pairs[i].read_hdr_v2(&mut br)?; | |
583 | self.pairs[i].start_ch = start_ch; | |
584 | validate!((chmap & pair_chmap) == 0); | |
585 | start_ch += self.pairs[i].mode.get_channels(); | |
586 | } | |
587 | self.channels = start_ch; | |
588 | }, | |
589 | _ => { return Err(DecoderError::InvalidData); } | |
590 | }; | |
591 | ||
592 | self.samples = self.pairs[0].samples / self.pairs[0].mode.get_channels(); | |
593 | validate!((self.samples >= 16) && (self.samples <= 1024)); | |
594 | if self.samples != self.dsp.size { | |
595 | self.dsp = CookDSP::new(self.samples); | |
596 | } | |
597 | self.chmap = NAChannelMap::from_ms_mapping(chmap); | |
598 | ||
599 | for i in 1..self.num_pairs { | |
600 | validate!((self.pairs[i].samples / self.pairs[i].mode.get_channels()) == self.samples); | |
601 | } | |
602 | ||
603 | let vector_bits = match self.samples { | |
604 | 16 | 32 | 64 | 128 | 256 => 5, | |
605 | 512 => 6, | |
606 | 1024 => 7, | |
607 | _ => unreachable!(), | |
608 | }; | |
609 | for pair in self.pairs.iter_mut() { | |
610 | match pair.mode { | |
611 | Mode::Mono => { | |
612 | pair.vector_bits = 5; | |
613 | }, | |
614 | Mode::Stereo => { | |
615 | pair.vector_bits = 5; | |
616 | pair.samples >>= 1; | |
617 | }, | |
618 | Mode::JointStereo => { | |
619 | pair.vector_bits = vector_bits; | |
620 | pair.samples >>= 1; | |
621 | }, | |
622 | }; | |
623 | } | |
624 | ||
625 | let ainfo = NAAudioInfo::new(ainfo.get_sample_rate(), self.channels as u8, | |
626 | SND_F32P_FORMAT, self.samples); | |
627 | self.info = info.replace_info(NACodecTypeInfo::Audio(ainfo)); | |
628 | ||
629 | Ok(()) | |
630 | } else { | |
631 | Err(DecoderError::InvalidData) | |
632 | } | |
633 | } | |
01613464 | 634 | fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> { |
594ca5ca KS |
635 | let info = pkt.get_stream().get_info(); |
636 | validate!(info.get_properties().is_audio()); | |
637 | let pktbuf = pkt.get_buffer(); | |
638 | validate!(pktbuf.len() > self.num_pairs * 2); | |
639 | ||
640 | let mut seg_size: [usize; MAX_PAIRS] = [0; MAX_PAIRS]; | |
641 | let mut seg_start: [usize; MAX_PAIRS+1] = [0; MAX_PAIRS+1]; | |
642 | ||
643 | let ainfo = self.info.get_properties().get_audio_info().unwrap(); | |
644 | ||
645 | seg_size[0] = pktbuf.len() - (self.num_pairs - 1); | |
646 | for i in 1..self.num_pairs { | |
647 | seg_size[i] = (pktbuf[pktbuf.len() - self.num_pairs + i] as usize) * 2; | |
648 | validate!(seg_size[i] != 0); | |
649 | let ret = seg_size[0].checked_sub(seg_size[i]); | |
650 | if let Some(val) = ret { | |
651 | seg_size[0] = val; | |
652 | } else { | |
653 | return Err(DecoderError::InvalidData); | |
654 | } | |
655 | } | |
656 | validate!(seg_size[0] != 0); | |
657 | seg_start[0] = 0; | |
658 | for i in 0..self.num_pairs { | |
659 | seg_start[i + 1] = seg_start[i] + seg_size[i]; | |
660 | } | |
661 | ||
662 | let mut abuf = alloc_audio_buffer(ainfo, self.samples, self.chmap.clone())?; | |
663 | ||
664 | for pair in 0..self.num_pairs { | |
665 | self.pairs[pair].decode(&mut self.dsp, &mut self.rnd, &self.codebooks, &pktbuf[seg_start[pair]..seg_start[pair + 1]], &mut self.src, &mut abuf)?; | |
666 | } | |
667 | ||
668 | let mut frm = NAFrame::new_from_pkt(pkt, self.info.replace_info(NACodecTypeInfo::Audio(ainfo)), abuf); | |
669 | frm.set_keyframe(true); | |
171860fc | 670 | Ok(frm.into_ref()) |
594ca5ca | 671 | } |
f9be4e75 KS |
672 | fn flush(&mut self) { |
673 | for pair in self.pairs.iter_mut() { | |
674 | pair.delay = [[0.0; MAX_SAMPLES]; 2]; | |
675 | } | |
676 | } | |
594ca5ca KS |
677 | } |
678 | ||
08a1fab7 | 679 | pub fn get_decoder() -> Box<dyn NADecoder + Send> { |
594ca5ca KS |
680 | Box::new(CookDecoder::new()) |
681 | } | |
682 | ||
683 | #[cfg(test)] | |
684 | mod test { | |
3167c45c KS |
685 | use nihav_core::codecs::RegisteredDecoders; |
686 | use nihav_core::demuxers::RegisteredDemuxers; | |
687 | use nihav_core::test::dec_video::*; | |
688 | use crate::codecs::realmedia_register_all_codecs; | |
689 | use crate::demuxers::realmedia_register_all_demuxers; | |
594ca5ca KS |
690 | #[test] |
691 | fn test_cook() { | |
3167c45c KS |
692 | let mut dmx_reg = RegisteredDemuxers::new(); |
693 | realmedia_register_all_demuxers(&mut dmx_reg); | |
694 | let mut dec_reg = RegisteredDecoders::new(); | |
695 | realmedia_register_all_codecs(&mut dec_reg); | |
696 | ||
594ca5ca KS |
697 | // let file = "assets/RV/rv30_weighted_mc.rm"; |
698 | let file = "assets/RV/multichannel.rma"; | |
5580b11b | 699 | test_decode_audio("realmedia", file, Some(2000), None/*Some("cook")*/, &dmx_reg, &dec_reg); |
594ca5ca KS |
700 | } |
701 | } | |
702 | ||
703 | const COOK_XOR_KEY: [u8; 4] = [ 0x37, 0xC5, 0x11, 0xF2 ]; | |
704 | ||
705 | const COOK_CPL_2BITS_BITS: &[u8; 3] = &[ 2, 1, 2 ]; | |
706 | const COOK_CPL_2BITS_CODES: &[u16; 3] = &[ 0x02, 0x00, 0x03 ]; | |
707 | const COOK_CPL_3BITS_BITS: &[u8; 7] = &[ 6, 5, 2, 1, 3, 4, 6 ]; | |
708 | const COOK_CPL_3BITS_CODES: &[u16; 7] = &[ 0x3e, 0x1e, 0x02, 0x00, 0x06, 0x0e, 0x3f ]; | |
709 | const COOK_CPL_4BITS_BITS: &[u8; 15] = &[ 8, 8, 7, 6, 5, 4, 3, 1, 3, 4, 5, 6, 7, 8, 8 ]; | |
710 | const COOK_CPL_4BITS_CODES: &[u16; 15] = &[ | |
711 | 0xfc, 0xfd, 0x7c, 0x3c, 0x1c, 0x0c, 0x04, 0x00, | |
712 | 0x05, 0x0d, 0x1d, 0x3d, 0x7d, 0xfe, 0xff | |
713 | ]; | |
714 | const COOK_CPL_5BITS_BITS: &[u8; 31] = &[ | |
715 | 10, 10, 10, 10, 9, 9, 8, 8, 7, 7, 6, 6, 5, 5, 3, 1, | |
716 | 3, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 10, 10 | |
717 | ]; | |
718 | const COOK_CPL_5BITS_CODES: &[u16; 31] = &[ | |
719 | 0x03F8, 0x03F9, 0x03FA, 0x03FB, 0x01F8, 0x01F9, 0x00F8, 0x00F9, | |
720 | 0x0078, 0x0079, 0x0038, 0x0039, 0x0018, 0x0019, 0x0004, 0x0000, | |
721 | 0x0005, 0x001A, 0x001B, 0x003A, 0x003B, 0x007A, 0x007B, 0x00FA, | |
722 | 0x00FB, 0x01FA, 0x01FB, 0x03FC, 0x03FD, 0x03FE, 0x03FF | |
723 | ]; | |
724 | const COOK_CPL_6BITS_BITS: &[u8; 63] = &[ | |
725 | 16, 15, 14, 13, 12, 11, 11, 11, 11, 10, 10, 10, 10, 9, 9, 9, | |
726 | 9, 9, 8, 8, 8, 8, 7, 7, 7, 7, 6, 6, 5, 5, 3, 1, | |
727 | 4, 5, 5, 6, 6, 7, 7, 7, 7, 8, 8, 8, 8, 9, 9, 9, | |
728 | 9, 10, 10, 10, 10, 10, 11, 11, 11, 11, 12, 13, 14, 14, 16 | |
729 | ]; | |
730 | const COOK_CPL_6BITS_CODES: &[u16; 63] = &[ | |
731 | 0xFFFE, 0x7FFE, 0x3FFC, 0x1FFC, 0x0FFC, 0x07F6, 0x07F7, 0x07F8, | |
732 | 0x07F9, 0x03F2, 0x03F3, 0x03F4, 0x03F5, 0x01F0, 0x01F1, 0x01F2, | |
733 | 0x01F3, 0x01F4, 0x00F0, 0x00F1, 0x00F2, 0x00F3, 0x0070, 0x0071, | |
734 | 0x0072, 0x0073, 0x0034, 0x0035, 0x0016, 0x0017, 0x0004, 0x0000, | |
735 | 0x000A, 0x0018, 0x0019, 0x0036, 0x0037, 0x0074, 0x0075, 0x0076, | |
736 | 0x0077, 0x00F4, 0x00F5, 0x00F6, 0x00F7, 0x01F5, 0x01F6, 0x01F7, | |
737 | 0x01F8, 0x03F6, 0x03F7, 0x03F8, 0x03F9, 0x03FA, 0x07FA, 0x07FB, | |
738 | 0x07FC, 0x07FD, 0x0FFD, 0x1FFD, 0x3FFD, 0x3FFE, 0xFFFF | |
739 | ]; | |
740 | ||
741 | const COOK_QUANT_BITS: [&[u8; 24]; 13] = [ | |
742 | &[ 4, 6, 5, 5, 4, 4, 4, 4, 4, 4, 3, 3, 3, 4, 5, 7, 8, 9, 11, 11, 12, 12, 12, 12 ], | |
743 | &[ 10, 8, 6, 5, 5, 4, 3, 3, 3, 3, 3, 3, 4, 5, 7, 9, 11, 12, 13, 15, 15, 15, 16, 16 ], | |
744 | &[ 12, 10, 8, 6, 5, 4, 4, 4, 4, 4, 4, 3, 3, 3, 4, 4, 5, 5, 7, 9, 11, 13, 14, 14 ], | |
745 | &[ 13, 10, 9, 9, 7, 7, 5, 5, 4, 3, 3, 3, 3, 3, 4, 4, 4, 5, 7, 9, 11, 13, 13, 13 ], | |
746 | &[ 12, 13, 10, 8, 6, 6, 5, 5, 4, 4, 3, 3, 3, 3, 3, 4, 5, 5, 6, 7, 9, 11, 14, 14 ], | |
747 | &[ 12, 11, 9, 8, 8, 7, 5, 4, 4, 3, 3, 3, 3, 3, 4, 4, 5, 5, 7, 8, 10, 13, 14, 14 ], | |
748 | &[ 15, 16, 15, 12, 10, 8, 6, 5, 4, 3, 3, 3, 2, 3, 4, 5, 5, 7, 9, 11, 13, 16, 16, 16 ], | |
749 | &[ 14, 14, 11, 10, 9, 7, 7, 5, 5, 4, 3, 3, 2, 3, 3, 4, 5, 7, 9, 9, 12, 14, 15, 15 ], | |
750 | &[ 9, 9, 9, 8, 7, 6, 5, 4, 3, 3, 3, 3, 3, 3, 4, 5, 6, 7, 8, 10, 11, 12, 13, 13 ], | |
751 | &[ 14, 12, 10, 8, 6, 6, 5, 4, 3, 3, 3, 3, 3, 3, 4, 5, 6, 8, 8, 9, 11, 14, 14, 14 ], | |
752 | &[ 13, 10, 9, 8, 6, 6, 5, 4, 4, 4, 3, 3, 2, 3, 4, 5, 6, 8, 9, 9, 11, 12, 14, 14 ], | |
753 | &[ 16, 13, 12, 11, 9, 6, 5, 5, 4, 4, 4, 3, 2, 3, 3, 4, 5, 7, 8, 10, 14, 16, 16, 16 ], | |
754 | &[ 13, 14, 14, 14, 10, 8, 7, 7, 5, 4, 3, 3, 2, 3, 3, 4, 5, 5, 7, 9, 11, 14, 14, 14 ], | |
755 | ]; | |
756 | const COOK_QUANT_CODES: [&[u16; 24]; 13] = [ | |
757 | &[ 0x0006, 0x003e, 0x001c, 0x001d, 0x0007, 0x0008, 0x0009, 0x000a, 0x000b, 0x000c, 0x0000, 0x0001, | |
758 | 0x0002, 0x000d, 0x001e, 0x007e, 0x00fe, 0x01fe, 0x07fc, 0x07fd, 0x0ffc, 0x0ffd, 0x0ffe, 0x0fff ], | |
759 | &[ 0x03fe, 0x00fe, 0x003e, 0x001c, 0x001d, 0x000c, 0x0000, 0x0001, 0x0002, 0x0003, 0x0004, 0x0005, | |
760 | 0x000d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x0ffe, 0x1ffe, 0x7ffc, 0x7ffd, 0x7ffe, 0xfffe, 0xffff ], | |
761 | &[ 0x0ffe, 0x03fe, 0x00fe, 0x003e, 0x001c, 0x0006, 0x0007, 0x0008, 0x0009, 0x000a, 0x000b, 0x0000, | |
762 | 0x0001, 0x0002, 0x000c, 0x000d, 0x001d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x1ffe, 0x3ffe, 0x3fff ], | |
763 | &[ 0x1ffc, 0x03fe, 0x01fc, 0x01fd, 0x007c, 0x007d, 0x001c, 0x001d, 0x000a, 0x0000, 0x0001, 0x0002, | |
764 | 0x0003, 0x0004, 0x000b, 0x000c, 0x000d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x1ffd, 0x1ffe, 0x1fff ], | |
765 | &[ 0x0ffe, 0x1ffe, 0x03fe, 0x00fe, 0x003c, 0x003d, 0x001a, 0x001b, 0x000a, 0x000b, 0x0000, 0x0001, | |
766 | 0x0002, 0x0003, 0x0004, 0x000c, 0x001c, 0x001d, 0x003e, 0x007e, 0x01fe, 0x07fe, 0x3ffe, 0x3fff ], | |
767 | &[ 0x0ffe, 0x07fe, 0x01fe, 0x00fc, 0x00fd, 0x007c, 0x001c, 0x000a, 0x000b, 0x0000, 0x0001, 0x0002, | |
768 | 0x0003, 0x0004, 0x000c, 0x000d, 0x001d, 0x001e, 0x007d, 0x00fe, 0x03fe, 0x1ffe, 0x3ffe, 0x3fff ], | |
769 | &[ 0x7ffc, 0xfffc, 0x7ffd, 0x0ffe, 0x03fe, 0x00fe, 0x003e, 0x001c, 0x000c, 0x0002, 0x0003, 0x0004, | |
770 | 0x0000, 0x0005, 0x000d, 0x001d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x1ffe, 0xfffd, 0xfffe, 0xffff ], | |
771 | &[ 0x3ffc, 0x3ffd, 0x07fe, 0x03fe, 0x01fc, 0x007c, 0x007d, 0x001c, 0x001d, 0x000c, 0x0002, 0x0003, | |
772 | 0x0000, 0x0004, 0x0005, 0x000d, 0x001e, 0x007e, 0x01fd, 0x01fe, 0x0ffe, 0x3ffe, 0x7ffe, 0x7fff ], | |
773 | &[ 0x01fc, 0x01fd, 0x01fe, 0x00fc, 0x007c, 0x003c, 0x001c, 0x000c, 0x0000, 0x0001, 0x0002, 0x0003, | |
774 | 0x0004, 0x0005, 0x000d, 0x001d, 0x003d, 0x007d, 0x00fd, 0x03fe, 0x07fe, 0x0ffe, 0x1ffe, 0x1fff ], | |
775 | &[ 0x3ffc, 0x0ffe, 0x03fe, 0x00fc, 0x003c, 0x003d, 0x001c, 0x000c, 0x0000, 0x0001, 0x0002, 0x0003, | |
776 | 0x0004, 0x0005, 0x000d, 0x001d, 0x003e, 0x00fd, 0x00fe, 0x01fe, 0x07fe, 0x3ffd, 0x3ffe, 0x3fff ], | |
777 | &[ 0x1ffe, 0x03fe, 0x01fc, 0x00fc, 0x003c, 0x003d, 0x001c, 0x000a, 0x000b, 0x000c, 0x0002, 0x0003, | |
778 | 0x0000, 0x0004, 0x000d, 0x001d, 0x003e, 0x00fd, 0x01fd, 0x01fe, 0x07fe, 0x0ffe, 0x3ffe, 0x3fff ], | |
779 | &[ 0xfffc, 0x1ffe, 0x0ffe, 0x07fe, 0x01fe, 0x003e, 0x001c, 0x001d, 0x000a, 0x000b, 0x000c, 0x0002, | |
780 | 0x0000, 0x0003, 0x0004, 0x000d, 0x001e, 0x007e, 0x00fe, 0x03fe, 0x3ffe, 0xfffd, 0xfffe, 0xffff ], | |
781 | &[ 0x1ffc, 0x3ffa, 0x3ffb, 0x3ffc, 0x03fe, 0x00fe, 0x007c, 0x007d, 0x001c, 0x000c, 0x0002, 0x0003, | |
782 | 0x0000, 0x0004, 0x0005, 0x000d, 0x001d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x3ffd, 0x3ffe, 0x3fff ], | |
783 | ]; | |
784 | ||
785 | const COOK_VQ0_BITS: &[u8; 191] = &[ | |
786 | 1, 4, 6, 6, 7, 7, 8, 8, 8, 9, 9, 10, | |
787 | 11, 11, 4, 5, 6, 7, 7, 8, 8, 9, 9, 9, | |
788 | 9, 10, 11, 11, 5, 6, 7, 8, 8, 9, 9, 9, | |
789 | 9, 10, 10, 10, 11, 12, 6, 7, 8, 9, 9, 9, | |
790 | 9, 10, 10, 10, 10, 11, 12, 13, 7, 7, 8, 9, | |
791 | 9, 9, 10, 10, 10, 10, 11, 11, 12, 13, 8, 8, | |
792 | 9, 9, 9, 10, 10, 10, 10, 11, 11, 12, 13, 14, | |
793 | 8, 8, 9, 9, 10, 10, 11, 11, 11, 12, 12, 13, | |
794 | 13, 15, 8, 8, 9, 9, 10, 10, 11, 11, 11, 12, | |
795 | 12, 13, 14, 15, 9, 9, 9, 10, 10, 10, 11, 11, | |
796 | 12, 13, 12, 14, 15, 16, 9, 9, 10, 10, 10, 10, | |
797 | 11, 12, 12, 14, 14, 16, 16, 0, 9, 9, 10, 10, | |
798 | 11, 11, 12, 13, 13, 14, 14, 15, 0, 0, 10, 10, | |
799 | 10, 11, 11, 12, 12, 13, 15, 15, 16, 0, 0, 0, | |
800 | 11, 11, 11, 12, 13, 13, 13, 15, 16, 16, 0, 0, | |
801 | 0, 0, 11, 11, 12, 13, 13, 14, 15, 16, 16 | |
802 | ]; | |
803 | const COOK_VQ0_CODES: &[u16; 191] = &[ | |
804 | 0x0000, 0x0008, 0x002c, 0x002d, 0x0062, 0x0063, 0x00d4, 0x00d5, | |
805 | 0x00d6, 0x01c6, 0x01c7, 0x03ca, 0x07d6, 0x07d7, 0x0009, 0x0014, | |
806 | 0x002e, 0x0064, 0x0065, 0x00d7, 0x00d8, 0x01c8, 0x01c9, 0x01ca, | |
807 | 0x01cb, 0x03cb, 0x07d8, 0x07d9, 0x0015, 0x002f, 0x0066, 0x00d9, | |
808 | 0x00da, 0x01cc, 0x01cd, 0x01ce, 0x01cf, 0x03cc, 0x03cd, 0x03ce, | |
809 | 0x07da, 0x0fe4, 0x0030, 0x0067, 0x00db, 0x01d0, 0x01d1, 0x01d2, | |
810 | 0x01d3, 0x03cf, 0x03d0, 0x03d1, 0x03d2, 0x07db, 0x0fe5, 0x1fea, | |
811 | 0x0068, 0x0069, 0x00dc, 0x01d4, 0x01d5, 0x01d6, 0x03d3, 0x03d4, | |
812 | 0x03d5, 0x03d6, 0x07dc, 0x07dd, 0x0fe6, 0x1feb, 0x00dd, 0x00de, | |
813 | 0x01d7, 0x01d8, 0x01d9, 0x03d7, 0x03d8, 0x03d9, 0x03da, 0x07de, | |
814 | 0x07df, 0x0fe7, 0x1fec, 0x3ff2, 0x00df, 0x00e0, 0x01da, 0x01db, | |
815 | 0x03db, 0x03dc, 0x07e0, 0x07e1, 0x07e2, 0x0fe8, 0x0fe9, 0x1fed, | |
816 | 0x1fee, 0x7ff4, 0x00e1, 0x00e2, 0x01dc, 0x01dd, 0x03dd, 0x03de, | |
817 | 0x07e3, 0x07e4, 0x07e5, 0x0fea, 0x0feb, 0x1fef, 0x3ff3, 0x7ff5, | |
818 | 0x01de, 0x01df, 0x01e0, 0x03df, 0x03e0, 0x03e1, 0x07e6, 0x07e7, | |
819 | 0x0fec, 0x1ff0, 0x0fed, 0x3ff4, 0x7ff6, 0xfff8, 0x01e1, 0x01e2, | |
820 | 0x03e2, 0x03e3, 0x03e4, 0x03e5, 0x07e8, 0x0fee, 0x0fef, 0x3ff5, | |
821 | 0x3ff6, 0xfff9, 0xfffa, 0xfffa, 0x01e3, 0x01e4, 0x03e6, 0x03e7, | |
822 | 0x07e9, 0x07ea, 0x0ff0, 0x1ff1, 0x1ff2, 0x3ff7, 0x3ff8, 0x7ff7, | |
823 | 0x7ff7, 0xfffa, 0x03e8, 0x03e9, 0x03ea, 0x07eb, 0x07ec, 0x0ff1, | |
824 | 0x0ff2, 0x1ff3, 0x7ff8, 0x7ff9, 0xfffb, 0x3ff8, 0x7ff7, 0x7ff7, | |
825 | 0x07ed, 0x07ee, 0x07ef, 0x0ff3, 0x1ff4, 0x1ff5, 0x1ff6, 0x7ffa, | |
826 | 0xfffc, 0xfffd, 0xfffb, 0xfffb, 0x3ff8, 0x7ff7, 0x07f0, 0x07f1, | |
827 | 0x0ff4, 0x1ff7, 0x1ff8, 0x3ff9, 0x7ffb, 0xfffe, 0xffff | |
828 | ]; | |
829 | const COOK_VQ1_BITS: &[u8; 97] = &[ | |
830 | 1, 4, 5, 6, 7, 8, 8, 9, 10, 10, 4, 5, | |
831 | 6, 7, 7, 8, 8, 9, 9, 11, 5, 5, 6, 7, | |
832 | 8, 8, 9, 9, 10, 11, 6, 6, 7, 8, 8, 9, | |
833 | 9, 10, 11, 12, 7, 7, 8, 8, 9, 9, 10, 11, | |
834 | 11, 13, 8, 8, 8, 9, 9, 10, 10, 11, 12, 14, | |
835 | 8, 8, 8, 9, 10, 11, 11, 12, 13, 15, 9, 9, | |
836 | 9, 10, 11, 12, 12, 14, 14, 0, 9, 9, 9, 10, | |
837 | 11, 12, 14, 16, 0, 0, 10, 10, 11, 12, 13, 14, 16 | |
838 | ]; | |
839 | const COOK_VQ1_CODES: &[u16; 97] = &[ | |
840 | 0x0000, 0x0008, 0x0014, 0x0030, 0x006a, 0x00e2, 0x00e3, 0x01e4, | |
841 | 0x03ec, 0x03ed, 0x0009, 0x0015, 0x0031, 0x006b, 0x006c, 0x00e4, | |
842 | 0x00e5, 0x01e5, 0x01e6, 0x07f0, 0x0016, 0x0017, 0x0032, 0x006d, | |
843 | 0x00e6, 0x00e7, 0x01e7, 0x01e8, 0x03ee, 0x07f1, 0x0033, 0x0034, | |
844 | 0x006e, 0x00e8, 0x00e9, 0x01e9, 0x01ea, 0x03ef, 0x07f2, 0x0ff6, | |
845 | 0x006f, 0x0070, 0x00ea, 0x00eb, 0x01eb, 0x01ec, 0x03f0, 0x07f3, | |
846 | 0x07f4, 0x1ffa, 0x00ec, 0x00ed, 0x00ee, 0x01ed, 0x01ee, 0x03f1, | |
847 | 0x03f2, 0x07f5, 0x0ff7, 0x3ffa, 0x00ef, 0x00f0, 0x00f1, 0x01ef, | |
848 | 0x03f3, 0x07f6, 0x07f7, 0x0ff8, 0x1ffb, 0x7ffe, 0x01f0, 0x01f1, | |
849 | 0x01f2, 0x03f4, 0x07f8, 0x0ff9, 0x0ffa, 0x3ffb, 0x3ffc, 0x0000, | |
850 | 0x01f3, 0x01f4, 0x01f5, 0x03f5, 0x07f9, 0x0ffb, 0x3ffd, 0xfffe, | |
851 | 0x0000, 0x0000, 0x03f6, 0x03f7, 0x07fa, 0x0ffc, 0x1ffc, 0x3ffe, | |
852 | 0xffff | |
853 | ]; | |
854 | const COOK_VQ2_BITS: &[u8; 48] = &[ | |
855 | 1, 4, 5, 7, 8, 9, 10, 3, 4, 5, 7, 8, | |
856 | 9, 10, 5, 5, 6, 7, 8, 10, 10, 7, 6, 7, | |
857 | 8, 9, 10, 12, 8, 8, 8, 9, 10, 12, 14, 8, | |
858 | 9, 9, 10, 11, 15, 16, 9, 10, 11, 12, 13, 16 | |
859 | ]; | |
860 | const COOK_VQ2_CODES: &[u16; 48] = &[ | |
861 | 0x0000, 0x000a, 0x0018, 0x0074, 0x00f2, 0x01f4, 0x03f6, 0x0004, 0x000b, 0x0019, 0x0075, 0x00f3, | |
862 | 0x01f5, 0x03f7, 0x001a, 0x001b, 0x0038, 0x0076, 0x00f4, 0x03f8, 0x03f9, 0x0077, 0x0039, 0x0078, | |
863 | 0x00f5, 0x01f6, 0x03fa, 0x0ffc, 0x00f6, 0x00f7, 0x00f8, 0x01f7, 0x03fb, 0x0ffd, 0x3ffe, 0x00f9, | |
864 | 0x01f8, 0x01f9, 0x03fc, 0x07fc, 0x7ffe, 0xfffe, 0x01fa, 0x03fd, 0x07fd, 0x0ffe, 0x1ffe, 0xffff | |
865 | ]; | |
866 | const COOK_VQ3_BITS: &[u8; 607] = &[ | |
867 | 2, 4, 6, 8, 10, 5, 5, 6, 8, 10, 7, 8, | |
868 | 8, 10, 12, 9, 9, 10, 12, 15, 10, 11, 13, 16, | |
869 | 16, 5, 6, 8, 10, 11, 5, 6, 8, 10, 12, 7, | |
870 | 7, 8, 10, 13, 9, 9, 10, 12, 15, 12, 11, 13, | |
871 | 16, 16, 7, 9, 10, 12, 15, 7, 8, 10, 12, 13, | |
872 | 9, 9, 11, 13, 16, 11, 11, 12, 14, 16, 12, 12, | |
873 | 14, 16, 0, 9, 11, 12, 16, 16, 9, 10, 13, 15, | |
874 | 16, 10, 11, 12, 16, 16, 13, 13, 16, 16, 16, 16, | |
875 | 16, 15, 16, 0, 11, 13, 16, 16, 15, 11, 13, 15, | |
876 | 16, 16, 13, 13, 16, 16, 0, 14, 16, 16, 16, 0, | |
877 | 16, 16, 0, 0, 0, 4, 6, 8, 10, 13, 6, 6, | |
878 | 8, 10, 13, 9, 8, 10, 12, 16, 10, 10, 11, 15, | |
879 | 16, 13, 12, 14, 16, 16, 5, 6, 8, 11, 13, 6, | |
880 | 6, 8, 10, 13, 8, 8, 9, 11, 14, 10, 10, 12, | |
881 | 12, 16, 13, 12, 13, 15, 16, 7, 8, 9, 12, 16, | |
882 | 7, 8, 10, 12, 14, 9, 9, 10, 13, 16, 11, 10, | |
883 | 12, 15, 16, 13, 13, 16, 16, 0, 9, 11, 13, 16, | |
884 | 16, 9, 10, 12, 15, 16, 10, 11, 13, 16, 16, 13, | |
885 | 12, 16, 16, 16, 16, 16, 16, 16, 0, 11, 13, 16, | |
886 | 16, 16, 11, 13, 16, 16, 16, 12, 13, 15, 16, 0, | |
887 | 16, 16, 16, 16, 0, 16, 16, 0, 0, 0, 6, 8, | |
888 | 11, 13, 16, 8, 8, 10, 12, 16, 11, 10, 11, 13, | |
889 | 16, 12, 13, 13, 15, 16, 16, 16, 14, 16, 0, 6, | |
890 | 8, 10, 13, 16, 8, 8, 10, 12, 16, 10, 10, 11, | |
891 | 13, 16, 13, 12, 13, 16, 16, 14, 14, 14, 16, 0, | |
892 | 8, 9, 11, 13, 16, 8, 9, 11, 16, 14, 10, 10, | |
893 | 12, 15, 16, 12, 12, 13, 16, 16, 15, 16, 16, 16, | |
894 | 0, 10, 12, 15, 16, 16, 10, 12, 12, 14, 16, 12, | |
895 | 12, 13, 16, 16, 14, 15, 16, 16, 0, 16, 16, 16, | |
896 | 0, 0, 12, 15, 15, 16, 0, 13, 13, 16, 16, 0, | |
897 | 14, 16, 16, 16, 0, 16, 16, 16, 0, 0, 0, 0, | |
898 | 0, 0, 0, 8, 10, 13, 15, 16, 10, 11, 13, 16, | |
899 | 16, 13, 13, 14, 16, 16, 16, 16, 16, 16, 16, 16, | |
900 | 16, 16, 16, 0, 8, 10, 11, 15, 16, 9, 10, 12, | |
901 | 16, 16, 12, 12, 15, 16, 16, 16, 14, 16, 16, 16, | |
902 | 16, 16, 16, 16, 0, 9, 11, 14, 16, 16, 10, 11, | |
903 | 13, 16, 16, 14, 13, 14, 16, 16, 16, 15, 15, 16, | |
904 | 0, 16, 16, 16, 0, 0, 11, 13, 16, 16, 16, 11, | |
905 | 13, 15, 16, 16, 13, 16, 16, 16, 0, 16, 16, 16, | |
906 | 16, 0, 16, 16, 0, 0, 0, 15, 16, 16, 16, 0, | |
907 | 14, 16, 16, 16, 0, 16, 16, 16, 0, 0, 16, 16, | |
908 | 0, 0, 0, 0, 0, 0, 0, 0, 9, 13, 16, 16, | |
909 | 16, 11, 13, 16, 16, 16, 14, 15, 16, 16, 0, 15, | |
910 | 16, 16, 16, 0, 16, 16, 0, 0, 0, 9, 13, 15, | |
911 | 15, 16, 12, 13, 14, 16, 16, 16, 15, 16, 16, 0, | |
912 | 16, 16, 16, 16, 0, 16, 16, 0, 0, 0, 11, 13, | |
913 | 15, 16, 0, 12, 14, 16, 16, 0, 16, 16, 16, 16, | |
914 | 0, 16, 16, 16, 0, 0, 0, 0, 0, 0, 0, 16, | |
915 | 16, 16, 16, 0, 16, 16, 16, 16, 0, 16, 16, 16, | |
916 | 0, 0, 16, 16, 0, 0, 0, 0, 0, 0, 0, 0, | |
917 | 16, 16, 0, 0, 0, 16, 16 | |
918 | ]; | |
919 | const COOK_VQ3_CODES: &[u16; 607] = &[ | |
920 | 0x0000, 0x0004, 0x0022, 0x00c6, 0x03b0, 0x000c, 0x000d, 0x0023, 0x00c7, 0x03b1, 0x005c, 0x00c8, | |
921 | 0x00c9, 0x03b2, 0x0fa4, 0x01c2, 0x01c3, 0x03b3, 0x0fa5, 0x7f72, 0x03b4, 0x07b2, 0x1f9a, 0xff24, | |
922 | 0xff25, 0x000e, 0x0024, 0x00ca, 0x03b5, 0x07b3, 0x000f, 0x0025, 0x00cb, 0x03b6, 0x0fa6, 0x005d, | |
923 | 0x005e, 0x00cc, 0x03b7, 0x1f9b, 0x01c4, 0x01c5, 0x03b8, 0x0fa7, 0x7f73, 0x0fa8, 0x07b4, 0x1f9c, | |
924 | 0xff26, 0xff27, 0x005f, 0x01c6, 0x03b9, 0x0fa9, 0x7f74, 0x0060, 0x00cd, 0x03ba, 0x0faa, 0x1f9d, | |
925 | 0x01c7, 0x01c8, 0x07b5, 0x1f9e, 0xff28, 0x07b6, 0x07b7, 0x0fab, 0x3fa2, 0xff29, 0x0fac, 0x0fad, | |
926 | 0x3fa3, 0xff2a, 0x3fa2, 0x01c9, 0x07b8, 0x0fae, 0xff2b, 0xff2c, 0x01ca, 0x03bb, 0x1f9f, 0x7f75, | |
927 | 0xff2d, 0x03bc, 0x07b9, 0x0faf, 0xff2e, 0xff2f, 0x1fa0, 0x1fa1, 0xff30, 0xff31, 0xff32, 0xff33, | |
928 | 0xff34, 0x7f76, 0xff35, 0xff31, 0x07ba, 0x1fa2, 0xff36, 0xff37, 0x7f77, 0x07bb, 0x1fa3, 0x7f78, | |
929 | 0xff38, 0xff39, 0x1fa4, 0x1fa5, 0xff3a, 0xff3b, 0xff2e, 0x3fa4, 0xff3c, 0xff3d, 0xff3e, 0xff31, | |
930 | 0xff3f, 0xff40, 0xff30, 0xff31, 0xff31, 0x0005, 0x0026, 0x00ce, 0x03bd, 0x1fa6, 0x0027, 0x0028, | |
931 | 0x00cf, 0x03be, 0x1fa7, 0x01cb, 0x00d0, 0x03bf, 0x0fb0, 0xff41, 0x03c0, 0x03c1, 0x07bc, 0x7f79, | |
932 | 0xff42, 0x1fa8, 0x0fb1, 0x3fa5, 0xff43, 0xff44, 0x0010, 0x0029, 0x00d1, 0x07bd, 0x1fa9, 0x002a, | |
933 | 0x002b, 0x00d2, 0x03c2, 0x1faa, 0x00d3, 0x00d4, 0x01cc, 0x07be, 0x3fa6, 0x03c3, 0x03c4, 0x0fb2, | |
934 | 0x0fb3, 0xff45, 0x1fab, 0x0fb4, 0x1fac, 0x7f7a, 0xff46, 0x0061, 0x00d5, 0x01cd, 0x0fb5, 0xff47, | |
935 | 0x0062, 0x00d6, 0x03c5, 0x0fb6, 0x3fa7, 0x01ce, 0x01cf, 0x03c6, 0x1fad, 0xff48, 0x07bf, 0x03c7, | |
936 | 0x0fb7, 0x7f7b, 0xff49, 0x1fae, 0x1faf, 0xff4a, 0xff4b, 0x7f7b, 0x01d0, 0x07c0, 0x1fb0, 0xff4c, | |
937 | 0xff4d, 0x01d1, 0x03c8, 0x0fb8, 0x7f7c, 0xff4e, 0x03c9, 0x07c1, 0x1fb1, 0xff4f, 0xff50, 0x1fb2, | |
938 | 0x0fb9, 0xff51, 0xff52, 0xff53, 0xff54, 0xff55, 0xff56, 0xff57, 0xff52, 0x07c2, 0x1fb3, 0xff58, | |
939 | 0xff59, 0xff5a, 0x07c3, 0x1fb4, 0xff5b, 0xff5c, 0xff5d, 0x0fba, 0x1fb5, 0x7f7d, 0xff5e, 0xff4f, | |
940 | 0xff5f, 0xff60, 0xff61, 0xff62, 0xff52, 0xff63, 0xff64, 0xff51, 0xff52, 0xff52, 0x002c, 0x00d7, | |
941 | 0x07c4, 0x1fb6, 0xff65, 0x00d8, 0x00d9, 0x03ca, 0x0fbb, 0xff66, 0x07c5, 0x03cb, 0x07c6, 0x1fb7, | |
942 | 0xff67, 0x0fbc, 0x1fb8, 0x1fb9, 0x7f7e, 0xff68, 0xff69, 0xff6a, 0x3fa8, 0xff6b, 0x7f7e, 0x002d, | |
943 | 0x00da, 0x03cc, 0x1fba, 0xff6c, 0x00db, 0x00dc, 0x03cd, 0x0fbd, 0xff6d, 0x03ce, 0x03cf, 0x07c7, | |
944 | 0x1fbb, 0xff6e, 0x1fbc, 0x0fbe, 0x1fbd, 0xff6f, 0xff70, 0x3fa9, 0x3faa, 0x3fab, 0xff71, 0xff6f, | |
945 | 0x00dd, 0x01d2, 0x07c8, 0x1fbe, 0xff72, 0x00de, 0x01d3, 0x07c9, 0xff73, 0x3fac, 0x03d0, 0x03d1, | |
946 | 0x0fbf, 0x7f7f, 0xff74, 0x0fc0, 0x0fc1, 0x1fbf, 0xff75, 0xff76, 0x7f80, 0xff77, 0xff78, 0xff79, | |
947 | 0xff75, 0x03d2, 0x0fc2, 0x7f81, 0xff7a, 0xff7b, 0x03d3, 0x0fc3, 0x0fc4, 0x3fad, 0xff7c, 0x0fc5, | |
948 | 0x0fc6, 0x1fc0, 0xff7d, 0xff7e, 0x3fae, 0x7f82, 0xff7f, 0xff80, 0xff80, 0xff81, 0xff82, 0xff83, | |
949 | 0xff80, 0xff80, 0x0fc7, 0x7f83, 0x7f84, 0xff84, 0xff7a, 0x1fc1, 0x1fc2, 0xff85, 0xff86, 0x3fad, | |
950 | 0x3faf, 0xff87, 0xff88, 0xff89, 0xff7d, 0xff8a, 0xff8b, 0xff8c, 0xff80, 0xff80, 0x3fae, 0x7f82, | |
951 | 0xff7f, 0xff80, 0xff80, 0x00df, 0x03d4, 0x1fc3, 0x7f85, 0xff8d, 0x03d5, 0x07ca, 0x1fc4, 0xff8e, | |
952 | 0xff8f, 0x1fc5, 0x1fc6, 0x3fb0, 0xff90, 0xff91, 0xff92, 0xff93, 0xff94, 0xff95, 0xff96, 0xff97, | |
953 | 0xff98, 0xff99, 0xff9a, 0xff95, 0x00e0, 0x03d6, 0x07cb, 0x7f86, 0xff9b, 0x01d4, 0x03d7, 0x0fc8, | |
954 | 0xff9c, 0xff9d, 0x0fc9, 0x0fca, 0x7f87, 0xff9e, 0xff9f, 0xffa0, 0x3fb1, 0xffa1, 0xffa2, 0xffa3, | |
955 | 0xffa4, 0xffa5, 0xffa6, 0xffa7, 0xffa2, 0x01d5, 0x07cc, 0x3fb2, 0xffa8, 0xffa9, 0x03d8, 0x07cd, | |
956 | 0x1fc7, 0xffaa, 0xffab, 0x3fb3, 0x1fc8, 0x3fb4, 0xffac, 0xffad, 0xffae, 0x7f88, 0x7f89, 0xffaf, | |
957 | 0xffaf, 0xffb0, 0xffb1, 0xffb2, 0xffaf, 0xffaf, 0x07ce, 0x1fc9, 0xffb3, 0xffb4, 0xffb5, 0x07cf, | |
958 | 0x1fca, 0x7f8a, 0xffb6, 0xffb7, 0x1fcb, 0xffb8, 0xffb9, 0xffba, 0xffba, 0xffbb, 0xffbc, 0xffbd, | |
959 | 0xffbe, 0xffbe, 0xffbf, 0xffc0, 0xffbd, 0xffbe, 0xffbe, 0x7f8b, 0xffc1, 0xffc2, 0xffc3, 0xffb4, | |
960 | 0x3fb5, 0xffc4, 0xffc5, 0xffc6, 0xffb6, 0xffc7, 0xffc8, 0xffc9, 0xffba, 0xffba, 0xffca, 0xffcb, | |
961 | 0xffbd, 0xffbe, 0xffbe, 0xffbb, 0xffbc, 0xffbd, 0xffbe, 0xffbe, 0x01d6, 0x1fcc, 0xffcc, 0xffcd, | |
962 | 0xffce, 0x07d0, 0x1fcd, 0xffcf, 0xffd0, 0xffd1, 0x3fb6, 0x7f8c, 0xffd2, 0xffd3, 0xff90, 0x7f8d, | |
963 | 0xffd4, 0xffd5, 0xffd6, 0xff95, 0xffd7, 0xffd8, 0xff94, 0xff95, 0xff95, 0x01d7, 0x1fce, 0x7f8e, | |
964 | 0x7f8f, 0xffd9, 0x0fcb, 0x1fcf, 0x3fb7, 0xffda, 0xffdb, 0xffdc, 0x7f90, 0xffdd, 0xffde, 0xff9e, | |
965 | 0xffdf, 0xffe0, 0xffe1, 0xffe2, 0xffa2, 0xffe3, 0xffe4, 0xffa1, 0xffa2, 0xffa2, 0x07d1, 0x1fd0, | |
966 | 0x7f91, 0xffe5, 0xffa8, 0x0fcc, 0x3fb8, 0xffe6, 0xffe7, 0xffaa, 0xffe8, 0xffe9, 0xffea, 0xffeb, | |
967 | 0xffac, 0xffec, 0xffed, 0xffee, 0xffaf, 0xffaf, 0xffae, 0x7f88, 0x7f89, 0xffaf, 0xffaf, 0xffef, | |
968 | 0xfff0, 0xfff1, 0xfff2, 0xffb4, 0xfff3, 0xfff4, 0xfff5, 0xfff6, 0xffb6, 0xfff7, 0xfff8, 0xfff9, | |
969 | 0xffba, 0xffba, 0xfffa, 0xfffb, 0xffbd, 0xffbe, 0xffbe, 0xffbb, 0xffbc, 0xffbd, 0xffbe, 0xffbe, | |
970 | 0xfffc, 0xfffd, 0xffb3, 0xffb4, 0xffb4, 0xfffe, 0xffff | |
971 | ]; | |
972 | const COOK_VQ4_BITS: &[u8; 246] = &[ | |
973 | 2, 4, 7, 10, 4, 5, 7, 10, 7, 8, 10, 14, | |
974 | 11, 11, 15, 15, 4, 5, 9, 12, 5, 5, 8, 12, | |
975 | 8, 7, 10, 15, 11, 11, 15, 15, 7, 9, 12, 15, | |
976 | 8, 8, 12, 15, 10, 10, 13, 15, 14, 14, 15, 0, | |
977 | 11, 13, 15, 15, 11, 13, 15, 15, 14, 15, 15, 0, | |
978 | 15, 15, 0, 0, 4, 5, 9, 13, 5, 6, 9, 13, | |
979 | 9, 9, 11, 15, 14, 13, 15, 15, 4, 6, 9, 12, | |
980 | 5, 6, 9, 13, 9, 8, 11, 15, 13, 12, 15, 15, | |
981 | 7, 9, 12, 15, 7, 8, 11, 15, 10, 10, 14, 15, | |
982 | 14, 15, 15, 0, 10, 12, 15, 15, 11, 13, 15, 15, | |
983 | 15, 15, 15, 0, 15, 15, 0, 0, 6, 9, 13, 14, | |
984 | 8, 9, 12, 15, 12, 12, 15, 15, 15, 15, 15, 0, | |
985 | 7, 9, 13, 15, 8, 9, 12, 15, 11, 12, 15, 15, | |
986 | 15, 15, 15, 0, 9, 11, 15, 15, 9, 11, 15, 15, | |
987 | 14, 14, 15, 0, 15, 15, 0, 0, 14, 15, 15, 0, | |
988 | 14, 15, 15, 0, 15, 15, 0, 0, 0, 0, 0, 0, | |
989 | 9, 12, 15, 15, 12, 13, 15, 15, 15, 15, 15, 0, | |
990 | 15, 15, 0, 0, 10, 12, 15, 15, 12, 14, 15, 15, | |
991 | 15, 15, 15, 0, 15, 15, 0, 0, 14, 15, 15, 0, | |
992 | 15, 15, 15, 0, 15, 15, 0, 0, 0, 0, 0, 0, | |
993 | 15, 15, 0, 0, 15, 15 | |
994 | ]; | |
995 | const COOK_VQ4_CODES: &[u16; 246] = &[ | |
996 | 0x0000, 0x0004, 0x006c, 0x03e6, 0x0005, 0x0012, 0x006d, 0x03e7, 0x006e, 0x00e8, 0x03e8, 0x3fc4, | |
997 | 0x07e0, 0x07e1, 0x7fa4, 0x7fa5, 0x0006, 0x0013, 0x01e2, 0x0fda, 0x0014, 0x0015, 0x00e9, 0x0fdb, | |
998 | 0x00ea, 0x006f, 0x03e9, 0x7fa6, 0x07e2, 0x07e3, 0x7fa7, 0x7fa8, 0x0070, 0x01e3, 0x0fdc, 0x7fa9, | |
999 | 0x00eb, 0x00ec, 0x0fdd, 0x7faa, 0x03ea, 0x03eb, 0x1fd6, 0x7fab, 0x3fc5, 0x3fc6, 0x7fac, 0x1fd6, | |
1000 | 0x07e4, 0x1fd7, 0x7fad, 0x7fae, 0x07e5, 0x1fd8, 0x7faf, 0x7fb0, 0x3fc7, 0x7fb1, 0x7fb2, 0x1fd6, | |
1001 | 0x7fb3, 0x7fb4, 0x1fd6, 0x1fd6, 0x0007, 0x0016, 0x01e4, 0x1fd9, 0x0017, 0x0032, 0x01e5, 0x1fda, | |
1002 | 0x01e6, 0x01e7, 0x07e6, 0x7fb5, 0x3fc8, 0x1fdb, 0x7fb6, 0x7fb7, 0x0008, 0x0033, 0x01e8, 0x0fde, | |
1003 | 0x0018, 0x0034, 0x01e9, 0x1fdc, 0x01ea, 0x00ed, 0x07e7, 0x7fb8, 0x1fdd, 0x0fdf, 0x7fb9, 0x7fba, | |
1004 | 0x0071, 0x01eb, 0x0fe0, 0x7fbb, 0x0072, 0x00ee, 0x07e8, 0x7fbc, 0x03ec, 0x03ed, 0x3fc9, 0x7fbd, | |
1005 | 0x3fca, 0x7fbe, 0x7fbf, 0x3fc9, 0x03ee, 0x0fe1, 0x7fc0, 0x7fc1, 0x07e9, 0x1fde, 0x7fc2, 0x7fc3, | |
1006 | 0x7fc4, 0x7fc5, 0x7fc6, 0x3fc9, 0x7fc7, 0x7fc8, 0x3fc9, 0x3fc9, 0x0035, 0x01ec, 0x1fdf, 0x3fcb, | |
1007 | 0x00ef, 0x01ed, 0x0fe2, 0x7fc9, 0x0fe3, 0x0fe4, 0x7fca, 0x7fcb, 0x7fcc, 0x7fcd, 0x7fce, 0x7fca, | |
1008 | 0x0073, 0x01ee, 0x1fe0, 0x7fcf, 0x00f0, 0x01ef, 0x0fe5, 0x7fd0, 0x07ea, 0x0fe6, 0x7fd1, 0x7fd2, | |
1009 | 0x7fd3, 0x7fd4, 0x7fd5, 0x7fd1, 0x01f0, 0x07eb, 0x7fd6, 0x7fd7, 0x01f1, 0x07ec, 0x7fd8, 0x7fd9, | |
1010 | 0x3fcc, 0x3fcd, 0x7fda, 0x7fda, 0x7fdb, 0x7fdc, 0x7fda, 0x7fda, 0x3fce, 0x7fdd, 0x7fde, 0x7fd6, | |
1011 | 0x3fcf, 0x7fdf, 0x7fe0, 0x7fd8, 0x7fe1, 0x7fe2, 0x7fda, 0x7fda, 0x3fcc, 0x3fcd, 0x7fda, 0x7fda, | |
1012 | 0x01f2, 0x0fe7, 0x7fe3, 0x7fe4, 0x0fe8, 0x1fe1, 0x7fe5, 0x7fe6, 0x7fe7, 0x7fe8, 0x7fe9, 0x7fca, | |
1013 | 0x7fea, 0x7feb, 0x7fca, 0x7fca, 0x03ef, 0x0fe9, 0x7fec, 0x7fed, 0x0fea, 0x3fd0, 0x7fee, 0x7fef, | |
1014 | 0x7ff0, 0x7ff1, 0x7ff2, 0x7fd1, 0x7ff3, 0x7ff4, 0x7fd1, 0x7fd1, 0x3fd1, 0x7ff5, 0x7ff6, 0x7fd6, | |
1015 | 0x7ff7, 0x7ff8, 0x7ff9, 0x7fd8, 0x7ffa, 0x7ffb, 0x7fda, 0x7fda, 0x3fcc, 0x3fcd, 0x7fda, 0x7fda, | |
1016 | 0x7ffc, 0x7ffd, 0x7fd6, 0x7fd6, 0x7ffe, 0x7fff | |
1017 | ]; | |
1018 | const COOK_VQ5_BITS: &[u8; 230] = &[ | |
1019 | 2, 4, 8, 4, 5, 9, 9, 10, 14, 4, 6, 11, | |
1020 | 5, 6, 12, 10, 11, 15, 9, 11, 15, 10, 13, 15, | |
1021 | 14, 15, 0, 4, 6, 12, 6, 7, 12, 12, 12, 15, | |
1022 | 5, 7, 13, 6, 7, 13, 12, 13, 15, 10, 12, 15, | |
1023 | 11, 13, 15, 15, 15, 0, 8, 13, 15, 11, 12, 15, | |
1024 | 15, 15, 0, 10, 13, 15, 12, 15, 15, 15, 15, 0, | |
1025 | 15, 15, 0, 15, 15, 0, 0, 0, 0, 4, 5, 11, | |
1026 | 5, 7, 12, 11, 12, 15, 6, 7, 13, 7, 8, 14, | |
1027 | 12, 14, 15, 11, 13, 15, 12, 13, 15, 15, 15, 0, | |
1028 | 5, 6, 13, 7, 8, 15, 12, 14, 15, 6, 8, 14, | |
1029 | 7, 8, 15, 14, 15, 15, 12, 12, 15, 12, 13, 15, | |
1030 | 15, 15, 0, 9, 13, 15, 12, 13, 15, 15, 15, 0, | |
1031 | 11, 13, 15, 13, 13, 15, 15, 15, 0, 14, 15, 0, | |
1032 | 15, 15, 0, 0, 0, 0, 8, 10, 15, 11, 12, 15, | |
1033 | 15, 15, 0, 10, 12, 15, 12, 13, 15, 15, 15, 0, | |
1034 | 14, 15, 0, 15, 15, 0, 0, 0, 0, 8, 12, 15, | |
1035 | 12, 13, 15, 15, 15, 0, 11, 13, 15, 13, 15, 15, | |
1036 | 15, 15, 0, 15, 15, 0, 15, 15, 0, 0, 0, 0, | |
1037 | 14, 15, 0, 15, 15, 0, 0, 0, 0, 15, 15, 0, | |
1038 | 15, 15 | |
1039 | ]; | |
1040 | const COOK_VQ5_CODES: &[u16; 230] = &[ | |
1041 | 0x0000, 0x0004, 0x00f0, 0x0005, 0x0012, 0x01f0, 0x01f1, 0x03e8, 0x3fce, 0x0006, 0x0030, 0x07de, | |
1042 | 0x0013, 0x0031, 0x0fd2, 0x03e9, 0x07df, 0x7fb0, 0x01f2, 0x07e0, 0x7fb1, 0x03ea, 0x1fd2, 0x7fb2, | |
1043 | 0x3fcf, 0x7fb3, 0x0031, 0x0007, 0x0032, 0x0fd3, 0x0033, 0x0070, 0x0fd4, 0x0fd5, 0x0fd6, 0x7fb4, | |
1044 | 0x0014, 0x0071, 0x1fd3, 0x0034, 0x0072, 0x1fd4, 0x0fd7, 0x1fd5, 0x7fb5, 0x03eb, 0x0fd8, 0x7fb6, | |
1045 | 0x07e1, 0x1fd6, 0x7fb7, 0x7fb8, 0x7fb9, 0x0072, 0x00f1, 0x1fd7, 0x7fba, 0x07e2, 0x0fd9, 0x7fbb, | |
1046 | 0x7fbc, 0x7fbd, 0x0070, 0x03ec, 0x1fd8, 0x7fbe, 0x0fda, 0x7fbf, 0x7fc0, 0x7fc1, 0x7fc2, 0x0072, | |
1047 | 0x7fc3, 0x7fc4, 0x0071, 0x7fc5, 0x7fc6, 0x0072, 0x0034, 0x0072, 0x0072, 0x0008, 0x0015, 0x07e3, | |
1048 | 0x0016, 0x0073, 0x0fdb, 0x07e4, 0x0fdc, 0x7fc7, 0x0035, 0x0074, 0x1fd9, 0x0075, 0x00f2, 0x3fd0, | |
1049 | 0x0fdd, 0x3fd1, 0x7fc8, 0x07e5, 0x1fda, 0x7fc9, 0x0fde, 0x1fdb, 0x7fca, 0x7fcb, 0x7fcc, 0x00f2, | |
1050 | 0x0017, 0x0036, 0x1fdc, 0x0076, 0x00f3, 0x7fcd, 0x0fdf, 0x3fd2, 0x7fce, 0x0037, 0x00f4, 0x3fd3, | |
1051 | 0x0077, 0x00f5, 0x7fcf, 0x3fd4, 0x7fd0, 0x7fd1, 0x0fe0, 0x0fe1, 0x7fd2, 0x0fe2, 0x1fdd, 0x7fd3, | |
1052 | 0x7fd4, 0x7fd5, 0x00f5, 0x01f3, 0x1fde, 0x7fd6, 0x0fe3, 0x1fdf, 0x7fd7, 0x7fd8, 0x7fd9, 0x00f3, | |
1053 | 0x07e6, 0x1fe0, 0x7fda, 0x1fe1, 0x1fe2, 0x7fdb, 0x7fdc, 0x7fdd, 0x00f5, 0x3fd5, 0x7fde, 0x00f4, | |
1054 | 0x7fdf, 0x7fe0, 0x00f5, 0x0077, 0x00f5, 0x00f5, 0x00f6, 0x03ed, 0x7fe1, 0x07e7, 0x0fe4, 0x7fe2, | |
1055 | 0x7fe3, 0x7fe4, 0x0073, 0x03ee, 0x0fe5, 0x7fe5, 0x0fe6, 0x1fe3, 0x7fe6, 0x7fe7, 0x7fe8, 0x00f2, | |
1056 | 0x3fd6, 0x7fe9, 0x0074, 0x7fea, 0x7feb, 0x00f2, 0x0075, 0x00f2, 0x00f2, 0x00f7, 0x0fe7, 0x7fec, | |
1057 | 0x0fe8, 0x1fe4, 0x7fed, 0x7fee, 0x7fef, 0x00f3, 0x07e8, 0x1fe5, 0x7ff0, 0x1fe6, 0x7ff1, 0x7ff2, | |
1058 | 0x7ff3, 0x7ff4, 0x00f5, 0x7ff5, 0x7ff6, 0x00f4, 0x7ff7, 0x7ff8, 0x00f5, 0x0077, 0x00f5, 0x00f5, | |
1059 | 0x3fd7, 0x7ff9, 0x0036, 0x7ffa, 0x7ffb, 0x00f3, 0x0076, 0x00f3, 0x00f3, 0x7ffc, 0x7ffd, 0x0000, | |
1060 | 0x7ffe, 0x7fff | |
1061 | ]; | |
1062 | const COOK_VQ6_BITS: &[u8; 32] = &[ | |
1063 | 1, 4, 4, 6, 4, 6, 6, 8, 4, 6, 6, 8, | |
1064 | 6, 9, 8, 10, 4, 6, 7, 8, 6, 9, 8, 11, | |
1065 | 6, 9, 8, 10, 8, 10, 9, 11 | |
1066 | ]; | |
1067 | const COOK_VQ6_CODES: &[u16; 32] = &[ | |
1068 | 0x0000, 0x0008, 0x0009, 0x0034, 0x000a, 0x0035, 0x0036, 0x00f6, | |
1069 | 0x000b, 0x0037, 0x0038, 0x00f7, 0x0039, 0x01fa, 0x00f8, 0x03fc, | |
1070 | 0x000c, 0x003a, 0x007a, 0x00f9, 0x003b, 0x01fb, 0x00fa, 0x07fe, | |
1071 | 0x003c, 0x01fc, 0x00fb, 0x03fd, 0x00fc, 0x03fe, 0x01fd, 0x07ff | |
1072 | ]; | |
1073 | ||
1074 | const COOK_CPL_SCALE2: &[f32; 5] = &[ | |
1075 | 1.0, 0.953020632266998, 0.70710676908493, 0.302905440330505, 0.0 | |
1076 | ]; | |
1077 | const COOK_CPL_SCALE3: &[f32; 9] = &[ | |
1078 | 1.0, 0.981279790401459, 0.936997592449188, 0.875934481620789, 0.70710676908493, | |
1079 | 0.482430040836334, 0.349335819482803, 0.192587479948997, 0.0 | |
1080 | ]; | |
1081 | const COOK_CPL_SCALE4: &[f32; 17] = &[ | |
1082 | 1.0, 0.991486728191376, 0.973249018192291, 0.953020632266998, 0.930133521556854, | |
1083 | 0.903453230857849, 0.870746195316315, 0.826180458068848, 0.70710676908493, | |
1084 | 0.563405573368073, 0.491732746362686, 0.428686618804932, 0.367221474647522, | |
1085 | 0.302905440330505, 0.229752898216248, 0.130207896232605, 0.0 | |
1086 | ]; | |
1087 | const COOK_CPL_SCALE5: &[f32; 33] = &[ | |
1088 | 1.0, 0.995926380157471, 0.987517595291138, 0.978726446628571, 0.969505727291107, | |
1089 | 0.95979779958725, 0.949531257152557, 0.938616216182709, 0.926936149597168, | |
1090 | 0.914336204528809, 0.900602877140045, 0.885426938533783, 0.868331849575043, | |
1091 | 0.84851086139679, 0.824381768703461, 0.791833400726318, 0.70710676908493, | |
1092 | 0.610737144947052, 0.566034197807312, 0.529177963733673, 0.495983630418777, | |
1093 | 0.464778542518616, 0.434642940759659, 0.404955863952637, 0.375219136476517, | |
1094 | 0.344963222742081, 0.313672333955765, 0.280692428350449, 0.245068684220314, | |
1095 | 0.205169528722763, 0.157508864998817, 0.0901700109243393, 0.0 | |
1096 | ]; | |
1097 | const COOK_CPL_SCALE6: &[f32; 65] = &[ | |
1098 | 1.0, 0.998005926609039, 0.993956744670868, 0.989822506904602, 0.985598564147949, | |
1099 | 0.981279790401459, 0.976860702037811, 0.972335040569305, 0.967696130275726, | |
1100 | 0.962936460971832, 0.958047747612000, 0.953020632266998, 0.947844684123993, | |
1101 | 0.942508161067963, 0.936997592449188, 0.931297719478607, 0.925390899181366, | |
1102 | 0.919256627559662, 0.912870943546295, 0.906205296516418, 0.899225592613220, | |
1103 | 0.891890347003937, 0.884148240089417, 0.875934481620789, 0.867165684700012, | |
1104 | 0.857730865478516, 0.847477376461029, 0.836184680461884, 0.823513329029083, | |
1105 | 0.808890223503113, 0.791194140911102, 0.767520070075989, 0.707106769084930, | |
1106 | 0.641024887561798, 0.611565053462982, 0.587959706783295, 0.567296981811523, | |
1107 | 0.548448026180267, 0.530831515789032, 0.514098942279816, 0.498019754886627, | |
1108 | 0.482430040836334, 0.467206478118896, 0.452251672744751, 0.437485188245773, | |
1109 | 0.422837972640991, 0.408248275518417, 0.393658757209778, 0.379014074802399, | |
1110 | 0.364258885383606, 0.349335819482803, 0.334183186292648, 0.318732559680939, | |
1111 | 0.302905440330505, 0.286608695983887, 0.269728302955627, 0.252119421958923, | |
1112 | 0.233590632677078, 0.213876649737358, 0.192587479948997, 0.169101938605309, | |
1113 | 0.142307326197624, 0.109772264957428, 0.0631198287010193, 0.0 | |
1114 | ]; | |
1115 | const COOK_CPL_SCALES: [&[f32]; 5] = [ | |
1116 | COOK_CPL_SCALE2, COOK_CPL_SCALE3, COOK_CPL_SCALE4, COOK_CPL_SCALE5, COOK_CPL_SCALE6 | |
1117 | ]; | |
1118 | ||
1119 | const COOK_CPL_BAND: [u8; MAX_SUBBANDS - 1] = [ | |
1120 | 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 11, 12, 12, 13, | |
1121 | 13, 14, 14, 14, 15, 15, 15, 15, 16, 16, 16, 16, 16, 17, 17, 17, | |
1122 | 17, 17, 17, 18, 18, 18, 18, 18, 18, 18, 19, 19, 19, 19, 19, 19, | |
1123 | 19, 19, 19 | |
1124 | ]; | |
1125 | ||
61d3e294 | 1126 | #[allow(clippy::approx_constant)] |
594ca5ca KS |
1127 | const COOK_DITHER_TAB: [f32; 9] = [ 0.0, 0.0, 0.0, 0.0, 0.0, 0.176777, 0.25, 0.707107, 1.0 ]; |
1128 | ||
1129 | const COOK_QUANT_CENTROID: [[f32; 14]; 7] = [ | |
1130 | [ 0.000, 0.392, 0.761, 1.120, 1.477, 1.832, 2.183, 2.541, 2.893, 3.245, 3.598, 3.942, 4.288, 4.724 ], | |
1131 | [ 0.000, 0.544, 1.060, 1.563, 2.068, 2.571, 3.072, 3.562, 4.070, 4.620, 0.000, 0.000, 0.000, 0.000 ], | |
1132 | [ 0.000, 0.746, 1.464, 2.180, 2.882, 3.584, 4.316, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000 ], | |
1133 | [ 0.000, 1.006, 2.000, 2.993, 3.985, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000 ], | |
1134 | [ 0.000, 1.321, 2.703, 3.983, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000 ], | |
1135 | [ 0.000, 1.657, 3.491, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000 ], | |
1136 | [ 0.000, 1.964, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000, 0.000 ] | |
1137 | ]; | |
1138 | ||
1139 | const COOK_EXP_BITS: [i32; NUM_CATEGORIES] = [ 52, 47, 43, 37, 29, 22, 16, 0 ]; |