Skip to content

add support to render all videos inside a folder #72

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
bladeclickers wants to merge 1 commit into
base: main
Choose a base branch
from
Open

add support to render all videos inside a folder #72

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions src/cli.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,9 @@ pub struct Arguments {
#[clap(long)]
pub vspipe_path: Option<PathBuf>,

#[clap(long, default_value_t = false)]
pub recursive: bool,

// misc io
/// Discard any audio tracks that'd pass to output
#[clap(long, visible_alias = "an", default_value_t = false)]
Expand Down
110 changes: 70 additions & 40 deletions src/video.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ use color_eyre::owo_colors::OwoColorize;
use ffprobe::FfProbe;
use rand::seq::IndexedRandom;
use rfd::FileDialog;
use std::{collections::HashMap, fs, path::PathBuf};
use std::{collections::HashMap, fs, path::{Path, PathBuf}};
use which::which;

#[derive(Debug, Clone)]
Expand Down Expand Up @@ -50,7 +50,6 @@ fn probe_video(input: &PathBuf) -> Option<FfProbe> {
}
};

// Try to open the file
let file = match fs::File::open(&path) {
Ok(file) => file,
_ => {
Expand All @@ -59,7 +58,6 @@ fn probe_video(input: &PathBuf) -> Option<FfProbe> {
}
};

// Check if the file is empty (0 bytes)
let metadata = file.metadata().expect("Error getting input file metadata");
if metadata.len() == 0 {
println!(
Expand Down Expand Up @@ -111,11 +109,6 @@ pub fn resolve_outpath(
"%FILENAME%-SM".to_string()
} else {
recipe.get("output", "file format").to_uppercase()
// .get("output")
// .expect("Failed getting [output] from recipe")
// .get("file format")
// .expect("Failed getting `[output] file format:` from recipe")
// .to_uppercase()
};

let out_dir = if args.outdir.is_some() {
Expand Down Expand Up @@ -143,7 +136,7 @@ pub fn resolve_outpath(
),
);
}
// create list of vars with section, key, and placeholder name

let variables = vec![
("interpolation", "fps", "INTERP_FPS"),
("interpolation", "speed", "SPEED"),
Expand All @@ -156,29 +149,23 @@ pub fn resolve_outpath(
("miscellaneous", "dedup threshold", "DEDUP"),
("pre-interp", "factor", "FACTOR"),
];
// loop through each var

for (section, key, var) in variables {
// check if file format string contains this var's placeholder
if format.contains(&format!("%{}%", var)) {
// get var value from recipe using section and key
let mut value = recipe.get(section, key);
// truncate weigting var if too long
if key == "weighting" && value.len() > 15 {
value = format!("{}..", &value[..15]);
}
// replace filename forbidden characters with underscores
value = value
.chars()
.map(|c| match c {
'<' | '>' | ':' | '"' | '/' | '\\' | '|' | '?' | '*' => '_',
_ => c,
})
.collect();
// skip this var if value is empty
if value.trim().is_empty() {
continue;
}
// replace placeholder with an actual value
format = format.replace(&format!("%{}%", var), &value);
}
}
Expand All @@ -189,11 +176,6 @@ pub fn resolve_outpath(
}

let rc_container = recipe.get("output", "container").trim().to_owned();
// .expect("Failed getting [output] from recipe")
// .get("container")
// .expect("Failed getting `[output] container:` from recipe")
// .trim();

let container: String = if rc_container.is_empty() {
println!("Defaulting output extension to .MP4");
String::from("MP4")
Expand All @@ -213,6 +195,40 @@ pub fn resolve_outpath(
out
}

/// Recursively collect all video files from a folder
fn collect_videos_from_folder(folder: &Path, recursive: bool) -> Vec<PathBuf> {
let mut videos = vec![];

if !folder.is_dir() {
println!("{} is not a folder, skipping.", folder.display());
return videos;
}

let entries = match fs::read_dir(folder) {
Ok(e) => e,
Err(e) => {
println!("Failed reading folder {}: {}", folder.display(), e);
return videos;
}
};

for entry in entries.filter_map(Result::ok) {
let path = entry.path();
if path.is_file() {
if crate::VIDEO_EXTENSIONS
.iter()
.any(|ext| path.extension().map_or(false, |e| e.eq_ignore_ascii_case(ext)))
{
videos.push(path);
}
} else if path.is_dir() && recursive {
videos.extend(collect_videos_from_folder(&path, recursive));
}
}

videos
}

/// Attempts to resolve and structure input structs from CLI arguments
pub fn resolve_input(args: &mut Arguments, recipe: &Recipe) -> Vec<Payload> {
let mut payloads: Vec<Payload> = vec![];
Expand Down Expand Up @@ -240,40 +256,54 @@ pub fn resolve_input(args: &mut Arguments, recipe: &Recipe) -> Vec<Payload> {
} else {
"ffplay"
};
panic!("You have the preview window enabled, but {term} is not installed/in PATH, ensure FULL FFmpeg is installed (ffmpeg, ffprobe and ffplay.");
panic!("You have the preview window enabled, but {term} is not installed/in PATH, ensure FULL FFmpeg is installed (ffmpeg, ffprobe and ffplay).");
}

// Option 1: launched a shortcut that had --tui in args
if args.tui && args.input.is_empty() && args.json.is_none() {
let input = FileDialog::new()
.add_filter("Video file", crate::VIDEO_EXTENSIONS)
.set_title("Select video(s) to queue to Smoothie")
.set_directory("/")
.pick_files();

dbg!(&input);
let input = if args.recursive {
// Folder picker for recursive mode
FileDialog::new()
.set_title("Select folder(s) to queue to Smoothie")
.pick_folders()
} else {
// Normal file picker
FileDialog::new()
.add_filter("Video file", crate::VIDEO_EXTENSIONS)
.set_title("Select video(s) to queue to Smoothie")
.pick_files()
};

args.input = match input {
Some(paths) => paths,
None => std::process::exit(0),
};
}

// Option 2: picked files in option 1 / used a shortcut Send to / the CLI
// Option 2: picked files / folders in args.input
if !args.input.is_empty() {
// input is a vector of paths
for vid in &mut args.input {
let probe = match probe_video(vid) {
let mut all_input_videos: Vec<PathBuf> = vec![];

for input in &args.input {
if input.is_file() {
all_input_videos.push(input.clone());
} else if input.is_dir() {
all_input_videos.extend(collect_videos_from_folder(
&input,
args.recursive,
));
} else {
println!("Input path {:?} does not exist, skipping.", input);
}
}

for vid in all_input_videos {
let probe = match probe_video(&vid) {
Some(probe) => probe,
None => continue, // filtered out
None => continue,
};

videos.push((
vid.canonicalize()
.expect("Failed getting full input file path"),
probe,
None,
));
videos.push((vid.canonicalize().expect("Failed canonicalize"), probe, None));
}

// Option 3: suckless-cut / Smoothie Pre-Render
Expand Down