]>
Commit | Line | Data |
---|---|---|
1 | use nihav_core::demuxers::*; | |
2 | ||
3 | struct IVFDemuxer<'a> { | |
4 | src: &'a mut ByteReader<'a>, | |
5 | nframes: u32, | |
6 | frameno: u32, | |
7 | } | |
8 | ||
9 | impl<'a> IVFDemuxer<'a> { | |
10 | fn new(src: &'a mut ByteReader<'a>) -> Self { | |
11 | IVFDemuxer { | |
12 | src, | |
13 | nframes: 0, | |
14 | frameno: 0, | |
15 | } | |
16 | } | |
17 | } | |
18 | ||
19 | impl<'a> DemuxCore<'a> for IVFDemuxer<'a> { | |
20 | fn open(&mut self, strmgr: &mut StreamManager, _seek_index: &mut SeekIndex) -> DemuxerResult<()> { | |
21 | let tag = self.src.read_tag()?; | |
22 | validate!(&tag == b"DKIF"); | |
23 | let ver = self.src.read_u16le()?; | |
24 | validate!(ver == 0); | |
25 | let hdr_len = self.src.read_u16le()? as usize; | |
26 | validate!(hdr_len >= 32); | |
27 | let fcc = self.src.read_tag()?; | |
28 | let codec_name = match &fcc { | |
29 | b"VP80" => "vp8", | |
30 | _ => "unknown", | |
31 | }; | |
32 | let width = self.src.read_u16le()? as usize; | |
33 | let height = self.src.read_u16le()? as usize; | |
34 | validate!(width > 0 && height > 0); | |
35 | let tb_den = self.src.read_u32le()?; | |
36 | let tb_num = self.src.read_u32le()?; | |
37 | self.nframes = self.src.read_u32le()?; | |
38 | self.frameno = 0; | |
39 | ||
40 | self.src.seek(SeekFrom::Start(hdr_len as u64))?; | |
41 | ||
42 | let vci = NACodecTypeInfo::Video(NAVideoInfo::new(width, height, false, YUV420_FORMAT)); | |
43 | let vinfo = NACodecInfo::new(codec_name, vci, None); | |
44 | if strmgr.add_stream(NAStream::new(StreamType::Video, 0, vinfo, tb_num, tb_den, u64::from(self.nframes))).is_none() { | |
45 | return Err(DemuxerError::MemoryError); | |
46 | } | |
47 | ||
48 | Ok(()) | |
49 | } | |
50 | ||
51 | fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> { | |
52 | while self.frameno < self.nframes { | |
53 | let fsize = self.src.read_u32le()? as usize; | |
54 | let tstamp = self.src.read_u64le()?; | |
55 | self.frameno += 1; | |
56 | ||
57 | if fsize == 0 { | |
58 | continue; | |
59 | } | |
60 | ||
61 | if let Some(stream) = strmgr.get_stream(0) { | |
62 | let (tb_num, tb_den) = stream.get_timebase(); | |
63 | let ts = NATimeInfo::new(Some(tstamp), None, None, tb_num, tb_den); | |
64 | return self.src.read_packet(stream, ts, false, fsize); | |
65 | } else { | |
66 | return Err(DemuxerError::InvalidData); | |
67 | } | |
68 | } | |
69 | Err(DemuxerError::EOF) | |
70 | } | |
71 | ||
72 | fn seek(&mut self, _time: NATimePoint, _seek_index: &SeekIndex) -> DemuxerResult<()> { | |
73 | Err(DemuxerError::NotPossible) | |
74 | } | |
75 | fn get_duration(&self) -> u64 { 0 } | |
76 | } | |
77 | ||
78 | impl<'a> NAOptionHandler for IVFDemuxer<'a> { | |
79 | fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] } | |
80 | fn set_options(&mut self, _options: &[NAOption]) { } | |
81 | fn query_option_value(&self, _name: &str) -> Option<NAValue> { None } | |
82 | } | |
83 | ||
84 | pub struct IVFDemuxerCreator { } | |
85 | ||
86 | impl DemuxerCreator for IVFDemuxerCreator { | |
87 | fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> { | |
88 | Box::new(IVFDemuxer::new(br)) | |
89 | } | |
90 | fn get_name(&self) -> &'static str { "ivf" } | |
91 | } | |
92 | ||
93 | #[cfg(test)] | |
94 | mod test { | |
95 | use super::*; | |
96 | use std::fs::File; | |
97 | ||
98 | #[test] | |
99 | fn test_ivf_demux() { | |
100 | let mut file = File::open("assets/Duck/VP8/vp80-00-comprehensive-001.ivf").unwrap(); | |
101 | let mut fr = FileReader::new_read(&mut file); | |
102 | let mut br = ByteReader::new(&mut fr); | |
103 | let mut dmx = IVFDemuxer::new(&mut br); | |
104 | let mut sm = StreamManager::new(); | |
105 | let mut si = SeekIndex::new(); | |
106 | dmx.open(&mut sm, &mut si).unwrap(); | |
107 | ||
108 | loop { | |
109 | let pktres = dmx.get_frame(&mut sm); | |
110 | if let Err(e) = pktres { | |
111 | if e == DemuxerError::EOF { break; } | |
112 | panic!("error"); | |
113 | } | |
114 | let pkt = pktres.unwrap(); | |
115 | println!("Got {}", pkt); | |
116 | } | |
117 | } | |
118 | } |