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

macro_rules! impl_indexed_extremum_statistic_image {
    ($ty:ty, $layout:ty, $min_index:path, $max_index:path) => {
        impl<'a> IndexedExtremumStatisticImage<$ty, $layout> for ImagePipeline<'a, $ty, $layout> {
            const OUTPUT_CHANNELS: usize = <$layout as ChannelLayout>::CHANNELS;

            fn min_index(
                stream_context: &StreamContext,
                source: &ImageView<'_, $ty, $layout>,
                output: &mut DeviceMemory<$ty>,
                index_x: &mut DeviceMemory<i32>,
                index_y: &mut DeviceMemory<i32>,
            ) -> Result<()> {
                $min_index(stream_context, source, output, index_x, index_y)
            }

            fn max_index(
                stream_context: &StreamContext,
                source: &ImageView<'_, $ty, $layout>,
                output: &mut DeviceMemory<$ty>,
                index_x: &mut DeviceMemory<i32>,
                index_y: &mut DeviceMemory<i32>,
            ) -> Result<()> {
                $max_index(stream_context, source, output, index_x, index_y)
            }
        }
    };
}

macro_rules! impl_pair_indexed_extremum_statistic_image {
    ($ty:ty, $layout:ty, $min_max_index:path) => {
        impl<'a> PairIndexedExtremumStatisticImage<$ty, $layout>
            for ImagePipeline<'a, $ty, $layout>
        {
            fn min_max_index(
                stream_context: &StreamContext,
                source: &ImageView<'_, $ty, $layout>,
                min: &mut DeviceMemory<$ty>,
                max: &mut DeviceMemory<$ty>,
                min_index: &mut DeviceMemory<Point>,
                max_index: &mut DeviceMemory<Point>,
            ) -> Result<()> {
                $min_max_index(stream_context, source, min, max, min_index, max_index)
            }
        }
    };
}

macro_rules! impl_channel_pair_indexed_extremum_statistic_image {
    ($ty:ty, $layout:ty, $min_max_index:path) => {
        impl<'a> ChannelPairIndexedExtremumStatisticImage<$ty, $layout>
            for ImagePipeline<'a, $ty, $layout>
        {
            fn min_max_index_channel(
                stream_context: &StreamContext,
                source: &ImageView<'_, $ty, $layout>,
                channel: usize,
                min: &mut DeviceMemory<$ty>,
                max: &mut DeviceMemory<$ty>,
                min_index: &mut DeviceMemory<Point>,
                max_index: &mut DeviceMemory<Point>,
            ) -> Result<()> {
                $min_max_index(
                    stream_context,
                    source,
                    channel,
                    min,
                    max,
                    min_index,
                    max_index,
                )
            }
        }
    };
}

macro_rules! impl_masked_pair_indexed_extremum_statistic_image {
    ($ty:ty, $min_max_index:path) => {
        impl<'a> MaskedPairIndexedExtremumStatisticImage<$ty, C1> for ImagePipeline<'a, $ty, C1> {
            fn min_max_index_masked(
                stream_context: &StreamContext,
                source: &ImageView<'_, $ty, C1>,
                mask: &MaskView<'_>,
                min: &mut DeviceMemory<$ty>,
                max: &mut DeviceMemory<$ty>,
                min_index: &mut DeviceMemory<Point>,
                max_index: &mut DeviceMemory<Point>,
            ) -> Result<()> {
                $min_max_index(stream_context, source, mask, min, max, min_index, max_index)
            }
        }
    };
}

macro_rules! impl_masked_channel_pair_indexed_extremum_statistic_image {
    ($ty:ty, $min_max_index:path) => {
        impl<'a> MaskedChannelPairIndexedExtremumStatisticImage<$ty, C3>
            for ImagePipeline<'a, $ty, C3>
        {
            fn min_max_index_channel_masked(
                stream_context: &StreamContext,
                source: &ImageView<'_, $ty, C3>,
                mask: &MaskView<'_>,
                channel: usize,
                min: &mut DeviceMemory<$ty>,
                max: &mut DeviceMemory<$ty>,
                min_index: &mut DeviceMemory<Point>,
                max_index: &mut DeviceMemory<Point>,
            ) -> Result<()> {
                $min_max_index(
                    stream_context,
                    source,
                    mask,
                    channel,
                    min,
                    max,
                    min_index,
                    max_index,
                )
            }
        }
    };
}