nihav-game/src/demuxers/bmv.rs

   1 use nihav_core::frame::*;
   2 use nihav_core::demuxers::*;
   3
   4 struct BMVDemuxer<'a> {
   5     src:        &'a mut ByteReader<'a>,
   6     vid_id:     usize,
   7     aud_id:     usize,
   8     vpos:       u64,
   9     apos:       u64,
  10     pkt_buf:    Vec<NAPacket>,
  11 }
  12
  13 impl<'a> DemuxCore<'a> for BMVDemuxer<'a> {
  14     fn open(&mut self, strmgr: &mut StreamManager, _seek_index: &mut SeekIndex) -> DemuxerResult<()> {
  15         let vhdr = NAVideoInfo::new(640, 429, false, PAL8_FORMAT);
  16         let vci = NACodecTypeInfo::Video(vhdr);
  17         let vinfo = NACodecInfo::new("bmv-video", vci, None);
  18         self.vid_id = strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, 1, 12, 0)).unwrap();
  19
  20         let ahdr = NAAudioInfo::new(22050, 2, SND_S16_FORMAT, 1);
  21         let ainfo = NACodecInfo::new("bmv-audio", NACodecTypeInfo::Audio(ahdr), None);
  22         self.aud_id = strmgr.add_stream(NAStream::new(StreamType::Audio, 1, ainfo, 1, 22050, 0)).unwrap();
  23
  24         self.vpos       = 0;
  25         self.apos       = 0;
  26         Ok(())
  27     }
  28
  29     fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
  30         if !self.pkt_buf.is_empty() {
  31             return Ok(self.pkt_buf.pop().unwrap());
  32         }
  33
  34         loop {
  35             let ctype                   = self.src.read_byte()?;
  36             if ctype == 0 { // NOP chunk
  37                 continue;
  38             }
  39             if ctype == 1 { return Err(DemuxerError::EOF); }
  40             let size                    = self.src.read_u24le()? as usize;
  41             validate!(size > 0);
  42             let asize;
  43             if (ctype & 0x20) != 0 {
  44                 let nblocks             = self.src.peek_byte()?;
  45                 asize = (nblocks as usize) * 65 + 1;
  46                 validate!(asize < size);
  47                 let stream = strmgr.get_stream(self.aud_id).unwrap();
  48                 let ts = stream.make_ts(Some(self.apos), None, None);
  49                 let apkt = self.src.read_packet(stream, ts, false, asize)?;
  50                 self.apos += u64::from(nblocks) * 32;
  51                 self.pkt_buf.push(apkt);
  52             } else {
  53                 asize = 0;
  54             }
  55             let mut buf: Vec<u8> = vec![0; size - asize + 1];
  56             buf[0] = ctype;
  57             self.src.read_buf(&mut buf[1..])?;
  58
  59             let stream = strmgr.get_stream(self.vid_id).unwrap();
  60             let ts = stream.make_ts(Some(self.vpos), None, None);
  61             let pkt = NAPacket::new(stream, ts, (ctype & 3) == 3, buf);
  62
  63             self.vpos += 1;
  64             return Ok(pkt);
  65         }
  66     }
  67
  68     fn seek(&mut self, _time: NATimePoint, _seek_index: &SeekIndex) -> DemuxerResult<()> {
  69         Err(DemuxerError::NotPossible)
  70     }
  71     fn get_duration(&self) -> u64 { 0 }
  72 }
  73
  74 impl<'a> NAOptionHandler for BMVDemuxer<'a> {
  75     fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
  76     fn set_options(&mut self, _options: &[NAOption]) { }
  77     fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
  78 }
  79
  80 impl<'a> BMVDemuxer<'a> {
  81     fn new(io: &'a mut ByteReader<'a>) -> Self {
  82         Self {
  83             src:        io,
  84             vid_id:     0,
  85             aud_id:     0,
  86             vpos:       0,
  87             apos:       0,
  88             pkt_buf:    Vec::with_capacity(1),
  89         }
  90     }
  91 }
  92
  93 pub struct BMVDemuxerCreator { }
  94
  95 impl DemuxerCreator for BMVDemuxerCreator {
  96     fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
  97         Box::new(BMVDemuxer::new(br))
  98     }
  99     fn get_name(&self) -> &'static str { "bmv" }
 100 }
 101
 102 struct BMV3Demuxer<'a> {
 103     src:        &'a mut ByteReader<'a>,
 104     vid_id:     usize,
 105     aud_id:     usize,
 106     vpos:       u64,
 107     apos:       u64,
 108     asize:      usize,
 109     ablob:      usize,
 110     pkt_buf:    Vec<NAPacket>,
 111 }
 112
 113 impl<'a> DemuxCore<'a> for BMV3Demuxer<'a> {
 114     fn open(&mut self, strmgr: &mut StreamManager, _seek_index: &mut SeekIndex) -> DemuxerResult<()> {
 115         let src = &mut self.src;
 116
 117         let mut magic = [0u8; 4];
 118                                             src.read_buf(&mut magic)?;
 119         validate!(&magic[0..] == b"BMVi");
 120         let size                            = src.read_u32le()?;
 121         validate!(size == 24);
 122         let _slot_size                      = src.read_u32le()? as usize;
 123         let nframes                         = src.read_u32le()? as usize;
 124         let _prefetch_slots                 = src.read_u16le()?;
 125         let _cache_size                     = src.read_u16le()?;
 126         let fps                             = src.read_u16le()? as u32;
 127         let audio_size                      = src.read_u16le()? as usize;
 128         let audio_blob_size                 = src.read_u16le()? as usize;
 129         let _audio_id                       = src.read_byte()?;
 130         let _video_id                       = src.read_byte()?;
 131         let width                           = src.read_u16le()? as usize;
 132         let height                          = src.read_u16le()? as usize;
 133         validate!(nframes > 0);
 134         validate!((width > 0) && (width <= 640));
 135         validate!((height > 0) && (height <= 432));
 136         validate!((audio_size > audio_blob_size) && (audio_blob_size > 0) && (audio_size % audio_blob_size == 0));
 137         let mut dta = [0u8; 4];
 138                                             src.read_buf(&mut dta)?;
 139         validate!(&dta[0..] == b"DATA");
 140         let data_size                       = src.read_u32le()? as usize;
 141         validate!(data_size > 0);
 142         self.asize = audio_size;
 143         self.ablob = audio_blob_size;
 144
 145         let vhdr = NAVideoInfo::new(width, height, false, RGB565_FORMAT);
 146         let vci = NACodecTypeInfo::Video(vhdr);
 147         let vinfo = NACodecInfo::new("bmv3-video", vci, None);
 148         self.vid_id = strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, 256, fps, nframes as u64)).unwrap();
 149
 150         let ahdr = NAAudioInfo::new(22050, 2, SND_S16_FORMAT, audio_blob_size);
 151         let ainfo = NACodecInfo::new("bmv3-audio", NACodecTypeInfo::Audio(ahdr), None);
 152         self.aud_id = strmgr.add_stream(NAStream::new(StreamType::Audio, 1, ainfo, 1, 22050, 0)).unwrap();
 153
 154         self.vpos       = 0;
 155         self.apos       = 0;
 156         Ok(())
 157     }
 158
 159     fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
 160         if !self.pkt_buf.is_empty() {
 161             return Ok(self.pkt_buf.pop().unwrap());
 162         }
 163
 164         loop {
 165             let ctype                       = self.src.read_byte()?;
 166             if ctype == 0 { // NOP chunk
 167                 continue;
 168             }
 169             if ctype == 1 { return Err(DemuxerError::EOF); }
 170             let size                    = self.src.read_u24le()? as usize;
 171             if size == 0 { continue; }
 172             let asize;
 173             if (ctype & 0x20) != 0 {
 174                 if (ctype & 0x40) != 0 {
 175                     asize = self.asize - self.ablob;
 176                 } else {
 177                     asize = self.asize;
 178                 }
 179                 validate!(asize <= size);
 180                 let mut buf: Vec<u8> = vec![0; asize + 1];
 181                 buf[0] = (self.src.tell() & 1) as u8;
 182                 self.src.read_buf(&mut buf[1..])?;
 183
 184                 let stream = strmgr.get_stream(self.aud_id).unwrap();
 185                 let ts = stream.make_ts(Some(self.apos), None, None);
 186                 let apkt = NAPacket::new(stream, ts, false, buf);
 187
 188                 self.apos += (asize as u64) / 41 * 32;
 189                 self.pkt_buf.push(apkt);
 190             } else {
 191                 asize = 0;
 192             }
 193             if size == asize {
 194                 if !self.pkt_buf.is_empty() {
 195                     return Ok(self.pkt_buf.pop().unwrap());
 196                 } else {
 197                     continue;
 198                 }
 199             }
 200             let mut buf: Vec<u8> = vec![0; size - asize + 1];
 201             buf[0] = ctype;
 202             self.src.read_buf(&mut buf[1..])?;
 203
 204             let stream = strmgr.get_stream(self.vid_id).unwrap();
 205             let ts = stream.make_ts(Some(self.vpos), None, None);
 206             let pkt = NAPacket::new(stream, ts, (ctype & 3) == 3, buf);
 207
 208             self.vpos += 1;
 209             return Ok(pkt);
 210         }
 211     }
 212
 213     fn seek(&mut self, _time: NATimePoint, _seek_index: &SeekIndex) -> DemuxerResult<()> {
 214         Err(DemuxerError::NotPossible)
 215     }
 216     fn get_duration(&self) -> u64 { 0 }
 217 }
 218
 219 impl<'a> NAOptionHandler for BMV3Demuxer<'a> {
 220     fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
 221     fn set_options(&mut self, _options: &[NAOption]) { }
 222     fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
 223 }
 224
 225 impl<'a> BMV3Demuxer<'a> {
 226     fn new(io: &'a mut ByteReader<'a>) -> Self {
 227         Self {
 228             src:        io,
 229             vid_id:     0,
 230             aud_id:     0,
 231             vpos:       0,
 232             apos:       0,
 233             asize:      0,
 234             ablob:      0,
 235             pkt_buf:    Vec::with_capacity(1),
 236         }
 237     }
 238 }
 239
 240 pub struct BMV3DemuxerCreator { }
 241
 242 impl DemuxerCreator for BMV3DemuxerCreator {
 243     fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
 244         Box::new(BMV3Demuxer::new(br))
 245     }
 246     fn get_name(&self) -> &'static str { "bmv3" }
 247 }
 248
 249 #[cfg(test)]
 250 mod test {
 251     use super::*;
 252     use std::fs::File;
 253
 254     // samples from https://samples.mplayerhq.hu/game-formats/bmv
 255     #[test]
 256     fn test_bmv_demux() {
 257         let mut file = File::open("assets/Game/DW2-MOUSE.BMV").unwrap();
 258         let mut fr = FileReader::new_read(&mut file);
 259         let mut br = ByteReader::new(&mut fr);
 260         let mut dmx = BMVDemuxer::new(&mut br);
 261         let mut sm = StreamManager::new();
 262         let mut si = SeekIndex::new();
 263         dmx.open(&mut sm, &mut si).unwrap();
 264         loop {
 265             let pktres = dmx.get_frame(&mut sm);
 266             if let Err(e) = pktres {
 267                 if (e as i32) == (DemuxerError::EOF as i32) { break; }
 268                 panic!("error");
 269             }
 270             let pkt = pktres.unwrap();
 271             println!("Got {}", pkt);
 272         }
 273     }
 274     #[test]
 275     fn test_bmv3_demux() {
 276         let mut file = File::open("assets/Game/DW3-Loffnote.bmv").unwrap();
 277         let mut fr = FileReader::new_read(&mut file);
 278         let mut br = ByteReader::new(&mut fr);
 279         let mut dmx = BMV3Demuxer::new(&mut br);
 280         let mut sm = StreamManager::new();
 281         let mut si = SeekIndex::new();
 282         dmx.open(&mut sm, &mut si).unwrap();
 283         loop {
 284             let pktres = dmx.get_frame(&mut sm);
 285             if let Err(e) = pktres {
 286                 if (e as i32) == (DemuxerError::EOF as i32) { break; }
 287                 panic!("error");
 288             }
 289             let pkt = pktres.unwrap();
 290             println!("Got {}", pkt);
 291         }
 292     }
 293 }