Merge pull request #22 from loiccoyle/fix_#14

fix weird color issue (#14)
Misterio77 · Jan 14, 2021 · 22d4e7f · 22d4e7f
2 parents 5c059e3 + 7652b60
commit 22d4e7f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/operations/generate.rs b/src/operations/generate.rs
@@ -161,11 +161,9 @@ pub fn generate(
     to_stdout: bool,
 ) -> Result<()> {
     let img_buffer = image::open(image_path)?;
-    let img_pixels = img_buffer
-        .as_flat_samples_u8()
-        .ok_or_else(|| anyhow!("Couldn't read provided file. Is it a valid image?"))?;
+    let img_pixels = img_buffer.to_rgba8().into_raw();
     let generated_colors =
-        color_thief::get_palette(img_pixels.as_slice(), color_thief::ColorFormat::Rgb, 1, 15)?;
+        color_thief::get_palette(img_pixels.as_slice(), color_thief::ColorFormat::Rgba, 1, 15)?;
 
     let mut colors: Vec<Rgb> = Vec::new();