use nihav_core::codecs::*;
use nihav_core::io::byteio::*;
-use nihav_codec_support::codecs::HAMShuffler;
+use nihav_codec_support::codecs::imaadpcm::*;
use std::str::FromStr;
use std::sync::Arc;
let dst = &mut dst[dpos..][..len];
br.read_buf(dst)?;
} else {
- let val = br.read_byte()?;
- for i in 0..len {
- dst[dpos + i] = val;
+ let val1 = br.read_byte()?;
+ let val2 = br.read_byte()?;
+ for i in (0..len).step_by(2) {
+ dst[dpos + i] = val1;
+ dst[dpos + i + 1] = val2;
}
}
dpos += len;
info: NACodecInfoRef,
pal: [u8; 768],
buf: Vec<u8>,
+ framebuf: Vec<u8>,
width: usize,
height: usize,
- hams: HAMShuffler,
+ xoff: usize,
+ yoff: usize,
+ is_16bit: bool,
+ is_24bit: bool,
+ ver1: u8,
+ ver2: u8,
}
impl VMDVideoDecoder {
info: NACodecInfoRef::default(),
pal: [0; 768],
buf: Vec::new(),
+ framebuf: Vec::new(),
width: 0,
height: 0,
- hams: HAMShuffler::default(),
+ xoff: 0,
+ yoff: 0,
+ is_16bit: false,
+ is_24bit: false,
+ ver1: 0,
+ ver2: 0,
}
}
- fn decode_frame(&mut self, br: &mut ByteReader, buf: &mut NAVideoBuffer<u8>) -> DecoderResult<bool> {
- let paloff = buf.get_offset(1);
- let stride = buf.get_stride(0);
- let data = buf.get_data_mut().unwrap();
- let dst = data.as_mut_slice();
-
+ fn decode_frame(&mut self, br: &mut ByteReader) -> DecoderResult<bool> {
let frame_x = br.read_u16le()? as usize;
let frame_y = br.read_u16le()? as usize;
- let frame_l = br.read_u16le()? as usize;
+ let frame_r = br.read_u16le()? as usize;
let frame_d = br.read_u16le()? as usize;
br.read_skip(1)?;
let flags = br.read_byte()?;
- let has_pal = (flags & 0x02) != 0;
- if (frame_x == 0xFFFF) && (frame_y == 0xFFFF) && (frame_l == 0xFFFF) && (frame_d == 0xFFFF) {
+ let has_pal = (flags & 0x02) != 0 && !self.is_16bit && !self.is_24bit;
+ if (frame_x == 0xFFFF) && (frame_y == 0xFFFF) && (frame_r == 0xFFFF) && (frame_d == 0xFFFF) {
return Ok(false);
}
- validate!(frame_l >= frame_x && frame_d >= frame_y);
- validate!(frame_l < self.width && frame_d < self.height);
+ validate!(frame_x >= self.xoff && frame_y >= self.yoff);
+ validate!(frame_r >= frame_x && frame_d >= frame_y);
+ validate!(frame_r - self.xoff < self.width && frame_d - self.yoff < self.height);
if has_pal {
br.read_skip(2)?;
}
}
- let dpal = &mut dst[paloff..][..768];
- dpal.copy_from_slice(&self.pal[0..]);
-
if br.left() == 0 { return Ok(false); }
- let w = frame_l + 1 - frame_x;
+ let bpp = if (!self.is_16bit && !self.is_24bit) || self.ver1 < 2 {
+ 1
+ } else if self.is_16bit {
+ 2
+ } else {
+ 3
+ };
+ let w = (frame_r + 1 - frame_x) * bpp;
let h = frame_d + 1 - frame_y;
- let dpos = frame_x + frame_y * stride;
+ let stride = self.width;
+ let dpos = (frame_x - self.xoff) * bpp + (frame_y - self.yoff) * stride;
let method = br.read_byte()?;
let is_intra;
lz_unpack(br, &mut self.buf)?;
let mut mr = MemoryReader::new_read(&self.buf);
let mut buf_br = ByteReader::new(&mut mr);
- is_intra = decode_frame_data(&mut buf_br, dst, dpos, stride, w, h, method & 0x7F)?;
+ is_intra = decode_frame_data(&mut buf_br, &mut self.framebuf, dpos, stride, w, h, method & 0x7F)?;
} else {
- is_intra = decode_frame_data(br, dst, dpos, stride, w, h, method & 0x7F)?;
+ is_intra = decode_frame_data(br, &mut self.framebuf, dpos, stride, w, h, method & 0x7F)?;
}
Ok(is_intra && frame_x == 0 && frame_y == 0 && w == self.width && h == self.height)
}
}
+const RGB555_FORMAT: NAPixelFormaton = NAPixelFormaton { model: ColorModel::RGB(RGBSubmodel::RGB), components: 3,
+ comp_info: [
+ Some(NAPixelChromaton{ h_ss: 0, v_ss: 0, packed: true, depth: 5, shift: 10, comp_offs: 0, next_elem: 2 }),
+ Some(NAPixelChromaton{ h_ss: 0, v_ss: 0, packed: true, depth: 5, shift: 5, comp_offs: 1, next_elem: 2 }),
+ Some(NAPixelChromaton{ h_ss: 0, v_ss: 0, packed: true, depth: 5, shift: 0, comp_offs: 2, next_elem: 2 }),
+ None, None],
+ elem_size: 2, be: false, alpha: false, palette: false };
+
impl NADecoder for VMDVideoDecoder {
fn init(&mut self, _supp: &mut NADecoderSupport, info: NACodecInfoRef) -> DecoderResult<()> {
if let NACodecTypeInfo::Video(vinfo) = info.get_properties() {
self.width = vinfo.get_width();
self.height = vinfo.get_height();
- let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(self.width, self.height, false, PAL8_FORMAT));
- self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();
validate!(info.get_extradata().is_some());
if let Some(ref edata) = info.get_extradata() {
validate!(edata.len() == 0x330);
let unp_size = read_u32le(&edata[800..])? as usize;
- validate!(unp_size < self.width * self.height * 3 + 64); // just for sanity
+ validate!(unp_size < self.width * self.height * 4 + 64); // just for sanity
self.buf.resize(unp_size, 0);
for i in 0..768 {
let el = edata[28 + i];
self.pal[i] = (el << 2) | (el >> 4);
}
+ self.xoff = read_u16le(&edata[8..])? as usize;
+ self.yoff = read_u16le(&edata[10..])? as usize;
+ self.ver1 = edata[2];
+ self.ver2 = edata[4];
+ } else {
+ unreachable!();
}
+ let (disp_width, fmt) = if self.ver2 < 5 {
+ (self.width, PAL8_FORMAT)
+ } else if self.ver2 < 13 {
+ self.is_24bit = true;
+ if self.ver1 >= 2 {
+ self.width *= 3;
+ }
+ validate!(self.width % 3 == 0);
+ (self.width / 3, RGB24_FORMAT)
+ } else {
+ self.is_16bit = true;
+ if self.ver1 >= 2 {
+ self.width *= 2;
+ }
+ (self.width / 2, RGB555_FORMAT)
+ };
+ self.framebuf = vec!(0; self.width * self.height);
+
+ let myinfo = NACodecTypeInfo::Video(NAVideoInfo::new(disp_width, self.height, false, fmt));
+ self.info = NACodecInfo::new_ref(info.get_name(), myinfo, info.get_extradata()).into_ref();
Ok(())
} else {
let mut mr = MemoryReader::new_read(&src);
let mut br = ByteReader::new(&mut mr);
- let mut buf;
- let bufret = self.hams.clone_ref();
- if let Some(bbuf) = bufret {
- buf = bbuf;
+ let is_intra = self.decode_frame(&mut br)?;
+
+ let bufinfo = alloc_video_buffer(self.info.get_properties().get_video_info().unwrap(), 4)?;
+ let videobuf;
+ if !self.is_16bit {
+ let mut buf = bufinfo.get_vbuf().unwrap();
+ let stride = buf.get_stride(0);
+ let paloff = buf.get_offset(1);
+ let data = buf.get_data_mut().unwrap();
+ for (inrow, outrow) in self.framebuf.chunks(self.width).zip(data.chunks_mut(stride)) {
+ outrow[..self.width].copy_from_slice(inrow);
+ }
+ if !self.is_24bit {
+ data[paloff..][..768].copy_from_slice(&self.pal);
+ }
+ videobuf = if !self.is_24bit { NABufferType::Video(buf) } else { NABufferType::VideoPacked(buf) };
} else {
- let bufinfo = alloc_video_buffer(self.info.get_properties().get_video_info().unwrap(), 4)?;
- buf = bufinfo.get_vbuf().unwrap();
- self.hams.add_frame(buf);
- buf = self.hams.get_output_frame().unwrap();
+ let mut buf = bufinfo.get_vbuf16().unwrap();
+ let stride = buf.get_stride(0);
+ let data = buf.get_data_mut().unwrap();
+ for (inrow, outrow) in self.framebuf.chunks(self.width).zip(data.chunks_mut(stride)) {
+ for i in (0..self.width).step_by(2) {
+ outrow[i >> 1] = read_u16le(&inrow[i..])?;
+ }
+ }
+ videobuf = NABufferType::Video16(buf);
}
- let is_intra = self.decode_frame(&mut br, &mut buf)?;
-
- let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), NABufferType::Video(buf));
+ let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), videobuf);
frm.set_keyframe(is_intra);
frm.set_frame_type(if is_intra { FrameType::I } else { FrameType::P });
Ok(frm.into_ref())
}
fn flush(&mut self) {
- self.hams.clear();
}
}
+impl NAOptionHandler for VMDVideoDecoder {
+ fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
+ fn set_options(&mut self, _options: &[NAOption]) { }
+ fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
+}
+
pub fn get_decoder_video() -> Box<dyn NADecoder + Send> {
Box::new(VMDVideoDecoder::new())
U8,
DPCM,
StereoDPCM,
+ ADPCM,
}
struct VMDAudioDecoder {
pred + i32::from(SOL_AUD_STEPS16[(val & 0x7F) as usize])
}
}
- fn cvt_u8(val: u8) -> u8 {
- if val < 128 {
- 127 - val
- } else {
- val
- }
- }
}
impl NADecoder for VMDAudioDecoder {
let edata = info.get_extradata();
let flags = if let Some(ref buf) = edata {
validate!(buf.len() >= 2);
- (buf[0] as u16) | ((buf[1] as u16) << 8)
+ u16::from(buf[0]) | (u16::from(buf[1]) << 8)
} else {
0
};
self.is_odd = (channels == 2) && ((self.blk_size & 1) != 0);
}
} else {
- fmt = SND_S16P_FORMAT;
- self.blk_size = (ainfo.get_block_len() + 1) * channels;
self.blk_align = ainfo.get_block_len();
- self.mode = VMDAudioMode::DPCM;
+ if (flags & 0x10) == 0 {
+ fmt = SND_S16P_FORMAT;
+ self.blk_size = (ainfo.get_block_len() + 1) * channels;
+ self.mode = VMDAudioMode::DPCM;
+ } else {
+ fmt = SND_S16_FORMAT;
+ self.blk_size = (ainfo.get_block_len() * channels + 1) / 2 + 3 * channels;
+ self.mode = VMDAudioMode::ADPCM;
+ }
};
self.ainfo = NAAudioInfo::new(ainfo.get_sample_rate(), ainfo.get_channels(), fmt, ainfo.get_block_len());
- self.info = info.replace_info(NACodecTypeInfo::Audio(self.ainfo.clone()));
+ self.info = info.replace_info(NACodecTypeInfo::Audio(self.ainfo));
self.chmap = NAChannelMap::from_str(if channels == 1 { "C" } else { "L,R" }).unwrap();
Ok(())
} else {
Err(DecoderError::InvalidData)
}
}
+ #[allow(clippy::identity_op)]
+ #[allow(clippy::cognitive_complexity)]
fn decode(&mut self, _supp: &mut NADecoderSupport, pkt: &NAPacket) -> DecoderResult<NAFrameRef> {
let info = pkt.get_stream().get_info();
if let NACodecTypeInfo::Audio(_) = info.get_properties() {
VMDAudioMode::DPCM => {
let mut adata = abuf.get_abuf_i16().unwrap();
let off1 = adata.get_offset(1);
- let mut dst = adata.get_data_mut().unwrap();
- self.decode_16bit(&mut dst, off1, &mut br, nblocks, mask)?;
+ let dst = adata.get_data_mut().unwrap();
+ self.decode_16bit(dst, off1, &mut br, nblocks, mask)?;
},
VMDAudioMode::U8 => {
let mut adata = abuf.get_abuf_u8().unwrap();
dst[doff + i] = br.read_byte()?;
}
} else {
- for i in 0..self.blk_size {
- let val = Self::cvt_u8(br.read_byte()?);
- dst[doff + i] = val;
- }
+ unreachable!();
}
doff += self.blk_align * channels;
mask >>= 1;
}
self.ch = ch;
},
+ VMDAudioMode::ADPCM => {
+ let mut adata = abuf.get_abuf_i16().unwrap();
+ let dst = adata.get_data_mut().unwrap();
+ let mut doff = 0;
+ if self.chmap.num_channels() == 1 {
+ let mut mask = mask;
+ let mut ima = IMAState::new();
+ for _ in 0..nblocks {
+ if (mask & 1) != 0 {
+ for i in 0..self.blk_align {
+ dst[doff + i] = 0;
+ }
+ doff += self.blk_align;
+ mask >>= 1;
+ continue;
+ }
+ let pred = br.read_u16le()? as i16;
+ let step = br.read_byte()?;
+ validate!((step as usize) < IMA_STEP_TABLE.len());
+ ima.reset(pred, step);
+ let mut b = 0;
+ for i in 0..self.blk_align {
+ if (i & 1) == 0 {
+ b = br.read_byte()?;
+ dst[doff] = ima.expand_sample(b >> 4);
+ } else {
+ dst[doff] = ima.expand_sample(b & 0xF);
+ }
+ doff += 1;
+ }
+ mask >>= 1;
+ }
+ } else {
+ let mut mask = mask;
+ let mut ima1 = IMAState::new();
+ let mut ima2 = IMAState::new();
+ for _ in 0..nblocks {
+ if (mask & 1) != 0 {
+ for i in 0..self.blk_align * 2 {
+ dst[doff + i] = 0;
+ }
+ doff += self.blk_align * 2;
+ mask >>= 1;
+ continue;
+ }
+ let pred1 = br.read_u16le()? as i16;
+ let pred2 = br.read_u16le()? as i16;
+ let step1 = br.read_byte()?;
+ let step2 = br.read_byte()?;
+ validate!((step1 as usize) < IMA_STEP_TABLE.len());
+ validate!((step2 as usize) < IMA_STEP_TABLE.len());
+ ima1.reset(pred1, step1);
+ ima2.reset(pred2, step2);
+ for _ in 0..self.blk_align {
+ let b = br.read_byte()?;
+ dst[doff] = ima1.expand_sample(b >> 4);
+ doff += 1;
+ dst[doff] = ima2.expand_sample(b & 0xF);
+ doff += 1;
+ }
+ mask >>= 1;
+ }
+ }
+ },
};
let mut frm = NAFrame::new_from_pkt(pkt, self.info.clone(), abuf);
}
}
+impl NAOptionHandler for VMDAudioDecoder {
+ fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
+ fn set_options(&mut self, _options: &[NAOption]) { }
+ fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
+}
+
pub fn get_decoder_audio() -> Box<dyn NADecoder + Send> {
Box::new(VMDAudioDecoder::new())
}
use nihav_core::codecs::RegisteredDecoders;
use nihav_core::demuxers::RegisteredDemuxers;
use nihav_codec_support::test::dec_video::*;
- use crate::game_register_all_codecs;
+ use crate::game_register_all_decoders;
use crate::game_register_all_demuxers;
+ // samples from https://samples.mplayerhq.hu/game-formats/sierra-vmd/ and various games
#[test]
fn test_vmd_video() {
let mut dmx_reg = RegisteredDemuxers::new();
game_register_all_demuxers(&mut dmx_reg);
let mut dec_reg = RegisteredDecoders::new();
- game_register_all_codecs(&mut dec_reg);
+ game_register_all_decoders(&mut dec_reg);
+
+ test_decoding("vmd", "vmd-video", "assets/Game/2832.VMD", Some(10), &dmx_reg, &dec_reg,
+ ExpectedTestResult::MD5Frames(vec![
+ [0xd29e0214, 0xf38ad154, 0xccbd381f, 0x3de1109c],
+ [0x904074eb, 0x202b1d6f, 0xe3f68538, 0xf0db641c],
+ [0x9c8b1b6c, 0xe205b8dc, 0xbfb07406, 0x993ace41],
+ [0x71ce4220, 0x8747fd05, 0x854dd86d, 0x2664cde5],
+ [0x3bc65fa4, 0xebb95292, 0xe0a0fea6, 0x0acfdea1],
+ [0x33982045, 0x8d11b69b, 0xac254a75, 0x63896a21],
+ [0xa667db33, 0x90e122d3, 0x2243da15, 0xcc4bffd2],
+ [0x518621c1, 0xb91412bc, 0x12312869, 0x141ef647],
+ [0x3069977e, 0x68fd3fa0, 0x2bfdb00d, 0x1e694684],
+ [0x246c12aa, 0x15137fb0, 0xa4b0fc3e, 0x626a2676],
+ [0x72cce7e3, 0x98506d04, 0xd4d8bbaf, 0x3cc5e32d]]));
+ }
+ #[test]
+ fn test_vmd_video_16bpp() {
+ let mut dmx_reg = RegisteredDemuxers::new();
+ game_register_all_demuxers(&mut dmx_reg);
+ let mut dec_reg = RegisteredDecoders::new();
+ game_register_all_decoders(&mut dec_reg);
+
+ test_decoding("vmd", "vmd-video", "assets/Game/HLP1000.VMD", Some(10), &dmx_reg, &dec_reg,
+ ExpectedTestResult::MD5Frames(vec![
+ [0x03d77d51, 0x8670ae24, 0x86184cc8, 0x9c928700],
+ [0xf4796f1b, 0x0f75a120, 0x62056509, 0xc83f1a2c],
+ [0xd9e6db4d, 0x7af82082, 0xac6a335c, 0x19b8438f],
+ [0x03d77d51, 0x8670ae24, 0x86184cc8, 0x9c928700],
+ [0xd9e6db4d, 0x7af82082, 0xac6a335c, 0x19b8438f],
+ [0xf4796f1b, 0x0f75a120, 0x62056509, 0xc83f1a2c]]));
+ }
+ #[test]
+ fn test_vmd_video_24bpp() {
+ let mut dmx_reg = RegisteredDemuxers::new();
+ game_register_all_demuxers(&mut dmx_reg);
+ let mut dec_reg = RegisteredDecoders::new();
+ game_register_all_decoders(&mut dec_reg);
+
+ test_decoding("vmd", "vmd-video", "assets/Game/02C.VMD", None, &dmx_reg, &dec_reg,
+ ExpectedTestResult::MD5([0xb580782c, 0xd7fb98c0, 0xaf9b83cc, 0xaea0846b]));
+ }
+ #[test]
+ fn test_vmd_audio_u8() {
+ let mut dmx_reg = RegisteredDemuxers::new();
+ game_register_all_demuxers(&mut dmx_reg);
+ let mut dec_reg = RegisteredDecoders::new();
+ game_register_all_decoders(&mut dec_reg);
+
+ test_decoding("vmd", "vmd-audio", "assets/Game/1491.VMD", None, &dmx_reg, &dec_reg,
+ ExpectedTestResult::MD5([0x75037601, 0xbc7b3976, 0x6e1c948b, 0xf05a3d6c]));
+ }
+ #[test]
+ fn test_vmd_audio_s16_old() {
+ let mut dmx_reg = RegisteredDemuxers::new();
+ game_register_all_demuxers(&mut dmx_reg);
+ let mut dec_reg = RegisteredDecoders::new();
+ game_register_all_decoders(&mut dec_reg);
+
+ test_decoding("vmd", "vmd-audio", "assets/Game/2832.VMD", None, &dmx_reg, &dec_reg,
+ ExpectedTestResult::MD5([0x32dcdf0e, 0xee058684, 0x43ed5bf1, 0x2ff18b5a]));
+ }
+ #[test]
+ fn test_vmd_audio_s16_new() {
+ let mut dmx_reg = RegisteredDemuxers::new();
+ game_register_all_demuxers(&mut dmx_reg);
+ let mut dec_reg = RegisteredDecoders::new();
+ game_register_all_decoders(&mut dec_reg);
-// let file = "assets/Game/1491.VMD";
- let file = "assets/Game/128.vmd";
- test_file_decoding("vmd", file, Some(10), true, false, None/*Some("vmd")*/, &dmx_reg, &dec_reg);
+ test_decoding("vmd", "vmd-audio", "assets/Game/1000.VMD", None, &dmx_reg, &dec_reg,
+ ExpectedTestResult::MD5([0xc36215d3, 0x96530a80, 0x89f1fa8e, 0x49da302b]));
}
#[test]
- fn test_vmd_audio() {
+ fn test_vmd_audio_ima_adpcm() {
let mut dmx_reg = RegisteredDemuxers::new();
game_register_all_demuxers(&mut dmx_reg);
let mut dec_reg = RegisteredDecoders::new();
- game_register_all_codecs(&mut dec_reg);
+ game_register_all_decoders(&mut dec_reg);
-// let file = "assets/Game/1491.VMD";
- let file = "assets/Game/128.vmd";
-// let file = "assets/Game/1000.VMD";
-// let file = "assets/Game/235.VMD";
- test_decode_audio("vmd", file, None, None/*Some("vmd")*/, &dmx_reg, &dec_reg);
+ test_decoding("vmd", "vmd-audio", "assets/Game/HLP1000.VMD", None, &dmx_reg, &dec_reg,
+ ExpectedTestResult::MD5([0x76a00405, 0xe4e5378d, 0x495b2a68, 0x4dffe042]));
}
}