progress indicator

rust-adventure · Aug 26, 2023 · f4e9a36 · f4e9a36
1 parent 92be0fa
commit f4e9a36
Show file tree

Hide file tree

Showing 3 changed files with 149 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -6,4 +6,5 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+indicatif = "0.17.6"
 itertools = "0.11.0"
diff --git a/src/main.rs b/src/main.rs
@@ -1,3 +1,4 @@
+use indicatif::ProgressIterator;
 use itertools::Itertools;
 use std::{fs, io};
 
@@ -8,6 +9,9 @@ const MAX_VALUE: u8 = 255;
 fn main() -> io::Result<()> {
     let pixels = (0..IMAGE_HEIGHT)
         .cartesian_product(0..IMAGE_WIDTH)
+        .progress_count(
+            IMAGE_HEIGHT as u64 * IMAGE_WIDTH as u64,
+        )
         .map(|(y, x)| {
             let r = x as f64 / (IMAGE_WIDTH - 1) as f64;
             let g = y as f64 / (IMAGE_HEIGHT - 1) as f64;