📄 ztrmm.c
字号:
}
}
}
} else {
/* Form B := alpha*B*A' or B := alpha*B*conjg( A' ). */
if (upper) {
for (j = 0; j < *n; ++j) {
for (i = *m - 1; i >= 0; --i) {
i__2 = i + j * *ldb;
temp.r = b[i__2].r, temp.i = b[i__2].i;
if (noconj) {
if (nounit) {
i__1 = i + i * *lda;
z__1.r = temp.r * a[i__1].r - temp.i * a[i__1].i,
z__1.i = temp.r * a[i__1].i + temp.i * a[i__1].r;
temp.r = z__1.r, temp.i = z__1.i;
}
for (k = 0; k < i; ++k) {
i__1 = k + i * *lda;
i__2 = k + j * *ldb;
temp.r += a[i__1].r * b[i__2].r - a[i__1].i * b[i__2].i,
temp.i += a[i__1].r * b[i__2].i + a[i__1].i * b[i__2].r;
}
} else {
if (nounit) {
i__1 = i + i * *lda;
z__1.r = temp.r * a[i__1].r + temp.i * a[i__1].i,
z__1.i = temp.i * a[i__1].r - temp.r * a[i__1].i;
temp.r = z__1.r, temp.i = z__1.i;
}
for (k = 0; k < i; ++k) {
i__1 = k + i * *lda;
i__2 = k + j * *ldb;
temp.r += a[i__1].r * b[i__2].r + a[i__1].i * b[i__2].i,
temp.i += a[i__1].r * b[i__2].i - a[i__1].i * b[i__2].r;
}
}
i__2 = i + j * *ldb;
b[i__2].r = alpha->r * temp.r - alpha->i * temp.i,
b[i__2].i = alpha->r * temp.i + alpha->i * temp.r;
}
}
} else {
for (j = 0; j < *n; ++j) {
for (i = 0; i < *m; ++i) {
i__2 = i + j * *ldb;
temp.r = b[i__2].r, temp.i = b[i__2].i;
if (noconj) {
if (nounit) {
i__1 = i + i * *lda;
z__1.r = temp.r * a[i__1].r - temp.i * a[i__1].i,
z__1.i = temp.r * a[i__1].i + temp.i * a[i__1].r;
temp.r = z__1.r, temp.i = z__1.i;
}
for (k = i + 1; k < *m; ++k) {
i__1 = k + i * *lda;
i__2 = k + j * *ldb;
temp.r += a[i__1].r * b[i__2].r - a[i__1].i * b[i__2].i,
temp.i += a[i__1].r * b[i__2].i + a[i__1].i * b[i__2].r;
}
} else {
if (nounit) {
i__1 = i + i * *lda;
z__1.r = temp.r * a[i__1].r + temp.i * a[i__1].i,
z__1.i = temp.i * a[i__1].r - temp.r * a[i__1].i;
temp.r = z__1.r, temp.i = z__1.i;
}
for (k = i + 1; k < *m; ++k) {
i__1 = k + i * *lda;
i__2 = k + j * *ldb;
temp.r += a[i__1].r * b[i__2].r + a[i__1].i * b[i__2].i,
temp.i += a[i__1].r * b[i__2].i - a[i__1].i * b[i__2].r;
}
}
i__2 = i + j * *ldb;
b[i__2].r = alpha->r * temp.r - alpha->i * temp.i,
b[i__2].i = alpha->r * temp.i + alpha->i * temp.r;
}
}
}
}
} else {
if (lsame_(transa, "N")) {
/* Form B := alpha*B*A. */
if (upper) {
for (j = *n - 1; j >= 0; --j) {
temp.r = alpha->r, temp.i = alpha->i;
if (nounit) {
i__1 = j + j * *lda;
z__1.r = temp.r * a[i__1].r - temp.i * a[i__1].i,
z__1.i = temp.r * a[i__1].i + temp.i * a[i__1].r;
temp.r = z__1.r, temp.i = z__1.i;
}
for (i = 0; i < *m; ++i) {
i__2 = i + j * *ldb;
z__1.r = temp.r * b[i__2].r - temp.i * b[i__2].i,
z__1.i = temp.r * b[i__2].i + temp.i * b[i__2].r;
b[i__2].r = z__1.r, b[i__2].i = z__1.i;
}
for (k = 0; k < j; ++k) {
i__1 = k + j * *lda;
if (a[i__1].r != 0. || a[i__1].i != 0.) {
temp.r = alpha->r * a[i__1].r - alpha->i * a[i__1].i,
temp.i = alpha->r * a[i__1].i + alpha->i * a[i__1].r;
for (i = 0; i < *m; ++i) {
i__2 = i + k * *ldb;
z__1.r = temp.r * b[i__2].r - temp.i * b[i__2].i,
z__1.i = temp.r * b[i__2].i + temp.i * b[i__2].r;
i__2 = i + j * *ldb;
b[i__2].r += z__1.r, b[i__2].i += z__1.i;
}
}
}
}
} else {
for (j = 0; j < *n; ++j) {
temp.r = alpha->r, temp.i = alpha->i;
if (nounit) {
i__1 = j + j * *lda;
z__1.r = temp.r * a[i__1].r - temp.i * a[i__1].i,
z__1.i = temp.r * a[i__1].i + temp.i * a[i__1].r;
temp.r = z__1.r, temp.i = z__1.i;
}
for (i = 0; i < *m; ++i) {
i__2 = i + j * *ldb;
z__1.r = temp.r * b[i__2].r - temp.i * b[i__2].i,
z__1.i = temp.r * b[i__2].i + temp.i * b[i__2].r;
b[i__2].r = z__1.r, b[i__2].i = z__1.i;
}
for (k = j + 1; k < *n; ++k) {
i__1 = k + j * *lda;
if (a[i__1].r != 0. || a[i__1].i != 0.) {
temp.r = alpha->r * a[i__1].r - alpha->i * a[i__1].i,
temp.i = alpha->r * a[i__1].i + alpha->i * a[i__1].r;
for (i = 0; i < *m; ++i) {
i__2 = i + k * *ldb;
z__1.r = temp.r * b[i__2].r - temp.i * b[i__2].i,
z__1.i = temp.r * b[i__2].i + temp.i * b[i__2].r;
i__2 = i + j * *ldb;
b[i__2].r += z__1.r, b[i__2].i += z__1.i;
}
}
}
}
}
} else {
/* Form B := alpha*B*A' or B := alpha*B*conjg( A' ). */
if (upper) {
for (k = 0; k < *n; ++k) {
for (j = 0; j < k; ++j) {
i__1 = j + k * *lda;
if (a[i__1].r != 0. || a[i__1].i != 0.) {
if (noconj) {
temp.r = alpha->r * a[i__1].r - alpha->i * a[i__1].i,
temp.i = alpha->i * a[i__1].r + alpha->r * a[i__1].i;
} else {
temp.r = alpha->r * a[i__1].r + alpha->i * a[i__1].i,
temp.i = alpha->i * a[i__1].r - alpha->r * a[i__1].i;
}
for (i = 0; i < *m; ++i) {
i__2 = i + k * *ldb;
z__1.r = temp.r * b[i__2].r - temp.i * b[i__2].i,
z__1.i = temp.r * b[i__2].i + temp.i * b[i__2].r;
i__2 = i + j * *ldb;
b[i__2].r += z__1.r, b[i__2].i += z__1.i;
}
}
}
temp.r = alpha->r, temp.i = alpha->i;
if (nounit) {
if (noconj) {
i__1 = k + k * *lda;
z__1.r = temp.r * a[i__1].r - temp.i * a[i__1].i,
z__1.i = temp.r * a[i__1].i + temp.i * a[i__1].r;
temp.r = z__1.r, temp.i = z__1.i;
} else {
i__1 = k + k * *lda;
z__1.r = temp.r * a[i__1].r + temp.i * a[i__1].i,
z__1.i = temp.i * a[i__1].r - temp.r * a[i__1].i;
temp.r = z__1.r, temp.i = z__1.i;
}
}
if (temp.r != 1. || temp.i != 0.) {
for (i = 0; i < *m; ++i) {
i__2 = i + k * *ldb;
z__1.r = temp.r * b[i__2].r - temp.i * b[i__2].i,
z__1.i = temp.r * b[i__2].i + temp.i * b[i__2].r;
b[i__2].r = z__1.r, b[i__2].i = z__1.i;
}
}
}
} else {
for (k = *n - 1; k >= 0; --k) {
for (j = k + 1; j < *n; ++j) {
i__1 = j + k * *lda;
if (a[i__1].r != 0. || a[i__1].i != 0.) {
if (noconj) {
temp.r = alpha->r * a[i__1].r - alpha->i * a[i__1].i,
temp.i = alpha->r * a[i__1].i + alpha->i * a[i__1].r;
} else {
temp.r = alpha->r * a[i__1].r + alpha->i * a[i__1].i,
temp.i = alpha->i * a[i__1].r - alpha->r * a[i__1].i;
}
for (i = 0; i < *m; ++i) {
i__2 = i + k * *ldb;
z__1.r = temp.r * b[i__2].r - temp.i * b[i__2].i,
z__1.i = temp.r * b[i__2].i + temp.i * b[i__2].r;
i__2 = i + j * *ldb;
b[i__2].r += z__1.r, b[i__2].i += z__1.i;
}
}
}
temp.r = alpha->r, temp.i = alpha->i;
if (nounit) {
if (noconj) {
i__1 = k + k * *lda;
z__1.r = temp.r * a[i__1].r - temp.i * a[i__1].i,
z__1.i = temp.i * a[i__1].r + temp.r * a[i__1].i;
temp.r = z__1.r, temp.i = z__1.i;
} else {
i__1 = k + k * *lda;
z__1.r = temp.r * a[i__1].r + temp.i * a[i__1].i,
z__1.i = temp.i * a[i__1].r - temp.r * a[i__1].i;
temp.r = z__1.r, temp.i = z__1.i;
}
}
if (temp.r != 1. || temp.i != 0.) {
for (i = 0; i < *m; ++i) {
i__2 = i + k * *ldb;
z__1.r = temp.r * b[i__2].r - temp.i * b[i__2].i,
z__1.i = temp.r * b[i__2].i + temp.i * b[i__2].r;
b[i__2].r = z__1.r, b[i__2].i = z__1.i;
}
}
}
}
}
}
} /* ztrmm_ */
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -