1 use nihav_core::frame::*;
2 use nihav_core::codecs::*;
3 use nihav_core::io::byteio::*;
5 #[derive(Clone,Copy,Debug,PartialEq)]
20 struct QVideoDecoder {
34 macro_rules! copy_tile {
35 ($self: expr, $doff: expr, $soff: expr) => {
38 for _y in 0..$self.tile_h {
39 for x in 0..$self.tile_w {
40 $self.frame[doff + x] = $self.frame[soff + x];
51 info: NACodecInfoRef::default(),
65 fn decode_mode7_tile(dst: &mut [u8], stride: usize, br: &mut ByteReader) -> DecoderResult<()> {
66 let op = br.peek_byte()?;
68 for dline in dst.chunks_mut(stride).take(4) {
69 br.read_buf(&mut dline[..4])?;
71 } else if op == 0xF8 || op == 0xFF {
76 let nclr = (op - 0xF6) as usize;
78 let mut pattern = br.read_u32le()?;
79 br.read_buf(&mut clr[..nclr])?;
80 for dline in dst.chunks_mut(stride).take(4) {
81 for el in dline[..4].iter_mut() {
82 *el = clr[(pattern & 3) as usize];
87 let mut pattern = br.read_u24le()?;
88 let pattern2 = br.read_u24le()?;
89 br.read_buf(&mut clr[..nclr])?;
90 for (y, dline) in dst.chunks_mut(stride).take(4).enumerate() {
91 for el in dline[..4].iter_mut() {
92 *el = clr[(pattern & 7) as usize];
104 fn decode_frame_v3(&mut self, br: &mut ByteReader, _ctype: u16) -> DecoderResult<()> {
105 let mut titer = self.tile_off.iter().enumerate();
106 let mut skip_mode = false;
107 while let Some((tile_no, &tile_off)) = titer.next() {
108 let op = br.read_byte()?;
111 let clr1 = br.read_byte()?;
113 for dline in self.frame[tile_off..].chunks_mut(self.w).take(self.tile_h) {
114 for el in dline[..self.tile_w].iter_mut() {
119 let mut pattern = br.read_u16le()?;
120 for dline in self.frame[tile_off..].chunks_mut(self.w).take(self.tile_h) {
121 for el in dline[..self.tile_w].iter_mut() {
122 *el = if (pattern & 0x8000) == 0 { clr0 } else { clr1 };
133 let run = br.read_byte()? as usize;
136 validate!(tile_no > 0);
137 let mut tile_off = tile_off;
140 copy_tile!(self, tile_off, self.tile_off[tile_no - 1]);
143 let (_tno, &toff) = titer.next().unwrap();
149 let off = br.read_u16le()? as usize;
150 validate!(tile_no + off < self.tile_off.len());
151 copy_tile!(self, tile_off, self.tile_off[tile_no + off]);
154 let off = br.read_u16le()? as usize;
155 validate!(off <= tile_no);
156 copy_tile!(self, tile_off, self.tile_off[tile_no - off]);
159 const MV_PART: [i8; 16] = [ 0, 4, 8, 12, 16, 20, 24, 28, -32, -4, -8, -12, -16, -20, -24, -28 ];
161 let idx = br.read_byte()? as usize;
162 let x = MV_PART[(idx & 0xF) as usize] as isize;
163 let y = MV_PART[(idx >> 4) as usize] as isize;
164 let src_off = (tile_off as isize) + x + y * (self.w as isize);
165 validate!(src_off >= 0);
166 validate!((src_off as usize) + self.tile_w + (self.tile_h - 1) * self.w <= self.w * self.h);
168 copy_tile!(self, tile_off, src_off as usize);
171 let off = (br.read_byte()? as usize) + 1;
172 validate!(tile_no + off < self.tile_off.len());
173 copy_tile!(self, tile_off, self.tile_off[tile_no + off]);
176 let off = (br.read_byte()? as usize) + 1;
177 validate!(off <= tile_no);
178 copy_tile!(self, tile_off, self.tile_off[tile_no - off]);
183 skip_mode = op == 0xFF;
189 fn decode_frame(&mut self, br: &mut ByteReader, _ctype: u16) -> DecoderResult<()> {
190 let mut titer = self.tile_off.iter().enumerate();
191 let mut last_mode = TileMode::Start;
193 while let Some((tile_no, &tile_off)) = titer.next() {
194 let op = br.read_byte()?;
197 let clr1 = br.read_byte()?;
199 for dline in self.frame[tile_off..].chunks_mut(self.w).take(self.tile_h) {
200 for el in dline[..self.tile_w].iter_mut() {
204 last_mode = TileMode::Fill;
206 let pat = br.read_byte()?;
207 let mut pattern = if pat < 128 {
208 last_mode = TileMode::ShortPattern(clr0, clr1);
209 self.patterns[pat as usize]
211 last_mode = TileMode::LongPattern(clr0, clr1);
212 u16::from(pat) | (u16::from(br.read_byte()?) << 8)
214 for dline in self.frame[tile_off..].chunks_mut(self.w).take(self.tile_h) {
215 for el in dline[..self.tile_w].iter_mut() {
216 *el = if (pattern & 0x8000) == 0 { clr0 } else { clr1 };
227 let run = br.read_byte()? as usize;
230 validate!(tile_no > 0);
231 validate!(last_mode != TileMode::Start);
232 let mut tile_no = tile_no;
233 let mut tile_off = tile_off;
235 let copy_off = match last_mode {
236 TileMode::Forward(off) => {
237 tile_no + (off as usize)
239 TileMode::Backward(off) => {
240 validate!(tile_no >= (off as usize));
241 tile_no - (off as usize)
243 TileMode::Skip => self.tile_off.len(),
246 if copy_off < self.tile_off.len() {
247 copy_tile!(self, tile_off, self.tile_off[copy_off]);
250 let (tno, &toff) = titer.next().unwrap();
255 last_mode = TileMode::Run;
258 let rtile = br.read_u16le()? as usize;
259 validate!(rtile < self.tile_off.len());
260 copy_tile!(self, tile_off, self.tile_off[rtile]);
261 last_mode = TileMode::Reuse;
266 let run = br.read_byte()? as usize;
268 let mut tile_no = tile_no;
269 let mut tile_off = tile_off;
272 TileMode::Start => return Err(DecoderError::InvalidData),
274 let clr = br.read_byte()?;
275 for dline in self.frame[tile_off..].chunks_mut(self.w).take(self.tile_h) {
276 for el in dline[..self.tile_w].iter_mut() {
281 TileMode::ShortPattern(clr0, clr1) => {
282 let pat = br.read_byte()?;
283 let mut pattern = if pat < 128 {
284 self.patterns[pat as usize]
286 u16::from(pat) | (u16::from(br.read_byte()?) << 8)
288 for dline in self.frame[tile_off..].chunks_mut(self.w).take(self.tile_h) {
289 for el in dline[..self.tile_w].iter_mut() {
290 *el = if (pattern & 0x8000) == 0 { clr0 } else { clr1 };
295 TileMode::LongPattern(clr0, clr1) => {
296 let mut pattern = br.read_u16le()?;
297 for dline in self.frame[tile_off..].chunks_mut(self.w).take(self.tile_h) {
298 for el in dline[..self.tile_w].iter_mut() {
299 *el = if (pattern & 0x8000) == 0 { clr0 } else { clr1 };
305 let rtile = br.read_u16le()? as usize;
306 validate!(rtile < self.tile_off.len());
307 copy_tile!(self, tile_off, self.tile_off[rtile]);
310 let idx = br.read_byte()? as usize;
311 let (x, y) = DEF_MVS[idx];
312 let src_off = (tile_off as isize) + (x as isize) * 4 + (y as isize) * 4 * (self.w as isize);
313 validate!(src_off >= 0);
314 validate!((src_off as usize) + self.tile_w + (self.tile_h - 1) * self.w <= self.w * self.h);
315 copy_tile!(self, tile_off, src_off as usize);
317 TileMode::Forward(_) => {
318 let off = (br.read_byte()? as usize) + 1;
319 validate!(tile_no + off < self.tile_off.len());
320 copy_tile!(self, tile_off, self.tile_off[tile_no + off]);
322 TileMode::Backward(_) => {
323 let off = (br.read_byte()? as usize) + 1;
324 validate!(off <= tile_no);
325 copy_tile!(self, tile_off, self.tile_off[tile_no - off]);
327 _ => unimplemented!(),
331 let (tno, &toff) = titer.next().unwrap();
338 validate!(self.tile_w == 4 && self.tile_h == 4);
339 let run = br.read_byte()? as usize;
342 let mut tile_off = tile_off;
344 Self::decode_mode7_tile(&mut self.frame[tile_off..], self.w, br)?;
347 let (_tno, &toff) = titer.next().unwrap();
356 last_mode = TileMode::FB;
359 let idx = br.read_byte()? as usize;
360 let (x, y) = DEF_MVS[idx];
361 let src_off = (tile_off as isize) + (x as isize) * 4 + (y as isize) * 4 * (self.w as isize);
362 validate!(src_off >= 0);
363 validate!((src_off as usize) + self.tile_w + (self.tile_h - 1) * self.w <= self.w * self.h);
365 copy_tile!(self, tile_off, src_off as usize);
366 last_mode = TileMode::MV;
369 let off = (br.read_byte()? as usize) + 1;
370 validate!(tile_no + off < self.tile_off.len());
371 copy_tile!(self, tile_off, self.tile_off[tile_no + off]);
372 last_mode = TileMode::Forward(off as u16);
375 let off = (br.read_byte()? as usize) + 1;
376 validate!(off <= tile_no);
377 copy_tile!(self, tile_off, self.tile_off[tile_no - off]);
378 last_mode = TileMode::Backward(off as u16);
381 last_mode = TileMode::Skip;
390 fn output_frame(&mut self, bufinfo: &mut NABufferType, w: usize, h: usize) {
391 let bufo = bufinfo.get_vbuf();
392 let mut buf = bufo.unwrap();
393 let paloff = buf.get_offset(1);
394 let stride = buf.get_stride(0);
395 let data = buf.get_data_mut().unwrap();
396 let dst = data.as_mut_slice();
398 dst[paloff..][..768].copy_from_slice(&self.pal);
399 for (dline, sline) in dst.chunks_mut(stride).zip(self.frame.chunks(w)).take(h) {
400 dline[..w].copy_from_slice(sline);
405 impl NADecoder for QVideoDecoder {
406 fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
407 if let NACodecTypeInfo::Video(_vinfo) = info.get_properties() {
410 if let Some(buf) = info.get_extradata() {
411 validate!(buf.len() >= 22);
412 w = read_u16le(&buf[4..])? as usize;
413 h = read_u16le(&buf[6..])? as usize;
414 self.tile_w = buf[8] as usize;
415 self.tile_h = buf[9] as usize;
416 validate!(self.tile_w > 0 && self.tile_h > 0);
417 if self.tile_w != 4 || self.tile_h != 4 {
418 return Err(DecoderError::NotImplemented);
420 self.version = buf[2];
421 if self.version != 3{
425 validate!((w % self.tile_w) == 0);
426 validate!((h % self.tile_h) == 0);
429 return Err(DecoderError::InvalidData);
431 let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(w, h, false, PAL8_FORMAT));
432 self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();
436 self.mode = match self.version {
442 self.frame.resize(w * h, 0);
444 self.tile_off = Vec::with_capacity((w / self.tile_w) * (h / self.tile_h));
446 for _y in (0..h).step_by(self.tile_h) {
447 for x in (0..w).step_by(self.tile_w) {
448 self.tile_off.push(off + x);
450 off += w * self.tile_h;
454 Err(DecoderError::InvalidData)
457 fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
458 let src = pkt.get_buffer();
459 validate!(!src.is_empty());
461 let mut mr = MemoryReader::new_read(&src);
462 let mut br = ByteReader::new(&mut mr);
464 while br.left() >= 6 {
465 let ctype = br.read_u16le()?;
466 let csize = br.read_u32le()? as usize;
467 validate!(csize <= (br.left() as usize));
470 validate!(csize <= 768);
471 br.read_buf(&mut self.pal[..csize])?;
472 for el in self.pal[..csize].iter_mut() {
473 *el = (*el << 2) | (*el >> 4);
476 2 | 3 | 4 | 9 | 11 => {
477 if self.version == 5 {
478 self.mode = if ctype == 9 || ctype == 11 { 7 } else { 6 };
480 if self.version == 3 {
481 self.decode_frame_v3(&mut br, ctype)?;
483 self.decode_frame(&mut br, ctype)?;
487 validate!(csize <= 256 && (csize & 1) == 0);
488 for el in self.patterns[..csize/2].iter_mut() {
489 *el = br.read_u16le()?;
493 self.mode = ctype as u8;
495 _ => return Err(DecoderError::InvalidData),
499 let mut bufinfo = alloc_video_buffer(self.info.get_properties().get_video_info().unwrap(), 0)?;
501 self.output_frame(&mut bufinfo, self.w, self.h);
503 let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo);
504 frm.set_frame_type(if pkt.is_keyframe() { FrameType::I } else { FrameType::P });
507 fn flush(&mut self) {
511 impl NAOptionHandler for QVideoDecoder {
512 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
513 fn set_options(&mut self, _options: &[NAOption]) { }
514 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
517 pub fn get_decoder() -> Box<dyn NADecoder + Send> {
518 Box::new(QVideoDecoder::new())
523 use nihav_core::codecs::RegisteredDecoders;
524 use nihav_core::demuxers::RegisteredDemuxers;
525 use nihav_codec_support::test::dec_video::*;
526 use crate::game_register_all_decoders;
527 use crate::game_register_all_demuxers;
531 let mut dmx_reg = RegisteredDemuxers::new();
532 game_register_all_demuxers(&mut dmx_reg);
533 let mut dec_reg = RegisteredDecoders::new();
534 game_register_all_decoders(&mut dec_reg);
536 test_decoding("legend-q", "legend-q-video", "assets/Game/dgate101.q", Some(31), &dmx_reg, &dec_reg,
537 ExpectedTestResult::MD5([0x9cc0014c, 0xf6332802, 0xfabeb715, 0xdfaa11c0]));
541 let mut dmx_reg = RegisteredDemuxers::new();
542 game_register_all_demuxers(&mut dmx_reg);
543 let mut dec_reg = RegisteredDecoders::new();
544 game_register_all_decoders(&mut dec_reg);
546 test_decoding("legend-q", "legend-q-video", "assets/Game/1925.Q", None, &dmx_reg, &dec_reg,
547 ExpectedTestResult::MD5([0xe1af971a, 0xfb509816, 0x9d60f5d6, 0xbcf48a3b]));
551 let mut dmx_reg = RegisteredDemuxers::new();
552 game_register_all_demuxers(&mut dmx_reg);
553 let mut dec_reg = RegisteredDecoders::new();
554 game_register_all_decoders(&mut dec_reg);
556 test_decoding("legend-q", "legend-q-video", "assets/Game/mc703.q", Some(16), &dmx_reg, &dec_reg,
557 ExpectedTestResult::MD5([0xf65ea3ce, 0x3052b2bb, 0xb10f8f69, 0x530d60f9]));
561 const DEF_MVS: [(i8, i8); 256] = [
562 ( 0, 8), ( 1, 8), ( 2, 8), ( 3, 8), ( 4, 8), ( 5, 8), ( 6, 8), ( 7, 8),
563 (-8, 8), (-1, 8), (-2, 8), (-3, 8), (-4, 8), (-5, 8), (-6, 8), (-7, 8),
564 ( 0, 9), ( 1, 9), ( 2, 9), ( 3, 9), ( 4, 9), ( 5, 9), ( 6, 9), ( 7, 9),
565 (-8, 9), (-1, 9), (-2, 9), (-3, 9), (-4, 9), (-5, 9), (-6, 9), (-7, 9),
566 ( 0, 2), ( 1, 2), ( 2, 2), ( 3, 2), ( 4, 2), ( 5, 2), ( 6, 2), ( 7, 2),
567 (-8, 2), (-1, 2), (-2, 2), (-3, 2), (-4, 2), (-5, 2), (-6, 2), (-7, 2),
568 ( 0, 3), ( 1, 3), ( 2, 3), ( 3, 3), ( 4, 3), ( 5, 3), ( 6, 3), ( 7, 3),
569 (-8, 3), (-1, 3), (-2, 3), (-3, 3), (-4, 3), (-5, 3), (-6, 3), (-7, 3),
570 ( 0, 4), ( 1, 4), ( 2, 4), ( 3, 4), ( 4, 4), ( 5, 4), ( 6, 4), ( 7, 4),
571 (-8, 4), (-1, 4), (-2, 4), (-3, 4), (-4, 4), (-5, 4), (-6, 4), (-7, 4),
572 ( 0, 5), ( 1, 5), ( 2, 5), ( 3, 5), ( 4, 5), ( 5, 5), ( 6, 5), ( 7, 5),
573 (-8, 5), (-1, 5), (-2, 5), (-3, 5), (-4, 5), (-5, 5), (-6, 5), (-7, 5),
574 ( 0, 6), ( 1, 6), ( 2, 6), ( 3, 6), ( 4, 6), ( 5, 6), ( 6, 6), ( 7, 6),
575 (-8, 6), (-1, 6), (-2, 6), (-3, 6), (-4, 6), (-5, 6), (-6, 6), (-7, 6),
576 ( 0, 7), ( 1, 7), ( 2, 7), ( 3, 7), ( 4, 7), ( 5, 7), ( 6, 7), ( 7, 7),
577 (-8, 7), (-1, 7), (-2, 7), (-3, 7), (-4, 7), (-5, 7), (-6, 7), (-7, 7),
578 ( 0,-8), ( 1,-8), ( 2,-8), ( 3,-8), ( 4,-8), ( 5,-8), ( 6,-8), ( 7,-8),
579 (-8,-8), (-1,-8), (-2,-8), (-3,-8), (-4,-8), (-5,-8), (-6,-8), (-7,-8),
580 ( 0,-9), ( 1,-9), ( 2,-9), ( 3,-9), ( 4,-9), ( 5,-9), ( 6,-9), ( 7,-9),
581 (-8,-9), (-1,-9), (-2,-9), (-3,-9), (-4,-9), (-5,-9), (-6,-9), (-7,-9),
582 ( 0,-2), ( 1,-2), ( 2,-2), ( 3,-2), ( 4,-2), ( 5,-2), ( 6,-2), ( 7,-2),
583 (-8,-2), (-1,-2), (-2,-2), (-3,-2), (-4,-2), (-5,-2), (-6,-2), (-7,-2),
584 ( 0,-3), ( 1,-3), ( 2,-3), ( 3,-3), ( 4,-3), ( 5,-3), ( 6,-3), ( 7,-3),
585 (-8,-3), (-1,-3), (-2,-3), (-3,-3), (-4,-3), (-5,-3), (-6,-3), (-7,-3),
586 ( 0,-4), ( 1,-4), ( 2,-4), ( 3,-4), ( 4,-4), ( 5,-4), ( 6,-4), ( 7,-4),
587 (-8,-4), (-1,-4), (-2,-4), (-3,-4), (-4,-4), (-5,-4), (-6,-4), (-7,-4),
588 ( 0,-5), ( 1,-5), ( 2,-5), ( 3,-5), ( 4,-5), ( 5,-5), ( 6,-5), ( 7,-5),
589 (-8,-5), (-1,-5), (-2,-5), (-3,-5), (-4,-5), (-5,-5), (-6,-5), (-7,-5),
590 ( 0,-6), ( 1,-6), ( 2,-6), ( 3,-6), ( 4,-6), ( 5,-6), ( 6,-6), ( 7,-6),
591 (-8,-6), (-1,-6), (-2,-6), (-3,-6), (-4,-6), (-5,-6), (-6,-6), (-7,-6),
592 ( 0,-7), ( 1,-7), ( 2,-7), ( 3,-7), ( 4,-7), ( 5,-7), ( 6,-7), ( 7,-7),
593 (-8,-7), (-1,-7), (-2,-7), (-3,-7), (-4,-7), (-5,-7), (-6,-7), (-7,-7)