use crate::matrix::Matrix;
use crate::matrix::MatrixError;
use crate::number::c64;
use lapack::{dgetrf, zgetrf};
use std::error::Error;
impl Matrix {
pub fn getrf(self) -> Result<(Matrix, Vec<i32>), Box<dyn Error>> {
let m = self.rows;
let n = self.cols;
let mut ipiv = vec![0; m.min(n)];
let mut info = 0;
let mut slf = self;
let m = m as i32;
let n = n as i32;
unsafe {
dgetrf(n, m, &mut slf.elems, n, &mut ipiv, &mut info);
}
match info {
0 => Ok((slf, ipiv)),
_ => Err(MatrixError::LapackRoutineError {
routine: "dgetrf".to_owned(),
info,
}
.into()),
}
}
}
impl Matrix<c64> {
pub fn getrf(self) -> Result<(Matrix<c64>, Vec<i32>), Box<dyn Error>> {
let m = self.rows;
let n = self.cols;
let mut ipiv = vec![0; m.min(n)];
let mut info = 0;
let mut slf = self;
let m = m as i32;
let n = n as i32;
unsafe {
zgetrf(n, m, &mut slf.elems, n, &mut ipiv, &mut info);
}
match info {
0 => Ok((slf, ipiv)),
_ => Err(MatrixError::LapackRoutineError {
routine: "zgetrf".to_owned(),
info,
}
.into()),
}
}
}