diff options
author | metamuffin <metamuffin@disroot.org> | 2023-03-09 19:04:33 +0100 |
---|---|---|
committer | metamuffin <metamuffin@disroot.org> | 2023-03-09 19:04:33 +0100 |
commit | 11b78570656bd0ca67594472765fc629aa25fd25 (patch) | |
tree | 184356867a6cc372c23cac1ad179919a92224d8b /lvc/codec/src/frameio.rs | |
parent | ac27a58c87bea9480f6603a731448f69c9c3b0fb (diff) | |
download | video-codec-experiments-11b78570656bd0ca67594472765fc629aa25fd25.tar video-codec-experiments-11b78570656bd0ca67594472765fc629aa25fd25.tar.bz2 video-codec-experiments-11b78570656bd0ca67594472765fc629aa25fd25.tar.zst |
move all logic to codec crate
Diffstat (limited to 'lvc/codec/src/frameio.rs')
-rw-r--r-- | lvc/codec/src/frameio.rs | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/lvc/codec/src/frameio.rs b/lvc/codec/src/frameio.rs new file mode 100644 index 0000000..f6cbcbf --- /dev/null +++ b/lvc/codec/src/frameio.rs @@ -0,0 +1,35 @@ +use crate::{Frame, Pixel, PixelValue, P2}; +use std::io::{Read, Result, Write}; + +impl Frame { + pub fn read(inp: &mut impl Read, size: P2) -> Result<Frame> { + let mut f = Frame::new(size); + + for y in 0..size.y { + for x in 0..size.x { + let mut cc = [0u8; 3]; + inp.read_exact(&mut cc)?; + f[P2 { x, y }] = Pixel { + r: cc[0] as PixelValue, + g: cc[1] as PixelValue, + b: cc[2] as PixelValue, + }; + } + } + Ok(f) + } + + pub fn write(out: &mut impl Write, frame: &Frame) -> Result<()> { + for y in 0..frame.size.y { + for x in 0..frame.size.x { + let p = frame[P2 { x, y }]; + let mut cc = [0u8; 3]; + cc[0] = p.r.clamp(0, 255) as u8; + cc[1] = p.g.clamp(0, 255) as u8; + cc[2] = p.b.clamp(0, 255) as u8; + out.write_all(&mut cc)?; + } + } + Ok(()) + } +} |