rmdec: factor out common code and add IVR demuxing support
[nihav.git] / src / demuxers / realmedia.rs
CommitLineData
ce52b3b5
KS
1use super::*;
2use super::DemuxerError::*;
3//use io::byteio::*;
4//use frame::*;
5use formats::*;
6use std::io::SeekFrom;
7use std::mem;
418bd0b8 8use std::fmt;
ce52b3b5
KS
9
10macro_rules! mktag {
11 ($a:expr, $b:expr, $c:expr, $d:expr) => ({
12 (($a as u32) << 24) | (($b as u32) << 16) | (($c as u32) << 8) | ($d as u32)
13 });
14 ($arr:expr) => ({
15 (($arr[0] as u32) << 24) | (($arr[1] as u32) << 16) | (($arr[2] as u32) << 8) | ($arr[3] as u32)
16 });
17}
18
02cb1e03
KS
19const RM_SIPRO_BLOCK_SIZES: [usize; 4] = [ 29, 19, 37, 20 ];
20const RM_SIPRO_SWAPS: [[u8; 2]; 38] = [
21 [ 0, 63 ], [ 1, 22 ], [ 2, 44 ], [ 3, 90 ],
22 [ 5, 81 ], [ 7, 31 ], [ 8, 86 ], [ 9, 58 ],
23 [ 10, 36 ], [ 12, 68 ], [ 13, 39 ], [ 14, 73 ],
24 [ 15, 53 ], [ 16, 69 ], [ 17, 57 ], [ 19, 88 ],
25 [ 20, 34 ], [ 21, 71 ], [ 24, 46 ], [ 25, 94 ],
26 [ 26, 54 ], [ 28, 75 ], [ 29, 50 ], [ 32, 70 ],
27 [ 33, 92 ], [ 35, 74 ], [ 38, 85 ], [ 40, 56 ],
28 [ 42, 87 ], [ 43, 65 ], [ 45, 59 ], [ 48, 79 ],
29 [ 49, 93 ], [ 51, 89 ], [ 55, 95 ], [ 61, 76 ],
30 [ 67, 83 ], [ 77, 80 ]
31];
32
ce52b3b5
KS
33struct RMVideoStream {
34 frame: Vec<u8>,
35 hdr_size: usize,
36 frame_size: usize,
37 frame_pos: usize,
38}
39
40impl RMVideoStream {
41 fn new() -> Self {
42 RMVideoStream {
43 frame: Vec::new(),
44 hdr_size: 0,
45 frame_size: 0,
46 frame_pos: 0,
47 }
48 }
49 fn flush(&mut self) {
50 self.frame.truncate(0);
51 self.frame_size = 0;
52 self.frame_pos = 0;
53 }
54 fn start_slice(&mut self, num_slices: usize, frame_size: usize, data: &[u8]) {
55 self.hdr_size = num_slices * 8 + 1;
56 self.frame.resize(frame_size + self.hdr_size, 0);
57 self.frame[0] = (num_slices - 1) as u8;
58 self.frame_pos = 0;
59 self.add_slice(1, data);
60 }
61 fn add_slice(&mut self, slice_no: usize, data: &[u8]) {
62 self.write_slice_info(slice_no);
9037cf6b 63 let dslice = &mut self.frame[self.hdr_size + self.frame_pos..][..data.len()];
ce52b3b5
KS
64 dslice.copy_from_slice(data);
65 self.frame_pos += data.len();
66 }
67 fn write_slice_info(&mut self, slice_no: usize) {
68 let off = 1 + (slice_no - 1) * 8;
69 self.frame[off + 0] = 0;
70 self.frame[off + 1] = 0;
71 self.frame[off + 2] = 0;
72 self.frame[off + 3] = 1;
73 self.frame[off + 4] = (self.frame_pos >> 24) as u8;
74 self.frame[off + 5] = (self.frame_pos >> 16) as u8;
75 self.frame[off + 6] = (self.frame_pos >> 8) as u8;
76 self.frame[off + 7] = (self.frame_pos >> 0) as u8;
77 }
78 fn get_frame_data(&mut self) -> Vec<u8> {
79 let mut v: Vec<u8> = Vec::new();
80 mem::swap(&mut v, &mut self.frame);
81 self.flush();
82 v
83 }
84}
85
b1ef37ee 86#[allow(dead_code)]
ce52b3b5
KS
87#[derive(Clone,Copy,PartialEq)]
88enum Deinterleaver {
89 None,
c19ce782 90 RA28_8,
ce52b3b5
KS
91 Generic,
92 Sipro,
93 VBR,
94}
95
b1ef37ee 96#[allow(dead_code)]
ce52b3b5
KS
97struct RMAudioStream {
98 deint: Deinterleaver,
c19ce782 99 iinfo: Option<InterleaveInfo>,
fe07b469
KS
100 buf: Vec<u8>,
101 sub_packet: usize,
ce52b3b5
KS
102}
103
c19ce782
KS
104const RM_ILEAVE_INT0: u32 = mktag!(b"Int0");
105const RM_ILEAVE_INT4: u32 = mktag!(b"Int4");
106const RM_ILEAVE_GENR: u32 = mktag!(b"genr");
107const RM_ILEAVE_SIPR: u32 = mktag!(b"sipr");
bb149b3b 108const RM_ILEAVE_VBRF: u32 = mktag!(b"vbrf");
c19ce782
KS
109const RM_ILEAVE_VBRS: u32 = mktag!(b"vbrs");
110
ce52b3b5 111impl RMAudioStream {
c19ce782
KS
112 fn new(iinfo: Option<InterleaveInfo>) -> Self {
113 let deint;
fe07b469 114 let mut buf: Vec<u8>;
c19ce782
KS
115 if let Some(info) = iinfo {
116 deint = match info.id {
117 RM_ILEAVE_INT0 => Deinterleaver::None,
118 RM_ILEAVE_INT4 => Deinterleaver::RA28_8,
119 RM_ILEAVE_GENR => Deinterleaver::Generic,
120 RM_ILEAVE_SIPR => Deinterleaver::Sipro,
bb149b3b 121 RM_ILEAVE_VBRF => Deinterleaver::VBR,
c19ce782
KS
122 RM_ILEAVE_VBRS => Deinterleaver::VBR,
123 _ => {println!("unknown deint {:X}", info.id); Deinterleaver::None },
124 };
fe07b469
KS
125 match deint {
126 Deinterleaver::None => { buf = Vec::new(); },
6669c73b
KS
127 Deinterleaver::RA28_8 |
128 Deinterleaver::Generic |
129 Deinterleaver::Sipro => {
fe07b469
KS
130 let bsize = (info.frame_size as usize) * (info.factor as usize);
131 buf = Vec::with_capacity(bsize);
132 buf.resize(bsize, 0u8);
133 },
480d951e
KS
134 Deinterleaver::VBR => {
135 buf = Vec::new();
136 },
fe07b469 137 };
c19ce782
KS
138 } else {
139 deint = Deinterleaver::None;
fe07b469
KS
140 buf = Vec::new();
141 }
142 RMAudioStream { deint: deint, iinfo: iinfo, buf: buf, sub_packet: 0 }
143 }
6669c73b 144 fn read_apackets(&mut self, queued_packets: &mut Vec<NAPacket>, src: &mut ByteReader, stream: Rc<NAStream>, ts: u32, keyframe: bool, payload_size: usize) -> DemuxerResult<NAPacket> {
fe07b469
KS
145 let (tb_num, tb_den) = stream.get_timebase();
146 let ts = NATimeInfo::new(Some(ts as u64), None, None, tb_num, tb_den);
147
148 if keyframe {
149 self.sub_packet = 0;
150 }
151 match self.deint {
6669c73b 152 Deinterleaver::None => { return src.read_packet(stream, ts, keyframe, payload_size); },
fe07b469
KS
153 Deinterleaver::RA28_8 => {
154 let iinfo = self.iinfo.unwrap();
155 let factor = iinfo.factor as usize;
156 let halffact = factor >> 1;
157 let fsize = iinfo.frame_size as usize;
158 let bsize = iinfo.block_size as usize;
159 let ppos = self.sub_packet;
160 for sb in 0..halffact {
161 let mut dst = &mut self.buf[sb * 2 * fsize + ppos * bsize..][..bsize];
162 src.read_buf(&mut dst)?;
163 }
164 self.sub_packet += 1;
165 if self.sub_packet == factor {
166 self.sub_packet = 0;
6669c73b 167 return Ok(NAPacket::new(stream, ts, true, self.buf.clone()));
fe07b469 168 } else {
6669c73b 169 return Err(DemuxerError::TryAgain);
fe07b469
KS
170 }
171 },
6669c73b
KS
172 Deinterleaver::Generic => {
173 let iinfo = self.iinfo.unwrap();
174 let factor = iinfo.factor as usize;
175 let fsize = iinfo.frame_size as usize;
176 let bsize = iinfo.block_size as usize;
177 let factor2 = fsize / bsize;
178 let ppos = self.sub_packet;
179
180 for sb in 0..factor2 {
181 let sb_pos = factor * sb + ((factor + 1) >> 1) * (ppos & 1) + (ppos >> 1);
182 let mut dst = &mut self.buf[bsize * sb_pos..][..bsize];
183 src.read_buf(&mut dst)?;
184 }
185 },
186 Deinterleaver::Sipro => {
187 let iinfo = self.iinfo.unwrap();
188 let fsize = iinfo.frame_size as usize;
189 let ppos = self.sub_packet;
190
191 let mut dst = &mut self.buf[fsize * ppos..][..fsize];
192 src.read_buf(&mut dst)?;
193 },
480d951e
KS
194 Deinterleaver::VBR => {
195 validate!(payload_size >= 5);
196 let hdrsizesize = src.read_u16be()?;
197 let num_entries = (hdrsizesize / 16) as usize;
198 validate!(payload_size >= num_entries * 3 + 2);
199 let mut sizes: Vec<usize> = Vec::with_capacity(num_entries);
200 let mut tot_size = 0;
201 for _ in 0..num_entries {
202 let sz = src.read_u16be()? as usize;
203 tot_size += sz;
204 sizes.push(sz);
205 }
206 validate!(tot_size + num_entries * 2 + 2 == payload_size);
207 let pkt_ts = NATimeInfo::new(None, None, None, tb_num, tb_den);
208 let mut first = true;
209 for size in sizes.iter() {
210 let cur_ts = if first { ts } else { pkt_ts };
211 first = false;
212 let pkt = src.read_packet(stream.clone(), cur_ts, true, *size)?;
213 queued_packets.push(pkt);
214 }
215 queued_packets.reverse();
216 let pkt0 = queued_packets.pop().unwrap();
217 return Ok(pkt0);
218 },
6669c73b
KS
219 };
220
221 let iinfo = self.iinfo.unwrap();
222 let factor = iinfo.factor as usize;
5dfd175c 223 let fsize = if iinfo.block_size != 0 { iinfo.block_size } else { iinfo.frame_size } as usize;
6669c73b
KS
224
225 self.sub_packet += 1;
226 if self.sub_packet < factor {
227 return Err(DemuxerError::TryAgain);
228 }
229
230 self.sub_packet = 0;
231
232 if self.deint == Deinterleaver::Sipro {
02cb1e03 233 sipro_restore(&mut self.buf, factor, fsize);
6669c73b
KS
234 }
235
236 let mut frames_iter = self.buf.chunks(fsize);
237 let pkt0 = frames_iter.next().unwrap();
238
239 let pkt_ts = NATimeInfo::new(None, None, None, tb_num, tb_den);
240 for pkts in frames_iter {
241 let pkt = NAPacket::new(stream.clone(), pkt_ts, true, pkts.to_vec());
242 queued_packets.push(pkt);
c19ce782 243 }
6669c73b
KS
244 queued_packets.reverse();
245 Ok(NAPacket::new(stream, ts, true, pkt0.to_vec()))
ce52b3b5
KS
246 }
247}
248
02cb1e03
KS
249fn sipro_restore(buf: &mut [u8], factor: usize, fsize: usize) {
250 let stride = factor * fsize * 2 / 96;
251 for i in 0..38 {
252 let mut sidx = (RM_SIPRO_SWAPS[i][0] as usize) * stride;
253 let mut didx = (RM_SIPRO_SWAPS[i][1] as usize) * stride;
254 for _ in 0..stride {
255 let in0 = buf[sidx >> 1];
256 let in1 = buf[didx >> 1];
257 let nib0 = (in0 >> ((sidx & 1) * 4)) & 0xF;
258 let nib1 = (in1 >> ((didx & 1) * 4)) & 0xF;
259
260 buf[didx >> 1] = (nib0 << (4 * (didx & 1))) | (in1 & (0xF << (4 * (!didx & 1))));
261 buf[sidx >> 1] = (nib1 << (4 * (sidx & 1))) | (in0 & (0xF << (4 * (!sidx & 1))));
262
263 sidx += 1;
264 didx += 1;
265 }
266 }
267}
268
d4df9ee1
KS
269struct SubstreamInfo {
270 id: u32,
271 map: Vec<u16>,
272 str_ids: Vec<u32>,
273}
274
275struct MLTIMapper {
276 sub_info: Vec<SubstreamInfo>,
418bd0b8 277 sstr_id: u32,
d4df9ee1
KS
278}
279
280impl MLTIMapper {
281 fn new() -> Self {
282 MLTIMapper {
283 sub_info: Vec::new(),
418bd0b8 284 sstr_id: 0x10000,
d4df9ee1
KS
285 }
286 }
287 fn add_stream(&mut self, stream_no: u32) {
288 let ssinfo = SubstreamInfo { id: stream_no, map: Vec::new(), str_ids: Vec::new() };
289 self.sub_info.push(ssinfo);
290 }
418bd0b8
KS
291 fn get_substream_no(&self) -> u32 {
292 self.sstr_id - 1
293 }
d4df9ee1
KS
294 fn find_idx(&self, stream_no: u32) -> Option<usize> {
295 self.sub_info.iter().position(|x| x.id == stream_no)
296 }
418bd0b8
KS
297 fn add_map_rule(&mut self, map_ss: u16) {
298 let idx = self.sub_info.len() - 1;
d4df9ee1
KS
299 self.sub_info[idx].map.push(map_ss);
300 }
418bd0b8
KS
301 fn add_substream(&mut self) {
302 let idx = self.sub_info.len() - 1;
303 self.sub_info[idx].str_ids.push(self.sstr_id);
304 self.sstr_id += 1;
d4df9ee1
KS
305 }
306 fn is_mlti_stream(&self, stream_no: u32) -> bool {
307 self.find_idx(stream_no).is_some()
308 }
309 fn find_substream(&self, stream_no: u32, grp: u16) -> Option<u32> {
310 if let Some(idx) = self.find_idx(stream_no) {
311 if (grp as usize) < self.sub_info[idx].map.len() {
312 let sub_id = self.sub_info[idx].map[grp as usize] as usize;
313 if sub_id < self.sub_info[idx].str_ids.len() {
314 return Some(self.sub_info[idx].str_ids[sub_id]);
315 }
316 }
317 }
318 None
319 }
320}
321
ce52b3b5
KS
322enum RMStreamType {
323 Audio(RMAudioStream),
324 Video(RMVideoStream),
325 Logical,
326 Unknown,
327}
328
418bd0b8
KS
329struct CommonStreamData {
330 streams: Vec<RMStreamType>,
331 str_ids: Vec<u32>,
332 mlti_mapper: MLTIMapper,
333}
334
335impl CommonStreamData {
336 fn new() -> Self {
337 CommonStreamData {
338 streams: Vec::new(),
339 str_ids: Vec::new(),
340 mlti_mapper: MLTIMapper::new(),
341 }
342 }
343 fn get_stream_id(&self, str_no: u32, pkt_grp: u16) -> u32 {
344 if !self.mlti_mapper.is_mlti_stream(str_no) {
345 str_no
346 } else {
347 self.mlti_mapper.find_substream(str_no, pkt_grp).unwrap()
348 }
349 }
350 fn find_stream(&self, stream_id: u32) -> Option<usize> {
351 self.str_ids.iter().position(|x| *x == stream_id)
352 }
353}
354
ce52b3b5
KS
355struct RealMediaDemuxer<'a> {
356 src: &'a mut ByteReader<'a>,
357 data_pos: u64,
770bef6f 358 next_data: u64,
db5cc44b 359 data_ver: u16,
ce52b3b5
KS
360 num_packets: u32,
361 cur_packet: u32,
362
418bd0b8 363 str_data: CommonStreamData,
ce52b3b5
KS
364
365 queued_pkts: Vec<NAPacket>,
366 slice_buf: Vec<u8>,
367}
368
369fn find_codec_name(registry: &[(&[u8;4], &'static str)], fcc: u32) -> &'static str {
370 for &(fourcc, name) in registry {
371 if mktag!(fourcc) == fcc { return name; }
372 }
373 "unknown"
374}
375
376fn read_14or30(src: &mut ByteReader) -> DemuxerResult<(bool, u32)> {
377 let tmp = src.read_u16be()?;
378 let flag = (tmp & 0x8000) != 0;
379 if (tmp & 0x4000) == 0x4000 {
380 Ok((flag, ((tmp & 0x3FFF) as u32)))
381 } else {
382 let val = ((tmp as u32) << 16) | (src.read_u16be()? as u32);
383 Ok((flag, val & 0x3FFFFFFF))
384 }
385}
386
387fn read_video_buf(src: &mut ByteReader, stream: Rc<NAStream>, ts: u32, keyframe: bool, frame_size: usize) -> DemuxerResult<NAPacket> {
388 let size = (frame_size as usize) + 9;
389 let mut vec: Vec<u8> = Vec::with_capacity(size);
390 vec.resize(size, 0);
391 //v[0] = 0; // 1 slice
392 vec[4] = 1;
393 src.read_buf(&mut vec[9..])?;
394
395 let (tb_num, tb_den) = stream.get_timebase();
396 let ts = NATimeInfo::new(Some(ts as u64), None, None, tb_num, tb_den);
397 Ok(NAPacket::new(stream, ts, keyframe, vec))
398}
399
400fn read_multiple_frame(src: &mut ByteReader, stream: Rc<NAStream>, keyframe: bool, skip_mtype: bool) -> DemuxerResult<NAPacket> {
401 if !skip_mtype {
402 let mtype = src.read_byte()?;
403 validate!(mtype == 0xC0);
404 }
405 let (_, frame_size) = read_14or30(src)?;
406 let (_, timestamp) = read_14or30(src)?;
c9aba47e
KS
407 let _seq_no = src.read_byte()?;
408//println!(" multiple frame size {} ts {} seq {}", frame_size, timestamp, seq_no);
ce52b3b5
KS
409
410 read_video_buf(src, stream, timestamp, keyframe, frame_size as usize)
411}
412
418bd0b8
KS
413struct RMDemuxCommon {}
414
415impl RMDemuxCommon {
416 fn parse_stream_info(str_data: &mut CommonStreamData, strmgr: &mut StreamManager, stream_no: u32, edata: &Vec<u8>) -> DemuxerResult<bool> {
417 let mut is_mlti = false;
418 let mut mr = MemoryReader::new_read(edata.as_slice());
419 let mut src = ByteReader::new(&mut mr);
420 let tag = src.read_u32be()?;
421 let tag2 = src.peek_u32be()?;
422//println!("tag1 {:X} tag2 {:X}", tag, tag2);
423 if tag == mktag!('.', 'r', 'a', 0xFD) {
424 Self::parse_audio_stream(strmgr, &mut str_data.streams, stream_no, &mut src, edata.as_slice())?;
425 } else if ((tag2 == mktag!('V', 'I', 'D', 'O')) || (tag2 == mktag!('I', 'M', 'A', 'G'))) && ((tag as usize) <= edata.len()) {
426 Self::parse_video_stream(strmgr, &mut str_data.streams, stream_no, &mut src, edata.as_slice(), tag2)?;
427 } else if tag == mktag!(b"LSD:") {
428 let extradata = Some(edata.clone());
429
430 src.read_skip(4)?; //version
431 let channels = src.read_u16be()?;
432 let samp_size = src.read_u16be()?;
433 let sample_rate = src.read_u32be()?;
434
435 println!("LSD sr {}, {} ch", sample_rate, channels);
436 let soniton = NASoniton::new(samp_size as u8, SONITON_FLAG_SIGNED);
437 let ahdr = NAAudioInfo::new(sample_rate, channels as u8, soniton, 1);
438 let nainfo = NACodecInfo::new("ralf", NACodecTypeInfo::Audio(ahdr), extradata);
439 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, stream_no as u32, nainfo, 1, sample_rate));
440 if res.is_none() { return Err(MemoryError); }
441 let astr = RMAudioStream::new(None);
442 str_data.streams.push(RMStreamType::Audio(astr));
443 } else if tag == mktag!(b"MLTI") {
444 is_mlti = true;
445 let num_rules = src.read_u16be()? as usize;
446 let mut max_sub = 0;
447 str_data.mlti_mapper.add_stream(stream_no);
448 for _ in 0..num_rules {
449 let substr = src.read_u16be()?;
450 max_sub = max_sub.max(substr);
451 str_data.mlti_mapper.add_map_rule(substr);
452 }
453 let num_substreams = src.read_u16be()? as usize;
454 validate!(num_substreams > (max_sub as usize));
455 for _ in 0..num_substreams {
456 let hdr_size = src.read_u32be()? as usize;
457 validate!(hdr_size > 8);
458 let pos = src.tell() as usize;
459 src.read_skip(hdr_size)?;
460 str_data.mlti_mapper.add_substream();
461 {
462 let hdrsrc = &edata[pos..][..hdr_size];
463 let mut mr = MemoryReader::new_read(hdrsrc);
464 let mut hsrc = ByteReader::new(&mut mr);
465
466 let tag = hsrc.read_u32be()?;
467 let tag2 = hsrc.peek_u32be()?;
468 let stream_no = str_data.mlti_mapper.get_substream_no();
469//todo check that all substreams are of the same type");
470 if tag == mktag!('.', 'r', 'a', 0xFD) {
471 Self::parse_audio_stream(strmgr, &mut str_data.streams, stream_no, &mut hsrc, hdrsrc)?;
472 } else if (tag2 == mktag!('V', 'I', 'D', 'O')) && ((tag as usize) <= hdr_size) {
473 Self::parse_video_stream(strmgr, &mut str_data.streams, stream_no, &mut hsrc, hdrsrc, tag2)?;
474 } else {
475println!("unknown MLTI substream {:08X} / {:08X}", tag, tag2);
476 return Err(DemuxerError::InvalidData);
477 }
478 str_data.str_ids.push(stream_no);
479 }
480 }
481 } else {
482 str_data.streams.push(RMStreamType::Logical);
483 }
484 Ok(is_mlti)
485 }
486 fn parse_audio_stream(strmgr: &mut StreamManager, streams: &mut Vec<RMStreamType>, stream_no: u32, src: &mut ByteReader, edata_: &[u8]) -> DemuxerResult<()> {
487 let ver = src.read_u16be()?;
488 let ainfo = match ver {
489 3 => {
490 parse_aformat3(src)?
491 },
492 4 => {
493 parse_aformat4(src)?
494 },
495 5 => {
496 parse_aformat5(src)?
497 },
498 _ => {
499 println!("unknown version {}", ver);
500 return Err(DemuxerError::InvalidData);
501 },
502 };
503println!(" got ainfo {:?}", ainfo);
504 let cname = find_codec_name(RM_AUDIO_CODEC_REGISTER, ainfo.fcc);
505 let blk_size = if ainfo.fcc != mktag!(b"sipr") {
506 ainfo.granularity as usize
507 } else {
508 validate!(ainfo.flavor <= 3);
509 RM_SIPRO_BLOCK_SIZES[ainfo.flavor as usize]
510 };
511 let srate = ainfo.sample_rate;
512 let soniton = NASoniton::new(ainfo.sample_size as u8, SONITON_FLAG_SIGNED);
513 let ahdr = NAAudioInfo::new(srate, ainfo.channels as u8, soniton, blk_size);
514 let extradata = if ainfo.edata_size == 0 {
515 None
516 } else {
517 let eslice = &edata_[(src.tell() as usize)..];
518 Some(eslice.to_vec())
519 };
520 let nainfo = NACodecInfo::new(cname, NACodecTypeInfo::Audio(ahdr), extradata);
521 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, stream_no as u32, nainfo, 1, srate));
522 if res.is_none() { return Err(MemoryError); }
523
524 let astr = RMAudioStream::new(ainfo.ileave_info);
525 streams.push(RMStreamType::Audio(astr));
526 Ok(())
527 }
528#[allow(unused_variables)]
529 fn parse_video_stream(strmgr: &mut StreamManager, streams: &mut Vec<RMStreamType>, stream_no: u32, src: &mut ByteReader, edata_: &[u8], tag2: u32) -> DemuxerResult<()> {
530 src.read_skip(4)?;
531 let fcc = src.read_u32be()?;
532 let width = src.read_u16be()? as usize;
533 let height = src.read_u16be()? as usize;
534 let bpp = src.read_u16be()?;
535 let pad_w = src.read_u16be()?;
536 let pad_h = src.read_u16be()?;
537 let fps;
538 if tag2 == mktag!('V', 'I', 'D', 'O') {
539 fps = src.read_u32be()?;
540 } else {
541 fps = 0x10000;
542 }
543 let extradata: Option<Vec<u8>>;
544 if src.left() > 0 {
545 let eslice = &edata_[(src.tell() as usize)..];
546 extradata = Some(eslice.to_vec());
547 } else {
548 extradata = None;
549 }
550 let cname = find_codec_name(RM_VIDEO_CODEC_REGISTER, fcc);
551
552 let vhdr = NAVideoInfo::new(width, height, false, RGB24_FORMAT);
553 let vinfo = NACodecInfo::new(cname, NACodecTypeInfo::Video(vhdr), extradata);
554 let res = strmgr.add_stream(NAStream::new(StreamType::Video, stream_no as u32, vinfo, 0x10000, fps));
555 if res.is_none() { return Err(DemuxerError::MemoryError); }
556
557 let vstr = RMVideoStream::new();
558 streams.push(RMStreamType::Video(vstr));
559 Ok(())
560 }
561#[allow(unused_variables)]
562 fn parse_packet_payload(src: &mut ByteReader, rmstream: &mut RMStreamType, stream: Rc<NAStream>, slice_buf: &mut Vec<u8>, queued_pkts: &mut Vec<NAPacket>, keyframe: bool, ts: u32, payload_size: usize) -> DemuxerResult<NAPacket> {
563 match rmstream {
564 RMStreamType::Video(ref mut vstr) => {
565
566 let pos = src.tell();
567 let b0 = src.read_byte()?;
568 match b0 >> 6 {
569 0 => { // partial frame
570 let b1 = src.read_byte()?;
571 let hdr1 = ((b0 as u16) << 8) | (b1 as u16);
572 let num_pkts = ((hdr1 >> 7) & 0x7F) as usize;
573 let packet_num = hdr1 & 0x7F;
574 let (_, frame_size) = read_14or30(src)?;
575 let (_, off) = read_14or30(src)?;
576 let seq_no = src.read_byte()?;
577//println!(" mode 0 pkt {}/{} off {}/{} seq {}", packet_num, num_pkts, off, frame_size, seq_no);
578 let hdr_skip = (src.tell() - pos) as usize;
579
580 let slice_size = (payload_size - hdr_skip) as usize;
581 slice_buf.resize(slice_size, 0);
582 src.read_buf(slice_buf.as_mut_slice())?;
583 if packet_num == 1 {
584 vstr.start_slice(num_pkts, frame_size as usize, slice_buf.as_slice());
585 } else {
586 vstr.add_slice(packet_num as usize, slice_buf.as_slice());
587 }
588 if (packet_num as usize) < num_pkts {
589 return Err(DemuxerError::TryAgain);
590 }
591 //todo: check if full frame is received
592 let (tb_num, tb_den) = stream.get_timebase();
593 let ts = NATimeInfo::new(Some(ts as u64), None, None, tb_num, tb_den);
594 let pkt = NAPacket::new(stream, ts, keyframe, vstr.get_frame_data());
595 Ok(pkt)
596 },
597 1 => { // whole frame
598 let seq_no = src.read_byte()?;
599//println!(" mode 1 seq {}", seq_no);
600 read_video_buf(src, stream, ts, keyframe, payload_size - 2)
601 },
602 2 => { // last partial frame
603 let b1 = src.read_byte()?;
604 let hdr1 = ((b0 as u16) << 8) | (b1 as u16);
605 let num_pkts = ((hdr1 >> 7) & 0x7F) as usize;
606 let packet_num = hdr1 & 0x7F;
607 let (_, frame_size) = read_14or30(src)?;
608 let (_, tail_size) = read_14or30(src)?;
609 let seq_no = src.read_byte()?;
610//println!(" mode 2 pkt {}/{} tail {}/{} seq {}", packet_num, num_pkts, tail_size, frame_size, seq_no);
611 slice_buf.resize(tail_size as usize, 0);
612 src.read_buf(slice_buf.as_mut_slice())?;
613 if packet_num == 1 && frame_size == tail_size {
614 vstr.start_slice(num_pkts, frame_size as usize, slice_buf.as_slice());
615 } else {
616 vstr.add_slice(packet_num as usize, slice_buf.as_slice());
617 }
618
619 while src.tell() < pos + (payload_size as u64) {
620 let res = read_multiple_frame(src, stream.clone(), false, false);
621 if res.is_err() { break; }
622 queued_pkts.push(res.unwrap());
623 }
624 queued_pkts.reverse();
625 let (tb_num, tb_den) = stream.get_timebase();
626 let ts = NATimeInfo::new(Some(ts as u64), None, None, tb_num, tb_den);
627 let pkt = NAPacket::new(stream, ts, keyframe, vstr.get_frame_data());
628 Ok(pkt)
629 },
630 _ => { // multiple frames
631//println!(" mode 3");
632 let res = read_multiple_frame(src, stream.clone(), keyframe, true);
633 if res.is_err() { return res; }
634 while src.tell() < pos + (payload_size as u64) {
635 let res = read_multiple_frame(src, stream.clone(), false, false);
636 if res.is_err() { break; }
637 queued_pkts.push(res.unwrap());
638 }
639 queued_pkts.reverse();
640 res
641 },
642 }
643 },
644 RMStreamType::Audio(ref mut astr) => {
645 astr.read_apackets(queued_pkts, src, stream, ts, keyframe, payload_size)
646 },
647 _ => {
648 src.read_skip(payload_size)?;
649 Err(DemuxerError::InvalidData)
650 },
651 }
652 }
653}
654
ce52b3b5
KS
655impl<'a> DemuxCore<'a> for RealMediaDemuxer<'a> {
656 #[allow(unused_variables)]
657 fn open(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<()> {
658 self.read_header(strmgr)?;
659 Ok(())
660 }
661
662#[allow(unused_variables)]
663 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
664 if !self.queued_pkts.is_empty() {
665 let pkt = self.queued_pkts.pop().unwrap();
666 return Ok(pkt);
667 }
668 loop {
770bef6f
KS
669 if self.cur_packet >= self.num_packets {
670 if (self.next_data != 0) && (self.next_data == self.src.tell()) {
671 let res = read_chunk(self.src);
672 if let Ok((id, size, ver)) = res {
673 self.data_pos = self.src.tell();
674 self.data_ver = ver;
675 if self.parse_data_start().is_ok() {
676 continue;
677 }
678 }
679 }
680 return Err(DemuxerError::EOF);
681 }
ce52b3b5
KS
682
683 let pkt_start = self.src.tell();
7348405b 684 let ver = self.src.read_u16be()?;
ce52b3b5
KS
685 validate!(ver <= 1);
686 let len = self.src.read_u16be()? as usize;
d4df9ee1 687 let str_no = self.src.read_u16be()? as u32;
ce52b3b5 688 let ts = self.src.read_u32be()?;
d4df9ee1 689 let pkt_grp;
770bef6f 690 let flags;
ce52b3b5 691 if ver == 0 {
d4df9ee1 692 pkt_grp = self.src.read_byte()? as u16;
770bef6f 693 flags = self.src.read_byte()?;
ce52b3b5 694 } else {
7348405b 695 //asm_rule = self.src.read_u16be()?;
d4df9ee1 696 //asm_flags = self.src.read_byte()?;
ce52b3b5 697 self.src.read_skip(2)?;
d4df9ee1 698 pkt_grp = 0;
770bef6f
KS
699 self.src.read_skip(1)?;
700 flags = 0;
ce52b3b5 701 }
ce52b3b5 702 let hdr_size = self.src.tell() - pkt_start;
d4df9ee1 703//println!("packet @{:X} size {} for {} ts {} grp {} flags {:X}", pkt_start, len, str_no, ts, pkt_grp, flags);
ce52b3b5
KS
704 self.cur_packet += 1;
705
706 let payload_size = len - (hdr_size as usize);
707
418bd0b8
KS
708 let stream_id = self.str_data.get_stream_id(str_no as u32, pkt_grp);
709 let sr = self.str_data.find_stream(stream_id);
ce52b3b5 710 if sr.is_none() {
c9aba47e 711//println!("stream {} not found", str_no);
ce52b3b5
KS
712 self.src.read_skip(payload_size)?;
713 return Err(DemuxerError::InvalidData);
714 }
715 let str_id = sr.unwrap();
d4df9ee1
KS
716
717 let streamres = strmgr.get_stream_by_id(stream_id);
c9aba47e
KS
718 if streamres.is_none() {
719 self.src.read_skip(payload_size)?;
720 continue;
721 }
722 let stream = streamres.unwrap();
723//println!(" stream {}", str_id);
d4df9ee1 724 if strmgr.is_ignored_id(stream_id) {
ce52b3b5
KS
725 self.src.read_skip(payload_size)?;
726 continue;
727 }
728 //todo skip unwanted packet
729 let keyframe = (flags & KEYFRAME_FLAG) != 0;
730
418bd0b8
KS
731 let ret = RMDemuxCommon::parse_packet_payload(&mut self.src, &mut self.str_data.streams[str_id], stream, &mut self.slice_buf, &mut self.queued_pkts, keyframe, ts, payload_size);
732 if let Err(DemuxerError::TryAgain) = ret {
733 continue;
734 } else {
735 return ret;
736 }
ce52b3b5
KS
737 }
738 }
739
740 #[allow(unused_variables)]
741 fn seek(&mut self, time: u64) -> DemuxerResult<()> {
742 Err(NotImplemented)
743 }
744}
745
746fn read_chunk(src: &mut ByteReader) -> DemuxerResult<(u32, u32, u16)> {
747 let id = src.read_u32be()?;
6132225f 748if id == 0 { return Ok((0, 0, 0)); }
ce52b3b5 749 let size = src.read_u32be()?;
9f45e528
KS
750if size == 0 {
751 let ver = src.read_u16be()?;
db5cc44b 752 validate!(ver <= 2);
9f45e528
KS
753 return Ok((id, 0x0FFFFFFF, ver));
754}
ce52b3b5
KS
755 validate!(size >= 10);
756 let ver = src.read_u16be()?;
db5cc44b 757 validate!(ver <= 2);
ce52b3b5
KS
758 Ok((id, size, ver))
759}
760
c19ce782
KS
761#[derive(Clone,Copy,Debug)]
762struct InterleaveInfo {
763 id: u32,
764 factor: u16,
765 block_size: u16,
766 frame_size: u16,
767}
768
769#[derive(Clone,Copy,Debug)]
770struct RealAudioInfo {
771 fcc: u32,
772 sample_rate: u32,
773 sample_size: u16,
02cb1e03 774 flavor: u16,
c19ce782
KS
775 channels: u16,
776 channel_mask: u32,
777 granularity: u32,
778 bytes_per_minute: u32,
779 total_bytes: u32,
780 edata_size: u32,
781 ileave_info: Option<InterleaveInfo>
782}
783
784fn skip_ra_metadata(src: &mut ByteReader) -> DemuxerResult<()> {
785 let title_len = src.read_byte()? as usize;
786 src.read_skip(title_len)?;
787 let author_len = src.read_byte()? as usize;
788 src.read_skip(author_len)?;
789 let copywrong_len = src.read_byte()? as usize;
790 src.read_skip(copywrong_len)?;
791 let comment_len = src.read_byte()? as usize;
792 src.read_skip(comment_len)?;
793 Ok(())
794}
795
796#[allow(unused_variables)]
797fn parse_aformat3(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
798 let start = src.tell();
799 let header_len = src.read_u16be()?;
800 validate!(header_len >= 24);
801 let flavor = src.read_u16be()?;
802 let granularity = src.read_u32be()?;
803 let bytes_per_minute = src.read_u32be()?;
804 let total_bytes = src.read_u32be()?;
805
806 skip_ra_metadata(src)?;
807
808 let _can_copy = src.read_byte()?;
809 let fcc_len = src.read_byte()?;
810 validate!(fcc_len == 4);
811 let fcc = src.read_u32be()?;
812
813 let end = src.tell();
814 validate!(end - start <= (header_len as u64) + 2);
815
816 Ok(RealAudioInfo {
02cb1e03
KS
817 fcc: fcc, flavor: flavor,
818 sample_rate: 8000, sample_size: 16, channels: 1, channel_mask: 0,
c19ce782
KS
819 granularity: granularity, bytes_per_minute: bytes_per_minute,
820 total_bytes: total_bytes, edata_size: 0,
821 ileave_info: None,
822 })
823}
824
825#[allow(unused_variables)]
826fn parse_aformat4(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
827 let start = src.tell();
828 src.read_skip(2)?; // zeroes
829 let id = src.read_u32be()?;
830 validate!(id == mktag!(b".ra4"));
831 let data_size = src.read_u32be()?;
832 let _ver4 = src.read_u16be()?; // should be 4
833 let header_size = src.read_u32be()?;
02cb1e03 834 let flavor = src.read_u16be()?;
c19ce782
KS
835 let granularity = src.read_u32be()?;
836 let total_bytes = src.read_u32be()?;
837 let bytes_per_minute = src.read_u32be()?;
838 let _bytes_per_minute2 = src.read_u32be()?;
839 let ileave_factor = src.read_u16be()?;
840 let ileave_block_size = src.read_u16be()?;
841 let _user_data = src.read_u16be()?;
842 let sample_rate = src.read_u32be()?;
843 let sample_size = src.read_u32be()?;
844 let channels = src.read_u16be()?;
845 let interleaver_id_len = src.read_byte()?;
846 validate!(interleaver_id_len == 4);
847 let interleaver_id = src.read_u32be()?;
848 let fcc_len = src.read_byte()?;
849 validate!(fcc_len == 4);
850 let fcc = src.read_u32be()?;
851 let is_interleaved = src.read_byte()?;
852 let _can_copy = src.read_byte()?;
853 let _stream_type = src.read_byte()?;
854
855 skip_ra_metadata(src)?;
856
857 let end = src.tell();
858 validate!(end - start <= (header_size as u64) + 10);
859
860 let ileave_info = if is_interleaved != 0 {
861 Some(InterleaveInfo {
fe07b469
KS
862 id: interleaver_id, factor: ileave_factor, block_size: granularity as u16,
863 frame_size: ileave_block_size,
c19ce782
KS
864 })
865 } else {
866 None
867 };
868
869 Ok(RealAudioInfo {
02cb1e03
KS
870 fcc: fcc, flavor: flavor,
871 sample_rate: sample_rate, sample_size: sample_size as u16, channels: channels, channel_mask: 0,
c19ce782
KS
872 granularity: granularity, bytes_per_minute: bytes_per_minute,
873 total_bytes: total_bytes & 0xFFFFFF, edata_size: 0,
874 ileave_info: ileave_info,
875 })
876}
877
878#[allow(unused_variables)]
879fn parse_aformat5(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
880 let start = src.tell();
881 src.read_skip(2)?; // zeroes
882 let id = src.read_u32be()?;
db5cc44b 883 validate!((id == mktag!(b".ra5")) || (id == mktag!(b".ra4")));
c19ce782
KS
884 let data_size = src.read_u32be()?;
885 let _ver5 = src.read_u16be()?; // should be 5
886 let header_size = src.read_u32be()?;
02cb1e03 887 let flavor = src.read_u16be()?;
c19ce782
KS
888 let granularity = src.read_u32be()?;
889 let total_bytes = src.read_u32be()?;
890 let bytes_per_minute = src.read_u32be()?;
891 let _bytes_per_minute2 = src.read_u32be()?;
892 let ileave_factor = src.read_u16be()?;
c19ce782 893 let frame_size = src.read_u16be()?;
6669c73b 894 let ileave_block_size = src.read_u16be()?;
c19ce782
KS
895 let user_data = src.read_u32be()?;
896 let _sample_rate1 = src.read_u16be()?;
897 let sample_rate = src.read_u32be()?;
898 let sample_size = src.read_u32be()?;
899 let channels = src.read_u16be()?;
900 let interleaver_id = src.read_u32be()?;
901 let fcc = src.read_u32be()?;
902 let is_interleaved = src.read_byte()?;
903 let _can_copy = src.read_byte()?;
904 let _stream_type = src.read_byte()?;
905 let has_ileave_pattern = src.read_byte()?;
906 if has_ileave_pattern != 0 {
907unimplemented!("ra5 interleave pattern");
908 }
843c8c87 909 let mut edata_size = src.read_u32be()?;
c19ce782 910 let end = src.tell();
db5cc44b
KS
911 if id == mktag!(b".ra5") {
912 validate!(end - start <= (header_size as u64) + 10);
bb149b3b 913// src.read_skip(((end - start) as usize) - (header_size as usize) - 10)?;
db5cc44b
KS
914 } else {
915 validate!(end - start <= (header_size as u64) + 15);
916 }
c19ce782
KS
917
918 let ileave_info = if is_interleaved != 0 {
919 Some(InterleaveInfo {
920 id: interleaver_id, factor: ileave_factor, block_size: ileave_block_size, frame_size: frame_size,
921 })
922 } else {
923 None
924 };
843c8c87
KS
925 if (fcc == mktag!(b"raac")) || (fcc == mktag!(b"racp")) {
926 validate!(edata_size > 1);
927 edata_size -= 1;
928 src.read_skip(1)?;
929 }
c19ce782
KS
930
931 Ok(RealAudioInfo {
02cb1e03
KS
932 fcc: fcc, flavor: flavor,
933 sample_rate: sample_rate, sample_size: sample_size as u16, channels: channels, channel_mask: 0,
c19ce782
KS
934 granularity: granularity, bytes_per_minute: bytes_per_minute,
935 total_bytes: total_bytes & 0xFFFFFF, edata_size: edata_size,
936 ileave_info: ileave_info,
937 })
938}
939
ce52b3b5
KS
940const RMVB_HDR_SIZE: u32 = 18;
941const RMVB_PROP_SIZE: u32 = 50;
942const KEYFRAME_FLAG: u8 = 0x02;
943
944impl<'a> RealMediaDemuxer<'a> {
945 fn new(io: &'a mut ByteReader<'a>) -> Self {
946 RealMediaDemuxer {
947 src: io,
948 data_pos: 0,
770bef6f 949 next_data: 0,
db5cc44b 950 data_ver: 0,
ce52b3b5
KS
951 num_packets: 0,
952 cur_packet: 0,
418bd0b8 953 str_data: CommonStreamData::new(),
ce52b3b5
KS
954 queued_pkts: Vec::new(),
955 slice_buf: Vec::new(),
956 }
957 }
958#[allow(unused_variables)]
959 fn read_header(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<()> {
960 let (id, size, ver) = read_chunk(self.src)?;
db5cc44b 961 validate!((id == mktag!(b".RMF")) || (id == mktag!(b".RMP")));
ce52b3b5
KS
962 validate!(size >= RMVB_HDR_SIZE);
963 let fver = self.src.read_u32be()?;
964 validate!(fver <= 1);
965 let num_hdr = self.src.read_u32be()? as usize;
966 validate!(num_hdr >= 1);
967 if size > RMVB_HDR_SIZE {
968 self.src.read_skip((size - RMVB_HDR_SIZE) as usize)?;
969 }
970
971 let (id, size, ver) = read_chunk(self.src)?;
db5cc44b
KS
972 let prop_size = if ver == 0 { RMVB_PROP_SIZE } else { RMVB_PROP_SIZE + 4 };
973 validate!(size >= prop_size);
974 validate!((ver == 0) || (ver == 2));
ce52b3b5
KS
975 let maxbr = self.src.read_u32be()?;
976 let avgbr = self.src.read_u32be()?;
977 let maxps = self.src.read_u32be()?;
978 let avgps = self.src.read_u32be()?;
979 let num_pkt = self.src.read_u32be()? as usize;
980 let duration = self.src.read_u32be()?;
981 let preroll = self.src.read_u32be()?;
db5cc44b
KS
982 if ver == 2 {
983 self.src.read_skip(4)?;
984 }
ce52b3b5
KS
985 let idx_off = self.src.read_u32be()?;
986 let data_off = self.src.read_u32be()?;
987 let num_streams = self.src.read_u16be()? as usize;
988 let flags = self.src.read_u16be()?;
db5cc44b
KS
989 if size > prop_size {
990 self.src.read_skip((size - prop_size) as usize)?;
ce52b3b5
KS
991 }
992
993 for _ in 0..num_hdr {
c9aba47e
KS
994 if self.src.is_eof() {
995 //warn maybe?
996 break;
997 }
998 let res = self.parse_chunk(strmgr);
999 match res {
1000 Ok(last) => { if last { break; } },
1001 Err(DemuxerError::IOError) => { break; },
b7b96b4c
KS
1002 Err(etype) => {
1003 if self.data_pos == 0 { // data is not found, report error
1004 return Err(etype);
1005 }
1006 },
c9aba47e 1007 };
ce52b3b5 1008 }
c9aba47e 1009//println!("now @ {:X} / {}", self.src.tell(), self.data_pos);
ce52b3b5
KS
1010 validate!(self.data_pos > 0);
1011 self.src.seek(SeekFrom::Start(self.data_pos))?;
770bef6f
KS
1012 self.parse_data_start()?;
1013 Ok(())
1014 }
1015 fn parse_data_start(&mut self) -> DemuxerResult<()> {
ce52b3b5 1016 let num_packets = self.src.read_u32be()?;
db5cc44b
KS
1017 if self.data_ver == 2 {
1018 self.src.read_skip(12)?; // zeroes?
1019 }
ce52b3b5 1020 let next_data_hdr = self.src.read_u32be()?;
9f45e528 1021 self.num_packets = if num_packets > 0 { num_packets } else { 0xFFFFFF };
ce52b3b5 1022 self.cur_packet = 0;
770bef6f 1023 self.next_data = next_data_hdr as u64;
ce52b3b5
KS
1024 Ok(())
1025 }
6132225f 1026 fn parse_chunk(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<bool> {
ce52b3b5
KS
1027 let (id, size, ver) = read_chunk(self.src)?;
1028 let end_pos = self.src.tell() - 10 + (size as u64);
1029
db5cc44b 1030 validate!((ver == 0) || (ver == 2));
ce52b3b5
KS
1031 if id == mktag!(b"CONT") { self.parse_content_desc()?; }
1032 else if id == mktag!(b"MDPR") { self.parse_mdpr(strmgr)?; }
db5cc44b 1033 else if id == mktag!(b"DATA") { if self.data_pos == 0 { self.data_ver = ver; self.data_pos = self.src.tell(); } }
ce52b3b5 1034 else if id == mktag!(b"INDX") { /* do nothing for now */ }
6132225f 1035 else if id == 0 { return Ok(true); }
ce52b3b5
KS
1036 else { println!("unknown chunk type {:08X}", id); }
1037
1038 let cpos = self.src.tell();
1039 if cpos < end_pos {
1040 self.src.read_skip((end_pos - cpos) as usize)?;
1041 }
6132225f 1042 Ok(false)
ce52b3b5
KS
1043 }
1044#[allow(unused_variables)]
1045 fn parse_content_desc(&mut self) -> DemuxerResult<()> {
1046 let title_len = self.src.read_u16be()? as usize;
1047 self.src.read_skip(title_len)?;
1048 let author_len = self.src.read_u16be()? as usize;
1049 self.src.read_skip(author_len)?;
1050 let copywrong_len = self.src.read_u16be()? as usize;
1051 self.src.read_skip(copywrong_len)?;
1052 let comment_len = self.src.read_u16be()? as usize;
1053 self.src.read_skip(comment_len)?;
1054 Ok(())
1055 }
1056#[allow(unused_variables)]
1057 fn parse_mdpr(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<()> {
d4df9ee1 1058 let stream_no = self.src.read_u16be()? as u32;
ce52b3b5
KS
1059//todo check stream_no for duplicates
1060 let maxbr = self.src.read_u32be()?;
1061 let avgbr = self.src.read_u32be()?;
1062 let maxps = self.src.read_u32be()?;
1063 let avgps = self.src.read_u32be()?;
1064 let start = self.src.read_u32be()?;
1065 let preroll = self.src.read_u32be()?;
1066 let duration = self.src.read_u32be()?;
1067 let sname_size = self.src.read_byte()? as usize;
1068 let sname = read_string_size(self.src, sname_size)?;
c9aba47e 1069//println!("str #{} sname = {} pkts {}/{} start {} preroll {}", stream_no, sname, maxps, avgps, start, preroll);
ce52b3b5
KS
1070 let mime_size = self.src.read_byte()? as usize;
1071 let mime = read_string_size(self.src, mime_size)?;
c9aba47e 1072//println!("mime = {}", mime);
ce52b3b5
KS
1073 let edata_size = self.src.read_u32be()? as usize;
1074 let edata: Option<Vec<u8>> = if edata_size == 0 { None } else {
1075 let mut edvec: Vec<u8> = Vec::with_capacity(edata_size);
1076 edvec.resize(edata_size, 0);
1077 self.src.read_buf(&mut edvec)?;
1078 Some(edvec)
1079 };
d4df9ee1 1080 let mut is_mlti = false;
ce52b3b5
KS
1081 if edata_size > 8 {
1082 if let Some(edata_) = edata {
418bd0b8 1083 is_mlti = RMDemuxCommon::parse_stream_info(&mut self.str_data, strmgr, stream_no, &edata_)?;
ce52b3b5
KS
1084 }
1085 } else {
418bd0b8 1086 self.str_data.streams.push(RMStreamType::Unknown);
ce52b3b5 1087 }
d4df9ee1 1088 if !is_mlti {
418bd0b8 1089 self.str_data.str_ids.push(stream_no);
d4df9ee1 1090 }
d4df9ee1 1091
ce52b3b5
KS
1092 Ok(())
1093 }
1094/*#[allow(unused_variables)]
1095 fn read_pkt_header(&mut self) -> DemuxerResult<()> {
1096 let ver = self.src.read_u16be()?;
1097 validate!(ver <= 1);
1098 let str_no = self.src.read_u16be()?;
1099 let timestamp = self.src.read_u32be()?;
1100 if ver == 0 {
1101 let pkt_group = self.src.read_byte()?;
1102 let pkt_flags = self.src.read_byte()?;
1103 } else {
1104 let asm_rule = self.src.read_u16be()?;
1105 let asm_flags = self.src.read_byte()?;
1106 }
1107 Ok(())
1108 }*/
1109}
1110
1111fn read_string(src: &mut ByteReader) -> DemuxerResult<String> {
1112 let mut vec: Vec<u8> = Vec::new();
1113 loop {
1114 let c = src.read_byte()?;
1115 if c == 0 { break; }
1116 vec.push(c);
1117 }
1118 let str = String::from_utf8(vec);
1119 if str.is_ok() {
1120 Ok(str.unwrap())
1121 } else {
1122 Ok(String::new())
1123 }
1124}
1125
1126fn read_string_size(src: &mut ByteReader, size: usize) -> DemuxerResult<String> {
1127 let mut vec: Vec<u8> = Vec::new();
1128 for _ in 0..size {
1129 let c = src.read_byte()?;
1130 vec.push(c);
1131 }
1132 let str = String::from_utf8(vec);
1133 if str.is_ok() {
1134 Ok(str.unwrap())
1135 } else {
1136 Ok(String::new())
1137 }
1138}
1139
1140#[allow(dead_code)]
1141#[allow(unused_variables)]
1142fn parse_rm_stream(io: &mut ByteReader) -> DemuxerResult<NAStream> {
1143 let mimetype = read_string(io)?;
1144 let strname = read_string(io)?;
7348405b
KS
1145 let strnum = io.read_u32be()?;
1146 let maxbr = io.read_u32be()?;
1147 let avgbr = io.read_u32be()?;
1148 let maxsize = io.read_u32be()?;
1149 let avgsize = io.read_u32be()?;
1150 let duration = io.read_u32be()?;
1151 let preroll = io.read_u32be()?;
1152 let start = io.read_u32be()?;
1153 let edatalen = io.read_u32be()? as usize;
ce52b3b5
KS
1154 let mut edata: Vec<u8> = Vec::with_capacity(edatalen);
1155 edata.resize(edatalen, 0);
1156 io.read_buf(&mut edata)?;
7348405b 1157 let numprops = io.read_u32be()? as usize;
ce52b3b5
KS
1158 //read properties
1159 unimplemented!();
1160}
1161
d1304e6e
KS
1162struct RealAudioDemuxer<'a> {
1163 src: &'a mut ByteReader<'a>,
1164 stream: Option<RMAudioStream>,
1165 data_start: u64,
1166 data_end: u64,
1167 blk_size: usize,
1168 queued_pkts: Vec<NAPacket>,
1169}
1170
1171impl<'a> DemuxCore<'a> for RealAudioDemuxer<'a> {
1172 #[allow(unused_variables)]
1173 fn open(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<()> {
1174 let magic = self.src.read_u32be()?;
1175 validate!(magic == mktag!(b".ra\xFD"));
1176 let ver = self.src.read_u16be()?;
1177 let ainfo = match ver {
1178 3 => {
1179 parse_aformat3(&mut self.src)?
1180 },
1181 4 => {
1182 parse_aformat4(&mut self.src)?
1183 },
1184 5 => {
1185 parse_aformat5(&mut self.src)?
1186 },
1187 _ => {
1188 println!("unknown version {}", ver);
1189 return Err(DemuxerError::InvalidData);
1190 },
1191 };
1192println!(" got ainfo {:?}", ainfo);
1193 let cname = find_codec_name(RM_AUDIO_CODEC_REGISTER, ainfo.fcc);
1194 let blk_size = if ainfo.fcc != mktag!(b"sipr") {
1195 ainfo.granularity as usize
1196 } else {
1197 validate!(ainfo.flavor <= 3);
1198 RM_SIPRO_BLOCK_SIZES[ainfo.flavor as usize]
1199 };
1200 let srate = ainfo.sample_rate;
1201 let soniton = NASoniton::new(ainfo.sample_size as u8, SONITON_FLAG_SIGNED);
1202 let ahdr = NAAudioInfo::new(srate, ainfo.channels as u8, soniton, blk_size);
1203 let extradata = if ainfo.edata_size == 0 {
1204 None
1205 } else {
1206 let mut dta: Vec<u8> = Vec::with_capacity(ainfo.edata_size as usize);
1207 dta.resize(ainfo.edata_size as usize, 0);
1208 self.src.read_buf(dta.as_mut_slice())?;
1209 Some(dta)
1210 };
1211 let nainfo = NACodecInfo::new(cname, NACodecTypeInfo::Audio(ahdr), extradata);
1212 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, 0, nainfo, 1, srate));
1213 if res.is_none() { return Err(MemoryError); }
1214
1215 let astr = RMAudioStream::new(ainfo.ileave_info);
1216 self.data_start = self.src.tell();
1217 self.data_end = if ainfo.total_bytes > 0 { self.src.tell() + (ainfo.total_bytes as u64) } else { 0 };
1218 self.blk_size = blk_size;
1219 self.stream = Some(astr);
1220
1221 Ok(())
1222 }
1223
1224 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
1225 if !self.queued_pkts.is_empty() {
1226 let pkt = self.queued_pkts.pop().unwrap();
1227 return Ok(pkt);
1228 }
1229 if (self.data_end != 0) && (self.src.tell() >= self.data_end) {
1230 return Err(DemuxerError::EOF);
1231 }
1232 let streamres = strmgr.get_stream_by_id(0);
1233 let stream = streamres.unwrap();
1234 if let Some(ref mut astr) = self.stream {
1235 loop {
1236 let ret = astr.read_apackets(&mut self.queued_pkts, &mut self.src, stream.clone(), 0, false, self.blk_size);
1237 if let Err(DemuxerError::TryAgain) = ret {
1238 continue;
1239 }
1240 return ret;
1241 }
1242 }
1243 Err(DemuxerError::NoSuchInput)
1244 }
1245
1246 #[allow(unused_variables)]
1247 fn seek(&mut self, time: u64) -> DemuxerResult<()> {
1248 Err(NotImplemented)
1249 }
1250}
1251
1252impl<'a> RealAudioDemuxer<'a> {
1253 fn new(io: &'a mut ByteReader<'a>) -> Self {
1254 RealAudioDemuxer {
1255 src: io,
1256 data_start: 0,
1257 data_end: 0,
1258 blk_size: 0,
1259 stream: None,
1260 queued_pkts: Vec::new(),
1261 }
1262 }
1263}
1264
418bd0b8
KS
1265enum IVRRecord {
1266 Invalid(u8),
1267 StreamProperties(usize),
1268 Packet { ts: u32, str: u32, flags: u32, len: usize, checksum: u32 },
1269 IntValue(Vec<u8>, u32),
1270 BinaryData(Vec<u8>, Vec<u8>),
1271 StringData(Vec<u8>, Vec<u8>),
1272 HeaderEnd,
1273 DataStart,
1274 DataEnd,
1275}
1276
1277impl IVRRecord {
1278 fn read_string(src: &mut ByteReader) -> DemuxerResult<Vec<u8>> {
1279 let len = src.read_u32be()? as usize;
1280 let mut val = Vec::with_capacity(len);
1281 val.resize(len, 0);
1282 src.read_buf(val.as_mut_slice())?;
1283 Ok(val)
1284 }
1285
1286 fn read(src: &mut ByteReader) -> DemuxerResult<Self> {
1287 let code = src.read_byte()?;
1288 match code {
1289 1 => {
1290 let val = src.read_u32be()? as usize;
1291 Ok(IVRRecord::StreamProperties(val))
1292 },
1293 2 => {
1294 let ts = src.read_u32be()?;
1295 let str = src.read_u16be()? as u32;
1296 let flags = src.read_u32be()?;
1297 let len = src.read_u32be()? as usize;
1298 let chk = src.read_u32be()?;
1299 Ok(IVRRecord::Packet { ts, str, flags, len, checksum: chk })
1300 },
1301 3 => {
1302 let name = Self::read_string(src)?;
1303 let len = src.read_u32be()?;
1304 validate!(len == 4);
1305 let val = src.read_u32be()?;
1306 Ok(IVRRecord::IntValue(name, val))
1307 },
1308 4 => {
1309 let name = Self::read_string(src)?;
1310 let len = src.read_u32be()? as usize;
1311 let mut val = Vec::with_capacity(len);
1312 val.resize(len, 0);
1313 src.read_buf(val.as_mut_slice())?;
1314 Ok(IVRRecord::BinaryData(name, val))
1315 },
1316 5 => {
1317 let name = Self::read_string(src)?;
1318 let val = Self::read_string(src)?;
1319 Ok(IVRRecord::StringData(name, val))
1320 },
1321 6 => Ok(IVRRecord::HeaderEnd),
1322 7 => {
1323 src.read_skip(8)?; // always zero?
1324 Ok(IVRRecord::DataEnd)
1325 },
1326 8 => {
1327 src.read_skip(8)?; // always zero?
1328 Ok(IVRRecord::DataStart)
1329 },
1330 _ => Ok(IVRRecord::Invalid(code)),
1331 }
1332 }
1333 fn is_data_start(&self) -> bool {
1334 match *self {
1335 IVRRecord::DataStart => true,
1336 _ => false,
1337 }
1338 }
1339}
1340
1341impl fmt::Display for IVRRecord {
1342 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
1343 match *self {
1344 IVRRecord::Invalid(typ) => write!(f, "Invalid({:02X})", typ),
1345 IVRRecord::StreamProperties(num) =>
1346 write!(f, "({} stream properties)", num),
1347 IVRRecord::Packet { ts, str, flags, len, checksum } =>
1348 write!(f, "paket({}, {}, {:X}, {}, {})", ts, str, flags, len, checksum),
1349 IVRRecord::IntValue(ref name, val) =>
1350 write!(f, "({} = {})", String::from_utf8_lossy(name), val),
1351 IVRRecord::BinaryData(ref name, ref val) =>
1352 write!(f, "({} = {} bytes)", String::from_utf8_lossy(name), val.len()),
1353 IVRRecord::StringData(ref name, ref val) =>
1354 write!(f, "({} = {})", String::from_utf8_lossy(name), String::from_utf8_lossy(val)),
1355 IVRRecord::HeaderEnd => write!(f, "header end"),
1356 IVRRecord::DataEnd => write!(f, "data end"),
1357 IVRRecord::DataStart => write!(f, "data start"),
1358 }
1359 }
1360}
1361
1362struct RecordDemuxer {
1363 start_pos: u64,
1364 cur_pos: u64,
1365 start_str: u32,
1366 remap_ids: Vec<u32>,
1367}
1368
1369impl RecordDemuxer {
1370 fn new(pos: u64, start_str: u32) -> Self {
1371 RecordDemuxer {
1372 start_pos: pos,
1373 cur_pos: pos,
1374 start_str: start_str,
1375 remap_ids: Vec::new(),
1376 }
1377 }
1378 fn parse_header(&mut self, src: &mut ByteReader, strmgr: &mut StreamManager, str_data: &mut CommonStreamData) -> DemuxerResult<()> {
1379 src.seek(SeekFrom::Start(self.cur_pos))?;
1380 let magic = src.read_u32be()?;
1381 validate!(magic == mktag!(b".REC"));
1382 let _smth = src.read_byte()?;
1383 let num_entries = src.read_u32be()? as usize;
1384 for _ in 0..num_entries {
1385 let _rec = IVRRecord::read(src)?;
1386//println!(" header rec {}", _rec);
1387 }
1388 let mut has_seek_table = false;
1389 let mut cur_str_no = 0;
1390 loop {
1391 let rec = IVRRecord::read(src)?;
1392 match rec {
1393 IVRRecord::HeaderEnd => { break; },
1394 IVRRecord::StreamProperties(num) => {
1395 let stream_no = cur_str_no + self.start_str;
1396 cur_str_no += 1;
1397 let mut parsed = false;
1398 let mut real_stream_no = 0;
1399 for _ in 0..num {
1400 let rec = IVRRecord::read(src)?;
1401//println!(" strm property {}", rec);
1402 match rec {
1403 IVRRecord::IntValue(ref name, val) => {
1404 if name == b"StreamNumber\0" {
1405 real_stream_no = val;
1406 }
1407 },
1408 IVRRecord::BinaryData(ref name, ref val) => {
1409 if name == b"OpaqueData\0" {
1410 validate!(!parsed);
1411 let is_mlti = RMDemuxCommon::parse_stream_info(str_data, strmgr, stream_no, val)?;
1412 if !is_mlti {
1413 str_data.str_ids.push(stream_no);
1414 }
1415 parsed = true;
1416 }
1417 },
1418 IVRRecord::StringData(ref name, ref val) => {
1419 if (name == b"SeekType\0") && (val != b"None\0") {
1420 has_seek_table = true;
1421 }
1422 },
1423 _ => { return Err(DemuxerError::InvalidData); }
1424 };
1425 }
1426 if !parsed {
1427 str_data.streams.push(RMStreamType::Unknown);
1428 str_data.str_ids.push(stream_no);
1429 }
1430 self.remap_ids.push(real_stream_no);
1431 },
1432 _ => {println!(" unexpected {}", rec); return Err(DemuxerError::InvalidData); }
1433 };
1434 }
1435 println!(" now @ {:X}", src.tell());
1436 let off0 = src.read_u32be()? as u64;
1437 let _off1 = src.read_u32be()?;
1438 let _off2 = src.read_u32be()?;
1439 validate!(off0 + self.start_pos == src.tell());
1440 println!(" has seek tab: {}", has_seek_table);
1441 if has_seek_table {
1442 src.read_skip(4)?;
1443 let data_off = src.read_u32be()? as u64;
1444 println!(" new off {:X}", data_off);
1445 let pos = src.tell();
1446 validate!(data_off + self.start_pos > pos);
1447 src.read_skip((data_off + self.start_pos - pos) as usize)?;
1448 let rec = IVRRecord::read(src)?;
1449 validate!(rec.is_data_start());
1450 } else {
1451 let ntype = src.peek_byte()?;
1452 validate!((ntype == 2) || (ntype == 7)); // packet or data end, no start
1453 }
1454
1455 self.cur_pos = src.tell();
1456
1457 Ok(())
1458 }
1459 fn get_packet(&mut self, src: &mut ByteReader, str_data: &mut CommonStreamData, strmgr: &StreamManager, queued_pkts: &mut Vec<NAPacket>, slice_buf: &mut Vec<u8>) -> DemuxerResult<NAPacket> {
1460 src.seek(SeekFrom::Start(self.cur_pos))?;
1461 loop {
1462 let rec = IVRRecord::read(src)?;
1463 match rec {
1464 IVRRecord::Packet { ts, str, flags: _, len, checksum: _ } => {
1465 let payload_size = len;
1466 let sr = self.remap_ids.iter().position(|x| *x == str);
1467 validate!(sr.is_some());
1468 let str_no = self.start_str + (sr.unwrap() as u32);
1469 let stream_id = str_data.get_stream_id(str_no as u32, 0/*pkt_grp*/);
1470 let sr = str_data.find_stream(stream_id);
1471 if sr.is_none() {
1472 src.read_skip(payload_size)?;
1473 return Err(DemuxerError::InvalidData);
1474 }
1475 let str_id = sr.unwrap();
1476
1477 let streamres = strmgr.get_stream_by_id(stream_id);
1478 if streamres.is_none() {
1479 src.read_skip(payload_size)?;
1480 continue;
1481 }
1482 let stream = streamres.unwrap();
1483 if strmgr.is_ignored_id(stream_id) {
1484 src.read_skip(payload_size)?;
1485 continue;
1486 }
1487 let keyframe = false;
1488 let ret = RMDemuxCommon::parse_packet_payload(src, &mut str_data.streams[str_id], stream, slice_buf, queued_pkts, keyframe, ts, payload_size);
1489 if let Err(DemuxerError::TryAgain) = ret {
1490 continue;
1491 } else {
1492 self.cur_pos = src.tell();
1493 return ret;
1494 }
1495 },
1496 IVRRecord::DataEnd => return Err(DemuxerError::EOF),
1497 _ => return Err(DemuxerError::InvalidData),
1498 }
1499 }
1500 }
1501}
1502
1503struct RealIVRDemuxer<'a> {
1504 src: &'a mut ByteReader<'a>,
1505 recs: Vec<RecordDemuxer>,
1506 cur_rec: usize,
1507 queued_pkts: Vec<NAPacket>,
1508 slice_buf: Vec<u8>,
1509 str_data: CommonStreamData,
1510}
1511
1512impl<'a> DemuxCore<'a> for RealIVRDemuxer<'a> {
1513 #[allow(unused_variables)]
1514 fn open(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<()> {
1515 let magic = self.src.peek_u32be()?;
1516 if magic == mktag!(b".REC") {
1517 let mut rec = RecordDemuxer::new(0, 0);
1518 rec.parse_header(&mut self.src, strmgr, &mut self.str_data)?;
1519 self.recs.push(rec);
1520 } else if magic == mktag!(b".R1M") {
1521println!("R1M kind");
1522 self.src.read_skip(4)?; // magic
1523 self.src.read_skip(3)?; // always 0, 1, 1 ?
1524 let _name = IVRRecord::read_string(&mut self.src)?;
1525 self.src.read_skip(1)?; // always 0?
1526 let len1 = self.src.read_u32be()? as u64;
1527 let off1 = self.src.read_u64be()?;
1528 let cpos = self.src.tell();
1529 validate!(off1 == len1 + cpos - 8);
1530 self.src.read_skip((off1 - cpos) as usize)?;
1531 loop {
1532 let typ = self.src.read_byte()?;
1533 println!(" at {:X} type {:02X}", self.src.tell(), typ);
1534 match typ {
1535 1 => {
1536 let len = self.src.read_u32be()?;
1537 self.src.read_skip(len as usize)?;
1538 },
1539 2 => {
1540 let len = self.src.read_u32be()? as u64;
1541 let pos = self.src.tell();
1542 let num_streams = self.str_data.streams.len() as u32;
1543 let mut rec = RecordDemuxer::new(pos + 12, num_streams);
1544 rec.parse_header(&mut self.src, strmgr, &mut self.str_data)?;
1545 self.recs.push(rec);
1546 self.src.seek(SeekFrom::Start(pos + len))?;
1547 },
1548 b'R' => {
1549 let mut buf: [u8; 2] = [0; 2];
1550 self.src.peek_buf(&mut buf)?;
1551 if (buf[0] == b'J') && (buf[1] == b'M') { // RJMx markers at the end of file
1552 break;
1553 } else {
1554 return Err(DemuxerError::InvalidData);
1555 }
1556 },
1557 _ => { return Err(DemuxerError::InvalidData); },
1558 };
1559 }
1560 } else {
1561 return Err(DemuxerError::InvalidData);
1562 }
1563
1564 Ok(())
1565 }
1566
1567 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
1568 if !self.queued_pkts.is_empty() {
1569 let pkt = self.queued_pkts.pop().unwrap();
1570 return Ok(pkt);
1571 }
1572 loop {
1573 if self.cur_rec >= self.recs.len() { return Err(DemuxerError::EOF); }
1574 let res = self.recs[self.cur_rec].get_packet(&mut self.src, &mut self.str_data, strmgr, &mut self.queued_pkts, &mut self.slice_buf);
1575 if let Err(DemuxerError::EOF) = res {
1576 self.cur_rec += 1;
1577 } else {
1578 return res;
1579 }
1580 }
1581 }
1582
1583 #[allow(unused_variables)]
1584 fn seek(&mut self, time: u64) -> DemuxerResult<()> {
1585 Err(NotImplemented)
1586 }
1587}
1588
1589impl<'a> RealIVRDemuxer<'a> {
1590 fn new(io: &'a mut ByteReader<'a>) -> Self {
1591 RealIVRDemuxer {
1592 src: io,
1593 recs: Vec::new(),
1594 cur_rec: 0,
1595 queued_pkts: Vec::new(),
1596 slice_buf: Vec::new(),
1597 str_data: CommonStreamData::new(),
1598 }
1599 }
1600}
1601
ce52b3b5
KS
1602static RM_VIDEO_CODEC_REGISTER: &'static [(&[u8;4], &str)] = &[
1603 (b"RV10", "realvideo1"),
1604 (b"RV20", "realvideo2"),
1605 (b"RVTR", "realvideo2"),
1606 (b"RV30", "realvideo3"),
1607 (b"RV40", "realvideo4"),
db5cc44b 1608 (b"RV60", "realvideo6"),
750b299c 1609 (b"CLV1", "clearvideo_rm"),
ce52b3b5
KS
1610];
1611
b1ef37ee 1612#[allow(dead_code)]
ce52b3b5
KS
1613static RM_AUDIO_CODEC_REGISTER: &'static [(&[u8;4], &str)] = &[
1614 (b"lpcJ", "ra14.4"),
1615 (b"28_8", "ra28.8"),
1616 (b"cook", "cook"),
1617 (b"dnet", "ac3"),
1618 (b"sipr", "sipro"),
1619 (b"atrc", "atrac3"),
1620 (b"LSD:", "ralf"),
1621 (b"raac", "aac"),
1622 (b"racp", "aac"),
1623];
1624
1625pub struct RealMediaDemuxerCreator { }
1626
1627impl DemuxerCreator for RealMediaDemuxerCreator {
1628 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<DemuxCore<'a> + 'a> {
1629 Box::new(RealMediaDemuxer::new(br))
1630 }
1631 fn get_name(&self) -> &'static str { "realmedia" }
1632}
1633
d1304e6e
KS
1634pub struct RealAudioDemuxerCreator { }
1635
1636impl DemuxerCreator for RealAudioDemuxerCreator {
1637 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<DemuxCore<'a> + 'a> {
1638 Box::new(RealAudioDemuxer::new(br))
1639 }
1640 fn get_name(&self) -> &'static str { "realaudio" }
1641}
1642
418bd0b8 1643pub struct RealIVRDemuxerCreator { }
d1304e6e
KS
1644
1645impl DemuxerCreator for RealIVRDemuxerCreator {
1646 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<DemuxCore<'a> + 'a> {
1647 Box::new(RealIVRDemuxer::new(br))
1648 }
1649 fn get_name(&self) -> &'static str { "real_ivr" }
418bd0b8 1650}
d1304e6e 1651
ce52b3b5
KS
1652#[cfg(test)]
1653mod test {
1654 use super::*;
1655 use std::fs::File;
1656
1657 #[test]
1658 fn test_rm_demux() {
1659 let mut file =
1660 File::open("assets/RV/rv10_dnet_640x352_realvideo_encoder_4.0.rm").unwrap();
1661// File::open("assets/RV/rv20_cook_640x352_realproducer_plus_8.51.rm").unwrap();
1662// File::open("assets/RV/rv20_svt_atrc_640x352_realproducer_plus_8.51.rm").unwrap();
1663// File::open("assets/RV/rv30_atrc_384x208_realproducer_plus_8.51.rm").unwrap();
1664// File::open("assets/RV/rv30_chroma_drift.rm").unwrap();
1665// File::open("assets/RV/rv30_weighted_mc.rm").unwrap();
1666// File::open("assets/RV/rv40_weighted_mc.rmvb").unwrap();
1667// File::open("assets/RV/rv40_weighted_mc_2.rmvb").unwrap();
1668// File::open("assets/RV/clv1_sipr_384x208_realvideo_encoder_4.0.rm").unwrap();
1669// File::open("assets/RV/luckynight.rmvb").unwrap();
1670// File::open("assets/RV/rv40_ralf.rmvb").unwrap();
1671 let mut fr = FileReader::new_read(&mut file);
1672 let mut br = ByteReader::new(&mut fr);
1673 let mut dmx = RealMediaDemuxer::new(&mut br);
1674 let mut sm = StreamManager::new();
1675 dmx.open(&mut sm).unwrap();
1676
1677 loop {
1678 let pktres = dmx.get_frame(&mut sm);
1679 if let Err(e) = pktres {
1680 if e == DemuxerError::EOF { break; }
1681 panic!("error");
1682 }
1683 let pkt = pktres.unwrap();
1684 println!("Got {}", pkt);
1685 }
41f11b2b 1686//panic!("the end");
ce52b3b5 1687 }
d1304e6e
KS
1688 #[test]
1689 fn test_ra_demux() {
1690 let mut file =
1691// File::open("assets/RV/welcome288.ra").unwrap();
1692 File::open("assets/RV/diemusik.ra").unwrap();
1693 let mut fr = FileReader::new_read(&mut file);
1694 let mut br = ByteReader::new(&mut fr);
1695 let mut dmx = RealAudioDemuxer::new(&mut br);
1696 let mut sm = StreamManager::new();
1697 dmx.open(&mut sm).unwrap();
1698
418bd0b8
KS
1699 loop {
1700 let pktres = dmx.get_frame(&mut sm);
1701 if let Err(e) = pktres {
1702 if e == DemuxerError::EOF { break; }
1703 panic!("error");
1704 }
1705 let pkt = pktres.unwrap();
1706 println!("Got {}", pkt);
1707 }
1708 }
1709 #[test]
1710 fn test_ivr_demux() {
1711 let mut file =
1712 File::open("assets/RV/Opener_rm_hi.ivr").unwrap();
1713// File::open("assets/RV/SherwinWilliamsCommercial.ivr").unwrap();
1714 let mut fr = FileReader::new_read(&mut file);
1715 let mut br = ByteReader::new(&mut fr);
1716 let mut dmx = RealIVRDemuxer::new(&mut br);
1717 let mut sm = StreamManager::new();
1718 dmx.open(&mut sm).unwrap();
1719
d1304e6e
KS
1720 loop {
1721 let pktres = dmx.get_frame(&mut sm);
1722 if let Err(e) = pktres {
1723 if e == DemuxerError::EOF { break; }
1724 panic!("error");
1725 }
1726 let pkt = pktres.unwrap();
1727 println!("Got {}", pkt);
1728 }
1729 }
ce52b3b5 1730}