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

use crate::{
    dim::{Dim2, Dim3, Dim4, DimDyn},
    matrix::MatrixBase,
    matrix_impl::Matrix,
    memory::Memory,
    num::Num,
};

pub trait Transpose {
    fn transpose(&mut self);
}

macro_rules! impl_transpose {
    ($dim:ty) => {
        impl<T: Num, M: Memory<Item = T>> Transpose for Matrix<M, $dim> {
            #[allow(clippy::almost_swapped)]
            fn transpose(&mut self) {
                let shape_stride = self.shape_stride();
                let transposed = shape_stride.transpose();

                self.update_shape(transposed.shape());
                self.update_stride(transposed.stride());
            }
        }
    };
}
impl_transpose!(Dim2);
impl_transpose!(Dim3);
impl_transpose!(Dim4);

impl<T: Num, M: Memory<Item = T>> Transpose for Matrix<M, DimDyn> {
    fn transpose(&mut self) {
        let shape_stride = self.shape_stride();
        let transposed = shape_stride.transpose();

        self.update_shape(transposed.shape());
        self.update_stride(transposed.stride());
    }
}

#[cfg(test)]
mod transpose {
    use crate::{
        matrix::{IndexItem, OwnedMatrix},
        matrix_impl::OwnedMatrixDyn,
    };

    use super::Transpose;

    #[test]
    fn transpose_2d() {
        let mut a = OwnedMatrixDyn::from_vec(vec![1., 2., 3., 4., 5., 6.], [2, 3]);
        a.transpose();
        assert_eq!(a.index_item([0, 0]), 1.);
        assert_eq!(a.index_item([0, 1]), 4.);
        assert_eq!(a.index_item([1, 0]), 2.);
        assert_eq!(a.index_item([1, 1]), 5.);
        assert_eq!(a.index_item([2, 0]), 3.);
        assert_eq!(a.index_item([2, 1]), 6.);
    }
}