nits

src/model.rs

@@ -23,13 +23,13 @@ pub struct Model {
     diffusivity: usize,
 
     // Current model iteration.
-    iteration: i32,
+    iteration: usize,
 
     // Color palette
     palette: Palette,
 
     // List of ImgData to be processed post-simulation into images
-    img_data_vec: Vec<ImgData>,
+    img_data_vec: Vec<(usize, ImgData)>,
 }
 
 impl Model {
@@ -143,15 +143,11 @@ impl Model {
         );
     }
 
-    fn size_of_imgdata_vec(&self) -> usize {
-        self.img_data_vec[0].size_of() * self.img_data_vec.len()
-    }
-
     fn save_image_data(&mut self) {
         let grids = ThinGridData::new_from_grid_vec(self.grids.clone());
-        let img_data = ImgData::new(grids, self.palette, self.iteration);
-        self.img_data_vec.push(img_data);
-        let size: usize = self.size_of_imgdata_vec();
+        let img_data = ImgData::new(grids, self.palette);
+        self.img_data_vec.push((self.iteration + 1, img_data));
+        let size: usize = std::mem::size_of_val(&self.img_data_vec);
         let mb = size / 1024 / 1024;
         // println!("{} B | {} KB | {} MB", size, size/1024, size/1024/1024);
 
@@ -171,7 +167,7 @@ impl Model {
         self.img_data_vec.clear();
     }
 
-    pub fn render_all_imgdata(&self) {
+    pub fn render_all_imgdata(&mut self) {
         if !Path::new("./tmp").exists() {
             std::fs::create_dir("./tmp").expect("could create directory");
         }
@@ -181,56 +177,15 @@ impl Model {
             "{spinner:.green} [{elapsed_precise}] [{bar:40.cyan/blue}] ({pos}/{len}, {percent}%, {per_sec})",
         ));
 
-        /*
-        for img in &self.img_data_vec {
-            Self::save_to_image(img.to_owned());
-            pb.inc(1);
-        }
-        pb.finish();
-        */
-
         self.img_data_vec
+            .drain(..)
+            .collect::<Vec<_>>()
             .par_iter()
             .progress_with(pb)
-            .for_each(|img| {
-                // Self::save_to_image(img.to_owned());
-                img.save_to_image();
+            .for_each(|(i, img)| {
+                img.to_image()
+                    .save(format!("./tmp/out_{}.png", i).as_str())
+                    .unwrap();
             });
     }
-
-    /*
-    pub fn save_to_image(imgdata: ImgData) {
-        let (width, height) = (imgdata.grids[0].width, imgdata.grids[0].height);
-        let mut img = image::RgbImage::new(width as u32, height as u32);
-
-        let max_values: Vec<_> = imgdata
-            .grids
-            .iter()
-            .map(|grid| grid.quantile(0.999) * 1.5)
-            .collect();
-
-        for y in 0..height {
-            for x in 0..width {
-                let i = y * width + x;
-                let (mut r, mut g, mut b) = (0.0_f32, 0.0_f32, 0.0_f32);
-                for (grid, max_value, color) in
-                    multizip((&imgdata.grids, &max_values, &imgdata.palette.colors))
-                {
-                    let mut t = (grid.data()[i] / max_value).clamp(0.0, 1.0);
-                    t = t.powf(1.0 / 2.2); // gamma correction
-                    r += color.0[0] as f32 * t;
-                    g += color.0[1] as f32 * t;
-                    b += color.0[2] as f32 * t;
-                }
-                r = r.clamp(0.0, 255.0);
-                g = g.clamp(0.0, 255.0);
-                b = b.clamp(0.0, 255.0);
-                img.put_pixel(x as u32, y as u32, image::Rgb([r as u8, g as u8, b as u8]));
-            }
-        }
-
-        img.save(format!("./tmp/out_{}.png", imgdata.iteration).as_str())
-            .unwrap();
-    }
-    */
 }