Merge pull request #1875 from Arvamer/avif_with_num_threads

AvifEncoder: Add with_num_threads()
image-rs · May 14, 2023 · c41c7d3 · c41c7d3
2 parents 124ebc7 + 2b3bb51
commit c41c7d3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/codecs/avif/encoder.rs b/src/codecs/avif/encoder.rs
@@ -81,6 +81,13 @@ impl<W: Write> AvifEncoder<W> {
             .with_internal_color_space(color_space.to_ravif());
         self
     }
+
+    /// Configures `rayon` thread pool size.
+    /// The default `None` is to use all threads in the default `rayon` thread pool.
+    pub fn with_num_threads(mut self, num_threads: Option<usize>) -> Self {
+        self.encoder = self.encoder.with_num_threads(num_threads);
+        self
+    }
 }
 
 impl<W: Write> ImageEncoder for AvifEncoder<W> {