smpl-utils 0.9.0

Small library for utility functions related to smpl-rs
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

use ndarray as nd;
use num_traits;
pub trait Gather1D<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> {
    fn gather(&self, indices: &[usize]) -> nd::Array1<T>;
}
impl<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> Gather1D<T> for nd::Array1<T> {
    fn gather(&self, indices: &[usize]) -> nd::Array1<T> {
        let mut res = nd::Array1::<T>::zeros(indices.len());
        for (i_out, &i_in) in indices.iter().enumerate() {
            res[i_out] = self[i_in];
        }
        res
    }
}
pub trait Gather2D<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> {
    fn gather(&self, indices_rows: &[usize], indices_cols: &[usize]) -> nd::Array2<T>;
}
impl<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> Gather2D<T> for nd::Array2<T> {
    fn gather(&self, indices_rows: &[usize], indices_cols: &[usize]) -> nd::Array2<T> {
        let mut res = nd::Array2::zeros((indices_rows.len(), indices_cols.len()));
        for (i_out, &i_in) in indices_rows.iter().enumerate() {
            for (j_out, &j_in) in indices_cols.iter().enumerate() {
                res[(i_out, j_out)] = self[(i_in, j_in)];
            }
        }
        res
    }
}
pub trait Gather3D<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> {
    fn gather(&self, indices_rows: &[usize], indices_cols: &[usize], indices_depth: &[usize]) -> nd::Array3<T>;
}
impl<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> Gather3D<T> for nd::Array3<T> {
    fn gather(&self, indices_rows: &[usize], indices_cols: &[usize], indices_depth: &[usize]) -> nd::Array3<T> {
        let mut res = nd::Array3::zeros((indices_rows.len(), indices_cols.len(), indices_depth.len()));
        for (i_out, &i_in) in indices_rows.iter().enumerate() {
            for (j_out, &j_in) in indices_cols.iter().enumerate() {
                for (k_out, &k_in) in indices_depth.iter().enumerate() {
                    res[(i_out, j_out, k_out)] = self[(i_in, j_in, k_in)];
                }
            }
        }
        res
    }
}
pub trait Scatter1D<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> {
    fn scatter(&self, indices: &[usize], dst: &mut nd::Array1<T>);
}
impl<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> Scatter1D<T> for nd::Array1<T> {
    fn scatter(&self, indices: &[usize], dst: &mut nd::Array1<T>) {
        for (i_in, &i_out) in indices.iter().enumerate() {
            dst[i_out] = self[i_in];
        }
    }
}
pub trait Scatter2D<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> {
    fn scatter(&self, indices_rows: &[usize], indices_cols: &[usize], dst: &mut nd::Array2<T>);
}
impl<T: nd::ScalarOperand + num_traits::identities::Zero + Copy> Scatter2D<T> for nd::Array2<T> {
    fn scatter(&self, indices_rows: &[usize], indices_cols: &[usize], dst: &mut nd::Array2<T>) {
        for (i_in, &i_out) in indices_rows.iter().enumerate() {
            for (j_in, &j_out) in indices_cols.iter().enumerate() {
                dst[(i_out, j_out)] = self[(i_in, j_in)];
            }
        }
    }
}