wee-matrix 0.1.0

A matrix library for Rust
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

// Example:
// let a = mat![1, 2, 3, 4; 5, 6, 7, 8; 9, 10, 11, 12];
// assert_eq!(a.dims(), (3, 4));
#[macro_export]
macro_rules! mat {
    ($( $( $x:expr ),* );*) => {{
        let mut v: Vec<Vec<f64>> = Vec::new();
        let mut nrows = 0;
        let mut ncols = 0;
        $(
            v.push(Vec::new());
            let mut tmp_ncols = 0;
            $(
                v[nrows].push($x as f64);
                tmp_ncols += 1;
            )*
            if nrows > 0 {
                assert_eq!(ncols, tmp_ncols);
            }
            ncols = tmp_ncols;
            nrows += 1;
        )*

        let mut data: Vec<f64> = Vec::new();
        for j in 0..ncols {
            for i in 0..nrows {
                data.push(v[i][j]);
            }
        }

        Matrix::from_vec(data, nrows, ncols)
    }}
}

#[cfg(test)]
mod tests {
    use Matrix;

    #[test]
    fn test_macro() {
        let a = mat![1, 2, 3, 4; 5, 6, 7, 8];
        assert_eq!(a.dims(), (2, 4));

        assert_eq!(a.get(0, 0).unwrap(), 1.0);
        assert_eq!(a.get(0, 1).unwrap(), 2.0);
        assert_eq!(a.get(0, 2).unwrap(), 3.0);
        assert_eq!(a.get(0, 3).unwrap(), 4.0);
        assert_eq!(a.get(1, 0).unwrap(), 5.0);
        assert_eq!(a.get(1, 1).unwrap(), 6.0);
        assert_eq!(a.get(1, 2).unwrap(), 7.0);
        assert_eq!(a.get(1, 3).unwrap(), 8.0);
        println!("{:#?}", a);

        let a = mat!(1, 2; 3.0, 4; 5.5, 6);
        assert_eq!(a.dims(), (3, 2));

        assert_eq!(a.get(0, 0).unwrap(), 1.0);
        assert_eq!(a.get(0, 1).unwrap(), 2.0);
        assert_eq!(a.get(1, 0).unwrap(), 3.0);
        assert_eq!(a.get(1, 1).unwrap(), 4.0);
        assert_eq!(a.get(2, 0).unwrap(), 5.5);
        assert_eq!(a.get(2, 1).unwrap(), 6.0);
        println!("{:#?}", a);

    }
}