]>
Commit | Line | Data |
---|---|---|
1 | //! Demuxer definitions. | |
2 | pub use crate::frame::*; | |
3 | pub use crate::io::byteio::*; | |
4 | pub use crate::options::*; | |
5 | ||
6 | /// A list specifying general demuxing errors. | |
7 | #[derive(Debug,Clone,Copy,PartialEq)] | |
8 | #[allow(dead_code)] | |
9 | pub enum DemuxerError { | |
10 | /// Reader got to end of stream. | |
11 | EOF, | |
12 | /// Demuxer encountered empty container. | |
13 | NoSuchInput, | |
14 | /// Demuxer encountered invalid input data. | |
15 | InvalidData, | |
16 | /// Data reading error. | |
17 | IOError, | |
18 | /// Feature is not implemented. | |
19 | NotImplemented, | |
20 | /// Allocation failed. | |
21 | MemoryError, | |
22 | /// The operation should be repeated. | |
23 | TryAgain, | |
24 | /// Seeking failed. | |
25 | SeekError, | |
26 | /// Operation cannot succeed in principle (e.g. seeking in a format not supporting seeking). | |
27 | NotPossible, | |
28 | } | |
29 | ||
30 | /// A specialised `Result` type for demuxing operations. | |
31 | pub type DemuxerResult<T> = Result<T, DemuxerError>; | |
32 | ||
33 | /// A trait for demuxing operations. | |
34 | pub trait DemuxCore<'a>: NAOptionHandler { | |
35 | /// Opens the input stream, reads required headers and prepares everything for packet demuxing. | |
36 | fn open(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()>; | |
37 | /// Demuxes a packet. | |
38 | fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket>; | |
39 | /// Seeks to the requested time. | |
40 | fn seek(&mut self, time: NATimePoint, seek_idx: &SeekIndex) -> DemuxerResult<()>; | |
41 | /// Returns container duration in milliseconds (zero if not available). | |
42 | fn get_duration(&self) -> u64; | |
43 | } | |
44 | ||
45 | /// An auxiliary trait to make bytestream reader read packet data. | |
46 | pub trait NAPacketReader { | |
47 | /// Reads input and constructs a packet containing it. | |
48 | fn read_packet(&mut self, strm: NAStreamRef, ts: NATimeInfo, keyframe: bool, size: usize) -> DemuxerResult<NAPacket>; | |
49 | /// Reads input into already existing packet. | |
50 | fn fill_packet(&mut self, pkt: &mut NAPacket) -> DemuxerResult<()>; | |
51 | } | |
52 | ||
53 | impl<'a> NAPacketReader for ByteReader<'a> { | |
54 | fn read_packet(&mut self, strm: NAStreamRef, ts: NATimeInfo, kf: bool, size: usize) -> DemuxerResult<NAPacket> { | |
55 | let mut buf: Vec<u8> = Vec::with_capacity(size); | |
56 | if buf.capacity() < size { return Err(DemuxerError::MemoryError); } | |
57 | buf.resize(size, 0); | |
58 | self.read_buf(buf.as_mut_slice())?; | |
59 | let pkt = NAPacket::new(strm, ts, kf, buf); | |
60 | Ok(pkt) | |
61 | } | |
62 | fn fill_packet(&mut self, pkt: &mut NAPacket) -> DemuxerResult<()> { | |
63 | let mut refbuf = pkt.get_buffer(); | |
64 | let buf = refbuf.as_mut().unwrap(); | |
65 | self.read_buf(buf.as_mut_slice())?; | |
66 | Ok(()) | |
67 | } | |
68 | } | |
69 | ||
70 | /// An auxiliary structure for operations with individual streams inside the container. | |
71 | #[derive(Default)] | |
72 | pub struct StreamManager { | |
73 | streams: Vec<NAStreamRef>, | |
74 | ignored: Vec<bool>, | |
75 | no_ign: bool, | |
76 | } | |
77 | ||
78 | impl StreamManager { | |
79 | /// Constructs a new instance of `StreamManager`. | |
80 | pub fn new() -> Self { | |
81 | StreamManager { | |
82 | streams: Vec::new(), | |
83 | ignored: Vec::new(), | |
84 | no_ign: true, | |
85 | } | |
86 | } | |
87 | /// Returns stream iterator. | |
88 | pub fn iter(&self) -> StreamIter { StreamIter::new(&self.streams) } | |
89 | ||
90 | /// Adds a new stream. | |
91 | pub fn add_stream(&mut self, stream: NAStream) -> Option<usize> { | |
92 | let stream_num = self.streams.len(); | |
93 | let mut stream = stream; | |
94 | stream.set_num(stream_num); | |
95 | self.streams.push(stream.into_ref()); | |
96 | self.ignored.push(false); | |
97 | Some(stream_num) | |
98 | } | |
99 | /// Adds a new stream from reference-counted object. | |
100 | pub fn add_stream_ref(&mut self, stream: NAStreamRef) -> Option<usize> { | |
101 | let stream_num = self.streams.len(); | |
102 | self.streams.push(stream); | |
103 | self.ignored.push(false); | |
104 | Some(stream_num) | |
105 | } | |
106 | /// Returns stream with the requested index. | |
107 | pub fn get_stream(&self, idx: usize) -> Option<NAStreamRef> { | |
108 | if idx < self.streams.len() { | |
109 | Some(self.streams[idx].clone()) | |
110 | } else { | |
111 | None | |
112 | } | |
113 | } | |
114 | /// Returns stream with the requested stream ID. | |
115 | pub fn get_stream_by_id(&self, id: u32) -> Option<NAStreamRef> { | |
116 | for i in 0..self.streams.len() { | |
117 | if self.streams[i].get_id() == id { | |
118 | return Some(self.streams[i].clone()); | |
119 | } | |
120 | } | |
121 | None | |
122 | } | |
123 | /// Returns the number of known streams. | |
124 | pub fn get_num_streams(&self) -> usize { self.streams.len() } | |
125 | /// Reports whether the stream is marked as ignored. | |
126 | pub fn is_ignored(&self, idx: usize) -> bool { | |
127 | if self.no_ign { | |
128 | true | |
129 | } else if idx < self.ignored.len() { | |
130 | self.ignored[idx] | |
131 | } else { | |
132 | false | |
133 | } | |
134 | } | |
135 | /// Reports whether the stream with certain ID is marked as ignored. | |
136 | pub fn is_ignored_id(&self, id: u32) -> bool { | |
137 | for i in 0..self.streams.len() { | |
138 | if self.streams[i].get_id() == id { | |
139 | return self.ignored[i]; | |
140 | } | |
141 | } | |
142 | false | |
143 | } | |
144 | /// Marks requested stream as ignored. | |
145 | pub fn set_ignored(&mut self, idx: usize) { | |
146 | if idx < self.ignored.len() { | |
147 | self.ignored[idx] = true; | |
148 | self.no_ign = false; | |
149 | } | |
150 | } | |
151 | /// Clears the ignored mark for the requested stream. | |
152 | pub fn set_unignored(&mut self, idx: usize) { | |
153 | if idx < self.ignored.len() { | |
154 | self.ignored[idx] = false; | |
155 | } | |
156 | } | |
157 | } | |
158 | ||
159 | /// Stream iterator. | |
160 | pub struct StreamIter<'a> { | |
161 | streams: &'a [NAStreamRef], | |
162 | pos: usize, | |
163 | } | |
164 | ||
165 | impl<'a> StreamIter<'a> { | |
166 | /// Constructs a new instance of `StreamIter`. | |
167 | pub fn new(streams: &'a [NAStreamRef]) -> Self { | |
168 | StreamIter { streams, pos: 0 } | |
169 | } | |
170 | } | |
171 | ||
172 | impl<'a> Iterator for StreamIter<'a> { | |
173 | type Item = NAStreamRef; | |
174 | ||
175 | fn next(&mut self) -> Option<Self::Item> { | |
176 | if self.pos >= self.streams.len() { return None; } | |
177 | let ret = self.streams[self.pos].clone(); | |
178 | self.pos += 1; | |
179 | Some(ret) | |
180 | } | |
181 | } | |
182 | ||
183 | /// Seeking modes. | |
184 | #[derive(Clone,Copy,PartialEq,Default)] | |
185 | pub enum SeekIndexMode { | |
186 | /// No seeking index present. | |
187 | #[default] | |
188 | None, | |
189 | /// Seeking index is present. | |
190 | Present, | |
191 | /// Seeking index should be constructed by the demuxer if possible. | |
192 | Automatic, | |
193 | } | |
194 | ||
195 | /// A structure holding seeking information. | |
196 | #[derive(Clone,Copy,Default)] | |
197 | pub struct SeekEntry { | |
198 | /// Time in milliseconds. | |
199 | pub time: u64, | |
200 | /// PTS | |
201 | pub pts: u64, | |
202 | /// Position in file. | |
203 | pub pos: u64, | |
204 | } | |
205 | ||
206 | /// Seeking information for individual streams. | |
207 | #[derive(Clone)] | |
208 | pub struct StreamSeekInfo { | |
209 | /// Stream ID. | |
210 | pub id: u32, | |
211 | /// Index is present. | |
212 | pub filled: bool, | |
213 | /// Packet seeking information. | |
214 | pub entries: Vec<SeekEntry>, | |
215 | } | |
216 | ||
217 | impl StreamSeekInfo { | |
218 | /// Constructs a new `StreamSeekInfo` instance. | |
219 | pub fn new(id: u32) -> Self { | |
220 | Self { | |
221 | id, | |
222 | filled: false, | |
223 | entries: Vec::new(), | |
224 | } | |
225 | } | |
226 | /// Adds new seeking point. | |
227 | pub fn add_entry(&mut self, entry: SeekEntry) { | |
228 | self.entries.push(entry); | |
229 | } | |
230 | /// Searches for an appropriate seek position before requested time. | |
231 | pub fn find_pos(&self, time: NATimePoint) -> Option<SeekEntry> { | |
232 | if time == NATimePoint::None { | |
233 | return None; | |
234 | } | |
235 | if !self.entries.is_empty() { | |
236 | // todo something faster like binary search | |
237 | let mut cand = None; | |
238 | for entry in self.entries.iter() { | |
239 | match time { | |
240 | NATimePoint::Milliseconds(ms) => { | |
241 | if entry.time <= ms { | |
242 | cand = Some(*entry); | |
243 | } else { | |
244 | break; | |
245 | } | |
246 | }, | |
247 | NATimePoint::PTS(pts) => { | |
248 | if entry.pts <= pts { | |
249 | cand = Some(*entry); | |
250 | } else { | |
251 | break; | |
252 | } | |
253 | }, | |
254 | NATimePoint::None => unreachable!(), | |
255 | }; | |
256 | } | |
257 | cand | |
258 | } else { | |
259 | None | |
260 | } | |
261 | } | |
262 | } | |
263 | ||
264 | /// Structure for holding seeking point search results. | |
265 | #[derive(Clone,Copy,Default)] | |
266 | pub struct SeekIndexResult { | |
267 | /// Packet PTS. | |
268 | pub pts: u64, | |
269 | /// Position in file. | |
270 | pub pos: u64, | |
271 | /// Stream ID. | |
272 | pub str_id: u32, | |
273 | } | |
274 | ||
275 | /// Seek information for the whole container. | |
276 | #[derive(Default)] | |
277 | pub struct SeekIndex { | |
278 | /// Seek information for individual streams. | |
279 | pub seek_info: Vec<StreamSeekInfo>, | |
280 | /// Seeking index mode. | |
281 | pub mode: SeekIndexMode, | |
282 | /// Ignore index flag. | |
283 | pub skip_index: bool, | |
284 | } | |
285 | ||
286 | impl SeekIndex { | |
287 | /// Constructs a new `SeekIndex` instance. | |
288 | pub fn new() -> Self { Self::default() } | |
289 | pub fn add_stream(&mut self, id: u32) -> usize { | |
290 | let ret = self.stream_id_to_index(id); | |
291 | if let Some(res) = ret { | |
292 | res | |
293 | } else { | |
294 | self.seek_info.push(StreamSeekInfo::new(id)); | |
295 | self.seek_info.len() - 1 | |
296 | } | |
297 | } | |
298 | /// Adds a new stream to the index. | |
299 | pub fn stream_id_to_index(&self, id: u32) -> Option<usize> { | |
300 | for (idx, strm) in self.seek_info.iter().enumerate() { | |
301 | if strm.id == id { | |
302 | return Some(idx); | |
303 | } | |
304 | } | |
305 | None | |
306 | } | |
307 | /// Returns stream reference for provided stream ID. | |
308 | pub fn get_stream_index(&mut self, id: u32) -> Option<&mut StreamSeekInfo> { | |
309 | self.seek_info.iter_mut().find(|stream| stream.id == id) | |
310 | } | |
311 | /// Adds seeking information to the index. | |
312 | pub fn add_entry(&mut self, id: u32, entry: SeekEntry) { | |
313 | let mut idx = self.stream_id_to_index(id); | |
314 | if idx.is_none() { | |
315 | idx = Some(self.add_stream(id)); | |
316 | } | |
317 | self.seek_info[idx.unwrap()].add_entry(entry); | |
318 | self.seek_info[idx.unwrap()].filled = true; | |
319 | } | |
320 | /// Searches for a seek position before requested time. | |
321 | pub fn find_pos(&self, time: NATimePoint) -> Option<SeekIndexResult> { | |
322 | let mut cand = None; | |
323 | for stream in self.seek_info.iter() { | |
324 | if !stream.filled { continue; } | |
325 | let res = stream.find_pos(time); | |
326 | if res.is_none() { continue; } | |
327 | let res = res.unwrap(); | |
328 | if cand.is_none() { | |
329 | cand = Some(SeekIndexResult { pts: res.pts, pos: res.pos, str_id: stream.id }); | |
330 | } else if let Some(entry) = cand { | |
331 | if res.pos < entry.pos { | |
332 | cand = Some(SeekIndexResult { pts: res.pts, pos: res.pos, str_id: stream.id }); | |
333 | } | |
334 | } | |
335 | } | |
336 | cand | |
337 | } | |
338 | } | |
339 | ||
340 | /// Demuxer structure with auxiliary data. | |
341 | pub struct Demuxer<'a> { | |
342 | dmx: Box<dyn DemuxCore<'a> + 'a>, | |
343 | streams: StreamManager, | |
344 | seek_idx: SeekIndex, | |
345 | } | |
346 | ||
347 | impl<'a> Demuxer<'a> { | |
348 | /// Constructs a new `Demuxer` instance. | |
349 | fn new(dmx: Box<dyn DemuxCore<'a> + 'a>, strmgr: StreamManager, seek_idx: SeekIndex) -> Self { | |
350 | Demuxer { | |
351 | dmx, | |
352 | streams: strmgr, | |
353 | seek_idx, | |
354 | } | |
355 | } | |
356 | /// Returns a stream reference by its number. | |
357 | pub fn get_stream(&self, idx: usize) -> Option<NAStreamRef> { | |
358 | self.streams.get_stream(idx) | |
359 | } | |
360 | /// Returns a stream reference by its ID. | |
361 | pub fn get_stream_by_id(&self, id: u32) -> Option<NAStreamRef> { | |
362 | self.streams.get_stream_by_id(id) | |
363 | } | |
364 | /// Reports the total number of streams. | |
365 | pub fn get_num_streams(&self) -> usize { | |
366 | self.streams.get_num_streams() | |
367 | } | |
368 | /// Returns a reference to the internal stream manager. | |
369 | pub fn get_stream_manager(&self) -> &StreamManager { | |
370 | &self.streams | |
371 | } | |
372 | /// Returns an iterator over streams. | |
373 | pub fn get_streams(&self) -> StreamIter { | |
374 | self.streams.iter() | |
375 | } | |
376 | /// Returns 'ignored' marker for requested stream. | |
377 | pub fn is_ignored_stream(&self, idx: usize) -> bool { | |
378 | self.streams.is_ignored(idx) | |
379 | } | |
380 | /// Sets 'ignored' marker for requested stream. | |
381 | pub fn set_ignored_stream(&mut self, idx: usize) { | |
382 | self.streams.set_ignored(idx) | |
383 | } | |
384 | /// Clears 'ignored' marker for requested stream. | |
385 | pub fn set_unignored_stream(&mut self, idx: usize) { | |
386 | self.streams.set_unignored(idx) | |
387 | } | |
388 | ||
389 | /// Demuxes a new packet from the container. | |
390 | pub fn get_frame(&mut self) -> DemuxerResult<NAPacket> { | |
391 | loop { | |
392 | let res = self.dmx.get_frame(&mut self.streams); | |
393 | if self.streams.no_ign || res.is_err() { return res; } | |
394 | let res = res.unwrap(); | |
395 | let idx = res.get_stream().get_num(); | |
396 | if !self.is_ignored_stream(idx) { | |
397 | return Ok(res); | |
398 | } | |
399 | } | |
400 | } | |
401 | /// Seeks to the requested time if possible. | |
402 | pub fn seek(&mut self, time: NATimePoint) -> DemuxerResult<()> { | |
403 | if self.seek_idx.skip_index { | |
404 | return Err(DemuxerError::NotPossible); | |
405 | } | |
406 | self.dmx.seek(time, &self.seek_idx) | |
407 | } | |
408 | /// Returns internal seek index. | |
409 | pub fn get_seek_index(&self) -> &SeekIndex { | |
410 | &self.seek_idx | |
411 | } | |
412 | /// Returns media duration reported by container or its streams. | |
413 | /// | |
414 | /// Duration is in milliseconds and set to zero when it is not available. | |
415 | pub fn get_duration(&self) -> u64 { | |
416 | let duration = self.dmx.get_duration(); | |
417 | if duration != 0 { | |
418 | return duration; | |
419 | } | |
420 | let mut duration = 0; | |
421 | for stream in self.streams.iter() { | |
422 | if stream.duration > 0 { | |
423 | let dur = NATimeInfo::ts_to_time(stream.duration, 1000, stream.tb_num, stream.tb_den); | |
424 | if duration < dur { | |
425 | duration = dur; | |
426 | } | |
427 | } | |
428 | } | |
429 | duration | |
430 | } | |
431 | } | |
432 | ||
433 | impl<'a> NAOptionHandler for Demuxer<'a> { | |
434 | fn get_supported_options(&self) -> &[NAOptionDefinition] { | |
435 | self.dmx.get_supported_options() | |
436 | } | |
437 | fn set_options(&mut self, options: &[NAOption]) { | |
438 | self.dmx.set_options(options); | |
439 | } | |
440 | fn query_option_value(&self, name: &str) -> Option<NAValue> { | |
441 | self.dmx.query_option_value(name) | |
442 | } | |
443 | } | |
444 | ||
445 | impl From<ByteIOError> for DemuxerError { | |
446 | fn from(_: ByteIOError) -> Self { DemuxerError::IOError } | |
447 | } | |
448 | ||
449 | /// The trait for creating demuxers. | |
450 | pub trait DemuxerCreator { | |
451 | /// Creates new demuxer instance that will use `ByteReader` source as an input. | |
452 | fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a>; | |
453 | /// Returns the name of current demuxer creator (equal to the container name it can demux). | |
454 | fn get_name(&self) -> &'static str; | |
455 | } | |
456 | ||
457 | /// Creates demuxer for a provided bytestream. | |
458 | pub fn create_demuxer<'a>(dmxcr: &dyn DemuxerCreator, br: &'a mut ByteReader<'a>) -> DemuxerResult<Demuxer<'a>> { | |
459 | let mut dmx = dmxcr.new_demuxer(br); | |
460 | let mut strmgr = StreamManager::new(); | |
461 | let mut seek_idx = SeekIndex::new(); | |
462 | dmx.open(&mut strmgr, &mut seek_idx)?; | |
463 | Ok(Demuxer::new(dmx, strmgr, seek_idx)) | |
464 | } | |
465 | ||
466 | /// Creates demuxer for a provided bytestream with options applied right after its creation. | |
467 | pub fn create_demuxer_with_options<'a>(dmxcr: &dyn DemuxerCreator, br: &'a mut ByteReader<'a>, opts: &[NAOption]) -> DemuxerResult<Demuxer<'a>> { | |
468 | let mut dmx = dmxcr.new_demuxer(br); | |
469 | dmx.set_options(opts); | |
470 | let mut strmgr = StreamManager::new(); | |
471 | let mut seek_idx = SeekIndex::new(); | |
472 | dmx.open(&mut strmgr, &mut seek_idx)?; | |
473 | Ok(Demuxer::new(dmx, strmgr, seek_idx)) | |
474 | } | |
475 | ||
476 | /// List of registered demuxers. | |
477 | #[derive(Default)] | |
478 | pub struct RegisteredDemuxers { | |
479 | dmxs: Vec<&'static dyn DemuxerCreator>, | |
480 | } | |
481 | ||
482 | impl RegisteredDemuxers { | |
483 | /// Constructs a new `RegisteredDemuxers` instance. | |
484 | pub fn new() -> Self { | |
485 | Self { dmxs: Vec::new() } | |
486 | } | |
487 | /// Registers a new demuxer. | |
488 | pub fn add_demuxer(&mut self, dmx: &'static dyn DemuxerCreator) { | |
489 | self.dmxs.push(dmx); | |
490 | } | |
491 | /// Searches for a demuxer that supports requested container format. | |
492 | pub fn find_demuxer(&self, name: &str) -> Option<&dyn DemuxerCreator> { | |
493 | self.dmxs.iter().find(|&&dmx| dmx.get_name() == name).copied() | |
494 | } | |
495 | /// Provides an iterator over currently registered demuxers. | |
496 | pub fn iter(&self) -> std::slice::Iter<&dyn DemuxerCreator> { | |
497 | self.dmxs.iter() | |
498 | } | |
499 | } | |
500 | ||
501 | /// A trait for raw data demuxing operations. | |
502 | pub trait RawDemuxCore<'a>: NAOptionHandler { | |
503 | /// Opens the input stream, reads required headers and prepares everything for packet demuxing. | |
504 | fn open(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()>; | |
505 | /// Reads a piece of raw data. | |
506 | fn get_data(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NARawData>; | |
507 | /// Seeks to the requested time. | |
508 | fn seek(&mut self, time: NATimePoint, seek_idx: &SeekIndex) -> DemuxerResult<()>; | |
509 | /// Returns container duration in milliseconds (zero if not available). | |
510 | fn get_duration(&self) -> u64; | |
511 | } | |
512 | ||
513 | /// Demuxer structure with auxiliary data. | |
514 | pub struct RawDemuxer<'a> { | |
515 | dmx: Box<dyn RawDemuxCore<'a> + 'a>, | |
516 | streams: StreamManager, | |
517 | seek_idx: SeekIndex, | |
518 | } | |
519 | ||
520 | impl<'a> RawDemuxer<'a> { | |
521 | /// Constructs a new `Demuxer` instance. | |
522 | fn new(dmx: Box<dyn RawDemuxCore<'a> + 'a>, strmgr: StreamManager, seek_idx: SeekIndex) -> Self { | |
523 | Self { | |
524 | dmx, | |
525 | streams: strmgr, | |
526 | seek_idx, | |
527 | } | |
528 | } | |
529 | /// Returns a stream reference by its number. | |
530 | pub fn get_stream(&self, idx: usize) -> Option<NAStreamRef> { | |
531 | self.streams.get_stream(idx) | |
532 | } | |
533 | /// Returns a stream reference by its ID. | |
534 | pub fn get_stream_by_id(&self, id: u32) -> Option<NAStreamRef> { | |
535 | self.streams.get_stream_by_id(id) | |
536 | } | |
537 | /// Reports the total number of streams. | |
538 | pub fn get_num_streams(&self) -> usize { | |
539 | self.streams.get_num_streams() | |
540 | } | |
541 | /// Returns a reference to the internal stream manager. | |
542 | pub fn get_stream_manager(&self) -> &StreamManager { | |
543 | &self.streams | |
544 | } | |
545 | /// Returns an iterator over streams. | |
546 | pub fn get_streams(&self) -> StreamIter { | |
547 | self.streams.iter() | |
548 | } | |
549 | /// Returns 'ignored' marker for requested stream. | |
550 | pub fn is_ignored_stream(&self, idx: usize) -> bool { | |
551 | self.streams.is_ignored(idx) | |
552 | } | |
553 | /// Sets 'ignored' marker for requested stream. | |
554 | pub fn set_ignored_stream(&mut self, idx: usize) { | |
555 | self.streams.set_ignored(idx) | |
556 | } | |
557 | /// Clears 'ignored' marker for requested stream. | |
558 | pub fn set_unignored_stream(&mut self, idx: usize) { | |
559 | self.streams.set_unignored(idx) | |
560 | } | |
561 | ||
562 | /// Demuxes a new piece of data from the container. | |
563 | pub fn get_data(&mut self) -> DemuxerResult<NARawData> { | |
564 | loop { | |
565 | let res = self.dmx.get_data(&mut self.streams); | |
566 | if self.streams.no_ign || res.is_err() { return res; } | |
567 | let res = res.unwrap(); | |
568 | let idx = res.get_stream().get_num(); | |
569 | if !self.is_ignored_stream(idx) { | |
570 | return Ok(res); | |
571 | } | |
572 | } | |
573 | } | |
574 | /// Seeks to the requested time if possible. | |
575 | pub fn seek(&mut self, time: NATimePoint) -> DemuxerResult<()> { | |
576 | if self.seek_idx.skip_index { | |
577 | return Err(DemuxerError::NotPossible); | |
578 | } | |
579 | self.dmx.seek(time, &self.seek_idx) | |
580 | } | |
581 | /// Returns internal seek index. | |
582 | pub fn get_seek_index(&self) -> &SeekIndex { | |
583 | &self.seek_idx | |
584 | } | |
585 | /// Returns media duration reported by container or its streams. | |
586 | /// | |
587 | /// Duration is in milliseconds and set to zero when it is not available. | |
588 | pub fn get_duration(&self) -> u64 { | |
589 | let duration = self.dmx.get_duration(); | |
590 | if duration != 0 { | |
591 | return duration; | |
592 | } | |
593 | let mut duration = 0; | |
594 | for stream in self.streams.iter() { | |
595 | if stream.duration > 0 { | |
596 | let dur = NATimeInfo::ts_to_time(stream.duration, 1000, stream.tb_num, stream.tb_den); | |
597 | if duration < dur { | |
598 | duration = dur; | |
599 | } | |
600 | } | |
601 | } | |
602 | duration | |
603 | } | |
604 | } | |
605 | ||
606 | impl<'a> NAOptionHandler for RawDemuxer<'a> { | |
607 | fn get_supported_options(&self) -> &[NAOptionDefinition] { | |
608 | self.dmx.get_supported_options() | |
609 | } | |
610 | fn set_options(&mut self, options: &[NAOption]) { | |
611 | self.dmx.set_options(options); | |
612 | } | |
613 | fn query_option_value(&self, name: &str) -> Option<NAValue> { | |
614 | self.dmx.query_option_value(name) | |
615 | } | |
616 | } | |
617 | ||
618 | /// The trait for creating raw data demuxers. | |
619 | pub trait RawDemuxerCreator { | |
620 | /// Creates new raw demuxer instance that will use `ByteReader` source as an input. | |
621 | fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn RawDemuxCore<'a> + 'a>; | |
622 | /// Tries to check whether the input can be demuxed with the demuxer. | |
623 | fn check_format(&self, br: &mut ByteReader) -> bool; | |
624 | /// Returns the name of current raw data demuxer creator (equal to the container name it can demux). | |
625 | fn get_name(&self) -> &'static str; | |
626 | } | |
627 | ||
628 | /// Creates raw data demuxer for a provided bytestream. | |
629 | pub fn create_raw_demuxer<'a>(dmxcr: &dyn RawDemuxerCreator, br: &'a mut ByteReader<'a>) -> DemuxerResult<RawDemuxer<'a>> { | |
630 | let mut dmx = dmxcr.new_demuxer(br); | |
631 | let mut strmgr = StreamManager::new(); | |
632 | let mut seek_idx = SeekIndex::new(); | |
633 | dmx.open(&mut strmgr, &mut seek_idx)?; | |
634 | Ok(RawDemuxer::new(dmx, strmgr, seek_idx)) | |
635 | } | |
636 | ||
637 | /// Creates raw data demuxer for a provided bytestream with options applied right after its creation. | |
638 | pub fn create_raw_demuxer_with_options<'a>(dmxcr: &dyn RawDemuxerCreator, br: &'a mut ByteReader<'a>, opts: &[NAOption]) -> DemuxerResult<RawDemuxer<'a>> { | |
639 | let mut dmx = dmxcr.new_demuxer(br); | |
640 | dmx.set_options(opts); | |
641 | let mut strmgr = StreamManager::new(); | |
642 | let mut seek_idx = SeekIndex::new(); | |
643 | dmx.open(&mut strmgr, &mut seek_idx)?; | |
644 | Ok(RawDemuxer::new(dmx, strmgr, seek_idx)) | |
645 | } | |
646 | ||
647 | /// List of registered demuxers. | |
648 | #[derive(Default)] | |
649 | pub struct RegisteredRawDemuxers { | |
650 | dmxs: Vec<&'static dyn RawDemuxerCreator>, | |
651 | } | |
652 | ||
653 | impl RegisteredRawDemuxers { | |
654 | /// Constructs a new `RegisteredDemuxers` instance. | |
655 | pub fn new() -> Self { | |
656 | Self { dmxs: Vec::new() } | |
657 | } | |
658 | /// Registers a new demuxer. | |
659 | pub fn add_demuxer(&mut self, dmx: &'static dyn RawDemuxerCreator) { | |
660 | self.dmxs.push(dmx); | |
661 | } | |
662 | /// Searches for a demuxer that supports requested container format. | |
663 | pub fn find_demuxer(&self, name: &str) -> Option<&dyn RawDemuxerCreator> { | |
664 | self.dmxs.iter().find(|&&dmx| dmx.get_name() == name).copied() | |
665 | } | |
666 | /// Provides an iterator over currently registered demuxers. | |
667 | pub fn iter(&self) -> std::slice::Iter<&dyn RawDemuxerCreator> { | |
668 | self.dmxs.iter() | |
669 | } | |
670 | } |