octave-nkf: liboctave/numeric/fEIG.cc annotate

annotate liboctave/numeric/fEIG.cc @ 20525:ff904ae0285b

eig: Return correct solution for a pair of hermitian matrices (bug #45511) * liboctave/numeric/EIG.cc (EIG::init): Use correct form of hermitian_init. * liboctave/numeric/fEIG.cc (FloatEIG::init): Likewise. * libinterp/corefcn/eig.cc: Add %!test cases. Thanks to Marco Caliari for identifying the fix.

author	Mike Miller <mtmiller@octave.org>
date	Tue, 15 Sep 2015 08:48:35 -0400
parents	4197fc428c7d
children

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
19731 4197fc428c7d maint: Update copyright notices for 2015. John W. Eaton <jwe@octave.org> parents: 17769 diff changeset	3 Copyright (C) 1994-2015 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
15018 3d8ace26c5b4 maint: Use Octave coding conventions for cuddled parentheses in liboctave/. Rik <rik@octave.org> parents: 14846 diff changeset	229 lambda.elem (j) = FloatComplex (wr.elem (j), wi.elem (j));
3d8ace26c5b4 maint: Use Octave coding conventions for cuddled parentheses in liboctave/. Rik <rik@octave.org> parents: 14846 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
15018 3d8ace26c5b4 maint: Use Octave coding conventions for cuddled parentheses in liboctave/. Rik <rik@octave.org> parents: 14846 diff changeset	579 lambda.elem (j) = FloatComplex (ar.elem (j) / beta.elem (j),
3d8ace26c5b4 maint: Use Octave coding conventions for cuddled parentheses in liboctave/. Rik <rik@octave.org> parents: 14846 diff changeset	580 ai.elem (j) / beta.elem (j));
3d8ace26c5b4 maint: Use Octave coding conventions for cuddled parentheses in liboctave/. Rik <rik@octave.org> parents: 14846 diff changeset	581 lambda.elem (j+1) = FloatComplex (ar.elem (j+1) / beta.elem (j+1),
3d8ace26c5b4 maint: Use Octave coding conventions for cuddled parentheses in liboctave/. Rik <rik@octave.org> parents: 14846 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
17769 49a5a4be04a1 maint: Use GNU style coding conventions for code in liboctave/ Rik <rik@octave.org> parents: 17744 diff changeset	602 FloatEIG::symmetric_init (const FloatMatrix& a, const FloatMatrix& b,
49a5a4be04a1 maint: Use GNU style coding conventions for code in liboctave/ Rik <rik@octave.org> parents: 17744 diff changeset	603 bool calc_ev)
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	604 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	605 octave_idx_type n = a.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	606 octave_idx_type nb = b.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	607
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	608 if (n != a.cols () \|\| nb != b.cols ())
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	609 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	610 (*current_liboctave_error_handler) ("EIG requires square matrix");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	611 return -1;
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
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	614 if (n != nb)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	615 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	616 (*current_liboctave_error_handler) ("EIG requires same size matrices");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	617 return -1;
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
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	620 octave_idx_type info = 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	621
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	622 FloatMatrix atmp = a;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	623 float *atmp_data = atmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	624
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	625 FloatMatrix btmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	626 float *btmp_data = btmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	627
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	628 FloatColumnVector wr (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	629 float *pwr = wr.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	630
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	631 octave_idx_type lwork = -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	632 float dummy_work;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	633
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	634 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	635 F77_CONST_CHAR_ARG2 ("U", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	636 n, atmp_data, n,
12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	637 btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	638 pwr, &dummy_work, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	639 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	640 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	641
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	642 if (info == 0)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	643 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	644 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	645 Array<float> work (dim_vector (lwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	646 float *pwork = work.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	647
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	648 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	649 F77_CONST_CHAR_ARG2 ("U", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	650 n, atmp_data, n,
12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	651 btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	652 pwr, pwork, lwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	653 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	654 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	655
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	656 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	657 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	658 (*current_liboctave_error_handler) ("unrecoverable error in ssygv");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	659 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	660 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	661
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	662 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	663 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	664 (*current_liboctave_error_handler) ("ssygv failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	665 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	666 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	667
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	668 lambda = FloatComplexColumnVector (wr);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	669 v = calc_ev ? FloatComplexMatrix (atmp) : FloatComplexMatrix ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	670 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	671 else
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	672 (*current_liboctave_error_handler) ("ssygv workspace query failed");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	673
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	674 return info;
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
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	677 octave_idx_type
17769 49a5a4be04a1 maint: Use GNU style coding conventions for code in liboctave/ Rik <rik@octave.org> parents: 17744 diff changeset	678 FloatEIG::init (const FloatComplexMatrix& a, const FloatComplexMatrix& b,
49a5a4be04a1 maint: Use GNU style coding conventions for code in liboctave/ Rik <rik@octave.org> parents: 17744 diff changeset	679 bool calc_ev)
8339 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 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	682 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	683 (*current_liboctave_error_handler)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	684 ("EIG: matrix contains Inf or NaN values");
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	685 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	686 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	687
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	688 octave_idx_type n = a.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	689 octave_idx_type nb = b.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	690
14846 460a3c6d8bf1 maint: Use Octave coding convention for cuddled parenthis in function calls with empty argument lists. Rik <octave@nomad.inbox5.com> parents: 14138 diff changeset	691 if (n != a.cols () \|\| nb != b.cols ())
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	692 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	693 (*current_liboctave_error_handler) ("EIG requires square matrix");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	694 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	695 }
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 if (n != nb)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	698 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	699 (*current_liboctave_error_handler) ("EIG requires same size matrices");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	700 return -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	701 }
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 octave_idx_type info = 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	704
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	705 FloatComplexMatrix tmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	706 FloatComplex *tmp_data = tmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	707
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	708 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	709 n, tmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	710 info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	711 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	712 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	713
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	714 if (a.is_hermitian () && b.is_hermitian () && info == 0)
20525 ff904ae0285b eig: Return correct solution for a pair of hermitian matrices (bug #45511) Mike Miller <mtmiller@octave.org> parents: 19731 diff changeset	715 return hermitian_init (a, b, calc_ev);
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	716
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	717 FloatComplexMatrix atmp = a;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	718 FloatComplex *atmp_data = atmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	719
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	720 FloatComplexMatrix btmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	721 FloatComplex *btmp_data = btmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	722
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	723 FloatComplexColumnVector alpha (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	724 FloatComplex *palpha = alpha.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	725
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	726 FloatComplexColumnVector beta (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	727 FloatComplex *pbeta = beta.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	728
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	729 octave_idx_type nvr = calc_ev ? n : 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	730 FloatComplexMatrix vtmp (nvr, nvr);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	731 FloatComplex *pv = vtmp.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	732
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	733 octave_idx_type lwork = -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	734 FloatComplex dummy_work;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	735
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	736 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	737 Array<float> rwork (dim_vector (lrwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	738 float *prwork = rwork.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	739
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	740 FloatComplex *dummy = 0;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	741 octave_idx_type idummy = 1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	742
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	743 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	744 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	745 n, atmp_data, n, btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	746 palpha, pbeta, dummy, idummy,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	747 pv, n, &dummy_work, lwork, prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	748 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	749 F77_CHAR_ARG_LEN (1)));
8339 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 if (info == 0)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	752 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	753 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	754 Array<FloatComplex> work (dim_vector (lwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	755 FloatComplex *pwork = work.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	756
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	757 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	758 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	759 n, atmp_data, n, btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	760 palpha, pbeta, dummy, idummy,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	761 pv, n, pwork, lwork, prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	762 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	763 F77_CHAR_ARG_LEN (1)));
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	764
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	765 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	766 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	767 (*current_liboctave_error_handler) ("unrecoverable error in cggev");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	768 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	769 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	770
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	771 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	772 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	773 (*current_liboctave_error_handler) ("cggev failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	774 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	775 }
8339 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 lambda.resize (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	778
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	779 for (octave_idx_type j = 0; j < n; j++)
15018 3d8ace26c5b4 maint: Use Octave coding conventions for cuddled parentheses in liboctave/. Rik <rik@octave.org> parents: 14846 diff changeset	780 lambda.elem (j) = alpha.elem (j) / beta.elem (j);
8339 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 v = vtmp;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	783 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	784 else
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	785 (*current_liboctave_error_handler) ("cggev workspace query failed");
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 return info;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	788 }
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	789
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	790 octave_idx_type
17769 49a5a4be04a1 maint: Use GNU style coding conventions for code in liboctave/ Rik <rik@octave.org> parents: 17744 diff changeset	791 FloatEIG::hermitian_init (const FloatComplexMatrix& a,
49a5a4be04a1 maint: Use GNU style coding conventions for code in liboctave/ Rik <rik@octave.org> parents: 17744 diff changeset	792 const FloatComplexMatrix& b, bool calc_ev)
8339 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 octave_idx_type n = a.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	795 octave_idx_type nb = b.rows ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	796
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	797 if (n != a.cols () \|\| nb != b.cols ())
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 (*current_liboctave_error_handler) ("EIG requires square matrix");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	800 return -1;
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
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	803 if (n != nb)
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 (*current_liboctave_error_handler) ("EIG requires same size matrices");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	806 return -1;
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
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	809 octave_idx_type info = 0;
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 atmp = a;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	812 FloatComplex *atmp_data = atmp.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 FloatComplexMatrix btmp = b;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	815 FloatComplex *btmp_data = btmp.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 FloatColumnVector wr (n);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	818 float *pwr = wr.fortran_vec ();
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 lwork = -1;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	821 FloatComplex dummy_work;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	822
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	823 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	824 Array<float> rwork (dim_vector (lrwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	825 float *prwork = rwork.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	826
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	827 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	828 F77_CONST_CHAR_ARG2 ("U", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	829 n, atmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	830 btmp_data, n,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	831 pwr, &dummy_work, lwork,
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	832 prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	833 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	834 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	835
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	836 if (info == 0)
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	837 {
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	838 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	839 Array<FloatComplex> work (dim_vector (lwork, 1));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	840 FloatComplex *pwork = work.fortran_vec ();
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	841
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	842 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	843 F77_CONST_CHAR_ARG2 ("U", 1),
11586 12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	844 n, atmp_data, n,
12df7854fa7c strip trailing whitespace from source files John W. Eaton <jwe@octave.org> parents: 11570 diff changeset	845 btmp_data, n,
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	846 pwr, pwork, lwork, prwork, info
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	847 F77_CHAR_ARG_LEN (1)
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	848 F77_CHAR_ARG_LEN (1)));
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	849
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	850 if (info < 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	851 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	852 (*current_liboctave_error_handler) ("unrecoverable error in zhegv");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	853 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	854 }
8339 18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	855
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	856 if (info > 0)
10314 07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	857 {
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	858 (*current_liboctave_error_handler) ("zhegv failed to converge");
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	859 return info;
07ebe522dac2 untabify liboctave C++ sources John W. Eaton <jwe@octave.org> parents: 10158 diff changeset	860 }
8339 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 lambda = FloatComplexColumnVector (wr);
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	863 v = calc_ev ? FloatComplexMatrix (atmp) : FloatComplexMatrix ();
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 else
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	866 (*current_liboctave_error_handler) ("zhegv workspace query failed");
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	867
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	868 return info;
18c4ded8612a Add generalized eigenvalue functions Jarkko Kaleva <d3roga@gmail.com> parents: 7789 diff changeset	869 }

Mercurial > octave-nkf

annotate liboctave/numeric/fEIG.cc @ 20525:ff904ae0285b