aboutsummaryrefslogtreecommitdiff
path: root/src/renderer.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/renderer.rs')
-rw-r--r--src/renderer.rs7
1 files changed, 3 insertions, 4 deletions
diff --git a/src/renderer.rs b/src/renderer.rs
index de65da6..bb33ddc 100644
--- a/src/renderer.rs
+++ b/src/renderer.rs
@@ -1,4 +1,4 @@
-use std::{fs, mem, path::Path, sync::mpsc, thread};
+use std::{fs, path::Path, thread};
use color_eyre::{
eyre::{bail, Result},
@@ -147,7 +147,7 @@ pub fn lazy_colorization<P: AsRef<Path>, F: Fn(usize) + Send + Sync>(
}
type Job = (u64, u64, Vec<u8>);
- let (tx, rx) = mpsc::sync_channel::<Job>(30);
+ let (tx, rx) = crossbeam_channel::bounded::<Job>(30);
thread::scope(|s| {
let saver = s.spawn(move || loop {
@@ -165,7 +165,7 @@ pub fn lazy_colorization<P: AsRef<Path>, F: Fn(usize) + Send + Sync>(
layer
.into_parallel_tiles()
- .try_for_each(|(tile_x, tile_y, tile)| {
+ .try_for_each_with(tx, |tx, (tile_x, tile_y, tile)| {
let colorized = colorize_tile(&tile, max.into());
let data = layer::compress_png_as_bytes(&colorized)?;
tx.send((tile_x, tile_y, data))?;
@@ -173,7 +173,6 @@ pub fn lazy_colorization<P: AsRef<Path>, F: Fn(usize) + Send + Sync>(
Ok::<(), Report>(())
})?;
- mem::drop(tx);
saver.join().unwrap()?;
Ok::<_, Report>(())
})?;