src/gromacs/linearalgebra/gmx_lapack/slasq3.c

   1 #include <math.h>
   2 #include "types/simple.h"
   3
   4 #include "../gmx_lapack.h"
   5 #include "lapack_limits.h"
   6
   7 void
   8 F77_FUNC(slasq3,SLASQ3)(int *i0,
   9                         int *n0,
  10                         float *z__,
  11                         int *pp,
  12                         float *dmin__,
  13                         float *sigma,
  14                         float *desig,
  15                         float *qmax,
  16                         int *nfail,
  17                         int *iter,
  18                         int *ndiv,
  19         int *ieee)
  20 {
  21
  22     int ttype = 0;
  23     float dmin1 = 0.;
  24     float dmin2 = 0.;
  25     float dn = 0.;
  26     float dn1 = 0.;
  27     float dn2 = 0.;
  28     float tau = 0.;
  29
  30     int i__1;
  31     float d__1, d__2;
  32     float s, t;
  33     int j4, nn;
  34     float eps, tol;
  35     int n0in, ipn4;
  36     float tol2, temp;
  37     --z__;
  38
  39     n0in = *n0;
  40     eps = GMX_FLOAT_EPS;
  41     tol = eps * 100.;
  42     d__1 = tol;
  43     tol2 = d__1 * d__1;
  44
  45
  46 L10:
  47
  48     if (*n0 < *i0) {
  49         return;
  50     }
  51     if (*n0 == *i0) {
  52         goto L20;
  53     }
  54     nn = (*n0 << 2) + *pp;
  55     if (*n0 == *i0 + 1) {
  56         goto L40;
  57     }
  58
  59     if (z__[nn - 5] > tol2 * (*sigma + z__[nn - 3]) && z__[nn - (*pp << 1) -
  60             4] > tol2 * z__[nn - 7]) {
  61         goto L30;
  62     }
  63
  64 L20:
  65
  66     z__[(*n0 << 2) - 3] = z__[(*n0 << 2) + *pp - 3] + *sigma;
  67     --(*n0);
  68     goto L10;
  69
  70 L30:
  71
  72     if (z__[nn - 9] > tol2 * *sigma && z__[nn - (*pp << 1) - 8] > tol2 * z__[
  73             nn - 11]) {
  74         goto L50;
  75     }
  76
  77 L40:
  78
  79     if (z__[nn - 3] > z__[nn - 7]) {
  80         s = z__[nn - 3];
  81         z__[nn - 3] = z__[nn - 7];
  82         z__[nn - 7] = s;
  83     }
  84     if (z__[nn - 5] > z__[nn - 3] * tol2) {
  85         t = (z__[nn - 7] - z__[nn - 3] + z__[nn - 5]) * .5;
  86         s = z__[nn - 3] * (z__[nn - 5] / t);
  87         if (s <= t) {
  88             s = z__[nn - 3] * (z__[nn - 5] / (t * (sqrt(s / t + 1.) + 1.)));
  89         } else {
  90             s = z__[nn - 3] * (z__[nn - 5] / (t + sqrt(t) * sqrt(t + s)));
  91         }
  92         t = z__[nn - 7] + (s + z__[nn - 5]);
  93         z__[nn - 3] *= z__[nn - 7] / t;
  94         z__[nn - 7] = t;
  95     }
  96     z__[(*n0 << 2) - 7] = z__[nn - 7] + *sigma;
  97     z__[(*n0 << 2) - 3] = z__[nn - 3] + *sigma;
  98     *n0 += -2;
  99     goto L10;
 100
 101 L50:
 102     if (*pp == 2) {
 103         *pp = 0;
 104     }
 105
 106     if (*dmin__ <= 0. || *n0 < n0in) {
 107         if (z__[(*i0 << 2) + *pp - 3] * 1.5 < z__[(*n0 << 2) + *pp - 3]) {
 108             ipn4 = 4*(*i0 + *n0);
 109             i__1 = 2*(*i0 + *n0 - 1);
 110             for (j4 = *i0 << 2; j4 <= i__1; j4 += 4) {
 111                 temp = z__[j4 - 3];
 112                 z__[j4 - 3] = z__[ipn4 - j4 - 3];
 113                 z__[ipn4 - j4 - 3] = temp;
 114                 temp = z__[j4 - 2];
 115                 z__[j4 - 2] = z__[ipn4 - j4 - 2];
 116                 z__[ipn4 - j4 - 2] = temp;
 117                 temp = z__[j4 - 1];
 118                 z__[j4 - 1] = z__[ipn4 - j4 - 5];
 119                 z__[ipn4 - j4 - 5] = temp;
 120                 temp = z__[j4];
 121                 z__[j4] = z__[ipn4 - j4 - 4];
 122                 z__[ipn4 - j4 - 4] = temp;
 123             }
 124             if (*n0 - *i0 <= 4) {
 125                 z__[(*n0 << 2) + *pp - 1] = z__[(*i0 << 2) + *pp - 1];
 126                 z__[(*n0 << 2) - *pp] = z__[(*i0 << 2) - *pp];
 127             }
 128             d__1 = dmin2, d__2 = z__[(*n0 << 2) + *pp - 1];
 129             dmin2 = ((d__1<d__2) ? d__1 : d__2);
 130             d__1 = z__[(*n0 << 2) + *pp - 1], d__2 = z__[(*i0 << 2) + *pp - 1]
 131                     , d__1 = ((d__1<d__2) ? d__1 : d__2), d__2 = z__[(*i0 << 2) + *pp + 3];
 132             z__[(*n0 << 2) + *pp - 1] = ((d__1<d__2) ? d__1 : d__2);
 133             d__1 = z__[(*n0 << 2) - *pp], d__2 = z__[(*i0 << 2) - *pp], d__1 =
 134                      ((d__1<d__2) ? d__1 : d__2), d__2 = z__[(*i0 << 2) - *pp + 4];
 135             z__[(*n0 << 2) - *pp] = ((d__1<d__2) ? d__1 : d__2);
 136             d__1 = *qmax;
 137             d__2 = z__[(*i0 << 2) + *pp - 3];
 138             d__1 = (d__1>d__2) ? d__1 : d__2;
 139             d__2 = z__[(*i0 << 2) + *pp + 1];
 140             *qmax = ((d__1>d__2) ? d__1 : d__2);
 141             *dmin__ = -0.;
 142         }
 143     }
 144
 145
 146     F77_FUNC(slasq4,SLASQ4)(i0, n0, &z__[1], pp, &n0in, dmin__, &dmin1, &dmin2, &dn, &dn1, &
 147             dn2, &tau, &ttype);
 148
 149 L70:
 150
 151     F77_FUNC(slasq5,SLASQ5)(i0, n0, &z__[1], pp, &tau, dmin__, &dmin1, &dmin2, &dn, &dn1, &
 152             dn2, ieee);
 153
 154     *ndiv += *n0 - *i0 + 2;
 155     ++(*iter);
 156
 157     if (*dmin__ >= 0. && dmin1 > 0.) {
 158
 159         goto L90;
 160
 161     } else if (*dmin__ < 0. && dmin1 > 0. && z__[4*(*n0 - 1) - *pp] < tol *
 162              (*sigma + dn1) && fabs(dn) < tol * *sigma) {
 163
 164         z__[4*(*n0 - 1) - *pp + 2] = 0.;
 165         *dmin__ = 0.;
 166         goto L90;
 167     } else if (*dmin__ < 0.) {
 168
 169         ++(*nfail);
 170         if (ttype < -22) {
 171
 172             tau = 0.;
 173         } else if (dmin1 > 0.) {
 174
 175             tau = (tau + *dmin__) * (1. - eps * 2.);
 176             ttype += -11;
 177         } else {
 178
 179             tau *= .25;
 180             ttype += -12;
 181         }
 182         goto L70;
 183     }
 184     else {
 185
 186         goto L80;
 187     }
 188
 189 L80:
 190     F77_FUNC(slasq6,SLASQ6)(i0, n0, &z__[1], pp, dmin__, &dmin1, &dmin2, &dn, &dn1, &dn2);
 191     *ndiv += *n0 - *i0 + 2;
 192     ++(*iter);
 193     tau = 0.;
 194
 195 L90:
 196     if (tau < *sigma) {
 197         *desig += tau;
 198         t = *sigma + *desig;
 199         *desig -= t - *sigma;
 200     } else {
 201         t = *sigma + tau;
 202         *desig = *sigma - (t - tau) + *desig;
 203     }
 204     *sigma = t;
 205
 206     return;
 207 }