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

use singe_cuda::memory::DeviceMemory;

use crate::{
    context::StreamContext,
    error::{Error, Result},
    image::view::{ChannelLayout, ImageView},
};

use super::{
    ImagePipeline,
    statistics::{HistogramEvenImage, HistogramRangeImage, ImageStatistic},
};

#[path = "statistics_count_in_range_methods.rs"]
mod count_in_range_methods;

impl<'a, T, L> ImagePipeline<'a, T, L>
where
    T: Copy,
    L: ChannelLayout,
{
    pub fn histogram_range_into<Level>(
        stream_context: &StreamContext,
        source: &ImageView<'_, T, L>,
        histogram: &mut DeviceMemory<i32>,
        levels: &[Level],
    ) -> Result<()>
    where
        Level: Copy,
        Self: HistogramRangeImage<T, L, Level>,
    {
        <Self as HistogramRangeImage<T, L, Level>>::histogram_range(
            stream_context,
            source,
            histogram,
            levels,
        )
    }

    pub fn histogram_range<Level>(self, levels: &[Level]) -> Result<ImageStatistic<i32>>
    where
        Level: Copy,
        Self: HistogramRangeImage<T, L, Level>,
    {
        let mut histogram = DeviceMemory::<i32>::create(histogram_range_bins(levels)?)?;
        {
            let source = self.view()?;
            <Self as HistogramRangeImage<T, L, Level>>::histogram_range(
                self.stream_context,
                &source,
                &mut histogram,
                levels,
            )?;
        }
        Ok(ImageStatistic::from_values(histogram))
    }
}

impl<'a, T, L> ImagePipeline<'a, T, L>
where
    T: Copy,
    L: ChannelLayout,
    Self: HistogramEvenImage<T, L>,
{
    pub fn histogram_even_into(
        stream_context: &StreamContext,
        source: &ImageView<'_, T, L>,
        histogram: &mut DeviceMemory<i32>,
        levels: i32,
        lower_level: i32,
        upper_level: i32,
    ) -> Result<()> {
        <Self as HistogramEvenImage<T, L>>::histogram_even(
            stream_context,
            source,
            histogram,
            levels,
            lower_level,
            upper_level,
        )
    }

    pub fn histogram_even(
        self,
        levels: i32,
        lower_level: i32,
        upper_level: i32,
    ) -> Result<ImageStatistic<i32>> {
        if levels < 2 {
            return Err(Error::OutOfRange {
                name: "levels".into(),
            });
        }
        let bins = usize::try_from(levels - 1).map_err(|_| Error::OutOfRange {
            name: "levels".into(),
        })?;
        let mut histogram = DeviceMemory::<i32>::create(bins)?;
        {
            let source = self.view()?;
            <Self as HistogramEvenImage<T, L>>::histogram_even(
                self.stream_context,
                &source,
                &mut histogram,
                levels,
                lower_level,
                upper_level,
            )?;
        }
        Ok(ImageStatistic::from_values(histogram))
    }
}

fn histogram_range_bins<Level>(levels: &[Level]) -> Result<usize> {
    if levels.len() < 2 {
        return Err(Error::OutOfRange {
            name: "levels".into(),
        });
    }
    Ok(levels.len() - 1)
}