dsalgo 0.3.10

A package for Datastructures and Algorithms.
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

#[derive(Debug, Clone, Copy, Eq, PartialEq)]

pub struct Matrix<T, const H: usize, const W: usize>([T; H * W])
where
    [(); H * W]:;

impl<T, const H: usize, const W: usize> Matrix<T, H, W>
where
    [(); H * W]:,
    T: Copy,
{
    pub fn new(fill_value: T) -> Self {
        Self([fill_value; H * W])
    }
}

impl<T, const H: usize, const W: usize> Default for Matrix<T, H, W>
where
    [(); H * W]:,
    T: Copy + Default,
{
    fn default() -> Self {
        Self::new(T::default())
    }
}

impl<T, const H: usize, const W: usize> From<[[T; W]; H]> for Matrix<T, H, W>
where
    [(); H * W]:,
    T: Copy + std::fmt::Debug,
{
    fn from(a: [[T; W]; H]) -> Self {
        Self(a.into_iter().flatten().collect::<Vec<_>>().try_into().unwrap())
    }
}

use std::ops::*;

impl<T, const H: usize, const W: usize> Index<(usize, usize)>
    for Matrix<T, H, W>
where
    [(); H * W]:,
{
    type Output = T;

    fn index(
        &self,
        index: (usize, usize),
    ) -> &Self::Output {
        &self.0[index.0 * W + index.1]
    }
}

impl<T, const H: usize, const W: usize> IndexMut<(usize, usize)>
    for Matrix<T, H, W>
where
    [(); H * W]:,
{
    fn index_mut(
        &mut self,
        index: (usize, usize),
    ) -> &mut Self::Output {
        &mut self.0[index.0 * W + index.1]
    }
}

impl<T, const H: usize, const W: usize> Matrix<T, H, W>
where
    [(); H * W]:,
    T: Copy + std::fmt::Debug,
{
    fn to_2d_array(&self) -> [[T; W]; H] {
        self.0
            .array_chunks::<W>()
            .map(|x| *x)
            .collect::<Vec<_>>()
            .try_into()
            .unwrap()
    }
}

impl<T, const H: usize, const W: usize> std::fmt::Display for Matrix<T, H, W>
where
    [(); H * W]:,
    T: std::fmt::Debug + Copy,
{
    fn fmt(
        &self,
        f: &mut std::fmt::Formatter<'_>,
    ) -> std::fmt::Result {
        let format_str = self
            .to_2d_array()
            .iter()
            .map(|row| format!("{:?}", row))
            .collect::<Vec<_>>()
            .join("\n");

        write!(f, "{}", format_str)
    }
}

#[cfg(test)]

mod tests {

    use super::*;

    #[test]

    fn test() {
        type Mat = Matrix<i64, 3, 2>;

        let mut a = Mat::default();

        a[(1, 1)] += 1;

        println!("{:?}", a);

        println!("{}", a);

        a = [[1, 2], [3, 4], [5, 6]].into();

        println!("{}", a);
    }
}