ezk-image 0.4.3

Convert pixel and color formats such and RGB, YUV (YCbCr), ICtCp
Documentation 
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

use crate::formats::visit_2x2::{Image2x2Visitor, visit};
use crate::formats::yuv422::{Yuv422Block, Yuv422Src};
use crate::planes::read_planes_mut;
use crate::primitive::Primitive;
use crate::vector::Vector;
use crate::{ConvertError, ImageMut, ImageRefExt};
use std::marker::PhantomData;

pub(crate) struct Write1Plane<'a, P, S>
where
    P: Primitive,
    S: Yuv422Src,
{
    yuyv: &'a mut [u8],

    yuyv_stride: usize,

    max_value: f32,

    i422_src: S,

    _m: PhantomData<&'a mut [P]>,
}

impl<'a, P, S> Write1Plane<'a, P, S>
where
    P: Primitive,
    S: Yuv422Src,
{
    pub(crate) fn write(dst: &'a mut dyn ImageMut, i422_src: S) -> Result<(), ConvertError> {
        dst.bounds_check()?;

        let dst_width = dst.width();
        let dst_height = dst.height();
        let dst_format = dst.format();

        let [(yuyv, yuyv_stride)] = read_planes_mut(dst.planes_mut())?;

        visit(
            dst_width,
            dst_height,
            Self {
                yuyv,
                yuyv_stride,
                max_value: crate::formats::max_value_for_bits(dst_format.bits_per_component()),
                i422_src,
                _m: PhantomData,
            },
        );

        Ok(())
    }

    unsafe fn write_yuyv<V: Vector>(&mut self, y: V, uv: V, offset0: usize)
    where
        P: Primitive,
    {
        let (yuyv00, yuyv01) = y.zip(uv);

        P::write_2x(&mut self.yuyv[offset0..], yuyv00, yuyv01);
    }
}

impl<P, S> Image2x2Visitor for Write1Plane<'_, P, S>
where
    P: Primitive,
    S: Yuv422Src,
{
    #[inline(always)]
    unsafe fn visit<V: Vector>(&mut self, x: usize, y: usize) {
        let Yuv422Block {
            y00,
            y01,
            y10,
            y11,
            u0,
            u1,
            v0,
            v1,
        } = self.i422_src.read::<V>(x, y);

        let y00 = y00.vmulf(self.max_value);
        let y01 = y01.vmulf(self.max_value);
        let y10 = y10.vmulf(self.max_value);
        let y11 = y11.vmulf(self.max_value);
        let u0 = u0.vmulf(self.max_value);
        let u1 = u1.vmulf(self.max_value);
        let v0 = v0.vmulf(self.max_value);
        let v1 = v1.vmulf(self.max_value);

        let offset0 = y * self.yuyv_stride + x * 2 * P::SIZE;
        let offset1 = (y + 1) * self.yuyv_stride + x * 2 * P::SIZE;

        let (uv00, uv01) = u0.zip(v0);
        let (uv10, uv11) = u1.zip(v1);

        self.write_yuyv(y00, uv00, offset0);
        self.write_yuyv(y01, uv01, offset0 + V::LEN * 2 * P::SIZE);
        self.write_yuyv(y10, uv10, offset1);
        self.write_yuyv(y11, uv11, offset1 + V::LEN * 2 * P::SIZE);
    }
}