convolve-image 0.4.0

A small library to perform convolution operations on images with arbitrarily-sized separable kernels
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

#![cfg(feature = "ndarray")]

use ndarray::{Dimension, Ix2, Ix3};

pub trait DimensionIterator {
    type Item: Copy;
    
    fn into_iter(self) -> impl IntoIterator<Item = Self::Item>;
}

impl DimensionIterator for Ix2 {
    type Item = (usize, usize);

    fn into_iter(self) -> impl IntoIterator<Item=Self::Item> {
        let [y, x] = self.slice() else {
            panic!("Expected 2d slice")
        };
        
        let y = *y;
        let x = *x;

        (0..y).flat_map(move |y| (0..x).map(move |x| (y, x)))
    }
}

impl DimensionIterator for Ix3 {
    type Item = (usize, usize, usize);

    fn into_iter(self) -> impl IntoIterator<Item=Self::Item> {
        let [y, x, z] = self.slice() else {
            panic!("Expected 3d slice")
        };

        let y = *y;
        let x = *x;
        let z = *z;

        (0..y).flat_map(move |y| (0..x).flat_map(move |x| (0..z).map(move |z| (y, x, z))))
    }
}