1 use nihav_core::codecs::*;
11 info: NACodecInfo::new_dummy(),
17 impl NADecoder for RawDecoder {
18 fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
19 if let NACodecTypeInfo::Video(_vinfo) = info.get_properties() {
20 self.info = info.clone();
23 Err(DecoderError::InvalidData)
26 fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
27 if let NACodecTypeInfo::Video(ref vinfo) = self.info.get_properties() {
28 let src = pkt.get_buffer();
29 let width = vinfo.width;
30 let height = vinfo.height;
31 if !vinfo.format.model.is_rgb() {
32 return Err(DecoderError::InvalidData);
34 let depth = if vinfo.format.is_paletted() { 8 } else { vinfo.format.get_total_depth() } as usize;
35 let buf = match depth {
37 for sd in pkt.side_data.iter() {
39 NASideData::Palette(_, ref pal) => {
40 for (dst, src) in self.pal.chunks_mut(3).zip(pal.chunks(4)) {
41 dst.copy_from_slice(&src[..3]);
49 let sstride = (vinfo.width + 3) & !3;
51 let buf = alloc_video_buffer(*vinfo, 0)?;
53 let mut frm = buf.get_vbuf().unwrap();
54 let dstride = frm.get_stride(0);
55 let paloff = frm.get_offset(1);
56 let dst = frm.get_data_mut().unwrap();
57 for (drow, srow) in dst.chunks_mut(dstride).zip(src.chunks(sstride)).take(height) {
58 drow[..width].copy_from_slice(&srow[..width]);
60 dst[paloff..][..768].copy_from_slice(&self.pal);
65 let sstride = (vinfo.width * 2 + 3) & !3;
67 let buf = alloc_video_buffer(*vinfo, 0)?;
69 let mut frm = buf.get_vbuf16().unwrap();
70 let dstride = frm.get_stride(0);
71 let dst = frm.get_data_mut().unwrap();
73 for (drow, srow) in dst.chunks_mut(dstride).zip(src.chunks(sstride)).take(height) {
74 for (dp, sp) in drow.iter_mut().zip(srow.chunks_exact(2)).take(width) {
75 *dp = u16::from(sp[0]) | (u16::from(sp[1]) << 8);
82 let ncomp = vinfo.format.components as usize;
83 let sstride = (width * (depth / 8) + 3) & !3;
84 let offs = vec![0; ncomp];
85 let mut strides = vec![0; ncomp];
87 NABufferType::VideoPacked(NAVideoBuffer::from_raw_parts(*vinfo, src, offs, strides).into_ref())
89 _ => return Err(DecoderError::NotImplemented),
92 let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), buf);
93 frm.set_keyframe(true);
94 frm.set_frame_type(FrameType::I);
97 Err(DecoderError::Bug)
100 fn flush(&mut self) {}
104 impl NAOptionHandler for RawDecoder {
105 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
106 fn set_options(&mut self, _options: &[NAOption]) { }
107 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
110 pub fn get_decoder() -> Box<dyn NADecoder + Send> {
111 Box::new(RawDecoder::new())
116 use nihav_core::codecs::RegisteredDecoders;
117 use nihav_core::demuxers::RegisteredDemuxers;
118 use nihav_codec_support::test::dec_video::*;
119 use crate::generic_register_all_decoders;
120 use crate::generic_register_all_demuxers;
122 fn test_rawvideo_ms_8() {
123 let mut dmx_reg = RegisteredDemuxers::new();
124 generic_register_all_demuxers(&mut dmx_reg);
125 let mut dec_reg = RegisteredDecoders::new();
126 generic_register_all_decoders(&mut dec_reg);
127 // sample: https://samples.mplayerhq.hu/V-codecs/Uncompressed/8bpp.avi
128 test_decoding("avi", "rawvideo-ms", "assets/Misc/8bpp.avi", Some(0), &dmx_reg,
129 &dec_reg, ExpectedTestResult::MD5([0xb6629439, 0x6ea482e9, 0x42c84d7c, 0x46c94431]));
132 fn test_rawvideo_ms_16() {
133 let mut dmx_reg = RegisteredDemuxers::new();
134 generic_register_all_demuxers(&mut dmx_reg);
135 let mut dec_reg = RegisteredDecoders::new();
136 generic_register_all_decoders(&mut dec_reg);
137 // sample: https://samples.mplayerhq.hu/V-codecs/Uncompressed/16bpp.avi
138 test_decoding("avi", "rawvideo-ms", "assets/Misc/16bpp.avi", Some(0), &dmx_reg,
139 &dec_reg, ExpectedTestResult::MD5([0xe80e16a1, 0x2d50659e, 0x413d24af, 0xea3bee05]));
142 fn test_rawvideo_ms_24() {
143 let mut dmx_reg = RegisteredDemuxers::new();
144 generic_register_all_demuxers(&mut dmx_reg);
145 let mut dec_reg = RegisteredDecoders::new();
146 generic_register_all_decoders(&mut dec_reg);
147 // sample:https://samples.mplayerhq.hu/V-codecs/Uncompressed/keve.avi
148 test_decoding("avi", "rawvideo-ms", "assets/Misc/keve.avi", Some(0), &dmx_reg,
149 &dec_reg, ExpectedTestResult::MD5([0x9514ac1f, 0x4512cc62, 0x069485ba, 0x084a1e63]));
152 fn test_rawvideo_ms_32() {
153 let mut dmx_reg = RegisteredDemuxers::new();
154 generic_register_all_demuxers(&mut dmx_reg);
155 let mut dec_reg = RegisteredDecoders::new();
156 generic_register_all_decoders(&mut dec_reg);
157 // sample: https://samples.mplayerhq.hu/V-codecs/Uncompressed/Logo-Uncompressed.zip
158 test_decoding("avi", "rawvideo-ms", "assets/Misc/VRMLuncompressed.avi", Some(0), &dmx_reg,
159 &dec_reg, ExpectedTestResult::MD5([0xf4c9d468, 0x8f42c576, 0xc8eb522a, 0x75f654b1]));