1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
use std::error::Error; use crate::matrix::*; use crate::number::Number; pub mod evd; pub mod ev; #[derive(Clone, Debug, Default, Hash)] pub struct SymmetricTridiagonalMatrix<T = f64> where T: Number, { d: Vec<T>, e: Vec<T>, } impl<T> SymmetricTridiagonalMatrix<T> where T: Number, { pub fn new(d: Vec<T>, e: Vec<T>) -> Result<Self, Box<dyn Error>> { if d.len().max(1) - 1 != e.len() { return Err(MatrixError::DimensionMismatch.into()); } Ok(Self { d, e }) } pub fn d(&self) -> &[T] { &self.d } pub fn e(&self) -> &[T] { &self.e } pub fn elems(self) -> (Vec<T>, Vec<T>) { (self.d, self.e) } pub fn mat(&self) -> Matrix<T> { let n = self.d.len(); let mut mat = Matrix::diag(&self.d); for i in 0..n - 1 { mat[i][i + 1] = self.e[i]; mat[i + 1][i] = self.e[i]; } mat } }