]> git.nihav.org Git - nihav.git/blame - nihav-game/src/demuxers/bmv.rs
avimux: do not record palette change chunks in OpenDML index
[nihav.git] / nihav-game / src / demuxers / bmv.rs
CommitLineData
9067c1f8
KS
1use nihav_core::frame::*;
2use nihav_core::demuxers::*;
3
4struct 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
13impl<'a> DemuxCore<'a> for BMVDemuxer<'a> {
33b5a8f0 14 fn open(&mut self, strmgr: &mut StreamManager, _seek_index: &mut SeekIndex) -> DemuxerResult<()> {
9067c1f8
KS
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);
a480a0de 18 self.vid_id = strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, 1, 12, 0)).unwrap();
9067c1f8
KS
19
20 let ahdr = NAAudioInfo::new(22050, 2, SND_S16_FORMAT, 1);
21 let ainfo = NACodecInfo::new("bmv-audio", NACodecTypeInfo::Audio(ahdr), None);
a480a0de 22 self.aud_id = strmgr.add_stream(NAStream::new(StreamType::Audio, 1, ainfo, 1, 22050, 0)).unwrap();
9067c1f8
KS
23
24 self.vpos = 0;
25 self.apos = 0;
26 Ok(())
27 }
28
29 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
e69b1148 30 if !self.pkt_buf.is_empty() {
9067c1f8
KS
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);
817e4872
KS
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)?;
e69b1148 50 self.apos += u64::from(nblocks) * 32;
9067c1f8
KS
51 self.pkt_buf.push(apkt);
52 } else {
53 asize = 0;
54 }
e69b1148 55 let mut buf: Vec<u8> = vec![0; size - asize + 1];
9067c1f8
KS
56 buf[0] = ctype;
57 self.src.read_buf(&mut buf[1..])?;
58
817e4872
KS
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);
9067c1f8
KS
62
63 self.vpos += 1;
64 return Ok(pkt);
65 }
66 }
67
24d99894 68 fn seek(&mut self, _time: NATimePoint, _seek_index: &SeekIndex) -> DemuxerResult<()> {
33b5a8f0 69 Err(DemuxerError::NotPossible)
9067c1f8 70 }
a480a0de 71 fn get_duration(&self) -> u64 { 0 }
9067c1f8
KS
72}
73
787b8d03
KS
74impl<'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
9067c1f8
KS
80impl<'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
93pub struct BMVDemuxerCreator { }
94
95impl DemuxerCreator for BMVDemuxerCreator {
6011e201 96 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
9067c1f8
KS
97 Box::new(BMVDemuxer::new(br))
98 }
99 fn get_name(&self) -> &'static str { "bmv" }
100}
101
ecda1cc1
KS
102struct 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
113impl<'a> DemuxCore<'a> for BMV3Demuxer<'a> {
33b5a8f0 114 fn open(&mut self, strmgr: &mut StreamManager, _seek_index: &mut SeekIndex) -> DemuxerResult<()> {
ecda1cc1
KS
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);
a480a0de 148 self.vid_id = strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, 256, fps, nframes as u64)).unwrap();
ecda1cc1
KS
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);
a480a0de 152 self.aud_id = strmgr.add_stream(NAStream::new(StreamType::Audio, 1, ainfo, 1, 22050, 0)).unwrap();
ecda1cc1
KS
153
154 self.vpos = 0;
155 self.apos = 0;
156 Ok(())
157 }
158
159 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
e69b1148 160 if !self.pkt_buf.is_empty() {
ecda1cc1
KS
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);
e69b1148 180 let mut buf: Vec<u8> = vec![0; asize + 1];
ecda1cc1
KS
181 buf[0] = (self.src.tell() & 1) as u8;
182 self.src.read_buf(&mut buf[1..])?;
183
817e4872
KS
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);
ecda1cc1
KS
187
188 self.apos += (asize as u64) / 41 * 32;
189 self.pkt_buf.push(apkt);
190 } else {
191 asize = 0;
192 }
726973ae 193 if size == asize {
e69b1148 194 if !self.pkt_buf.is_empty() {
726973ae
KS
195 return Ok(self.pkt_buf.pop().unwrap());
196 } else {
197 continue;
198 }
199 }
e69b1148 200 let mut buf: Vec<u8> = vec![0; size - asize + 1];
ecda1cc1
KS
201 buf[0] = ctype;
202 self.src.read_buf(&mut buf[1..])?;
203
817e4872
KS
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);
ecda1cc1
KS
207
208 self.vpos += 1;
209 return Ok(pkt);
210 }
211 }
212
24d99894 213 fn seek(&mut self, _time: NATimePoint, _seek_index: &SeekIndex) -> DemuxerResult<()> {
33b5a8f0 214 Err(DemuxerError::NotPossible)
ecda1cc1 215 }
a480a0de 216 fn get_duration(&self) -> u64 { 0 }
ecda1cc1
KS
217}
218
787b8d03
KS
219impl<'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
ecda1cc1
KS
225impl<'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
240pub struct BMV3DemuxerCreator { }
241
242impl DemuxerCreator for BMV3DemuxerCreator {
6011e201 243 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
ecda1cc1
KS
244 Box::new(BMV3Demuxer::new(br))
245 }
246 fn get_name(&self) -> &'static str { "bmv3" }
247}
248
9067c1f8
KS
249#[cfg(test)]
250mod test {
251 use super::*;
252 use std::fs::File;
253
886cde48 254 // samples from https://samples.mplayerhq.hu/game-formats/bmv
9067c1f8
KS
255 #[test]
256 fn test_bmv_demux() {
1678d59a 257 let mut file = File::open("assets/Game/DW2-MOUSE.BMV").unwrap();
9067c1f8
KS
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();
caf0f37e
KS
262 let mut si = SeekIndex::new();
263 dmx.open(&mut sm, &mut si).unwrap();
9067c1f8
KS
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 }
ecda1cc1
KS
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();
caf0f37e
KS
281 let mut si = SeekIndex::new();
282 dmx.open(&mut sm, &mut si).unwrap();
ecda1cc1
KS
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 }
9067c1f8 293}