1 use nihav_core::frame::*;
2 use nihav_core::demuxers::*;
6 src: &'a mut ByteReader<'a>,
17 impl<'a> DemuxCore<'a> for QDemuxer<'a> {
18 #[allow(unused_variables)]
19 fn open(&mut self, strmgr: &mut StreamManager, _seek_index: &mut SeekIndex) -> DemuxerResult<()> {
20 let src = &mut self.src;
22 let mut hdr = [0; 22];
23 src.read_buf(&mut hdr)?;
24 validate!(hdr[0] == 0x39);
25 validate!(hdr[1] == 0x68);
27 validate!(version >= 3 && version <= 7);
28 let mut width = read_u16le(&hdr[4..])? as usize;
29 let mut height = read_u16le(&hdr[6..])? as usize;
31 width *= hdr[8] as usize;
32 height *= hdr[9] as usize;
34 validate!(width > 0 && width <= 800);
35 validate!(height > 0 && height <= 600);
37 self.nframes = read_u16le(&hdr[10..])? as usize;
38 validate!(self.nframes > 0);
39 let fps = if hdr[16] == 0 { 5 } else { hdr[16] as u32 };
40 self.duration = (self.nframes as u64) * 1000 / u64::from(fps);
41 let asize = if version > 3 {
45 let vhdr = NAVideoInfo::new(width, height, false, PAL8_FORMAT);
46 let vci = NACodecTypeInfo::Video(vhdr);
47 let vinfo = NACodecInfo::new("legend-q-video", vci, Some(hdr.to_vec()));
48 self.v_id = strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, 1, fps, self.nframes as u64));
50 let ntype = self.src.peek_byte()?;
52 let _ = self.src.read_u16le()?;
53 let size = self.src.read_u32le()? as usize;
54 validate!(size >= 44);
55 let mut buf = vec![0; size];
56 self.src.read_buf(&mut buf)?;
57 let arate = read_u32le(&buf[24..])?;
59 let channels = buf[22];
60 let abits = buf[34] as usize;
61 validate!(abits == 8 || abits == 16);
62 self.bps = (channels as usize) * abits / 8;
63 let bsize = read_u16le(&buf[32..])? as usize;
64 let ahdr = NAAudioInfo::new(arate, channels as u8, if abits == 16 { SND_S16_FORMAT } else { SND_U8_FORMAT }, bsize);
65 let ainfo = NACodecInfo::new("pcm", NACodecTypeInfo::Audio(ahdr), None);
66 self.a_id = strmgr.add_stream(NAStream::new(StreamType::Audio, 1, ainfo, 1, arate, 2));
72 self.side_data.clear();
76 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
78 let ctype = self.src.read_u16le()?;
79 let size = self.src.read_u32le()? as usize;
81 0xFFFF => return Err(DemuxerError::EOF),
83 if let Some(a_id) = self.a_id {
84 let stream = strmgr.get_stream(a_id).unwrap();
85 let ts = stream.make_ts(Some(self.apts), None, None);
86 self.apts += (size / self.bps) as u64;
87 return self.src.read_packet(stream, ts, true, size);
89 self.src.read_skip(size)?;
93 validate!(size <= 768);
94 let cur_len = self.side_data.len();
95 self.side_data.resize(cur_len + size + 6, 0);
96 self.side_data[cur_len] = ctype as u8;
97 write_u32le(&mut self.side_data[cur_len + 2..], size as u32)?;
98 self.src.read_buf(&mut self.side_data[cur_len + 6..])?;
101 validate!(self.v_id.is_some());
102 let stream = strmgr.get_stream(self.v_id.unwrap_or(0)).unwrap();
103 let ts = stream.make_ts(Some(self.vpts), None, None);
106 let cur_len = self.side_data.len();
107 self.side_data.resize(cur_len + size + 6, 0);
108 self.side_data[cur_len] = ctype as u8;
109 self.side_data[cur_len] = ctype as u8;
110 write_u32le(&mut self.side_data[cur_len + 2..], size as u32)?;
111 if let Err(err) = self.src.read_buf(&mut self.side_data[cur_len + 6..]) {
112 self.side_data.truncate(cur_len);
113 return Err(err.into());
115 let mut buf = Vec::new();
116 std::mem::swap(&mut buf, &mut self.side_data);
117 return Ok(NAPacket::new(stream, ts, self.vpts == 1, buf));
120 validate!(size <= 256);
121 let cur_len = self.side_data.len();
122 self.side_data.resize(cur_len + size + 6, 0);
123 self.side_data[cur_len] = ctype as u8;
124 write_u32le(&mut self.side_data[cur_len + 2..], size as u32)?;
125 self.src.read_buf(&mut self.side_data[cur_len + 6..])?;
128 self.side_data.push(ctype as u8);
129 self.side_data.push(0);
130 self.side_data.push(0);
131 self.side_data.push(0);
132 self.side_data.push(0);
133 self.side_data.push(0);
135 8 => return Err(DemuxerError::InvalidData), //should be handled before main loop
136 9 => { // first part of interlaced frame
137 let cur_len = self.side_data.len();
138 self.side_data.resize(cur_len + size + 6, 0);
139 self.side_data[cur_len] = ctype as u8;
140 write_u32le(&mut self.side_data[cur_len + 2..], size as u32)?;
141 self.src.read_buf(&mut self.side_data[cur_len + 6..])?;
144 self.src.read_skip(size)?;
150 fn seek(&mut self, _time: NATimePoint, _seek_index: &SeekIndex) -> DemuxerResult<()> {
151 Err(DemuxerError::NotPossible)
153 fn get_duration(&self) -> u64 { self.duration }
155 impl<'a> NAOptionHandler for QDemuxer<'a> {
156 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
157 fn set_options(&mut self, _options: &[NAOption]) { }
158 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
160 impl<'a> QDemuxer<'a> {
161 fn new(io: &'a mut ByteReader<'a>) -> Self {
171 side_data: Vec::with_capacity(256 + 6),
176 pub struct QDemuxerCreator { }
178 impl DemuxerCreator for QDemuxerCreator {
179 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
180 Box::new(QDemuxer::new(br))
182 fn get_name(&self) -> &'static str { "legend-q" }
190 // samples from Death Gate, Mission Critical and Shannara games
192 fn test_q_demux_v3() {
193 let mut file = File::open("assets/Game/dgate101.q").unwrap();
194 let mut fr = FileReader::new_read(&mut file);
195 let mut br = ByteReader::new(&mut fr);
196 let mut dmx = QDemuxer::new(&mut br);
197 let mut sm = StreamManager::new();
198 let mut si = SeekIndex::new();
199 dmx.open(&mut sm, &mut si).unwrap();
201 let pktres = dmx.get_frame(&mut sm);
202 if let Err(e) = pktres {
203 if (e as i32) == (DemuxerError::EOF as i32) { break; }
206 let pkt = pktres.unwrap();
207 println!("Got {}", pkt);
212 fn test_q_demux_v4() {
213 let mut file = File::open("assets/Game/1425A5.Q").unwrap();
214 let mut fr = FileReader::new_read(&mut file);
215 let mut br = ByteReader::new(&mut fr);
216 let mut dmx = QDemuxer::new(&mut br);
217 let mut sm = StreamManager::new();
218 let mut si = SeekIndex::new();
219 dmx.open(&mut sm, &mut si).unwrap();
221 let pktres = dmx.get_frame(&mut sm);
222 if let Err(e) = pktres {
223 if (e as i32) == (DemuxerError::EOF as i32) { break; }
226 let pkt = pktres.unwrap();
227 println!("Got {}", pkt);
232 fn test_q_demux_v5() {
233 let mut file = File::open("assets/Game/mc703.q").unwrap();
234 let mut fr = FileReader::new_read(&mut file);
235 let mut br = ByteReader::new(&mut fr);
236 let mut dmx = QDemuxer::new(&mut br);
237 let mut sm = StreamManager::new();
238 let mut si = SeekIndex::new();
239 dmx.open(&mut sm, &mut si).unwrap();
241 let pktres = dmx.get_frame(&mut sm);
242 if let Err(e) = pktres {
243 if (e as i32) == (DemuxerError::EOF as i32) { break; }
246 let pkt = pktres.unwrap();
247 println!("Got {}", pkt);