use std::ops::{Sub, SubAssign};
use crate::matrix::Matrix;
impl<const M: usize, const N: usize> Sub<&Matrix<M, N>> for &Matrix<M, N> {
type Output = Matrix<M, N>;
fn sub(self, rhs: &Matrix<M, N>) -> Self::Output {
let mut result = Matrix::zeros();
for m in 0..M {
for n in 0..N {
result[(m, n)] = self[(m, n)] - rhs[(m, n)];
}
}
result
}
}
#[allow(clippy::op_ref)]
impl<const M: usize, const N: usize> Sub<Matrix<M, N>> for &Matrix<M, N> {
type Output = Matrix<M, N>;
fn sub(self, rhs: Matrix<M, N>) -> Self::Output {
self - &rhs
}
}
#[allow(clippy::op_ref)]
impl<const M: usize, const N: usize> Sub<&Matrix<M, N>> for Matrix<M, N> {
type Output = Matrix<M, N>;
fn sub(self, rhs: &Matrix<M, N>) -> Self::Output {
&self - rhs
}
}
impl<const M: usize, const N: usize> Sub<Matrix<M, N>> for Matrix<M, N> {
type Output = Matrix<M, N>;
fn sub(self, rhs: Matrix<M, N>) -> Self::Output {
&self - &rhs
}
}
impl<const M: usize, const N: usize> SubAssign<&Matrix<M, N>> for Matrix<M, N> {
fn sub_assign(&mut self, rhs: &Self) {
for m in 0..M {
for n in 0..N {
self[(m, n)] -= rhs[(m, n)];
}
}
}
}
impl<const M: usize, const N: usize> SubAssign<Matrix<M, N>> for Matrix<M, N> {
fn sub_assign(&mut self, rhs: Self) {
*self -= &rhs;
}
}