X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=src%2Fdemuxers%2Fmod.rs;h=943ca5463bd4b00f6f82fa55aceb91338bf59506;hb=HEAD;hp=5e337a50d4311ce56f56d16c2ce91370247b3a25;hpb=3f7c7cfd270b5b75917508aee6ca5877433a984d;p=nihav.git diff --git a/src/demuxers/mod.rs b/src/demuxers/mod.rs deleted file mode 100644 index 5e337a5..0000000 --- a/src/demuxers/mod.rs +++ /dev/null @@ -1,225 +0,0 @@ -#[cfg(feature="demuxer_gdv")] -pub mod gdv; -#[cfg(feature="demuxer_avi")] -pub mod avi; - -use std::fmt; -use std::rc::Rc; -use frame::*; -use std::collections::HashMap; -use io::byteio::*; - -/// Possible stream types. -#[derive(Debug,Clone,Copy)] -#[allow(dead_code)] -pub enum StreamType { - /// video stream - Video, - /// audio stream - Audio, - /// subtitles - Subtitles, - /// any data stream (or might be an unrecognized audio/video stream) - Data, - /// nonexistent stream - None, -} - -impl fmt::Display for StreamType { - fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { - match *self { - StreamType::Video => write!(f, "Video"), - StreamType::Audio => write!(f, "Audio"), - StreamType::Subtitles => write!(f, "Subtitles"), - StreamType::Data => write!(f, "Data"), - StreamType::None => write!(f, "-"), - } - } -} - - -#[allow(dead_code)] -#[derive(Clone)] -pub struct NAStream { - media_type: StreamType, - id: u32, - num: usize, - info: Rc, -} - -impl NAStream { - pub fn new(mt: StreamType, id: u32, info: NACodecInfo) -> Self { - NAStream { media_type: mt, id: id, num: 0, info: Rc::new(info) } - } - pub fn get_id(&self) -> u32 { self.id } - pub fn get_num(&self) -> usize { self.num } - pub fn set_num(&mut self, num: usize) { self.num = num; } - pub fn get_info(&self) -> Rc { self.info.clone() } -} - -impl fmt::Display for NAStream { - fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { - write!(f, "({}#{} - {})", self.media_type, self.id, self.info.get_properties()) - } -} - -#[allow(dead_code)] -pub struct NAPacket { - stream: Rc, - pts: Option, - dts: Option, - duration: Option, - buffer: Rc>, - keyframe: bool, -// options: HashMap>, -} - -impl NAPacket { - pub fn new(str: Rc, pts: Option, dts: Option, dur: Option, kf: bool, vec: Vec) -> Self { -// let mut vec: Vec = Vec::new(); -// vec.resize(size, 0); - NAPacket { stream: str, pts: pts, dts: dts, duration: dur, keyframe: kf, buffer: Rc::new(vec) } - } - pub fn get_stream(&self) -> Rc { self.stream.clone() } - pub fn get_pts(&self) -> Option { self.pts } - pub fn get_dts(&self) -> Option { self.dts } - pub fn get_duration(&self) -> Option { self.duration } - pub fn is_keyframe(&self) -> bool { self.keyframe } - pub fn get_buffer(&self) -> Rc> { self.buffer.clone() } -} - -impl Drop for NAPacket { - fn drop(&mut self) {} -} - -impl fmt::Display for NAPacket { - fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { - let mut foo = format!("[pkt for {} size {}", self.stream, self.buffer.len()); - if let Some(pts) = self.pts { foo = format!("{} pts {}", foo, pts); } - if let Some(dts) = self.dts { foo = format!("{} dts {}", foo, dts); } - if let Some(dur) = self.duration { foo = format!("{} duration {}", foo, dur); } - if self.keyframe { foo = format!("{} kf", foo); } - foo = foo + "]"; - write!(f, "{}", foo) - } -} - -#[derive(Debug)] -#[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, pts: Option, dts: Option, dur: Option, 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, pts: Option, dts: Option, dur: Option, 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, pts, dts, dur, 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(()) - } -} - -pub 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 - } - } - 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 } -} - -pub trait FrameFromPacket { - fn new_from_pkt(pkt: &NAPacket, info: Rc) -> NAFrame; - fn fill_timestamps(&mut self, pkt: &NAPacket); -} - -impl FrameFromPacket for NAFrame { - fn new_from_pkt(pkt: &NAPacket, info: Rc) -> NAFrame { - NAFrame::new(pkt.pts, pkt.dts, pkt.duration, info, HashMap::new()) - } - fn fill_timestamps(&mut self, pkt: &NAPacket) { - self.set_pts(pkt.pts); - self.set_dts(pkt.dts); - self.set_duration(pkt.duration); - } -} - -///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 -}