C_LAPACK: Fixes to make it compile with MSVC (#3605)
[platform/upstream/openblas.git] / lapack-netlib / SRC / dsteqr.c
1 #include <math.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <stdio.h>
5 #include <complex.h>
6 #ifdef complex
7 #undef complex
8 #endif
9 #ifdef I
10 #undef I
11 #endif
12
13 #if defined(_WIN64)
14 typedef long long BLASLONG;
15 typedef unsigned long long BLASULONG;
16 #else
17 typedef long BLASLONG;
18 typedef unsigned long BLASULONG;
19 #endif
20
21 #ifdef LAPACK_ILP64
22 typedef BLASLONG blasint;
23 #if defined(_WIN64)
24 #define blasabs(x) llabs(x)
25 #else
26 #define blasabs(x) labs(x)
27 #endif
28 #else
29 typedef int blasint;
30 #define blasabs(x) abs(x)
31 #endif
32
33 typedef blasint integer;
34
35 typedef unsigned int uinteger;
36 typedef char *address;
37 typedef short int shortint;
38 typedef float real;
39 typedef double doublereal;
40 typedef struct { real r, i; } complex;
41 typedef struct { doublereal r, i; } doublecomplex;
42 #ifdef _MSC_VER
43 static inline _Fcomplex Cf(complex *z) {_Fcomplex zz={z->r , z->i}; return zz;}
44 static inline _Dcomplex Cd(doublecomplex *z) {_Dcomplex zz={z->r , z->i};return zz;}
45 static inline _Fcomplex * _pCf(complex *z) {return (_Fcomplex*)z;}
46 static inline _Dcomplex * _pCd(doublecomplex *z) {return (_Dcomplex*)z;}
47 #else
48 static inline _Complex float Cf(complex *z) {return z->r + z->i*_Complex_I;}
49 static inline _Complex double Cd(doublecomplex *z) {return z->r + z->i*_Complex_I;}
50 static inline _Complex float * _pCf(complex *z) {return (_Complex float*)z;}
51 static inline _Complex double * _pCd(doublecomplex *z) {return (_Complex double*)z;}
52 #endif
53 #define pCf(z) (*_pCf(z))
54 #define pCd(z) (*_pCd(z))
55 typedef int logical;
56 typedef short int shortlogical;
57 typedef char logical1;
58 typedef char integer1;
59
60 #define TRUE_ (1)
61 #define FALSE_ (0)
62
63 /* Extern is for use with -E */
64 #ifndef Extern
65 #define Extern extern
66 #endif
67
68 /* I/O stuff */
69
70 typedef int flag;
71 typedef int ftnlen;
72 typedef int ftnint;
73
74 /*external read, write*/
75 typedef struct
76 {       flag cierr;
77         ftnint ciunit;
78         flag ciend;
79         char *cifmt;
80         ftnint cirec;
81 } cilist;
82
83 /*internal read, write*/
84 typedef struct
85 {       flag icierr;
86         char *iciunit;
87         flag iciend;
88         char *icifmt;
89         ftnint icirlen;
90         ftnint icirnum;
91 } icilist;
92
93 /*open*/
94 typedef struct
95 {       flag oerr;
96         ftnint ounit;
97         char *ofnm;
98         ftnlen ofnmlen;
99         char *osta;
100         char *oacc;
101         char *ofm;
102         ftnint orl;
103         char *oblnk;
104 } olist;
105
106 /*close*/
107 typedef struct
108 {       flag cerr;
109         ftnint cunit;
110         char *csta;
111 } cllist;
112
113 /*rewind, backspace, endfile*/
114 typedef struct
115 {       flag aerr;
116         ftnint aunit;
117 } alist;
118
119 /* inquire */
120 typedef struct
121 {       flag inerr;
122         ftnint inunit;
123         char *infile;
124         ftnlen infilen;
125         ftnint  *inex;  /*parameters in standard's order*/
126         ftnint  *inopen;
127         ftnint  *innum;
128         ftnint  *innamed;
129         char    *inname;
130         ftnlen  innamlen;
131         char    *inacc;
132         ftnlen  inacclen;
133         char    *inseq;
134         ftnlen  inseqlen;
135         char    *indir;
136         ftnlen  indirlen;
137         char    *infmt;
138         ftnlen  infmtlen;
139         char    *inform;
140         ftnint  informlen;
141         char    *inunf;
142         ftnlen  inunflen;
143         ftnint  *inrecl;
144         ftnint  *innrec;
145         char    *inblank;
146         ftnlen  inblanklen;
147 } inlist;
148
149 #define VOID void
150
151 union Multitype {       /* for multiple entry points */
152         integer1 g;
153         shortint h;
154         integer i;
155         /* longint j; */
156         real r;
157         doublereal d;
158         complex c;
159         doublecomplex z;
160         };
161
162 typedef union Multitype Multitype;
163
164 struct Vardesc {        /* for Namelist */
165         char *name;
166         char *addr;
167         ftnlen *dims;
168         int  type;
169         };
170 typedef struct Vardesc Vardesc;
171
172 struct Namelist {
173         char *name;
174         Vardesc **vars;
175         int nvars;
176         };
177 typedef struct Namelist Namelist;
178
179 #define abs(x) ((x) >= 0 ? (x) : -(x))
180 #define dabs(x) (fabs(x))
181 #define f2cmin(a,b) ((a) <= (b) ? (a) : (b))
182 #define f2cmax(a,b) ((a) >= (b) ? (a) : (b))
183 #define dmin(a,b) (f2cmin(a,b))
184 #define dmax(a,b) (f2cmax(a,b))
185 #define bit_test(a,b)   ((a) >> (b) & 1)
186 #define bit_clear(a,b)  ((a) & ~((uinteger)1 << (b)))
187 #define bit_set(a,b)    ((a) |  ((uinteger)1 << (b)))
188
189 #define abort_() { sig_die("Fortran abort routine called", 1); }
190 #define c_abs(z) (cabsf(Cf(z)))
191 #define c_cos(R,Z) { pCf(R)=ccos(Cf(Z)); }
192 #ifdef _MSC_VER
193 #define c_div(c, a, b) {Cf(c)._Val[0] = (Cf(a)._Val[0]/Cf(b)._Val[0]); Cf(c)._Val[1]=(Cf(a)._Val[1]/Cf(b)._Val[1]);}
194 #define z_div(c, a, b) {Cd(c)._Val[0] = (Cd(a)._Val[0]/Cd(b)._Val[0]); Cd(c)._Val[1]=(Cd(a)._Val[1]/df(b)._Val[1]);}
195 #else
196 #define c_div(c, a, b) {pCf(c) = Cf(a)/Cf(b);}
197 #define z_div(c, a, b) {pCd(c) = Cd(a)/Cd(b);}
198 #endif
199 #define c_exp(R, Z) {pCf(R) = cexpf(Cf(Z));}
200 #define c_log(R, Z) {pCf(R) = clogf(Cf(Z));}
201 #define c_sin(R, Z) {pCf(R) = csinf(Cf(Z));}
202 //#define c_sqrt(R, Z) {*(R) = csqrtf(Cf(Z));}
203 #define c_sqrt(R, Z) {pCf(R) = csqrtf(Cf(Z));}
204 #define d_abs(x) (fabs(*(x)))
205 #define d_acos(x) (acos(*(x)))
206 #define d_asin(x) (asin(*(x)))
207 #define d_atan(x) (atan(*(x)))
208 #define d_atn2(x, y) (atan2(*(x),*(y)))
209 #define d_cnjg(R, Z) { pCd(R) = conj(Cd(Z)); }
210 #define r_cnjg(R, Z) { pCf(R) = conjf(Cf(Z)); }
211 #define d_cos(x) (cos(*(x)))
212 #define d_cosh(x) (cosh(*(x)))
213 #define d_dim(__a, __b) ( *(__a) > *(__b) ? *(__a) - *(__b) : 0.0 )
214 #define d_exp(x) (exp(*(x)))
215 #define d_imag(z) (cimag(Cd(z)))
216 #define r_imag(z) (cimagf(Cf(z)))
217 #define d_int(__x) (*(__x)>0 ? floor(*(__x)) : -floor(- *(__x)))
218 #define r_int(__x) (*(__x)>0 ? floor(*(__x)) : -floor(- *(__x)))
219 #define d_lg10(x) ( 0.43429448190325182765 * log(*(x)) )
220 #define r_lg10(x) ( 0.43429448190325182765 * log(*(x)) )
221 #define d_log(x) (log(*(x)))
222 #define d_mod(x, y) (fmod(*(x), *(y)))
223 #define u_nint(__x) ((__x)>=0 ? floor((__x) + .5) : -floor(.5 - (__x)))
224 #define d_nint(x) u_nint(*(x))
225 #define u_sign(__a,__b) ((__b) >= 0 ? ((__a) >= 0 ? (__a) : -(__a)) : -((__a) >= 0 ? (__a) : -(__a)))
226 #define d_sign(a,b) u_sign(*(a),*(b))
227 #define r_sign(a,b) u_sign(*(a),*(b))
228 #define d_sin(x) (sin(*(x)))
229 #define d_sinh(x) (sinh(*(x)))
230 #define d_sqrt(x) (sqrt(*(x)))
231 #define d_tan(x) (tan(*(x)))
232 #define d_tanh(x) (tanh(*(x)))
233 #define i_abs(x) abs(*(x))
234 #define i_dnnt(x) ((integer)u_nint(*(x)))
235 #define i_len(s, n) (n)
236 #define i_nint(x) ((integer)u_nint(*(x)))
237 #define i_sign(a,b) ((integer)u_sign((integer)*(a),(integer)*(b)))
238 #define pow_dd(ap, bp) ( pow(*(ap), *(bp)))
239 #define pow_si(B,E) spow_ui(*(B),*(E))
240 #define pow_ri(B,E) spow_ui(*(B),*(E))
241 #define pow_di(B,E) dpow_ui(*(B),*(E))
242 #define pow_zi(p, a, b) {pCd(p) = zpow_ui(Cd(a), *(b));}
243 #define pow_ci(p, a, b) {pCf(p) = cpow_ui(Cf(a), *(b));}
244 #define pow_zz(R,A,B) {pCd(R) = cpow(Cd(A),*(B));}
245 #define s_cat(lpp, rpp, rnp, np, llp) {         ftnlen i, nc, ll; char *f__rp, *lp;     ll = (llp); lp = (lpp);         for(i=0; i < (int)*(np); ++i) {                 nc = ll;                if((rnp)[i] < nc) nc = (rnp)[i];                ll -= nc;               f__rp = (rpp)[i];               while(--nc >= 0) *lp++ = *(f__rp)++;         }  while(--ll >= 0) *lp++ = ' '; }
246 #define s_cmp(a,b,c,d) ((integer)strncmp((a),(b),f2cmin((c),(d))))
247 #define s_copy(A,B,C,D) { int __i,__m; for (__i=0, __m=f2cmin((C),(D)); __i<__m && (B)[__i] != 0; ++__i) (A)[__i] = (B)[__i]; }
248 #define sig_die(s, kill) { exit(1); }
249 #define s_stop(s, n) {exit(0);}
250 static char junk[] = "\n@(#)LIBF77 VERSION 19990503\n";
251 #define z_abs(z) (cabs(Cd(z)))
252 #define z_exp(R, Z) {pCd(R) = cexp(Cd(Z));}
253 #define z_sqrt(R, Z) {pCd(R) = csqrt(Cd(Z));}
254 #define myexit_() break;
255 #define mycycle() continue;
256 #define myceiling(w) {ceil(w)}
257 #define myhuge(w) {HUGE_VAL}
258 //#define mymaxloc_(w,s,e,n) {if (sizeof(*(w)) == sizeof(double)) dmaxloc_((w),*(s),*(e),n); else dmaxloc_((w),*(s),*(e),n);}
259 #define mymaxloc(w,s,e,n) {dmaxloc_(w,*(s),*(e),n)}
260
261 /* procedure parameter types for -A and -C++ */
262
263 #define F2C_proc_par_types 1
264 #ifdef __cplusplus
265 typedef logical (*L_fp)(...);
266 #else
267 typedef logical (*L_fp)();
268 #endif
269
270 static float spow_ui(float x, integer n) {
271         float pow=1.0; unsigned long int u;
272         if(n != 0) {
273                 if(n < 0) n = -n, x = 1/x;
274                 for(u = n; ; ) {
275                         if(u & 01) pow *= x;
276                         if(u >>= 1) x *= x;
277                         else break;
278                 }
279         }
280         return pow;
281 }
282 static double dpow_ui(double x, integer n) {
283         double pow=1.0; unsigned long int u;
284         if(n != 0) {
285                 if(n < 0) n = -n, x = 1/x;
286                 for(u = n; ; ) {
287                         if(u & 01) pow *= x;
288                         if(u >>= 1) x *= x;
289                         else break;
290                 }
291         }
292         return pow;
293 }
294 #ifdef _MSC_VER
295 static _Fcomplex cpow_ui(complex x, integer n) {
296         complex pow={1.0,0.0}; unsigned long int u;
297                 if(n != 0) {
298                 if(n < 0) n = -n, x.r = 1/x.r, x.i=1/x.i;
299                 for(u = n; ; ) {
300                         if(u & 01) pow.r *= x.r, pow.i *= x.i;
301                         if(u >>= 1) x.r *= x.r, x.i *= x.i;
302                         else break;
303                 }
304         }
305         _Fcomplex p={pow.r, pow.i};
306         return p;
307 }
308 #else
309 static _Complex float cpow_ui(_Complex float x, integer n) {
310         _Complex float pow=1.0; unsigned long int u;
311         if(n != 0) {
312                 if(n < 0) n = -n, x = 1/x;
313                 for(u = n; ; ) {
314                         if(u & 01) pow *= x;
315                         if(u >>= 1) x *= x;
316                         else break;
317                 }
318         }
319         return pow;
320 }
321 #endif
322 #ifdef _MSC_VER
323 static _Dcomplex zpow_ui(_Dcomplex x, integer n) {
324         _Dcomplex pow={1.0,0.0}; unsigned long int u;
325         if(n != 0) {
326                 if(n < 0) n = -n, x._Val[0] = 1/x._Val[0], x._Val[1] =1/x._Val[1];
327                 for(u = n; ; ) {
328                         if(u & 01) pow._Val[0] *= x._Val[0], pow._Val[1] *= x._Val[1];
329                         if(u >>= 1) x._Val[0] *= x._Val[0], x._Val[1] *= x._Val[1];
330                         else break;
331                 }
332         }
333         _Dcomplex p = {pow._Val[0], pow._Val[1]};
334         return p;
335 }
336 #else
337 static _Complex double zpow_ui(_Complex double x, integer n) {
338         _Complex double pow=1.0; unsigned long int u;
339         if(n != 0) {
340                 if(n < 0) n = -n, x = 1/x;
341                 for(u = n; ; ) {
342                         if(u & 01) pow *= x;
343                         if(u >>= 1) x *= x;
344                         else break;
345                 }
346         }
347         return pow;
348 }
349 #endif
350 static integer pow_ii(integer x, integer n) {
351         integer pow; unsigned long int u;
352         if (n <= 0) {
353                 if (n == 0 || x == 1) pow = 1;
354                 else if (x != -1) pow = x == 0 ? 1/x : 0;
355                 else n = -n;
356         }
357         if ((n > 0) || !(n == 0 || x == 1 || x != -1)) {
358                 u = n;
359                 for(pow = 1; ; ) {
360                         if(u & 01) pow *= x;
361                         if(u >>= 1) x *= x;
362                         else break;
363                 }
364         }
365         return pow;
366 }
367 static integer dmaxloc_(double *w, integer s, integer e, integer *n)
368 {
369         double m; integer i, mi;
370         for(m=w[s-1], mi=s, i=s+1; i<=e; i++)
371                 if (w[i-1]>m) mi=i ,m=w[i-1];
372         return mi-s+1;
373 }
374 static integer smaxloc_(float *w, integer s, integer e, integer *n)
375 {
376         float m; integer i, mi;
377         for(m=w[s-1], mi=s, i=s+1; i<=e; i++)
378                 if (w[i-1]>m) mi=i ,m=w[i-1];
379         return mi-s+1;
380 }
381 static inline void cdotc_(complex *z, integer *n_, complex *x, integer *incx_, complex *y, integer *incy_) {
382         integer n = *n_, incx = *incx_, incy = *incy_, i;
383 #ifdef _MSC_VER
384         _Fcomplex zdotc = {0.0, 0.0};
385         if (incx == 1 && incy == 1) {
386                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
387                         zdotc._Val[0] += conjf(Cf(&x[i]))._Val[0] * Cf(&y[i])._Val[0];
388                         zdotc._Val[1] += conjf(Cf(&x[i]))._Val[1] * Cf(&y[i])._Val[1];
389                 }
390         } else {
391                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
392                         zdotc._Val[0] += conjf(Cf(&x[i*incx]))._Val[0] * Cf(&y[i*incy])._Val[0];
393                         zdotc._Val[1] += conjf(Cf(&x[i*incx]))._Val[1] * Cf(&y[i*incy])._Val[1];
394                 }
395         }
396         pCf(z) = zdotc;
397 }
398 #else
399         _Complex float zdotc = 0.0;
400         if (incx == 1 && incy == 1) {
401                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
402                         zdotc += conjf(Cf(&x[i])) * Cf(&y[i]);
403                 }
404         } else {
405                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
406                         zdotc += conjf(Cf(&x[i*incx])) * Cf(&y[i*incy]);
407                 }
408         }
409         pCf(z) = zdotc;
410 }
411 #endif
412 static inline void zdotc_(doublecomplex *z, integer *n_, doublecomplex *x, integer *incx_, doublecomplex *y, integer *incy_) {
413         integer n = *n_, incx = *incx_, incy = *incy_, i;
414 #ifdef _MSC_VER
415         _Dcomplex zdotc = {0.0, 0.0};
416         if (incx == 1 && incy == 1) {
417                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
418                         zdotc._Val[0] += conj(Cd(&x[i]))._Val[0] * Cd(&y[i])._Val[0];
419                         zdotc._Val[1] += conj(Cd(&x[i]))._Val[1] * Cd(&y[i])._Val[1];
420                 }
421         } else {
422                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
423                         zdotc._Val[0] += conj(Cd(&x[i*incx]))._Val[0] * Cd(&y[i*incy])._Val[0];
424                         zdotc._Val[1] += conj(Cd(&x[i*incx]))._Val[1] * Cd(&y[i*incy])._Val[1];
425                 }
426         }
427         pCd(z) = zdotc;
428 }
429 #else
430         _Complex double zdotc = 0.0;
431         if (incx == 1 && incy == 1) {
432                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
433                         zdotc += conj(Cd(&x[i])) * Cd(&y[i]);
434                 }
435         } else {
436                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
437                         zdotc += conj(Cd(&x[i*incx])) * Cd(&y[i*incy]);
438                 }
439         }
440         pCd(z) = zdotc;
441 }
442 #endif  
443 static inline void cdotu_(complex *z, integer *n_, complex *x, integer *incx_, complex *y, integer *incy_) {
444         integer n = *n_, incx = *incx_, incy = *incy_, i;
445 #ifdef _MSC_VER
446         _Fcomplex zdotc = {0.0, 0.0};
447         if (incx == 1 && incy == 1) {
448                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
449                         zdotc._Val[0] += Cf(&x[i])._Val[0] * Cf(&y[i])._Val[0];
450                         zdotc._Val[1] += Cf(&x[i])._Val[1] * Cf(&y[i])._Val[1];
451                 }
452         } else {
453                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
454                         zdotc._Val[0] += Cf(&x[i*incx])._Val[0] * Cf(&y[i*incy])._Val[0];
455                         zdotc._Val[1] += Cf(&x[i*incx])._Val[1] * Cf(&y[i*incy])._Val[1];
456                 }
457         }
458         pCf(z) = zdotc;
459 }
460 #else
461         _Complex float zdotc = 0.0;
462         if (incx == 1 && incy == 1) {
463                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
464                         zdotc += Cf(&x[i]) * Cf(&y[i]);
465                 }
466         } else {
467                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
468                         zdotc += Cf(&x[i*incx]) * Cf(&y[i*incy]);
469                 }
470         }
471         pCf(z) = zdotc;
472 }
473 #endif
474 static inline void zdotu_(doublecomplex *z, integer *n_, doublecomplex *x, integer *incx_, doublecomplex *y, integer *incy_) {
475         integer n = *n_, incx = *incx_, incy = *incy_, i;
476 #ifdef _MSC_VER
477         _Dcomplex zdotc = {0.0, 0.0};
478         if (incx == 1 && incy == 1) {
479                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
480                         zdotc._Val[0] += Cd(&x[i])._Val[0] * Cd(&y[i])._Val[0];
481                         zdotc._Val[1] += Cd(&x[i])._Val[1] * Cd(&y[i])._Val[1];
482                 }
483         } else {
484                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
485                         zdotc._Val[0] += Cd(&x[i*incx])._Val[0] * Cd(&y[i*incy])._Val[0];
486                         zdotc._Val[1] += Cd(&x[i*incx])._Val[1] * Cd(&y[i*incy])._Val[1];
487                 }
488         }
489         pCd(z) = zdotc;
490 }
491 #else
492         _Complex double zdotc = 0.0;
493         if (incx == 1 && incy == 1) {
494                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
495                         zdotc += Cd(&x[i]) * Cd(&y[i]);
496                 }
497         } else {
498                 for (i=0;i<n;i++) { /* zdotc = zdotc + dconjg(x(i))* y(i) */
499                         zdotc += Cd(&x[i*incx]) * Cd(&y[i*incy]);
500                 }
501         }
502         pCd(z) = zdotc;
503 }
504 #endif
505 /*  -- translated by f2c (version 20000121).
506    You must link the resulting object file with the libraries:
507         -lf2c -lm   (in that order)
508 */
509
510
511
512
513 /* Table of constant values */
514
515 static doublereal c_b9 = 0.;
516 static doublereal c_b10 = 1.;
517 static integer c__0 = 0;
518 static integer c__1 = 1;
519 static integer c__2 = 2;
520
521 /* > \brief \b DSTEQR */
522
523 /*  =========== DOCUMENTATION =========== */
524
525 /* Online html documentation available at */
526 /*            http://www.netlib.org/lapack/explore-html/ */
527
528 /* > \htmlonly */
529 /* > Download DSTEQR + dependencies */
530 /* > <a href="http://www.netlib.org/cgi-bin/netlibfiles.tgz?format=tgz&filename=/lapack/lapack_routine/dsteqr.
531 f"> */
532 /* > [TGZ]</a> */
533 /* > <a href="http://www.netlib.org/cgi-bin/netlibfiles.zip?format=zip&filename=/lapack/lapack_routine/dsteqr.
534 f"> */
535 /* > [ZIP]</a> */
536 /* > <a href="http://www.netlib.org/cgi-bin/netlibfiles.txt?format=txt&filename=/lapack/lapack_routine/dsteqr.
537 f"> */
538 /* > [TXT]</a> */
539 /* > \endhtmlonly */
540
541 /*  Definition: */
542 /*  =========== */
543
544 /*       SUBROUTINE DSTEQR( COMPZ, N, D, E, Z, LDZ, WORK, INFO ) */
545
546 /*       CHARACTER          COMPZ */
547 /*       INTEGER            INFO, LDZ, N */
548 /*       DOUBLE PRECISION   D( * ), E( * ), WORK( * ), Z( LDZ, * ) */
549
550
551 /* > \par Purpose: */
552 /*  ============= */
553 /* > */
554 /* > \verbatim */
555 /* > */
556 /* > DSTEQR computes all eigenvalues and, optionally, eigenvectors of a */
557 /* > symmetric tridiagonal matrix using the implicit QL or QR method. */
558 /* > The eigenvectors of a full or band symmetric matrix can also be found */
559 /* > if DSYTRD or DSPTRD or DSBTRD has been used to reduce this matrix to */
560 /* > tridiagonal form. */
561 /* > \endverbatim */
562
563 /*  Arguments: */
564 /*  ========== */
565
566 /* > \param[in] COMPZ */
567 /* > \verbatim */
568 /* >          COMPZ is CHARACTER*1 */
569 /* >          = 'N':  Compute eigenvalues only. */
570 /* >          = 'V':  Compute eigenvalues and eigenvectors of the original */
571 /* >                  symmetric matrix.  On entry, Z must contain the */
572 /* >                  orthogonal matrix used to reduce the original matrix */
573 /* >                  to tridiagonal form. */
574 /* >          = 'I':  Compute eigenvalues and eigenvectors of the */
575 /* >                  tridiagonal matrix.  Z is initialized to the identity */
576 /* >                  matrix. */
577 /* > \endverbatim */
578 /* > */
579 /* > \param[in] N */
580 /* > \verbatim */
581 /* >          N is INTEGER */
582 /* >          The order of the matrix.  N >= 0. */
583 /* > \endverbatim */
584 /* > */
585 /* > \param[in,out] D */
586 /* > \verbatim */
587 /* >          D is DOUBLE PRECISION array, dimension (N) */
588 /* >          On entry, the diagonal elements of the tridiagonal matrix. */
589 /* >          On exit, if INFO = 0, the eigenvalues in ascending order. */
590 /* > \endverbatim */
591 /* > */
592 /* > \param[in,out] E */
593 /* > \verbatim */
594 /* >          E is DOUBLE PRECISION array, dimension (N-1) */
595 /* >          On entry, the (n-1) subdiagonal elements of the tridiagonal */
596 /* >          matrix. */
597 /* >          On exit, E has been destroyed. */
598 /* > \endverbatim */
599 /* > */
600 /* > \param[in,out] Z */
601 /* > \verbatim */
602 /* >          Z is DOUBLE PRECISION array, dimension (LDZ, N) */
603 /* >          On entry, if  COMPZ = 'V', then Z contains the orthogonal */
604 /* >          matrix used in the reduction to tridiagonal form. */
605 /* >          On exit, if INFO = 0, then if  COMPZ = 'V', Z contains the */
606 /* >          orthonormal eigenvectors of the original symmetric matrix, */
607 /* >          and if COMPZ = 'I', Z contains the orthonormal eigenvectors */
608 /* >          of the symmetric tridiagonal matrix. */
609 /* >          If COMPZ = 'N', then Z is not referenced. */
610 /* > \endverbatim */
611 /* > */
612 /* > \param[in] LDZ */
613 /* > \verbatim */
614 /* >          LDZ is INTEGER */
615 /* >          The leading dimension of the array Z.  LDZ >= 1, and if */
616 /* >          eigenvectors are desired, then  LDZ >= f2cmax(1,N). */
617 /* > \endverbatim */
618 /* > */
619 /* > \param[out] WORK */
620 /* > \verbatim */
621 /* >          WORK is DOUBLE PRECISION array, dimension (f2cmax(1,2*N-2)) */
622 /* >          If COMPZ = 'N', then WORK is not referenced. */
623 /* > \endverbatim */
624 /* > */
625 /* > \param[out] INFO */
626 /* > \verbatim */
627 /* >          INFO is INTEGER */
628 /* >          = 0:  successful exit */
629 /* >          < 0:  if INFO = -i, the i-th argument had an illegal value */
630 /* >          > 0:  the algorithm has failed to find all the eigenvalues in */
631 /* >                a total of 30*N iterations; if INFO = i, then i */
632 /* >                elements of E have not converged to zero; on exit, D */
633 /* >                and E contain the elements of a symmetric tridiagonal */
634 /* >                matrix which is orthogonally similar to the original */
635 /* >                matrix. */
636 /* > \endverbatim */
637
638 /*  Authors: */
639 /*  ======== */
640
641 /* > \author Univ. of Tennessee */
642 /* > \author Univ. of California Berkeley */
643 /* > \author Univ. of Colorado Denver */
644 /* > \author NAG Ltd. */
645
646 /* > \date December 2016 */
647
648 /* > \ingroup auxOTHERcomputational */
649
650 /*  ===================================================================== */
651 /* Subroutine */ int dsteqr_(char *compz, integer *n, doublereal *d__, 
652         doublereal *e, doublereal *z__, integer *ldz, doublereal *work, 
653         integer *info)
654 {
655     /* System generated locals */
656     integer z_dim1, z_offset, i__1, i__2;
657     doublereal d__1, d__2;
658
659     /* Local variables */
660     integer lend, jtot;
661     extern /* Subroutine */ int dlae2_(doublereal *, doublereal *, doublereal 
662             *, doublereal *, doublereal *);
663     doublereal b, c__, f, g;
664     integer i__, j, k, l, m;
665     doublereal p, r__, s;
666     extern logical lsame_(char *, char *);
667     extern /* Subroutine */ int dlasr_(char *, char *, char *, integer *, 
668             integer *, doublereal *, doublereal *, doublereal *, integer *);
669     doublereal anorm;
670     extern /* Subroutine */ int dswap_(integer *, doublereal *, integer *, 
671             doublereal *, integer *);
672     integer l1;
673     extern /* Subroutine */ int dlaev2_(doublereal *, doublereal *, 
674             doublereal *, doublereal *, doublereal *, doublereal *, 
675             doublereal *);
676     integer lendm1, lendp1;
677     extern doublereal dlapy2_(doublereal *, doublereal *);
678     integer ii;
679     extern doublereal dlamch_(char *);
680     integer mm, iscale;
681     extern /* Subroutine */ int dlascl_(char *, integer *, integer *, 
682             doublereal *, doublereal *, integer *, integer *, doublereal *, 
683             integer *, integer *), dlaset_(char *, integer *, integer 
684             *, doublereal *, doublereal *, doublereal *, integer *);
685     doublereal safmin;
686     extern /* Subroutine */ int dlartg_(doublereal *, doublereal *, 
687             doublereal *, doublereal *, doublereal *);
688     doublereal safmax;
689     extern /* Subroutine */ int xerbla_(char *, integer *, ftnlen);
690     extern doublereal dlanst_(char *, integer *, doublereal *, doublereal *);
691     extern /* Subroutine */ int dlasrt_(char *, integer *, doublereal *, 
692             integer *);
693     integer lendsv;
694     doublereal ssfmin;
695     integer nmaxit, icompz;
696     doublereal ssfmax;
697     integer lm1, mm1, nm1;
698     doublereal rt1, rt2, eps;
699     integer lsv;
700     doublereal tst, eps2;
701
702
703 /*  -- LAPACK computational routine (version 3.7.0) -- */
704 /*  -- LAPACK is a software package provided by Univ. of Tennessee,    -- */
705 /*  -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..-- */
706 /*     December 2016 */
707
708
709 /*  ===================================================================== */
710
711
712 /*     Test the input parameters. */
713
714     /* Parameter adjustments */
715     --d__;
716     --e;
717     z_dim1 = *ldz;
718     z_offset = 1 + z_dim1 * 1;
719     z__ -= z_offset;
720     --work;
721
722     /* Function Body */
723     *info = 0;
724
725     if (lsame_(compz, "N")) {
726         icompz = 0;
727     } else if (lsame_(compz, "V")) {
728         icompz = 1;
729     } else if (lsame_(compz, "I")) {
730         icompz = 2;
731     } else {
732         icompz = -1;
733     }
734     if (icompz < 0) {
735         *info = -1;
736     } else if (*n < 0) {
737         *info = -2;
738     } else if (*ldz < 1 || icompz > 0 && *ldz < f2cmax(1,*n)) {
739         *info = -6;
740     }
741     if (*info != 0) {
742         i__1 = -(*info);
743         xerbla_("DSTEQR", &i__1, (ftnlen)6);
744         return 0;
745     }
746
747 /*     Quick return if possible */
748
749     if (*n == 0) {
750         return 0;
751     }
752
753     if (*n == 1) {
754         if (icompz == 2) {
755             z__[z_dim1 + 1] = 1.;
756         }
757         return 0;
758     }
759
760 /*     Determine the unit roundoff and over/underflow thresholds. */
761
762     eps = dlamch_("E");
763 /* Computing 2nd power */
764     d__1 = eps;
765     eps2 = d__1 * d__1;
766     safmin = dlamch_("S");
767     safmax = 1. / safmin;
768     ssfmax = sqrt(safmax) / 3.;
769     ssfmin = sqrt(safmin) / eps2;
770
771 /*     Compute the eigenvalues and eigenvectors of the tridiagonal */
772 /*     matrix. */
773
774     if (icompz == 2) {
775         dlaset_("Full", n, n, &c_b9, &c_b10, &z__[z_offset], ldz);
776     }
777
778     nmaxit = *n * 30;
779     jtot = 0;
780
781 /*     Determine where the matrix splits and choose QL or QR iteration */
782 /*     for each block, according to whether top or bottom diagonal */
783 /*     element is smaller. */
784
785     l1 = 1;
786     nm1 = *n - 1;
787
788 L10:
789     if (l1 > *n) {
790         goto L160;
791     }
792     if (l1 > 1) {
793         e[l1 - 1] = 0.;
794     }
795     if (l1 <= nm1) {
796         i__1 = nm1;
797         for (m = l1; m <= i__1; ++m) {
798             tst = (d__1 = e[m], abs(d__1));
799             if (tst == 0.) {
800                 goto L30;
801             }
802             if (tst <= sqrt((d__1 = d__[m], abs(d__1))) * sqrt((d__2 = d__[m 
803                     + 1], abs(d__2))) * eps) {
804                 e[m] = 0.;
805                 goto L30;
806             }
807 /* L20: */
808         }
809     }
810     m = *n;
811
812 L30:
813     l = l1;
814     lsv = l;
815     lend = m;
816     lendsv = lend;
817     l1 = m + 1;
818     if (lend == l) {
819         goto L10;
820     }
821
822 /*     Scale submatrix in rows and columns L to LEND */
823
824     i__1 = lend - l + 1;
825     anorm = dlanst_("M", &i__1, &d__[l], &e[l]);
826     iscale = 0;
827     if (anorm == 0.) {
828         goto L10;
829     }
830     if (anorm > ssfmax) {
831         iscale = 1;
832         i__1 = lend - l + 1;
833         dlascl_("G", &c__0, &c__0, &anorm, &ssfmax, &i__1, &c__1, &d__[l], n, 
834                 info);
835         i__1 = lend - l;
836         dlascl_("G", &c__0, &c__0, &anorm, &ssfmax, &i__1, &c__1, &e[l], n, 
837                 info);
838     } else if (anorm < ssfmin) {
839         iscale = 2;
840         i__1 = lend - l + 1;
841         dlascl_("G", &c__0, &c__0, &anorm, &ssfmin, &i__1, &c__1, &d__[l], n, 
842                 info);
843         i__1 = lend - l;
844         dlascl_("G", &c__0, &c__0, &anorm, &ssfmin, &i__1, &c__1, &e[l], n, 
845                 info);
846     }
847
848 /*     Choose between QL and QR iteration */
849
850     if ((d__1 = d__[lend], abs(d__1)) < (d__2 = d__[l], abs(d__2))) {
851         lend = lsv;
852         l = lendsv;
853     }
854
855     if (lend > l) {
856
857 /*        QL Iteration */
858
859 /*        Look for small subdiagonal element. */
860
861 L40:
862         if (l != lend) {
863             lendm1 = lend - 1;
864             i__1 = lendm1;
865             for (m = l; m <= i__1; ++m) {
866 /* Computing 2nd power */
867                 d__2 = (d__1 = e[m], abs(d__1));
868                 tst = d__2 * d__2;
869                 if (tst <= eps2 * (d__1 = d__[m], abs(d__1)) * (d__2 = d__[m 
870                         + 1], abs(d__2)) + safmin) {
871                     goto L60;
872                 }
873 /* L50: */
874             }
875         }
876
877         m = lend;
878
879 L60:
880         if (m < lend) {
881             e[m] = 0.;
882         }
883         p = d__[l];
884         if (m == l) {
885             goto L80;
886         }
887
888 /*        If remaining matrix is 2-by-2, use DLAE2 or SLAEV2 */
889 /*        to compute its eigensystem. */
890
891         if (m == l + 1) {
892             if (icompz > 0) {
893                 dlaev2_(&d__[l], &e[l], &d__[l + 1], &rt1, &rt2, &c__, &s);
894                 work[l] = c__;
895                 work[*n - 1 + l] = s;
896                 dlasr_("R", "V", "B", n, &c__2, &work[l], &work[*n - 1 + l], &
897                         z__[l * z_dim1 + 1], ldz);
898             } else {
899                 dlae2_(&d__[l], &e[l], &d__[l + 1], &rt1, &rt2);
900             }
901             d__[l] = rt1;
902             d__[l + 1] = rt2;
903             e[l] = 0.;
904             l += 2;
905             if (l <= lend) {
906                 goto L40;
907             }
908             goto L140;
909         }
910
911         if (jtot == nmaxit) {
912             goto L140;
913         }
914         ++jtot;
915
916 /*        Form shift. */
917
918         g = (d__[l + 1] - p) / (e[l] * 2.);
919         r__ = dlapy2_(&g, &c_b10);
920         g = d__[m] - p + e[l] / (g + d_sign(&r__, &g));
921
922         s = 1.;
923         c__ = 1.;
924         p = 0.;
925
926 /*        Inner loop */
927
928         mm1 = m - 1;
929         i__1 = l;
930         for (i__ = mm1; i__ >= i__1; --i__) {
931             f = s * e[i__];
932             b = c__ * e[i__];
933             dlartg_(&g, &f, &c__, &s, &r__);
934             if (i__ != m - 1) {
935                 e[i__ + 1] = r__;
936             }
937             g = d__[i__ + 1] - p;
938             r__ = (d__[i__] - g) * s + c__ * 2. * b;
939             p = s * r__;
940             d__[i__ + 1] = g + p;
941             g = c__ * r__ - b;
942
943 /*           If eigenvectors are desired, then save rotations. */
944
945             if (icompz > 0) {
946                 work[i__] = c__;
947                 work[*n - 1 + i__] = -s;
948             }
949
950 /* L70: */
951         }
952
953 /*        If eigenvectors are desired, then apply saved rotations. */
954
955         if (icompz > 0) {
956             mm = m - l + 1;
957             dlasr_("R", "V", "B", n, &mm, &work[l], &work[*n - 1 + l], &z__[l 
958                     * z_dim1 + 1], ldz);
959         }
960
961         d__[l] -= p;
962         e[l] = g;
963         goto L40;
964
965 /*        Eigenvalue found. */
966
967 L80:
968         d__[l] = p;
969
970         ++l;
971         if (l <= lend) {
972             goto L40;
973         }
974         goto L140;
975
976     } else {
977
978 /*        QR Iteration */
979
980 /*        Look for small superdiagonal element. */
981
982 L90:
983         if (l != lend) {
984             lendp1 = lend + 1;
985             i__1 = lendp1;
986             for (m = l; m >= i__1; --m) {
987 /* Computing 2nd power */
988                 d__2 = (d__1 = e[m - 1], abs(d__1));
989                 tst = d__2 * d__2;
990                 if (tst <= eps2 * (d__1 = d__[m], abs(d__1)) * (d__2 = d__[m 
991                         - 1], abs(d__2)) + safmin) {
992                     goto L110;
993                 }
994 /* L100: */
995             }
996         }
997
998         m = lend;
999
1000 L110:
1001         if (m > lend) {
1002             e[m - 1] = 0.;
1003         }
1004         p = d__[l];
1005         if (m == l) {
1006             goto L130;
1007         }
1008
1009 /*        If remaining matrix is 2-by-2, use DLAE2 or SLAEV2 */
1010 /*        to compute its eigensystem. */
1011
1012         if (m == l - 1) {
1013             if (icompz > 0) {
1014                 dlaev2_(&d__[l - 1], &e[l - 1], &d__[l], &rt1, &rt2, &c__, &s)
1015                         ;
1016                 work[m] = c__;
1017                 work[*n - 1 + m] = s;
1018                 dlasr_("R", "V", "F", n, &c__2, &work[m], &work[*n - 1 + m], &
1019                         z__[(l - 1) * z_dim1 + 1], ldz);
1020             } else {
1021                 dlae2_(&d__[l - 1], &e[l - 1], &d__[l], &rt1, &rt2);
1022             }
1023             d__[l - 1] = rt1;
1024             d__[l] = rt2;
1025             e[l - 1] = 0.;
1026             l += -2;
1027             if (l >= lend) {
1028                 goto L90;
1029             }
1030             goto L140;
1031         }
1032
1033         if (jtot == nmaxit) {
1034             goto L140;
1035         }
1036         ++jtot;
1037
1038 /*        Form shift. */
1039
1040         g = (d__[l - 1] - p) / (e[l - 1] * 2.);
1041         r__ = dlapy2_(&g, &c_b10);
1042         g = d__[m] - p + e[l - 1] / (g + d_sign(&r__, &g));
1043
1044         s = 1.;
1045         c__ = 1.;
1046         p = 0.;
1047
1048 /*        Inner loop */
1049
1050         lm1 = l - 1;
1051         i__1 = lm1;
1052         for (i__ = m; i__ <= i__1; ++i__) {
1053             f = s * e[i__];
1054             b = c__ * e[i__];
1055             dlartg_(&g, &f, &c__, &s, &r__);
1056             if (i__ != m) {
1057                 e[i__ - 1] = r__;
1058             }
1059             g = d__[i__] - p;
1060             r__ = (d__[i__ + 1] - g) * s + c__ * 2. * b;
1061             p = s * r__;
1062             d__[i__] = g + p;
1063             g = c__ * r__ - b;
1064
1065 /*           If eigenvectors are desired, then save rotations. */
1066
1067             if (icompz > 0) {
1068                 work[i__] = c__;
1069                 work[*n - 1 + i__] = s;
1070             }
1071
1072 /* L120: */
1073         }
1074
1075 /*        If eigenvectors are desired, then apply saved rotations. */
1076
1077         if (icompz > 0) {
1078             mm = l - m + 1;
1079             dlasr_("R", "V", "F", n, &mm, &work[m], &work[*n - 1 + m], &z__[m 
1080                     * z_dim1 + 1], ldz);
1081         }
1082
1083         d__[l] -= p;
1084         e[lm1] = g;
1085         goto L90;
1086
1087 /*        Eigenvalue found. */
1088
1089 L130:
1090         d__[l] = p;
1091
1092         --l;
1093         if (l >= lend) {
1094             goto L90;
1095         }
1096         goto L140;
1097
1098     }
1099
1100 /*     Undo scaling if necessary */
1101
1102 L140:
1103     if (iscale == 1) {
1104         i__1 = lendsv - lsv + 1;
1105         dlascl_("G", &c__0, &c__0, &ssfmax, &anorm, &i__1, &c__1, &d__[lsv], 
1106                 n, info);
1107         i__1 = lendsv - lsv;
1108         dlascl_("G", &c__0, &c__0, &ssfmax, &anorm, &i__1, &c__1, &e[lsv], n, 
1109                 info);
1110     } else if (iscale == 2) {
1111         i__1 = lendsv - lsv + 1;
1112         dlascl_("G", &c__0, &c__0, &ssfmin, &anorm, &i__1, &c__1, &d__[lsv], 
1113                 n, info);
1114         i__1 = lendsv - lsv;
1115         dlascl_("G", &c__0, &c__0, &ssfmin, &anorm, &i__1, &c__1, &e[lsv], n, 
1116                 info);
1117     }
1118
1119 /*     Check for no convergence to an eigenvalue after a total */
1120 /*     of N*MAXIT iterations. */
1121
1122     if (jtot < nmaxit) {
1123         goto L10;
1124     }
1125     i__1 = *n - 1;
1126     for (i__ = 1; i__ <= i__1; ++i__) {
1127         if (e[i__] != 0.) {
1128             ++(*info);
1129         }
1130 /* L150: */
1131     }
1132     goto L190;
1133
1134 /*     Order eigenvalues and eigenvectors. */
1135
1136 L160:
1137     if (icompz == 0) {
1138
1139 /*        Use Quick Sort */
1140
1141         dlasrt_("I", n, &d__[1], info);
1142
1143     } else {
1144
1145 /*        Use Selection Sort to minimize swaps of eigenvectors */
1146
1147         i__1 = *n;
1148         for (ii = 2; ii <= i__1; ++ii) {
1149             i__ = ii - 1;
1150             k = i__;
1151             p = d__[i__];
1152             i__2 = *n;
1153             for (j = ii; j <= i__2; ++j) {
1154                 if (d__[j] < p) {
1155                     k = j;
1156                     p = d__[j];
1157                 }
1158 /* L170: */
1159             }
1160             if (k != i__) {
1161                 d__[k] = d__[i__];
1162                 d__[i__] = p;
1163                 dswap_(n, &z__[i__ * z_dim1 + 1], &c__1, &z__[k * z_dim1 + 1],
1164                          &c__1);
1165             }
1166 /* L180: */
1167         }
1168     }
1169
1170 L190:
1171     return 0;
1172
1173 /*     End of DSTEQR */
1174
1175 } /* dsteqr_ */
1176