pictorus-blocks 0.0.0

Implementations of Pictorus blocks.
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

//! Traits for inter-operation with nalgebra

// NOTE this only supports nalgebra 0.33.x
use nalgebra::{Const, Dim, MatrixView, MatrixViewMut};
use pictorus_traits::Matrix;

pub trait MatrixExt {
    type NROWS: Dim;
    type NCOLS: Dim;
    type Elem;

    fn as_view(&self) -> MatrixView<Self::Elem, Self::NROWS, Self::NCOLS>;
    fn as_view_mut(&mut self) -> MatrixViewMut<Self::Elem, Self::NROWS, Self::NCOLS>;
    fn from_view(view: &MatrixView<Self::Elem, Self::NROWS, Self::NCOLS>) -> Self;
}

impl<T, const NROWS: usize, const NCOLS: usize> MatrixExt for Matrix<NROWS, NCOLS, T>
where
    T: pictorus_traits::Scalar + nalgebra::Scalar,
{
    type NROWS = Const<NROWS>;
    type NCOLS = Const<NCOLS>;
    type Elem = T;

    fn as_view(&self) -> MatrixView<Self::Elem, Self::NROWS, Self::NCOLS> {
        MatrixView::<Self::Elem, Self::NROWS, Self::NCOLS>::from_slice(self.data.as_flattened())
    }

    fn as_view_mut(&mut self) -> MatrixViewMut<Self::Elem, Self::NROWS, Self::NCOLS> {
        MatrixViewMut::<Self::Elem, Self::NROWS, Self::NCOLS>::from_slice(
            self.data.as_flattened_mut(),
        )
    }

    fn from_view(view: &MatrixView<Self::Elem, Self::NROWS, Self::NCOLS>) -> Self {
        let mut m = pictorus_traits::Matrix::<NROWS, NCOLS, T>::zeroed();
        m.as_view_mut().copy_from(view);
        m
    }
}