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

use singe_cuda::types::Complex32;
use singe_npp_sys as sys;

use crate::{
    context::StreamContext,
    error::{Error, Result},
    image::view::{C1, ImageView, ImageViewMut},
    try_ffi,
    types::{DataTypeLike, Size},
};

pub trait MagnitudeC1: DataTypeLike + Sized {
    type Output: DataTypeLike;

    fn magnitude(
        stream_context: &StreamContext,
        source: &ImageView<'_, Self, C1>,
        destination: &mut ImageViewMut<'_, Self::Output, C1>,
    ) -> Result<()>;
}

pub fn magnitude<T>(
    stream_context: &StreamContext,
    source: &ImageView<'_, T, C1>,
    destination: &mut ImageViewMut<'_, T::Output, C1>,
) -> Result<()>
where
    T: MagnitudeC1,
{
    T::magnitude(stream_context, source, destination)
}

pub trait MagnitudeSquaredC1: DataTypeLike + Sized {
    type Output: DataTypeLike;

    fn magnitude_squared(
        stream_context: &StreamContext,
        source: &ImageView<'_, Self, C1>,
        destination: &mut ImageViewMut<'_, Self::Output, C1>,
    ) -> Result<()>;
}

pub fn magnitude_squared<T>(
    stream_context: &StreamContext,
    source: &ImageView<'_, T, C1>,
    destination: &mut ImageViewMut<'_, T::Output, C1>,
) -> Result<()>
where
    T: MagnitudeSquaredC1,
{
    T::magnitude_squared(stream_context, source, destination)
}

pub(crate) fn magnitude_f32_complex_c1(
    stream_context: &StreamContext,
    source: &ImageView<'_, Complex32, C1>,
    destination: &mut ImageViewMut<'_, f32, C1>,
) -> Result<()> {
    validate_same_size(source.size(), destination.size())?;

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

pub(crate) fn magnitude_squared_f32_complex_c1(
    stream_context: &StreamContext,
    source: &ImageView<'_, Complex32, C1>,
    destination: &mut ImageViewMut<'_, f32, C1>,
) -> Result<()> {
    validate_same_size(source.size(), destination.size())?;

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

impl MagnitudeC1 for Complex32 {
    type Output = f32;

    fn magnitude(
        stream_context: &StreamContext,
        source: &ImageView<'_, Self, C1>,
        destination: &mut ImageViewMut<'_, Self::Output, C1>,
    ) -> Result<()> {
        magnitude_f32_complex_c1(stream_context, source, destination)
    }
}

impl MagnitudeSquaredC1 for Complex32 {
    type Output = f32;

    fn magnitude_squared(
        stream_context: &StreamContext,
        source: &ImageView<'_, Self, C1>,
        destination: &mut ImageViewMut<'_, Self::Output, C1>,
    ) -> Result<()> {
        magnitude_squared_f32_complex_c1(stream_context, source, destination)
    }
}

fn validate_same_size(source: Size, destination: Size) -> Result<()> {
    if source == destination {
        return Ok(());
    }
    Err(Error::SizeMismatch {
        name: "image size".into(),
        expected: source,
        actual: destination,
    })
}