1 use nihav_core::formats;
2 use nihav_core::codecs::*;
3 use nihav_core::io::byteio::*;
6 const FRAME_W: usize = 640;
7 const FRAME_H: usize = 429;
9 const BMV_INTRA: u8 = 0x03;
10 const BMV_SCROLL: u8 = 0x04;
11 const BMV_PAL: u8 = 0x08;
12 const BMV_COMMAND: u8 = 0x10;
13 const BMV_PRINT: u8 = 0x80;
15 struct BMVReader<'a> {
23 impl<'a> BMVReader<'a> {
24 fn new(src: &'a [u8], fwd: bool) -> Self {
25 let pos = if fwd { 0 } else { src.len() - 1 };
26 Self { src, pos, fwd, nibble: 0, saved: false }
28 fn advance(&mut self) {
30 if self.pos < self.src.len() - 1 { self.pos += 1; }
32 if self.pos > 0 { self.pos -= 1; }
35 fn get_byte(&mut self) -> u8 {
36 let ret = self.src[self.pos];
40 fn get_nibble(&mut self) -> u8 {
45 let val = self.get_byte();
47 self.nibble = val >> 4;
53 struct BMVWriter<'a> {
60 impl<'a> BMVWriter<'a> {
61 fn new(data: &'a mut [u8], fwd: bool, off: isize) -> Self {
62 let pos = if fwd { 0 } else { data.len() - 1 };
63 Self { data, pos, fwd, off }
65 fn is_at_end(&self) -> bool {
67 self.pos == self.data.len() - 1
72 fn advance(&mut self) {
74 if self.pos < self.data.len() - 1 { self.pos += 1; }
76 if self.pos > 0 { self.pos -= 1; }
79 fn put_byte(&mut self, val: u8) {
80 self.data[self.pos] = val;
83 fn copy(&mut self, len: usize) {
85 let saddr = (self.pos as isize) + self.off;
86 if saddr < 0 { continue; }
87 self.data[self.pos] = self.data[saddr as usize];
91 fn repeat(&mut self, len: usize) {
92 let last = if self.fwd { self.data[self.pos - 1] } else { self.data[self.pos + 1] };
99 struct BMVVideoDecoder {
100 info: NACodecInfoRef,
102 frame: [u8; FRAME_W * FRAME_H],
105 impl BMVVideoDecoder {
108 info: NACodecInfoRef::default(), pal: [0; 768], frame: [0; FRAME_W * FRAME_H],
111 fn decode_frame(&mut self, src: &[u8], bufinfo: &mut NABufferType, line: i16) -> DecoderResult<()> {
112 let bufo = bufinfo.get_vbuf();
113 let mut buf = bufo.unwrap();
114 let paloff = buf.get_offset(1);
115 let stride = buf.get_stride(0);
116 let data = buf.get_data_mut().unwrap();
117 let dst = data.as_mut_slice();
119 let fwd = (line <= -640) || (line >= 0);
121 let mut br = BMVReader::new(src, fwd);
122 let mut bw = BMVWriter::new(&mut self.frame, fwd, line as isize);
124 while !bw.is_at_end() {
128 let nib = br.get_nibble() as usize;
129 if (nib & 0xC) != 0 {
144 let len = (val >> 1) - 1;
148 2 => for _ in 0..len { bw.put_byte(br.get_byte()); },
154 for y in 0..FRAME_H {
155 for x in 0..FRAME_W {
156 dst[y * stride + x] = self.frame[y * FRAME_W + x];
160 let dpal = &mut dst[paloff..][..768];
161 dpal.copy_from_slice(&self.pal[0..]);
167 impl NADecoder for BMVVideoDecoder {
168 fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
169 if let NACodecTypeInfo::Video(_vinfo) = info.get_properties() {
170 let fmt = NAPixelFormaton::new(ColorModel::RGB(RGBSubmodel::RGB),
171 Some(NAPixelChromaton::new(0, 0, true, 8, 0, 0, 3)),
172 Some(NAPixelChromaton::new(0, 0, true, 8, 0, 1, 3)),
173 Some(NAPixelChromaton::new(0, 0, true, 8, 0, 2, 3)),
175 FORMATON_FLAG_PALETTE, 3);
176 let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(FRAME_W, FRAME_H, false, fmt));
177 self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();
181 Err(DecoderError::InvalidData)
184 fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
185 let src = pkt.get_buffer();
186 validate!(src.len() > 1);
188 let mut mr = MemoryReader::new_read(&src);
189 let mut br = ByteReader::new(&mut mr);
190 let flags = br.read_byte()?;
192 if (flags & BMV_COMMAND) != 0 {
193 let size = if (flags & BMV_PRINT) != 0 { 8 } else { 10 };
196 if (flags & BMV_PAL) != 0 {
197 br.read_buf(&mut self.pal)?;
200 if (flags & BMV_SCROLL) != 0 {
201 line = br.read_u16le()? as i16;
202 } else if (flags & BMV_INTRA) == BMV_INTRA {
207 let pos = br.tell() as usize;
209 let mut bufinfo = alloc_video_buffer(self.info.get_properties().get_video_info().unwrap(), 0)?;
211 self.decode_frame(&src[pos..], &mut bufinfo, line)?;
213 let is_intra = (flags & BMV_INTRA) == BMV_INTRA;
214 let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), bufinfo);
215 frm.set_keyframe(is_intra);
216 frm.set_frame_type(if is_intra { FrameType::I } else { FrameType::P });
219 fn flush(&mut self) {
223 impl NAOptionHandler for BMVVideoDecoder {
224 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
225 fn set_options(&mut self, _options: &[NAOption]) { }
226 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
230 pub fn get_decoder_video() -> Box<dyn NADecoder + Send> {
231 Box::new(BMVVideoDecoder::new())
234 struct BMVAudioDecoder {
239 const BMV_AUD_SCALES: [i32; 16] = [ 16512, 8256, 4128, 2064, 1032, 516, 258, 192, 129, 88, 64, 56, 48, 40, 36, 32 ];
241 impl BMVAudioDecoder {
244 ainfo: NAAudioInfo::new(0, 1, formats::SND_S16P_FORMAT, 0),
245 chmap: NAChannelMap::new(),
250 fn scale_sample(samp: u8, scale: i32) -> i16 {
251 let val = (i32::from(samp as i8) * scale) >> 5;
254 } else if val > 32767 {
261 impl NADecoder for BMVAudioDecoder {
262 fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
263 if let NACodecTypeInfo::Audio(ainfo) = info.get_properties() {
264 self.ainfo = NAAudioInfo::new(ainfo.get_sample_rate(), ainfo.get_channels(), formats::SND_S16P_FORMAT, 32);
265 self.chmap = NAChannelMap::from_str("L,R").unwrap();
268 Err(DecoderError::InvalidData)
271 fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
272 let info = pkt.get_stream().get_info();
273 if let NACodecTypeInfo::Audio(_) = info.get_properties() {
274 let pktbuf = pkt.get_buffer();
275 validate!(pktbuf.len() > 1);
276 let nblocks = pktbuf[0] as usize;
277 validate!(pktbuf.len() == 1 + 65 * nblocks);
278 let samples = nblocks * 32;
279 let abuf = alloc_audio_buffer(self.ainfo, samples, self.chmap.clone())?;
280 let mut adata = abuf.get_abuf_i16().unwrap();
281 let off1 = adata.get_offset(1);
282 let dst = adata.get_data_mut().unwrap();
283 let psrc = &pktbuf[1..];
284 for (n, src) in psrc.chunks_exact(65).enumerate() {
285 let code = src[0].rotate_right(1);
286 let scale0 = BMV_AUD_SCALES[(code & 0xF) as usize];
287 let scale1 = BMV_AUD_SCALES[(code >> 4) as usize];
289 let aoff1 = aoff0 + off1;
290 let data = &src[1..];
291 for (i, samp) in data.chunks_exact(2).enumerate() {
292 dst[aoff0 + i] = scale_sample(samp[0], scale0);
293 dst[aoff1 + i] = scale_sample(samp[1], scale1);
296 let mut frm = NAFrame::new_from_pkt(pkt, info, abuf);
297 frm.set_duration(Some(samples as u64));
298 frm.set_keyframe(false);
301 Err(DecoderError::InvalidData)
304 fn flush(&mut self) {
308 impl NAOptionHandler for BMVAudioDecoder {
309 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
310 fn set_options(&mut self, _options: &[NAOption]) { }
311 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
314 pub fn get_decoder_audio() -> Box<dyn NADecoder + Send> {
315 Box::new(BMVAudioDecoder::new())
320 use nihav_core::codecs::RegisteredDecoders;
321 use nihav_core::demuxers::RegisteredDemuxers;
322 use nihav_codec_support::test::dec_video::*;
323 use crate::game_register_all_codecs;
324 use crate::game_register_all_demuxers;
326 fn test_bmv_video() {
327 let mut dmx_reg = RegisteredDemuxers::new();
328 game_register_all_demuxers(&mut dmx_reg);
329 let mut dec_reg = RegisteredDecoders::new();
330 game_register_all_codecs(&mut dec_reg);
332 test_decoding("bmv", "bmv-video", "assets/Game/WILDCAT.BMV", Some(40), &dmx_reg, &dec_reg,
333 ExpectedTestResult::MD5([0x9e91bb16, 0xc1edafc9, 0x4ef3171f, 0x0f3f6181]));
336 fn test_bmv_audio() {
337 let mut dmx_reg = RegisteredDemuxers::new();
338 game_register_all_demuxers(&mut dmx_reg);
339 let mut dec_reg = RegisteredDecoders::new();
340 game_register_all_codecs(&mut dec_reg);
342 test_decoding("bmv", "bmv-audio", "assets/Game/PERFECT.BMV", None, &dmx_reg, &dec_reg,
343 ExpectedTestResult::MD5([0x90b9ace4, 0x5fc19938, 0x7f534560, 0x32589cdf]));