Merge pull request #25 from Polochon-street/add-binary-example
Add binary example to make playlist from a folder
This commit is contained in:
commit
13899e6b58
3 changed files with 92 additions and 2 deletions
32
Cargo.lock
generated
32
Cargo.lock
generated
|
@ -28,9 +28,9 @@ dependencies = [
|
|||
|
||||
[[package]]
|
||||
name = "anyhow"
|
||||
version = "1.0.44"
|
||||
version = "1.0.45"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "61604a8f862e1d5c3229fdd78f8b02c68dcf73a4c4b05fd636d12240aaa242c1"
|
||||
checksum = "ee10e43ae4a853c0a3591d4e2ada1719e553be18199d9da9d4a83f5927c2f5c7"
|
||||
|
||||
[[package]]
|
||||
name = "atty"
|
||||
|
@ -79,12 +79,15 @@ checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
|
|||
name = "bliss-audio"
|
||||
version = "0.4.1"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"bliss-audio-aubio-rs",
|
||||
"crossbeam",
|
||||
"env_logger",
|
||||
"ffmpeg-next",
|
||||
"glob",
|
||||
"lazy_static 1.4.0",
|
||||
"log",
|
||||
"mime_guess",
|
||||
"ndarray",
|
||||
"ndarray-npy",
|
||||
"ndarray-stats",
|
||||
|
@ -616,6 +619,22 @@ dependencies = [
|
|||
"autocfg",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "mime"
|
||||
version = "0.3.16"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2a60c7ce501c71e03a9c9c0d35b861413ae925bd979cc7a4e30d060069aaac8d"
|
||||
|
||||
[[package]]
|
||||
name = "mime_guess"
|
||||
version = "2.0.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2684d4c2e97d99848d30b324b00c8fcc7e5c897b7cbb5819b09e7c90e8baf212"
|
||||
dependencies = [
|
||||
"mime",
|
||||
"unicase",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "miniz_oxide"
|
||||
version = "0.4.4"
|
||||
|
@ -1177,6 +1196,15 @@ version = "0.1.3"
|
|||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "56dee185309b50d1f11bfedef0fe6d036842e3fb77413abef29f8f8d1c5d4c1c"
|
||||
|
||||
[[package]]
|
||||
name = "unicase"
|
||||
version = "2.6.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "50f37be617794602aabbeee0be4f259dc1778fabe05e2d67ee8f79326d5cb4f6"
|
||||
dependencies = [
|
||||
"version_check",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "unicode-segmentation"
|
||||
version = "1.8.0"
|
||||
|
|
|
@ -44,3 +44,8 @@ bliss-audio-aubio-rs = "0.2.0"
|
|||
strum = "0.21"
|
||||
strum_macros = "0.21"
|
||||
serde = { version = "1.0", optional = true, features = ["derive"] }
|
||||
|
||||
[dev-dependencies]
|
||||
mime_guess = "2.0.3"
|
||||
glob = "0.3.0"
|
||||
anyhow = "1.0.45"
|
||||
|
|
57
examples/playlist.rs
Normal file
57
examples/playlist.rs
Normal file
|
@ -0,0 +1,57 @@
|
|||
use anyhow::Result;
|
||||
use bliss_audio::distance::{closest_to_first_song, dedup_playlist, euclidean_distance};
|
||||
use bliss_audio::{library::analyze_paths_streaming, Song};
|
||||
use glob::glob;
|
||||
use mime_guess;
|
||||
use std::env;
|
||||
use std::fs;
|
||||
use std::path::Path;
|
||||
|
||||
/* Analyzes a folder recursively, and make a playlist out of the file
|
||||
* provided by the user. */
|
||||
// TODO still:
|
||||
// * Save the results somewhere to avoid analyzing stuff over and over
|
||||
// * Make the output file configurable
|
||||
// * Allow to choose between outputing to stdout and a file
|
||||
fn main() -> Result<()> {
|
||||
let args: Vec<String> = env::args().skip(1).collect();
|
||||
if args.len() > 3 || args.len() < 2 {
|
||||
println!("Usage: ./playlist <folder> <file>");
|
||||
println!(
|
||||
"Creates a playlist of all audio files in a folder (recursively), \
|
||||
starting with <file>, and outputs the result both to stdout and \
|
||||
a `playlist.m3u` file in the current folder."
|
||||
);
|
||||
return Ok(());
|
||||
}
|
||||
let folder = &args[0];
|
||||
let file = fs::canonicalize(&args[1])?;
|
||||
let pattern = Path::new(folder).join("**").join("*");
|
||||
let songs = glob(&pattern.to_string_lossy())?
|
||||
.map(|e| fs::canonicalize(e.unwrap()).unwrap())
|
||||
.filter(|e| match mime_guess::from_path(e).first() {
|
||||
Some(m) => m.type_() == "audio",
|
||||
None => false,
|
||||
})
|
||||
.map(|x| x.to_string_lossy().to_string())
|
||||
.collect::<Vec<String>>();
|
||||
let rx = analyze_paths_streaming(songs)?;
|
||||
let first_song = Song::new(file)?;
|
||||
let mut analyzed_songs = vec![first_song.to_owned()];
|
||||
for (path, result) in rx.iter() {
|
||||
match result {
|
||||
Ok(song) => analyzed_songs.push(song),
|
||||
Err(e) => println!("error analyzing {}: {}", path, e),
|
||||
};
|
||||
}
|
||||
closest_to_first_song(&first_song, &mut analyzed_songs, euclidean_distance);
|
||||
dedup_playlist(&mut analyzed_songs, None);
|
||||
let playlist = analyzed_songs
|
||||
.iter()
|
||||
.map(|s| s.path.to_string_lossy().to_string())
|
||||
.collect::<Vec<String>>()
|
||||
.join("\n");
|
||||
println!("{}", playlist);
|
||||
fs::write("./playlist.m3u", playlist)?;
|
||||
Ok(())
|
||||
}
|
Loading…
Reference in a new issue