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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
use super::EncodeConfig;
use crate::{
block::{AdvancedReference, Block},
frame::Frame,
refsampler::Sampler,
view::View,
};
pub fn default(
view: &View<&Frame>,
prev: &View<&Frame>,
config: &EncodeConfig,
max_diff: f64,
) -> (f64, Block) {
let mut pm = AdvancedReference::default();
let sampler = Sampler::from_refblock(prev.clone(), &pm);
let mut diff = View::diff_sampler(&view, &sampler);
if diff < max_diff {
(diff, Block::REFZERO)
} else {
loop {
let (mut d, mut p) = (diff, pm.clone());
if config.do_translate {
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 4);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 4);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 4);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 4);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 2);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 2);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 2);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 2);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 1);
}
if config.do_value_scale {
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.value_scale += 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.value_scale -= 1);
}
if config.do_linear_transform {
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.a -= 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.a += 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.b -= 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.b += 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.c -= 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.c += 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.d -= 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.transform.d += 1);
}
if d >= diff {
break (diff, Block::AdvancedReference(pm));
}
diff = d;
pm = p;
}
}
}
pub fn partial(
view: &View<&Frame>,
prev: &View<&Frame>,
_config: &EncodeConfig,
max_diff: f64,
) -> (f64, Block) {
let mut pm = AdvancedReference::default();
let sampler = Sampler::from_refblock(prev.clone(), &pm);
let mut diff = View::diff_sampler(&view, &sampler);
if diff < max_diff {
(diff, Block::REFZERO)
} else {
loop {
let (mut d, mut p) = (diff, pm.clone());
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 2);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 2);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 2);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 2);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x += 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.x -= 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y += 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.translation.y -= 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.value_scale += 1);
pk(&view, &prev, &mut d, &mut p, &pm, |p| p.value_scale -= 1);
if d >= diff {
break (diff, Block::AdvancedReference(pm));
}
diff = d;
pm = p;
}
}
}
#[inline]
fn pk<F: FnMut(&mut AdvancedReference) -> ()>(
view: &View<&Frame>,
prev: &View<&Frame>,
diff: &mut f64,
params: &mut AdvancedReference,
initial_params: &AdvancedReference,
mut f: F,
) {
let mut p = initial_params.clone();
f(&mut p);
let sampler = Sampler::from_refblock(prev.clone(), &p);
let d = View::diff_sampler(view, &sampler);
if d < *diff {
*diff = d;
*params = p;
}
}
|
