diff options
Diffstat (limited to 'evc/src/bin/decode.rs')
-rw-r--r-- | evc/src/bin/decode.rs | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/evc/src/bin/decode.rs b/evc/src/bin/decode.rs new file mode 100644 index 0000000..293181a --- /dev/null +++ b/evc/src/bin/decode.rs @@ -0,0 +1,60 @@ +#![feature(box_patterns)] +use std::io::{self, BufReader, BufWriter}; + +use clap::Parser; +use evc::{ + block::{Block, BlockInner}, + frame::Frame, + header::Header, + ser::Source, + view::View, +}; + +#[derive(Parser)] +#[clap(about, version)] +pub struct DecodeArgs {} + +fn main() -> io::Result<()> { + 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>()?; + eprintln!("{header:?}"); + let size = header.resolution; + + let mut prev = Frame::new(size); + loop { + let block = Block::read(&mut input, size)?; + let mut frame = Frame::new(size); + + blit_block( + &block, + frame.view_mut((0, 0), size), + prev.view((0, 0), size), + ); + + frame.write(&mut output)?; + prev = frame; + } +} + +fn blit_block(block: &Block, mut target: View<&mut Frame>, prev: View<&Frame>) { + match &block.inner { + BlockInner::Literal(pixels) => { + for x in 0..block.size.0 { + for y in 0..block.size.1 { + target[(x, y)] = pixels[x + y * block.size.0] + } + } + } + BlockInner::Split(box [a, b]) => { + let [at, bt] = target.split_mut_unsafe(); + let [ap, bp] = prev.split(); + blit_block(a, at, ap); + blit_block(b, bt, bp); + } + BlockInner::Reference { translation: _ } => target.copy_from(&prev), + } +} |