X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=src%2Fdemuxers%2Fmod.rs;h=943ca5463bd4b00f6f82fa55aceb91338bf59506;hb=HEAD;hp=14aac9380761bff34c4ff1fdb6161fca95f3b373;hpb=20ef4353905883ca289782858ccfcd7d2146fa42;p=nihav.git diff --git a/src/demuxers/mod.rs b/src/demuxers/mod.rs deleted file mode 100644 index 14aac93..0000000 --- a/src/demuxers/mod.rs +++ /dev/null @@ -1,173 +0,0 @@ -pub mod gdv; - -use std::fmt; -use std::rc::Rc; -use frame::*; -//use std::collections::HashMap; -use io::byteio::*; - -#[derive(Debug,Clone,Copy)] -#[allow(dead_code)] -pub enum StreamType { - Video, - Audio, - Subtitles, - Data, -} - -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"), - } - } -} - - -#[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) - } -} - -#[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_buffer(&self) -> Rc> { self.buffer.clone() } -} - -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_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); } - if res.unwrap() < buf.len() { 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); } - if res.unwrap() < buf.len() { 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 - } -} - -impl From for DemuxerError { - fn from(_: ByteIOError) -> Self { DemuxerError::IOError } -} - -//impl NADemuxerBuilder { -// #[allow(unused_variables)] -// pub fn create_demuxer(name: &str, url: &str) -> DemuxerResult>> { -// unimplemented!() -// } -//}