del-geo-core 0.1.38

2D/3D geometry utility codes
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

pub fn mult_vec<Real, const N: usize, const NN: usize>(a: &[Real; NN], b: &[Real; N]) -> [Real; N]
where
    Real: num_traits::Float,
{
    assert_eq!(NN, N * N);
    let mut c = [Real::zero(); N];
    for i in 0..N {
        for j in 0..N {
            c[i] = c[i] + a[i + N * j] * b[j];
        }
    }
    c
}

pub fn add_in_place<Real, const NN: usize>(a: &mut [Real; NN], b: &[Real; NN])
where
    Real: num_traits::Float,
{
    a.iter_mut()
        .zip(b.iter())
        .for_each(|(va, vb)| *va = *va + *vb);
}

pub fn mult_mat_col_major<Real, const N: usize, const NN: usize>(
    a: &[Real; NN],
    b: &[Real; NN],
) -> [Real; NN]
where
    Real: num_traits::Float,
{
    assert_eq!(NN, N * N);
    let mut c = [Real::zero(); NN];
    for i in 0..N {
        for j in 0..N {
            for k in 0..N {
                c[i + N * j] = c[i + N * j] + a[i + N * k] * b[k + N * j];
            }
        }
    }
    c
}

pub fn try_inverse<Real, const N: usize, const NN: usize>(b: &[Real; NN]) -> Option<[Real; NN]>
where
    Real: num_traits::Float,
{
    crate::matn_row_major::try_inverse::<Real, N, NN>(b)
}

pub fn sub_in_place<Real, const NN: usize>(a: &mut [Real; NN], b: &[Real; NN])
where
    Real: num_traits::Float,
{
    a.iter_mut()
        .zip(b.iter())
        .for_each(|(va, &vb)| *va = (*va) - vb);
}

pub fn transpose<Real, const N: usize, const NN: usize>(a: &[Real; NN]) -> [Real; NN]
where
    Real: num_traits::Float,
{
    let mut c = [Real::zero(); NN];
    for i in 0..N {
        for j in 0..N {
            c[i + N * j] = a[j + N * i];
        }
    }
    c
}