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

use super::*;

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

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

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

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

        $(
            impl $trait<$layout> for $ty {
                fn $method(
                    stream_context: &StreamContext,
                    source: &ImageView<'_, Self, $layout>,
                    destination: &mut ImageViewMut<'_, Self, $layout>,
                ) -> Result<()> {
                    $direct(stream_context, source, destination)
                }
            }
        )*
    };
}

impl_morphology_3x3!(dilate_3x3_u8_c1, u8, C1, nppiDilate3x3_8u_C1R_Ctx);
impl_morphology_3x3!(dilate_3x3_u8_c3, u8, C3, nppiDilate3x3_8u_C3R_Ctx);
impl_morphology_3x3!(dilate_3x3_u8_c4, u8, C4, nppiDilate3x3_8u_C4R_Ctx);
impl_morphology_3x3!(dilate_3x3_u8_ac4, u8, AC4, nppiDilate3x3_8u_AC4R_Ctx);
impl_morphology_3x3!(dilate_3x3_u16_c1, u16, C1, nppiDilate3x3_16u_C1R_Ctx);
impl_morphology_3x3!(dilate_3x3_u16_c3, u16, C3, nppiDilate3x3_16u_C3R_Ctx);
impl_morphology_3x3!(dilate_3x3_u16_c4, u16, C4, nppiDilate3x3_16u_C4R_Ctx);
impl_morphology_3x3!(dilate_3x3_u16_ac4, u16, AC4, nppiDilate3x3_16u_AC4R_Ctx);
impl_morphology_3x3!(dilate_3x3_f32_c1, f32, C1, nppiDilate3x3_32f_C1R_Ctx);
impl_morphology_3x3!(dilate_3x3_f32_c3, f32, C3, nppiDilate3x3_32f_C3R_Ctx);
impl_morphology_3x3!(dilate_3x3_f32_c4, f32, C4, nppiDilate3x3_32f_C4R_Ctx);
impl_morphology_3x3!(dilate_3x3_f32_ac4, f32, AC4, nppiDilate3x3_32f_AC4R_Ctx);
impl_morphology_3x3!(dilate_3x3_f64_c1, f64, C1, nppiDilate3x3_64f_C1R_Ctx);

impl_morphology_3x3!(erode_3x3_u8_c1, u8, C1, nppiErode3x3_8u_C1R_Ctx);
impl_morphology_3x3!(erode_3x3_u8_c3, u8, C3, nppiErode3x3_8u_C3R_Ctx);
impl_morphology_3x3!(erode_3x3_u8_c4, u8, C4, nppiErode3x3_8u_C4R_Ctx);
impl_morphology_3x3!(erode_3x3_u8_ac4, u8, AC4, nppiErode3x3_8u_AC4R_Ctx);
impl_morphology_3x3!(erode_3x3_u16_c1, u16, C1, nppiErode3x3_16u_C1R_Ctx);
impl_morphology_3x3!(erode_3x3_u16_c3, u16, C3, nppiErode3x3_16u_C3R_Ctx);
impl_morphology_3x3!(erode_3x3_u16_c4, u16, C4, nppiErode3x3_16u_C4R_Ctx);
impl_morphology_3x3!(erode_3x3_u16_ac4, u16, AC4, nppiErode3x3_16u_AC4R_Ctx);
impl_morphology_3x3!(erode_3x3_f32_c1, f32, C1, nppiErode3x3_32f_C1R_Ctx);
impl_morphology_3x3!(erode_3x3_f32_c3, f32, C3, nppiErode3x3_32f_C3R_Ctx);
impl_morphology_3x3!(erode_3x3_f32_c4, f32, C4, nppiErode3x3_32f_C4R_Ctx);
impl_morphology_3x3!(erode_3x3_f32_ac4, f32, AC4, nppiErode3x3_32f_AC4R_Ctx);
impl_morphology_3x3!(erode_3x3_f64_c1, f64, C1, nppiErode3x3_64f_C1R_Ctx);

impl_generic_morphology_3x3!(Dilate3x3C1, dilate_3x3, dilate_3x3_c1, C1, [
    u8 => dilate_3x3_u8_c1,
    u16 => dilate_3x3_u16_c1,
    f32 => dilate_3x3_f32_c1,
    f64 => dilate_3x3_f64_c1,
]);
impl_generic_morphology_3x3!(Dilate3x3C3, dilate_3x3, dilate_3x3_c3, C3, [
    u8 => dilate_3x3_u8_c3,
    u16 => dilate_3x3_u16_c3,
    f32 => dilate_3x3_f32_c3,
]);
impl_generic_morphology_3x3!(Dilate3x3C4, dilate_3x3, dilate_3x3_c4, C4, [
    u8 => dilate_3x3_u8_c4,
    u16 => dilate_3x3_u16_c4,
    f32 => dilate_3x3_f32_c4,
]);
impl_generic_morphology_3x3!(Dilate3x3Ac4, dilate_3x3, dilate_3x3_ac4, AC4, [
    u8 => dilate_3x3_u8_ac4,
    u16 => dilate_3x3_u16_ac4,
    f32 => dilate_3x3_f32_ac4,
]);
impl_generic_morphology_3x3!(Erode3x3C1, erode_3x3, erode_3x3_c1, C1, [
    u8 => erode_3x3_u8_c1,
    u16 => erode_3x3_u16_c1,
    f32 => erode_3x3_f32_c1,
    f64 => erode_3x3_f64_c1,
]);
impl_generic_morphology_3x3!(Erode3x3C3, erode_3x3, erode_3x3_c3, C3, [
    u8 => erode_3x3_u8_c3,
    u16 => erode_3x3_u16_c3,
    f32 => erode_3x3_f32_c3,
]);
impl_generic_morphology_3x3!(Erode3x3C4, erode_3x3, erode_3x3_c4, C4, [
    u8 => erode_3x3_u8_c4,
    u16 => erode_3x3_u16_c4,
    f32 => erode_3x3_f32_c4,
]);
impl_generic_morphology_3x3!(Erode3x3Ac4, erode_3x3, erode_3x3_ac4, AC4, [
    u8 => erode_3x3_u8_ac4,
    u16 => erode_3x3_u16_ac4,
    f32 => erode_3x3_f32_ac4,
]);