M-Labs/nalgebra
3
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

UDU: Expand to Dim from DimName

Browse Source 
Signed-off-by: Christopher Rabotin <christopher.rabotin@gmail.com>
This commit is contained in:
Christopher Rabotin 2020-09-26 18:34:35 -06:00 committed by Crozet Sébastien
parent 5a7ed61e9b
commit e9933e5c91
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/linalg/udu.rs
View File

@ -3,13 +3,13 @@ use serde::{Deserialize, Serialize};
use crate::allocator::Allocator;
use crate::base::{DefaultAllocator, MatrixN};
use crate::dimension::DimName;
use crate::dimension::Dim;
use simba::scalar::ComplexField;
/// UDU factorization
#[cfg_attr(feature = "serde-serialize", derive(Serialize, Deserialize))]
#[derive(Clone, Debug)]
pub struct UDU<N: ComplexField, D: DimName>
pub struct UDU<N: ComplexField, D: Dim>
where
DefaultAllocator: Allocator<N, D, D>,
{
@ -19,14 +19,14 @@ where
pub d: MatrixN<N, D>,
}
impl<N: ComplexField, D: DimName> Copy for UDU<N, D>
impl<N: ComplexField, D: Dim> Copy for UDU<N, D>
where
DefaultAllocator: Allocator<N, D, D>,
MatrixN<N, D>: Copy,
{
}
impl<N: ComplexField, D: DimName> UDU<N, D>
impl<N: ComplexField, D: Dim> UDU<N, D>
where
DefaultAllocator: Allocator<N, D, D>,
{
@ -34,10 +34,11 @@ where
/// NOTE: The provided matrix MUST be symmetric, and no verification is done in this regard.
/// Ref.: "Optimal control and estimation-Dover Publications", Robert F. Stengel, (1994) page 360
pub fn new(p: MatrixN<N, D>) -> Self {
let mut d = MatrixN::<N, D>::zeros();
let mut u = MatrixN::<N, D>::zeros();
let n = p.ncols();
let n_as_dim = D::from_usize(n);
let mut d = MatrixN::<N, D>::zeros_generic(n_as_dim, n_as_dim);
let mut u = MatrixN::<N, D>::zeros_generic(n_as_dim, n_as_dim);
d[(n - 1, n - 1)] = p[(n - 1, n - 1)];
u[(n - 1, n - 1)] = N::one();