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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
use anyhow::bail;
use crate::{
pixel::Pixel,
ser::{Ser, Sink, Source},
vec2::Vec2,
};
#[derive(Clone, Debug)]
pub enum Block {
Literal(Vec<Pixel>),
Split(Box<[Block; 2]>),
Reference { translation: Vec2 },
}
impl Block {
pub fn write(&self, sink: &mut impl std::io::Write) -> anyhow::Result<()> {
match &self {
Block::Literal(pixels) => {
sink.put(0u8)?;
pixels.write(sink)?;
}
Block::Split(box [a, b]) => {
sink.put(1u8)?;
a.write(sink)?;
b.write(sink)?;
}
Block::Reference { translation: _ } => {
sink.put(2u8)?;
// sink.put(*translation)?;
}
}
Ok(())
}
pub fn read(source: &mut impl std::io::Read, size: Vec2) -> anyhow::Result<Self> {
Ok(match source.get::<u8>()? {
0 => Block::Literal(source.get()?),
1 => Block::Split(Box::new({
let vert = size.x > size.y;
let asize = if vert {
(size.x / 2, size.y).into()
} else {
(size.x, size.y / 2).into()
};
let bsize = if vert {
(size.x - size.x / 2, size.y).into()
} else {
(size.x, size.y - size.y / 2).into()
};
let a = Block::read(source, asize)?;
let b = Block::read(source, bsize)?;
[a, b]
})),
2 => Block::Reference {
translation: Vec2::ZERO, //source.get()?,
},
x => bail!("corrupt block type ({})", x),
})
}
}
impl Block {
pub fn is_literal(&self) -> bool {
matches!(self, Block::Literal(..))
}
}
|
