X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=src%2Fdemuxers%2Fmod.rs;h=1b044a642f0a1643c051cf10efb5ce45d8d00cfd;hb=HEAD;hp=aaf92e4cf36d421b8eeb3befabb7e2aa20a884cd;hpb=e189501ed92ffd55123f9fceb7e4c78ddf1be936;p=nihav.git diff --git a/src/demuxers/mod.rs b/src/demuxers/mod.rs deleted file mode 100644 index aaf92e4..0000000 --- a/src/demuxers/mod.rs +++ /dev/null @@ -1,113 +0,0 @@ -#[cfg(feature="demuxer_gdv")] -pub mod gdv; -#[cfg(feature="demuxer_avi")] -pub mod avi; - -use std::rc::Rc; -use frame::*; -use io::byteio::*; - -#[derive(Debug,Clone,Copy,PartialEq)] -#[allow(dead_code)] -pub enum DemuxerError { - EOF, - NoSuchInput, - InvalidData, - IOError, - NotImplemented, - MemoryError, -} - -type DemuxerResult = Result; - -pub trait Demux<'a> { - fn open(&mut self) -> DemuxerResult<()>; - fn get_num_streams(&self) -> usize; - fn get_stream(&self, idx: usize) -> Option>; - fn get_frame(&mut self) -> DemuxerResult; - fn seek(&mut self, time: u64) -> DemuxerResult<()>; -} - -pub trait NAPacketReader { - fn read_packet(&mut self, str: Rc, ts: NATimeInfo, keyframe: bool, size: usize) -> DemuxerResult; - fn fill_packet(&mut self, pkt: &mut NAPacket) -> DemuxerResult<()>; -} - -impl<'a> NAPacketReader for ByteReader<'a> { - fn read_packet(&mut self, str: Rc, ts: NATimeInfo, kf: bool, size: usize) -> DemuxerResult { - let mut buf: Vec = Vec::with_capacity(size); - if buf.capacity() < size { return Err(DemuxerError::MemoryError); } - buf.resize(size, 0); - let res = self.read_buf(buf.as_mut_slice()); - if let Err(_) = res { return Err(DemuxerError::IOError); } - let pkt = NAPacket::new(str, ts, kf, buf); - Ok(pkt) - } - fn fill_packet(&mut self, pkt: &mut NAPacket) -> DemuxerResult<()> { - let mut refbuf = pkt.get_buffer(); - let mut buf = Rc::make_mut(&mut refbuf); - let res = self.read_buf(buf.as_mut_slice()); - if let Err(_) = res { return Err(DemuxerError::IOError); } - Ok(()) - } -} - -struct Demuxer { - streams: Vec>, -} - -impl Demuxer { - pub fn new() -> Self { Demuxer { streams: Vec::new() } } - pub fn add_stream(&mut self, stream: NAStream) -> Option { - let stream_num = self.streams.len(); - let mut str = stream.clone(); - str.set_num(stream_num); - self.streams.push(Rc::new(str)); - Some(stream_num) - } - pub fn get_stream(&self, idx: usize) -> Option> { - if idx < self.streams.len() { - Some(self.streams[idx].clone()) - } else { - None - } - } - #[allow(dead_code)] - pub fn get_stream_by_id(&self, id: u32) -> Option> { - for i in 0..self.streams.len() { - if self.streams[i].get_id() == id { - return Some(self.streams[i].clone()); - } - } - None - } - pub fn get_num_streams(&self) -> usize { self.streams.len() } -} - -impl From for DemuxerError { - fn from(_: ByteIOError) -> Self { DemuxerError::IOError } -} - -///The structure used to create demuxers. -pub trait DemuxerCreator { - /// Create new demuxer instance that will use `ByteReader` source as an input. - fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box + 'a>; - /// Get the name of current demuxer creator. - fn get_name(&self) -> &'static str; -} - -const DEMUXERS: &[&'static DemuxerCreator] = &[ -#[cfg(feature="demuxer_avi")] - &avi::AVIDemuxerCreator {}, -#[cfg(feature="demuxer_gdv")] - &gdv::GDVDemuxerCreator {}, -]; - -pub fn find_demuxer(name: &str) -> Option<&DemuxerCreator> { - for &dmx in DEMUXERS { - if dmx.get_name() == name { - return Some(dmx); - } - } - None -}