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 54
use std::error::Error; use crate::{matrix::*, number::Number}; #[derive(Clone, Debug, Default, Hash)] pub struct BidiagonalMatrix<T = f64> where T: Number, { d: Vec<T>, e: Vec<T>, } impl<T> BidiagonalMatrix<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, upper: bool) -> Matrix<T> { let n = self.d.len(); let mut mat = Matrix::diag(&self.d); if upper { for i in 0..n - 1 { mat[i + 1][i] = self.e[i]; } } else { for i in 0..n - 1 { mat[i][i + 1] = self.e[i]; } } mat } }