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)
16 const RM_SIPRO_BLOCK_SIZES: [usize; 4] = [ 29, 19, 37, 20 ];
17 const RM_SIPRO_SWAPS: [[u8; 2]; 38] = [
18 [ 0, 63 ], [ 1, 22 ], [ 2, 44 ], [ 3, 90 ],
19 [ 5, 81 ], [ 7, 31 ], [ 8, 86 ], [ 9, 58 ],
20 [ 10, 36 ], [ 12, 68 ], [ 13, 39 ], [ 14, 73 ],
21 [ 15, 53 ], [ 16, 69 ], [ 17, 57 ], [ 19, 88 ],
22 [ 20, 34 ], [ 21, 71 ], [ 24, 46 ], [ 25, 94 ],
23 [ 26, 54 ], [ 28, 75 ], [ 29, 50 ], [ 32, 70 ],
24 [ 33, 92 ], [ 35, 74 ], [ 38, 85 ], [ 40, 56 ],
25 [ 42, 87 ], [ 43, 65 ], [ 45, 59 ], [ 48, 79 ],
26 [ 49, 93 ], [ 51, 89 ], [ 55, 95 ], [ 61, 76 ],
27 [ 67, 83 ], [ 77, 80 ]
30 struct RMVideoStream {
47 self.frame.truncate(0);
51 fn start_slice(&mut self, num_slices: usize, frame_size: usize, data: &[u8]) {
52 self.hdr_size = num_slices * 8 + 1;
53 self.frame.resize(frame_size + self.hdr_size, 0);
54 self.frame[0] = (num_slices - 1) as u8;
56 self.add_slice(1, data);
58 fn add_slice(&mut self, slice_no: usize, data: &[u8]) {
59 self.write_slice_info(slice_no);
60 let dslice = &mut self.frame[self.hdr_size + self.frame_pos..][..data.len()];
61 dslice.copy_from_slice(data);
62 self.frame_pos += data.len();
64 fn write_slice_info(&mut self, slice_no: usize) {
65 let off = 1 + (slice_no - 1) * 8;
66 self.frame[off + 0] = 0;
67 self.frame[off + 1] = 0;
68 self.frame[off + 2] = 0;
69 self.frame[off + 3] = 1;
70 self.frame[off + 4] = (self.frame_pos >> 24) as u8;
71 self.frame[off + 5] = (self.frame_pos >> 16) as u8;
72 self.frame[off + 6] = (self.frame_pos >> 8) as u8;
73 self.frame[off + 7] = (self.frame_pos >> 0) as u8;
75 fn get_frame_data(&mut self) -> Vec<u8> {
76 let mut v: Vec<u8> = Vec::new();
77 mem::swap(&mut v, &mut self.frame);
83 #[derive(Clone,Copy,PartialEq)]
92 struct RMAudioStream {
94 iinfo: Option<InterleaveInfo>,
99 const RM_ILEAVE_INT0: u32 = mktag!(b"Int0");
100 const RM_ILEAVE_INT4: u32 = mktag!(b"Int4");
101 const RM_ILEAVE_GENR: u32 = mktag!(b"genr");
102 const RM_ILEAVE_SIPR: u32 = mktag!(b"sipr");
103 const RM_ILEAVE_VBRF: u32 = mktag!(b"vbrf");
104 const RM_ILEAVE_VBRS: u32 = mktag!(b"vbrs");
107 fn new(iinfo: Option<InterleaveInfo>) -> Self {
110 if let Some(info) = iinfo {
111 deint = match info.id {
112 RM_ILEAVE_INT0 => Deinterleaver::None,
113 RM_ILEAVE_INT4 => Deinterleaver::RA28_8,
114 RM_ILEAVE_GENR => Deinterleaver::Generic,
115 RM_ILEAVE_SIPR => Deinterleaver::Sipro,
116 RM_ILEAVE_VBRF => Deinterleaver::VBR,
117 RM_ILEAVE_VBRS => Deinterleaver::VBR,
118 _ => {println!("unknown deint {:X}", info.id); Deinterleaver::None },
121 Deinterleaver::None => { buf = Vec::new(); },
122 Deinterleaver::RA28_8 |
123 Deinterleaver::Generic |
124 Deinterleaver::Sipro => {
125 let bsize = (info.frame_size as usize) * (info.factor as usize);
126 buf = vec![0; bsize];
128 Deinterleaver::VBR => {
133 deint = Deinterleaver::None;
136 RMAudioStream { deint, iinfo, buf, sub_packet: 0 }
138 fn read_apackets(&mut self, queued_packets: &mut Vec<NAPacket>, src: &mut ByteReader, stream: NAStreamRef, ts: u32, keyframe: bool, payload_size: usize) -> DemuxerResult<NAPacket> {
139 let (tb_num, tb_den) = stream.get_timebase();
140 let ts = NATimeInfo::new(Some(ts as u64), None, None, tb_num, tb_den);
146 Deinterleaver::None => { return src.read_packet(stream, ts, keyframe, payload_size); },
147 Deinterleaver::RA28_8 => {
148 let iinfo = self.iinfo.unwrap();
149 let factor = iinfo.factor as usize;
150 let halffact = factor >> 1;
151 let fsize = iinfo.frame_size as usize;
152 let bsize = iinfo.block_size as usize;
153 let ppos = self.sub_packet;
154 for sb in 0..halffact {
155 let mut dst = &mut self.buf[sb * 2 * fsize + ppos * bsize..][..bsize];
156 src.read_buf(&mut dst)?;
158 self.sub_packet += 1;
159 if self.sub_packet == factor {
161 return Ok(NAPacket::new(stream, ts, true, self.buf.clone()));
163 return Err(DemuxerError::TryAgain);
166 Deinterleaver::Generic => {
167 let iinfo = self.iinfo.unwrap();
168 let factor = iinfo.factor as usize;
169 let fsize = iinfo.frame_size as usize;
170 let bsize = iinfo.block_size as usize;
171 let factor2 = fsize / bsize;
172 let ppos = self.sub_packet;
174 for sb in 0..factor2 {
175 let sb_pos = factor * sb + ((factor + 1) >> 1) * (ppos & 1) + (ppos >> 1);
176 let mut dst = &mut self.buf[bsize * sb_pos..][..bsize];
177 src.read_buf(&mut dst)?;
180 Deinterleaver::Sipro => {
181 let iinfo = self.iinfo.unwrap();
182 let fsize = iinfo.frame_size as usize;
183 let ppos = self.sub_packet;
185 let mut dst = &mut self.buf[fsize * ppos..][..fsize];
186 src.read_buf(&mut dst)?;
188 Deinterleaver::VBR => {
189 validate!(payload_size >= 5);
190 let hdrsizesize = src.read_u16be()?;
191 let num_entries = (hdrsizesize / 16) as usize;
192 validate!(payload_size >= num_entries * 3 + 2);
193 let mut sizes: Vec<usize> = Vec::with_capacity(num_entries);
194 let mut tot_size = 0;
195 for _ in 0..num_entries {
196 let sz = src.read_u16be()? as usize;
200 validate!(tot_size + num_entries * 2 + 2 == payload_size);
201 let pkt_ts = NATimeInfo::new(None, None, None, tb_num, tb_den);
202 let mut first = true;
203 for size in sizes.iter() {
204 let cur_ts = if first { ts } else { pkt_ts };
206 let pkt = src.read_packet(stream.clone(), cur_ts, true, *size)?;
207 queued_packets.push(pkt);
209 queued_packets.reverse();
210 let pkt0 = queued_packets.pop().unwrap();
215 let iinfo = self.iinfo.unwrap();
216 let factor = iinfo.factor as usize;
217 let fsize = if iinfo.block_size != 0 { iinfo.block_size } else { iinfo.frame_size } as usize;
219 self.sub_packet += 1;
220 if self.sub_packet < factor {
221 return Err(DemuxerError::TryAgain);
226 if self.deint == Deinterleaver::Sipro {
227 sipro_restore(&mut self.buf, factor, iinfo.frame_size as usize);
230 let mut frames_iter = self.buf.chunks(fsize);
231 let pkt0 = frames_iter.next().unwrap();
233 let pkt_ts = NATimeInfo::new(None, None, None, tb_num, tb_den);
234 for pkts in frames_iter {
235 let pkt = NAPacket::new(stream.clone(), pkt_ts, true, pkts.to_vec());
236 queued_packets.push(pkt);
238 queued_packets.reverse();
239 Ok(NAPacket::new(stream, ts, true, pkt0.to_vec()))
243 fn sipro_restore(buf: &mut [u8], factor: usize, fsize: usize) {
244 let stride = factor * fsize * 2 / 96;
246 let mut sidx = (RM_SIPRO_SWAPS[i][0] as usize) * stride;
247 let mut didx = (RM_SIPRO_SWAPS[i][1] as usize) * stride;
249 let in0 = buf[sidx >> 1];
250 let in1 = buf[didx >> 1];
251 let nib0 = (in0 >> ((sidx & 1) * 4)) & 0xF;
252 let nib1 = (in1 >> ((didx & 1) * 4)) & 0xF;
254 buf[didx >> 1] = (nib0 << (4 * (didx & 1))) | (in1 & (0xF << (4 * (!didx & 1))));
255 buf[sidx >> 1] = (nib1 << (4 * (sidx & 1))) | (in0 & (0xF << (4 * (!sidx & 1))));
263 struct SubstreamInfo {
270 sub_info: Vec<SubstreamInfo>,
277 sub_info: Vec::new(),
281 fn add_stream(&mut self, stream_no: u32) {
282 let ssinfo = SubstreamInfo { id: stream_no, map: Vec::new(), str_ids: Vec::new() };
283 self.sub_info.push(ssinfo);
285 fn get_substream_no(&self) -> u32 {
288 fn find_idx(&self, stream_no: u32) -> Option<usize> {
289 self.sub_info.iter().position(|x| x.id == stream_no)
291 fn add_map_rule(&mut self, map_ss: u16) {
292 let idx = self.sub_info.len() - 1;
293 self.sub_info[idx].map.push(map_ss);
295 fn add_substream(&mut self) {
296 let idx = self.sub_info.len() - 1;
297 self.sub_info[idx].str_ids.push(self.sstr_id);
300 fn is_mlti_stream(&self, stream_no: u32) -> bool {
301 self.find_idx(stream_no).is_some()
303 fn find_substream(&self, stream_no: u32, grp: u16) -> Option<u32> {
304 if let Some(idx) = self.find_idx(stream_no) {
305 if (grp as usize) < self.sub_info[idx].map.len() {
306 let sub_id = self.sub_info[idx].map[grp as usize] as usize;
307 if sub_id < self.sub_info[idx].str_ids.len() {
308 return Some(self.sub_info[idx].str_ids[sub_id]);
317 Audio(RMAudioStream),
318 Video(RMVideoStream),
323 struct CommonStreamData {
324 streams: Vec<RMStreamType>,
326 mlti_mapper: MLTIMapper,
329 impl CommonStreamData {
334 mlti_mapper: MLTIMapper::new(),
337 fn get_stream_id(&self, str_no: u32, pkt_grp: u16) -> u32 {
338 if !self.mlti_mapper.is_mlti_stream(str_no) {
341 self.mlti_mapper.find_substream(str_no, pkt_grp).unwrap()
344 fn find_stream(&self, stream_id: u32) -> Option<usize> {
345 self.str_ids.iter().position(|x| *x == stream_id)
349 struct RealMediaDemuxer<'a> {
350 src: &'a mut ByteReader<'a>,
358 str_data: CommonStreamData,
360 data_chunks: Vec<(u64, u32, u16)>,
361 cur_data_chunk: usize,
363 queued_pkts: Vec<NAPacket>,
367 fn find_codec_name(registry: &[(&[u8;4], &'static str)], fcc: u32) -> &'static str {
368 for &(fourcc, name) in registry {
369 if mktag!(fourcc) == fcc { return name; }
374 fn read_14or30(src: &mut ByteReader) -> DemuxerResult<(bool, u32)> {
375 let tmp = src.read_u16be()?;
376 let flag = (tmp & 0x8000) != 0;
377 if (tmp & 0x4000) == 0x4000 {
378 Ok((flag, ((tmp & 0x3FFF) as u32)))
380 let val = ((tmp as u32) << 16) | (src.read_u16be()? as u32);
381 Ok((flag, val & 0x3FFFFFFF))
385 fn read_video_buf(src: &mut ByteReader, stream: NAStreamRef, ts: u32, keyframe: bool, frame_size: usize) -> DemuxerResult<NAPacket> {
386 let size = (frame_size as usize) + 9;
387 let mut vec: Vec<u8> = vec![0; size];
388 //v[0] = 0; // 1 slice
390 src.read_buf(&mut vec[9..])?;
392 let (tb_num, tb_den) = stream.get_timebase();
393 let ts = NATimeInfo::new(Some(ts as u64), None, None, tb_num, tb_den);
394 Ok(NAPacket::new(stream, ts, keyframe, vec))
397 fn read_multiple_frame(src: &mut ByteReader, stream: NAStreamRef, keyframe: bool, skip_mtype: bool) -> DemuxerResult<NAPacket> {
399 let mtype = src.read_byte()?;
400 validate!(mtype == 0xC0);
402 let (_, frame_size) = read_14or30(src)?;
403 let (_, timestamp) = read_14or30(src)?;
404 let _seq_no = src.read_byte()?;
405 //println!(" multiple frame size {} ts {} seq {}", frame_size, timestamp, seq_no);
407 read_video_buf(src, stream, timestamp, keyframe, frame_size as usize)
410 struct RMDemuxCommon {}
413 fn parse_stream_info(str_data: &mut CommonStreamData, strmgr: &mut StreamManager, stream_no: u32, edata: &Vec<u8>) -> DemuxerResult<bool> {
414 let mut is_mlti = false;
415 let mut mr = MemoryReader::new_read(edata.as_slice());
416 let mut src = ByteReader::new(&mut mr);
417 let tag = src.read_u32be()?;
418 let tag2 = src.peek_u32be()?;
419 //println!("tag1 {:X} tag2 {:X}", tag, tag2);
420 if tag == mktag!('.', 'r', 'a', 0xFD) {
421 Self::parse_audio_stream(strmgr, &mut str_data.streams, stream_no, &mut src, edata.as_slice())?;
422 } else if ((tag2 == mktag!('V', 'I', 'D', 'O')) || (tag2 == mktag!('I', 'M', 'A', 'G'))) && ((tag as usize) <= edata.len()) {
423 Self::parse_video_stream(strmgr, &mut str_data.streams, stream_no, &mut src, edata.as_slice(), tag2)?;
424 } else if tag == mktag!(b"LSD:") {
425 let extradata = Some(edata.to_owned());
427 src.read_skip(4)?; //version
428 let channels = src.read_u16be()?;
429 let samp_size = src.read_u16be()?;
430 let sample_rate = src.read_u32be()?;
432 println!("LSD sr {}, {} ch", sample_rate, channels);
433 let soniton = NASoniton::new(samp_size as u8, SONITON_FLAG_SIGNED);
434 let ahdr = NAAudioInfo::new(sample_rate, channels as u8, soniton, 1);
435 let nainfo = NACodecInfo::new("ralf", NACodecTypeInfo::Audio(ahdr), extradata);
436 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, stream_no as u32, nainfo, 1, 1000));
437 if res.is_none() { return Err(MemoryError); }
438 let astr = RMAudioStream::new(None);
439 str_data.streams.push(RMStreamType::Audio(astr));
440 } else if tag == mktag!(b"MLTI") {
442 let num_rules = src.read_u16be()? as usize;
444 str_data.mlti_mapper.add_stream(stream_no);
445 for _ in 0..num_rules {
446 let substr = src.read_u16be()?;
447 max_sub = max_sub.max(substr);
448 str_data.mlti_mapper.add_map_rule(substr);
450 let num_substreams = src.read_u16be()? as usize;
451 validate!(num_substreams > (max_sub as usize));
452 for _ in 0..num_substreams {
453 let hdr_size = src.read_u32be()? as usize;
454 validate!(hdr_size > 8);
455 let pos = src.tell() as usize;
456 src.read_skip(hdr_size)?;
457 str_data.mlti_mapper.add_substream();
459 let hdrsrc = &edata[pos..][..hdr_size];
460 let mut mr = MemoryReader::new_read(hdrsrc);
461 let mut hsrc = ByteReader::new(&mut mr);
463 let tag = hsrc.read_u32be()?;
464 let tag2 = hsrc.peek_u32be()?;
465 let stream_no = str_data.mlti_mapper.get_substream_no();
466 //todo check that all substreams are of the same type");
467 if tag == mktag!('.', 'r', 'a', 0xFD) {
468 Self::parse_audio_stream(strmgr, &mut str_data.streams, stream_no, &mut hsrc, hdrsrc)?;
469 } else if (tag2 == mktag!('V', 'I', 'D', 'O')) && ((tag as usize) <= hdr_size) {
470 Self::parse_video_stream(strmgr, &mut str_data.streams, stream_no, &mut hsrc, hdrsrc, tag2)?;
472 println!("unknown MLTI substream {:08X} / {:08X}", tag, tag2);
473 return Err(DemuxerError::InvalidData);
475 str_data.str_ids.push(stream_no);
479 str_data.streams.push(RMStreamType::Logical);
483 fn parse_audio_stream(strmgr: &mut StreamManager, streams: &mut Vec<RMStreamType>, stream_no: u32, src: &mut ByteReader, edata_: &[u8]) -> DemuxerResult<()> {
484 let ver = src.read_u16be()?;
485 let ainfo = match ver {
496 println!("unknown version {}", ver);
497 return Err(DemuxerError::InvalidData);
500 println!(" got ainfo {:?}", ainfo);
501 let cname = find_codec_name(RM_AUDIO_CODEC_REGISTER, ainfo.fcc);
502 let blk_size = if ainfo.fcc != mktag!(b"sipr") {
503 ainfo.granularity as usize
505 validate!(ainfo.flavor <= 3);
506 RM_SIPRO_BLOCK_SIZES[ainfo.flavor as usize]
508 let srate = ainfo.sample_rate;
509 let soniton = NASoniton::new(ainfo.sample_size as u8, SONITON_FLAG_SIGNED);
510 let ahdr = NAAudioInfo::new(srate, ainfo.channels as u8, soniton, blk_size);
511 let extradata = if ainfo.edata_size == 0 {
514 let eslice = &edata_[(src.tell() as usize)..];
515 Some(eslice.to_vec())
517 let nainfo = NACodecInfo::new(cname, NACodecTypeInfo::Audio(ahdr), extradata);
518 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, stream_no as u32, nainfo, 1, 1000));
519 if res.is_none() { return Err(MemoryError); }
521 let astr = RMAudioStream::new(ainfo.ileave_info);
522 streams.push(RMStreamType::Audio(astr));
525 #[allow(unused_variables)]
526 fn parse_video_stream(strmgr: &mut StreamManager, streams: &mut Vec<RMStreamType>, stream_no: u32, src: &mut ByteReader, edata_: &[u8], tag2: u32) -> DemuxerResult<()> {
528 let fcc = src.read_u32be()?;
529 let width = src.read_u16be()? as usize;
530 let height = src.read_u16be()? as usize;
531 let bpp = src.read_u16be()?;
532 let pad_w = src.read_u16be()?;
533 let pad_h = src.read_u16be()?;
535 if tag2 == mktag!('V', 'I', 'D', 'O') {
536 _fps = src.read_u32be()?;
540 let extradata: Option<Vec<u8>>;
542 let eslice = &edata_[(src.tell() as usize)..];
543 extradata = Some(eslice.to_vec());
547 let cname = find_codec_name(RM_VIDEO_CODEC_REGISTER, fcc);
549 let vhdr = NAVideoInfo::new(width, height, false, RGB24_FORMAT);
550 let vinfo = NACodecInfo::new(cname, NACodecTypeInfo::Video(vhdr), extradata);
551 let res = strmgr.add_stream(NAStream::new(StreamType::Video, stream_no as u32, vinfo, 1, 1000));
552 if res.is_none() { return Err(DemuxerError::MemoryError); }
554 let vstr = RMVideoStream::new();
555 streams.push(RMStreamType::Video(vstr));
558 #[allow(unused_variables)]
559 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> {
561 RMStreamType::Video(ref mut vstr) => {
563 let pos = src.tell();
564 let b0 = src.read_byte()?;
566 0 => { // partial frame
567 let b1 = src.read_byte()?;
568 let hdr1 = ((b0 as u16) << 8) | (b1 as u16);
569 let num_pkts = ((hdr1 >> 7) & 0x7F) as usize;
570 let packet_num = hdr1 & 0x7F;
571 let (_, frame_size) = read_14or30(src)?;
572 let (_, off) = read_14or30(src)?;
573 let seq_no = src.read_byte()?;
574 //println!(" mode 0 pkt {}/{} off {}/{} seq {}", packet_num, num_pkts, off, frame_size, seq_no);
575 let hdr_skip = (src.tell() - pos) as usize;
577 let slice_size = (payload_size - hdr_skip) as usize;
578 slice_buf.resize(slice_size, 0);
579 src.read_buf(slice_buf.as_mut_slice())?;
581 vstr.start_slice(num_pkts, frame_size as usize, slice_buf.as_slice());
583 vstr.add_slice(packet_num as usize, slice_buf.as_slice());
585 if (packet_num as usize) < num_pkts {
586 return Err(DemuxerError::TryAgain);
588 //todo: check if full frame is received
589 let (tb_num, tb_den) = stream.get_timebase();
590 let ts = NATimeInfo::new(Some(ts as u64), None, None, tb_num, tb_den);
591 let pkt = NAPacket::new(stream, ts, keyframe, vstr.get_frame_data());
594 1 => { // whole frame
595 let seq_no = src.read_byte()?;
596 //println!(" mode 1 seq {}", seq_no);
597 read_video_buf(src, stream, ts, keyframe, payload_size - 2)
599 2 => { // last partial frame
600 let b1 = src.read_byte()?;
601 let hdr1 = ((b0 as u16) << 8) | (b1 as u16);
602 let num_pkts = ((hdr1 >> 7) & 0x7F) as usize;
603 let packet_num = hdr1 & 0x7F;
604 let (_, frame_size) = read_14or30(src)?;
605 let (_, tail_size) = read_14or30(src)?;
606 let seq_no = src.read_byte()?;
607 //println!(" mode 2 pkt {}/{} tail {}/{} seq {}", packet_num, num_pkts, tail_size, frame_size, seq_no);
608 slice_buf.resize(tail_size as usize, 0);
609 src.read_buf(slice_buf.as_mut_slice())?;
610 if packet_num == 1 && frame_size == tail_size {
611 vstr.start_slice(num_pkts, frame_size as usize, slice_buf.as_slice());
613 vstr.add_slice(packet_num as usize, slice_buf.as_slice());
616 while src.tell() < pos + (payload_size as u64) {
617 let res = read_multiple_frame(src, stream.clone(), false, false);
618 if res.is_err() { break; }
619 queued_pkts.push(res.unwrap());
621 queued_pkts.reverse();
622 let (tb_num, tb_den) = stream.get_timebase();
623 let ts = NATimeInfo::new(Some(ts as u64), None, None, tb_num, tb_den);
624 let pkt = NAPacket::new(stream, ts, keyframe, vstr.get_frame_data());
627 _ => { // multiple frames
628 //println!(" mode 3");
629 let res = read_multiple_frame(src, stream.clone(), keyframe, true);
630 if res.is_err() { return res; }
631 while src.tell() < pos + (payload_size as u64) {
632 let res = read_multiple_frame(src, stream.clone(), false, false);
633 if res.is_err() { break; }
634 queued_pkts.push(res.unwrap());
636 queued_pkts.reverse();
641 RMStreamType::Audio(ref mut astr) => {
642 astr.read_apackets(queued_pkts, src, stream, ts, keyframe, payload_size)
645 src.read_skip(payload_size)?;
646 Err(DemuxerError::InvalidData)
652 impl<'a> DemuxCore<'a> for RealMediaDemuxer<'a> {
653 fn open(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
654 self.read_header(strmgr, seek_idx)?;
658 #[allow(unused_variables)]
659 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
660 if !self.queued_pkts.is_empty() {
661 let pkt = self.queued_pkts.pop().unwrap();
665 if (self.cur_packet >= self.num_packets) || (self.src.tell() >= self.data_end) {
666 self.cur_data_chunk += 1;
667 if self.cur_data_chunk < self.data_chunks.len() {
668 let (pos, _, _) = self.data_chunks[self.cur_data_chunk];
669 self.src.seek(SeekFrom::Start(pos))?;
670 let res = read_chunk(self.src);
671 if let Ok((id, size, ver)) = res {
672 self.data_pos = self.src.tell();
674 self.data_end = self.data_pos + (size as u64);
675 if self.parse_data_start().is_ok() {
680 return Err(DemuxerError::EOF);
683 let pkt_start = self.src.tell();
684 let ver = self.src.read_u16be()?;
686 let len = self.src.read_u16be()? as usize;
687 let str_no = self.src.read_u16be()? as u32;
688 let ts = self.src.read_u32be()?;
692 pkt_grp = self.src.read_byte()? as u16;
693 flags = self.src.read_byte()?;
695 //asm_rule = self.src.read_u16be()?;
696 //asm_flags = self.src.read_byte()?;
697 self.src.read_skip(2)?;
699 self.src.read_skip(1)?;
702 let hdr_size = self.src.tell() - pkt_start;
703 //println!("packet @{:X} size {} for {} ts {} grp {} flags {:X}", pkt_start, len, str_no, ts, pkt_grp, flags);
704 self.cur_packet += 1;
706 let payload_size = len - (hdr_size as usize);
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);
711 //println!("stream {} not found", str_no);
712 self.src.read_skip(payload_size)?;
713 return Err(DemuxerError::InvalidData);
715 let str_id = sr.unwrap();
717 let streamres = strmgr.get_stream_by_id(stream_id);
718 if streamres.is_none() {
719 self.src.read_skip(payload_size)?;
722 let stream = streamres.unwrap();
723 //println!(" stream {}", str_id);
724 if strmgr.is_ignored_id(stream_id) {
725 self.src.read_skip(payload_size)?;
728 //todo skip unwanted packet
729 let keyframe = (flags & KEYFRAME_FLAG) != 0;
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 {
740 #[allow(unused_variables)]
741 fn seek(&mut self, time: u64, seek_idx: &SeekIndex) -> DemuxerResult<()> {
742 self.queued_pkts.clear();
743 let ret = seek_idx.find_pos(time);
745 return Err(DemuxerError::SeekError);
747 let ret = ret.unwrap();
748 let seek_pos = ret.pos;
749 for (pos, size, ver) in self.data_chunks.iter() {
750 if seek_pos < *pos { continue; }
751 let end = *pos + (*size as u64);
754 self.data_pos = seek_pos;
755 self.data_ver = *ver;
757 self.src.seek(SeekFrom::Start(seek_pos))?;
761 Err(DemuxerError::SeekError)
765 fn read_chunk(src: &mut ByteReader) -> DemuxerResult<(u32, u32, u16)> {
766 let id = src.read_u32be()?;
767 if id == 0 { return Ok((0, 0, 0)); }
768 let size = src.read_u32be()?;
770 let ver = src.read_u16be()?;
772 return Ok((id, 0x0FFFFFFF, ver));
774 validate!(size >= 10);
775 let ver = src.read_u16be()?;
780 #[derive(Clone,Copy,Debug)]
781 struct InterleaveInfo {
788 #[derive(Clone,Copy,Debug)]
789 struct RealAudioInfo {
797 bytes_per_minute: u32,
800 ileave_info: Option<InterleaveInfo>
803 fn skip_ra_metadata(src: &mut ByteReader) -> DemuxerResult<()> {
804 let title_len = src.read_byte()? as usize;
805 src.read_skip(title_len)?;
806 let author_len = src.read_byte()? as usize;
807 src.read_skip(author_len)?;
808 let copywrong_len = src.read_byte()? as usize;
809 src.read_skip(copywrong_len)?;
810 let comment_len = src.read_byte()? as usize;
811 src.read_skip(comment_len)?;
815 #[allow(unused_variables)]
816 fn parse_aformat3(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
817 let start = src.tell();
818 let header_len = src.read_u16be()?;
819 validate!(header_len >= 24);
820 let flavor = src.read_u16be()?;
821 let granularity = src.read_u32be()?;
822 let bytes_per_minute = src.read_u32be()?;
823 let total_bytes = src.read_u32be()?;
825 skip_ra_metadata(src)?;
827 let _can_copy = src.read_byte()?;
828 let fcc_len = src.read_byte()?;
829 validate!(fcc_len == 4);
830 let fcc = src.read_u32be()?;
832 let end = src.tell();
833 validate!(end - start <= (header_len as u64) + 2);
837 sample_rate: 8000, sample_size: 16, channels: 1, channel_mask: 0,
838 granularity, bytes_per_minute,
839 total_bytes, edata_size: 0,
844 #[allow(unused_variables)]
845 fn parse_aformat4(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
846 let start = src.tell();
847 src.read_skip(2)?; // zeroes
848 let id = src.read_u32be()?;
849 validate!(id == mktag!(b".ra4"));
850 let data_size = src.read_u32be()?;
851 let _ver4 = src.read_u16be()?; // should be 4
852 let header_size = src.read_u32be()?;
853 let flavor = src.read_u16be()?;
854 let granularity = src.read_u32be()?;
855 let total_bytes = src.read_u32be()?;
856 let bytes_per_minute = src.read_u32be()?;
857 let _bytes_per_minute2 = src.read_u32be()?;
858 let ileave_factor = src.read_u16be()?;
859 let ileave_block_size = src.read_u16be()?;
860 let _user_data = src.read_u16be()?;
861 let sample_rate = src.read_u32be()?;
862 let sample_size = src.read_u32be()?;
863 let channels = src.read_u16be()?;
864 let interleaver_id_len = src.read_byte()?;
865 validate!(interleaver_id_len == 4);
866 let interleaver_id = src.read_u32be()?;
867 let fcc_len = src.read_byte()?;
868 validate!(fcc_len == 4);
869 let fcc = src.read_u32be()?;
870 let is_interleaved = src.read_byte()?;
871 let _can_copy = src.read_byte()?;
872 let _stream_type = src.read_byte()?;
874 skip_ra_metadata(src)?;
876 let end = src.tell();
877 validate!(end - start <= (header_size as u64) + 10);
879 let ileave_info = if is_interleaved != 0 {
880 Some(InterleaveInfo {
881 id: interleaver_id, factor: ileave_factor, block_size: granularity as u16,
882 frame_size: ileave_block_size,
890 sample_rate, sample_size: sample_size as u16, channels, channel_mask: 0,
891 granularity, bytes_per_minute,
892 total_bytes: total_bytes & 0xFFFFFF, edata_size: 0,
897 #[allow(unused_variables)]
898 fn parse_aformat5(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
899 let start = src.tell();
900 src.read_skip(2)?; // zeroes
901 let id = src.read_u32be()?;
902 validate!((id == mktag!(b".ra5")) || (id == mktag!(b".ra4")));
903 let data_size = src.read_u32be()?;
904 let _ver5 = src.read_u16be()?; // should be 5
905 let header_size = src.read_u32be()?;
906 let flavor = src.read_u16be()?;
907 let granularity = src.read_u32be()?;
908 let total_bytes = src.read_u32be()?;
909 let bytes_per_minute = src.read_u32be()?;
910 let _bytes_per_minute2 = src.read_u32be()?;
911 let ileave_factor = src.read_u16be()?;
912 let frame_size = src.read_u16be()?;
913 let ileave_block_size = src.read_u16be()?;
914 let user_data = src.read_u32be()?;
915 let _sample_rate1 = src.read_u16be()?;
916 let sample_rate = src.read_u32be()?;
917 let sample_size = src.read_u32be()?;
918 let channels = src.read_u16be()?;
919 let interleaver_id = src.read_u32be()?;
920 let fcc = src.read_u32be()?;
921 let is_interleaved = src.read_byte()?;
922 let _can_copy = src.read_byte()?;
923 let _stream_type = src.read_byte()?;
924 let has_ileave_pattern = src.read_byte()?;
925 if has_ileave_pattern != 0 {
926 unimplemented!("ra5 interleave pattern");
928 let mut edata_size = src.read_u32be()?;
929 let end = src.tell();
930 if id == mktag!(b".ra5") {
931 validate!(end - start <= (header_size as u64) + 10);
932 // src.read_skip(((end - start) as usize) - (header_size as usize) - 10)?;
934 validate!(end - start <= (header_size as u64) + 15);
937 let ileave_info = if is_interleaved != 0 {
938 Some(InterleaveInfo {
939 id: interleaver_id, factor: ileave_factor, block_size: ileave_block_size, frame_size,
944 if (fcc == mktag!(b"raac")) || (fcc == mktag!(b"racp")) {
945 validate!(edata_size > 1);
952 sample_rate, sample_size: sample_size as u16, channels, channel_mask: 0,
953 granularity, bytes_per_minute,
954 total_bytes: total_bytes & 0xFFFFFF, edata_size,
959 const RMVB_HDR_SIZE: u32 = 18;
960 const RMVB_PROP_SIZE: u32 = 50;
961 const KEYFRAME_FLAG: u8 = 0x02;
963 impl<'a> RealMediaDemuxer<'a> {
964 fn new(io: &'a mut ByteReader<'a>) -> Self {
973 str_data: CommonStreamData::new(),
974 data_chunks: Vec::new(),
976 queued_pkts: Vec::new(),
977 slice_buf: Vec::new(),
980 #[allow(unused_variables)]
981 fn read_header(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
982 let (id, size, ver) = read_chunk(self.src)?;
983 validate!((id == mktag!(b".RMF")) || (id == mktag!(b".RMP")));
984 validate!(size >= RMVB_HDR_SIZE);
985 let fver = self.src.read_u32be()?;
986 validate!(fver <= 1);
987 let num_hdr = self.src.read_u32be()? as usize;
988 validate!(num_hdr >= 1);
989 if size > RMVB_HDR_SIZE {
990 self.src.read_skip((size - RMVB_HDR_SIZE) as usize)?;
993 let (id, size, ver) = read_chunk(self.src)?;
994 let prop_size = if ver == 0 { RMVB_PROP_SIZE } else { RMVB_PROP_SIZE + 4 };
995 validate!(size >= prop_size);
996 validate!((ver == 0) || (ver == 2));
997 let maxbr = self.src.read_u32be()?;
998 let avgbr = self.src.read_u32be()?;
999 let maxps = self.src.read_u32be()?;
1000 let avgps = self.src.read_u32be()?;
1001 let num_pkt = self.src.read_u32be()? as usize;
1002 let duration = self.src.read_u32be()?;
1003 let preroll = self.src.read_u32be()?;
1005 self.src.read_skip(4)?;
1007 let idx_off = self.src.read_u32be()?;
1008 let data_off = self.src.read_u32be()?;
1009 let num_streams = self.src.read_u16be()? as usize;
1010 let flags = self.src.read_u16be()?;
1011 if size > prop_size {
1012 self.src.read_skip((size - prop_size) as usize)?;
1015 for _ in 0..num_hdr {
1016 if self.src.is_eof() {
1020 let res = self.parse_chunk(strmgr, seek_idx);
1022 Ok(last) => { if last { break; } },
1023 Err(DemuxerError::IOError) => { break; },
1025 if self.data_chunks.len() == 0 { // data is not found, report error
1031 //println!("now @ {:X} / {}", self.src.tell(), self.data_pos);
1032 validate!(self.data_chunks.len() > 0);
1033 self.cur_data_chunk = 0;
1034 let (pos, size, ver) = self.data_chunks[self.cur_data_chunk];
1035 self.data_pos = pos;
1036 self.data_ver = ver;
1037 self.data_end = pos + (size as u64);
1038 self.src.seek(SeekFrom::Start(self.data_pos + 10))?;
1039 self.parse_data_start()?;
1042 fn parse_data_start(&mut self) -> DemuxerResult<()> {
1043 let num_packets = self.src.read_u32be()?;
1044 if self.data_ver == 2 {
1045 self.src.read_skip(12)?; // zeroes?
1047 let next_data_hdr = self.src.read_u32be()?;
1048 self.num_packets = if num_packets > 0 { num_packets } else { 0xFFFFFF };
1049 self.cur_packet = 0;
1050 self.next_data = next_data_hdr as u64;
1053 fn parse_chunk(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<bool> {
1054 let (id, size, ver) = read_chunk(self.src)?;
1055 let end_pos = self.src.tell() - 10 + (size as u64);
1057 validate!((ver == 0) || (ver == 2));
1058 if id == mktag!(b"CONT") { self.parse_content_desc()?; }
1059 else if id == mktag!(b"MDPR") { self.parse_mdpr(strmgr)?; }
1060 else if id == mktag!(b"DATA") {
1061 self.data_chunks.push((self.src.tell() - 10, size, ver));
1063 else if id == mktag!(b"INDX") {
1064 if !seek_idx.skip_index {
1065 self.parse_index(seek_idx, (size as usize) - 10, ver)?;
1068 else if id == 0 { return Ok(true); }
1069 else { println!("unknown chunk type {:08X}", id); }
1071 let cpos = self.src.tell();
1073 self.src.read_skip((end_pos - cpos) as usize)?;
1077 #[allow(unused_variables)]
1078 fn parse_content_desc(&mut self) -> DemuxerResult<()> {
1079 let title_len = self.src.read_u16be()? as usize;
1080 self.src.read_skip(title_len)?;
1081 let author_len = self.src.read_u16be()? as usize;
1082 self.src.read_skip(author_len)?;
1083 let copywrong_len = self.src.read_u16be()? as usize;
1084 self.src.read_skip(copywrong_len)?;
1085 let comment_len = self.src.read_u16be()? as usize;
1086 self.src.read_skip(comment_len)?;
1089 #[allow(unused_variables)]
1090 fn parse_mdpr(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<()> {
1091 let stream_no = self.src.read_u16be()? as u32;
1092 //todo check stream_no for duplicates
1093 let maxbr = self.src.read_u32be()?;
1094 let avgbr = self.src.read_u32be()?;
1095 let maxps = self.src.read_u32be()?;
1096 let avgps = self.src.read_u32be()?;
1097 let start = self.src.read_u32be()?;
1098 let preroll = self.src.read_u32be()?;
1099 let duration = self.src.read_u32be()?;
1100 let sname_size = self.src.read_byte()? as usize;
1101 let sname = read_string_size(self.src, sname_size)?;
1102 //println!("str #{} sname = {} pkts {}/{} start {} preroll {}", stream_no, sname, maxps, avgps, start, preroll);
1103 let mime_size = self.src.read_byte()? as usize;
1104 let mime = read_string_size(self.src, mime_size)?;
1105 //println!("mime = {}", mime);
1106 let edata_size = self.src.read_u32be()? as usize;
1107 let edata: Option<Vec<u8>> = if edata_size == 0 { None } else {
1108 let mut edvec: Vec<u8> = vec![0; edata_size];
1109 self.src.read_buf(&mut edvec)?;
1112 let mut is_mlti = false;
1114 if let Some(edata_) = edata {
1115 is_mlti = RMDemuxCommon::parse_stream_info(&mut self.str_data, strmgr, stream_no, &edata_)?;
1118 self.str_data.streams.push(RMStreamType::Unknown);
1121 self.str_data.str_ids.push(stream_no);
1126 fn parse_index(&mut self, seek_idx: &mut SeekIndex, chunk_size: usize, ver: u16) -> DemuxerResult<()> {
1127 if ver != 0 { return Ok(()); }
1128 let num_entries = self.src.read_u32be()? as usize;
1129 let str_id = self.src.read_u16be()? as u32;
1130 let _next_idx = self.src.read_u32be()?;
1131 validate!(chunk_size == num_entries * 14 + 10);
1132 if num_entries == 0 { return Ok(()); }
1134 seek_idx.add_stream(str_id);
1135 let idx = seek_idx.get_stream_index(str_id).unwrap();
1136 for _ in 0..num_entries {
1137 let ver = self.src.read_u16be()? as u32;
1138 validate!(ver == 0);
1139 let ts = self.src.read_u32be()? as u64;
1140 let pos = self.src.read_u32be()? as u64;
1141 let _pkt_no = self.src.read_u32be()?;
1142 idx.add_entry(SeekEntry { time: ts, pts: 0, pos });
1145 seek_idx.mode = SeekIndexMode::Present;
1150 fn read_string_size(src: &mut ByteReader, size: usize) -> DemuxerResult<String> {
1151 let mut vec: Vec<u8> = Vec::new();
1153 let c = src.read_byte()?;
1156 let str = String::from_utf8(vec);
1164 struct RealAudioDemuxer<'a> {
1165 src: &'a mut ByteReader<'a>,
1166 stream: Option<RMAudioStream>,
1170 queued_pkts: Vec<NAPacket>,
1173 impl<'a> DemuxCore<'a> for RealAudioDemuxer<'a> {
1174 #[allow(unused_variables)]
1175 fn open(&mut self, strmgr: &mut StreamManager, _seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
1176 let magic = self.src.read_u32be()?;
1177 validate!(magic == mktag!(b".ra\xFD"));
1178 let ver = self.src.read_u16be()?;
1179 let ainfo = match ver {
1181 parse_aformat3(&mut self.src)?
1184 parse_aformat4(&mut self.src)?
1187 parse_aformat5(&mut self.src)?
1190 println!("unknown version {}", ver);
1191 return Err(DemuxerError::InvalidData);
1194 println!(" got ainfo {:?}", ainfo);
1195 let cname = find_codec_name(RM_AUDIO_CODEC_REGISTER, ainfo.fcc);
1196 let blk_size = if ainfo.fcc != mktag!(b"sipr") {
1197 ainfo.granularity as usize
1199 validate!(ainfo.flavor <= 3);
1200 RM_SIPRO_BLOCK_SIZES[ainfo.flavor as usize]
1202 let srate = ainfo.sample_rate;
1203 let soniton = NASoniton::new(ainfo.sample_size as u8, SONITON_FLAG_SIGNED);
1204 let ahdr = NAAudioInfo::new(srate, ainfo.channels as u8, soniton, blk_size);
1205 let extradata = if ainfo.edata_size == 0 {
1208 let mut dta: Vec<u8> = vec![0; ainfo.edata_size as usize];
1209 self.src.read_buf(dta.as_mut_slice())?;
1212 let nainfo = NACodecInfo::new(cname, NACodecTypeInfo::Audio(ahdr), extradata);
1213 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, 0, nainfo, 1, srate));
1214 if res.is_none() { return Err(MemoryError); }
1216 let astr = RMAudioStream::new(ainfo.ileave_info);
1217 self.data_start = self.src.tell();
1218 self.data_end = if ainfo.total_bytes > 0 { self.src.tell() + (ainfo.total_bytes as u64) } else { 0 };
1219 self.blk_size = blk_size;
1220 self.stream = Some(astr);
1225 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
1226 if !self.queued_pkts.is_empty() {
1227 let pkt = self.queued_pkts.pop().unwrap();
1230 if (self.data_end != 0) && (self.src.tell() >= self.data_end) {
1231 return Err(DemuxerError::EOF);
1233 let streamres = strmgr.get_stream_by_id(0);
1234 let stream = streamres.unwrap();
1235 if let Some(ref mut astr) = self.stream {
1237 let ret = astr.read_apackets(&mut self.queued_pkts, &mut self.src, stream.clone(), 0, false, self.blk_size);
1238 if let Err(DemuxerError::TryAgain) = ret {
1244 Err(DemuxerError::NoSuchInput)
1247 #[allow(unused_variables)]
1248 fn seek(&mut self, time: u64, seek_idx: &SeekIndex) -> DemuxerResult<()> {
1253 impl<'a> RealAudioDemuxer<'a> {
1254 fn new(io: &'a mut ByteReader<'a>) -> Self {
1261 queued_pkts: Vec::new(),
1268 StreamProperties(usize),
1269 Packet { ts: u32, str: u32, flags: u32, len: usize, checksum: u32 },
1270 IntValue(Vec<u8>, u32),
1271 BinaryData(Vec<u8>, Vec<u8>),
1272 StringData(Vec<u8>, Vec<u8>),
1279 fn read_string(src: &mut ByteReader) -> DemuxerResult<Vec<u8>> {
1280 let len = src.read_u32be()? as usize;
1281 let mut val = vec![0; len];
1282 src.read_buf(val.as_mut_slice())?;
1286 fn read(src: &mut ByteReader) -> DemuxerResult<Self> {
1287 let code = src.read_byte()?;
1290 let val = src.read_u32be()? as usize;
1291 Ok(IVRRecord::StreamProperties(val))
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 validate!((len > 0) && (len < (1 << 24)));
1300 Ok(IVRRecord::Packet { ts, str, flags, len, checksum: chk })
1303 let name = Self::read_string(src)?;
1304 let len = src.read_u32be()?;
1305 validate!(len == 4);
1306 let val = src.read_u32be()?;
1307 Ok(IVRRecord::IntValue(name, val))
1310 let name = Self::read_string(src)?;
1311 let len = src.read_u32be()? as usize;
1312 let mut val = vec![0; len];
1313 src.read_buf(val.as_mut_slice())?;
1314 Ok(IVRRecord::BinaryData(name, val))
1317 let name = Self::read_string(src)?;
1318 let val = Self::read_string(src)?;
1319 Ok(IVRRecord::StringData(name, val))
1321 6 => Ok(IVRRecord::HeaderEnd),
1323 src.read_skip(8)?; // always zero?
1324 Ok(IVRRecord::DataEnd)
1327 src.read_skip(8)?; // always zero?
1328 Ok(IVRRecord::DataStart)
1330 _ => Ok(IVRRecord::Invalid(code)),
1333 fn is_data_start(&self) -> bool {
1335 IVRRecord::DataStart => true,
1341 impl fmt::Display for IVRRecord {
1342 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
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"),
1362 struct RecordDemuxer {
1366 remap_ids: Vec<u32>,
1369 impl RecordDemuxer {
1370 fn new(pos: u64, start_str: u32) -> Self {
1375 remap_ids: Vec::new(),
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);
1388 let mut has_seek_table = false;
1389 let mut cur_str_no = 0;
1391 let rec = IVRRecord::read(src)?;
1393 IVRRecord::HeaderEnd => { break; },
1394 IVRRecord::StreamProperties(num) => {
1395 let stream_no = cur_str_no + self.start_str;
1397 let mut parsed = false;
1398 let mut real_stream_no = 0;
1400 let rec = IVRRecord::read(src)?;
1401 //println!(" strm property {}", rec);
1403 IVRRecord::IntValue(ref name, val) => {
1404 if name == b"StreamNumber\0" {
1405 real_stream_no = val;
1408 IVRRecord::BinaryData(ref name, ref val) => {
1409 if name == b"OpaqueData\0" {
1411 let is_mlti = RMDemuxCommon::parse_stream_info(str_data, strmgr, stream_no, val)?;
1413 str_data.str_ids.push(stream_no);
1418 IVRRecord::StringData(ref name, ref val) => {
1419 if (name == b"SeekType\0") && (val != b"None\0") {
1420 has_seek_table = true;
1423 _ => { return Err(DemuxerError::InvalidData); }
1427 str_data.streams.push(RMStreamType::Unknown);
1428 str_data.str_ids.push(stream_no);
1430 self.remap_ids.push(real_stream_no);
1432 _ => {println!(" unexpected {}", rec); return Err(DemuxerError::InvalidData); }
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);
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());
1451 let ntype = src.peek_byte()?;
1452 validate!((ntype == 2) || (ntype == 7)); // packet or data end, no start
1455 self.cur_pos = src.tell();
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))?;
1462 let rec = IVRRecord::read(src)?;
1464 IVRRecord::Packet { ts, str, flags, len, .. } => {
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 pkt_grp = ((flags >> 8) & 0xFF) as u16;
1470 let stream_id = str_data.get_stream_id(str_no as u32, pkt_grp);
1471 let sr = str_data.find_stream(stream_id);
1473 src.read_skip(payload_size)?;
1474 return Err(DemuxerError::InvalidData);
1476 let str_id = sr.unwrap();
1478 let streamres = strmgr.get_stream_by_id(stream_id);
1479 if streamres.is_none() {
1480 src.read_skip(payload_size)?;
1483 let stream = streamres.unwrap();
1484 if strmgr.is_ignored_id(stream_id) {
1485 src.read_skip(payload_size)?;
1488 let keyframe = false;
1489 let ret = RMDemuxCommon::parse_packet_payload(src, &mut str_data.streams[str_id], stream, slice_buf, queued_pkts, keyframe, ts, payload_size);
1490 if let Err(DemuxerError::TryAgain) = ret {
1493 self.cur_pos = src.tell();
1497 IVRRecord::DataEnd => return Err(DemuxerError::EOF),
1498 _ => return Err(DemuxerError::InvalidData),
1504 struct RealIVRDemuxer<'a> {
1505 src: &'a mut ByteReader<'a>,
1506 recs: Vec<RecordDemuxer>,
1508 queued_pkts: Vec<NAPacket>,
1510 str_data: CommonStreamData,
1513 impl<'a> DemuxCore<'a> for RealIVRDemuxer<'a> {
1514 #[allow(unused_variables)]
1515 fn open(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
1516 let magic = self.src.peek_u32be()?;
1517 if magic == mktag!(b".REC") {
1518 let mut rec = RecordDemuxer::new(0, 0);
1519 rec.parse_header(&mut self.src, strmgr, &mut self.str_data)?;
1520 self.recs.push(rec);
1521 } else if magic == mktag!(b".R1M") {
1522 println!("R1M kind");
1523 self.src.read_skip(4)?; // magic
1524 self.src.read_skip(3)?; // always 0, 1, 1 ?
1525 let _name = IVRRecord::read_string(&mut self.src)?;
1526 self.src.read_skip(1)?; // always 0?
1527 let len1 = self.src.read_u32be()? as u64;
1528 let off1 = self.src.read_u64be()?;
1529 let cpos = self.src.tell();
1530 validate!(off1 == len1 + cpos - 8);
1531 self.src.read_skip((off1 - cpos) as usize)?;
1533 let typ = self.src.read_byte()?;
1534 println!(" at {:X} type {:02X}", self.src.tell(), typ);
1537 let len = self.src.read_u32be()?;
1538 self.src.read_skip(len as usize)?;
1541 let len = self.src.read_u32be()? as u64;
1542 let pos = self.src.tell();
1544 let num_streams = self.str_data.streams.len() as u32;
1545 let mut rec = RecordDemuxer::new(pos + 12, num_streams);
1546 rec.parse_header(&mut self.src, strmgr, &mut self.str_data)?;
1547 self.recs.push(rec);
1549 self.src.seek(SeekFrom::Start(pos + len))?;
1552 let mut buf: [u8; 2] = [0; 2];
1553 self.src.peek_buf(&mut buf)?;
1554 if (buf[0] == b'J') && (buf[1] == b'M') { // RJMx markers at the end of file
1557 return Err(DemuxerError::InvalidData);
1560 _ => { return Err(DemuxerError::InvalidData); },
1564 return Err(DemuxerError::InvalidData);
1570 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
1571 if !self.queued_pkts.is_empty() {
1572 let pkt = self.queued_pkts.pop().unwrap();
1576 if self.cur_rec >= self.recs.len() { return Err(DemuxerError::EOF); }
1577 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);
1578 if let Err(DemuxerError::EOF) = res {
1586 #[allow(unused_variables)]
1587 fn seek(&mut self, time: u64, seek_idx: &SeekIndex) -> DemuxerResult<()> {
1592 impl<'a> RealIVRDemuxer<'a> {
1593 fn new(io: &'a mut ByteReader<'a>) -> Self {
1598 queued_pkts: Vec::new(),
1599 slice_buf: Vec::new(),
1600 str_data: CommonStreamData::new(),
1605 static RM_VIDEO_CODEC_REGISTER: &'static [(&[u8;4], &str)] = &[
1606 (b"RV10", "realvideo1"),
1607 (b"RV20", "realvideo2"),
1608 (b"RVTR", "realvideo2"),
1609 (b"RV30", "realvideo3"),
1610 (b"RV40", "realvideo4"),
1611 (b"RV60", "realvideo6"),
1612 (b"CLV1", "clearvideo_rm"),
1615 static RM_AUDIO_CODEC_REGISTER: &'static [(&[u8;4], &str)] = &[
1616 (b"lpcJ", "ra14.4"),
1617 (b"28_8", "ra28.8"),
1621 (b"atrc", "atrac3"),
1627 pub struct RealMediaDemuxerCreator { }
1629 impl DemuxerCreator for RealMediaDemuxerCreator {
1630 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
1631 Box::new(RealMediaDemuxer::new(br))
1633 fn get_name(&self) -> &'static str { "realmedia" }
1636 pub struct RealAudioDemuxerCreator { }
1638 impl DemuxerCreator for RealAudioDemuxerCreator {
1639 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
1640 Box::new(RealAudioDemuxer::new(br))
1642 fn get_name(&self) -> &'static str { "realaudio" }
1645 pub struct RealIVRDemuxerCreator { }
1647 impl DemuxerCreator for RealIVRDemuxerCreator {
1648 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
1649 Box::new(RealIVRDemuxer::new(br))
1651 fn get_name(&self) -> &'static str { "real_ivr" }
1660 fn test_rm_demux() {
1662 File::open("assets/RV/rv10_dnet_640x352_realvideo_encoder_4.0.rm").unwrap();
1663 // File::open("assets/RV/rv20_cook_640x352_realproducer_plus_8.51.rm").unwrap();
1664 // File::open("assets/RV/rv20_svt_atrc_640x352_realproducer_plus_8.51.rm").unwrap();
1665 // File::open("assets/RV/rv30_atrc_384x208_realproducer_plus_8.51.rm").unwrap();
1666 // File::open("assets/RV/rv30_chroma_drift.rm").unwrap();
1667 // File::open("assets/RV/rv30_weighted_mc.rm").unwrap();
1668 // File::open("assets/RV/rv40_weighted_mc.rmvb").unwrap();
1669 // File::open("assets/RV/rv40_weighted_mc_2.rmvb").unwrap();
1670 // File::open("assets/RV/clv1_sipr_384x208_realvideo_encoder_4.0.rm").unwrap();
1671 // File::open("assets/RV/luckynight.rmvb").unwrap();
1672 // File::open("assets/RV/rv40_ralf.rmvb").unwrap();
1673 let mut fr = FileReader::new_read(&mut file);
1674 let mut br = ByteReader::new(&mut fr);
1675 let mut dmx = RealMediaDemuxer::new(&mut br);
1676 let mut sm = StreamManager::new();
1677 let mut si = SeekIndex::new();
1678 dmx.open(&mut sm, &mut si).unwrap();
1681 let pktres = dmx.get_frame(&mut sm);
1682 if let Err(e) = pktres {
1683 if e == DemuxerError::EOF { break; }
1686 let pkt = pktres.unwrap();
1687 println!("Got {}", pkt);
1689 //panic!("the end");
1692 fn test_ra_demux() {
1694 // File::open("assets/RV/welcome288.ra").unwrap();
1695 File::open("assets/RV/diemusik.ra").unwrap();
1696 let mut fr = FileReader::new_read(&mut file);
1697 let mut br = ByteReader::new(&mut fr);
1698 let mut dmx = RealAudioDemuxer::new(&mut br);
1699 let mut sm = StreamManager::new();
1700 let mut si = SeekIndex::new();
1701 dmx.open(&mut sm, &mut si).unwrap();
1704 let pktres = dmx.get_frame(&mut sm);
1705 if let Err(e) = pktres {
1706 if e == DemuxerError::EOF { break; }
1709 let pkt = pktres.unwrap();
1710 println!("Got {}", pkt);
1714 fn test_ivr_demux() {
1716 File::open("assets/RV/Opener_rm_hi.ivr").unwrap();
1717 // File::open("assets/RV/SherwinWilliamsCommercial.ivr").unwrap();
1718 let mut fr = FileReader::new_read(&mut file);
1719 let mut br = ByteReader::new(&mut fr);
1720 let mut dmx = RealIVRDemuxer::new(&mut br);
1721 let mut sm = StreamManager::new();
1722 let mut si = SeekIndex::new();
1723 dmx.open(&mut sm, &mut si).unwrap();
1726 let pktres = dmx.get_frame(&mut sm);
1727 if let Err(e) = pktres {
1728 if e == DemuxerError::EOF { break; }
1731 let pkt = pktres.unwrap();
1732 println!("Got {}", pkt);