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

use crate::{
    context::StreamContext,
    error::Result,
    image::{
        geometry,
        view::{C1, ImageView, PlanarImageView, PlanarImageViewMut},
    },
    pipeline::{ImageAllocator, Workspace},
    types::Size,
};

use super::{CreatePlanarImage, PlanarImage, PlanarRemap};

impl<T, const C: usize> PlanarImage<T, C> {
    pub(in crate::pipeline::image) fn transform_geometry<G>(
        self,
        stream_context: &StreamContext,
        destination_size: Size,
        geometry: &G,
        operation: for<'source, 'destination> fn(
            &StreamContext,
            &G,
            &PlanarImageView<'source, T, C>,
            &mut PlanarImageViewMut<'destination, T, C>,
        ) -> Result<()>,
    ) -> Result<Self>
    where
        T: Copy,
        Workspace: ImageAllocator<T, C1>,
        Self: CreatePlanarImage<T, C>,
    {
        let mut destination = <Self as CreatePlanarImage<T, C>>::create(destination_size)?;

        {
            let source = <Self as CreatePlanarImage<T, C>>::view(&self)?;
            let mut destination_view =
                <Self as CreatePlanarImage<T, C>>::view_mut(&mut destination)?;
            operation(stream_context, geometry, &source, &mut destination_view)?;
        }

        Ok(destination)
    }

    pub(in crate::pipeline::image) fn remap_geometry<M>(
        self,
        stream_context: &StreamContext,
        remap: &geometry::Remap,
        x_map: &ImageView<'_, M, C1>,
        y_map: &ImageView<'_, M, C1>,
        operation: PlanarRemap<T, M, C>,
    ) -> Result<Self>
    where
        T: Copy,
        M: Copy,
        Workspace: ImageAllocator<T, C1>,
        Self: CreatePlanarImage<T, C>,
    {
        let mut destination = <Self as CreatePlanarImage<T, C>>::create(x_map.size())?;

        {
            let source = <Self as CreatePlanarImage<T, C>>::view(&self)?;
            let mut destination_view =
                <Self as CreatePlanarImage<T, C>>::view_mut(&mut destination)?;
            operation(
                stream_context,
                remap,
                &source,
                x_map,
                y_map,
                &mut destination_view,
            )?;
        }

        Ok(destination)
    }
}