Mercurial > octave
diff liboctave/numeric/chol.h @ 21269:3c8a3d35661a
better use of templates for Cholesky factorization
* liboctave/numeric/chol.h, liboctave/numeric/chol.cc: New files
generated from CmplxCHOL.cc, fCmplxCHOL.cc, floatCHOL.cc, CmplxCHOL.h,
dbleCHOL.cc, dbleCHOL.h, fCmplxCHOL.h, and floatCHOL.h and converted
to templates.
* liboctave/numeric/module.mk: Update.
* __qp__.cc, chol.cc, CMatrix.cc, CMatrix.h, dMatrix.cc, dMatrix.h,
fCMatrix.cc, fCMatrix.h, fMatrix.cc, fMatrix.h, eigs-base.cc,
mx-defs.h, mx-ext.h: Use new classes.
author | John W. Eaton <jwe@octave.org> |
---|---|
date | Tue, 16 Feb 2016 02:47:29 -0500 |
parents | liboctave/numeric/dbleCHOL.h@1473547f50f5 |
children | 6ca3acf5fad8 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/liboctave/numeric/chol.h Tue Feb 16 02:47:29 2016 -0500 @@ -0,0 +1,101 @@ +/* + +Copyright (C) 1994-2015 John W. Eaton +Copyright (C) 2008-2009 Jaroslav Hajek + +This file is part of Octave. + +Octave is free software; you can redistribute it and/or modify it +under the terms of the GNU General Public License as published by the +Free Software Foundation; either version 3 of the License, or (at your +option) any later version. + +Octave is distributed in the hope that it will be useful, but WITHOUT +ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or +FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License +for more details. + +You should have received a copy of the GNU General Public License +along with Octave; see the file COPYING. If not, see +<http://www.gnu.org/licenses/>. + +*/ + +#if ! defined (octave_chol_h) +#define octave_chol_h 1 + +#include "octave-config.h" + +template <typename T> +class +chol +{ +public: + + typedef typename T::column_vector_type VT; + typedef typename T::real_elt_type COND_T; + + chol (void) : chol_mat (), xrcond (0) { } + + chol (const T& a, bool upper = true, bool calc_cond = false) + : chol_mat (), xrcond (0) + { + init (a, upper, calc_cond); + } + + chol (const T& a, octave_idx_type& info, bool upper = true, + bool calc_cond = false) + : chol_mat (), xrcond (0) + { + info = init (a, upper, calc_cond); + } + + chol (const chol& a) + : chol_mat (a.chol_mat), xrcond (a.xrcond) { } + + chol& operator = (const chol& a) + { + if (this != &a) + { + chol_mat = a.chol_mat; + xrcond = a.xrcond; + } + + return *this; + } + + T chol_matrix (void) const { return chol_mat; } + + COND_T rcond (void) const { return xrcond; } + + // Compute the inverse of a matrix using the Cholesky factorization. + T inverse (void) const; + + void set (const T& R); + + void update (const VT& u); + + octave_idx_type downdate (const VT& u); + + octave_idx_type insert_sym (const VT& u, octave_idx_type j); + + void delete_sym (octave_idx_type j); + + void shift_sym (octave_idx_type i, octave_idx_type j); + +private: + + T chol_mat; + + COND_T xrcond; + + bool is_upper; + + octave_idx_type init (const T& a, bool upper, bool calc_cond); +}; + +template <typename T> +T +chol2inv (const T& r); + +#endif