diff options
author | Kostya Shishkov <kostya.shishkov@gmail.com> | 2024-09-14 11:39:39 +0200 |
---|---|---|
committer | Kostya Shishkov <kostya.shishkov@gmail.com> | 2024-09-14 11:39:39 +0200 |
commit | 4efc9d7e0b8b03a36e62c54aa94edc5662649071 (patch) | |
tree | 03d4cd62052474f1c32e023d948163b010b1a26d /nihav-game/src/codecs | |
parent | 26869f5fada89416a5714307552e47962ae54c39 (diff) | |
download | nihav-master.tar.gz |
Diffstat (limited to 'nihav-game/src/codecs')
-rw-r--r-- | nihav-game/src/codecs/imax.rs | 166 |
1 files changed, 0 insertions, 166 deletions
diff --git a/nihav-game/src/codecs/imax.rs b/nihav-game/src/codecs/imax.rs deleted file mode 100644 index de0bf98..0000000 --- a/nihav-game/src/codecs/imax.rs +++ /dev/null @@ -1,166 +0,0 @@ -use nihav_core::codecs::*; -use nihav_core::io::byteio::*; - -const FRAME_W: usize = 320; -const FRAME_H: usize = 160; - -struct IMAXDecoder { - info: NACodecInfoRef, - pal: [u8; 768], - frame: [u8; FRAME_W * FRAME_H], - hist: [u8; 32768], - hist_pos: usize, -} - -impl IMAXDecoder { - fn new() -> Self { - Self { - info: NACodecInfoRef::default(), - pal: [0; 768], - frame: [0; FRAME_W * FRAME_H], - hist: [0; 32768], - hist_pos: 0, - } - } -} - -impl NADecoder for IMAXDecoder { - fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> { - if let NACodecTypeInfo::Video(_vinfo) = info.get_properties() { - /*let fmt = NAPixelFormaton::new(ColorModel::RGB(RGBSubmodel::RGB), - Some(NAPixelChromaton::new(0, 0, true, 8, 0, 0, 3)), - Some(NAPixelChromaton::new(0, 0, true, 8, 0, 1, 3)), - Some(NAPixelChromaton::new(0, 0, true, 8, 0, 2, 3)), - None, None, - FORMATON_FLAG_PALETTE, 3);*/ - let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(FRAME_W, FRAME_H, false, PAL8_FORMAT)); - self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref(); - - Ok(()) - } else { - Err(DecoderError::InvalidData) - } - } - fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> { - let src = pkt.get_buffer(); - validate!(src.len() > 0); - - for sd in pkt.side_data.iter() { - if let NASideData::Palette(true, ref pal) = sd { - for (dst, src) in self.pal.chunks_mut(3).zip(pal.chunks(4)) { - dst[0] = src[0]; - dst[1] = src[1]; - dst[2] = src[2]; - } - break; - } - } - - let mut mr = MemoryReader::new_read(&src); - let mut br = ByteReader::new(&mut mr); - - let mut is_intra = true; - let mut is_skip = true; - let mut idx = 0; - while idx < self.frame.len() { - let v = br.read_byte()?; - let op = v >> 6; - let len = (v & 0x3F) as usize; - match op { - 0 => { - validate!(idx + len <= self.frame.len()); - idx += len; - is_intra = false; - }, - 1 => { - if len == 0 { - let off = br.read_u16le()? as usize; - let len = br.read_byte()? as usize; - validate!(idx + len <= self.frame.len()); - validate!(off + len <= self.hist.len()); - self.frame[idx..][..len].copy_from_slice(&self.hist[off..][..len]); - } else { - validate!(idx + len <= self.frame.len()); - br.read_buf(&mut self.frame[idx..][..len])?; - if self.hist_pos + len <= self.hist.len() { - self.hist[self.hist_pos..][..len].copy_from_slice(&self.frame[idx..][..len]); - self.hist_pos += len; - } - idx += len; - } - is_skip = false; - }, - 2 => { - let pix = br.read_byte()?; - validate!(idx + len <= self.frame.len()); - for _ in 0..len { - self.frame[idx] = pix; - idx += 1; - } - is_skip = false; - }, - _ => { - let len2 = br.read_byte()? as usize; - let skip_len = len * 64 + len2; - validate!(idx + skip_len <= self.frame.len()); - idx += skip_len; - is_intra = false; - }, - }; - } - - let bufinfo = if !is_skip { - let binfo = alloc_video_buffer(self.info.get_properties().get_video_info().unwrap(), 0)?; - let mut vbuf = binfo.get_vbuf().unwrap(); - let paloff = vbuf.get_offset(1); - let stride = vbuf.get_stride(0); - let data = vbuf.get_data_mut().unwrap(); - for (drow, srow) in data.chunks_mut(stride).zip(self.frame.chunks(FRAME_W)) { - drow[..FRAME_W].copy_from_slice(srow); - } - data[paloff..][..768].copy_from_slice(&self.pal); - binfo - } else { - NABufferType::None - }; - - let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo); - frm.set_keyframe(is_intra); - let ftype = if is_skip { FrameType::Skip } else if is_intra { FrameType::I } else { FrameType::P }; - frm.set_frame_type(ftype); - Ok(frm.into_ref()) - } - fn flush(&mut self) { - } -} - -impl NAOptionHandler for IMAXDecoder { - fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] } - fn set_options(&mut self, _options: &[NAOption]) { } - fn query_option_value(&self, _name: &str) -> Option<NAValue> { None } -} - -pub fn get_decoder() -> Box<dyn NADecoder + Send> { - Box::new(IMAXDecoder::new()) -} - -#[cfg(test)] -mod test { - use nihav_core::codecs::RegisteredDecoders; - use nihav_core::demuxers::RegisteredDemuxers; - use nihav_codec_support::test::dec_video::*; - use crate::game_register_all_decoders; - use crate::game_register_all_demuxers; - #[test] - fn test_imax_video() { - let mut dmx_reg = RegisteredDemuxers::new(); - game_register_all_demuxers(&mut dmx_reg); - let mut dec_reg = RegisteredDecoders::new(); - game_register_all_decoders(&mut dec_reg); - - // sample from Fable game - test_decoding("fable-imax", "fable-imax", "assets/Game/present.imx", - Some(64), &dmx_reg, &dec_reg, - ExpectedTestResult::MD5([0x775e1326, 0x7aa63674, 0x9b8aec54, 0x5caee2e3])); - } -} |