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

use crate::structure::matrix::Shape;

pub trait MutFP {
    type Scalar;
    fn mut_map<F>(&mut self, f: F)
    where
        F: Fn(Self::Scalar) -> Self::Scalar;
    fn mut_zip_with<F>(&mut self, f: F, other: &Self)
    where
        F: Fn(Self::Scalar, Self::Scalar) -> Self::Scalar;
}

pub trait MutMatrix {
    type Scalar;

    unsafe fn col_mut(&mut self, idx: usize) -> Vec<*mut Self::Scalar>;
    unsafe fn row_mut(&mut self, idx: usize) -> Vec<*mut Self::Scalar>;
    unsafe fn swap(&mut self, idx1: usize, idx2: usize, shape: Shape);
    unsafe fn swap_with_perm(&mut self, p: &Vec<(usize, usize)>, shape: Shape);
}

// Mutable trait for Vector in Parallel (Uses Rayon crate)
pub trait ParallelMutFP {
    type Scalar;
    fn par_mut_map<F>(&mut self, f: F)
    where
        F: Fn(Self::Scalar) -> Self::Scalar + Send + Sync;
    fn par_mut_zip_with<F>(&mut self, f: F, other: &Self)
    where
        F: Fn(Self::Scalar, Self::Scalar) -> Self::Scalar + Send + Sync;
}