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

use super::*;

pub(crate) fn copy_subpixel_u8_c1(
    stream_context: &StreamContext,
    source: &ImageView<'_, u8, C1>,
    destination: &mut ImageViewMut<'_, u8, C1>,
    x_shift: f32,
    y_shift: f32,
) -> Result<()> {
    validate_subpixel_copy(source.size(), destination.size(), x_shift, y_shift)?;

    unsafe {
        try_ffi!(sys::nppiCopySubpix_8u_C1R_Ctx(
            source.as_ptr().cast(),
            source.step(),
            destination.as_mut_ptr().cast(),
            destination.step(),
            destination.size().into(),
            x_shift,
            y_shift,
            stream_context.as_raw(),
        ))?;
    }
    Ok(())
}

macro_rules! impl_copy_subpixel {
    ($name:ident, $ty:ty, $layout:ty, $ffi:ident) => {
        pub(crate) fn $name(
            stream_context: &StreamContext,
            source: &ImageView<'_, $ty, $layout>,
            destination: &mut ImageViewMut<'_, $ty, $layout>,
            x_shift: f32,
            y_shift: f32,
        ) -> Result<()> {
            validate_subpixel_copy(source.size(), destination.size(), x_shift, y_shift)?;

            unsafe {
                try_ffi!(sys::$ffi(
                    source.as_ptr().cast(),
                    source.step(),
                    destination.as_mut_ptr().cast(),
                    destination.step(),
                    destination.size().into(),
                    x_shift,
                    y_shift,
                    stream_context.as_raw(),
                ))?;
            }
            Ok(())
        }
    };
}

macro_rules! impl_generic_copy_subpixel_operation {
    ($trait:ident, $function:ident, $layout:ty, [$($ty:ty => $direct:ident),+ $(,)?]) => {
        pub trait $trait<Layout>: DataTypeLike + Sized {
            fn dispatch(
                stream_context: &StreamContext,
                source: &ImageView<'_, Self, Layout>,
                destination: &mut ImageViewMut<'_, Self, Layout>,
                x_shift: f32,
                y_shift: f32,
            ) -> Result<()>;
        }

        $(
            impl $trait<$layout> for $ty {
                fn dispatch(
                    stream_context: &StreamContext,
                    source: &ImageView<'_, Self, $layout>,
                    destination: &mut ImageViewMut<'_, Self, $layout>,
                    x_shift: f32,
                    y_shift: f32,
                ) -> Result<()> {
                    $direct(stream_context, source, destination, x_shift, y_shift)
                }
            }
        )+

        pub fn $function<T>(
            stream_context: &StreamContext,
            source: &ImageView<'_, T, $layout>,
            destination: &mut ImageViewMut<'_, T, $layout>,
            x_shift: f32,
            y_shift: f32,
        ) -> Result<()>
        where
            T: $trait<$layout>,
        {
            T::dispatch(stream_context, source, destination, x_shift, y_shift)
        }
    };
}

impl_copy_subpixel!(copy_subpixel_u8_c3, u8, C3, nppiCopySubpix_8u_C3R_Ctx);
impl_copy_subpixel!(copy_subpixel_u8_c4, u8, C4, nppiCopySubpix_8u_C4R_Ctx);
impl_copy_subpixel!(copy_subpixel_u8_ac4, u8, AC4, nppiCopySubpix_8u_AC4R_Ctx);
impl_copy_subpixel!(copy_subpixel_u16_c1, u16, C1, nppiCopySubpix_16u_C1R_Ctx);
impl_copy_subpixel!(copy_subpixel_u16_c3, u16, C3, nppiCopySubpix_16u_C3R_Ctx);
impl_copy_subpixel!(copy_subpixel_u16_c4, u16, C4, nppiCopySubpix_16u_C4R_Ctx);
impl_copy_subpixel!(copy_subpixel_u16_ac4, u16, AC4, nppiCopySubpix_16u_AC4R_Ctx);
impl_copy_subpixel!(copy_subpixel_i16_c1, i16, C1, nppiCopySubpix_16s_C1R_Ctx);
impl_copy_subpixel!(copy_subpixel_i16_c3, i16, C3, nppiCopySubpix_16s_C3R_Ctx);
impl_copy_subpixel!(copy_subpixel_i16_c4, i16, C4, nppiCopySubpix_16s_C4R_Ctx);
impl_copy_subpixel!(copy_subpixel_i16_ac4, i16, AC4, nppiCopySubpix_16s_AC4R_Ctx);
impl_copy_subpixel!(copy_subpixel_i32_c1, i32, C1, nppiCopySubpix_32s_C1R_Ctx);
impl_copy_subpixel!(copy_subpixel_i32_c3, i32, C3, nppiCopySubpix_32s_C3R_Ctx);
impl_copy_subpixel!(copy_subpixel_i32_c4, i32, C4, nppiCopySubpix_32s_C4R_Ctx);
impl_copy_subpixel!(copy_subpixel_i32_ac4, i32, AC4, nppiCopySubpix_32s_AC4R_Ctx);
impl_copy_subpixel!(copy_subpixel_f32_c1, f32, C1, nppiCopySubpix_32f_C1R_Ctx);
impl_copy_subpixel!(copy_subpixel_f32_c3, f32, C3, nppiCopySubpix_32f_C3R_Ctx);
impl_copy_subpixel!(copy_subpixel_f32_c4, f32, C4, nppiCopySubpix_32f_C4R_Ctx);
impl_copy_subpixel!(copy_subpixel_f32_ac4, f32, AC4, nppiCopySubpix_32f_AC4R_Ctx);
impl_generic_copy_subpixel_operation!(
    CopySubpixelC1,
    copy_subpixel_c1,
    C1,
    [
        u8 => copy_subpixel_u8_c1,
        u16 => copy_subpixel_u16_c1,
        i16 => copy_subpixel_i16_c1,
        i32 => copy_subpixel_i32_c1,
        f32 => copy_subpixel_f32_c1,
    ]
);
impl_generic_copy_subpixel_operation!(
    CopySubpixelC3,
    copy_subpixel_c3,
    C3,
    [
        u8 => copy_subpixel_u8_c3,
        u16 => copy_subpixel_u16_c3,
        i16 => copy_subpixel_i16_c3,
        i32 => copy_subpixel_i32_c3,
        f32 => copy_subpixel_f32_c3,
    ]
);
impl_generic_copy_subpixel_operation!(
    CopySubpixelC4,
    copy_subpixel_c4,
    C4,
    [
        u8 => copy_subpixel_u8_c4,
        u16 => copy_subpixel_u16_c4,
        i16 => copy_subpixel_i16_c4,
        i32 => copy_subpixel_i32_c4,
        f32 => copy_subpixel_f32_c4,
    ]
);
impl_generic_copy_subpixel_operation!(
    CopySubpixelAC4,
    copy_subpixel_ac4,
    AC4,
    [
        u8 => copy_subpixel_u8_ac4,
        u16 => copy_subpixel_u16_ac4,
        i16 => copy_subpixel_i16_ac4,
        i32 => copy_subpixel_i32_ac4,
        f32 => copy_subpixel_f32_ac4,
    ]
);