338 lines
13 KiB
C++
338 lines
13 KiB
C++
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
#include "lmp_f2c.h"
|
|
static integer c__1 = 1;
|
|
static doublereal c_b8 = -1.;
|
|
static doublereal c_b9 = 1.;
|
|
int dlasyf_(char *uplo, integer *n, integer *nb, integer *kb, doublereal *a, integer *lda,
|
|
integer *ipiv, doublereal *w, integer *ldw, integer *info, ftnlen uplo_len)
|
|
{
|
|
integer a_dim1, a_offset, w_dim1, w_offset, i__1, i__2, i__3, i__4, i__5;
|
|
doublereal d__1, d__2, d__3;
|
|
double sqrt(doublereal);
|
|
integer j, k;
|
|
doublereal t, r1, d11, d21, d22;
|
|
integer jb, jj, kk, jp, kp, kw, kkw, imax, jmax;
|
|
doublereal alpha;
|
|
extern int dscal_(integer *, doublereal *, doublereal *, integer *),
|
|
dgemm_(char *, char *, integer *, integer *, integer *, doublereal *, doublereal *,
|
|
integer *, doublereal *, integer *, doublereal *, doublereal *, integer *, ftnlen,
|
|
ftnlen);
|
|
extern logical lsame_(char *, char *, ftnlen, ftnlen);
|
|
extern int dgemv_(char *, integer *, integer *, doublereal *, doublereal *, integer *,
|
|
doublereal *, integer *, doublereal *, doublereal *, integer *, ftnlen),
|
|
dcopy_(integer *, doublereal *, integer *, doublereal *, integer *),
|
|
dswap_(integer *, doublereal *, integer *, doublereal *, integer *);
|
|
integer kstep;
|
|
doublereal absakk;
|
|
extern integer idamax_(integer *, doublereal *, integer *);
|
|
doublereal colmax, rowmax;
|
|
a_dim1 = *lda;
|
|
a_offset = 1 + a_dim1;
|
|
a -= a_offset;
|
|
--ipiv;
|
|
w_dim1 = *ldw;
|
|
w_offset = 1 + w_dim1;
|
|
w -= w_offset;
|
|
*info = 0;
|
|
alpha = (sqrt(17.) + 1.) / 8.;
|
|
if (lsame_(uplo, (char *)"U", (ftnlen)1, (ftnlen)1)) {
|
|
k = *n;
|
|
L10:
|
|
kw = *nb + k - *n;
|
|
if (k <= *n - *nb + 1 && *nb < *n || k < 1) {
|
|
goto L30;
|
|
}
|
|
dcopy_(&k, &a[k * a_dim1 + 1], &c__1, &w[kw * w_dim1 + 1], &c__1);
|
|
if (k < *n) {
|
|
i__1 = *n - k;
|
|
dgemv_((char *)"No transpose", &k, &i__1, &c_b8, &a[(k + 1) * a_dim1 + 1], lda,
|
|
&w[k + (kw + 1) * w_dim1], ldw, &c_b9, &w[kw * w_dim1 + 1], &c__1, (ftnlen)12);
|
|
}
|
|
kstep = 1;
|
|
absakk = (d__1 = w[k + kw * w_dim1], abs(d__1));
|
|
if (k > 1) {
|
|
i__1 = k - 1;
|
|
imax = idamax_(&i__1, &w[kw * w_dim1 + 1], &c__1);
|
|
colmax = (d__1 = w[imax + kw * w_dim1], abs(d__1));
|
|
} else {
|
|
colmax = 0.;
|
|
}
|
|
if (max(absakk, colmax) == 0.) {
|
|
if (*info == 0) {
|
|
*info = k;
|
|
}
|
|
kp = k;
|
|
} else {
|
|
if (absakk >= alpha * colmax) {
|
|
kp = k;
|
|
} else {
|
|
dcopy_(&imax, &a[imax * a_dim1 + 1], &c__1, &w[(kw - 1) * w_dim1 + 1], &c__1);
|
|
i__1 = k - imax;
|
|
dcopy_(&i__1, &a[imax + (imax + 1) * a_dim1], lda, &w[imax + 1 + (kw - 1) * w_dim1],
|
|
&c__1);
|
|
if (k < *n) {
|
|
i__1 = *n - k;
|
|
dgemv_((char *)"No transpose", &k, &i__1, &c_b8, &a[(k + 1) * a_dim1 + 1], lda,
|
|
&w[imax + (kw + 1) * w_dim1], ldw, &c_b9, &w[(kw - 1) * w_dim1 + 1],
|
|
&c__1, (ftnlen)12);
|
|
}
|
|
i__1 = k - imax;
|
|
jmax = imax + idamax_(&i__1, &w[imax + 1 + (kw - 1) * w_dim1], &c__1);
|
|
rowmax = (d__1 = w[jmax + (kw - 1) * w_dim1], abs(d__1));
|
|
if (imax > 1) {
|
|
i__1 = imax - 1;
|
|
jmax = idamax_(&i__1, &w[(kw - 1) * w_dim1 + 1], &c__1);
|
|
d__2 = rowmax, d__3 = (d__1 = w[jmax + (kw - 1) * w_dim1], abs(d__1));
|
|
rowmax = max(d__2, d__3);
|
|
}
|
|
if (absakk >= alpha * colmax * (colmax / rowmax)) {
|
|
kp = k;
|
|
} else if ((d__1 = w[imax + (kw - 1) * w_dim1], abs(d__1)) >= alpha * rowmax) {
|
|
kp = imax;
|
|
dcopy_(&k, &w[(kw - 1) * w_dim1 + 1], &c__1, &w[kw * w_dim1 + 1], &c__1);
|
|
} else {
|
|
kp = imax;
|
|
kstep = 2;
|
|
}
|
|
}
|
|
kk = k - kstep + 1;
|
|
kkw = *nb + kk - *n;
|
|
if (kp != kk) {
|
|
a[kp + kp * a_dim1] = a[kk + kk * a_dim1];
|
|
i__1 = kk - 1 - kp;
|
|
dcopy_(&i__1, &a[kp + 1 + kk * a_dim1], &c__1, &a[kp + (kp + 1) * a_dim1], lda);
|
|
if (kp > 1) {
|
|
i__1 = kp - 1;
|
|
dcopy_(&i__1, &a[kk * a_dim1 + 1], &c__1, &a[kp * a_dim1 + 1], &c__1);
|
|
}
|
|
if (k < *n) {
|
|
i__1 = *n - k;
|
|
dswap_(&i__1, &a[kk + (k + 1) * a_dim1], lda, &a[kp + (k + 1) * a_dim1], lda);
|
|
}
|
|
i__1 = *n - kk + 1;
|
|
dswap_(&i__1, &w[kk + kkw * w_dim1], ldw, &w[kp + kkw * w_dim1], ldw);
|
|
}
|
|
if (kstep == 1) {
|
|
dcopy_(&k, &w[kw * w_dim1 + 1], &c__1, &a[k * a_dim1 + 1], &c__1);
|
|
r1 = 1. / a[k + k * a_dim1];
|
|
i__1 = k - 1;
|
|
dscal_(&i__1, &r1, &a[k * a_dim1 + 1], &c__1);
|
|
} else {
|
|
if (k > 2) {
|
|
d21 = w[k - 1 + kw * w_dim1];
|
|
d11 = w[k + kw * w_dim1] / d21;
|
|
d22 = w[k - 1 + (kw - 1) * w_dim1] / d21;
|
|
t = 1. / (d11 * d22 - 1.);
|
|
d21 = t / d21;
|
|
i__1 = k - 2;
|
|
for (j = 1; j <= i__1; ++j) {
|
|
a[j + (k - 1) * a_dim1] =
|
|
d21 * (d11 * w[j + (kw - 1) * w_dim1] - w[j + kw * w_dim1]);
|
|
a[j + k * a_dim1] =
|
|
d21 * (d22 * w[j + kw * w_dim1] - w[j + (kw - 1) * w_dim1]);
|
|
}
|
|
}
|
|
a[k - 1 + (k - 1) * a_dim1] = w[k - 1 + (kw - 1) * w_dim1];
|
|
a[k - 1 + k * a_dim1] = w[k - 1 + kw * w_dim1];
|
|
a[k + k * a_dim1] = w[k + kw * w_dim1];
|
|
}
|
|
}
|
|
if (kstep == 1) {
|
|
ipiv[k] = kp;
|
|
} else {
|
|
ipiv[k] = -kp;
|
|
ipiv[k - 1] = -kp;
|
|
}
|
|
k -= kstep;
|
|
goto L10;
|
|
L30:
|
|
i__1 = -(*nb);
|
|
for (j = (k - 1) / *nb * *nb + 1; i__1 < 0 ? j >= 1 : j <= 1; j += i__1) {
|
|
i__2 = *nb, i__3 = k - j + 1;
|
|
jb = min(i__2, i__3);
|
|
i__2 = j + jb - 1;
|
|
for (jj = j; jj <= i__2; ++jj) {
|
|
i__3 = jj - j + 1;
|
|
i__4 = *n - k;
|
|
dgemv_((char *)"No transpose", &i__3, &i__4, &c_b8, &a[j + (k + 1) * a_dim1], lda,
|
|
&w[jj + (kw + 1) * w_dim1], ldw, &c_b9, &a[j + jj * a_dim1], &c__1,
|
|
(ftnlen)12);
|
|
}
|
|
i__2 = j - 1;
|
|
i__3 = *n - k;
|
|
dgemm_((char *)"No transpose", (char *)"Transpose", &i__2, &jb, &i__3, &c_b8, &a[(k + 1) * a_dim1 + 1],
|
|
lda, &w[j + (kw + 1) * w_dim1], ldw, &c_b9, &a[j * a_dim1 + 1], lda, (ftnlen)12,
|
|
(ftnlen)9);
|
|
}
|
|
j = k + 1;
|
|
L60:
|
|
jj = j;
|
|
jp = ipiv[j];
|
|
if (jp < 0) {
|
|
jp = -jp;
|
|
++j;
|
|
}
|
|
++j;
|
|
if (jp != jj && j <= *n) {
|
|
i__1 = *n - j + 1;
|
|
dswap_(&i__1, &a[jp + j * a_dim1], lda, &a[jj + j * a_dim1], lda);
|
|
}
|
|
if (j < *n) {
|
|
goto L60;
|
|
}
|
|
*kb = *n - k;
|
|
} else {
|
|
k = 1;
|
|
L70:
|
|
if (k >= *nb && *nb < *n || k > *n) {
|
|
goto L90;
|
|
}
|
|
i__1 = *n - k + 1;
|
|
dcopy_(&i__1, &a[k + k * a_dim1], &c__1, &w[k + k * w_dim1], &c__1);
|
|
i__1 = *n - k + 1;
|
|
i__2 = k - 1;
|
|
dgemv_((char *)"No transpose", &i__1, &i__2, &c_b8, &a[k + a_dim1], lda, &w[k + w_dim1], ldw, &c_b9,
|
|
&w[k + k * w_dim1], &c__1, (ftnlen)12);
|
|
kstep = 1;
|
|
absakk = (d__1 = w[k + k * w_dim1], abs(d__1));
|
|
if (k < *n) {
|
|
i__1 = *n - k;
|
|
imax = k + idamax_(&i__1, &w[k + 1 + k * w_dim1], &c__1);
|
|
colmax = (d__1 = w[imax + k * w_dim1], abs(d__1));
|
|
} else {
|
|
colmax = 0.;
|
|
}
|
|
if (max(absakk, colmax) == 0.) {
|
|
if (*info == 0) {
|
|
*info = k;
|
|
}
|
|
kp = k;
|
|
} else {
|
|
if (absakk >= alpha * colmax) {
|
|
kp = k;
|
|
} else {
|
|
i__1 = imax - k;
|
|
dcopy_(&i__1, &a[imax + k * a_dim1], lda, &w[k + (k + 1) * w_dim1], &c__1);
|
|
i__1 = *n - imax + 1;
|
|
dcopy_(&i__1, &a[imax + imax * a_dim1], &c__1, &w[imax + (k + 1) * w_dim1], &c__1);
|
|
i__1 = *n - k + 1;
|
|
i__2 = k - 1;
|
|
dgemv_((char *)"No transpose", &i__1, &i__2, &c_b8, &a[k + a_dim1], lda, &w[imax + w_dim1],
|
|
ldw, &c_b9, &w[k + (k + 1) * w_dim1], &c__1, (ftnlen)12);
|
|
i__1 = imax - k;
|
|
jmax = k - 1 + idamax_(&i__1, &w[k + (k + 1) * w_dim1], &c__1);
|
|
rowmax = (d__1 = w[jmax + (k + 1) * w_dim1], abs(d__1));
|
|
if (imax < *n) {
|
|
i__1 = *n - imax;
|
|
jmax = imax + idamax_(&i__1, &w[imax + 1 + (k + 1) * w_dim1], &c__1);
|
|
d__2 = rowmax, d__3 = (d__1 = w[jmax + (k + 1) * w_dim1], abs(d__1));
|
|
rowmax = max(d__2, d__3);
|
|
}
|
|
if (absakk >= alpha * colmax * (colmax / rowmax)) {
|
|
kp = k;
|
|
} else if ((d__1 = w[imax + (k + 1) * w_dim1], abs(d__1)) >= alpha * rowmax) {
|
|
kp = imax;
|
|
i__1 = *n - k + 1;
|
|
dcopy_(&i__1, &w[k + (k + 1) * w_dim1], &c__1, &w[k + k * w_dim1], &c__1);
|
|
} else {
|
|
kp = imax;
|
|
kstep = 2;
|
|
}
|
|
}
|
|
kk = k + kstep - 1;
|
|
if (kp != kk) {
|
|
a[kp + kp * a_dim1] = a[kk + kk * a_dim1];
|
|
i__1 = kp - kk - 1;
|
|
dcopy_(&i__1, &a[kk + 1 + kk * a_dim1], &c__1, &a[kp + (kk + 1) * a_dim1], lda);
|
|
if (kp < *n) {
|
|
i__1 = *n - kp;
|
|
dcopy_(&i__1, &a[kp + 1 + kk * a_dim1], &c__1, &a[kp + 1 + kp * a_dim1], &c__1);
|
|
}
|
|
if (k > 1) {
|
|
i__1 = k - 1;
|
|
dswap_(&i__1, &a[kk + a_dim1], lda, &a[kp + a_dim1], lda);
|
|
}
|
|
dswap_(&kk, &w[kk + w_dim1], ldw, &w[kp + w_dim1], ldw);
|
|
}
|
|
if (kstep == 1) {
|
|
i__1 = *n - k + 1;
|
|
dcopy_(&i__1, &w[k + k * w_dim1], &c__1, &a[k + k * a_dim1], &c__1);
|
|
if (k < *n) {
|
|
r1 = 1. / a[k + k * a_dim1];
|
|
i__1 = *n - k;
|
|
dscal_(&i__1, &r1, &a[k + 1 + k * a_dim1], &c__1);
|
|
}
|
|
} else {
|
|
if (k < *n - 1) {
|
|
d21 = w[k + 1 + k * w_dim1];
|
|
d11 = w[k + 1 + (k + 1) * w_dim1] / d21;
|
|
d22 = w[k + k * w_dim1] / d21;
|
|
t = 1. / (d11 * d22 - 1.);
|
|
d21 = t / d21;
|
|
i__1 = *n;
|
|
for (j = k + 2; j <= i__1; ++j) {
|
|
a[j + k * a_dim1] =
|
|
d21 * (d11 * w[j + k * w_dim1] - w[j + (k + 1) * w_dim1]);
|
|
a[j + (k + 1) * a_dim1] =
|
|
d21 * (d22 * w[j + (k + 1) * w_dim1] - w[j + k * w_dim1]);
|
|
}
|
|
}
|
|
a[k + k * a_dim1] = w[k + k * w_dim1];
|
|
a[k + 1 + k * a_dim1] = w[k + 1 + k * w_dim1];
|
|
a[k + 1 + (k + 1) * a_dim1] = w[k + 1 + (k + 1) * w_dim1];
|
|
}
|
|
}
|
|
if (kstep == 1) {
|
|
ipiv[k] = kp;
|
|
} else {
|
|
ipiv[k] = -kp;
|
|
ipiv[k + 1] = -kp;
|
|
}
|
|
k += kstep;
|
|
goto L70;
|
|
L90:
|
|
i__1 = *n;
|
|
i__2 = *nb;
|
|
for (j = k; i__2 < 0 ? j >= i__1 : j <= i__1; j += i__2) {
|
|
i__3 = *nb, i__4 = *n - j + 1;
|
|
jb = min(i__3, i__4);
|
|
i__3 = j + jb - 1;
|
|
for (jj = j; jj <= i__3; ++jj) {
|
|
i__4 = j + jb - jj;
|
|
i__5 = k - 1;
|
|
dgemv_((char *)"No transpose", &i__4, &i__5, &c_b8, &a[jj + a_dim1], lda, &w[jj + w_dim1],
|
|
ldw, &c_b9, &a[jj + jj * a_dim1], &c__1, (ftnlen)12);
|
|
}
|
|
if (j + jb <= *n) {
|
|
i__3 = *n - j - jb + 1;
|
|
i__4 = k - 1;
|
|
dgemm_((char *)"No transpose", (char *)"Transpose", &i__3, &jb, &i__4, &c_b8, &a[j + jb + a_dim1],
|
|
lda, &w[j + w_dim1], ldw, &c_b9, &a[j + jb + j * a_dim1], lda, (ftnlen)12,
|
|
(ftnlen)9);
|
|
}
|
|
}
|
|
j = k - 1;
|
|
L120:
|
|
jj = j;
|
|
jp = ipiv[j];
|
|
if (jp < 0) {
|
|
jp = -jp;
|
|
--j;
|
|
}
|
|
--j;
|
|
if (jp != jj && j >= 1) {
|
|
dswap_(&j, &a[jp + a_dim1], lda, &a[jj + a_dim1], lda);
|
|
}
|
|
if (j > 1) {
|
|
goto L120;
|
|
}
|
|
*kb = k - 1;
|
|
}
|
|
return 0;
|
|
}
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|