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

use singe_cuda::memory::DeviceMemory;

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

use super::ImagePipeline;

pub use self::outputs::{
    ContiguousImage, ImageHistograms, ImageIndexedMinMax, ImageIndexedStatistic,
    ImageMeanStandardDeviation, ImageMinMax, ImageSquaredIntegral, ImageStatistic,
};
pub(super) use super::statistics_dispatch::*;

#[path = "statistics_outputs.rs"]
mod outputs;

impl<'a, T, L> ImagePipeline<'a, T, L> {
    pub fn even_levels_i32(levels: &mut [i32], lower_level: i32, upper_level: i32) -> Result<()> {
        statistics::even_levels_i32(levels, lower_level, upper_level)
    }

    pub fn even_levels<U: statistics::EvenLevels>(
        levels: &mut [U],
        lower_level: U,
        upper_level: U,
    ) -> Result<()> {
        statistics::even_levels(levels, lower_level, upper_level)
    }
}

pub(super) type BatchQualityMetric<T, L> = for<'source_0, 'source_1> fn(
    &StreamContext,
    &[ImageView<'source_0, T, L>],
    &[ImageView<'source_1, T, L>],
    &mut DeviceMemory<f32>,
) -> Result<()>;