Compare commits

...

5 commits
0.4.0 ... main

5 changed files with 97 additions and 65 deletions

View file

@ -1,6 +1,6 @@
[package] [package]
name = "matrix-basic" name = "matrix-basic"
version = "0.4.0" version = "0.5.0"
edition = "2021" edition = "2021"
authors = ["Sayantan Santra <sayantan[dot]santra689[at]gmail[dot]com"] authors = ["Sayantan Santra <sayantan[dot]santra689[at]gmail[dot]com"]
license = "GPL-3.0" license = "GPL-3.0"

View file

@ -1,10 +1,11 @@
[![crate.io badge](https://img.shields.io/crates/d/matrix-basic)](https://crates.io/crates/matrix-basic) [![crate.io badge](https://img.shields.io/crates/d/matrix-basic)](https://crates.io/crates/matrix-basic)
# `matrix-basic` # `matrix-basic`
### A Rust crate for very basic matrix operations ### A Rust crate for very basic matrix operations.
This is a crate for very basic matrix operations with any type that supports addition, substraction, This is a crate for very basic matrix operations with any type that supports addition, substraction, multiplication,
and multiplication. Additional properties might be needed for certain operations. negation, has a zero defined, and implements the Copy trait. Additional properties (e.g. division, existence of one etc.)
might be needed for certain operations.
I created it mostly to learn how to use generic types and traits. I created it mostly to learn how to use generic types and traits.

29
src/errors.rs Normal file
View file

@ -0,0 +1,29 @@
use std::{
error::Error,
fmt::{self, Display, Formatter},
};
/// Error type for using in this crate. Mostly to reduce writing
/// error description every time.
#[derive(Debug, PartialEq)]
pub enum MatrixError {
/// Provided matrix isn't square.
NotSquare,
/// provided matrix is singular.
Singular,
/// Provided array has unequal rows.
UnequalRows,
}
impl Display for MatrixError {
fn fmt(&self, f: &mut Formatter) -> fmt::Result {
let out = match *self {
Self::NotSquare => "provided matrix isn't square",
Self::Singular => "provided matrix is singular",
Self::UnequalRows => "provided array has unequal rows",
};
write!(f, "{out}")
}
}
impl Error for MatrixError {}

View file

@ -8,6 +8,7 @@
//! //!
//! Sayantan Santra (2023) //! Sayantan Santra (2023)
use errors::MatrixError;
use num::{ use num::{
traits::{One, Zero}, traits::{One, Zero},
Integer, Integer,
@ -18,6 +19,7 @@ use std::{
result::Result, result::Result,
}; };
pub mod errors;
mod tests; mod tests;
/// Trait a type must satisfy to be element of a matrix. This is /// Trait a type must satisfy to be element of a matrix. This is
@ -52,7 +54,7 @@ pub struct Matrix<T: ToMatrix> {
} }
impl<T: ToMatrix> Matrix<T> { impl<T: ToMatrix> Matrix<T> {
/// Creates a matrix from given 2D "array" in a `Vec<Vec<T>>` form. /// Creates a matrix from given 2D "array" in a [`Vec<Vec<T>>`] form.
/// It'll throw an error if all the given rows aren't of the same size. /// It'll throw an error if all the given rows aren't of the same size.
/// # Example /// # Example
/// ``` /// ```
@ -62,7 +64,7 @@ impl<T: ToMatrix> Matrix<T> {
/// will create the following matrix: /// will create the following matrix:
/// ⌈1, 2, 3⌉ /// ⌈1, 2, 3⌉
/// ⌊4, 5, 6⌋ /// ⌊4, 5, 6⌋
pub fn from(entries: Vec<Vec<T>>) -> Result<Matrix<T>, &'static str> { pub fn from(entries: Vec<Vec<T>>) -> Result<Matrix<T>, MatrixError> {
let mut equal_rows = true; let mut equal_rows = true;
let row_len = entries[0].len(); let row_len = entries[0].len();
for row in &entries { for row in &entries {
@ -74,7 +76,7 @@ impl<T: ToMatrix> Matrix<T> {
if equal_rows { if equal_rows {
Ok(Matrix { entries }) Ok(Matrix { entries })
} else { } else {
Err("Unequal rows.") Err(MatrixError::UnequalRows)
} }
} }
@ -152,7 +154,7 @@ impl<T: ToMatrix> Matrix<T> {
/// let m = Matrix::from(vec![vec![1, 2], vec![3, 4]]).unwrap(); /// let m = Matrix::from(vec![vec![1, 2], vec![3, 4]]).unwrap();
/// assert_eq!(m.det(), Ok(-2)); /// assert_eq!(m.det(), Ok(-2));
/// ``` /// ```
pub fn det(&self) -> Result<T, &'static str> { pub fn det(&self) -> Result<T, MatrixError> {
if self.is_square() { if self.is_square() {
// It's a recursive algorithm using minors. // It's a recursive algorithm using minors.
// TODO: Implement a faster algorithm. // TODO: Implement a faster algorithm.
@ -173,7 +175,7 @@ impl<T: ToMatrix> Matrix<T> {
}; };
Ok(out) Ok(out)
} else { } else {
Err("Provided matrix isn't square.") Err(MatrixError::NotSquare)
} }
} }
@ -187,7 +189,7 @@ impl<T: ToMatrix> Matrix<T> {
/// let m = Matrix::from(vec![vec![1.0, 2.0], vec![3.0, 4.0]]).unwrap(); /// let m = Matrix::from(vec![vec![1.0, 2.0], vec![3.0, 4.0]]).unwrap();
/// assert_eq!(m.det_in_field(), Ok(-2.0)); /// assert_eq!(m.det_in_field(), Ok(-2.0));
/// ``` /// ```
pub fn det_in_field(&self) -> Result<T, &'static str> pub fn det_in_field(&self) -> Result<T, MatrixError>
where where
T: One, T: One,
T: PartialEq, T: PartialEq,
@ -227,7 +229,7 @@ impl<T: ToMatrix> Matrix<T> {
} }
Ok(multiplier) Ok(multiplier)
} else { } else {
Err("Provided matrix isn't square.") Err(MatrixError::NotSquare)
} }
} }
@ -351,7 +353,7 @@ impl<T: ToMatrix> Matrix<T> {
/// let m = Matrix::from(vec![vec![1, 2], vec![3, 4]]).unwrap(); /// let m = Matrix::from(vec![vec![1, 2], vec![3, 4]]).unwrap();
/// assert_eq!(m.trace(), Ok(5)); /// assert_eq!(m.trace(), Ok(5));
/// ``` /// ```
pub fn trace(self) -> Result<T, &'static str> { pub fn trace(self) -> Result<T, MatrixError> {
if self.is_square() { if self.is_square() {
let mut out = self.entries[0][0]; let mut out = self.entries[0][0];
for i in 1..self.height() { for i in 1..self.height() {
@ -359,7 +361,7 @@ impl<T: ToMatrix> Matrix<T> {
} }
Ok(out) Ok(out)
} else { } else {
Err("Provided matrix isn't square.") Err(MatrixError::NotSquare)
} }
} }
@ -408,7 +410,7 @@ impl<T: ToMatrix> Matrix<T> {
/// let n = Matrix::from(vec![vec![-2.0, 1.0], vec![1.5, -0.5]]).unwrap(); /// let n = Matrix::from(vec![vec![-2.0, 1.0], vec![1.5, -0.5]]).unwrap();
/// assert_eq!(m.inverse(), Ok(n)); /// assert_eq!(m.inverse(), Ok(n));
/// ``` /// ```
pub fn inverse(&self) -> Result<Self, &'static str> pub fn inverse(&self) -> Result<Self, MatrixError>
where where
T: Div<Output = T>, T: Div<Output = T>,
T: One, T: One,
@ -436,7 +438,7 @@ impl<T: ToMatrix> Matrix<T> {
} }
} }
if zero_column { if zero_column {
return Err("Provided matrix is singular."); return Err(MatrixError::Singular);
} }
} }
for j in (i + 1)..h { for j in (i + 1)..h {
@ -454,7 +456,7 @@ impl<T: ToMatrix> Matrix<T> {
// Then we reduce the rows // Then we reduce the rows
for i in 0..h { for i in 0..h {
if rows[i][i] == T::zero() { if rows[i][i] == T::zero() {
return Err("Provided matrix is singular."); return Err(MatrixError::Singular);
} }
let divisor = rows[i][i]; let divisor = rows[i][i];
for entry in rows[i].iter_mut().skip(i) { for entry in rows[i].iter_mut().skip(i) {
@ -477,7 +479,7 @@ impl<T: ToMatrix> Matrix<T> {
Ok(Matrix { entries: out }) Ok(Matrix { entries: out })
} else { } else {
Err("Provided matrix isn't square.") Err(MatrixError::NotSquare)
} }
} }
@ -496,7 +498,7 @@ impl<T: Mul<Output = T> + ToMatrix> Mul for Matrix<T> {
fn mul(self, other: Self) -> Self::Output { fn mul(self, other: Self) -> Self::Output {
let width = self.width(); let width = self.width();
if width != other.height() { if width != other.height() {
panic!("Row length of first matrix must be same as column length of second matrix."); panic!("row length of first matrix != column length of second matrix");
} else { } else {
let mut out = Vec::new(); let mut out = Vec::new();
for row in self.rows() { for row in self.rows() {
@ -527,7 +529,7 @@ impl<T: Mul<Output = T> + ToMatrix> Add for Matrix<T> {
} }
Matrix { entries: out } Matrix { entries: out }
} else { } else {
panic!("Both matrices must be of same dimensions."); panic!("provided matrices have different dimensions");
} }
} }
} }
@ -551,54 +553,20 @@ impl<T: ToMatrix> Sub for Matrix<T> {
if self.height() == other.height() && self.width() == other.width() { if self.height() == other.height() && self.width() == other.width() {
self + -other self + -other
} else { } else {
panic!("Both matrices must be of same dimensions."); panic!("provided matrices have different dimensions");
} }
} }
} }
/// Trait for conversion between matrices of different types. /// Trait for conversion between matrices of different types.
/// It only has a [`matrix_into()`](Self::matrix_into()) method. /// It only has a [`matrix_from()`](Self::matrix_from()) method.
/// This is needed since negative trait bound are not supported in stable Rust /// This is needed since negative trait bound are not supported in stable Rust
/// yet, so we'll have a conflict trying to implement [`From`]. /// yet, so we'll have a conflict trying to implement [`From`].
/// I plan to change this to the default From trait as soon as some sort /// I plan to change this to the default From trait as soon as some sort
/// of specialization system is implemented. /// of specialization system is implemented.
/// You can track this issue [here](https://github.com/rust-lang/rust/issues/42721). /// You can track this issue [here](https://github.com/rust-lang/rust/issues/42721).
pub trait MatrixInto<T: ToMatrix> { pub trait MatrixFrom<T: ToMatrix> {
/// Method for converting a matrix into a matrix of type [`Matrix<T>`]. /// Method for getting a matrix of a new type from a matrix of type [`Matrix<T>`].
/// # Example
/// ```
/// use matrix_basic::Matrix;
/// use matrix_basic::MatrixInto;
///
/// let a = Matrix::from(vec![vec![1, 2, 3], vec![0, 1, 2]]).unwrap();
/// let b = Matrix::from(vec![vec![1.0, 2.0, 3.0], vec![0.0, 1.0, 2.0]]).unwrap();
/// let c: Matrix<f64> = a.matrix_into(); // Type annotation is needed here
///
/// assert_eq!(c, b);
/// ```
fn matrix_into(self) -> Matrix<T>;
}
/// Blanket implementation of [`MatrixInto`] for converting [`Matrix<S>`] to [`Matrix<T>`] whenever
/// `S` implements [`Into(T)`]. Look at [`matrix_into`](Self::matrix_into()).
impl<T: ToMatrix, S: ToMatrix + Into<T>> MatrixInto<T> for Matrix<S> {
fn matrix_into(self) -> Matrix<T> {
let mut out = Vec::new();
for row in self.entries {
let mut new_row: Vec<T> = Vec::new();
for entry in row {
new_row.push(entry.into());
}
out.push(new_row)
}
Matrix { entries: out }
}
}
/// Sister trait of [`MatrixInto`]. Basically does the same thing, just with a
/// different syntax.
pub trait MatrixFrom<T> {
/// Method for getting a matrix from another matrix of type [`Matrix<T>`].
/// # Example /// # Example
/// ``` /// ```
/// use matrix_basic::Matrix; /// use matrix_basic::Matrix;
@ -610,14 +578,48 @@ pub trait MatrixFrom<T> {
/// ///
/// assert_eq!(c, b); /// assert_eq!(c, b);
/// ``` /// ```
fn matrix_from(input: T) -> Self; fn matrix_from(input: Matrix<T>) -> Self;
} }
/// Blanket implementation of [`MatrixFrom`] for [`Matrix<S>`] whenever `T` (which is actually some)[`Matrix<U>`] implements /// Blanket implementation of [`MatrixFrom<T>`] for converting [`Matrix<S>`] to [`Matrix<T>`] whenever
/// [`MatrixInto<S>`]. /// `S` implements [`From(T)`]. Look at [`matrix_into`](Self::matrix_into()).
impl<T: MatrixInto<S>, S: ToMatrix> MatrixFrom<T> for Matrix<S> { impl<T: ToMatrix, S: ToMatrix + From<T>> MatrixFrom<T> for Matrix<S> {
fn matrix_from(input: T) -> Self { fn matrix_from(input: Matrix<T>) -> Self {
let out: Matrix<S> = input.matrix_into(); let mut out = Vec::new();
out for row in input.entries {
let mut new_row: Vec<S> = Vec::new();
for entry in row {
new_row.push(entry.into());
}
out.push(new_row)
}
Matrix { entries: out }
}
}
/// Sister trait of [`MatrixFrom`]. Basically does the same thing, just with a
/// different syntax.
pub trait MatrixInto<T> {
/// Method for converting a matrix [`Matrix<T>`] to another type.
/// # Example
/// ```
/// use matrix_basic::Matrix;
/// use matrix_basic::MatrixInto;
///
/// let a = Matrix::from(vec![vec![1, 2, 3], vec![0, 1, 2]]).unwrap();
/// let b = Matrix::from(vec![vec![1.0, 2.0, 3.0], vec![0.0, 1.0, 2.0]]).unwrap();
/// let c: Matrix<f64> = a.matrix_into(); // Type annotation is needed here
///
///
/// assert_eq!(c, b);
/// ```
fn matrix_into(self) -> T;
}
/// Blanket implementation of [`MatrixInto<T>`] for [`Matrix<S>`] whenever `T`
/// (which is actually some)[`Matrix<U>`] implements [`MatrixFrom<S>`].
impl<T: MatrixFrom<S>, S: ToMatrix> MatrixInto<T> for Matrix<S> {
fn matrix_into(self) -> T {
T::matrix_from(self)
} }
} }

View file

@ -72,7 +72,7 @@ fn conversion_test() {
let b = Matrix::from(vec![vec![1.0, 2.0, 3.0], vec![0.0, 1.0, 2.0]]).unwrap(); let b = Matrix::from(vec![vec![1.0, 2.0, 3.0], vec![0.0, 1.0, 2.0]]).unwrap();
use crate::MatrixInto; use crate::MatrixInto;
assert_eq!(a.clone().matrix_into(), b); assert_eq!(b, a.clone().matrix_into());
use crate::MatrixFrom; use crate::MatrixFrom;
let c = Matrix::<f64>::matrix_from(a); let c = Matrix::<f64>::matrix_from(a);