From c887f5fef803e720052898b9561028dfd50e51db Mon Sep 17 00:00:00 2001 From: gennyble Date: Tue, 16 Jan 2024 15:06:29 -0600 Subject: ability to choose kmeans implementation --- squash/Cargo.toml | 2 +- squash/src/main.rs | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) (limited to 'squash') diff --git a/squash/Cargo.toml b/squash/Cargo.toml index 7d33b5b..971afce 100644 --- a/squash/Cargo.toml +++ b/squash/Cargo.toml @@ -11,7 +11,7 @@ repository = "https://github.com/gennyble/colorsquash/tree/main/squash" [dependencies] # the meat 'o the thing! the meaning behind it all -colorsquash = { path = "..", version = "0.2.0", features = ["gifed", "kmeans"] } +colorsquash = { path = "..", version = "0.2.0", features = ["gifed"] } # just useful tools for writing binaries anyhow = "1.0.75" diff --git a/squash/src/main.rs b/squash/src/main.rs index a7cc66b..24d557f 100644 --- a/squash/src/main.rs +++ b/squash/src/main.rs @@ -31,10 +31,12 @@ fn main() -> Result<(), anyhow::Error> { builder = builder.selector(sorsel); } - cli::Selector::Kmeans => builder = builder.selector(Kmeans), + cli::Selector::Kmeans => builder = builder.selector(Kmeans { max_iter: 10 }), }; + let start = std::time::Instant::now(); let mut squasher = builder.build(&image.data); + println!("{:.2}ms", start.elapsed().as_secs_f32()); let size = squasher.map_over(&mut image.data); image.data.resize(size, 0); -- cgit 1.4.1-3-g733a5