make all codec crates export just register_all functions and document them
[nihav.git] / nihav-realmedia / src / codecs / cook.rs
CommitLineData
5641dccf
KS
1use nihav_core::formats::*;
2use nihav_core::frame::*;
3use nihav_core::codecs::*;
5641dccf
KS
4use nihav_core::dsp::mdct::IMDCT;
5use nihav_core::io::bitreader::*;
6use nihav_core::io::byteio::{ByteReader, MemoryReader};
7use nihav_core::io::codebook::*;
8use nihav_core::io::intcode::*;
594ca5ca
KS
9use std::f32::consts;
10use std::mem::swap;
11
12#[derive(Debug,Clone,Copy,PartialEq)]
13enum Mode {
14 Mono,
15 Stereo,
16 JointStereo,
17}
18
19impl Mode {
e07387c7
KS
20 fn get_channels(self) -> usize {
21 match self {
594ca5ca
KS
22 Mode::Mono => 1,
23 _ => 2,
24 }
25 }
26}
27
28struct CookBookReader {
29 bits: &'static [u8],
30 codes: &'static [u16],
31}
32impl 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
39struct Codebooks {
40 cpl_cb: [Codebook<u16>; 5],
41 quant_cb: Vec<Codebook<u16>>,
42 vq_cb: [Codebook<u16>; 7],
43}
44
45impl 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
84struct 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
94impl 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
118trait ClipCat {
119 fn clip_cat(&self) -> usize;
120}
121
122impl ClipCat for i32 {
123 fn clip_cat(&self) -> usize { ((*self).max(0) as usize).min(NUM_CATEGORIES - 1) }
124}
125
126const BAND_SIZE: usize = 20;
127const MAX_SAMPLES: usize = MAX_SUBBANDS * BAND_SIZE;
128const MAX_PAIRS: usize = 5;
129const MAX_SUBBANDS: usize = 52;
130const NUM_CATEGORIES: usize = 8;
131
132#[derive(Clone,Copy)]
133struct 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
152impl 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 }
fa90ccfb 305 let mut br = BitReader::new(&buf[..src.len()], BitReaderMode::BE);
594ca5ca
KS
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
465const COOK_VQ_GROUP_SIZE: [usize; 7] = [ 2, 2, 2, 4, 4, 5, 5 ];
466const COOK_NUM_VQ_GROUPS: [usize; 7] = [ 10, 10, 10, 5, 5, 4, 4 ];
467const COOK_VQ_INV_RADIX: [u32; 7] = [ 74899, 104858, 149797, 209716, 262144, 349526, 524288 ];
468const COOK_VQ_MULT: [u32; 7] = [ 13, 9, 6, 4, 3, 2, 1 ];
469fn 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
508struct RND {
509 state: u32,
510}
511
512impl 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
522struct 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
535impl 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
552impl 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 679pub fn get_decoder() -> Box<dyn NADecoder + Send> {
594ca5ca
KS
680 Box::new(CookDecoder::new())
681}
682
683#[cfg(test)]
684mod test {
3167c45c
KS
685 use nihav_core::codecs::RegisteredDecoders;
686 use nihav_core::demuxers::RegisteredDemuxers;
687 use nihav_core::test::dec_video::*;
e64739f8
KS
688 use crate::realmedia_register_all_codecs;
689 use crate::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
703const COOK_XOR_KEY: [u8; 4] = [ 0x37, 0xC5, 0x11, 0xF2 ];
704
705const COOK_CPL_2BITS_BITS: &[u8; 3] = &[ 2, 1, 2 ];
706const COOK_CPL_2BITS_CODES: &[u16; 3] = &[ 0x02, 0x00, 0x03 ];
707const COOK_CPL_3BITS_BITS: &[u8; 7] = &[ 6, 5, 2, 1, 3, 4, 6 ];
708const COOK_CPL_3BITS_CODES: &[u16; 7] = &[ 0x3e, 0x1e, 0x02, 0x00, 0x06, 0x0e, 0x3f ];
709const COOK_CPL_4BITS_BITS: &[u8; 15] = &[ 8, 8, 7, 6, 5, 4, 3, 1, 3, 4, 5, 6, 7, 8, 8 ];
710const COOK_CPL_4BITS_CODES: &[u16; 15] = &[
711 0xfc, 0xfd, 0x7c, 0x3c, 0x1c, 0x0c, 0x04, 0x00,
712 0x05, 0x0d, 0x1d, 0x3d, 0x7d, 0xfe, 0xff
713];
714const 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];
718const 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];
724const 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];
730const 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
741const 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];
756const 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
785const 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];
803const 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];
829const 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];
839const 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];
854const 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];
860const 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];
866const 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];
919const 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];
972const 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];
995const 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];
1018const 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];
1040const 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];
1062const 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];
1067const 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
1074const COOK_CPL_SCALE2: &[f32; 5] = &[
1075 1.0, 0.953020632266998, 0.70710676908493, 0.302905440330505, 0.0
1076];
1077const 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];
1081const 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];
1087const 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];
1097const 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];
1115const COOK_CPL_SCALES: [&[f32]; 5] = [
1116 COOK_CPL_SCALE2, COOK_CPL_SCALE3, COOK_CPL_SCALE4, COOK_CPL_SCALE5, COOK_CPL_SCALE6
1117];
1118
1119const 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
1127const COOK_DITHER_TAB: [f32; 9] = [ 0.0, 0.0, 0.0, 0.0, 0.0, 0.176777, 0.25, 0.707107, 1.0 ];
1128
1129const 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
1139const COOK_EXP_BITS: [i32; NUM_CATEGORIES] = [ 52, 47, 43, 37, 29, 22, 16, 0 ];