ztrsm.c

SuperLU 2.2版本。对大型、稀疏、非对称的线性系统的直接求解

		i__1 = *n;
		for (j = 1; j <= *n; ++j) {
		    i__2 = *m;
		    for (i = 1; i <= *m; ++i) {
			i__3 = i + j * b_dim1;
			z__1.r = alpha->r * B(i,j).r - alpha->i * B(i,j).i, 
				z__1.i = alpha->r * B(i,j).i + alpha->i * B(i,j).r;
			temp.r = z__1.r, temp.i = z__1.i;
			if (noconj) {
			    i__3 = i - 1;
			    for (k = 1; k <= i-1; ++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;
/* L110: */
			    }
			    if (nounit) {
				z_div(&z__1, &temp, &A(i,i));
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			} else {
			    i__3 = i - 1;
			    for (k = 1; k <= i-1; ++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;
/* L120: */
			    }
			    if (nounit) {
				d_cnjg(&z__2, &A(i,i));
				z_div(&z__1, &temp, &z__2);
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			}
			i__3 = i + j * b_dim1;
			B(i,j).r = temp.r, B(i,j).i = temp.i;
/* L130: */
		    }
/* L140: */
		}
	    } else {
		i__1 = *n;
		for (j = 1; j <= *n; ++j) {
		    for (i = *m; i >= 1; --i) {
			i__2 = i + j * b_dim1;
			z__1.r = alpha->r * B(i,j).r - alpha->i * B(i,j).i, 
				z__1.i = alpha->r * B(i,j).i + alpha->i * B(i,j).r;
			temp.r = z__1.r, temp.i = z__1.i;
			if (noconj) {
			    i__2 = *m;
			    for (k = i + 1; k <= *m; ++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;
/* L150: */
			    }
			    if (nounit) {
				z_div(&z__1, &temp, &A(i,i));
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			} else {
			    i__2 = *m;
			    for (k = i + 1; k <= *m; ++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;
/* L160: */
			    }
			    if (nounit) {
				d_cnjg(&z__2, &A(i,i));
				z_div(&z__1, &temp, &z__2);
				temp.r = z__1.r, temp.i = z__1.i;
			    }
			}
			i__2 = i + j * b_dim1;
			B(i,j).r = temp.r, B(i,j).i = temp.i;
/* L170: */
		    }
/* L180: */
		}
	    }
	}
    } else {
	if (lsame_(transa, "N")) {

/*           Form  B := alpha*B*inv( A ). */

	    if (upper) {
		i__1 = *n;
		for (j = 1; j <= *n; ++j) {
		    if (alpha->r != 1. || alpha->i != 0.) {
			i__2 = *m;
			for (i = 1; i <= *m; ++i) {
			    i__3 = i + j * b_dim1;
			    i__4 = i + j * b_dim1;
			    z__1.r = alpha->r * B(i,j).r - alpha->i * B(i,j)
				    .i, z__1.i = alpha->r * B(i,j).i + 
				    alpha->i * B(i,j).r;
			    B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L190: */
			}
		    }
		    i__2 = j - 1;
		    for (k = 1; k <= j-1; ++k) {
			i__3 = k + j * a_dim1;
			if (A(k,j).r != 0. || A(k,j).i != 0.) {
			    i__3 = *m;
			    for (i = 1; i <= *m; ++i) {
				i__4 = i + j * b_dim1;
				i__5 = i + j * b_dim1;
				i__6 = k + j * a_dim1;
				i__7 = i + k * b_dim1;
				z__2.r = A(k,j).r * B(i,k).r - A(k,j).i * 
					B(i,k).i, z__2.i = A(k,j).r * B(i,k).i + A(k,j).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;
/* L200: */
			    }
			}
/* L210: */
		    }
		    if (nounit) {
			z_div(&z__1, &c_b1, &A(j,j));
			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;
/* L220: */
			}
		    }
/* L230: */
		}
	    } else {
		for (j = *n; j >= 1; --j) {
		    if (alpha->r != 1. || alpha->i != 0.) {
			i__1 = *m;
			for (i = 1; i <= *m; ++i) {
			    i__2 = i + j * b_dim1;
			    i__3 = i + j * b_dim1;
			    z__1.r = alpha->r * B(i,j).r - alpha->i * B(i,j)
				    .i, z__1.i = alpha->r * B(i,j).i + 
				    alpha->i * B(i,j).r;
			    B(i,j).r = z__1.r, B(i,j).i = z__1.i;
/* L240: */
			}
		    }
		    i__1 = *n;
		    for (k = j + 1; k <= *n; ++k) {
			i__2 = k + j * a_dim1;
			if (A(k,j).r != 0. || A(k,j).i != 0.) {
			    i__2 = *m;
			    for (i = 1; i <= *m; ++i) {
				i__3 = i + j * b_dim1;
				i__4 = i + j * b_dim1;
				i__5 = k + j * a_dim1;
				i__6 = i + k * b_dim1;
				z__2.r = A(k,j).r * B(i,k).r - A(k,j).i * 
					B(i,k).i, z__2.i = A(k,j).r * B(i,k).i + A(k,j).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: */
		    }
		    if (nounit) {
			z_div(&z__1, &c_b1, &A(j,j));
			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;
/* L270: */
			}
		    }
/* L280: */
		}
	    }
	} else {

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

	    if (upper) {
		for (k = *n; k >= 1; --k) {
		    if (nounit) {
			if (noconj) {
			    z_div(&z__1, &c_b1, &A(k,k));
			    temp.r = z__1.r, temp.i = z__1.i;
			} else {
			    d_cnjg(&z__2, &A(k,k));
			    z_div(&z__1, &c_b1, &z__2);
			    temp.r = z__1.r, temp.i = z__1.i;
			}
			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;
/* L290: */
			}
		    }
		    i__1 = k - 1;
		    for (j = 1; j <= k-1; ++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;
				temp.r = A(j,k).r, temp.i = A(j,k).i;
			    } else {
				d_cnjg(&z__1, &A(j,k));
				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;
/* L300: */
			    }
			}
/* L310: */
		    }
		    if (alpha->r != 1. || alpha->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 = alpha->r * B(i,k).r - alpha->i * B(i,k)
				    .i, z__1.i = alpha->r * B(i,k).i + 
				    alpha->i * B(i,k).r;
			    B(i,k).r = z__1.r, B(i,k).i = z__1.i;
/* L320: */
			}
		    }
/* L330: */
		}
	    } else {
		i__1 = *n;
		for (k = 1; k <= *n; ++k) {
		    if (nounit) {
			if (noconj) {
			    z_div(&z__1, &c_b1, &A(k,k));
			    temp.r = z__1.r, temp.i = z__1.i;
			} else {
			    d_cnjg(&z__2, &A(k,k));
			    z_div(&z__1, &c_b1, &z__2);
			    temp.r = z__1.r, temp.i = z__1.i;
			}
			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;
/* L340: */
			}
		    }
		    i__2 = *n;
		    for (j = k + 1; j <= *n; ++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;
				temp.r = A(j,k).r, temp.i = A(j,k).i;
			    } else {
				d_cnjg(&z__1, &A(j,k));
				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;
/* L350: */
			    }
			}
/* L360: */
		    }
		    if (alpha->r != 1. || alpha->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 = alpha->r * B(i,k).r - alpha->i * B(i,k)
				    .i, z__1.i = alpha->r * B(i,k).i + 
				    alpha->i * B(i,k).r;
			    B(i,k).r = z__1.r, B(i,k).i = z__1.i;
/* L370: */
			}
		    }
/* L380: */
		}
	    }
	}
    }

    return 0;

/*     End of ZTRSM . */

} /* ztrsm_ */