nalgebra/src/linalg/determinant.rs

62 lines
2.2 KiB
Rust
Raw Normal View History

2020-03-21 19:16:46 +08:00
use simba::scalar::ComplexField;
2019-03-23 21:29:07 +08:00
use crate::base::allocator::Allocator;
use crate::base::dimension::DimMin;
use crate::base::storage::Storage;
use crate::base::{DefaultAllocator, SquareMatrix};
2019-03-23 21:29:07 +08:00
use crate::linalg::LU;
2021-04-11 17:00:38 +08:00
impl<T: ComplexField, D: DimMin<D, Output = D>, S: Storage<T, D, D>> SquareMatrix<T, D, S> {
/// Computes the matrix determinant.
///
/// If the matrix has a dimension larger than 3, an LU decomposition is used.
#[inline]
#[must_use = "This function does not mutate self. You should use the return value."]
2021-04-11 17:00:38 +08:00
pub fn determinant(&self) -> T
2020-04-06 00:49:48 +08:00
where
2021-04-11 17:00:38 +08:00
DefaultAllocator: Allocator<T, D, D> + Allocator<(usize, usize), D>,
2020-04-06 00:49:48 +08:00
{
2018-02-02 19:26:35 +08:00
assert!(
self.is_square(),
"Unable to compute the determinant of a non-square matrix."
);
let dim = self.shape().0;
unsafe {
match dim {
2021-04-11 17:00:38 +08:00
0 => T::one(),
2018-12-03 04:00:08 +08:00
1 => *self.get_unchecked((0, 0)),
2 => {
2018-12-03 04:00:08 +08:00
let m11 = *self.get_unchecked((0, 0));
let m12 = *self.get_unchecked((0, 1));
let m21 = *self.get_unchecked((1, 0));
let m22 = *self.get_unchecked((1, 1));
m11 * m22 - m21 * m12
2018-02-02 19:26:35 +08:00
}
3 => {
2018-12-03 04:00:08 +08:00
let m11 = *self.get_unchecked((0, 0));
let m12 = *self.get_unchecked((0, 1));
let m13 = *self.get_unchecked((0, 2));
2018-12-03 04:00:08 +08:00
let m21 = *self.get_unchecked((1, 0));
let m22 = *self.get_unchecked((1, 1));
let m23 = *self.get_unchecked((1, 2));
2018-12-03 04:00:08 +08:00
let m31 = *self.get_unchecked((2, 0));
let m32 = *self.get_unchecked((2, 1));
let m33 = *self.get_unchecked((2, 2));
let minor_m12_m23 = m22 * m33 - m32 * m23;
let minor_m11_m23 = m21 * m33 - m31 * m23;
let minor_m11_m22 = m21 * m32 - m31 * m22;
m11 * minor_m12_m23 - m12 * minor_m11_m23 + m13 * minor_m11_m22
}
2018-02-02 19:26:35 +08:00
_ => LU::new(self.clone_owned()).determinant(),
}
}
}
}