X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=nihav-qt%2Fsrc%2Fcodecs%2Frpza.rs;fp=nihav-qt%2Fsrc%2Fcodecs%2Frpza.rs;h=a3b2e5fafd784bdf89d5198619450f424a3e86f6;hb=4c1582cf2e275af7c0f4a2c1a397fed5b68d31d5;hp=0000000000000000000000000000000000000000;hpb=d341f57a0caf409d7dcc258b396cdee2080be399;p=nihav.git diff --git a/nihav-qt/src/codecs/rpza.rs b/nihav-qt/src/codecs/rpza.rs new file mode 100644 index 0000000..a3b2e5f --- /dev/null +++ b/nihav-qt/src/codecs/rpza.rs @@ -0,0 +1,217 @@ +use nihav_core::codecs::*; +use nihav_core::io::byteio::*; +use nihav_codec_support::codecs::HAMShuffler; + +const RGB555_FORMAT: NAPixelFormaton = NAPixelFormaton { + model: ColorModel::RGB(RGBSubmodel::RGB), components: 3, + comp_info: [ + Some(NAPixelChromaton{ h_ss: 0, v_ss: 0, packed: true, depth: 5, shift: 10, comp_offs: 0, next_elem: 2 }), + Some(NAPixelChromaton{ h_ss: 0, v_ss: 0, packed: true, depth: 5, shift: 5, comp_offs: 0, next_elem: 2 }), + Some(NAPixelChromaton{ h_ss: 0, v_ss: 0, packed: true, depth: 5, shift: 0, comp_offs: 0, next_elem: 2 }), + None, None], + elem_size: 2, be: false, alpha: false, palette: false }; + +#[derive(Default)] +struct RpzaDecoder { + info: NACodecInfoRef, + hams: HAMShuffler, + width: usize, + height: usize, +} + +impl RpzaDecoder { + fn new() -> Self { + Self::default() + } + fn put_block(dst: &mut [u16], dstride: usize, block: &[u16; 16]) { + for (line, src) in dst.chunks_mut(dstride).take(4).zip(block.chunks(4)) { + (&mut line[..4]).copy_from_slice(src); + } + } +} + +fn div3(a: u16, b: u16) -> u16 { + let r0 = (a >> 10) & 0x1F; + let g0 = (a >> 5) & 0x1F; + let b0 = a & 0x1F; + let r1 = (b >> 10) & 0x1F; + let g1 = (b >> 5) & 0x1F; + let b1 = b & 0x1F; + + let r = (r0 * 21 + r1 * 11) / 32; + let g = (g0 * 21 + g1 * 11) / 32; + let b = (b0 * 21 + b1 * 11) / 32; + + (r << 10) | (g << 5) | b +} + +impl NADecoder for RpzaDecoder { + fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> { + if let NACodecTypeInfo::Video(vinfo) = info.get_properties() { + self.width = (vinfo.get_width() + 3) & !3; + self.height = (vinfo.get_height() + 3) & !3; + let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(self.width, self.height, false, RGB555_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 { + let src = pkt.get_buffer(); + validate!(src.len() >= 2); + let mut mr = MemoryReader::new_read(src.as_slice()); + let mut br = ByteReader::new(&mut mr); + + let id = br.read_byte()?; + validate!(id == 0xE1); + let length = br.read_u24be()? as usize; + validate!(length == src.len()); + + let bufret = self.hams.clone_ref(); + let mut buf; + if let Some(bbuf) = bufret { + buf = bbuf; + } else { + let bufinfo = alloc_video_buffer(self.info.get_properties().get_video_info().unwrap(), 2)?; + buf = bufinfo.get_vbuf16().unwrap(); + self.hams.add_frame(buf); + buf = self.hams.get_output_frame().unwrap(); + } + let frm = NASimpleVideoFrame::from_video_buf(&mut buf).unwrap(); + let stride = frm.stride[0]; + let mut doff = frm.offset[0]; + + let mut x = 0; + let nblocks = (self.width / 4) * (self.height / 4); + let mut blockpos = 0; + let mut has_skips = false; + let mut block = [0u16; 16]; + while blockpos < nblocks { + let opcode = br.read_byte()?; + if (opcode & 0x80) == 0 { + let b2 = br.read_byte()?; + let clr0 = u16::from(opcode) * 256 + u16::from(b2); + if (br.peek_byte()? & 0x80) == 0 { + block[0] = clr0; + for el in block.iter_mut().skip(1) { + *el = br.read_u16be()?; + } + } else { + let clr3 = br.read_u16be()?; + let clr1 = div3(clr0, clr3); + let clr2 = div3(clr3, clr0); + let clr = [clr3, clr2, clr1, clr0]; + let flags = br.read_u32be()? as usize; + for i in 0..16 { + block[i] = clr[(flags >> (30 - i * 2)) & 3]; + } + } + Self::put_block(&mut frm.data[doff + x..], stride, &block); + x += 4; + if x == self.width { + x = 0; + doff += stride * 4; + } + blockpos += 1; + continue; + } + let len = ((opcode & 0x1F) as usize) + 1; + validate!(blockpos + len <= nblocks); + match opcode >> 5 { + 4 => { + has_skips = true; + for _ in 0..len { + x += 4; + if x == self.width { + x = 0; + doff += stride * 4; + } + } + }, + 5 => { + let clr = br.read_u16be()?; + block = [clr; 16]; + for _ in 0..len { + Self::put_block(&mut frm.data[doff + x..], stride, &block); + x += 4; + if x == self.width { + x = 0; + doff += stride * 4; + } + } + }, + 6 => { + let clr0 = br.read_u16be()?; + let clr3 = br.read_u16be()?; + let clr1 = div3(clr0, clr3); + let clr2 = div3(clr3, clr0); + let clr = [clr3, clr2, clr1, clr0]; + for _ in 0..len { + let flags = br.read_u32be()? as usize; + for i in 0..16 { + block[i] = clr[(flags >> (30 - i * 2)) & 3]; + } + Self::put_block(&mut frm.data[doff + x..], stride, &block); + x += 4; + if x == self.width { + x = 0; + doff += stride * 4; + } + } + }, + _ => { + return Err(DecoderError::InvalidData); + }, + }; + blockpos += len; + } + + let buftype = NABufferType::Video16(buf); + + let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), buftype); + frm.set_keyframe(!has_skips); + frm.set_frame_type(if !has_skips { FrameType::I } else { FrameType::P }); + Ok(frm.into_ref()) + } + fn flush(&mut self) { + self.hams.clear(); + } +} + +impl NAOptionHandler for RpzaDecoder { + fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] } + fn set_options(&mut self, _options: &[NAOption]) { } + fn query_option_value(&self, _name: &str) -> Option { None } +} + +pub fn get_decoder() -> Box { + Box::new(RpzaDecoder::new()) +} + +#[cfg(test)] +mod test { + use nihav_core::codecs::RegisteredDecoders; + use nihav_core::demuxers::RegisteredDemuxers; + use nihav_codec_support::test::dec_video::*; + use crate::qt_register_all_codecs; + use nihav_commonfmt::generic_register_all_demuxers; + #[test] + fn test_rpza() { + let mut dmx_reg = RegisteredDemuxers::new(); + generic_register_all_demuxers(&mut dmx_reg); + let mut dec_reg = RegisteredDecoders::new(); + qt_register_all_codecs(&mut dec_reg); + + test_decoding("mov", "apple-video", "assets/QT/aletrek-rpza.mov", Some(6), &dmx_reg, &dec_reg, + ExpectedTestResult::MD5Frames(vec![ + [0xe7cfc941, 0xa448841b, 0x75afc888, 0x94aa064b], + [0xadaaec50, 0xff7319ff, 0xa1f3f64a, 0xc40c2985], + [0xdfa5c4b9, 0xdac2d22b, 0x14f9f281, 0x7295eae7], + [0x3f3420b9, 0xbe48b885, 0x91b0fb51, 0xd71462ac], + [0xcaae1580, 0x16eecce3, 0x2ca0dd4b, 0x7f9c62e3], + [0xb51fc759, 0xe1cfc171, 0xda854767, 0x878f6e17], + [0x8d3abf6d, 0xeecec7a2, 0x5832e5d6, 0x86145fc9]])); + } +}