51 lines
1.3 KiB
Rust
51 lines
1.3 KiB
Rust
use crate::StreamError;
|
|
use crate::{FrameData, VideoStream, VideoStreamIterator};
|
|
use std::path::{Path, PathBuf};
|
|
use std::fs;
|
|
use std::io;
|
|
use std::io::ErrorKind;
|
|
|
|
pub struct FileSource {
|
|
source: PathBuf,
|
|
frame_size: (u32, u32),
|
|
}
|
|
|
|
impl FileSource {
|
|
pub fn new(source: &Path, frame_size: (u32, u32)) -> Result<Self, CaptureError> {
|
|
let source = fs::canonicalize(source)?;
|
|
if !source.exists() {
|
|
return Err(CaptureError::IoError(io::Error::new(ErrorKind::NotFound, "File not found")));
|
|
}
|
|
Ok(Self {
|
|
source,
|
|
frame_size,
|
|
})
|
|
}
|
|
}
|
|
|
|
impl IntoIterator for FileSource {
|
|
type Item = Result<Option<FrameData>, StreamError>;
|
|
type IntoIter = VideoStreamIterator;
|
|
|
|
fn into_iter(self) -> Self::IntoIter {
|
|
let pipeline_description = format!(
|
|
"uridecodebin uri=file://{} ! videoconvert ! videoscale ! capsfilter caps=\"video/x-raw, width={}, height={}\"",
|
|
self.source.to_string_lossy(),
|
|
self.frame_size.0,
|
|
self.frame_size.1
|
|
);
|
|
VideoStream::new(pipeline_description).into_iter()
|
|
}
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub enum CaptureError {
|
|
IoError(io::Error),
|
|
}
|
|
|
|
impl From<io::Error> for CaptureError {
|
|
fn from(err: io::Error) -> Self {
|
|
CaptureError::IoError(err)
|
|
}
|
|
}
|