russell_lab 1.13.0

Scientific laboratory for linear algebra and numerical mathematics
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

use russell_lab::*;

fn main() -> Result<(), StrError> {
    // allocate matrices
    #[rustfmt::skip]
    let a = ComplexMatrix::from(&[
        [cpx!(1.0, 0.0), cpx!(2.0, 0.0), cpx!( 0.0,  1.0), cpx!(1.0, 0.0), cpx!(-1.0, 0.0)],
        [cpx!(2.0, 0.0), cpx!(3.0, 0.0), cpx!(-1.0, -1.0), cpx!(1.0, 0.0), cpx!( 1.0, 0.0)],
        [cpx!(1.0, 0.0), cpx!(2.0, 0.0), cpx!( 0.0,  1.0), cpx!(4.0, 0.0), cpx!(-1.0, 0.0)],
        [cpx!(4.0, 0.0), cpx!(0.0, 0.0), cpx!( 3.0, -1.0), cpx!(1.0, 0.0), cpx!( 1.0, 0.0)],
    ]);
    #[rustfmt::skip]
    let b = ComplexMatrix::from(&[
        [cpx!(1.0, 0.0), cpx!(0.0, 0.0), cpx!(0.0,  1.0)],
        [cpx!(0.0, 0.0), cpx!(0.0, 0.0), cpx!(3.0, -1.0)],
        [cpx!(0.0, 0.0), cpx!(0.0, 0.0), cpx!(1.0,  1.0)],
        [cpx!(1.0, 0.0), cpx!(0.0, 0.0), cpx!(1.0, -1.0)],
        [cpx!(0.0, 0.0), cpx!(2.0, 0.0), cpx!(0.0,  1.0)],
    ]);
    const NOISE: Complex64 = cpx!(123.0, 456.0);
    #[rustfmt::skip]
    let mut c = ComplexMatrix::from(&[
        [NOISE, NOISE, NOISE],
        [NOISE, NOISE, NOISE],
        [NOISE, NOISE, NOISE],
        [NOISE, NOISE, NOISE],
    ]);

    // multiply matrices
    // c = (0.5 - 2i) ⋅ a ⋅ b
    let alpha = cpx!(0.5, -2.0);
    let beta = cpx!(0.0, 0.0);
    complex_mat_mat_mul(&mut c, alpha, &a, &b, beta)?;

    // check
    #[rustfmt::skip]
    let correct = &[
        [cpx!(1.0, -4.0), cpx!(-1.0,  4.0), cpx!(-1.0, -13.0)],
        [cpx!(1.5, -6.0), cpx!( 1.0, -4.0), cpx!(-1.0, -21.5)],
        [cpx!(2.5,-10.0), cpx!(-1.0,  4.0), cpx!(-5.5, -20.5)],
        [cpx!(2.5,-10.0), cpx!( 1.0, -4.0), cpx!(14.5,  -7.0)],
    ];
    complex_mat_approx_eq(&c, correct, 1e-15);
    Ok(())
}