evc/src/bin/decode.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

#![feature(box_patterns)]
use anyhow::Context;
use clap::Parser;
use evc::{
    block::Block,
    codec::decode::decode_block,
    debug::draw_debug,
    format::{header::Header, ser::Source},
    frame::Frame,
};
use log::info;
use std::io::{BufReader, BufWriter};

#[derive(Parser)]
#[clap(about, version)]
pub struct DecodeArgs {
    #[arg(long)]
    debug: bool,
}

fn main() -> anyhow::Result<()> {
    env_logger::init_from_env("LOG");
    let args = DecodeArgs::parse();

    let mut input = BufReader::new(std::io::stdin());
    let mut output = BufWriter::new(std::io::stdout());

    let header = input.get::<Header>().context("reading header")?;
    info!("{header:?}");
    let size = header.resolution;

    let mut prev = Frame::new(size);
    for i in 0.. {
        info!("decode frame {i}");

        let block = Block::read(&mut input, size).context("reading encoded frame")?;
        let mut frame = Frame::new(size);

        decode_block(&block, frame.view_mut(), prev.view());

        if args.debug {
            let mut f2 = frame.clone();
            draw_debug(&block, f2.view_mut());
            f2.write(&mut output).context("writing raw frame")?;
        } else {
            frame.write(&mut output).context("writing raw frame")?;
        }

        prev = frame;
    }
    drop(input);
    Ok(())
}