octave-nkf: liboctave/fEIG.cc annotate

annotate liboctave/fEIG.cc @ 14557:e8e86ae3abbc

make diag (x, m, n) return a proper diagonal matrix object (bug #36099) * Array.h, Array.cc (Array<T>::diag (octave_idx_type, octave_idx_type) const): New function. * CMatrix.h, CMatrix.cc (ComplexMatrix::diag (octave_idx_type, octave_idx_type) const): New function. * dMatrix.h, dMatrix.cc (Matrix::diag (octave_idx_type, octave_idx_type) const): New function. * fCMatrix.h, fCMatrix.cc (FloatComplexMatrix::diag (octave_idx_type, octave_idx_type) const): New function. * fMatrix.h, fMatrix.cc (FloatMatrix::diag (octave_idx_type, octave_idx_type) const): New function. * CNDArray.cc, CNDArray.h (ComplexNDArray::diag (octave_idx_type, octave_idx_type) const): New forwarding function. * boolNDArray.cc, boolNDArray.h (boolNDArray::diag (octave_idx_type, octave_idx_type) const): New forwarding function. * chNDArray.cc, chNDArray.h (charNDArray::diag (octave_idx_type, octave_idx_type) const): New forwarding function. * dNDArray.cc, dNDArray.h (NDArray::diag (octave_idx_type, octave_idx_type) const): New forwarding function. * fCNDArray.cc, fCNDArray.h (FloatComplexNDArray::diag (octave_idx_type, octave_idx_type) const): New forwarding function. * fNDArray.cc, fNDArray.h (FloatNDArray::diag (octave_idx_type, octave_idx_type) const): New forwarding function. * intNDArray.cc, intNDArray.h (intNDArray<T>::diag (octave_idx_type, octave_idx_type) const): New forwarding function. * Cell.cc, Cell.h (Cell::diag (octave_idx_type, octave_idx_type) const): New function. * ov.h (octave_value::diag (octave_idx_type, octave_idx_type)): New function. * ov-base.h, ov-base.cc (octave_base_value::diag (octave_idx_type, octave_idx_type) const): New virtual function and default implementation. * ov-base-mat.h (octave_base_matrix<T>::diag (octave_idx_type, octave_idx_type) const): New function. * ov-base-scalar.cc, ov-base-scalar.h (octave_base_scalar<T>::diag (octave_idx_type, octave_idx_type)): New function. * ov-complex.cc, ov-complex.h (octave_complex::diag (octave_idx_type, octave_idx_type) const): New function. * ov-cx-mat.cc, ov-complex.h (octave_complex_matrix::diag (octave_idx_type, octave_idx_type) const): New function. * ov-float.cc, ov-float.h (octave_float_scalar::diag (octave_idx_type, octave_idx_type) const): New function. * ov-flt-complex.cc, ov-flt-complex.h (octave_float_complex::diag (octave_idx_type, octave_idx_type) const): New function. * ov-flt-cx-mat.cc, ov-flt-cx-mat.h (octave_float_complex_matrix::diag (octave_idx_type, octave_idx_type) const): New function. * ov-flt-re-mat.cc, ov-flt-re-mat.h (octave_float_matrix::diag (octave_idx_type, octave_idx_type) const): New function. * ov-range.cc, ov-range.h (octave_range::diag (octave_idx_type, octave_idx_type) const): New function. * ov-re-mat.cc, ov-re-mat.h (octave_matrix::diag (octave_idx_type, octave_idx_type) const): New function. * ov-scalar.cc, ov-scalar.h (octave_scalar::diag (octave_idx_type, octave_idx_type) const): New function. * data.cc (Fdiag): Use two-arg octave_value::diag method for dispatching. New tests.

author	John W. Eaton <jwe@octave.org>
date	Thu, 12 Apr 2012 16:27:39 -0400
parents	72c96de7a403
children	460a3c6d8bf1

rev	line source
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	1 /*
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	2
14138 72c96de7a403 maint: update copyright notices for 2012 John W. Eaton <jwe@octave.org> parents: 11586 diff changeset	3 Copyright (C) 1994-2012 John W. Eaton
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	4
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	5 This file is part of Octave.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	6
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	7 Octave is free software; you can redistribute it and/or modify it
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	8 under the terms of the GNU General Public License as published by the
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	9 Free Software Foundation; either version 3 of the License, or (at your
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	10 option) any later version.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	11
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	12 Octave is distributed in the hope that it will be useful, but WITHOUT
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	13 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	15 for more details.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	16
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	17 You should have received a copy of the GNU General Public License
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	18 along with Octave; see the file COPYING. If not, see
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	19 <http://www.gnu.org/licenses/>.
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	20
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	21 */
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	22
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	23 #ifdef HAVE_CONFIG_H
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	24 #include <config.h>
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	25 #endif
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	26
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	27 #include "fEIG.h"
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	28 #include "fColVector.h"
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	29 #include "f77-fcn.h"
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	30 #include "lo-error.h"
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	31
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	32 extern "C"
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	33 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	34 F77_RET_T
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	35 F77_FUNC (sgeev, SGEEV) (F77_CONST_CHAR_ARG_DECL,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	36 F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	37 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	38 const octave_idx_type&, float, float, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	39 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	40 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	41 const octave_idx_type&, octave_idx_type&
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	42 F77_CHAR_ARG_LEN_DECL
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	43 F77_CHAR_ARG_LEN_DECL);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	44
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	45 F77_RET_T
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	46 F77_FUNC (cgeev, CGEEV) (F77_CONST_CHAR_ARG_DECL,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	47 F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	48 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	49 const octave_idx_type&, FloatComplex, FloatComplex,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	50 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	51 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	52 const octave_idx_type&, float*, octave_idx_type&
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	53 F77_CHAR_ARG_LEN_DECL
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	54 F77_CHAR_ARG_LEN_DECL);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	55
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	56 F77_RET_T
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	57 F77_FUNC (ssyev, SSYEV) (F77_CONST_CHAR_ARG_DECL,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	58 F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	59 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	60 const octave_idx_type&, float, float,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	61 const octave_idx_type&, octave_idx_type&
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	62 F77_CHAR_ARG_LEN_DECL
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	63 F77_CHAR_ARG_LEN_DECL);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	64
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	65 F77_RET_T
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	66 F77_FUNC (cheev, CHEEV) (F77_CONST_CHAR_ARG_DECL,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	67 F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	68 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	69 const octave_idx_type&, float, FloatComplex,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	70 const octave_idx_type&, float*, octave_idx_type&
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	71 F77_CHAR_ARG_LEN_DECL
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	72 F77_CHAR_ARG_LEN_DECL);
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	73
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	74 F77_RET_T
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	75 F77_FUNC (spotrf, SPOTRF) (F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	76 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	77 const octave_idx_type&, octave_idx_type&
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	78 F77_CHAR_ARG_LEN_DECL
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	79 F77_CHAR_ARG_LEN_DECL);
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	80
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	81 F77_RET_T
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	82 F77_FUNC (cpotrf, CPOTRF) (F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	83 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	84 const octave_idx_type&, octave_idx_type&
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	85 F77_CHAR_ARG_LEN_DECL
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	86 F77_CHAR_ARG_LEN_DECL);
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	87
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	88 F77_RET_T
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	89 F77_FUNC (sggev, SGGEV) (F77_CONST_CHAR_ARG_DECL,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	90 F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	91 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	92 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	93 const octave_idx_type&, float, float, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	94 float, const octave_idx_type&, float,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	95 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	96 const octave_idx_type&, octave_idx_type&
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	97 F77_CHAR_ARG_LEN_DECL
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	98 F77_CHAR_ARG_LEN_DECL);
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	99
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	100 F77_RET_T
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	101 F77_FUNC (ssygv, SSYGV) (const octave_idx_type&,
11518 141b3fb5cef7 style fixes John W. Eaton <jwe@octave.org> parents: 11495 diff changeset	102 F77_CONST_CHAR_ARG_DECL,
141b3fb5cef7 style fixes John W. Eaton <jwe@octave.org> parents: 11495 diff changeset	103 F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	104 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	105 const octave_idx_type&, float*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	106 const octave_idx_type&, float, float,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	107 const octave_idx_type&, octave_idx_type&
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	108 F77_CHAR_ARG_LEN_DECL
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	109 F77_CHAR_ARG_LEN_DECL);
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	110
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	111 F77_RET_T
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	112 F77_FUNC (cggev, CGGEV) (F77_CONST_CHAR_ARG_DECL,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	113 F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	114 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	115 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	116 const octave_idx_type&, FloatComplex*,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	117 FloatComplex, FloatComplex,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	118 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	119 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	120 const octave_idx_type&, float*, octave_idx_type&
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	121 F77_CHAR_ARG_LEN_DECL
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	122 F77_CHAR_ARG_LEN_DECL);
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	123
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	124 F77_RET_T
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	125 F77_FUNC (chegv, CHEGV) (const octave_idx_type&,
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	126 F77_CONST_CHAR_ARG_DECL,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	127 F77_CONST_CHAR_ARG_DECL,
11495 8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	128 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	129 const octave_idx_type&, FloatComplex*,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	130 const octave_idx_type&, float, FloatComplex,
8a5e980da6aa style fixes John W. Eaton <jwe@octave.org> parents: 10350 diff changeset	131 const octave_idx_type&, float*, octave_idx_type&
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	132 F77_CHAR_ARG_LEN_DECL
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	133 F77_CHAR_ARG_LEN_DECL);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	134 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	135
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	136 octave_idx_type
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	137 FloatEIG::init (const FloatMatrix& a, bool calc_ev)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	138 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	139 if (a.any_element_is_inf_or_nan ())
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	140 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	141 (*current_liboctave_error_handler)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	142 ("EIG: matrix contains Inf or NaN values");
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	143 return -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	144 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	145
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	146 if (a.is_symmetric ())
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	147 return symmetric_init (a, calc_ev);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	148
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	149 octave_idx_type n = a.rows ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	150
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	151 if (n != a.cols ())
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	152 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	153 (*current_liboctave_error_handler) ("EIG requires square matrix");
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	154 return -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	155 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	156
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	157 octave_idx_type info = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	158
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	159 FloatMatrix atmp = a;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	160 float *tmp_data = atmp.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	161
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	162 Array<float> wr (dim_vector (n, 1));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	163 float *pwr = wr.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	164
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	165 Array<float> wi (dim_vector (n, 1));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	166 float *pwi = wi.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	167
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	168 volatile octave_idx_type nvr = calc_ev ? n : 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	169 FloatMatrix vr (nvr, nvr);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	170 float *pvr = vr.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	171
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	172 octave_idx_type lwork = -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	173 float dummy_work;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	174
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	175 float *dummy = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	176 octave_idx_type idummy = 1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	177
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	178 F77_XFCN (sgeev, SGEEV, (F77_CONST_CHAR_ARG2 ("N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	179 F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	180 n, tmp_data, n, pwr, pwi, dummy,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	181 idummy, pvr, n, &dummy_work, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	182 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	183 F77_CHAR_ARG_LEN (1)));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	184
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	185 if (info == 0)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	186 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	187 lwork = static_cast<octave_idx_type> (dummy_work);
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	188 Array<float> work (dim_vector (lwork, 1));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	189 float *pwork = work.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	190
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	191 F77_XFCN (sgeev, SGEEV, (F77_CONST_CHAR_ARG2 ("N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	192 F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	193 n, tmp_data, n, pwr, pwi, dummy,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	194 idummy, pvr, n, pwork, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	195 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	196 F77_CHAR_ARG_LEN (1)));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	197
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	198 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	199 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	200 (*current_liboctave_error_handler) ("unrecoverable error in sgeev");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	201 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	202 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	203
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	204 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	205 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	206 (*current_liboctave_error_handler) ("sgeev failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	207 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	208 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	209
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	210 lambda.resize (n);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	211 v.resize (nvr, nvr);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	212
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	213 for (octave_idx_type j = 0; j < n; j++)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	214 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	215 if (wi.elem (j) == 0.0)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	216 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	217 lambda.elem (j) = FloatComplex (wr.elem (j));
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	218 for (octave_idx_type i = 0; i < nvr; i++)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	219 v.elem (i, j) = vr.elem (i, j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	220 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	221 else
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	222 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	223 if (j+1 >= n)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	224 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	225 (*current_liboctave_error_handler) ("EIG: internal error");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	226 return -1;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	227 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	228
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	229 lambda.elem(j) = FloatComplex (wr.elem(j), wi.elem(j));
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	230 lambda.elem(j+1) = FloatComplex (wr.elem(j+1), wi.elem(j+1));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	231
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	232 for (octave_idx_type i = 0; i < nvr; i++)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	233 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	234 float real_part = vr.elem (i, j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	235 float imag_part = vr.elem (i, j+1);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	236 v.elem (i, j) = FloatComplex (real_part, imag_part);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	237 v.elem (i, j+1) = FloatComplex (real_part, -imag_part);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	238 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	239 j++;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	240 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	241 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	242 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	243 else
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	244 (*current_liboctave_error_handler) ("sgeev workspace query failed");
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	245
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	246 return info;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	247 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	248
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	249 octave_idx_type
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	250 FloatEIG::symmetric_init (const FloatMatrix& a, bool calc_ev)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	251 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	252 octave_idx_type n = a.rows ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	253
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	254 if (n != a.cols ())
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	255 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	256 (*current_liboctave_error_handler) ("EIG requires square matrix");
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	257 return -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	258 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	259
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	260 octave_idx_type info = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	261
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	262 FloatMatrix atmp = a;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	263 float *tmp_data = atmp.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	264
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	265 FloatColumnVector wr (n);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	266 float *pwr = wr.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	267
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	268 octave_idx_type lwork = -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	269 float dummy_work;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	270
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	271 F77_XFCN (ssyev, SSYEV, (F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	272 F77_CONST_CHAR_ARG2 ("U", 1),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	273 n, tmp_data, n, pwr, &dummy_work, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	274 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	275 F77_CHAR_ARG_LEN (1)));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	276
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	277 if (info == 0)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	278 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	279 lwork = static_cast<octave_idx_type> (dummy_work);
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	280 Array<float> work (dim_vector (lwork, 1));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	281 float *pwork = work.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	282
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	283 F77_XFCN (ssyev, SSYEV, (F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	284 F77_CONST_CHAR_ARG2 ("U", 1),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	285 n, tmp_data, n, pwr, pwork, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	286 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	287 F77_CHAR_ARG_LEN (1)));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	288
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	289 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	290 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	291 (*current_liboctave_error_handler) ("unrecoverable error in ssyev");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	292 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	293 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	294
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	295 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	296 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	297 (*current_liboctave_error_handler) ("ssyev failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	298 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	299 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	300
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	301 lambda = FloatComplexColumnVector (wr);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	302 v = calc_ev ? FloatComplexMatrix (atmp) : FloatComplexMatrix ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	303 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	304 else
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	305 (*current_liboctave_error_handler) ("ssyev workspace query failed");
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	306
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	307 return info;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	308 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	309
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	310 octave_idx_type
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	311 FloatEIG::init (const FloatComplexMatrix& a, bool calc_ev)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	312 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	313 if (a.any_element_is_inf_or_nan ())
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	314 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	315 (*current_liboctave_error_handler)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	316 ("EIG: matrix contains Inf or NaN values");
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	317 return -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	318 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	319
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	320 if (a.is_hermitian ())
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	321 return hermitian_init (a, calc_ev);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	322
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	323 octave_idx_type n = a.rows ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	324
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	325 if (n != a.cols ())
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	326 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	327 (*current_liboctave_error_handler) ("EIG requires square matrix");
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	328 return -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	329 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	330
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	331 octave_idx_type info = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	332
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	333 FloatComplexMatrix atmp = a;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	334 FloatComplex *tmp_data = atmp.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	335
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	336 FloatComplexColumnVector w (n);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	337 FloatComplex *pw = w.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	338
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	339 octave_idx_type nvr = calc_ev ? n : 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	340 FloatComplexMatrix vtmp (nvr, nvr);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	341 FloatComplex *pv = vtmp.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	342
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	343 octave_idx_type lwork = -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	344 FloatComplex dummy_work;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	345
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	346 octave_idx_type lrwork = 2*n;
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	347 Array<float> rwork (dim_vector (lrwork, 1));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	348 float *prwork = rwork.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	349
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	350 FloatComplex *dummy = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	351 octave_idx_type idummy = 1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	352
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	353 F77_XFCN (cgeev, CGEEV, (F77_CONST_CHAR_ARG2 ("N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	354 F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	355 n, tmp_data, n, pw, dummy, idummy,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	356 pv, n, &dummy_work, lwork, prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	357 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	358 F77_CHAR_ARG_LEN (1)));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	359
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	360 if (info == 0)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	361 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	362 lwork = static_cast<octave_idx_type> (dummy_work.real ());
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	363 Array<FloatComplex> work (dim_vector (lwork, 1));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	364 FloatComplex *pwork = work.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	365
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	366 F77_XFCN (cgeev, CGEEV, (F77_CONST_CHAR_ARG2 ("N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	367 F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	368 n, tmp_data, n, pw, dummy, idummy,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	369 pv, n, pwork, lwork, prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	370 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	371 F77_CHAR_ARG_LEN (1)));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	372
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	373 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	374 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	375 (*current_liboctave_error_handler) ("unrecoverable error in cgeev");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	376 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	377 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	378
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	379 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	380 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	381 (*current_liboctave_error_handler) ("cgeev failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	382 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	383 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	384
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	385 lambda = w;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	386 v = vtmp;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	387 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	388 else
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	389 (*current_liboctave_error_handler) ("cgeev workspace query failed");
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	390
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	391 return info;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	392 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	393
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	394 octave_idx_type
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	395 FloatEIG::hermitian_init (const FloatComplexMatrix& a, bool calc_ev)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	396 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	397 octave_idx_type n = a.rows ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	398
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	399 if (n != a.cols ())
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	400 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	401 (*current_liboctave_error_handler) ("EIG requires square matrix");
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	402 return -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	403 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	404
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	405 octave_idx_type info = 0;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	406
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	407 FloatComplexMatrix atmp = a;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	408 FloatComplex *tmp_data = atmp.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	409
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	410 FloatColumnVector wr (n);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	411 float *pwr = wr.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	412
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	413 octave_idx_type lwork = -1;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	414 FloatComplex dummy_work;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	415
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	416 octave_idx_type lrwork = 3*n;
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	417 Array<float> rwork (dim_vector (lrwork, 1));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	418 float *prwork = rwork.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	419
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	420 F77_XFCN (cheev, CHEEV, (F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	421 F77_CONST_CHAR_ARG2 ("U", 1),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	422 n, tmp_data, n, pwr, &dummy_work, lwork,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	423 prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	424 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	425 F77_CHAR_ARG_LEN (1)));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	426
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	427 if (info == 0)
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	428 {
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	429 lwork = static_cast<octave_idx_type> (dummy_work.real ());
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	430 Array<FloatComplex> work (dim_vector (lwork, 1));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	431 FloatComplex *pwork = work.fortran_vec ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	432
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	433 F77_XFCN (cheev, CHEEV, (F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	434 F77_CONST_CHAR_ARG2 ("U", 1),
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	435 n, tmp_data, n, pwr, pwork, lwork, prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	436 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	437 F77_CHAR_ARG_LEN (1)));
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	438
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	439 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	440 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	441 (*current_liboctave_error_handler) ("unrecoverable error in cheev");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	442 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	443 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	444
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	445 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	446 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	447 (*current_liboctave_error_handler) ("cheev failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	448 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	449 }
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	450
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	451 lambda = FloatComplexColumnVector (wr);
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	452 v = calc_ev ? FloatComplexMatrix (atmp) : FloatComplexMatrix ();
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	453 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	454 else
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	455 (*current_liboctave_error_handler) ("cheev workspace query failed");
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	456
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	457 return info;
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	458 }
82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: diff changeset	459
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	460 octave_idx_type
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	461 FloatEIG::init (const FloatMatrix& a, const FloatMatrix& b, bool calc_ev)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	462 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	463 if (a.any_element_is_inf_or_nan () \|\| b.any_element_is_inf_or_nan ())
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	464 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	465 (*current_liboctave_error_handler)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	466 ("EIG: matrix contains Inf or NaN values");
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	467 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	468 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	469
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	470 octave_idx_type n = a.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	471 octave_idx_type nb = b.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	472
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	473 if (n != a.cols () \|\| nb != b.cols ())
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	474 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	475 (*current_liboctave_error_handler) ("EIG requires square matrix");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	476 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	477 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	478
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	479 if (n != nb)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	480 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	481 (*current_liboctave_error_handler) ("EIG requires same size matrices");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	482 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	483 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	484
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	485 octave_idx_type info = 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	486
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	487 FloatMatrix tmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	488 float *tmp_data = tmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	489
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	490 F77_XFCN (spotrf, SPOTRF, (F77_CONST_CHAR_ARG2 ("L", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	491 n, tmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	492 info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	493 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	494 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	495
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	496 if (a.is_symmetric () && b.is_symmetric () && info == 0)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	497 return symmetric_init (a, b, calc_ev);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	498
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	499 FloatMatrix atmp = a;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	500 float *atmp_data = atmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	501
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	502 FloatMatrix btmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	503 float *btmp_data = btmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	504
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	505 Array<float> ar (dim_vector (n, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	506 float *par = ar.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	507
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	508 Array<float> ai (dim_vector (n, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	509 float *pai = ai.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	510
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	511 Array<float> beta (dim_vector (n, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	512 float *pbeta = beta.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	513
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	514 volatile octave_idx_type nvr = calc_ev ? n : 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	515 FloatMatrix vr (nvr, nvr);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	516 float *pvr = vr.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	517
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	518 octave_idx_type lwork = -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	519 float dummy_work;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	520
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	521 float *dummy = 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	522 octave_idx_type idummy = 1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	523
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	524 F77_XFCN (sggev, SGGEV, (F77_CONST_CHAR_ARG2 ("N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	525 F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	526 n, atmp_data, n, btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	527 par, pai, pbeta,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	528 dummy, idummy, pvr, n,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	529 &dummy_work, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	530 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	531 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	532
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	533 if (info == 0)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	534 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	535 lwork = static_cast<octave_idx_type> (dummy_work);
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	536 Array<float> work (dim_vector (lwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	537 float *pwork = work.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	538
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	539 F77_XFCN (sggev, SGGEV, (F77_CONST_CHAR_ARG2 ("N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	540 F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	541 n, atmp_data, n, btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	542 par, pai, pbeta,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	543 dummy, idummy, pvr, n,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	544 pwork, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	545 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	546 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	547
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	548 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	549 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	550 (*current_liboctave_error_handler) ("unrecoverable error in sggev");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	551 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	552 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	553
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	554 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	555 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	556 (*current_liboctave_error_handler) ("sggev failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	557 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	558 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	559
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	560 lambda.resize (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	561 v.resize (nvr, nvr);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	562
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	563 for (octave_idx_type j = 0; j < n; j++)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	564 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	565 if (ai.elem (j) == 0.0)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	566 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	567 lambda.elem (j) = FloatComplex (ar.elem (j) / beta.elem (j));
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	568 for (octave_idx_type i = 0; i < nvr; i++)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	569 v.elem (i, j) = vr.elem (i, j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	570 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	571 else
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	572 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	573 if (j+1 >= n)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	574 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	575 (*current_liboctave_error_handler) ("EIG: internal error");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	576 return -1;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	577 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	578
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	579 lambda.elem(j) = FloatComplex (ar.elem(j) / beta.elem (j),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	580 ai.elem(j) / beta.elem (j));
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	581 lambda.elem(j+1) = FloatComplex (ar.elem(j+1) / beta.elem (j+1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	582 ai.elem(j+1) / beta.elem (j+1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	583
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	584 for (octave_idx_type i = 0; i < nvr; i++)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	585 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	586 float real_part = vr.elem (i, j);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	587 float imag_part = vr.elem (i, j+1);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	588 v.elem (i, j) = FloatComplex (real_part, imag_part);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	589 v.elem (i, j+1) = FloatComplex (real_part, -imag_part);
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	590 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	591 j++;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	592 }
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	593 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	594 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	595 else
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	596 (*current_liboctave_error_handler) ("sggev workspace query failed");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	597
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	598 return info;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	599 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	600
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	601 octave_idx_type
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	602 FloatEIG::symmetric_init (const FloatMatrix& a, const FloatMatrix& b, bool calc_ev)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	603 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	604 octave_idx_type n = a.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	605 octave_idx_type nb = b.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	606
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	607 if (n != a.cols () \|\| nb != b.cols ())
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	608 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	609 (*current_liboctave_error_handler) ("EIG requires square matrix");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	610 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	611 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	612
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	613 if (n != nb)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	614 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	615 (*current_liboctave_error_handler) ("EIG requires same size matrices");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	616 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	617 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	618
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	619 octave_idx_type info = 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	620
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	621 FloatMatrix atmp = a;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	622 float *atmp_data = atmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	623
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	624 FloatMatrix btmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	625 float *btmp_data = btmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	626
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	627 FloatColumnVector wr (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	628 float *pwr = wr.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	629
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	630 octave_idx_type lwork = -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	631 float dummy_work;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	632
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	633 F77_XFCN (ssygv, SSYGV, (1, F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	634 F77_CONST_CHAR_ARG2 ("U", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	635 n, atmp_data, n,
12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	636 btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	637 pwr, &dummy_work, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	638 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	639 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	640
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	641 if (info == 0)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	642 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	643 lwork = static_cast<octave_idx_type> (dummy_work);
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	644 Array<float> work (dim_vector (lwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	645 float *pwork = work.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	646
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	647 F77_XFCN (ssygv, SSYGV, (1, F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	648 F77_CONST_CHAR_ARG2 ("U", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	649 n, atmp_data, n,
12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	650 btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	651 pwr, pwork, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	652 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	653 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	654
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	655 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	656 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	657 (*current_liboctave_error_handler) ("unrecoverable error in ssygv");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	658 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	659 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	660
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	661 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	662 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	663 (*current_liboctave_error_handler) ("ssygv failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	664 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	665 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	666
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	667 lambda = FloatComplexColumnVector (wr);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	668 v = calc_ev ? FloatComplexMatrix (atmp) : FloatComplexMatrix ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	669 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	670 else
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	671 (*current_liboctave_error_handler) ("ssygv workspace query failed");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	672
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	673 return info;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	674 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	675
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	676 octave_idx_type
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	677 FloatEIG::init (const FloatComplexMatrix& a, const FloatComplexMatrix& b, bool calc_ev)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	678 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	679 if (a.any_element_is_inf_or_nan () \|\| b.any_element_is_inf_or_nan ())
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	680 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	681 (*current_liboctave_error_handler)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	682 ("EIG: matrix contains Inf or NaN values");
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	683 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	684 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	685
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	686 octave_idx_type n = a.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	687 octave_idx_type nb = b.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	688
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	689 if (n != a.cols () \|\| nb != b.cols())
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	690 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	691 (*current_liboctave_error_handler) ("EIG requires square matrix");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	692 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	693 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	694
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	695 if (n != nb)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	696 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	697 (*current_liboctave_error_handler) ("EIG requires same size matrices");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	698 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	699 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	700
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	701 octave_idx_type info = 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	702
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	703 FloatComplexMatrix tmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	704 FloatComplex *tmp_data = tmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	705
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	706 F77_XFCN (cpotrf, CPOTRF, (F77_CONST_CHAR_ARG2 ("L", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	707 n, tmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	708 info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	709 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	710 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	711
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	712 if (a.is_hermitian () && b.is_hermitian () && info == 0)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	713 return hermitian_init (a, calc_ev);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	714
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	715 FloatComplexMatrix atmp = a;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	716 FloatComplex *atmp_data = atmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	717
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	718 FloatComplexMatrix btmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	719 FloatComplex *btmp_data = btmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	720
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	721 FloatComplexColumnVector alpha (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	722 FloatComplex *palpha = alpha.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	723
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	724 FloatComplexColumnVector beta (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	725 FloatComplex *pbeta = beta.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	726
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	727 octave_idx_type nvr = calc_ev ? n : 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	728 FloatComplexMatrix vtmp (nvr, nvr);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	729 FloatComplex *pv = vtmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	730
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	731 octave_idx_type lwork = -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	732 FloatComplex dummy_work;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	733
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	734 octave_idx_type lrwork = 8*n;
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	735 Array<float> rwork (dim_vector (lrwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	736 float *prwork = rwork.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	737
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	738 FloatComplex *dummy = 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	739 octave_idx_type idummy = 1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	740
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	741 F77_XFCN (cggev, CGGEV, (F77_CONST_CHAR_ARG2 ("N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	742 F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	743 n, atmp_data, n, btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	744 palpha, pbeta, dummy, idummy,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	745 pv, n, &dummy_work, lwork, prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	746 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	747 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	748
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	749 if (info == 0)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	750 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	751 lwork = static_cast<octave_idx_type> (dummy_work.real ());
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	752 Array<FloatComplex> work (dim_vector (lwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	753 FloatComplex *pwork = work.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	754
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	755 F77_XFCN (cggev, CGGEV, (F77_CONST_CHAR_ARG2 ("N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	756 F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	757 n, atmp_data, n, btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	758 palpha, pbeta, dummy, idummy,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	759 pv, n, pwork, lwork, prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	760 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	761 F77_CHAR_ARG_LEN (1)));
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	762
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	763 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	764 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	765 (*current_liboctave_error_handler) ("unrecoverable error in cggev");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	766 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	767 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	768
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	769 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	770 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	771 (*current_liboctave_error_handler) ("cggev failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	772 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	773 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	774
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	775 lambda.resize (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	776
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	777 for (octave_idx_type j = 0; j < n; j++)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	778 lambda.elem (j) = alpha.elem (j) / beta.elem(j);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	779
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	780 v = vtmp;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	781 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	782 else
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	783 (*current_liboctave_error_handler) ("cggev workspace query failed");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	784
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	785 return info;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	786 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	787
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	788 octave_idx_type
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	789 FloatEIG::hermitian_init (const FloatComplexMatrix& a, const FloatComplexMatrix& b, bool calc_ev)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	790 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	791 octave_idx_type n = a.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	792 octave_idx_type nb = b.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	793
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	794 if (n != a.cols () \|\| nb != b.cols ())
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	795 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	796 (*current_liboctave_error_handler) ("EIG requires square matrix");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	797 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	798 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	799
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	800 if (n != nb)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	801 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	802 (*current_liboctave_error_handler) ("EIG requires same size matrices");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	803 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	804 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	805
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	806 octave_idx_type info = 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	807
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	808 FloatComplexMatrix atmp = a;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	809 FloatComplex *atmp_data = atmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	810
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	811 FloatComplexMatrix btmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	812 FloatComplex *btmp_data = btmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	813
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	814 FloatColumnVector wr (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	815 float *pwr = wr.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	816
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	817 octave_idx_type lwork = -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	818 FloatComplex dummy_work;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	819
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	820 octave_idx_type lrwork = 3*n;
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	821 Array<float> rwork (dim_vector (lrwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	822 float *prwork = rwork.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	823
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	824 F77_XFCN (chegv, CHEGV, (1, F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	825 F77_CONST_CHAR_ARG2 ("U", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	826 n, atmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	827 btmp_data, n,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	828 pwr, &dummy_work, lwork,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	829 prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	830 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	831 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	832
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	833 if (info == 0)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	834 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	835 lwork = static_cast<octave_idx_type> (dummy_work.real ());
11570 57632dea2446 attempt better backward compatibility for Array constructors John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	836 Array<FloatComplex> work (dim_vector (lwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	837 FloatComplex *pwork = work.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	838
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	839 F77_XFCN (chegv, CHEGV, (1, F77_CONST_CHAR_ARG2 (calc_ev ? "V" : "N", 1),
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	840 F77_CONST_CHAR_ARG2 ("U", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	841 n, atmp_data, n,
12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	842 btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	843 pwr, pwork, lwork, prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	844 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	845 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	846
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	847 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	848 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	849 (*current_liboctave_error_handler) ("unrecoverable error in zhegv");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	850 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	851 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	852
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	853 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	854 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	855 (*current_liboctave_error_handler) ("zhegv failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	856 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	857 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	858
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	859 lambda = FloatComplexColumnVector (wr);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	860 v = calc_ev ? FloatComplexMatrix (atmp) : FloatComplexMatrix ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	861 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	862 else
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	863 (*current_liboctave_error_handler) ("zhegv workspace query failed");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	864
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	865 return info;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	866 }

Mercurial > hg > octave-nkf

annotate liboctave/fEIG.cc @ 14557:e8e86ae3abbc