singe-npp 0.1.0-alpha.8

Safe Rust wrappers for NVIDIA Performance Primitives library (NPP).
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164

use super::*;

pub(crate) fn copy_u8_c3_to_p3(
    stream_context: &StreamContext,
    source: &ImageView<'_, u8, C3>,
    destination: &mut PlanarImageViewMut<'_, u8, 3>,
) -> Result<()> {
    validate_same_size(source.size(), destination.planes()[0].size())?;

    let destination_planes = destination
        .planes_mut()
        .each_mut()
        .map(ImageViewMut::as_mut_ptr);

    unsafe {
        try_ffi!(sys::nppiCopy_8u_C3P3R_Ctx(
            source.as_ptr().cast(),
            source.step(),
            destination_planes.as_ptr().cast(),
            destination.planes()[0].step(),
            source.size().into(),
            stream_context.as_raw(),
        ))?;
    }
    Ok(())
}

pub(crate) fn copy_u8_c4_to_p4(
    stream_context: &StreamContext,
    source: &ImageView<'_, u8, C4>,
    destination: &mut PlanarImageViewMut<'_, u8, 4>,
) -> Result<()> {
    validate_same_size(source.size(), destination.planes()[0].size())?;

    let destination_planes = destination
        .planes_mut()
        .each_mut()
        .map(ImageViewMut::as_mut_ptr);

    unsafe {
        try_ffi!(sys::nppiCopy_8u_C4P4R_Ctx(
            source.as_ptr().cast(),
            source.step(),
            destination_planes.as_ptr().cast(),
            destination.planes()[0].step(),
            source.size().into(),
            stream_context.as_raw(),
        ))?;
    }
    Ok(())
}

pub(crate) fn copy_u8_p3_to_c3(
    stream_context: &StreamContext,
    source: &PlanarImageView<'_, u8, 3>,
    destination: &mut ImageViewMut<'_, u8, C3>,
) -> Result<()> {
    validate_same_size(source.planes()[0].size(), destination.size())?;

    let source_planes = source.planes().map(|plane| plane.as_ptr());

    unsafe {
        try_ffi!(sys::nppiCopy_8u_P3C3R_Ctx(
            source_planes.as_ptr().cast(),
            source.planes()[0].step(),
            destination.as_mut_ptr().cast(),
            destination.step(),
            destination.size().into(),
            stream_context.as_raw(),
        ))?;
    }
    Ok(())
}

pub(crate) fn copy_u8_p4_to_c4(
    stream_context: &StreamContext,
    source: &PlanarImageView<'_, u8, 4>,
    destination: &mut ImageViewMut<'_, u8, C4>,
) -> Result<()> {
    validate_same_size(source.planes()[0].size(), destination.size())?;

    let source_planes = source.planes().map(|plane| plane.as_ptr());

    unsafe {
        try_ffi!(sys::nppiCopy_8u_P4C4R_Ctx(
            source_planes.as_ptr().cast(),
            source.planes()[0].step(),
            destination.as_mut_ptr().cast(),
            destination.step(),
            destination.size().into(),
            stream_context.as_raw(),
        ))?;
    }
    Ok(())
}

impl_copy_packed_to_planar!(copy_u16_c3_to_p3, u16, C3, 3, nppiCopy_16u_C3P3R_Ctx);
impl_copy_packed_to_planar!(copy_u16_c4_to_p4, u16, C4, 4, nppiCopy_16u_C4P4R_Ctx);
impl_copy_packed_to_planar!(copy_i16_c3_to_p3, i16, C3, 3, nppiCopy_16s_C3P3R_Ctx);
impl_copy_packed_to_planar!(copy_i16_c4_to_p4, i16, C4, 4, nppiCopy_16s_C4P4R_Ctx);
impl_copy_packed_to_planar!(copy_i32_c3_to_p3, i32, C3, 3, nppiCopy_32s_C3P3R_Ctx);
impl_copy_packed_to_planar!(copy_i32_c4_to_p4, i32, C4, 4, nppiCopy_32s_C4P4R_Ctx);
impl_copy_packed_to_planar!(copy_f32_c3_to_p3, f32, C3, 3, nppiCopy_32f_C3P3R_Ctx);
impl_copy_packed_to_planar!(copy_f32_c4_to_p4, f32, C4, 4, nppiCopy_32f_C4P4R_Ctx);
impl_copy_planar_to_packed!(copy_u16_p3_to_c3, u16, C3, 3, nppiCopy_16u_P3C3R_Ctx);
impl_copy_planar_to_packed!(copy_u16_p4_to_c4, u16, C4, 4, nppiCopy_16u_P4C4R_Ctx);
impl_copy_planar_to_packed!(copy_i16_p3_to_c3, i16, C3, 3, nppiCopy_16s_P3C3R_Ctx);
impl_copy_planar_to_packed!(copy_i16_p4_to_c4, i16, C4, 4, nppiCopy_16s_P4C4R_Ctx);
impl_copy_planar_to_packed!(copy_i32_p3_to_c3, i32, C3, 3, nppiCopy_32s_P3C3R_Ctx);
impl_copy_planar_to_packed!(copy_i32_p4_to_c4, i32, C4, 4, nppiCopy_32s_P4C4R_Ctx);
impl_copy_planar_to_packed!(copy_f32_p3_to_c3, f32, C3, 3, nppiCopy_32f_P3C3R_Ctx);
impl_copy_planar_to_packed!(copy_f32_p4_to_c4, f32, C4, 4, nppiCopy_32f_P4C4R_Ctx);
impl_generic_copy_packed_to_planar_operation!(
    CopyC3ToP3,
    copy_c3_to_p3,
    C3,
    3,
    [
        u8 => copy_u8_c3_to_p3,
        u16 => copy_u16_c3_to_p3,
        i16 => copy_i16_c3_to_p3,
        i32 => copy_i32_c3_to_p3,
        f32 => copy_f32_c3_to_p3,
    ]
);
impl_generic_copy_packed_to_planar_operation!(
    CopyC4ToP4,
    copy_c4_to_p4,
    C4,
    4,
    [
        u8 => copy_u8_c4_to_p4,
        u16 => copy_u16_c4_to_p4,
        i16 => copy_i16_c4_to_p4,
        i32 => copy_i32_c4_to_p4,
        f32 => copy_f32_c4_to_p4,
    ]
);
impl_generic_copy_planar_to_packed_operation!(
    CopyP3ToC3,
    copy_p3_to_c3,
    C3,
    3,
    [
        u8 => copy_u8_p3_to_c3,
        u16 => copy_u16_p3_to_c3,
        i16 => copy_i16_p3_to_c3,
        i32 => copy_i32_p3_to_c3,
        f32 => copy_f32_p3_to_c3,
    ]
);
impl_generic_copy_planar_to_packed_operation!(
    CopyP4ToC4,
    copy_p4_to_c4,
    C4,
    4,
    [
        u8 => copy_u8_p4_to_c4,
        u16 => copy_u16_p4_to_c4,
        i16 => copy_i16_p4_to_c4,
        i32 => copy_i32_p4_to_c4,
        f32 => copy_f32_p4_to_c4,
    ]
);