add some first benchmarks

1 files changed, 11 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index b5bbdb3..e066a95 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,6 +1,7 @@
 use std::{env, fs, path::PathBuf};
 
 use color_eyre::eyre::Result;
+use indicatif::ProgressBar;
 
 mod gpx;
 mod layer;
@@ -21,10 +22,18 @@ fn main() -> Result<()> {
 
     for zoom in 0..=19 {
         println!("Doing level {zoom}");
-        let counter = renderer::render_heatcounter(zoom, &tracks);
+
+        let bar = ProgressBar::new(tracks.len().try_into().unwrap());
+        let counter =
+            renderer::render_heatcounter(zoom, &tracks, |x| bar.inc(x.try_into().unwrap()));
+        bar.finish();
+
         let target = ["tiles", &zoom.to_string()].iter().collect::<PathBuf>();
         fs::create_dir(&target)?;
-        renderer::lazy_colorization(&counter, &target)?;
+
+        let bar = ProgressBar::new(counter.tile_count().try_into().unwrap());
+        renderer::lazy_colorization(&counter, &target, |x| bar.inc(x.try_into().unwrap()))?;
+        bar.finish();
     }
 
     Ok(())