perf: multithreading recursive mode (#13)

fgardt · Mar 21, 2024 · a11567b · a11567b
1 parent 7a2dee7
commit a11567b
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -27,4 +27,5 @@ clap = { version = "4.5", features = ["derive"] }
 env_logger = "0.10"
 image = { version = "0.25", features = ["png"] }
 log = "0.4"
+rayon = "1.9.0"
 thiserror = "1.0"
diff --git a/src/image_util.rs b/src/image_util.rs
@@ -81,7 +81,6 @@ pub fn crop_images(images: &mut Vec<RgbaImage>) -> ImgUtilResult<(i32, i32)> {
     let mut max_x = std::u32::MIN;
     let mut max_y = std::u32::MIN;
 
-    // TODO: parallelize this
     for image in images.iter() {
         // ensure image has same size
         if image.width() != raw_width || image.height() != raw_height {

diff --git a/src/main.rs b/src/main.rs
@@ -7,6 +7,7 @@ use std::{
 use clap::{Args, Parser, Subcommand};
 use image::{ImageBuffer, RgbaImage};
 use image_util::ImageBufferExt;
+use rayon::prelude::*;
 
 #[macro_use]
 extern crate log;
@@ -132,11 +133,11 @@ impl SpritesheetArgs {
             return Ok(());
         }
 
-        for source in sources {
-            if let Err(err) = generate_spritesheet(self, &source) {
+        sources.par_iter().for_each(|source| {
+            if let Err(err) = generate_spritesheet(self, source) {
                 error!("{}: {err}", source.display(),);
             }
-        }
+        });
 
         Ok(())
     }