Skip to content

Commit

Permalink
add another couple of dependent functions
Browse files Browse the repository at this point in the history
  • Loading branch information
akohlmey committed Jan 8, 2025
1 parent f3425eb commit 7480460
Show file tree
Hide file tree
Showing 2 changed files with 210 additions and 0 deletions.
95 changes: 95 additions & 0 deletions lib/linalg/dlarf1f.cpp
Original file line number Diff line number Diff line change
@@ -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
115 changes: 115 additions & 0 deletions lib/linalg/zlarf1f.cpp
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit 7480460

Please sign in to comment.