forked from M-Labs/nalgebra
finally got the correct type for insert column
This commit is contained in:
parent
2f18aee212
commit
e583e37d1c
@ -6,8 +6,9 @@ use alga::general::ComplexField;
|
|||||||
use crate::allocator::Allocator;
|
use crate::allocator::Allocator;
|
||||||
use crate::base::{DefaultAllocator, Matrix, MatrixMN, MatrixN, SquareMatrix};
|
use crate::base::{DefaultAllocator, Matrix, MatrixMN, MatrixN, SquareMatrix};
|
||||||
use crate::constraint::{SameNumberOfRows, ShapeConstraint};
|
use crate::constraint::{SameNumberOfRows, ShapeConstraint};
|
||||||
use crate::dimension::{Dim, DimSub, Dynamic, U1};
|
use crate::dimension::{Dim, DimAdd, DimSum, DimSub, Dynamic, U1};
|
||||||
use crate::storage::{Storage, StorageMut};
|
use crate::storage::{Storage, StorageMut};
|
||||||
|
use crate::base::allocator::Reallocator;
|
||||||
|
|
||||||
/// The Cholesky decomposition of a symmetric-definite-positive matrix.
|
/// The Cholesky decomposition of a symmetric-definite-positive matrix.
|
||||||
#[cfg_attr(feature = "serde-serialize", derive(Serialize, Deserialize))]
|
#[cfg_attr(feature = "serde-serialize", derive(Serialize, Deserialize))]
|
||||||
@ -188,6 +189,34 @@ where
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Updates the decomposition such that we get the decomposition of a matrix with the given column `c` in the `j`th position.
|
||||||
|
/// Since the matrix is square, an identical row will be added in the `j`th row.
|
||||||
|
pub fn insert_column<R2: Dim, S2>(
|
||||||
|
self,
|
||||||
|
j: usize,
|
||||||
|
c: &Matrix<N, R2, U1, S2>,
|
||||||
|
) -> Cholesky<N, DimSum<D, U1>>
|
||||||
|
where
|
||||||
|
D: DimAdd<U1>,
|
||||||
|
DefaultAllocator: Reallocator<N, D, D, D, DimSum<D, U1>> + Reallocator<N, D, DimSum<D, U1>, DimSum<D, U1>, DimSum<D, U1>>,
|
||||||
|
S2: Storage<N, R2, U1>,
|
||||||
|
ShapeConstraint: SameNumberOfRows<R2, DimSum<D, U1>>,
|
||||||
|
{
|
||||||
|
let n = c.nrows();
|
||||||
|
assert_eq!(
|
||||||
|
n,
|
||||||
|
self.chol.nrows() + 1,
|
||||||
|
"The new column must have the size of the factored matrix plus one."
|
||||||
|
);
|
||||||
|
assert!(j < n, "j needs to be within the bound of the new matrix.");
|
||||||
|
// TODO what is the fastest way to produce the new matrix ?
|
||||||
|
let chol= self.chol.insert_column(j, N::zero()).insert_row(j, N::zero());
|
||||||
|
|
||||||
|
// TODO see https://en.wikipedia.org/wiki/Cholesky_decomposition#Updating_the_decomposition
|
||||||
|
unimplemented!();
|
||||||
|
Cholesky { chol }
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<N: ComplexField, D: DimSub<Dynamic>, S: Storage<N, D, D>> SquareMatrix<N, D, S>
|
impl<N: ComplexField, D: DimSub<Dynamic>, S: Storage<N, D, D>> SquareMatrix<N, D, S>
|
||||||
|
Loading…
Reference in New Issue
Block a user