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
use yuv::YUV;
pub fn yuv_444<'a, T: Copy + 'a, YRowsIter: 'a, URowsIter: 'a, VRowsIter: 'a>(y: YRowsIter, u: URowsIter, v: VRowsIter) -> impl Iterator<Item = YUV<T>> + 'a
where YRowsIter: Iterator<Item=&'a [T]>,
URowsIter: Iterator<Item=&'a [T]>,
VRowsIter: Iterator<Item=&'a [T]>
{
y.zip(u.zip(v))
.flat_map(|(y,(u,v))| {
y.iter().copied().zip(u.iter().copied().zip(v.iter().copied()))
.map(|(y,(u,v))| YUV{y,u,v})
})
}
pub fn yuv_422<'a, T: Copy + 'a, YRowsIter: 'a, URowsIter: 'a, VRowsIter: 'a>(y: YRowsIter, u: URowsIter, v: VRowsIter) -> impl Iterator<Item = YUV<T>> + 'a
where YRowsIter: Iterator<Item=&'a [T]>,
URowsIter: Iterator<Item=&'a [T]>,
VRowsIter: Iterator<Item=&'a [T]>
{
y.zip(u.zip(v))
.flat_map(|(y,(u,v))| {
let u = u.iter().copied().flat_map(|u_px| std::iter::repeat(u_px).take(2));
let v = v.iter().copied().flat_map(|v_px| std::iter::repeat(v_px).take(2));
y.iter().copied().zip(u.zip(v))
.map(|(y,(u,v))| YUV{y,u,v})
})
}
pub fn yuv_420<'a, T: Copy + 'a, YRowsIter: 'a, URowsIter: 'a, VRowsIter: 'a>(y: YRowsIter, u: URowsIter, v: VRowsIter) -> impl Iterator<Item = YUV<T>> + 'a
where YRowsIter: Iterator<Item=&'a [T]>,
URowsIter: Iterator<Item=&'a [T]>,
VRowsIter: Iterator<Item=&'a [T]>
{
let u = u.flat_map(|u_row| std::iter::repeat(u_row).take(2));
let v = v.flat_map(|v_row| std::iter::repeat(v_row).take(2));
y.zip(u.zip(v))
.flat_map(|(y,(u,v))| {
let u = u.iter().copied().flat_map(|u_px| std::iter::repeat(u_px).take(2));
let v = v.iter().copied().flat_map(|v_px| std::iter::repeat(v_px).take(2));
y.iter().copied().zip(u.zip(v))
.map(|(y,(u,v))| YUV{y,u,v})
})
}