use crate::matrix::ge::Matrix;
use crate::number::Number;
use std::ops::{Index, IndexMut};
impl<T> Index<usize> for Matrix<T>
where
T: Number,
{
type Output = [T];
fn index(&self, index: usize) -> &Self::Output {
let i = self.rows * index;
&self.elems[i..i + self.rows]
}
}
impl<T> IndexMut<usize> for Matrix<T>
where
T: Number,
{
fn index_mut(&mut self, index: usize) -> &mut Self::Output {
let i = self.rows * index;
&mut self.elems[i..i + self.rows]
}
}
impl<T> Index<(usize, usize)> for Matrix<T>
where
T: Number,
{
type Output = T;
fn index(&self, index: (usize, usize)) -> &Self::Output {
&self.elems[index.0 + index.1 * self.rows]
}
}
impl<T> IndexMut<(usize, usize)> for Matrix<T>
where
T: Number,
{
fn index_mut(&mut self, index: (usize, usize)) -> &mut Self::Output {
&mut self.elems[index.0 + index.1 * self.rows]
}
}
mod tests {
#[allow(unused_imports)]
use crate::*;
#[test]
fn it_works() {
let a = mat!(
1.0, 2.0;
3.0, 4.0
);
assert_eq!(a[0], [1.0, 3.0]);
assert_eq!(a[1], [2.0, 4.0]);
}
#[test]
fn it_works2() {
let a = mat!(
1.0, 2.0;
3.0, 4.0
);
assert_eq!(a[(0, 0)], 1.0);
assert_eq!(a[(0, 1)], 2.0);
assert_eq!(a[(1, 0)], 3.0);
assert_eq!(a[(1, 1)], 4.0);
}
}