ztrmm.c

来自「算断裂的」· C语言 代码 · 共 689 行 · 第 1/2 页

					temp.i * A(i,i).r;
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			    i__2 = i - 1;
			    for (k = 1; k <= i-1; ++k) {
				i__3 = k + i * a_dim1;
				i__4 = k + j * b_dim1;
				z__2.r = A(k,i).r * B(k,j).r - A(k,i).i * 
					B(k,j).i, z__2.i = A(k,i).r * B(k,j).i + A(k,i).i * B(k,j).r;
				z__1.r = temp.r + z__2.r, z__1.i = temp.i + 
					z__2.i;
				temp.r = z__1.r, temp.i = z__1.i;
/* L90: */
			    }
			} else {
			    if (nounit) {
				d_cnjg(&z__2, &A(i,i));
				z__1.r = temp.r * z__2.r - temp.i * z__2.i, 
					z__1.i = temp.r * z__2.i + temp.i * 
					z__2.r;
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			    i__2 = i - 1;
			    for (k = 1; k <= i-1; ++k) {
				d_cnjg(&z__3, &A(k,i));
				i__3 = k + j * b_dim1;
				z__2.r = z__3.r * B(k,j).r - z__3.i * B(k,j)
					.i, z__2.i = z__3.r * B(k,j).i + 
					z__3.i * B(k,j).r;
				z__1.r = temp.r + z__2.r, z__1.i = temp.i + 
					z__2.i;
				temp.r = z__1.r, temp.i = z__1.i;
/* L100: */
			    }
			}
			i__2 = i + j * b_dim1;
			z__1.r = alpha->r * temp.r - alpha->i * temp.i, 
				z__1.i = alpha->r * temp.i + alpha->i * 
				temp.r;
			B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L110: */
		    }
/* L120: */
		}
	    } else {
		i__1 = *n;
		for (j = 1; j <= *n; ++j) {
		    i__2 = *m;
		    for (i = 1; i <= *m; ++i) {
			i__3 = i + j * b_dim1;
			temp.r = B(i,j).r, temp.i = B(i,j).i;
			if (noconj) {
			    if (nounit) {
				i__3 = i + i * a_dim1;
				z__1.r = temp.r * A(i,i).r - temp.i * A(i,i)
					.i, z__1.i = temp.r * A(i,i).i + 
					temp.i * A(i,i).r;
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			    i__3 = *m;
			    for (k = i + 1; k <= *m; ++k) {
				i__4 = k + i * a_dim1;
				i__5 = k + j * b_dim1;
				z__2.r = A(k,i).r * B(k,j).r - A(k,i).i * 
					B(k,j).i, z__2.i = A(k,i).r * B(k,j).i + A(k,i).i * B(k,j).r;
				z__1.r = temp.r + z__2.r, z__1.i = temp.i + 
					z__2.i;
				temp.r = z__1.r, temp.i = z__1.i;
/* L130: */
			    }
			} else {
			    if (nounit) {
				d_cnjg(&z__2, &A(i,i));
				z__1.r = temp.r * z__2.r - temp.i * z__2.i, 
					z__1.i = temp.r * z__2.i + temp.i * 
					z__2.r;
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			    i__3 = *m;
			    for (k = i + 1; k <= *m; ++k) {
				d_cnjg(&z__3, &A(k,i));
				i__4 = k + j * b_dim1;
				z__2.r = z__3.r * B(k,j).r - z__3.i * B(k,j)
					.i, z__2.i = z__3.r * B(k,j).i + 
					z__3.i * B(k,j).r;
				z__1.r = temp.r + z__2.r, z__1.i = temp.i + 
					z__2.i;
				temp.r = z__1.r, temp.i = z__1.i;
/* L140: */
			    }
			}
			i__3 = i + j * b_dim1;
			z__1.r = alpha->r * temp.r - alpha->i * temp.i, 
				z__1.i = alpha->r * temp.i + alpha->i * 
				temp.r;
			B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L150: */
		    }
/* L160: */
		}
	    }
	}
    } else {
	if (lsame_(transa, "N")) {

/*           Form  B := alpha*B*A. */

	    if (upper) {
		for (j = *n; j >= 1; --j) {
		    temp.r = alpha->r, temp.i = alpha->i;
		    if (nounit) {
			i__1 = j + j * a_dim1;
			z__1.r = temp.r * A(j,j).r - temp.i * A(j,j).i, 
				z__1.i = temp.r * A(j,j).i + temp.i * A(j,j)
				.r;
			temp.r = z__1.r, temp.i = z__1.i;
		    }
		    i__1 = *m;
		    for (i = 1; i <= *m; ++i) {
			i__2 = i + j * b_dim1;
			i__3 = i + j * b_dim1;
			z__1.r = temp.r * B(i,j).r - temp.i * B(i,j).i, 
				z__1.i = temp.r * B(i,j).i + temp.i * B(i,j)
				.r;
			B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L170: */
		    }
		    i__1 = j - 1;
		    for (k = 1; k <= j-1; ++k) {
			i__2 = k + j * a_dim1;
			if (A(k,j).r != 0. || A(k,j).i != 0.) {
			    i__2 = k + j * a_dim1;
			    z__1.r = alpha->r * A(k,j).r - alpha->i * A(k,j)
				    .i, z__1.i = alpha->r * A(k,j).i + 
				    alpha->i * A(k,j).r;
			    temp.r = z__1.r, temp.i = z__1.i;
			    i__2 = *m;
			    for (i = 1; i <= *m; ++i) {
				i__3 = i + j * b_dim1;
				i__4 = i + j * b_dim1;
				i__5 = i + k * b_dim1;
				z__2.r = temp.r * B(i,k).r - temp.i * B(i,k)
					.i, z__2.i = temp.r * B(i,k).i + 
					temp.i * B(i,k).r;
				z__1.r = B(i,j).r + z__2.r, z__1.i = B(i,j)
					.i + z__2.i;
				B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L180: */
			    }
			}
/* L190: */
		    }
/* L200: */
		}
	    } else {
		i__1 = *n;
		for (j = 1; j <= *n; ++j) {
		    temp.r = alpha->r, temp.i = alpha->i;
		    if (nounit) {
			i__2 = j + j * a_dim1;
			z__1.r = temp.r * A(j,j).r - temp.i * A(j,j).i, 
				z__1.i = temp.r * A(j,j).i + temp.i * A(j,j)
				.r;
			temp.r = z__1.r, temp.i = z__1.i;
		    }
		    i__2 = *m;
		    for (i = 1; i <= *m; ++i) {
			i__3 = i + j * b_dim1;
			i__4 = i + j * b_dim1;
			z__1.r = temp.r * B(i,j).r - temp.i * B(i,j).i, 
				z__1.i = temp.r * B(i,j).i + temp.i * B(i,j)
				.r;
			B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L210: */
		    }
		    i__2 = *n;
		    for (k = j + 1; k <= *n; ++k) {
			i__3 = k + j * a_dim1;
			if (A(k,j).r != 0. || A(k,j).i != 0.) {
			    i__3 = k + j * a_dim1;
			    z__1.r = alpha->r * A(k,j).r - alpha->i * A(k,j)
				    .i, z__1.i = alpha->r * A(k,j).i + 
				    alpha->i * A(k,j).r;
			    temp.r = z__1.r, temp.i = z__1.i;
			    i__3 = *m;
			    for (i = 1; i <= *m; ++i) {
				i__4 = i + j * b_dim1;
				i__5 = i + j * b_dim1;
				i__6 = i + k * b_dim1;
				z__2.r = temp.r * B(i,k).r - temp.i * B(i,k)
					.i, z__2.i = temp.r * B(i,k).i + 
					temp.i * B(i,k).r;
				z__1.r = B(i,j).r + z__2.r, z__1.i = B(i,j)
					.i + z__2.i;
				B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L220: */
			    }
			}
/* L230: */
		    }
/* L240: */
		}
	    }
	} else {

/*           Form  B := alpha*B*A'   or   B := alpha*B*conjg( A' )
. */

	    if (upper) {
		i__1 = *n;
		for (k = 1; k <= *n; ++k) {
		    i__2 = k - 1;
		    for (j = 1; j <= k-1; ++j) {
			i__3 = j + k * a_dim1;
			if (A(j,k).r != 0. || A(j,k).i != 0.) {
			    if (noconj) {
				i__3 = j + k * a_dim1;
				z__1.r = alpha->r * A(j,k).r - alpha->i * A(j,k).i, z__1.i = alpha->r * A(j,k)
					.i + alpha->i * A(j,k).r;
				temp.r = z__1.r, temp.i = z__1.i;
			    } else {
				d_cnjg(&z__2, &A(j,k));
				z__1.r = alpha->r * z__2.r - alpha->i * 
					z__2.i, z__1.i = alpha->r * z__2.i + 
					alpha->i * z__2.r;
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			    i__3 = *m;
			    for (i = 1; i <= *m; ++i) {
				i__4 = i + j * b_dim1;
				i__5 = i + j * b_dim1;
				i__6 = i + k * b_dim1;
				z__2.r = temp.r * B(i,k).r - temp.i * B(i,k)
					.i, z__2.i = temp.r * B(i,k).i + 
					temp.i * B(i,k).r;
				z__1.r = B(i,j).r + z__2.r, z__1.i = B(i,j)
					.i + z__2.i;
				B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L250: */
			    }
			}
/* L260: */
		    }
		    temp.r = alpha->r, temp.i = alpha->i;
		    if (nounit) {
			if (noconj) {
			    i__2 = k + k * a_dim1;
			    z__1.r = temp.r * A(k,k).r - temp.i * A(k,k).i, 
				    z__1.i = temp.r * A(k,k).i + temp.i * A(k,k).r;
			    temp.r = z__1.r, temp.i = z__1.i;
			} else {
			    d_cnjg(&z__2, &A(k,k));
			    z__1.r = temp.r * z__2.r - temp.i * z__2.i, 
				    z__1.i = temp.r * z__2.i + temp.i * 
				    z__2.r;
			    temp.r = z__1.r, temp.i = z__1.i;
			}
		    }
		    if (temp.r != 1. || temp.i != 0.) {
			i__2 = *m;
			for (i = 1; i <= *m; ++i) {
			    i__3 = i + k * b_dim1;
			    i__4 = i + k * b_dim1;
			    z__1.r = temp.r * B(i,k).r - temp.i * B(i,k).i, 
				    z__1.i = temp.r * B(i,k).i + temp.i * B(i,k).r;
			    B(i,k).r = z__1.r, B(i,k).i = z__1.i;
/* L270: */
			}
		    }
/* L280: */
		}
	    } else {
		for (k = *n; k >= 1; --k) {
		    i__1 = *n;
		    for (j = k + 1; j <= *n; ++j) {
			i__2 = j + k * a_dim1;
			if (A(j,k).r != 0. || A(j,k).i != 0.) {
			    if (noconj) {
				i__2 = j + k * a_dim1;
				z__1.r = alpha->r * A(j,k).r - alpha->i * A(j,k).i, z__1.i = alpha->r * A(j,k)
					.i + alpha->i * A(j,k).r;
				temp.r = z__1.r, temp.i = z__1.i;
			    } else {
				d_cnjg(&z__2, &A(j,k));
				z__1.r = alpha->r * z__2.r - alpha->i * 
					z__2.i, z__1.i = alpha->r * z__2.i + 
					alpha->i * z__2.r;
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			    i__2 = *m;
			    for (i = 1; i <= *m; ++i) {
				i__3 = i + j * b_dim1;
				i__4 = i + j * b_dim1;
				i__5 = i + k * b_dim1;
				z__2.r = temp.r * B(i,k).r - temp.i * B(i,k)
					.i, z__2.i = temp.r * B(i,k).i + 
					temp.i * B(i,k).r;
				z__1.r = B(i,j).r + z__2.r, z__1.i = B(i,j)
					.i + z__2.i;
				B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L290: */
			    }
			}
/* L300: */
		    }
		    temp.r = alpha->r, temp.i = alpha->i;
		    if (nounit) {
			if (noconj) {
			    i__1 = k + k * a_dim1;
			    z__1.r = temp.r * A(k,k).r - temp.i * A(k,k).i, 
				    z__1.i = temp.r * A(k,k).i + temp.i * A(k,k).r;
			    temp.r = z__1.r, temp.i = z__1.i;
			} else {
			    d_cnjg(&z__2, &A(k,k));
			    z__1.r = temp.r * z__2.r - temp.i * z__2.i, 
				    z__1.i = temp.r * z__2.i + temp.i * 
				    z__2.r;
			    temp.r = z__1.r, temp.i = z__1.i;
			}
		    }
		    if (temp.r != 1. || temp.i != 0.) {
			i__1 = *m;
			for (i = 1; i <= *m; ++i) {
			    i__2 = i + k * b_dim1;
			    i__3 = i + k * b_dim1;
			    z__1.r = temp.r * B(i,k).r - temp.i * B(i,k).i, 
				    z__1.i = temp.r * B(i,k).i + temp.i * B(i,k).r;
			    B(i,k).r = z__1.r, B(i,k).i = z__1.i;
/* L310: */
			}
		    }
/* L320: */
		}
	    }
	}
    }

    return 0;

/*     End of ZTRMM . */

} /* ztrmm_ */