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