works very well

1 files changed, 84 insertions, 17 deletions

diff --git a/lvc/src/main.rs b/lvc/src/main.rs
index 413e4a4..48df911 100644
--- a/lvc/src/main.rs
+++ b/lvc/src/main.rs
@@ -1,30 +1,97 @@
-use lvc::{debug::draw_debug, decode::decode, encode::encode, Frame, Pixel, PixelValue, View, P2};
+use bincode::config::standard;
+use clap::{Parser, Subcommand};
+use lvc::{
+    debug::draw_debug,
+    decode::decode,
+    encode::{encode, EncodeConfig},
+    Block, Frame, Pixel, PixelValue, View, P2,
+};
 use std::io::{stdin, stdout, BufReader, BufWriter, Read, Write};
 
+#[derive(Parser)]
+#[clap(about, version)]
+struct Args {
+    // Width of the video signal
+    #[arg(short, long)]
+    width: u16,
+    // Height of the video signal
+    #[arg(short, long)]
+    height: u16,
+    #[clap(subcommand)]
+    action: Action,
+}
+#[derive(Clone, Subcommand)]
+enum Action {
+    // Compress video
+    Encode {
+        #[arg(short, long, default_value_t = 1024)]
+        block_size: usize,
+        #[arg(short, long, default_value_t = 10_000)]
+        threshold: u32,
+    },
+    // Decompress video
+    Decode {
+        #[arg(short, long)]
+        debug: bool,
+    },
+}
+
 fn main() {
-    let size = P2 { x: 1920, y: 1080 };
+    let args = Args::parse();
 
-    let mut last_frame = Frame::new(size);
-    let mut debug_frame = Some(Frame::new(size));
+    let size = P2 {
+        x: args.width as i32,
+        y: args.height as i32,
+    };
+    match args.action {
+        Action::Decode { debug } => {
+            let mut frame = Frame::new(size);
+            let mut last_frame = Frame::new(size);
+            let mut debug_frame = if debug { Some(Frame::new(size)) } else { None };
 
-    let mut stdin = BufReader::new(stdin());
-    let mut stdout = BufWriter::new(stdout());
+            let mut stdin = BufReader::new(stdin());
+            let mut stdout = BufWriter::new(stdout());
 
-    loop {
-        let mut frame = read_frame(&mut stdin, size);
+            loop {
+                let b: Block = bincode::decode_from_std_read(&mut stdin, standard()).unwrap();
+                decode(&last_frame, &mut frame, View::all(size), &b);
 
-        let b = encode(&last_frame, &frame, View::all(size));
-        decode(&last_frame, &mut frame, View::all(size), &b);
+                if let Some(debug_frame) = &mut debug_frame {
+                    debug_frame.pixels.copy_from_slice(&frame.pixels);
+                    draw_debug(debug_frame, View::all(size), &b);
+                    write_frame(&mut stdout, &debug_frame);
+                } else {
+                    write_frame(&mut stdout, &frame);
+                }
 
-        if let Some(debug_frame) = &mut debug_frame {
-            debug_frame.pixels.copy_from_slice(&frame.pixels);
-            draw_debug(debug_frame, View::all(size), &b);
-            write_frame(&mut stdout, &debug_frame);
-        } else {
-            write_frame(&mut stdout, &frame);
+                last_frame.pixels.copy_from_slice(&frame.pixels);
+            }
         }
+        Action::Encode {
+            block_size,
+            threshold,
+        } => {
+            let config = EncodeConfig {
+                threshold,
+                max_block_size: block_size,
+            };
+
+            let mut last_frame = Frame::new(size);
 
-        last_frame = frame;
+            let mut stdin = BufReader::new(stdin());
+            let mut stdout = BufWriter::new(stdout());
+
+            for frame_number in 0.. {
+                eprintln!("encode frame {frame_number}");
+                let mut frame = read_frame(&mut stdin, size);
+
+                let b: Block = encode(&last_frame, &frame, View::all(size), &config);
+                bincode::encode_into_std_write(&b, &mut stdout, standard()).unwrap();
+
+                decode(&last_frame, &mut frame, View::all(size), &b);
+                last_frame = frame;
+            }
+        }
     }
 }