Mercurial > hg > octave-lyh
diff liboctave/base-lu.cc @ 1991:413f6a81868f
[project @ 1996-03-03 00:40:53 by jwe]
Initial revision
author | jwe |
---|---|
date | Sun, 03 Mar 1996 00:40:53 +0000 |
parents | |
children | 1b57120c997b |
line wrap: on
line diff
new file mode 100644 --- /dev/null +++ b/liboctave/base-lu.cc @@ -0,0 +1,105 @@ +// -*- C++ -*- +/* + +Copyright (C) 1996 John W. Eaton + +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 2, 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, write to the Free +Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + +*/ + +#if defined (__GNUG__) +#pragma implementation +#endif + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#include "base-lu.h" + +template <class lu_type, class p_type> +lu_type +base_lu <lu_type, p_type> :: L (void) const +{ + int n = ipvt.length (); + + lu_type l (n, n, 0.0); + + for (int i = 0; i < n; i++) + { + l.xelem (i, i) = 1.0; + for (int j = 0; j < i; j++) + l.xelem (i, j) = a_fact.xelem (i, j); + } + + return l; +} + +template <class lu_type, class p_type> +lu_type +base_lu <lu_type, p_type> :: U (void) const +{ + int n = ipvt.length (); + + lu_type u (n, n, 0.0); + + for (int i = 0; i < n; i++) + { + for (int j = i; j < n; j++) + u.xelem (i, j) = a_fact.xelem (i, j); + } + + return u; +} + +template <class lu_type, class p_type> +p_type +base_lu <lu_type, p_type> :: P (void) const +{ + int n = ipvt.length (); + + Array<int> pvt (n); + + for (int i = 0; i < n; i++) + pvt.xelem (i) = i; + + for (int i = 0; i < n - 1; i++) + { + int k = ipvt.xelem (i); + + if (k != i) + { + int tmp = pvt.xelem (k); + pvt.xelem (k) = pvt.xelem (i); + pvt.xelem (i) = tmp; + } + } + + p_type p (n, n, 0.0); + + for (int i = 0; i < n; i++) + p.xelem (i, pvt.xelem (i)) = 1.0; + + return p; +} + +/* +;;; Local Variables: *** +;;; mode: C++ *** +;;; page-delimiter: "^/\\*" *** +;;; End: *** +*/