yscv-video 0.1.8

Video decoding (H.264, HEVC), MP4 parsing, and camera I/O
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use crate::{Frame, FrameSource, VideoError};

/// Pull-based frame stream wrapper with deterministic optional limit.
#[derive(Debug)]
pub struct FrameStream<S: FrameSource> {
    source: S,
    max_frames: Option<usize>,
    emitted: usize,
}

impl<S: FrameSource> FrameStream<S> {
    pub fn new(source: S) -> Self {
        Self {
            source,
            max_frames: None,
            emitted: 0,
        }
    }

    pub fn with_max_frames(mut self, max_frames: usize) -> Self {
        self.max_frames = Some(max_frames);
        self
    }

    pub fn try_next(&mut self) -> Result<Option<Frame>, VideoError> {
        if let Some(limit) = self.max_frames
            && self.emitted >= limit
        {
            return Ok(None);
        }
        let frame = self.source.next_frame()?;
        if frame.is_some() {
            self.emitted += 1;
        }
        Ok(frame)
    }
}