1 use nihav_core::frame::*;
2 use nihav_core::demuxers::*;
4 struct BMVDemuxer<'a> {
5 src: &'a mut ByteReader<'a>,
10 pkt_buf: Vec<NAPacket>,
13 impl<'a> DemuxCore<'a> for BMVDemuxer<'a> {
14 #[allow(unused_variables)]
15 fn open(&mut self, strmgr: &mut StreamManager, _seek_index: &mut SeekIndex) -> DemuxerResult<()> {
16 let src = &mut self.src;
18 let vhdr = NAVideoInfo::new(640, 429, false, PAL8_FORMAT);
19 let vci = NACodecTypeInfo::Video(vhdr);
20 let vinfo = NACodecInfo::new("bmv-video", vci, None);
21 self.vid_id = strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, 1, 12)).unwrap();
23 let ahdr = NAAudioInfo::new(22050, 2, SND_S16_FORMAT, 1);
24 let ainfo = NACodecInfo::new("bmv-audio", NACodecTypeInfo::Audio(ahdr), None);
25 self.aud_id = strmgr.add_stream(NAStream::new(StreamType::Audio, 1, ainfo, 1, 22050)).unwrap();
32 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
33 if !self.pkt_buf.is_empty() {
34 return Ok(self.pkt_buf.pop().unwrap());
38 let ctype = self.src.read_byte()?;
39 if ctype == 0 { // NOP chunk
42 if ctype == 1 { return Err(DemuxerError::EOF); }
43 let size = self.src.read_u24le()? as usize;
46 if (ctype & 0x20) != 0 {
47 let nblocks = self.src.peek_byte()?;
48 asize = (nblocks as usize) * 65 + 1;
49 validate!(asize < size);
50 let str = strmgr.get_stream(self.aud_id).unwrap();
51 let (tb_num, tb_den) = str.get_timebase();
52 let ts = NATimeInfo::new(Some(self.apos), None, None, tb_num, tb_den);
53 let apkt = self.src.read_packet(str, ts, false, asize)?;
54 self.apos += u64::from(nblocks) * 32;
55 self.pkt_buf.push(apkt);
59 let mut buf: Vec<u8> = vec![0; size - asize + 1];
61 self.src.read_buf(&mut buf[1..])?;
63 let str = strmgr.get_stream(self.vid_id).unwrap();
64 let (tb_num, tb_den) = str.get_timebase();
65 let ts = NATimeInfo::new(Some(self.vpos), None, None, tb_num, tb_den);
66 let pkt = NAPacket::new(str, ts, (ctype & 3) == 3, buf);
73 fn seek(&mut self, _time: NATimePoint, _seek_index: &SeekIndex) -> DemuxerResult<()> {
74 Err(DemuxerError::NotPossible)
78 impl<'a> NAOptionHandler for BMVDemuxer<'a> {
79 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
80 fn set_options(&mut self, _options: &[NAOption]) { }
81 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
84 impl<'a> BMVDemuxer<'a> {
85 fn new(io: &'a mut ByteReader<'a>) -> Self {
92 pkt_buf: Vec::with_capacity(1),
97 pub struct BMVDemuxerCreator { }
99 impl DemuxerCreator for BMVDemuxerCreator {
100 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
101 Box::new(BMVDemuxer::new(br))
103 fn get_name(&self) -> &'static str { "bmv" }
106 struct BMV3Demuxer<'a> {
107 src: &'a mut ByteReader<'a>,
114 pkt_buf: Vec<NAPacket>,
117 impl<'a> DemuxCore<'a> for BMV3Demuxer<'a> {
118 #[allow(unused_variables)]
119 #[allow(clippy::cast_lossless)]
120 fn open(&mut self, strmgr: &mut StreamManager, _seek_index: &mut SeekIndex) -> DemuxerResult<()> {
121 let src = &mut self.src;
123 let mut magic = [0u8; 4];
124 src.read_buf(&mut magic)?;
125 validate!(&magic[0..] == b"BMVi");
126 let size = src.read_u32le()?;
127 validate!(size == 24);
128 let _slot_size = src.read_u32le()? as usize;
129 let nframes = src.read_u32le()? as usize;
130 let _prefetch_slots = src.read_u16le()?;
131 let _cache_size = src.read_u16le()?;
132 let fps = src.read_u16le()? as u32;
133 let audio_size = src.read_u16le()? as usize;
134 let audio_blob_size = src.read_u16le()? as usize;
135 let _audio_id = src.read_byte()?;
136 let _video_id = src.read_byte()?;
137 let width = src.read_u16le()? as usize;
138 let height = src.read_u16le()? as usize;
139 validate!(nframes > 0);
140 validate!((width > 0) && (width <= 640));
141 validate!((height > 0) && (height <= 432));
142 validate!((audio_size > audio_blob_size) && (audio_blob_size > 0) && (audio_size % audio_blob_size == 0));
143 let mut dta = [0u8; 4];
144 src.read_buf(&mut dta)?;
145 validate!(&dta[0..] == b"DATA");
146 let data_size = src.read_u32le()? as usize;
147 validate!(data_size > 0);
148 self.asize = audio_size;
149 self.ablob = audio_blob_size;
151 let vhdr = NAVideoInfo::new(width, height, false, RGB565_FORMAT);
152 let vci = NACodecTypeInfo::Video(vhdr);
153 let vinfo = NACodecInfo::new("bmv3-video", vci, None);
154 self.vid_id = strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, 256, fps)).unwrap();
156 let ahdr = NAAudioInfo::new(22050, 2, SND_S16_FORMAT, audio_blob_size);
157 let ainfo = NACodecInfo::new("bmv3-audio", NACodecTypeInfo::Audio(ahdr), None);
158 self.aud_id = strmgr.add_stream(NAStream::new(StreamType::Audio, 1, ainfo, 1, 22050)).unwrap();
165 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
166 if !self.pkt_buf.is_empty() {
167 return Ok(self.pkt_buf.pop().unwrap());
171 let ctype = self.src.read_byte()?;
172 if ctype == 0 { // NOP chunk
175 if ctype == 1 { return Err(DemuxerError::EOF); }
176 let size = self.src.read_u24le()? as usize;
177 if size == 0 { continue; }
179 if (ctype & 0x20) != 0 {
180 if (ctype & 0x40) != 0 {
181 asize = self.asize - self.ablob;
185 validate!(asize <= size);
186 let mut buf: Vec<u8> = vec![0; asize + 1];
187 buf[0] = (self.src.tell() & 1) as u8;
188 self.src.read_buf(&mut buf[1..])?;
190 let str = strmgr.get_stream(self.aud_id).unwrap();
191 let (tb_num, tb_den) = str.get_timebase();
192 let ts = NATimeInfo::new(Some(self.apos), None, None, tb_num, tb_den);
193 let apkt = NAPacket::new(str, ts, false, buf);
195 self.apos += (asize as u64) / 41 * 32;
196 self.pkt_buf.push(apkt);
201 if !self.pkt_buf.is_empty() {
202 return Ok(self.pkt_buf.pop().unwrap());
207 let mut buf: Vec<u8> = vec![0; size - asize + 1];
209 self.src.read_buf(&mut buf[1..])?;
211 let str = strmgr.get_stream(self.vid_id).unwrap();
212 let (tb_num, tb_den) = str.get_timebase();
213 let ts = NATimeInfo::new(Some(self.vpos), None, None, tb_num, tb_den);
214 let pkt = NAPacket::new(str, ts, (ctype & 3) == 3, buf);
221 fn seek(&mut self, _time: NATimePoint, _seek_index: &SeekIndex) -> DemuxerResult<()> {
222 Err(DemuxerError::NotPossible)
226 impl<'a> NAOptionHandler for BMV3Demuxer<'a> {
227 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
228 fn set_options(&mut self, _options: &[NAOption]) { }
229 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
232 impl<'a> BMV3Demuxer<'a> {
233 fn new(io: &'a mut ByteReader<'a>) -> Self {
242 pkt_buf: Vec::with_capacity(1),
247 pub struct BMV3DemuxerCreator { }
249 impl DemuxerCreator for BMV3DemuxerCreator {
250 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
251 Box::new(BMV3Demuxer::new(br))
253 fn get_name(&self) -> &'static str { "bmv3" }
262 fn test_bmv_demux() {
263 let mut file = File::open("assets/Game/DW2-MOUSE.BMV").unwrap();
264 let mut fr = FileReader::new_read(&mut file);
265 let mut br = ByteReader::new(&mut fr);
266 let mut dmx = BMVDemuxer::new(&mut br);
267 let mut sm = StreamManager::new();
268 let mut si = SeekIndex::new();
269 dmx.open(&mut sm, &mut si).unwrap();
271 let pktres = dmx.get_frame(&mut sm);
272 if let Err(e) = pktres {
273 if (e as i32) == (DemuxerError::EOF as i32) { break; }
276 let pkt = pktres.unwrap();
277 println!("Got {}", pkt);
281 fn test_bmv3_demux() {
282 let mut file = File::open("assets/Game/DW3-Loffnote.bmv").unwrap();
283 let mut fr = FileReader::new_read(&mut file);
284 let mut br = ByteReader::new(&mut fr);
285 let mut dmx = BMV3Demuxer::new(&mut br);
286 let mut sm = StreamManager::new();
287 let mut si = SeekIndex::new();
288 dmx.open(&mut sm, &mut si).unwrap();
290 let pktres = dmx.get_frame(&mut sm);
291 if let Err(e) = pktres {
292 if (e as i32) == (DemuxerError::EOF as i32) { break; }
295 let pkt = pktres.unwrap();
296 println!("Got {}", pkt);