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
69
70
71
|
use crate::diff::diff;
use crate::impls::ToArray;
use crate::split::split;
use crate::{Block, Frame, Ref, View};
pub fn encode(last_frame: &Frame, frame: &Frame, view: View) -> Block {
if view.size().area() > 1024 {
let [av, bv] = split(view);
return Block::Split(
rayon::join(
|| Box::new(encode(last_frame, frame, av)),
|| Box::new(encode(last_frame, frame, bv)),
)
.to_array(),
);
}
let mut r = Ref::default();
let mut d = diff([last_frame, frame], view, r);
for granularity in [4, 2, 1] {
for _ in 0..10 {
let (nd, nrp) = optimize_ref(last_frame, frame, view, d, r, granularity);
if nd < d {
r = nrp;
d = nd;
} else {
break;
}
}
}
if d < 10000 {
return Block::Ref(r);
} else {
Block::Lit(frame.export(view))
}
}
pub fn optimize_ref(
last_frame: &Frame,
frame: &Frame,
view: View,
mut d: u32,
mut r: Ref,
granularity: i32,
) -> (u32, Ref) {
let mut n = |f: fn(&mut Ref, i32)| {
let mut r2 = r;
f(&mut r2, granularity);
let d2 = diff([last_frame, frame], view, r2);
if d2 < d {
d = d2;
r = r2;
}
};
n(|r, g| r.pos_off.x += g);
n(|r, g| r.pos_off.x -= g);
n(|r, g| r.pos_off.y += g);
n(|r, g| r.pos_off.y -= g);
n(|r, g| r.color_off.r += (g as i16) << 2);
n(|r, g| r.color_off.r -= (g as i16) << 2);
n(|r, g| r.color_off.g += (g as i16) << 2);
n(|r, g| r.color_off.g -= (g as i16) << 2);
n(|r, g| r.color_off.b += (g as i16) << 2);
n(|r, g| r.color_off.b -= (g as i16) << 2);
(d, r)
}
|
