#include "blaswrap.h" #include "f2c.h" /* Subroutine */ int zgttrf_(integer *n, doublecomplex *dl, doublecomplex * d__, doublecomplex *du, doublecomplex *du2, integer *ipiv, integer * info) { /* -- LAPACK routine (version 3.0) -- Univ. of Tennessee, Univ. of California Berkeley, NAG Ltd., Courant Institute, Argonne National Lab, and Rice University June 30, 1999 Purpose ======= ZGTTRF computes an LU factorization of a complex tridiagonal matrix A using elimination with partial pivoting and row interchanges. The factorization has the form A = L * U where L is a product of permutation and unit lower bidiagonal matrices and U is upper triangular with nonzeros in only the main diagonal and first two superdiagonals. Arguments ========= N (input) INTEGER The order of the matrix A. DL (input/output) COMPLEX*16 array, dimension (N-1) On entry, DL must contain the (n-1) sub-diagonal elements of A. On exit, DL is overwritten by the (n-1) multipliers that define the matrix L from the LU factorization of A. D (input/output) COMPLEX*16 array, dimension (N) On entry, D must contain the diagonal elements of A. On exit, D is overwritten by the n diagonal elements of the upper triangular matrix U from the LU factorization of A. DU (input/output) COMPLEX*16 array, dimension (N-1) On entry, DU must contain the (n-1) super-diagonal elements of A. On exit, DU is overwritten by the (n-1) elements of the first super-diagonal of U. DU2 (output) COMPLEX*16 array, dimension (N-2) On exit, DU2 is overwritten by the (n-2) elements of the second super-diagonal of U. IPIV (output) INTEGER array, dimension (N) The pivot indices; for 1 <= i <= n, row i of the matrix was interchanged with row IPIV(i). IPIV(i) will always be either i or i+1; IPIV(i) = i indicates a row interchange was not required. INFO (output) INTEGER = 0: successful exit < 0: if INFO = -k, the k-th argument had an illegal value > 0: if INFO = k, U(k,k) is exactly zero. The factorization has been completed, but the factor U is exactly singular, and division by zero will occur if it is used to solve a system of equations. ===================================================================== Parameter adjustments */ /* System generated locals */ integer i__1, i__2, i__3, i__4; doublereal d__1, d__2, d__3, d__4; doublecomplex z__1, z__2; /* Builtin functions */ double d_imag(doublecomplex *); void z_div(doublecomplex *, doublecomplex *, doublecomplex *); /* Local variables */ static doublecomplex fact, temp; static integer i__; extern /* Subroutine */ int xerbla_(char *, integer *); --ipiv; --du2; --du; --d__; --dl; /* Function Body */ *info = 0; if (*n < 0) { *info = -1; i__1 = -(*info); xerbla_("ZGTTRF", &i__1); return 0; } /* Quick return if possible */ if (*n == 0) { return 0; } /* Initialize IPIV(i) = i and DU2(i) = 0 */ i__1 = *n; for (i__ = 1; i__ <= i__1; ++i__) { ipiv[i__] = i__; /* L10: */ } i__1 = *n - 2; for (i__ = 1; i__ <= i__1; ++i__) { i__2 = i__; du2[i__2].r = 0., du2[i__2].i = 0.; /* L20: */ } i__1 = *n - 2; for (i__ = 1; i__ <= i__1; ++i__) { i__2 = i__; i__3 = i__; if ((d__1 = d__[i__2].r, abs(d__1)) + (d__2 = d_imag(&d__[i__]), abs( d__2)) >= (d__3 = dl[i__3].r, abs(d__3)) + (d__4 = d_imag(&dl[ i__]), abs(d__4))) { /* No row interchange required, eliminate DL(I) */ i__2 = i__; if ((d__1 = d__[i__2].r, abs(d__1)) + (d__2 = d_imag(&d__[i__]), abs(d__2)) != 0.) { z_div(&z__1, &dl[i__], &d__[i__]); fact.r = z__1.r, fact.i = z__1.i; i__2 = i__; dl[i__2].r = fact.r, dl[i__2].i = fact.i; i__2 = i__ + 1; i__3 = i__ + 1; i__4 = i__; z__2.r = fact.r * du[i__4].r - fact.i * du[i__4].i, z__2.i = fact.r * du[i__4].i + fact.i * du[i__4].r; z__1.r = d__[i__3].r - z__2.r, z__1.i = d__[i__3].i - z__2.i; d__[i__2].r = z__1.r, d__[i__2].i = z__1.i; } } else { /* Interchange rows I and I+1, eliminate DL(I) */ z_div(&z__1, &d__[i__], &dl[i__]); fact.r = z__1.r, fact.i = z__1.i; i__2 = i__; i__3 = i__; d__[i__2].r = dl[i__3].r, d__[i__2].i = dl[i__3].i; i__2 = i__; dl[i__2].r = fact.r, dl[i__2].i = fact.i; i__2 = i__; temp.r = du[i__2].r, temp.i = du[i__2].i; i__2 = i__; i__3 = i__ + 1; du[i__2].r = d__[i__3].r, du[i__2].i = d__[i__3].i; i__2 = i__ + 1; i__3 = i__ + 1; z__2.r = fact.r * d__[i__3].r - fact.i * d__[i__3].i, z__2.i = fact.r * d__[i__3].i + fact.i * d__[i__3].r; z__1.r = temp.r - z__2.r, z__1.i = temp.i - z__2.i; d__[i__2].r = z__1.r, d__[i__2].i = z__1.i; i__2 = i__; i__3 = i__ + 1; du2[i__2].r = du[i__3].r, du2[i__2].i = du[i__3].i; i__2 = i__ + 1; z__2.r = -fact.r, z__2.i = -fact.i; i__3 = i__ + 1; z__1.r = z__2.r * du[i__3].r - z__2.i * du[i__3].i, z__1.i = z__2.r * du[i__3].i + z__2.i * du[i__3].r; du[i__2].r = z__1.r, du[i__2].i = z__1.i; ipiv[i__] = i__ + 1; } /* L30: */ } if (*n > 1) { i__ = *n - 1; i__1 = i__; i__2 = i__; if ((d__1 = d__[i__1].r, abs(d__1)) + (d__2 = d_imag(&d__[i__]), abs( d__2)) >= (d__3 = dl[i__2].r, abs(d__3)) + (d__4 = d_imag(&dl[ i__]), abs(d__4))) { i__1 = i__; if ((d__1 = d__[i__1].r, abs(d__1)) + (d__2 = d_imag(&d__[i__]), abs(d__2)) != 0.) { z_div(&z__1, &dl[i__], &d__[i__]); fact.r = z__1.r, fact.i = z__1.i; i__1 = i__; dl[i__1].r = fact.r, dl[i__1].i = fact.i; i__1 = i__ + 1; i__2 = i__ + 1; i__3 = i__; z__2.r = fact.r * du[i__3].r - fact.i * du[i__3].i, z__2.i = fact.r * du[i__3].i + fact.i * du[i__3].r; z__1.r = d__[i__2].r - z__2.r, z__1.i = d__[i__2].i - z__2.i; d__[i__1].r = z__1.r, d__[i__1].i = z__1.i; } } else { z_div(&z__1, &d__[i__], &dl[i__]); fact.r = z__1.r, fact.i = z__1.i; i__1 = i__; i__2 = i__; d__[i__1].r = dl[i__2].r, d__[i__1].i = dl[i__2].i; i__1 = i__; dl[i__1].r = fact.r, dl[i__1].i = fact.i; i__1 = i__; temp.r = du[i__1].r, temp.i = du[i__1].i; i__1 = i__; i__2 = i__ + 1; du[i__1].r = d__[i__2].r, du[i__1].i = d__[i__2].i; i__1 = i__ + 1; i__2 = i__ + 1; z__2.r = fact.r * d__[i__2].r - fact.i * d__[i__2].i, z__2.i = fact.r * d__[i__2].i + fact.i * d__[i__2].r; z__1.r = temp.r - z__2.r, z__1.i = temp.i - z__2.i; d__[i__1].r = z__1.r, d__[i__1].i = z__1.i; ipiv[i__] = i__ + 1; } } /* Check for a zero on the diagonal of U. */ i__1 = *n; for (i__ = 1; i__ <= i__1; ++i__) { i__2 = i__; if ((d__1 = d__[i__2].r, abs(d__1)) + (d__2 = d_imag(&d__[i__]), abs( d__2)) == 0.) { *info = i__; goto L50; } /* L40: */ } L50: return 0; /* End of ZGTTRF */ } /* zgttrf_ */