1 files changed, 15 insertions, 6 deletions
diff --git a/evc/src/bin/encode.rs b/evc/src/bin/encode.rs
index 135a0e8..e1f6730 100644
--- a/evc/src/bin/encode.rs
+++ b/evc/src/bin/encode.rs
@@ -1,9 +1,9 @@
 use anyhow::Context;
 use clap::Parser;
-use evc::{
+use indicatif::ProgressBar;
+use libreschmux::{
     codec::{
-        compress::compress_block,
-        decode::decode_block,
+        decode::{decode_block, DecodeConfig},
         encode::{encode_block, EncodeConfig, EncodeMode},
     },
     format::{header::Header, ser::Sink},
@@ -31,6 +31,9 @@ pub struct EncodeArgs {
     #[arg(long)]
     no_translate: bool,
 
+    #[arg[short = 'L', long]]
+    length: Option<usize>,
+
     #[arg(short, long, default_value = "8")]
     jobs: usize,
 
@@ -67,10 +70,12 @@ fn main() -> anyhow::Result<()> {
     output
         .put(Header {
             resolution: size,
-            frame_count: 0,
+            frame_count: args.length.unwrap_or(usize::MAX),
         })
         .context("writing header")?;
 
+    let progress_bar = args.length.map(|len| ProgressBar::new(len as u64));
+
     let mut prev_frame = Frame::new(size);
     for i in 0.. {
         info!("encode frame {i}");
@@ -79,7 +84,7 @@ fn main() -> anyhow::Result<()> {
         let v1 = frame.view();
         let v2 = prev_frame.view();
 
-        let (error, mut root) = encode_block(v1, v2, &config);
+        let (error, root) = encode_block(v1, v2, &config);
 
         // compress_block(
         //     &mut root,
@@ -97,11 +102,15 @@ fn main() -> anyhow::Result<()> {
             error / frame.view().area() as f64
         );
 
+        if let Some(progress_bar) = &progress_bar {
+            progress_bar.inc(1);
+        }
+
         decode_block(
             &root,
             frame.view_mut(),
             prev_frame.view(),
-            &evc::codec::decode::DecodeConfig {
+            &DecodeConfig {
                 max_threads: config.max_threads,
             },
         );