switch to better FFT interface and more flexible FFT implementation
[nihav.git] / nihav-realmedia / src / codecs / cook.rs
1 use nihav_core::formats::*;
2 use nihav_core::frame::*;
3 use nihav_core::codecs::*;
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::*;
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 {
20 fn get_channels(&self) -> usize {
21 match *self {
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 {
77 cpl_cb: cpl_cb,
78 quant_cb: quant_cb,
79 vq_cb: vq_cb,
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;
114 CookDSP { imdct: IMDCT::new(samples*2, false), window: window, out: [0.0; 2048], size, pow_tab, hpow_tab, gain_tab }
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
145 block: [[f32; MAX_SAMPLES * 2]; 2],
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
166 block: [[0.0; MAX_SAMPLES * 2]; 2],
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]);
311 self.block[channel] = [0.0; MAX_SAMPLES * 2];
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];
340 for i in cstart..cend+1 {
341 self.decouple[i] = br.read_cb(cb)? as u8;
342 }
343 } else {
344 for i in cstart..cend+1 {
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
369 self.block[channel] = [0.0; MAX_SAMPLES * 2];
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();
433 let output = adata.get_data_mut().unwrap();
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 {
523 info: NACodecInfoRef,
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 {
553 fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
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
570 self.pairs[0].mode = Mode::Mono;
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 }
634 fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
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);
670 Ok(frm.into_ref())
671 }
672 }
673
674 pub fn get_decoder() -> Box<NADecoder> {
675 Box::new(CookDecoder::new())
676 }
677
678 #[cfg(test)]
679 mod test {
680 use nihav_core::codecs::RegisteredDecoders;
681 use nihav_core::demuxers::RegisteredDemuxers;
682 use nihav_core::test::dec_video::*;
683 use crate::codecs::realmedia_register_all_codecs;
684 use crate::demuxers::realmedia_register_all_demuxers;
685 #[test]
686 fn test_cook() {
687 let mut dmx_reg = RegisteredDemuxers::new();
688 realmedia_register_all_demuxers(&mut dmx_reg);
689 let mut dec_reg = RegisteredDecoders::new();
690 realmedia_register_all_codecs(&mut dec_reg);
691
692 // let file = "assets/RV/rv30_weighted_mc.rm";
693 let file = "assets/RV/multichannel.rma";
694 test_decode_audio("realmedia", file, Some(2000), "cook", &dmx_reg, &dec_reg);
695 }
696 }
697
698 const COOK_XOR_KEY: [u8; 4] = [ 0x37, 0xC5, 0x11, 0xF2 ];
699
700 const COOK_CPL_2BITS_BITS: &[u8; 3] = &[ 2, 1, 2 ];
701 const COOK_CPL_2BITS_CODES: &[u16; 3] = &[ 0x02, 0x00, 0x03 ];
702 const COOK_CPL_3BITS_BITS: &[u8; 7] = &[ 6, 5, 2, 1, 3, 4, 6 ];
703 const COOK_CPL_3BITS_CODES: &[u16; 7] = &[ 0x3e, 0x1e, 0x02, 0x00, 0x06, 0x0e, 0x3f ];
704 const COOK_CPL_4BITS_BITS: &[u8; 15] = &[ 8, 8, 7, 6, 5, 4, 3, 1, 3, 4, 5, 6, 7, 8, 8 ];
705 const COOK_CPL_4BITS_CODES: &[u16; 15] = &[
706 0xfc, 0xfd, 0x7c, 0x3c, 0x1c, 0x0c, 0x04, 0x00,
707 0x05, 0x0d, 0x1d, 0x3d, 0x7d, 0xfe, 0xff
708 ];
709 const COOK_CPL_5BITS_BITS: &[u8; 31] = &[
710 10, 10, 10, 10, 9, 9, 8, 8, 7, 7, 6, 6, 5, 5, 3, 1,
711 3, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 10, 10
712 ];
713 const COOK_CPL_5BITS_CODES: &[u16; 31] = &[
714 0x03F8, 0x03F9, 0x03FA, 0x03FB, 0x01F8, 0x01F9, 0x00F8, 0x00F9,
715 0x0078, 0x0079, 0x0038, 0x0039, 0x0018, 0x0019, 0x0004, 0x0000,
716 0x0005, 0x001A, 0x001B, 0x003A, 0x003B, 0x007A, 0x007B, 0x00FA,
717 0x00FB, 0x01FA, 0x01FB, 0x03FC, 0x03FD, 0x03FE, 0x03FF
718 ];
719 const COOK_CPL_6BITS_BITS: &[u8; 63] = &[
720 16, 15, 14, 13, 12, 11, 11, 11, 11, 10, 10, 10, 10, 9, 9, 9,
721 9, 9, 8, 8, 8, 8, 7, 7, 7, 7, 6, 6, 5, 5, 3, 1,
722 4, 5, 5, 6, 6, 7, 7, 7, 7, 8, 8, 8, 8, 9, 9, 9,
723 9, 10, 10, 10, 10, 10, 11, 11, 11, 11, 12, 13, 14, 14, 16
724 ];
725 const COOK_CPL_6BITS_CODES: &[u16; 63] = &[
726 0xFFFE, 0x7FFE, 0x3FFC, 0x1FFC, 0x0FFC, 0x07F6, 0x07F7, 0x07F8,
727 0x07F9, 0x03F2, 0x03F3, 0x03F4, 0x03F5, 0x01F0, 0x01F1, 0x01F2,
728 0x01F3, 0x01F4, 0x00F0, 0x00F1, 0x00F2, 0x00F3, 0x0070, 0x0071,
729 0x0072, 0x0073, 0x0034, 0x0035, 0x0016, 0x0017, 0x0004, 0x0000,
730 0x000A, 0x0018, 0x0019, 0x0036, 0x0037, 0x0074, 0x0075, 0x0076,
731 0x0077, 0x00F4, 0x00F5, 0x00F6, 0x00F7, 0x01F5, 0x01F6, 0x01F7,
732 0x01F8, 0x03F6, 0x03F7, 0x03F8, 0x03F9, 0x03FA, 0x07FA, 0x07FB,
733 0x07FC, 0x07FD, 0x0FFD, 0x1FFD, 0x3FFD, 0x3FFE, 0xFFFF
734 ];
735
736 const COOK_QUANT_BITS: [&[u8; 24]; 13] = [
737 &[ 4, 6, 5, 5, 4, 4, 4, 4, 4, 4, 3, 3, 3, 4, 5, 7, 8, 9, 11, 11, 12, 12, 12, 12 ],
738 &[ 10, 8, 6, 5, 5, 4, 3, 3, 3, 3, 3, 3, 4, 5, 7, 9, 11, 12, 13, 15, 15, 15, 16, 16 ],
739 &[ 12, 10, 8, 6, 5, 4, 4, 4, 4, 4, 4, 3, 3, 3, 4, 4, 5, 5, 7, 9, 11, 13, 14, 14 ],
740 &[ 13, 10, 9, 9, 7, 7, 5, 5, 4, 3, 3, 3, 3, 3, 4, 4, 4, 5, 7, 9, 11, 13, 13, 13 ],
741 &[ 12, 13, 10, 8, 6, 6, 5, 5, 4, 4, 3, 3, 3, 3, 3, 4, 5, 5, 6, 7, 9, 11, 14, 14 ],
742 &[ 12, 11, 9, 8, 8, 7, 5, 4, 4, 3, 3, 3, 3, 3, 4, 4, 5, 5, 7, 8, 10, 13, 14, 14 ],
743 &[ 15, 16, 15, 12, 10, 8, 6, 5, 4, 3, 3, 3, 2, 3, 4, 5, 5, 7, 9, 11, 13, 16, 16, 16 ],
744 &[ 14, 14, 11, 10, 9, 7, 7, 5, 5, 4, 3, 3, 2, 3, 3, 4, 5, 7, 9, 9, 12, 14, 15, 15 ],
745 &[ 9, 9, 9, 8, 7, 6, 5, 4, 3, 3, 3, 3, 3, 3, 4, 5, 6, 7, 8, 10, 11, 12, 13, 13 ],
746 &[ 14, 12, 10, 8, 6, 6, 5, 4, 3, 3, 3, 3, 3, 3, 4, 5, 6, 8, 8, 9, 11, 14, 14, 14 ],
747 &[ 13, 10, 9, 8, 6, 6, 5, 4, 4, 4, 3, 3, 2, 3, 4, 5, 6, 8, 9, 9, 11, 12, 14, 14 ],
748 &[ 16, 13, 12, 11, 9, 6, 5, 5, 4, 4, 4, 3, 2, 3, 3, 4, 5, 7, 8, 10, 14, 16, 16, 16 ],
749 &[ 13, 14, 14, 14, 10, 8, 7, 7, 5, 4, 3, 3, 2, 3, 3, 4, 5, 5, 7, 9, 11, 14, 14, 14 ],
750 ];
751 const COOK_QUANT_CODES: [&[u16; 24]; 13] = [
752 &[ 0x0006, 0x003e, 0x001c, 0x001d, 0x0007, 0x0008, 0x0009, 0x000a, 0x000b, 0x000c, 0x0000, 0x0001,
753 0x0002, 0x000d, 0x001e, 0x007e, 0x00fe, 0x01fe, 0x07fc, 0x07fd, 0x0ffc, 0x0ffd, 0x0ffe, 0x0fff ],
754 &[ 0x03fe, 0x00fe, 0x003e, 0x001c, 0x001d, 0x000c, 0x0000, 0x0001, 0x0002, 0x0003, 0x0004, 0x0005,
755 0x000d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x0ffe, 0x1ffe, 0x7ffc, 0x7ffd, 0x7ffe, 0xfffe, 0xffff ],
756 &[ 0x0ffe, 0x03fe, 0x00fe, 0x003e, 0x001c, 0x0006, 0x0007, 0x0008, 0x0009, 0x000a, 0x000b, 0x0000,
757 0x0001, 0x0002, 0x000c, 0x000d, 0x001d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x1ffe, 0x3ffe, 0x3fff ],
758 &[ 0x1ffc, 0x03fe, 0x01fc, 0x01fd, 0x007c, 0x007d, 0x001c, 0x001d, 0x000a, 0x0000, 0x0001, 0x0002,
759 0x0003, 0x0004, 0x000b, 0x000c, 0x000d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x1ffd, 0x1ffe, 0x1fff ],
760 &[ 0x0ffe, 0x1ffe, 0x03fe, 0x00fe, 0x003c, 0x003d, 0x001a, 0x001b, 0x000a, 0x000b, 0x0000, 0x0001,
761 0x0002, 0x0003, 0x0004, 0x000c, 0x001c, 0x001d, 0x003e, 0x007e, 0x01fe, 0x07fe, 0x3ffe, 0x3fff ],
762 &[ 0x0ffe, 0x07fe, 0x01fe, 0x00fc, 0x00fd, 0x007c, 0x001c, 0x000a, 0x000b, 0x0000, 0x0001, 0x0002,
763 0x0003, 0x0004, 0x000c, 0x000d, 0x001d, 0x001e, 0x007d, 0x00fe, 0x03fe, 0x1ffe, 0x3ffe, 0x3fff ],
764 &[ 0x7ffc, 0xfffc, 0x7ffd, 0x0ffe, 0x03fe, 0x00fe, 0x003e, 0x001c, 0x000c, 0x0002, 0x0003, 0x0004,
765 0x0000, 0x0005, 0x000d, 0x001d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x1ffe, 0xfffd, 0xfffe, 0xffff ],
766 &[ 0x3ffc, 0x3ffd, 0x07fe, 0x03fe, 0x01fc, 0x007c, 0x007d, 0x001c, 0x001d, 0x000c, 0x0002, 0x0003,
767 0x0000, 0x0004, 0x0005, 0x000d, 0x001e, 0x007e, 0x01fd, 0x01fe, 0x0ffe, 0x3ffe, 0x7ffe, 0x7fff ],
768 &[ 0x01fc, 0x01fd, 0x01fe, 0x00fc, 0x007c, 0x003c, 0x001c, 0x000c, 0x0000, 0x0001, 0x0002, 0x0003,
769 0x0004, 0x0005, 0x000d, 0x001d, 0x003d, 0x007d, 0x00fd, 0x03fe, 0x07fe, 0x0ffe, 0x1ffe, 0x1fff ],
770 &[ 0x3ffc, 0x0ffe, 0x03fe, 0x00fc, 0x003c, 0x003d, 0x001c, 0x000c, 0x0000, 0x0001, 0x0002, 0x0003,
771 0x0004, 0x0005, 0x000d, 0x001d, 0x003e, 0x00fd, 0x00fe, 0x01fe, 0x07fe, 0x3ffd, 0x3ffe, 0x3fff ],
772 &[ 0x1ffe, 0x03fe, 0x01fc, 0x00fc, 0x003c, 0x003d, 0x001c, 0x000a, 0x000b, 0x000c, 0x0002, 0x0003,
773 0x0000, 0x0004, 0x000d, 0x001d, 0x003e, 0x00fd, 0x01fd, 0x01fe, 0x07fe, 0x0ffe, 0x3ffe, 0x3fff ],
774 &[ 0xfffc, 0x1ffe, 0x0ffe, 0x07fe, 0x01fe, 0x003e, 0x001c, 0x001d, 0x000a, 0x000b, 0x000c, 0x0002,
775 0x0000, 0x0003, 0x0004, 0x000d, 0x001e, 0x007e, 0x00fe, 0x03fe, 0x3ffe, 0xfffd, 0xfffe, 0xffff ],
776 &[ 0x1ffc, 0x3ffa, 0x3ffb, 0x3ffc, 0x03fe, 0x00fe, 0x007c, 0x007d, 0x001c, 0x000c, 0x0002, 0x0003,
777 0x0000, 0x0004, 0x0005, 0x000d, 0x001d, 0x001e, 0x007e, 0x01fe, 0x07fe, 0x3ffd, 0x3ffe, 0x3fff ],
778 ];
779
780 const COOK_VQ0_BITS: &[u8; 191] = &[
781 1, 4, 6, 6, 7, 7, 8, 8, 8, 9, 9, 10,
782 11, 11, 4, 5, 6, 7, 7, 8, 8, 9, 9, 9,
783 9, 10, 11, 11, 5, 6, 7, 8, 8, 9, 9, 9,
784 9, 10, 10, 10, 11, 12, 6, 7, 8, 9, 9, 9,
785 9, 10, 10, 10, 10, 11, 12, 13, 7, 7, 8, 9,
786 9, 9, 10, 10, 10, 10, 11, 11, 12, 13, 8, 8,
787 9, 9, 9, 10, 10, 10, 10, 11, 11, 12, 13, 14,
788 8, 8, 9, 9, 10, 10, 11, 11, 11, 12, 12, 13,
789 13, 15, 8, 8, 9, 9, 10, 10, 11, 11, 11, 12,
790 12, 13, 14, 15, 9, 9, 9, 10, 10, 10, 11, 11,
791 12, 13, 12, 14, 15, 16, 9, 9, 10, 10, 10, 10,
792 11, 12, 12, 14, 14, 16, 16, 0, 9, 9, 10, 10,
793 11, 11, 12, 13, 13, 14, 14, 15, 0, 0, 10, 10,
794 10, 11, 11, 12, 12, 13, 15, 15, 16, 0, 0, 0,
795 11, 11, 11, 12, 13, 13, 13, 15, 16, 16, 0, 0,
796 0, 0, 11, 11, 12, 13, 13, 14, 15, 16, 16
797 ];
798 const COOK_VQ0_CODES: &[u16; 191] = &[
799 0x0000, 0x0008, 0x002c, 0x002d, 0x0062, 0x0063, 0x00d4, 0x00d5,
800 0x00d6, 0x01c6, 0x01c7, 0x03ca, 0x07d6, 0x07d7, 0x0009, 0x0014,
801 0x002e, 0x0064, 0x0065, 0x00d7, 0x00d8, 0x01c8, 0x01c9, 0x01ca,
802 0x01cb, 0x03cb, 0x07d8, 0x07d9, 0x0015, 0x002f, 0x0066, 0x00d9,
803 0x00da, 0x01cc, 0x01cd, 0x01ce, 0x01cf, 0x03cc, 0x03cd, 0x03ce,
804 0x07da, 0x0fe4, 0x0030, 0x0067, 0x00db, 0x01d0, 0x01d1, 0x01d2,
805 0x01d3, 0x03cf, 0x03d0, 0x03d1, 0x03d2, 0x07db, 0x0fe5, 0x1fea,
806 0x0068, 0x0069, 0x00dc, 0x01d4, 0x01d5, 0x01d6, 0x03d3, 0x03d4,
807 0x03d5, 0x03d6, 0x07dc, 0x07dd, 0x0fe6, 0x1feb, 0x00dd, 0x00de,
808 0x01d7, 0x01d8, 0x01d9, 0x03d7, 0x03d8, 0x03d9, 0x03da, 0x07de,
809 0x07df, 0x0fe7, 0x1fec, 0x3ff2, 0x00df, 0x00e0, 0x01da, 0x01db,
810 0x03db, 0x03dc, 0x07e0, 0x07e1, 0x07e2, 0x0fe8, 0x0fe9, 0x1fed,
811 0x1fee, 0x7ff4, 0x00e1, 0x00e2, 0x01dc, 0x01dd, 0x03dd, 0x03de,
812 0x07e3, 0x07e4, 0x07e5, 0x0fea, 0x0feb, 0x1fef, 0x3ff3, 0x7ff5,
813 0x01de, 0x01df, 0x01e0, 0x03df, 0x03e0, 0x03e1, 0x07e6, 0x07e7,
814 0x0fec, 0x1ff0, 0x0fed, 0x3ff4, 0x7ff6, 0xfff8, 0x01e1, 0x01e2,
815 0x03e2, 0x03e3, 0x03e4, 0x03e5, 0x07e8, 0x0fee, 0x0fef, 0x3ff5,
816 0x3ff6, 0xfff9, 0xfffa, 0xfffa, 0x01e3, 0x01e4, 0x03e6, 0x03e7,
817 0x07e9, 0x07ea, 0x0ff0, 0x1ff1, 0x1ff2, 0x3ff7, 0x3ff8, 0x7ff7,
818 0x7ff7, 0xfffa, 0x03e8, 0x03e9, 0x03ea, 0x07eb, 0x07ec, 0x0ff1,
819 0x0ff2, 0x1ff3, 0x7ff8, 0x7ff9, 0xfffb, 0x3ff8, 0x7ff7, 0x7ff7,
820 0x07ed, 0x07ee, 0x07ef, 0x0ff3, 0x1ff4, 0x1ff5, 0x1ff6, 0x7ffa,
821 0xfffc, 0xfffd, 0xfffb, 0xfffb, 0x3ff8, 0x7ff7, 0x07f0, 0x07f1,
822 0x0ff4, 0x1ff7, 0x1ff8, 0x3ff9, 0x7ffb, 0xfffe, 0xffff
823 ];
824 const COOK_VQ1_BITS: &[u8; 97] = &[
825 1, 4, 5, 6, 7, 8, 8, 9, 10, 10, 4, 5,
826 6, 7, 7, 8, 8, 9, 9, 11, 5, 5, 6, 7,
827 8, 8, 9, 9, 10, 11, 6, 6, 7, 8, 8, 9,
828 9, 10, 11, 12, 7, 7, 8, 8, 9, 9, 10, 11,
829 11, 13, 8, 8, 8, 9, 9, 10, 10, 11, 12, 14,
830 8, 8, 8, 9, 10, 11, 11, 12, 13, 15, 9, 9,
831 9, 10, 11, 12, 12, 14, 14, 0, 9, 9, 9, 10,
832 11, 12, 14, 16, 0, 0, 10, 10, 11, 12, 13, 14, 16
833 ];
834 const COOK_VQ1_CODES: &[u16; 97] = &[
835 0x0000, 0x0008, 0x0014, 0x0030, 0x006a, 0x00e2, 0x00e3, 0x01e4,
836 0x03ec, 0x03ed, 0x0009, 0x0015, 0x0031, 0x006b, 0x006c, 0x00e4,
837 0x00e5, 0x01e5, 0x01e6, 0x07f0, 0x0016, 0x0017, 0x0032, 0x006d,
838 0x00e6, 0x00e7, 0x01e7, 0x01e8, 0x03ee, 0x07f1, 0x0033, 0x0034,
839 0x006e, 0x00e8, 0x00e9, 0x01e9, 0x01ea, 0x03ef, 0x07f2, 0x0ff6,
840 0x006f, 0x0070, 0x00ea, 0x00eb, 0x01eb, 0x01ec, 0x03f0, 0x07f3,
841 0x07f4, 0x1ffa, 0x00ec, 0x00ed, 0x00ee, 0x01ed, 0x01ee, 0x03f1,
842 0x03f2, 0x07f5, 0x0ff7, 0x3ffa, 0x00ef, 0x00f0, 0x00f1, 0x01ef,
843 0x03f3, 0x07f6, 0x07f7, 0x0ff8, 0x1ffb, 0x7ffe, 0x01f0, 0x01f1,
844 0x01f2, 0x03f4, 0x07f8, 0x0ff9, 0x0ffa, 0x3ffb, 0x3ffc, 0x0000,
845 0x01f3, 0x01f4, 0x01f5, 0x03f5, 0x07f9, 0x0ffb, 0x3ffd, 0xfffe,
846 0x0000, 0x0000, 0x03f6, 0x03f7, 0x07fa, 0x0ffc, 0x1ffc, 0x3ffe,
847 0xffff
848 ];
849 const COOK_VQ2_BITS: &[u8; 48] = &[
850 1, 4, 5, 7, 8, 9, 10, 3, 4, 5, 7, 8,
851 9, 10, 5, 5, 6, 7, 8, 10, 10, 7, 6, 7,
852 8, 9, 10, 12, 8, 8, 8, 9, 10, 12, 14, 8,
853 9, 9, 10, 11, 15, 16, 9, 10, 11, 12, 13, 16
854 ];
855 const COOK_VQ2_CODES: &[u16; 48] = &[
856 0x0000, 0x000a, 0x0018, 0x0074, 0x00f2, 0x01f4, 0x03f6, 0x0004, 0x000b, 0x0019, 0x0075, 0x00f3,
857 0x01f5, 0x03f7, 0x001a, 0x001b, 0x0038, 0x0076, 0x00f4, 0x03f8, 0x03f9, 0x0077, 0x0039, 0x0078,
858 0x00f5, 0x01f6, 0x03fa, 0x0ffc, 0x00f6, 0x00f7, 0x00f8, 0x01f7, 0x03fb, 0x0ffd, 0x3ffe, 0x00f9,
859 0x01f8, 0x01f9, 0x03fc, 0x07fc, 0x7ffe, 0xfffe, 0x01fa, 0x03fd, 0x07fd, 0x0ffe, 0x1ffe, 0xffff
860 ];
861 const COOK_VQ3_BITS: &[u8; 607] = &[
862 2, 4, 6, 8, 10, 5, 5, 6, 8, 10, 7, 8,
863 8, 10, 12, 9, 9, 10, 12, 15, 10, 11, 13, 16,
864 16, 5, 6, 8, 10, 11, 5, 6, 8, 10, 12, 7,
865 7, 8, 10, 13, 9, 9, 10, 12, 15, 12, 11, 13,
866 16, 16, 7, 9, 10, 12, 15, 7, 8, 10, 12, 13,
867 9, 9, 11, 13, 16, 11, 11, 12, 14, 16, 12, 12,
868 14, 16, 0, 9, 11, 12, 16, 16, 9, 10, 13, 15,
869 16, 10, 11, 12, 16, 16, 13, 13, 16, 16, 16, 16,
870 16, 15, 16, 0, 11, 13, 16, 16, 15, 11, 13, 15,
871 16, 16, 13, 13, 16, 16, 0, 14, 16, 16, 16, 0,
872 16, 16, 0, 0, 0, 4, 6, 8, 10, 13, 6, 6,
873 8, 10, 13, 9, 8, 10, 12, 16, 10, 10, 11, 15,
874 16, 13, 12, 14, 16, 16, 5, 6, 8, 11, 13, 6,
875 6, 8, 10, 13, 8, 8, 9, 11, 14, 10, 10, 12,
876 12, 16, 13, 12, 13, 15, 16, 7, 8, 9, 12, 16,
877 7, 8, 10, 12, 14, 9, 9, 10, 13, 16, 11, 10,
878 12, 15, 16, 13, 13, 16, 16, 0, 9, 11, 13, 16,
879 16, 9, 10, 12, 15, 16, 10, 11, 13, 16, 16, 13,
880 12, 16, 16, 16, 16, 16, 16, 16, 0, 11, 13, 16,
881 16, 16, 11, 13, 16, 16, 16, 12, 13, 15, 16, 0,
882 16, 16, 16, 16, 0, 16, 16, 0, 0, 0, 6, 8,
883 11, 13, 16, 8, 8, 10, 12, 16, 11, 10, 11, 13,
884 16, 12, 13, 13, 15, 16, 16, 16, 14, 16, 0, 6,
885 8, 10, 13, 16, 8, 8, 10, 12, 16, 10, 10, 11,
886 13, 16, 13, 12, 13, 16, 16, 14, 14, 14, 16, 0,
887 8, 9, 11, 13, 16, 8, 9, 11, 16, 14, 10, 10,
888 12, 15, 16, 12, 12, 13, 16, 16, 15, 16, 16, 16,
889 0, 10, 12, 15, 16, 16, 10, 12, 12, 14, 16, 12,
890 12, 13, 16, 16, 14, 15, 16, 16, 0, 16, 16, 16,
891 0, 0, 12, 15, 15, 16, 0, 13, 13, 16, 16, 0,
892 14, 16, 16, 16, 0, 16, 16, 16, 0, 0, 0, 0,
893 0, 0, 0, 8, 10, 13, 15, 16, 10, 11, 13, 16,
894 16, 13, 13, 14, 16, 16, 16, 16, 16, 16, 16, 16,
895 16, 16, 16, 0, 8, 10, 11, 15, 16, 9, 10, 12,
896 16, 16, 12, 12, 15, 16, 16, 16, 14, 16, 16, 16,
897 16, 16, 16, 16, 0, 9, 11, 14, 16, 16, 10, 11,
898 13, 16, 16, 14, 13, 14, 16, 16, 16, 15, 15, 16,
899 0, 16, 16, 16, 0, 0, 11, 13, 16, 16, 16, 11,
900 13, 15, 16, 16, 13, 16, 16, 16, 0, 16, 16, 16,
901 16, 0, 16, 16, 0, 0, 0, 15, 16, 16, 16, 0,
902 14, 16, 16, 16, 0, 16, 16, 16, 0, 0, 16, 16,
903 0, 0, 0, 0, 0, 0, 0, 0, 9, 13, 16, 16,
904 16, 11, 13, 16, 16, 16, 14, 15, 16, 16, 0, 15,
905 16, 16, 16, 0, 16, 16, 0, 0, 0, 9, 13, 15,
906 15, 16, 12, 13, 14, 16, 16, 16, 15, 16, 16, 0,
907 16, 16, 16, 16, 0, 16, 16, 0, 0, 0, 11, 13,
908 15, 16, 0, 12, 14, 16, 16, 0, 16, 16, 16, 16,
909 0, 16, 16, 16, 0, 0, 0, 0, 0, 0, 0, 16,
910 16, 16, 16, 0, 16, 16, 16, 16, 0, 16, 16, 16,
911 0, 0, 16, 16, 0, 0, 0, 0, 0, 0, 0, 0,
912 16, 16, 0, 0, 0, 16, 16
913 ];
914 const COOK_VQ3_CODES: &[u16; 607] = &[
915 0x0000, 0x0004, 0x0022, 0x00c6, 0x03b0, 0x000c, 0x000d, 0x0023, 0x00c7, 0x03b1, 0x005c, 0x00c8,
916 0x00c9, 0x03b2, 0x0fa4, 0x01c2, 0x01c3, 0x03b3, 0x0fa5, 0x7f72, 0x03b4, 0x07b2, 0x1f9a, 0xff24,
917 0xff25, 0x000e, 0x0024, 0x00ca, 0x03b5, 0x07b3, 0x000f, 0x0025, 0x00cb, 0x03b6, 0x0fa6, 0x005d,
918 0x005e, 0x00cc, 0x03b7, 0x1f9b, 0x01c4, 0x01c5, 0x03b8, 0x0fa7, 0x7f73, 0x0fa8, 0x07b4, 0x1f9c,
919 0xff26, 0xff27, 0x005f, 0x01c6, 0x03b9, 0x0fa9, 0x7f74, 0x0060, 0x00cd, 0x03ba, 0x0faa, 0x1f9d,
920 0x01c7, 0x01c8, 0x07b5, 0x1f9e, 0xff28, 0x07b6, 0x07b7, 0x0fab, 0x3fa2, 0xff29, 0x0fac, 0x0fad,
921 0x3fa3, 0xff2a, 0x3fa2, 0x01c9, 0x07b8, 0x0fae, 0xff2b, 0xff2c, 0x01ca, 0x03bb, 0x1f9f, 0x7f75,
922 0xff2d, 0x03bc, 0x07b9, 0x0faf, 0xff2e, 0xff2f, 0x1fa0, 0x1fa1, 0xff30, 0xff31, 0xff32, 0xff33,
923 0xff34, 0x7f76, 0xff35, 0xff31, 0x07ba, 0x1fa2, 0xff36, 0xff37, 0x7f77, 0x07bb, 0x1fa3, 0x7f78,
924 0xff38, 0xff39, 0x1fa4, 0x1fa5, 0xff3a, 0xff3b, 0xff2e, 0x3fa4, 0xff3c, 0xff3d, 0xff3e, 0xff31,
925 0xff3f, 0xff40, 0xff30, 0xff31, 0xff31, 0x0005, 0x0026, 0x00ce, 0x03bd, 0x1fa6, 0x0027, 0x0028,
926 0x00cf, 0x03be, 0x1fa7, 0x01cb, 0x00d0, 0x03bf, 0x0fb0, 0xff41, 0x03c0, 0x03c1, 0x07bc, 0x7f79,
927 0xff42, 0x1fa8, 0x0fb1, 0x3fa5, 0xff43, 0xff44, 0x0010, 0x0029, 0x00d1, 0x07bd, 0x1fa9, 0x002a,
928 0x002b, 0x00d2, 0x03c2, 0x1faa, 0x00d3, 0x00d4, 0x01cc, 0x07be, 0x3fa6, 0x03c3, 0x03c4, 0x0fb2,
929 0x0fb3, 0xff45, 0x1fab, 0x0fb4, 0x1fac, 0x7f7a, 0xff46, 0x0061, 0x00d5, 0x01cd, 0x0fb5, 0xff47,
930 0x0062, 0x00d6, 0x03c5, 0x0fb6, 0x3fa7, 0x01ce, 0x01cf, 0x03c6, 0x1fad, 0xff48, 0x07bf, 0x03c7,
931 0x0fb7, 0x7f7b, 0xff49, 0x1fae, 0x1faf, 0xff4a, 0xff4b, 0x7f7b, 0x01d0, 0x07c0, 0x1fb0, 0xff4c,
932 0xff4d, 0x01d1, 0x03c8, 0x0fb8, 0x7f7c, 0xff4e, 0x03c9, 0x07c1, 0x1fb1, 0xff4f, 0xff50, 0x1fb2,
933 0x0fb9, 0xff51, 0xff52, 0xff53, 0xff54, 0xff55, 0xff56, 0xff57, 0xff52, 0x07c2, 0x1fb3, 0xff58,
934 0xff59, 0xff5a, 0x07c3, 0x1fb4, 0xff5b, 0xff5c, 0xff5d, 0x0fba, 0x1fb5, 0x7f7d, 0xff5e, 0xff4f,
935 0xff5f, 0xff60, 0xff61, 0xff62, 0xff52, 0xff63, 0xff64, 0xff51, 0xff52, 0xff52, 0x002c, 0x00d7,
936 0x07c4, 0x1fb6, 0xff65, 0x00d8, 0x00d9, 0x03ca, 0x0fbb, 0xff66, 0x07c5, 0x03cb, 0x07c6, 0x1fb7,
937 0xff67, 0x0fbc, 0x1fb8, 0x1fb9, 0x7f7e, 0xff68, 0xff69, 0xff6a, 0x3fa8, 0xff6b, 0x7f7e, 0x002d,
938 0x00da, 0x03cc, 0x1fba, 0xff6c, 0x00db, 0x00dc, 0x03cd, 0x0fbd, 0xff6d, 0x03ce, 0x03cf, 0x07c7,
939 0x1fbb, 0xff6e, 0x1fbc, 0x0fbe, 0x1fbd, 0xff6f, 0xff70, 0x3fa9, 0x3faa, 0x3fab, 0xff71, 0xff6f,
940 0x00dd, 0x01d2, 0x07c8, 0x1fbe, 0xff72, 0x00de, 0x01d3, 0x07c9, 0xff73, 0x3fac, 0x03d0, 0x03d1,
941 0x0fbf, 0x7f7f, 0xff74, 0x0fc0, 0x0fc1, 0x1fbf, 0xff75, 0xff76, 0x7f80, 0xff77, 0xff78, 0xff79,
942 0xff75, 0x03d2, 0x0fc2, 0x7f81, 0xff7a, 0xff7b, 0x03d3, 0x0fc3, 0x0fc4, 0x3fad, 0xff7c, 0x0fc5,
943 0x0fc6, 0x1fc0, 0xff7d, 0xff7e, 0x3fae, 0x7f82, 0xff7f, 0xff80, 0xff80, 0xff81, 0xff82, 0xff83,
944 0xff80, 0xff80, 0x0fc7, 0x7f83, 0x7f84, 0xff84, 0xff7a, 0x1fc1, 0x1fc2, 0xff85, 0xff86, 0x3fad,
945 0x3faf, 0xff87, 0xff88, 0xff89, 0xff7d, 0xff8a, 0xff8b, 0xff8c, 0xff80, 0xff80, 0x3fae, 0x7f82,
946 0xff7f, 0xff80, 0xff80, 0x00df, 0x03d4, 0x1fc3, 0x7f85, 0xff8d, 0x03d5, 0x07ca, 0x1fc4, 0xff8e,
947 0xff8f, 0x1fc5, 0x1fc6, 0x3fb0, 0xff90, 0xff91, 0xff92, 0xff93, 0xff94, 0xff95, 0xff96, 0xff97,
948 0xff98, 0xff99, 0xff9a, 0xff95, 0x00e0, 0x03d6, 0x07cb, 0x7f86, 0xff9b, 0x01d4, 0x03d7, 0x0fc8,
949 0xff9c, 0xff9d, 0x0fc9, 0x0fca, 0x7f87, 0xff9e, 0xff9f, 0xffa0, 0x3fb1, 0xffa1, 0xffa2, 0xffa3,
950 0xffa4, 0xffa5, 0xffa6, 0xffa7, 0xffa2, 0x01d5, 0x07cc, 0x3fb2, 0xffa8, 0xffa9, 0x03d8, 0x07cd,
951 0x1fc7, 0xffaa, 0xffab, 0x3fb3, 0x1fc8, 0x3fb4, 0xffac, 0xffad, 0xffae, 0x7f88, 0x7f89, 0xffaf,
952 0xffaf, 0xffb0, 0xffb1, 0xffb2, 0xffaf, 0xffaf, 0x07ce, 0x1fc9, 0xffb3, 0xffb4, 0xffb5, 0x07cf,
953 0x1fca, 0x7f8a, 0xffb6, 0xffb7, 0x1fcb, 0xffb8, 0xffb9, 0xffba, 0xffba, 0xffbb, 0xffbc, 0xffbd,
954 0xffbe, 0xffbe, 0xffbf, 0xffc0, 0xffbd, 0xffbe, 0xffbe, 0x7f8b, 0xffc1, 0xffc2, 0xffc3, 0xffb4,
955 0x3fb5, 0xffc4, 0xffc5, 0xffc6, 0xffb6, 0xffc7, 0xffc8, 0xffc9, 0xffba, 0xffba, 0xffca, 0xffcb,
956 0xffbd, 0xffbe, 0xffbe, 0xffbb, 0xffbc, 0xffbd, 0xffbe, 0xffbe, 0x01d6, 0x1fcc, 0xffcc, 0xffcd,
957 0xffce, 0x07d0, 0x1fcd, 0xffcf, 0xffd0, 0xffd1, 0x3fb6, 0x7f8c, 0xffd2, 0xffd3, 0xff90, 0x7f8d,
958 0xffd4, 0xffd5, 0xffd6, 0xff95, 0xffd7, 0xffd8, 0xff94, 0xff95, 0xff95, 0x01d7, 0x1fce, 0x7f8e,
959 0x7f8f, 0xffd9, 0x0fcb, 0x1fcf, 0x3fb7, 0xffda, 0xffdb, 0xffdc, 0x7f90, 0xffdd, 0xffde, 0xff9e,
960 0xffdf, 0xffe0, 0xffe1, 0xffe2, 0xffa2, 0xffe3, 0xffe4, 0xffa1, 0xffa2, 0xffa2, 0x07d1, 0x1fd0,
961 0x7f91, 0xffe5, 0xffa8, 0x0fcc, 0x3fb8, 0xffe6, 0xffe7, 0xffaa, 0xffe8, 0xffe9, 0xffea, 0xffeb,
962 0xffac, 0xffec, 0xffed, 0xffee, 0xffaf, 0xffaf, 0xffae, 0x7f88, 0x7f89, 0xffaf, 0xffaf, 0xffef,
963 0xfff0, 0xfff1, 0xfff2, 0xffb4, 0xfff3, 0xfff4, 0xfff5, 0xfff6, 0xffb6, 0xfff7, 0xfff8, 0xfff9,
964 0xffba, 0xffba, 0xfffa, 0xfffb, 0xffbd, 0xffbe, 0xffbe, 0xffbb, 0xffbc, 0xffbd, 0xffbe, 0xffbe,
965 0xfffc, 0xfffd, 0xffb3, 0xffb4, 0xffb4, 0xfffe, 0xffff
966 ];
967 const COOK_VQ4_BITS: &[u8; 246] = &[
968 2, 4, 7, 10, 4, 5, 7, 10, 7, 8, 10, 14,
969 11, 11, 15, 15, 4, 5, 9, 12, 5, 5, 8, 12,
970 8, 7, 10, 15, 11, 11, 15, 15, 7, 9, 12, 15,
971 8, 8, 12, 15, 10, 10, 13, 15, 14, 14, 15, 0,
972 11, 13, 15, 15, 11, 13, 15, 15, 14, 15, 15, 0,
973 15, 15, 0, 0, 4, 5, 9, 13, 5, 6, 9, 13,
974 9, 9, 11, 15, 14, 13, 15, 15, 4, 6, 9, 12,
975 5, 6, 9, 13, 9, 8, 11, 15, 13, 12, 15, 15,
976 7, 9, 12, 15, 7, 8, 11, 15, 10, 10, 14, 15,
977 14, 15, 15, 0, 10, 12, 15, 15, 11, 13, 15, 15,
978 15, 15, 15, 0, 15, 15, 0, 0, 6, 9, 13, 14,
979 8, 9, 12, 15, 12, 12, 15, 15, 15, 15, 15, 0,
980 7, 9, 13, 15, 8, 9, 12, 15, 11, 12, 15, 15,
981 15, 15, 15, 0, 9, 11, 15, 15, 9, 11, 15, 15,
982 14, 14, 15, 0, 15, 15, 0, 0, 14, 15, 15, 0,
983 14, 15, 15, 0, 15, 15, 0, 0, 0, 0, 0, 0,
984 9, 12, 15, 15, 12, 13, 15, 15, 15, 15, 15, 0,
985 15, 15, 0, 0, 10, 12, 15, 15, 12, 14, 15, 15,
986 15, 15, 15, 0, 15, 15, 0, 0, 14, 15, 15, 0,
987 15, 15, 15, 0, 15, 15, 0, 0, 0, 0, 0, 0,
988 15, 15, 0, 0, 15, 15
989 ];
990 const COOK_VQ4_CODES: &[u16; 246] = &[
991 0x0000, 0x0004, 0x006c, 0x03e6, 0x0005, 0x0012, 0x006d, 0x03e7, 0x006e, 0x00e8, 0x03e8, 0x3fc4,
992 0x07e0, 0x07e1, 0x7fa4, 0x7fa5, 0x0006, 0x0013, 0x01e2, 0x0fda, 0x0014, 0x0015, 0x00e9, 0x0fdb,
993 0x00ea, 0x006f, 0x03e9, 0x7fa6, 0x07e2, 0x07e3, 0x7fa7, 0x7fa8, 0x0070, 0x01e3, 0x0fdc, 0x7fa9,
994 0x00eb, 0x00ec, 0x0fdd, 0x7faa, 0x03ea, 0x03eb, 0x1fd6, 0x7fab, 0x3fc5, 0x3fc6, 0x7fac, 0x1fd6,
995 0x07e4, 0x1fd7, 0x7fad, 0x7fae, 0x07e5, 0x1fd8, 0x7faf, 0x7fb0, 0x3fc7, 0x7fb1, 0x7fb2, 0x1fd6,
996 0x7fb3, 0x7fb4, 0x1fd6, 0x1fd6, 0x0007, 0x0016, 0x01e4, 0x1fd9, 0x0017, 0x0032, 0x01e5, 0x1fda,
997 0x01e6, 0x01e7, 0x07e6, 0x7fb5, 0x3fc8, 0x1fdb, 0x7fb6, 0x7fb7, 0x0008, 0x0033, 0x01e8, 0x0fde,
998 0x0018, 0x0034, 0x01e9, 0x1fdc, 0x01ea, 0x00ed, 0x07e7, 0x7fb8, 0x1fdd, 0x0fdf, 0x7fb9, 0x7fba,
999 0x0071, 0x01eb, 0x0fe0, 0x7fbb, 0x0072, 0x00ee, 0x07e8, 0x7fbc, 0x03ec, 0x03ed, 0x3fc9, 0x7fbd,
1000 0x3fca, 0x7fbe, 0x7fbf, 0x3fc9, 0x03ee, 0x0fe1, 0x7fc0, 0x7fc1, 0x07e9, 0x1fde, 0x7fc2, 0x7fc3,
1001 0x7fc4, 0x7fc5, 0x7fc6, 0x3fc9, 0x7fc7, 0x7fc8, 0x3fc9, 0x3fc9, 0x0035, 0x01ec, 0x1fdf, 0x3fcb,
1002 0x00ef, 0x01ed, 0x0fe2, 0x7fc9, 0x0fe3, 0x0fe4, 0x7fca, 0x7fcb, 0x7fcc, 0x7fcd, 0x7fce, 0x7fca,
1003 0x0073, 0x01ee, 0x1fe0, 0x7fcf, 0x00f0, 0x01ef, 0x0fe5, 0x7fd0, 0x07ea, 0x0fe6, 0x7fd1, 0x7fd2,
1004 0x7fd3, 0x7fd4, 0x7fd5, 0x7fd1, 0x01f0, 0x07eb, 0x7fd6, 0x7fd7, 0x01f1, 0x07ec, 0x7fd8, 0x7fd9,
1005 0x3fcc, 0x3fcd, 0x7fda, 0x7fda, 0x7fdb, 0x7fdc, 0x7fda, 0x7fda, 0x3fce, 0x7fdd, 0x7fde, 0x7fd6,
1006 0x3fcf, 0x7fdf, 0x7fe0, 0x7fd8, 0x7fe1, 0x7fe2, 0x7fda, 0x7fda, 0x3fcc, 0x3fcd, 0x7fda, 0x7fda,
1007 0x01f2, 0x0fe7, 0x7fe3, 0x7fe4, 0x0fe8, 0x1fe1, 0x7fe5, 0x7fe6, 0x7fe7, 0x7fe8, 0x7fe9, 0x7fca,
1008 0x7fea, 0x7feb, 0x7fca, 0x7fca, 0x03ef, 0x0fe9, 0x7fec, 0x7fed, 0x0fea, 0x3fd0, 0x7fee, 0x7fef,
1009 0x7ff0, 0x7ff1, 0x7ff2, 0x7fd1, 0x7ff3, 0x7ff4, 0x7fd1, 0x7fd1, 0x3fd1, 0x7ff5, 0x7ff6, 0x7fd6,
1010 0x7ff7, 0x7ff8, 0x7ff9, 0x7fd8, 0x7ffa, 0x7ffb, 0x7fda, 0x7fda, 0x3fcc, 0x3fcd, 0x7fda, 0x7fda,
1011 0x7ffc, 0x7ffd, 0x7fd6, 0x7fd6, 0x7ffe, 0x7fff
1012 ];
1013 const COOK_VQ5_BITS: &[u8; 230] = &[
1014 2, 4, 8, 4, 5, 9, 9, 10, 14, 4, 6, 11,
1015 5, 6, 12, 10, 11, 15, 9, 11, 15, 10, 13, 15,
1016 14, 15, 0, 4, 6, 12, 6, 7, 12, 12, 12, 15,
1017 5, 7, 13, 6, 7, 13, 12, 13, 15, 10, 12, 15,
1018 11, 13, 15, 15, 15, 0, 8, 13, 15, 11, 12, 15,
1019 15, 15, 0, 10, 13, 15, 12, 15, 15, 15, 15, 0,
1020 15, 15, 0, 15, 15, 0, 0, 0, 0, 4, 5, 11,
1021 5, 7, 12, 11, 12, 15, 6, 7, 13, 7, 8, 14,
1022 12, 14, 15, 11, 13, 15, 12, 13, 15, 15, 15, 0,
1023 5, 6, 13, 7, 8, 15, 12, 14, 15, 6, 8, 14,
1024 7, 8, 15, 14, 15, 15, 12, 12, 15, 12, 13, 15,
1025 15, 15, 0, 9, 13, 15, 12, 13, 15, 15, 15, 0,
1026 11, 13, 15, 13, 13, 15, 15, 15, 0, 14, 15, 0,
1027 15, 15, 0, 0, 0, 0, 8, 10, 15, 11, 12, 15,
1028 15, 15, 0, 10, 12, 15, 12, 13, 15, 15, 15, 0,
1029 14, 15, 0, 15, 15, 0, 0, 0, 0, 8, 12, 15,
1030 12, 13, 15, 15, 15, 0, 11, 13, 15, 13, 15, 15,
1031 15, 15, 0, 15, 15, 0, 15, 15, 0, 0, 0, 0,
1032 14, 15, 0, 15, 15, 0, 0, 0, 0, 15, 15, 0,
1033 15, 15
1034 ];
1035 const COOK_VQ5_CODES: &[u16; 230] = &[
1036 0x0000, 0x0004, 0x00f0, 0x0005, 0x0012, 0x01f0, 0x01f1, 0x03e8, 0x3fce, 0x0006, 0x0030, 0x07de,
1037 0x0013, 0x0031, 0x0fd2, 0x03e9, 0x07df, 0x7fb0, 0x01f2, 0x07e0, 0x7fb1, 0x03ea, 0x1fd2, 0x7fb2,
1038 0x3fcf, 0x7fb3, 0x0031, 0x0007, 0x0032, 0x0fd3, 0x0033, 0x0070, 0x0fd4, 0x0fd5, 0x0fd6, 0x7fb4,
1039 0x0014, 0x0071, 0x1fd3, 0x0034, 0x0072, 0x1fd4, 0x0fd7, 0x1fd5, 0x7fb5, 0x03eb, 0x0fd8, 0x7fb6,
1040 0x07e1, 0x1fd6, 0x7fb7, 0x7fb8, 0x7fb9, 0x0072, 0x00f1, 0x1fd7, 0x7fba, 0x07e2, 0x0fd9, 0x7fbb,
1041 0x7fbc, 0x7fbd, 0x0070, 0x03ec, 0x1fd8, 0x7fbe, 0x0fda, 0x7fbf, 0x7fc0, 0x7fc1, 0x7fc2, 0x0072,
1042 0x7fc3, 0x7fc4, 0x0071, 0x7fc5, 0x7fc6, 0x0072, 0x0034, 0x0072, 0x0072, 0x0008, 0x0015, 0x07e3,
1043 0x0016, 0x0073, 0x0fdb, 0x07e4, 0x0fdc, 0x7fc7, 0x0035, 0x0074, 0x1fd9, 0x0075, 0x00f2, 0x3fd0,
1044 0x0fdd, 0x3fd1, 0x7fc8, 0x07e5, 0x1fda, 0x7fc9, 0x0fde, 0x1fdb, 0x7fca, 0x7fcb, 0x7fcc, 0x00f2,
1045 0x0017, 0x0036, 0x1fdc, 0x0076, 0x00f3, 0x7fcd, 0x0fdf, 0x3fd2, 0x7fce, 0x0037, 0x00f4, 0x3fd3,
1046 0x0077, 0x00f5, 0x7fcf, 0x3fd4, 0x7fd0, 0x7fd1, 0x0fe0, 0x0fe1, 0x7fd2, 0x0fe2, 0x1fdd, 0x7fd3,
1047 0x7fd4, 0x7fd5, 0x00f5, 0x01f3, 0x1fde, 0x7fd6, 0x0fe3, 0x1fdf, 0x7fd7, 0x7fd8, 0x7fd9, 0x00f3,
1048 0x07e6, 0x1fe0, 0x7fda, 0x1fe1, 0x1fe2, 0x7fdb, 0x7fdc, 0x7fdd, 0x00f5, 0x3fd5, 0x7fde, 0x00f4,
1049 0x7fdf, 0x7fe0, 0x00f5, 0x0077, 0x00f5, 0x00f5, 0x00f6, 0x03ed, 0x7fe1, 0x07e7, 0x0fe4, 0x7fe2,
1050 0x7fe3, 0x7fe4, 0x0073, 0x03ee, 0x0fe5, 0x7fe5, 0x0fe6, 0x1fe3, 0x7fe6, 0x7fe7, 0x7fe8, 0x00f2,
1051 0x3fd6, 0x7fe9, 0x0074, 0x7fea, 0x7feb, 0x00f2, 0x0075, 0x00f2, 0x00f2, 0x00f7, 0x0fe7, 0x7fec,
1052 0x0fe8, 0x1fe4, 0x7fed, 0x7fee, 0x7fef, 0x00f3, 0x07e8, 0x1fe5, 0x7ff0, 0x1fe6, 0x7ff1, 0x7ff2,
1053 0x7ff3, 0x7ff4, 0x00f5, 0x7ff5, 0x7ff6, 0x00f4, 0x7ff7, 0x7ff8, 0x00f5, 0x0077, 0x00f5, 0x00f5,
1054 0x3fd7, 0x7ff9, 0x0036, 0x7ffa, 0x7ffb, 0x00f3, 0x0076, 0x00f3, 0x00f3, 0x7ffc, 0x7ffd, 0x0000,
1055 0x7ffe, 0x7fff
1056 ];
1057 const COOK_VQ6_BITS: &[u8; 32] = &[
1058 1, 4, 4, 6, 4, 6, 6, 8, 4, 6, 6, 8,
1059 6, 9, 8, 10, 4, 6, 7, 8, 6, 9, 8, 11,
1060 6, 9, 8, 10, 8, 10, 9, 11
1061 ];
1062 const COOK_VQ6_CODES: &[u16; 32] = &[
1063 0x0000, 0x0008, 0x0009, 0x0034, 0x000a, 0x0035, 0x0036, 0x00f6,
1064 0x000b, 0x0037, 0x0038, 0x00f7, 0x0039, 0x01fa, 0x00f8, 0x03fc,
1065 0x000c, 0x003a, 0x007a, 0x00f9, 0x003b, 0x01fb, 0x00fa, 0x07fe,
1066 0x003c, 0x01fc, 0x00fb, 0x03fd, 0x00fc, 0x03fe, 0x01fd, 0x07ff
1067 ];
1068
1069 const COOK_CPL_SCALE2: &[f32; 5] = &[
1070 1.0, 0.953020632266998, 0.70710676908493, 0.302905440330505, 0.0
1071 ];
1072 const COOK_CPL_SCALE3: &[f32; 9] = &[
1073 1.0, 0.981279790401459, 0.936997592449188, 0.875934481620789, 0.70710676908493,
1074 0.482430040836334, 0.349335819482803, 0.192587479948997, 0.0
1075 ];
1076 const COOK_CPL_SCALE4: &[f32; 17] = &[
1077 1.0, 0.991486728191376, 0.973249018192291, 0.953020632266998, 0.930133521556854,
1078 0.903453230857849, 0.870746195316315, 0.826180458068848, 0.70710676908493,
1079 0.563405573368073, 0.491732746362686, 0.428686618804932, 0.367221474647522,
1080 0.302905440330505, 0.229752898216248, 0.130207896232605, 0.0
1081 ];
1082 const COOK_CPL_SCALE5: &[f32; 33] = &[
1083 1.0, 0.995926380157471, 0.987517595291138, 0.978726446628571, 0.969505727291107,
1084 0.95979779958725, 0.949531257152557, 0.938616216182709, 0.926936149597168,
1085 0.914336204528809, 0.900602877140045, 0.885426938533783, 0.868331849575043,
1086 0.84851086139679, 0.824381768703461, 0.791833400726318, 0.70710676908493,
1087 0.610737144947052, 0.566034197807312, 0.529177963733673, 0.495983630418777,
1088 0.464778542518616, 0.434642940759659, 0.404955863952637, 0.375219136476517,
1089 0.344963222742081, 0.313672333955765, 0.280692428350449, 0.245068684220314,
1090 0.205169528722763, 0.157508864998817, 0.0901700109243393, 0.0
1091 ];
1092 const COOK_CPL_SCALE6: &[f32; 65] = &[
1093 1.0, 0.998005926609039, 0.993956744670868, 0.989822506904602, 0.985598564147949,
1094 0.981279790401459, 0.976860702037811, 0.972335040569305, 0.967696130275726,
1095 0.962936460971832, 0.958047747612000, 0.953020632266998, 0.947844684123993,
1096 0.942508161067963, 0.936997592449188, 0.931297719478607, 0.925390899181366,
1097 0.919256627559662, 0.912870943546295, 0.906205296516418, 0.899225592613220,
1098 0.891890347003937, 0.884148240089417, 0.875934481620789, 0.867165684700012,
1099 0.857730865478516, 0.847477376461029, 0.836184680461884, 0.823513329029083,
1100 0.808890223503113, 0.791194140911102, 0.767520070075989, 0.707106769084930,
1101 0.641024887561798, 0.611565053462982, 0.587959706783295, 0.567296981811523,
1102 0.548448026180267, 0.530831515789032, 0.514098942279816, 0.498019754886627,
1103 0.482430040836334, 0.467206478118896, 0.452251672744751, 0.437485188245773,
1104 0.422837972640991, 0.408248275518417, 0.393658757209778, 0.379014074802399,
1105 0.364258885383606, 0.349335819482803, 0.334183186292648, 0.318732559680939,
1106 0.302905440330505, 0.286608695983887, 0.269728302955627, 0.252119421958923,
1107 0.233590632677078, 0.213876649737358, 0.192587479948997, 0.169101938605309,
1108 0.142307326197624, 0.109772264957428, 0.0631198287010193, 0.0
1109 ];
1110 const COOK_CPL_SCALES: [&[f32]; 5] = [
1111 COOK_CPL_SCALE2, COOK_CPL_SCALE3, COOK_CPL_SCALE4, COOK_CPL_SCALE5, COOK_CPL_SCALE6
1112 ];
1113
1114 const COOK_CPL_BAND: [u8; MAX_SUBBANDS - 1] = [
1115 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 11, 12, 12, 13,
1116 13, 14, 14, 14, 15, 15, 15, 15, 16, 16, 16, 16, 16, 17, 17, 17,
1117 17, 17, 17, 18, 18, 18, 18, 18, 18, 18, 19, 19, 19, 19, 19, 19,
1118 19, 19, 19
1119 ];
1120
1121 const COOK_DITHER_TAB: [f32; 9] = [ 0.0, 0.0, 0.0, 0.0, 0.0, 0.176777, 0.25, 0.707107, 1.0 ];
1122
1123 const COOK_QUANT_CENTROID: [[f32; 14]; 7] = [
1124 [ 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 ],
1125 [ 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 ],
1126 [ 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 ],
1127 [ 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 ],
1128 [ 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 ],
1129 [ 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 ],
1130 [ 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 ]
1131 ];
1132
1133 const COOK_EXP_BITS: [i32; NUM_CATEGORIES] = [ 52, 47, 43, 37, 29, 22, 16, 0 ];