2 use nihav_core::demuxers::DemuxerError::*;
8 ($a:expr, $b:expr, $c:expr, $d:expr) => ({
9 (($a as u32) << 24) | (($b as u32) << 16) | (($c as u32) << 8) | ($d as u32)
12 (($arr[0] as u32) << 24) | (($arr[1] as u32) << 16) | (($arr[2] as u32) << 8) | ($arr[3] as u32)
17 fn read_size(&mut self, ver: u16) -> ByteIOResult<u64>;
20 impl<'a> ReadSize for ByteReader<'a> {
21 fn read_size(&mut self, ver: u16) -> ByteIOResult<u64> {
23 0 => Ok(u64::from(self.read_u32be()?)),
24 2 => self.read_u64be(),
30 const RM_SIPRO_BLOCK_SIZES: [usize; 4] = [ 29, 19, 37, 20 ];
31 const RM_SIPRO_SWAPS: [[u8; 2]; 38] = [
32 [ 0, 63 ], [ 1, 22 ], [ 2, 44 ], [ 3, 90 ],
33 [ 5, 81 ], [ 7, 31 ], [ 8, 86 ], [ 9, 58 ],
34 [ 10, 36 ], [ 12, 68 ], [ 13, 39 ], [ 14, 73 ],
35 [ 15, 53 ], [ 16, 69 ], [ 17, 57 ], [ 19, 88 ],
36 [ 20, 34 ], [ 21, 71 ], [ 24, 46 ], [ 25, 94 ],
37 [ 26, 54 ], [ 28, 75 ], [ 29, 50 ], [ 32, 70 ],
38 [ 33, 92 ], [ 35, 74 ], [ 38, 85 ], [ 40, 56 ],
39 [ 42, 87 ], [ 43, 65 ], [ 45, 59 ], [ 48, 79 ],
40 [ 49, 93 ], [ 51, 89 ], [ 55, 95 ], [ 61, 76 ],
41 [ 67, 83 ], [ 77, 80 ]
44 struct RMVideoStream {
65 fn start_slice(&mut self, num_slices: usize, frame_size: usize, data: &[u8]) {
66 self.hdr_size = num_slices * 8 + 1;
67 self.frame.resize(frame_size + self.hdr_size, 0);
68 self.frame[0] = (num_slices - 1) as u8;
70 self.add_slice(1, data).unwrap();
72 fn add_slice(&mut self, slice_no: usize, data: &[u8]) -> DemuxerResult<()> {
73 validate!(self.hdr_size + self.frame_pos + data.len() <= self.frame.len());
74 self.write_slice_info(slice_no);
75 let dslice = &mut self.frame[self.hdr_size + self.frame_pos..][..data.len()];
76 dslice.copy_from_slice(data);
77 self.frame_pos += data.len();
80 fn write_slice_info(&mut self, slice_no: usize) {
81 let off = 1 + (slice_no - 1) * 8;
82 self.frame[off + 0] = 0;
83 self.frame[off + 1] = 0;
84 self.frame[off + 2] = 0;
85 self.frame[off + 3] = 1;
86 self.frame[off + 4] = (self.frame_pos >> 24) as u8;
87 self.frame[off + 5] = (self.frame_pos >> 16) as u8;
88 self.frame[off + 6] = (self.frame_pos >> 8) as u8;
89 self.frame[off + 7] = (self.frame_pos >> 0) as u8;
91 fn get_frame_data(&mut self) -> Vec<u8> {
92 let mut v: Vec<u8> = Vec::new();
93 mem::swap(&mut v, &mut self.frame);
99 #[derive(Clone,Copy,PartialEq)]
108 struct RMAudioStream {
109 deint: Deinterleaver,
110 iinfo: Option<InterleaveInfo>,
115 const RM_ILEAVE_INT0: u32 = mktag!(b"Int0");
116 const RM_ILEAVE_INT4: u32 = mktag!(b"Int4");
117 const RM_ILEAVE_GENR: u32 = mktag!(b"genr");
118 const RM_ILEAVE_SIPR: u32 = mktag!(b"sipr");
119 const RM_ILEAVE_VBRF: u32 = mktag!(b"vbrf");
120 const RM_ILEAVE_VBRS: u32 = mktag!(b"vbrs");
123 fn new(iinfo: Option<InterleaveInfo>) -> Self {
126 if let Some(info) = iinfo {
127 deint = match info.id {
128 RM_ILEAVE_INT0 => Deinterleaver::None,
129 RM_ILEAVE_INT4 => Deinterleaver::RA28_8,
130 RM_ILEAVE_GENR => Deinterleaver::Generic,
131 RM_ILEAVE_SIPR => Deinterleaver::Sipro,
132 RM_ILEAVE_VBRF => Deinterleaver::VBR,
133 RM_ILEAVE_VBRS => Deinterleaver::VBR,
134 _ => {println!("unknown deint {:X}", info.id); Deinterleaver::None },
137 Deinterleaver::None => { buf = Vec::new(); },
138 Deinterleaver::RA28_8 |
139 Deinterleaver::Generic |
140 Deinterleaver::Sipro => {
141 let bsize = (info.frame_size as usize) * (info.factor as usize);
142 buf = vec![0; bsize];
144 Deinterleaver::VBR => {
149 deint = Deinterleaver::None;
152 RMAudioStream { deint, iinfo, buf, sub_packet: 0 }
154 fn read_apackets(&mut self, queued_packets: &mut Vec<NAPacket>, src: &mut ByteReader, stream: NAStreamRef, ts: u32, keyframe: bool, payload_size: usize) -> DemuxerResult<NAPacket> {
155 let ts = stream.make_ts(Some(ts as u64), None, None);
161 Deinterleaver::None => { return src.read_packet(stream, ts, keyframe, payload_size); },
162 Deinterleaver::RA28_8 => {
163 let iinfo = self.iinfo.unwrap();
164 let factor = iinfo.factor as usize;
165 let halffact = factor >> 1;
166 let fsize = iinfo.frame_size as usize;
167 let bsize = iinfo.block_size as usize;
168 let ppos = self.sub_packet;
169 for sb in 0..halffact {
170 let dst = &mut self.buf[sb * 2 * fsize + ppos * bsize..][..bsize];
173 self.sub_packet += 1;
174 if self.sub_packet == factor {
176 return Ok(NAPacket::new(stream, ts, true, self.buf.clone()));
178 return Err(DemuxerError::TryAgain);
181 Deinterleaver::Generic => {
182 let iinfo = self.iinfo.unwrap();
183 let factor = iinfo.factor as usize;
184 let fsize = iinfo.frame_size as usize;
185 let bsize = iinfo.block_size as usize;
186 let factor2 = fsize / bsize;
187 let ppos = self.sub_packet;
189 for sb in 0..factor2 {
190 let sb_pos = factor * sb + ((factor + 1) >> 1) * (ppos & 1) + (ppos >> 1);
191 let dst = &mut self.buf[bsize * sb_pos..][..bsize];
195 Deinterleaver::Sipro => {
196 let iinfo = self.iinfo.unwrap();
197 let fsize = iinfo.frame_size as usize;
198 let ppos = self.sub_packet;
200 let dst = &mut self.buf[fsize * ppos..][..fsize];
203 Deinterleaver::VBR => {
204 validate!(payload_size >= 5);
205 let hdrsizesize = src.read_u16be()?;
206 let num_entries = (hdrsizesize / 16) as usize;
207 validate!(payload_size >= num_entries * 3 + 2);
208 let mut sizes: Vec<usize> = Vec::with_capacity(num_entries);
209 let mut tot_size = 0;
210 for _ in 0..num_entries {
211 let sz = src.read_u16be()? as usize;
215 validate!(tot_size + num_entries * 2 + 2 == payload_size);
216 let pkt_ts = stream.make_ts(None, None, None);
217 let mut first = true;
218 for size in sizes.iter() {
219 let cur_ts = if first { ts } else { pkt_ts };
221 let pkt = src.read_packet(stream.clone(), cur_ts, true, *size)?;
222 queued_packets.push(pkt);
224 queued_packets.reverse();
225 let pkt0 = queued_packets.pop().unwrap();
230 let iinfo = self.iinfo.unwrap();
231 let factor = iinfo.factor as usize;
232 let fsize = if iinfo.block_size != 0 { iinfo.block_size } else { iinfo.frame_size } as usize;
234 self.sub_packet += 1;
235 if self.sub_packet < factor {
236 return Err(DemuxerError::TryAgain);
241 if self.deint == Deinterleaver::Sipro {
242 sipro_restore(&mut self.buf, factor, iinfo.frame_size as usize);
245 let mut frames_iter = self.buf.chunks(fsize);
246 let pkt0 = frames_iter.next().unwrap();
248 let pkt_ts = stream.make_ts(None, None, None);
249 for pkts in frames_iter {
250 let pkt = NAPacket::new(stream.clone(), pkt_ts, true, pkts.to_vec());
251 queued_packets.push(pkt);
253 queued_packets.reverse();
254 Ok(NAPacket::new(stream, ts, true, pkt0.to_vec()))
258 fn sipro_restore(buf: &mut [u8], factor: usize, fsize: usize) {
259 let stride = factor * fsize * 2 / 96;
261 let mut sidx = (RM_SIPRO_SWAPS[i][0] as usize) * stride;
262 let mut didx = (RM_SIPRO_SWAPS[i][1] as usize) * stride;
264 let in0 = buf[sidx >> 1];
265 let in1 = buf[didx >> 1];
266 let nib0 = (in0 >> ((sidx & 1) * 4)) & 0xF;
267 let nib1 = (in1 >> ((didx & 1) * 4)) & 0xF;
269 buf[didx >> 1] = (nib0 << (4 * (didx & 1))) | (in1 & (0xF << (4 * (!didx & 1))));
270 buf[sidx >> 1] = (nib1 << (4 * (sidx & 1))) | (in0 & (0xF << (4 * (!sidx & 1))));
278 struct SubstreamInfo {
285 sub_info: Vec<SubstreamInfo>,
292 sub_info: Vec::new(),
296 fn add_stream(&mut self, stream_no: u32) {
297 let ssinfo = SubstreamInfo { id: stream_no, map: Vec::new(), str_ids: Vec::new() };
298 self.sub_info.push(ssinfo);
300 fn get_substream_no(&self) -> u32 {
303 fn find_idx(&self, stream_no: u32) -> Option<usize> {
304 self.sub_info.iter().position(|x| x.id == stream_no)
306 fn add_map_rule(&mut self, map_ss: u16) {
307 let idx = self.sub_info.len() - 1;
308 self.sub_info[idx].map.push(map_ss);
310 fn add_substream(&mut self) {
311 let idx = self.sub_info.len() - 1;
312 self.sub_info[idx].str_ids.push(self.sstr_id);
315 fn is_mlti_stream(&self, stream_no: u32) -> bool {
316 self.find_idx(stream_no).is_some()
318 fn find_substream(&self, stream_no: u32, grp: u16) -> Option<u32> {
319 if let Some(idx) = self.find_idx(stream_no) {
320 if (grp as usize) < self.sub_info[idx].map.len() {
321 let sub_id = self.sub_info[idx].map[grp as usize] as usize;
322 if sub_id < self.sub_info[idx].str_ids.len() {
323 return Some(self.sub_info[idx].str_ids[sub_id]);
332 Audio(RMAudioStream),
333 Video(RMVideoStream),
338 struct CommonStreamData {
339 streams: Vec<RMStreamType>,
341 mlti_mapper: MLTIMapper,
344 impl CommonStreamData {
349 mlti_mapper: MLTIMapper::new(),
352 fn get_stream_id(&self, str_no: u32, pkt_grp: u16) -> u32 {
353 if !self.mlti_mapper.is_mlti_stream(str_no) {
356 self.mlti_mapper.find_substream(str_no, pkt_grp).unwrap()
359 fn find_stream(&self, stream_id: u32) -> Option<usize> {
360 self.str_ids.iter().position(|x| *x == stream_id)
364 struct RealMediaDemuxer<'a> {
365 src: &'a mut ByteReader<'a>,
373 str_data: CommonStreamData,
375 data_chunks: Vec<(u64, u32, u16)>,
376 cur_data_chunk: usize,
378 queued_pkts: Vec<NAPacket>,
382 fn find_codec_name(registry: &[(&[u8;4], &'static str)], fcc: u32) -> &'static str {
383 for &(fourcc, name) in registry {
384 if mktag!(fourcc) == fcc { return name; }
389 fn read_14or30(src: &mut ByteReader) -> DemuxerResult<(bool, u32)> {
390 let tmp = src.read_u16be()?;
391 let flag = (tmp & 0x8000) != 0;
392 if (tmp & 0x4000) == 0x4000 {
393 Ok((flag, ((tmp & 0x3FFF) as u32)))
395 let val = ((tmp as u32) << 16) | (src.read_u16be()? as u32);
396 Ok((flag, val & 0x3FFFFFFF))
400 fn read_video_buf(src: &mut ByteReader, stream: NAStreamRef, ts: u32, keyframe: bool, frame_size: usize) -> DemuxerResult<NAPacket> {
401 let size = (frame_size as usize) + 9;
402 let mut vec: Vec<u8> = vec![0; size];
403 //v[0] = 0; // 1 slice
405 src.read_buf(&mut vec[9..])?;
407 let ts = stream.make_ts(Some(ts as u64), None, None);
408 Ok(NAPacket::new(stream, ts, keyframe, vec))
411 fn read_multiple_frame(src: &mut ByteReader, stream: NAStreamRef, keyframe: bool, skip_mtype: bool) -> DemuxerResult<NAPacket> {
413 let mtype = src.read_byte()?;
414 validate!(mtype == 0xC0);
416 let (_, frame_size) = read_14or30(src)?;
417 let (_, timestamp) = read_14or30(src)?;
418 let _seq_no = src.read_byte()?;
420 read_video_buf(src, stream, timestamp, keyframe, frame_size as usize)
423 struct RMDemuxCommon {}
426 fn parse_stream_info(str_data: &mut CommonStreamData, strmgr: &mut StreamManager, stream_no: u32, edata: &[u8], duration: u32) -> DemuxerResult<bool> {
427 let mut is_mlti = false;
428 let mut mr = MemoryReader::new_read(edata);
429 let mut src = ByteReader::new(&mut mr);
430 let tag = src.read_u32be()?;
431 let tag2 = src.peek_u32be()?;
432 if tag == mktag!('.', 'r', 'a', 0xFD) {
433 Self::parse_audio_stream(strmgr, &mut str_data.streams, stream_no, &mut src, edata, duration)?;
434 } else if ((tag2 == mktag!('V', 'I', 'D', 'O')) || (tag2 == mktag!('I', 'M', 'A', 'G'))) && ((tag as usize) <= edata.len()) {
435 Self::parse_video_stream(strmgr, &mut str_data.streams, stream_no, &mut src, edata, tag2, duration)?;
436 } else if tag == mktag!(b"LSD:") {
437 let extradata = Some(edata.to_owned());
439 src.read_skip(4)?; //version
440 let channels = src.read_u16be()?;
441 let samp_size = src.read_u16be()?;
442 let sample_rate = src.read_u32be()?;
444 let soniton = NASoniton::new(samp_size as u8, SONITON_FLAG_SIGNED);
445 let ahdr = NAAudioInfo::new(sample_rate, channels as u8, soniton, 1);
446 let nainfo = NACodecInfo::new("ralf", NACodecTypeInfo::Audio(ahdr), extradata);
447 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, stream_no as u32, nainfo, 1, 1000, u64::from(duration)));
448 if res.is_none() { return Err(MemoryError); }
449 let astr = RMAudioStream::new(None);
450 str_data.streams.push(RMStreamType::Audio(astr));
451 } else if tag == mktag!(b"MLTI") {
453 let num_rules = src.read_u16be()? as usize;
455 str_data.mlti_mapper.add_stream(stream_no);
456 for _ in 0..num_rules {
457 let substr = src.read_u16be()?;
458 max_sub = max_sub.max(substr);
459 str_data.mlti_mapper.add_map_rule(substr);
461 let num_substreams = src.read_u16be()? as usize;
462 validate!(num_substreams > (max_sub as usize));
463 for _ in 0..num_substreams {
464 let hdr_size = src.read_u32be()? as usize;
465 validate!(hdr_size > 8);
466 let pos = src.tell() as usize;
467 src.read_skip(hdr_size)?;
468 str_data.mlti_mapper.add_substream();
470 let hdrsrc = &edata[pos..][..hdr_size];
471 let mut mr = MemoryReader::new_read(hdrsrc);
472 let mut hsrc = ByteReader::new(&mut mr);
474 let tag = hsrc.read_u32be()?;
475 let tag2 = hsrc.peek_u32be()?;
476 let stream_no = str_data.mlti_mapper.get_substream_no();
477 //todo check that all substreams are of the same type");
478 if tag == mktag!('.', 'r', 'a', 0xFD) {
479 Self::parse_audio_stream(strmgr, &mut str_data.streams, stream_no, &mut hsrc, hdrsrc, duration)?;
480 } else if (tag2 == mktag!('V', 'I', 'D', 'O')) && ((tag as usize) <= hdr_size) {
481 Self::parse_video_stream(strmgr, &mut str_data.streams, stream_no, &mut hsrc, hdrsrc, tag2, duration)?;
483 println!("unknown MLTI substream {:08X} / {:08X}", tag, tag2);
484 return Err(DemuxerError::InvalidData);
486 str_data.str_ids.push(stream_no);
490 str_data.streams.push(RMStreamType::Logical);
494 fn parse_audio_stream(strmgr: &mut StreamManager, streams: &mut Vec<RMStreamType>, stream_no: u32, src: &mut ByteReader, edata_: &[u8], duration: u32) -> DemuxerResult<()> {
495 let ver = src.read_u16be()?;
496 let ainfo = match ver {
507 println!("unknown version {}", ver);
508 return Err(DemuxerError::InvalidData);
511 let cname = find_codec_name(RM_AUDIO_CODEC_REGISTER, ainfo.fcc);
512 let blk_size = if ainfo.fcc != mktag!(b"sipr") {
513 ainfo.granularity as usize
515 validate!(ainfo.flavor <= 3);
516 RM_SIPRO_BLOCK_SIZES[ainfo.flavor as usize]
518 let srate = ainfo.sample_rate;
519 let soniton = NASoniton::new(ainfo.sample_size as u8, SONITON_FLAG_SIGNED);
520 let ahdr = NAAudioInfo::new(srate, ainfo.channels as u8, soniton, blk_size);
521 let extradata = if ainfo.edata_size == 0 {
524 let eslice = &edata_[(src.tell() as usize)..];
525 Some(eslice.to_vec())
527 let duration = if duration == 0 { ainfo.get_duration(1000) } else { u64::from(duration) };
528 let nainfo = NACodecInfo::new(cname, NACodecTypeInfo::Audio(ahdr), extradata);
529 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, stream_no as u32, nainfo, 1, 1000, duration));
530 if res.is_none() { return Err(MemoryError); }
532 let astr = RMAudioStream::new(ainfo.ileave_info);
533 streams.push(RMStreamType::Audio(astr));
536 #[allow(unused_variables)]
537 fn parse_video_stream(strmgr: &mut StreamManager, streams: &mut Vec<RMStreamType>, stream_no: u32, src: &mut ByteReader, edata_: &[u8], tag2: u32, duration: u32) -> DemuxerResult<()> {
539 let fcc = src.read_u32be()?;
540 let width = src.read_u16be()? as usize;
541 let height = src.read_u16be()? as usize;
542 let bpp = src.read_u16be()?;
543 let pad_w = src.read_u16be()?;
544 let pad_h = src.read_u16be()?;
546 if tag2 == mktag!('V', 'I', 'D', 'O') {
547 _fps = src.read_u32be()?;
551 let extradata: Option<Vec<u8>>;
553 let eslice = &edata_[(src.tell() as usize)..];
554 extradata = Some(eslice.to_vec());
558 let cname = find_codec_name(RM_VIDEO_CODEC_REGISTER, fcc);
560 let vhdr = NAVideoInfo::new(width, height, false, RGB24_FORMAT);
561 let vinfo = NACodecInfo::new(cname, NACodecTypeInfo::Video(vhdr), extradata);
562 let res = strmgr.add_stream(NAStream::new(StreamType::Video, stream_no as u32, vinfo, 1, 1000, u64::from(duration)));
563 if res.is_none() { return Err(DemuxerError::MemoryError); }
565 let vstr = RMVideoStream::new();
566 streams.push(RMStreamType::Video(vstr));
569 #[allow(unused_variables)]
570 fn parse_packet_payload(src: &mut ByteReader, rmstream: &mut RMStreamType, stream: NAStreamRef, slice_buf: &mut Vec<u8>, queued_pkts: &mut Vec<NAPacket>, keyframe: bool, ts: u32, payload_size: usize) -> DemuxerResult<NAPacket> {
572 RMStreamType::Video(ref mut vstr) => {
574 let pos = src.tell();
575 let b0 = src.read_byte()?;
577 0 => { // partial frame
578 let b1 = src.read_byte()?;
579 let hdr1 = ((b0 as u16) << 8) | (b1 as u16);
580 let num_pkts = ((hdr1 >> 7) & 0x7F) as usize;
581 let packet_num = hdr1 & 0x7F;
582 let (_, frame_size) = read_14or30(src)?;
583 let (_, off) = read_14or30(src)?;
584 let seq_no = src.read_byte()?;
585 let hdr_skip = (src.tell() - pos) as usize;
587 let slice_size = (payload_size - hdr_skip) as usize;
588 slice_buf.resize(slice_size, 0);
589 src.read_buf(slice_buf.as_mut_slice())?;
591 vstr.start_slice(num_pkts, frame_size as usize, slice_buf.as_slice());
593 vstr.add_slice(packet_num as usize, slice_buf.as_slice())?;
595 if (packet_num as usize) < num_pkts {
596 return Err(DemuxerError::TryAgain);
598 //todo: check if full frame is received
599 let ts = stream.make_ts(Some(ts as u64), None, None);
600 let pkt = NAPacket::new(stream, ts, keyframe, vstr.get_frame_data());
603 1 => { // whole frame
604 let seq_no = src.read_byte()?;
605 read_video_buf(src, stream, ts, keyframe, payload_size - 2)
607 2 => { // last partial frame
608 let b1 = src.read_byte()?;
609 let hdr1 = ((b0 as u16) << 8) | (b1 as u16);
610 let num_pkts = ((hdr1 >> 7) & 0x7F) as usize;
611 let packet_num = hdr1 & 0x7F;
612 let (_, frame_size) = read_14or30(src)?;
613 let (_, tail_size) = read_14or30(src)?;
614 let seq_no = src.read_byte()?;
615 slice_buf.resize(tail_size as usize, 0);
616 src.read_buf(slice_buf.as_mut_slice())?;
617 if packet_num == 1 && frame_size == tail_size {
618 vstr.start_slice(num_pkts, frame_size as usize, slice_buf.as_slice());
620 vstr.add_slice(packet_num as usize, slice_buf.as_slice())?;
623 while src.tell() < pos + (payload_size as u64) {
624 let res = read_multiple_frame(src, stream.clone(), false, false);
625 if res.is_err() { break; }
626 queued_pkts.push(res.unwrap());
628 queued_pkts.reverse();
629 let ts = stream.make_ts(Some(ts as u64), None, None);
630 let pkt = NAPacket::new(stream, ts, keyframe, vstr.get_frame_data());
633 _ => { // multiple frames
634 let res = read_multiple_frame(src, stream.clone(), keyframe, true);
635 if res.is_err() { return res; }
636 while src.tell() < pos + (payload_size as u64) {
637 let res = read_multiple_frame(src, stream.clone(), false, false);
638 if res.is_err() { break; }
639 queued_pkts.push(res.unwrap());
641 queued_pkts.reverse();
646 RMStreamType::Audio(ref mut astr) => {
647 astr.read_apackets(queued_pkts, src, stream, ts, keyframe, payload_size)
650 src.read_skip(payload_size)?;
651 Err(DemuxerError::InvalidData)
657 impl<'a> DemuxCore<'a> for RealMediaDemuxer<'a> {
658 fn open(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
659 self.read_header(strmgr, seek_idx)?;
663 #[allow(unused_variables)]
664 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
665 if !self.queued_pkts.is_empty() {
666 let pkt = self.queued_pkts.pop().unwrap();
670 if (self.cur_packet >= self.num_packets) || (self.src.tell() >= self.data_end) {
671 self.cur_data_chunk += 1;
672 if self.cur_data_chunk < self.data_chunks.len() {
673 let (pos, _, _) = self.data_chunks[self.cur_data_chunk];
674 self.src.seek(SeekFrom::Start(pos))?;
675 let res = read_chunk(self.src);
676 if let Ok((id, size, ver)) = res {
677 self.data_pos = self.src.tell();
679 self.data_end = self.data_pos + (size as u64);
680 if self.parse_data_start().is_ok() {
685 return Err(DemuxerError::EOF);
688 let pkt_start = self.src.tell();
689 let ver = self.src.read_u16be()?;
691 let len = self.src.read_u16be()? as usize;
692 let str_no = self.src.read_u16be()? as u32;
693 let ts = self.src.read_u32be()?;
697 pkt_grp = self.src.read_byte()? as u16;
698 flags = self.src.read_byte()?;
700 //asm_rule = self.src.read_u16be()?;
701 //asm_flags = self.src.read_byte()?;
702 self.src.read_skip(2)?;
704 self.src.read_skip(1)?;
707 let hdr_size = self.src.tell() - pkt_start;
708 //println!("packet @{:X} size {} for {} ts {} grp {} flags {:X}", pkt_start, len, str_no, ts, pkt_grp, flags);
709 self.cur_packet += 1;
711 let payload_size = len - (hdr_size as usize);
713 let stream_id = self.str_data.get_stream_id(str_no as u32, pkt_grp);
714 let sr = self.str_data.find_stream(stream_id);
716 self.src.read_skip(payload_size)?;
717 return Err(DemuxerError::InvalidData);
719 let str_id = sr.unwrap();
721 let streamres = strmgr.get_stream_by_id(stream_id);
722 if streamres.is_none() {
723 self.src.read_skip(payload_size)?;
726 let stream = streamres.unwrap();
727 if strmgr.is_ignored_id(stream_id) {
728 self.src.read_skip(payload_size)?;
731 //todo skip unwanted packet
732 let keyframe = (flags & KEYFRAME_FLAG) != 0;
734 let ret = RMDemuxCommon::parse_packet_payload(self.src, &mut self.str_data.streams[str_id], stream, &mut self.slice_buf, &mut self.queued_pkts, keyframe, ts, payload_size);
735 if let Err(DemuxerError::TryAgain) = ret {
743 #[allow(unused_variables)]
744 fn seek(&mut self, time: NATimePoint, seek_idx: &SeekIndex) -> DemuxerResult<()> {
745 self.queued_pkts.clear();
746 let ret = seek_idx.find_pos(time);
748 return Err(DemuxerError::SeekError);
750 let ret = ret.unwrap();
751 let seek_pos = ret.pos;
752 for (pos, size, ver) in self.data_chunks.iter() {
753 if seek_pos < *pos { continue; }
754 let end = *pos + (*size as u64);
757 self.data_pos = seek_pos;
758 self.data_ver = *ver;
760 self.src.seek(SeekFrom::Start(seek_pos))?;
764 Err(DemuxerError::SeekError)
767 fn get_duration(&self) -> u64 { 0 }
770 impl<'a> NAOptionHandler for RealMediaDemuxer<'a> {
771 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
772 fn set_options(&mut self, _options: &[NAOption]) { }
773 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
776 fn read_chunk(src: &mut ByteReader) -> DemuxerResult<(u32, u32, u16)> {
777 let id = src.read_u32be()?;
778 if id == 0 { return Ok((0, 0, 0)); }
779 let size = src.read_u32be()?;
781 let ver = src.read_u16be()?;
783 return Ok((id, 0x0FFFFFFF, ver));
785 validate!(size >= 10);
786 let ver = src.read_u16be()?;
791 #[derive(Clone,Copy,Debug)]
792 struct InterleaveInfo {
799 #[derive(Clone,Copy,Debug)]
801 struct RealAudioInfo {
809 bytes_per_minute: u32,
812 ileave_info: Option<InterleaveInfo>
816 fn get_duration(&self, base: u32) -> u64 {
817 if self.bytes_per_minute != 0 {
818 u64::from(self.total_bytes) * 60 * u64::from(base) / u64::from(self.bytes_per_minute)
825 fn skip_ra_metadata(src: &mut ByteReader) -> DemuxerResult<()> {
826 let title_len = src.read_byte()? as usize;
827 src.read_skip(title_len)?;
828 let author_len = src.read_byte()? as usize;
829 src.read_skip(author_len)?;
830 let copywrong_len = src.read_byte()? as usize;
831 src.read_skip(copywrong_len)?;
832 let comment_len = src.read_byte()? as usize;
833 src.read_skip(comment_len)?;
837 #[allow(unused_variables)]
838 fn parse_aformat3(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
839 let start = src.tell();
840 let header_len = src.read_u16be()?;
841 validate!(header_len >= 24);
842 let flavor = src.read_u16be()?;
843 let granularity = src.read_u32be()?;
844 let bytes_per_minute = src.read_u32be()?;
845 let total_bytes = src.read_u32be()?;
847 skip_ra_metadata(src)?;
849 let _can_copy = src.read_byte()?;
850 let fcc_len = src.read_byte()?;
851 validate!(fcc_len == 4);
852 let fcc = src.read_u32be()?;
854 let end = src.tell();
855 validate!(end - start <= (header_len as u64) + 2);
859 sample_rate: 8000, sample_size: 16, channels: 1, channel_mask: 0,
860 granularity, bytes_per_minute,
861 total_bytes, edata_size: 0,
866 #[allow(unused_variables)]
867 fn parse_aformat4(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
868 let start = src.tell();
869 src.read_skip(2)?; // zeroes
870 let id = src.read_u32be()?;
871 validate!(id == mktag!(b".ra4"));
872 let data_size = src.read_u32be()?;
873 let _ver4 = src.read_u16be()?; // should be 4
874 let header_size = src.read_u32be()?;
875 let flavor = src.read_u16be()?;
876 let granularity = src.read_u32be()?;
877 let total_bytes = src.read_u32be()?;
878 let bytes_per_minute = src.read_u32be()?;
879 let _bytes_per_minute2 = src.read_u32be()?;
880 let ileave_factor = src.read_u16be()?;
881 let ileave_block_size = src.read_u16be()?;
882 let _user_data = src.read_u16be()?;
883 let sample_rate = src.read_u32be()?;
884 let sample_size = src.read_u32be()?;
885 let channels = src.read_u16be()?;
886 let interleaver_id_len = src.read_byte()?;
887 validate!(interleaver_id_len == 4);
888 let interleaver_id = src.read_u32be()?;
889 let fcc_len = src.read_byte()?;
890 validate!(fcc_len == 4);
891 let fcc = src.read_u32be()?;
892 let is_interleaved = src.read_byte()?;
893 let _can_copy = src.read_byte()?;
894 let _stream_type = src.read_byte()?;
896 skip_ra_metadata(src)?;
898 let end = src.tell();
899 validate!(end - start <= (header_size as u64) + 10);
901 let ileave_info = if is_interleaved != 0 {
902 Some(InterleaveInfo {
903 id: interleaver_id, factor: ileave_factor, block_size: granularity as u16,
904 frame_size: ileave_block_size,
912 sample_rate, sample_size: sample_size as u16, channels, channel_mask: 0,
913 granularity, bytes_per_minute,
914 total_bytes: total_bytes & 0xFFFFFF, edata_size: 0,
919 #[allow(unused_variables)]
920 fn parse_aformat5(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
921 let start = src.tell();
922 src.read_skip(2)?; // zeroes
923 let id = src.read_u32be()?;
924 validate!((id == mktag!(b".ra5")) || (id == mktag!(b".ra4")));
925 let data_size = src.read_u32be()?;
926 let _ver5 = src.read_u16be()?; // should be 5
927 let header_size = src.read_u32be()?;
928 let flavor = src.read_u16be()?;
929 let granularity = src.read_u32be()?;
930 let total_bytes = src.read_u32be()?;
931 let bytes_per_minute = src.read_u32be()?;
932 let _bytes_per_minute2 = src.read_u32be()?;
933 let ileave_factor = src.read_u16be()?;
934 let frame_size = src.read_u16be()?;
935 let ileave_block_size = src.read_u16be()?;
936 let user_data = src.read_u32be()?;
937 let _sample_rate1 = src.read_u16be()?;
938 let sample_rate = src.read_u32be()?;
939 let sample_size = src.read_u32be()?;
940 let channels = src.read_u16be()?;
941 let interleaver_id = src.read_u32be()?;
942 let fcc = src.read_u32be()?;
943 let is_interleaved = src.read_byte()?;
944 let _can_copy = src.read_byte()?;
945 let _stream_type = src.read_byte()?;
946 let has_ileave_pattern = src.read_byte()?;
947 if has_ileave_pattern != 0 {
948 unimplemented!("ra5 interleave pattern");
950 let mut edata_size = src.read_u32be()?;
951 let end = src.tell();
952 if id == mktag!(b".ra5") {
953 validate!(end - start <= (header_size as u64) + 10);
954 // src.read_skip(((end - start) as usize) - (header_size as usize) - 10)?;
956 validate!(end - start <= (header_size as u64) + 15);
959 let ileave_info = if is_interleaved != 0 {
960 Some(InterleaveInfo {
961 id: interleaver_id, factor: ileave_factor, block_size: ileave_block_size, frame_size,
966 if (fcc == mktag!(b"raac")) || (fcc == mktag!(b"racp")) {
967 validate!(edata_size > 1);
974 sample_rate, sample_size: sample_size as u16, channels, channel_mask: 0,
975 granularity, bytes_per_minute,
976 total_bytes: total_bytes & 0xFFFFFF, edata_size,
981 const RMVB_HDR_SIZE: u32 = 18;
982 const RMVB_PROP_SIZE: u32 = 50;
983 const KEYFRAME_FLAG: u8 = 0x02;
985 impl<'a> RealMediaDemuxer<'a> {
986 fn new(io: &'a mut ByteReader<'a>) -> Self {
995 str_data: CommonStreamData::new(),
996 data_chunks: Vec::new(),
998 queued_pkts: Vec::new(),
999 slice_buf: Vec::new(),
1002 #[allow(unused_variables)]
1003 fn read_header(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
1004 let (id, size, ver) = read_chunk(self.src)?;
1005 validate!((id == mktag!(b".RMF")) || (id == mktag!(b".RMP")));
1006 validate!(size >= RMVB_HDR_SIZE);
1007 let fver = self.src.read_u32be()?;
1008 validate!(fver <= 1);
1009 let num_hdr = self.src.read_u32be()? as usize;
1010 validate!(num_hdr >= 1);
1011 if size > RMVB_HDR_SIZE {
1012 self.src.read_skip((size - RMVB_HDR_SIZE) as usize)?;
1015 let (id, size, ver) = read_chunk(self.src)?;
1016 let prop_size = if ver == 0 { RMVB_PROP_SIZE } else { RMVB_PROP_SIZE + 4 };
1017 validate!(size >= prop_size);
1018 validate!((ver == 0) || (ver == 2));
1019 let maxbr = self.src.read_u32be()?;
1020 let avgbr = self.src.read_u32be()?;
1021 let maxps = self.src.read_u32be()?;
1022 let avgps = self.src.read_u32be()?;
1023 let num_pkt = self.src.read_u32be()? as usize;
1024 let duration = self.src.read_u32be()?;
1025 let preroll = self.src.read_u32be()?;
1026 let idx_off = self.src.read_size(ver)?;
1027 let data_off = self.src.read_u32be()?;
1028 let num_streams = self.src.read_u16be()? as usize;
1029 let flags = self.src.read_u16be()?;
1030 if size > prop_size {
1031 self.src.read_skip((size - prop_size) as usize)?;
1034 for _ in 0..num_hdr {
1035 if self.src.is_eof() {
1039 let res = self.parse_chunk(strmgr, seek_idx);
1041 Ok(last) => { if last { break; } },
1042 Err(DemuxerError::IOError) => { break; },
1044 if self.data_chunks.is_empty() { // data is not found, report error
1050 validate!(!self.data_chunks.is_empty());
1051 self.cur_data_chunk = 0;
1052 let (pos, size, ver) = self.data_chunks[self.cur_data_chunk];
1053 self.data_pos = pos;
1054 self.data_ver = ver;
1055 self.data_end = pos + (size as u64);
1056 self.src.seek(SeekFrom::Start(self.data_pos + 10))?;
1057 self.parse_data_start()?;
1060 fn parse_data_start(&mut self) -> DemuxerResult<()> {
1061 let num_packets = self.src.read_u32be()?;
1062 if self.data_ver == 2 {
1063 self.src.read_skip(12)?; // zeroes?
1065 let next_data_hdr = self.src.read_u32be()?;
1066 self.num_packets = if num_packets > 0 { num_packets } else { 0xFFFFFF };
1067 self.cur_packet = 0;
1068 self.next_data = next_data_hdr as u64;
1071 fn parse_chunk(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<bool> {
1072 let (id, size, ver) = read_chunk(self.src)?;
1073 let end_pos = self.src.tell() - 10 + (size as u64);
1075 validate!((ver == 0) || (ver == 2));
1076 if id == mktag!(b"CONT") { self.parse_content_desc()?; }
1077 else if id == mktag!(b"MDPR") { self.parse_mdpr(strmgr)?; }
1078 else if id == mktag!(b"DATA") {
1079 self.data_chunks.push((self.src.tell() - 10, size, ver));
1081 else if id == mktag!(b"INDX") {
1082 if !seek_idx.skip_index {
1083 self.parse_index(seek_idx, (size as usize) - 10, ver)?;
1086 else if id == 0 { return Ok(true); }
1087 else { println!("unknown chunk type {:08X}", id); }
1089 let cpos = self.src.tell();
1091 self.src.read_skip((end_pos - cpos) as usize)?;
1095 #[allow(unused_variables)]
1096 fn parse_content_desc(&mut self) -> DemuxerResult<()> {
1097 let title_len = self.src.read_u16be()? as usize;
1098 self.src.read_skip(title_len)?;
1099 let author_len = self.src.read_u16be()? as usize;
1100 self.src.read_skip(author_len)?;
1101 let copywrong_len = self.src.read_u16be()? as usize;
1102 self.src.read_skip(copywrong_len)?;
1103 let comment_len = self.src.read_u16be()? as usize;
1104 self.src.read_skip(comment_len)?;
1107 #[allow(unused_variables)]
1108 fn parse_mdpr(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<()> {
1109 let stream_no = self.src.read_u16be()? as u32;
1110 //todo check stream_no for duplicates
1111 let maxbr = self.src.read_u32be()?;
1112 let avgbr = self.src.read_u32be()?;
1113 let maxps = self.src.read_u32be()?;
1114 let avgps = self.src.read_u32be()?;
1115 let start = self.src.read_u32be()?;
1116 let preroll = self.src.read_u32be()?;
1117 let duration = self.src.read_u32be()?;
1118 let sname_size = self.src.read_byte()? as usize;
1119 let sname = read_string_size(self.src, sname_size)?;
1120 let mime_size = self.src.read_byte()? as usize;
1121 let mime = read_string_size(self.src, mime_size)?;
1122 let edata_size = self.src.read_u32be()? as usize;
1123 let edata: Option<Vec<u8>> = if edata_size == 0 { None } else {
1124 let mut edvec: Vec<u8> = vec![0; edata_size];
1125 self.src.read_buf(&mut edvec)?;
1128 let mut is_mlti = false;
1130 if let Some(edata_) = edata {
1131 is_mlti = RMDemuxCommon::parse_stream_info(&mut self.str_data, strmgr, stream_no, &edata_, duration)?;
1134 self.str_data.streams.push(RMStreamType::Unknown);
1137 self.str_data.str_ids.push(stream_no);
1142 fn parse_index(&mut self, seek_idx: &mut SeekIndex, chunk_size: usize, ver: u16) -> DemuxerResult<()> {
1143 if ver != 0 && ver != 2 { return Ok(()); }
1144 let num_entries = self.src.read_u32be()? as usize;
1145 let str_id = self.src.read_u16be()? as u32;
1146 let _next_idx = self.src.read_size(ver)?;
1148 validate!(chunk_size == num_entries * 14 + 10);
1150 validate!(chunk_size == num_entries * 18 + 14);
1152 if num_entries == 0 { return Ok(()); }
1154 seek_idx.add_stream(str_id);
1155 let idx = seek_idx.get_stream_index(str_id).unwrap();
1156 for _ in 0..num_entries {
1157 let iver = self.src.read_u16be()?;
1158 validate!(iver == ver);
1159 let ts = self.src.read_u32be()? as u64;
1160 let pos = self.src.read_size(ver)?;
1161 let _pkt_no = self.src.read_u32be()?;
1162 idx.add_entry(SeekEntry { time: ts, pts: 0, pos });
1165 seek_idx.mode = SeekIndexMode::Present;
1170 fn read_string_size(src: &mut ByteReader, size: usize) -> DemuxerResult<String> {
1171 let mut vec: Vec<u8> = Vec::new();
1173 let c = src.read_byte()?;
1176 if let Ok(res) = String::from_utf8(vec) {
1183 struct RealAudioDemuxer<'a> {
1184 src: &'a mut ByteReader<'a>,
1185 stream: Option<RMAudioStream>,
1189 queued_pkts: Vec<NAPacket>,
1192 impl<'a> DemuxCore<'a> for RealAudioDemuxer<'a> {
1193 #[allow(unused_variables)]
1194 fn open(&mut self, strmgr: &mut StreamManager, _seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
1195 let magic = self.src.read_u32be()?;
1196 validate!(magic == mktag!(b".ra\xFD"));
1197 let ver = self.src.read_u16be()?;
1198 let ainfo = match ver {
1200 parse_aformat3(self.src)?
1203 parse_aformat4(self.src)?
1206 parse_aformat5(self.src)?
1209 println!("unknown version {}", ver);
1210 return Err(DemuxerError::InvalidData);
1213 let cname = find_codec_name(RM_AUDIO_CODEC_REGISTER, ainfo.fcc);
1214 let blk_size = if ainfo.fcc != mktag!(b"sipr") {
1215 ainfo.granularity as usize
1217 validate!(ainfo.flavor <= 3);
1218 RM_SIPRO_BLOCK_SIZES[ainfo.flavor as usize]
1220 let srate = ainfo.sample_rate;
1221 let soniton = NASoniton::new(ainfo.sample_size as u8, SONITON_FLAG_SIGNED);
1222 let ahdr = NAAudioInfo::new(srate, ainfo.channels as u8, soniton, blk_size);
1223 let extradata = if ainfo.edata_size == 0 {
1226 let mut dta: Vec<u8> = vec![0; ainfo.edata_size as usize];
1227 self.src.read_buf(dta.as_mut_slice())?;
1230 let nainfo = NACodecInfo::new(cname, NACodecTypeInfo::Audio(ahdr), extradata);
1231 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, 0, nainfo, 1, srate, ainfo.get_duration(ainfo.sample_rate)));
1232 if res.is_none() { return Err(MemoryError); }
1234 let astr = RMAudioStream::new(ainfo.ileave_info);
1235 self.data_start = self.src.tell();
1236 self.data_end = if ainfo.total_bytes > 0 { self.src.tell() + (ainfo.total_bytes as u64) } else { 0 };
1237 self.blk_size = blk_size;
1238 self.stream = Some(astr);
1243 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
1244 if !self.queued_pkts.is_empty() {
1245 let pkt = self.queued_pkts.pop().unwrap();
1248 if (self.data_end != 0) && (self.src.tell() >= self.data_end) {
1249 return Err(DemuxerError::EOF);
1251 let streamres = strmgr.get_stream_by_id(0);
1252 let stream = streamres.unwrap();
1253 if let Some(ref mut astr) = self.stream {
1255 let ret = astr.read_apackets(&mut self.queued_pkts, self.src, stream.clone(), 0, false, self.blk_size);
1256 if let Err(DemuxerError::TryAgain) = ret {
1262 Err(DemuxerError::NoSuchInput)
1265 #[allow(unused_variables)]
1266 fn seek(&mut self, time: NATimePoint, seek_idx: &SeekIndex) -> DemuxerResult<()> {
1270 fn get_duration(&self) -> u64 { 0 }
1273 impl<'a> NAOptionHandler for RealAudioDemuxer<'a> {
1274 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
1275 fn set_options(&mut self, _options: &[NAOption]) { }
1276 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
1279 impl<'a> RealAudioDemuxer<'a> {
1280 fn new(io: &'a mut ByteReader<'a>) -> Self {
1287 queued_pkts: Vec::new(),
1294 StreamProperties(usize),
1295 Packet { ts: u32, strm: u32, flags: u32, len: usize, checksum: u32 },
1296 IntValue(Vec<u8>, u32),
1297 BinaryData(Vec<u8>, Vec<u8>),
1298 StringData(Vec<u8>, Vec<u8>),
1305 fn read_string(src: &mut ByteReader) -> DemuxerResult<Vec<u8>> {
1306 let len = src.read_u32be()? as usize;
1307 let mut val = vec![0; len];
1308 src.read_buf(val.as_mut_slice())?;
1312 fn read(src: &mut ByteReader) -> DemuxerResult<Self> {
1313 let code = src.read_byte()?;
1316 let val = src.read_u32be()? as usize;
1317 Ok(IVRRecord::StreamProperties(val))
1320 let ts = src.read_u32be()?;
1321 let strm = src.read_u16be()? as u32;
1322 let flags = src.read_u32be()?;
1323 let len = src.read_u32be()? as usize;
1324 let chk = src.read_u32be()?;
1325 validate!((len > 0) && (len < (1 << 24)));
1326 Ok(IVRRecord::Packet { ts, strm, flags, len, checksum: chk })
1329 let name = Self::read_string(src)?;
1330 let len = src.read_u32be()?;
1331 validate!(len == 4);
1332 let val = src.read_u32be()?;
1333 Ok(IVRRecord::IntValue(name, val))
1336 let name = Self::read_string(src)?;
1337 let len = src.read_u32be()? as usize;
1338 let mut val = vec![0; len];
1339 src.read_buf(val.as_mut_slice())?;
1340 Ok(IVRRecord::BinaryData(name, val))
1343 let name = Self::read_string(src)?;
1344 let val = Self::read_string(src)?;
1345 Ok(IVRRecord::StringData(name, val))
1347 6 => Ok(IVRRecord::HeaderEnd),
1349 src.read_skip(8)?; // always zero?
1350 Ok(IVRRecord::DataEnd)
1353 src.read_skip(8)?; // always zero?
1354 Ok(IVRRecord::DataStart)
1356 _ => Ok(IVRRecord::Invalid(code)),
1359 fn is_data_start(&self) -> bool {
1360 matches!(*self, IVRRecord::DataStart)
1364 impl fmt::Display for IVRRecord {
1365 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
1367 IVRRecord::Invalid(typ) => write!(f, "Invalid({:02X})", typ),
1368 IVRRecord::StreamProperties(num) =>
1369 write!(f, "({} stream properties)", num),
1370 IVRRecord::Packet { ts, strm, flags, len, checksum } =>
1371 write!(f, "paket({}, {}, {:X}, {}, {})", ts, strm, flags, len, checksum),
1372 IVRRecord::IntValue(ref name, val) =>
1373 write!(f, "({} = {})", String::from_utf8_lossy(name), val),
1374 IVRRecord::BinaryData(ref name, ref val) =>
1375 write!(f, "({} = {} bytes)", String::from_utf8_lossy(name), val.len()),
1376 IVRRecord::StringData(ref name, ref val) =>
1377 write!(f, "({} = {})", String::from_utf8_lossy(name), String::from_utf8_lossy(val)),
1378 IVRRecord::HeaderEnd => write!(f, "header end"),
1379 IVRRecord::DataEnd => write!(f, "data end"),
1380 IVRRecord::DataStart => write!(f, "data start"),
1385 struct RecordDemuxer {
1389 remap_ids: Vec<u32>,
1392 impl RecordDemuxer {
1393 fn new(pos: u64, start_str: u32) -> Self {
1398 remap_ids: Vec::new(),
1401 fn parse_header(&mut self, src: &mut ByteReader, strmgr: &mut StreamManager, str_data: &mut CommonStreamData) -> DemuxerResult<()> {
1402 src.seek(SeekFrom::Start(self.cur_pos))?;
1403 let magic = src.read_u32be()?;
1404 validate!(magic == mktag!(b".REC"));
1405 let _smth = src.read_byte()?;
1406 let num_entries = src.read_u32be()? as usize;
1407 for _ in 0..num_entries {
1408 let _rec = IVRRecord::read(src)?;
1410 let mut has_seek_table = false;
1411 let mut cur_str_no = 0;
1413 let rec = IVRRecord::read(src)?;
1415 IVRRecord::HeaderEnd => { break; },
1416 IVRRecord::StreamProperties(num) => {
1417 let stream_no = cur_str_no + self.start_str;
1419 let mut parsed = false;
1420 let mut real_stream_no = 0;
1421 let mut duration = 0;
1423 let rec = IVRRecord::read(src)?;
1425 IVRRecord::IntValue(ref name, val) => {
1426 if name == b"StreamNumber\0" {
1427 real_stream_no = val;
1429 if name == b"Duration\0" {
1433 IVRRecord::BinaryData(ref name, ref val) => {
1434 if name == b"OpaqueData\0" {
1436 let is_mlti = RMDemuxCommon::parse_stream_info(str_data, strmgr, stream_no, val, duration)?;
1438 str_data.str_ids.push(stream_no);
1443 IVRRecord::StringData(ref name, ref val) => {
1444 if (name == b"SeekType\0") && (val != b"None\0") {
1445 has_seek_table = true;
1448 _ => { return Err(DemuxerError::InvalidData); }
1452 str_data.streams.push(RMStreamType::Unknown);
1453 str_data.str_ids.push(stream_no);
1455 self.remap_ids.push(real_stream_no);
1457 _ => {println!(" unexpected {}", rec); return Err(DemuxerError::InvalidData); }
1460 let off0 = src.read_u32be()? as u64;
1461 let _off1 = src.read_u32be()?;
1462 let _off2 = src.read_u32be()?;
1463 validate!(off0 + self.start_pos == src.tell());
1466 let data_off = src.read_u32be()? as u64;
1467 let pos = src.tell();
1468 validate!(data_off + self.start_pos > pos);
1469 src.read_skip((data_off + self.start_pos - pos) as usize)?;
1470 let rec = IVRRecord::read(src)?;
1471 validate!(rec.is_data_start());
1473 let ntype = src.peek_byte()?;
1474 validate!((ntype == 2) || (ntype == 7)); // packet or data end, no start
1477 self.cur_pos = src.tell();
1481 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> {
1482 src.seek(SeekFrom::Start(self.cur_pos))?;
1484 let rec = IVRRecord::read(src)?;
1486 IVRRecord::Packet { ts, strm, flags, len, .. } => {
1487 let payload_size = len;
1488 let sr = self.remap_ids.iter().position(|x| *x == strm);
1489 validate!(sr.is_some());
1490 let str_no = self.start_str + (sr.unwrap() as u32);
1491 let pkt_grp = ((flags >> 8) & 0xFF) as u16;
1492 let stream_id = str_data.get_stream_id(str_no as u32, pkt_grp);
1493 let sr = str_data.find_stream(stream_id);
1495 src.read_skip(payload_size)?;
1496 return Err(DemuxerError::InvalidData);
1498 let str_id = sr.unwrap();
1500 let streamres = strmgr.get_stream_by_id(stream_id);
1501 if streamres.is_none() {
1502 src.read_skip(payload_size)?;
1505 let stream = streamres.unwrap();
1506 if strmgr.is_ignored_id(stream_id) {
1507 src.read_skip(payload_size)?;
1510 let keyframe = false;
1511 let ret = RMDemuxCommon::parse_packet_payload(src, &mut str_data.streams[str_id], stream, slice_buf, queued_pkts, keyframe, ts, payload_size);
1512 if let Err(DemuxerError::TryAgain) = ret {
1515 self.cur_pos = src.tell();
1519 IVRRecord::DataEnd => return Err(DemuxerError::EOF),
1520 _ => return Err(DemuxerError::InvalidData),
1526 struct RealIVRDemuxer<'a> {
1527 src: &'a mut ByteReader<'a>,
1528 recs: Vec<RecordDemuxer>,
1530 queued_pkts: Vec<NAPacket>,
1532 str_data: CommonStreamData,
1535 impl<'a> DemuxCore<'a> for RealIVRDemuxer<'a> {
1536 #[allow(unused_variables)]
1537 fn open(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
1538 let magic = self.src.peek_u32be()?;
1539 if magic == mktag!(b".REC") {
1540 let mut rec = RecordDemuxer::new(0, 0);
1541 rec.parse_header(self.src, strmgr, &mut self.str_data)?;
1542 self.recs.push(rec);
1543 } else if magic == mktag!(b".R1M") {
1544 self.src.read_skip(4)?; // magic
1545 self.src.read_skip(3)?; // always 0, 1, 1 ?
1546 let _name = IVRRecord::read_string(self.src)?;
1547 self.src.read_skip(1)?; // always 0?
1548 let len1 = self.src.read_u32be()? as u64;
1549 let off1 = self.src.read_u64be()?;
1550 let cpos = self.src.tell();
1551 validate!(off1 == len1 + cpos - 8);
1552 self.src.read_skip((off1 - cpos) as usize)?;
1554 let typ = self.src.read_byte()?;
1557 let len = self.src.read_u32be()?;
1558 self.src.read_skip(len as usize)?;
1561 let len = self.src.read_u32be()? as u64;
1562 let pos = self.src.tell();
1564 let num_streams = self.str_data.streams.len() as u32;
1565 let mut rec = RecordDemuxer::new(pos + 12, num_streams);
1566 rec.parse_header(self.src, strmgr, &mut self.str_data)?;
1567 self.recs.push(rec);
1569 self.src.seek(SeekFrom::Start(pos + len))?;
1572 let mut buf: [u8; 2] = [0; 2];
1573 self.src.peek_buf(&mut buf)?;
1574 if (buf[0] == b'J') && (buf[1] == b'M') { // RJMx markers at the end of file
1577 return Err(DemuxerError::InvalidData);
1580 _ => { return Err(DemuxerError::InvalidData); },
1584 return Err(DemuxerError::InvalidData);
1590 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
1591 if !self.queued_pkts.is_empty() {
1592 let pkt = self.queued_pkts.pop().unwrap();
1596 if self.cur_rec >= self.recs.len() { return Err(DemuxerError::EOF); }
1597 let res = self.recs[self.cur_rec].get_packet(self.src, &mut self.str_data, strmgr, &mut self.queued_pkts, &mut self.slice_buf);
1598 if let Err(DemuxerError::EOF) = res {
1606 #[allow(unused_variables)]
1607 fn seek(&mut self, time: NATimePoint, seek_idx: &SeekIndex) -> DemuxerResult<()> {
1611 fn get_duration(&self) -> u64 { 0 }
1614 impl<'a> NAOptionHandler for RealIVRDemuxer<'a> {
1615 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
1616 fn set_options(&mut self, _options: &[NAOption]) { }
1617 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
1620 impl<'a> RealIVRDemuxer<'a> {
1621 fn new(io: &'a mut ByteReader<'a>) -> Self {
1626 queued_pkts: Vec::new(),
1627 slice_buf: Vec::new(),
1628 str_data: CommonStreamData::new(),
1633 static RM_VIDEO_CODEC_REGISTER: &[(&[u8;4], &str)] = &[
1634 (b"RV10", "realvideo1"),
1635 (b"RV20", "realvideo2"),
1636 (b"RVTR", "realvideo2"),
1637 (b"RV30", "realvideo3"),
1638 (b"RV40", "realvideo4"),
1639 (b"RV60", "realvideo6"),
1640 (b"CLV1", "clearvideo_rm"),
1643 static RM_AUDIO_CODEC_REGISTER: &[(&[u8;4], &str)] = &[
1644 (b"lpcJ", "ra14.4"),
1645 (b"28_8", "ra28.8"),
1649 (b"atrc", "atrac3"),
1655 pub struct RealMediaDemuxerCreator { }
1657 impl DemuxerCreator for RealMediaDemuxerCreator {
1658 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
1659 Box::new(RealMediaDemuxer::new(br))
1661 fn get_name(&self) -> &'static str { "realmedia" }
1664 pub struct RealAudioDemuxerCreator { }
1666 impl DemuxerCreator for RealAudioDemuxerCreator {
1667 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
1668 Box::new(RealAudioDemuxer::new(br))
1670 fn get_name(&self) -> &'static str { "realaudio" }
1673 pub struct RealIVRDemuxerCreator { }
1675 impl DemuxerCreator for RealIVRDemuxerCreator {
1676 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
1677 Box::new(RealIVRDemuxer::new(br))
1679 fn get_name(&self) -> &'static str { "real_ivr" }
1688 fn test_rm_demux() {
1689 // sample from a private collection
1691 File::open("assets/RV/rv10_dnet_640x352_realvideo_encoder_4.0.rm").unwrap();
1692 // File::open("assets/RV/rv20_cook_640x352_realproducer_plus_8.51.rm").unwrap();
1693 // File::open("assets/RV/rv20_svt_atrc_640x352_realproducer_plus_8.51.rm").unwrap();
1694 // File::open("assets/RV/rv30_atrc_384x208_realproducer_plus_8.51.rm").unwrap();
1695 // File::open("assets/RV/rv30_chroma_drift.rm").unwrap();
1696 // File::open("assets/RV/rv30_weighted_mc.rm").unwrap();
1697 // File::open("assets/RV/rv40_weighted_mc.rmvb").unwrap();
1698 // File::open("assets/RV/rv40_weighted_mc_2.rmvb").unwrap();
1699 // File::open("assets/RV/clv1_sipr_384x208_realvideo_encoder_4.0.rm").unwrap();
1700 // File::open("assets/RV/luckynight.rmvb").unwrap();
1701 // File::open("assets/RV/rv40_ralf.rmvb").unwrap();
1702 let mut fr = FileReader::new_read(&mut file);
1703 let mut br = ByteReader::new(&mut fr);
1704 let mut dmx = RealMediaDemuxer::new(&mut br);
1705 let mut sm = StreamManager::new();
1706 let mut si = SeekIndex::new();
1707 dmx.open(&mut sm, &mut si).unwrap();
1710 let pktres = dmx.get_frame(&mut sm);
1711 if let Err(e) = pktres {
1712 if e == DemuxerError::EOF { break; }
1715 let pkt = pktres.unwrap();
1716 println!("Got {}", pkt);
1718 //panic!("the end");
1721 fn test_ra_demux() {
1722 // sample: https://samples.mplayerhq.hu/real//RA/ra_with_comment_field/diemusik.ra
1724 // File::open("assets/RV/welcome288.ra").unwrap();
1725 File::open("assets/RV/diemusik.ra").unwrap();
1726 let mut fr = FileReader::new_read(&mut file);
1727 let mut br = ByteReader::new(&mut fr);
1728 let mut dmx = RealAudioDemuxer::new(&mut br);
1729 let mut sm = StreamManager::new();
1730 let mut si = SeekIndex::new();
1731 dmx.open(&mut sm, &mut si).unwrap();
1734 let pktres = dmx.get_frame(&mut sm);
1735 if let Err(e) = pktres {
1736 if e == DemuxerError::EOF { break; }
1739 let pkt = pktres.unwrap();
1740 println!("Got {}", pkt);
1744 fn test_ivr_demux() {
1745 // sample: https://samples.mplayerhq.hu/real/ivr/Opener_rm_hi.ivr
1747 File::open("assets/RV/Opener_rm_hi.ivr").unwrap();
1748 // File::open("assets/RV/SherwinWilliamsCommercial.ivr").unwrap();
1749 let mut fr = FileReader::new_read(&mut file);
1750 let mut br = ByteReader::new(&mut fr);
1751 let mut dmx = RealIVRDemuxer::new(&mut br);
1752 let mut sm = StreamManager::new();
1753 let mut si = SeekIndex::new();
1754 dmx.open(&mut sm, &mut si).unwrap();
1757 let pktres = dmx.get_frame(&mut sm);
1758 if let Err(e) = pktres {
1759 if e == DemuxerError::EOF { break; }
1762 let pkt = pktres.unwrap();
1763 println!("Got {}", pkt);