From 617aa6a046792afc03659306eec5facb730ffc79 Mon Sep 17 00:00:00 2001 From: Tanguy Mervin Date: Mon, 5 Jan 2026 19:17:26 +0100 Subject: [PATCH] fix: added clamping when input number of colors to extract is greater than the actual colors in image --- src/lib.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/lib.rs b/src/lib.rs index 163ced8..72205d4 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -121,7 +121,8 @@ pub fn extract>( used.sort_unstable_by(|a, b| b.0.cmp(&a.0)); - let top_used = &used[..number_of_color]; + let nmin = number_of_color.min(used.len()); + let top_used = &used[..nmin]; let sum_counts: u32 = top_used.iter().map(|&(count, _)| count).sum(); let mut colors = Vec::with_capacity(number_of_color);