feat(cli): encode in parallel (#44)

elringus · Sep 10, 2024 · 87b1b60 · 87b1b60
1 parent 4acc8fd
commit 87b1b60
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/crates/Cargo.lock b/crates/Cargo.lock
diff --git a/crates/cli/Cargo.toml b/crates/cli/Cargo.toml
@@ -16,6 +16,7 @@ image = { version = "0.25", default-features = false, features = [
     "jpeg",
     "bmp"
 ] }
+rayon = "1.10.0"
 clap = { version = "4.5", features = ["derive"] }
 clap_derive = "4.5"
 indicatif = "0.17"
diff --git a/crates/cli/src/lib.rs b/crates/cli/src/lib.rs
@@ -3,6 +3,7 @@ mod json;
 pub mod models;
 
 use crate::models::*;
+use rayon::prelude::*;
 use sprite_dicing::{DicedSprite, Prefs, Progress, SourceSprite, Texture};
 use std::{fs, path::Path, path::PathBuf};
 
@@ -69,9 +70,8 @@ fn eval_sprite_id(root: &Path, path: &Path, separator: &str) -> String {
 }
 
 fn write_atlases(tex: Vec<Texture>, dir: &Path, fmt: &AtlasFormat, prefs: &Prefs) -> Result<()> {
-    let total = tex.len();
-    tex.into_iter().enumerate().try_for_each(|(idx, tex)| {
-        Progress::report(prefs, 4, idx, total, "Encoding atlas textures");
+    Progress::report(prefs, 4, 0, tex.len(), "Encoding atlas textures");
+    tex.into_par_iter().enumerate().try_for_each(|(idx, tex)| {
         let name = format!("atlas_{idx}.{}", fmt.extension());
         write_atlas(&dir.join(name), tex)
     })