anime: gif-image

rog-anime/src/anime_gif.rs

@@ -1,7 +1,8 @@
 use serde_derive::{Deserialize, Serialize};
 use std::{fs::File, path::Path, time::Duration};
+use glam::Vec2;
 
-use crate::{error::AnimeError, AniMeDataBuffer, AniMeDiagonal};
+use crate::{AniMeDataBuffer, AniMeDiagonal, AniMeImage, error::AnimeError, Pixel};
 
 #[derive(Debug, Clone, Deserialize, Serialize)]
 pub struct AniMeFrame {
@@ -25,7 +26,7 @@ impl AniMeFrame {
 pub struct AniMeGif(Vec<AniMeFrame>);
 
 impl AniMeGif {
-    pub fn new(file_name: &Path, brightness: f32) -> Result<Self, AnimeError> {
+    pub fn create_diagonal_gif(file_name: &Path, brightness: f32) -> Result<Self, AnimeError> {
         let mut frames = Vec::new();
         let mut matrix = AniMeDiagonal::new();
 
@@ -57,6 +58,47 @@ impl AniMeGif {
         Ok(Self(frames))
     }
 
+    pub fn create_png_gif(file_name: &Path, brightness: f32) -> Result<Self, AnimeError> {
+        let mut frames = Vec::new();
+
+        let mut decoder = gif::DecodeOptions::new();
+        // Configure the decoder such that it will expand the image to RGBA.
+        decoder.set_color_output(gif::ColorOutput::RGBA);
+        // Read the file header
+        let file = File::open(file_name)?;
+        let mut decoder = decoder.read_info(file)?;
+
+        let width = decoder.width();
+        let pixels: Vec<Pixel> = vec![Pixel::default(); (decoder.width() as u32 * decoder.height() as u32) as usize];
+        let mut image = AniMeImage::new(Vec2::new(1.0, 1.0), 0.0, Vec2::new(0.0, 0.0),
+        brightness, pixels, decoder.width() as u32);
+
+        while let Some(frame) = decoder.read_next_frame()? {
+            let wait = frame.delay * 10;
+            for (y, row) in frame.buffer.chunks(frame.width as usize * 4).enumerate() {
+                for (x, px) in row.chunks(4).enumerate() {
+                    if px[3] != 255 {
+                        // should be t but not in some gifs? What, ASUS, what?
+                        continue;
+                    }
+                    let pos = (x + frame.left as usize) + ((y + frame.top as usize) * width as usize);
+                    image.get_mut()[pos] =
+                        Pixel {
+                            color: ((px[0] as u32 + px[1] as u32 + px[2] as u32) / 3),
+                            alpha: 1.0,
+                        };
+                }
+            }
+            image.update();
+
+            frames.push(AniMeFrame {
+                data: <AniMeDataBuffer>::from(&image),
+                delay: Duration::from_millis(wait as u64),
+            });
+        }
+        Ok(Self(frames))
+    }
+
     pub fn frames(&self) -> &[AniMeFrame] {
         &self.0
     }