diff options
| author | hachem <im@hachem.wtf> | 2024-09-24 21:33:51 +0200 |
|---|---|---|
| committer | hachem <im@hachem.wtf> | 2024-09-24 21:33:51 +0200 |
| commit | c07ab7911bd50d0060841690a863333a9a3f226d (patch) | |
| tree | 5deb12fc8234d553318e4615ae21ddfcc6db4ab4 /libpsi-core/src/maths/matrix_ops.rs | |
| parent | 26010345eedc716c70e8a4219005db19b266c960 (diff) | |
rename project
Diffstat (limited to 'libpsi-core/src/maths/matrix_ops.rs')
| -rw-r--r-- | libpsi-core/src/maths/matrix_ops.rs | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/libpsi-core/src/maths/matrix_ops.rs b/libpsi-core/src/maths/matrix_ops.rs new file mode 100644 index 0000000..15ab28b --- /dev/null +++ b/libpsi-core/src/maths/matrix_ops.rs @@ -0,0 +1,62 @@ +use super::{Float, Matrix}; +use core::ops; + +impl<T: Float> ops::Add<&Matrix<T>> for Matrix<T> { + type Output = Option<Matrix<T>>; + + fn add(self, other: &Matrix<T>) -> Self::Output { + self.add_to(other) + } +} + +impl<T: Float> ops::Sub<&Matrix<T>> for Matrix<T> { + type Output = Option<Matrix<T>>; + + fn sub(self, other: &Matrix<T>) -> Self::Output { + self.subtract(other) + } +} + +impl<T: Float> ops::Mul<T> for Matrix<T> { + type Output = Matrix<T>; + + fn mul(self, scalar: T) -> Self::Output { + self.scale(scalar) + } +} + +impl<T: Float> ops::Div<T> for Matrix<T> { + type Output = Matrix<T>; + + fn div(self, scalar: T) -> Self::Output { + self.scale(T::one() / scalar) + } +} + +impl<T: Float> ops::AddAssign<&Matrix<T>> for Matrix<T> { + fn add_assign(&mut self, other: &Matrix<T>) { + if let Some(result) = self.add_to(other) { + *self = result; + } + } +} + +impl<T: Float> ops::SubAssign<&Matrix<T>> for Matrix<T> { + fn sub_assign(&mut self, other: &Matrix<T>) { + if let Some(result) = self.subtract(other) { + *self = result; + } + } +} + +impl<T: Float> ops::MulAssign<T> for Matrix<T> { + fn mul_assign(&mut self, scalar: T) { + *self = self.scale(scalar); + } +} + +impl<T: Float> ops::DivAssign<T> for Matrix<T> { + fn div_assign(&mut self, scalar: T) { + *self = self.scale(T::one() / scalar); + } +} |
