about summary refs log tree commit diff
path: root/squash/src/main.rs
diff options
context:
space:
mode:
authorgennyble <gen@nyble.dev>2024-01-16 15:06:29 -0600
committergennyble <gen@nyble.dev>2024-01-16 15:06:29 -0600
commitc887f5fef803e720052898b9561028dfd50e51db (patch)
tree0ea9c8c9e6c4603d0626bc12e653b6421f16f78a /squash/src/main.rs
parentbbbac45d835dd40c3fb53f8c7f9a2731783841e8 (diff)
downloadcolorsquash-c887f5fef803e720052898b9561028dfd50e51db.tar.gz
colorsquash-c887f5fef803e720052898b9561028dfd50e51db.zip
ability to choose kmeans implementation
Diffstat (limited to 'squash/src/main.rs')
-rw-r--r--squash/src/main.rs4
1 files changed, 3 insertions, 1 deletions
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);