diff --git a/lib/linalg/dlarf1f.cpp b/lib/linalg/dlarf1f.cpp new file mode 100644 index 0000000000..2b94d7d472 --- /dev/null +++ b/lib/linalg/dlarf1f.cpp @@ -0,0 +1,95 @@ +#ifdef __cplusplus +extern "C" { +#endif +#include "lmp_f2c.h" +static doublereal c_b4 = 1.; +static doublereal c_b5 = 0.; +static integer c__1 = 1; +int dlarf1f_(char *side, integer *m, integer *n, doublereal *v, integer *incv, doublereal *tau, + doublereal *c__, integer *ldc, doublereal *work, ftnlen side_len) +{ + integer c_dim1, c_offset, i__1; + doublereal d__1; + integer i__; + logical applyleft; + extern int dger_(integer *, integer *, doublereal *, doublereal *, integer *, doublereal *, + integer *, doublereal *, integer *), + dscal_(integer *, doublereal *, doublereal *, integer *); + extern logical lsame_(char *, char *, ftnlen, ftnlen); + extern int dgemv_(char *, integer *, integer *, doublereal *, doublereal *, integer *, + doublereal *, integer *, doublereal *, doublereal *, integer *, ftnlen); + integer lastc; + extern int daxpy_(integer *, doublereal *, doublereal *, integer *, doublereal *, integer *); + integer lastv; + extern integer iladlc_(integer *, integer *, doublereal *, integer *), + iladlr_(integer *, integer *, doublereal *, integer *); + --v; + c_dim1 = *ldc; + c_offset = 1 + c_dim1; + c__ -= c_offset; + --work; + applyleft = lsame_(side, (char *)"L", (ftnlen)1, (ftnlen)1); + lastv = 1; + lastc = 0; + if (*tau != 0.) { + if (applyleft) { + lastv = *m; + } else { + lastv = *n; + } + if (*incv > 0) { + i__ = (lastv - 1) * *incv + 1; + } else { + i__ = 1; + } + while (lastv > 1 && v[i__] == 0.) { + --lastv; + i__ -= *incv; + } + if (applyleft) { + lastc = iladlc_(&lastv, n, &c__[c_offset], ldc); + } else { + lastc = iladlr_(m, &lastv, &c__[c_offset], ldc); + } + } + if (lastc == 0) { + return 0; + } + if (applyleft) { + if (lastv == 1) { + d__1 = 1. - *tau; + dscal_(&lastc, &d__1, &c__[c_offset], ldc); + } else { + i__1 = lastv - 1; + dgemv_((char *)"T", &i__1, &lastc, &c_b4, &c__[c_dim1 + 2], ldc, &v[*incv + 1], incv, &c_b5, + &work[1], &c__1, (ftnlen)1); + daxpy_(&lastc, &c_b4, &c__[c_offset], ldc, &work[1], &c__1); + d__1 = -(*tau); + daxpy_(&lastc, &d__1, &work[1], &c__1, &c__[c_offset], ldc); + i__1 = lastv - 1; + d__1 = -(*tau); + dger_(&i__1, &lastc, &d__1, &v[*incv + 1], incv, &work[1], &c__1, &c__[c_dim1 + 2], + ldc); + } + } else { + if (lastv == 1) { + d__1 = 1. - *tau; + dscal_(&lastc, &d__1, &c__[c_offset], &c__1); + } else { + i__1 = lastv - 1; + dgemv_((char *)"N", &lastc, &i__1, &c_b4, &c__[(c_dim1 << 1) + 1], ldc, &v[*incv + 1], incv, + &c_b5, &work[1], &c__1, (ftnlen)1); + daxpy_(&lastc, &c_b4, &c__[c_offset], &c__1, &work[1], &c__1); + d__1 = -(*tau); + daxpy_(&lastc, &d__1, &work[1], &c__1, &c__[c_offset], &c__1); + i__1 = lastv - 1; + d__1 = -(*tau); + dger_(&lastc, &i__1, &d__1, &work[1], &c__1, &v[*incv + 1], incv, + &c__[(c_dim1 << 1) + 1], ldc); + } + } + return 0; +} +#ifdef __cplusplus +} +#endif diff --git a/lib/linalg/zlarf1f.cpp b/lib/linalg/zlarf1f.cpp new file mode 100644 index 0000000000..3a1362de40 --- /dev/null +++ b/lib/linalg/zlarf1f.cpp @@ -0,0 +1,115 @@ +#ifdef __cplusplus +extern "C" { +#endif +#include "lmp_f2c.h" +static doublecomplex c_b1 = {1., 0.}; +static doublecomplex c_b2 = {0., 0.}; +static integer c__1 = 1; +int zlarf1f_(char *side, integer *m, integer *n, doublecomplex *v, integer *incv, + doublecomplex *tau, doublecomplex *c__, integer *ldc, doublecomplex *work, + ftnlen side_len) +{ + integer c_dim1, c_offset, i__1, i__2, i__3; + doublecomplex z__1, z__2, z__3; + void d_lmp_cnjg(doublecomplex *, doublecomplex *); + integer i__; + logical applyleft; + extern logical lsame_(char *, char *, ftnlen, ftnlen); + integer lastc; + extern int zgerc_(integer *, integer *, doublecomplex *, doublecomplex *, integer *, + doublecomplex *, integer *, doublecomplex *, integer *), + zscal_(integer *, doublecomplex *, doublecomplex *, integer *), + zgemv_(char *, integer *, integer *, doublecomplex *, doublecomplex *, integer *, + doublecomplex *, integer *, doublecomplex *, doublecomplex *, integer *, ftnlen); + integer lastv; + extern int zaxpy_(integer *, doublecomplex *, doublecomplex *, integer *, doublecomplex *, + integer *); + extern integer ilazlc_(integer *, integer *, doublecomplex *, integer *), + ilazlr_(integer *, integer *, doublecomplex *, integer *); + --v; + c_dim1 = *ldc; + c_offset = 1 + c_dim1; + c__ -= c_offset; + --work; + applyleft = lsame_(side, (char *)"L", (ftnlen)1, (ftnlen)1); + lastv = 1; + lastc = 0; + if (tau->r != 0. || tau->i != 0.) { + if (applyleft) { + lastv = *m; + } else { + lastv = *n; + } + if (*incv > 0) { + i__ = (lastv - 1) * *incv + 1; + } else { + i__ = 1; + } + for (;;) { + i__1 = i__; + if (!(lastv > 1 && (v[i__1].r == 0. && v[i__1].i == 0.))) break; + --lastv; + i__ -= *incv; + } + if (applyleft) { + lastc = ilazlc_(&lastv, n, &c__[c_offset], ldc); + } else { + lastc = ilazlr_(m, &lastv, &c__[c_offset], ldc); + } + } + if (lastc == 0) { + return 0; + } + if (applyleft) { + if (lastv == 1) { + z__1.r = 1. - tau->r, z__1.i = 0. - tau->i; + zscal_(&lastc, &z__1, &c__[c_offset], ldc); + } else { + i__1 = lastv - 1; + zgemv_((char *)"C", &i__1, &lastc, &c_b1, &c__[c_dim1 + 2], ldc, &v[*incv + 1], incv, &c_b2, + &work[1], &c__1, (ftnlen)1); + i__1 = lastc; + for (i__ = 1; i__ <= i__1; ++i__) { + i__2 = i__; + i__3 = i__; + d_lmp_cnjg(&z__2, &c__[i__ * c_dim1 + 1]); + z__1.r = work[i__3].r + z__2.r, z__1.i = work[i__3].i + z__2.i; + work[i__2].r = z__1.r, work[i__2].i = z__1.i; + } + i__1 = lastc; + for (i__ = 1; i__ <= i__1; ++i__) { + i__2 = i__ * c_dim1 + 1; + i__3 = i__ * c_dim1 + 1; + d_lmp_cnjg(&z__3, &work[i__]); + z__2.r = tau->r * z__3.r - tau->i * z__3.i, + z__2.i = tau->r * z__3.i + tau->i * z__3.r; + z__1.r = c__[i__3].r - z__2.r, z__1.i = c__[i__3].i - z__2.i; + c__[i__2].r = z__1.r, c__[i__2].i = z__1.i; + } + i__1 = lastv - 1; + z__1.r = -tau->r, z__1.i = -tau->i; + zgerc_(&i__1, &lastc, &z__1, &v[*incv + 1], incv, &work[1], &c__1, &c__[c_dim1 + 2], + ldc); + } + } else { + if (lastv == 1) { + z__1.r = 1. - tau->r, z__1.i = 0. - tau->i; + zscal_(&lastc, &z__1, &c__[c_offset], &c__1); + } else { + i__1 = lastv - 1; + zgemv_((char *)"N", &lastc, &i__1, &c_b1, &c__[(c_dim1 << 1) + 1], ldc, &v[*incv + 1], incv, + &c_b2, &work[1], &c__1, (ftnlen)1); + zaxpy_(&lastc, &c_b1, &c__[c_offset], &c__1, &work[1], &c__1); + z__1.r = -tau->r, z__1.i = -tau->i; + zaxpy_(&lastc, &z__1, &work[1], &c__1, &c__[c_offset], &c__1); + i__1 = lastv - 1; + z__1.r = -tau->r, z__1.i = -tau->i; + zgerc_(&lastc, &i__1, &z__1, &work[1], &c__1, &v[*incv + 1], incv, + &c__[(c_dim1 << 1) + 1], ldc); + } + } + return 0; +} +#ifdef __cplusplus +} +#endif