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
pub trait Diag
{
    type Output;
    
    /// Returns the diagonal of the given matrix
    /// 
    /// {aᵢᵢ}
    /// 
    /// # Examples
    /// 
    /// ```rust
    /// let a = [
    ///     [1.0, 2.0],
    ///     [3.0, 4.0]
    /// ];
    /// assert_eq!(a.diag(), [1.0, 4.0]);
    /// ```
    fn diag(&self) -> Self::Output;
}

impl<F, const L: usize, const H: usize> Diag for [[F; L]; H]
where F: Copy
{
    type Output = Vec<F>;
    fn diag(&self) -> Self::Output
    {
        (0..L.min(H)).map(|i| self[i][i]).collect()
    }
}