#include "f2c.h" #include "blaswrap.h" /* Subroutine */ int claptm_(char *uplo, integer *n, integer *nrhs, real * alpha, real *d__, complex *e, complex *x, integer *ldx, real *beta, complex *b, integer *ldb) { /* System generated locals */ integer b_dim1, b_offset, x_dim1, x_offset, i__1, i__2, i__3, i__4, i__5, i__6, i__7, i__8, i__9; complex q__1, q__2, q__3, q__4, q__5, q__6, q__7; /* Builtin functions */ void r_cnjg(complex *, complex *); /* Local variables */ integer i__, j; extern logical lsame_(char *, char *); /* -- LAPACK auxiliary routine (version 3.1) -- */ /* Univ. of Tennessee, Univ. of California Berkeley and NAG Ltd.. */ /* November 2006 */ /* .. Scalar Arguments .. */ /* .. */ /* .. Array Arguments .. */ /* .. */ /* Purpose */ /* ======= */ /* CLAPTM multiplies an N by NRHS matrix X by a Hermitian tridiagonal */ /* matrix A and stores the result in a matrix B. The operation has the */ /* form */ /* B := alpha * A * X + beta * B */ /* where alpha may be either 1. or -1. and beta may be 0., 1., or -1. */ /* Arguments */ /* ========= */ /* UPLO (input) CHARACTER */ /* Specifies whether the superdiagonal or the subdiagonal of the */ /* tridiagonal matrix A is stored. */ /* = 'U': Upper, E is the superdiagonal of A. */ /* = 'L': Lower, E is the subdiagonal of A. */ /* N (input) INTEGER */ /* The order of the matrix A. N >= 0. */ /* NRHS (input) INTEGER */ /* The number of right hand sides, i.e., the number of columns */ /* of the matrices X and B. */ /* ALPHA (input) REAL */ /* The scalar alpha. ALPHA must be 1. or -1.; otherwise, */ /* it is assumed to be 0. */ /* D (input) REAL array, dimension (N) */ /* The n diagonal elements of the tridiagonal matrix A. */ /* E (input) COMPLEX array, dimension (N-1) */ /* The (n-1) subdiagonal or superdiagonal elements of A. */ /* X (input) COMPLEX array, dimension (LDX,NRHS) */ /* The N by NRHS matrix X. */ /* LDX (input) INTEGER */ /* The leading dimension of the array X. LDX >= max(N,1). */ /* BETA (input) REAL */ /* The scalar beta. BETA must be 0., 1., or -1.; otherwise, */ /* it is assumed to be 1. */ /* B (input/output) COMPLEX array, dimension (LDB,NRHS) */ /* On entry, the N by NRHS matrix B. */ /* On exit, B is overwritten by the matrix expression */ /* B := alpha * A * X + beta * B. */ /* LDB (input) INTEGER */ /* The leading dimension of the array B. LDB >= max(N,1). */ /* ===================================================================== */ /* .. Parameters .. */ /* .. */ /* .. Local Scalars .. */ /* .. */ /* .. External Functions .. */ /* .. */ /* .. Intrinsic Functions .. */ /* .. */ /* .. Executable Statements .. */ /* Parameter adjustments */ --d__; --e; x_dim1 = *ldx; x_offset = 1 + x_dim1; x -= x_offset; b_dim1 = *ldb; b_offset = 1 + b_dim1; b -= b_offset; /* Function Body */ if (*n == 0) { return 0; } if (*beta == 0.f) { i__1 = *nrhs; for (j = 1; j <= i__1; ++j) { i__2 = *n; for (i__ = 1; i__ <= i__2; ++i__) { i__3 = i__ + j * b_dim1; b[i__3].r = 0.f, b[i__3].i = 0.f; /* L10: */ } /* L20: */ } } else if (*beta == -1.f) { i__1 = *nrhs; for (j = 1; j <= i__1; ++j) { i__2 = *n; for (i__ = 1; i__ <= i__2; ++i__) { i__3 = i__ + j * b_dim1; i__4 = i__ + j * b_dim1; q__1.r = -b[i__4].r, q__1.i = -b[i__4].i; b[i__3].r = q__1.r, b[i__3].i = q__1.i; /* L30: */ } /* L40: */ } } if (*alpha == 1.f) { if (lsame_(uplo, "U")) { /* Compute B := B + A*X, where E is the superdiagonal of A. */ i__1 = *nrhs; for (j = 1; j <= i__1; ++j) { if (*n == 1) { i__2 = j * b_dim1 + 1; i__3 = j * b_dim1 + 1; i__4 = j * x_dim1 + 1; q__2.r = d__[1] * x[i__4].r, q__2.i = d__[1] * x[i__4].i; q__1.r = b[i__3].r + q__2.r, q__1.i = b[i__3].i + q__2.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; } else { i__2 = j * b_dim1 + 1; i__3 = j * b_dim1 + 1; i__4 = j * x_dim1 + 1; q__3.r = d__[1] * x[i__4].r, q__3.i = d__[1] * x[i__4].i; q__2.r = b[i__3].r + q__3.r, q__2.i = b[i__3].i + q__3.i; i__5 = j * x_dim1 + 2; q__4.r = e[1].r * x[i__5].r - e[1].i * x[i__5].i, q__4.i = e[1].r * x[i__5].i + e[1].i * x[i__5].r; q__1.r = q__2.r + q__4.r, q__1.i = q__2.i + q__4.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; i__2 = *n + j * b_dim1; i__3 = *n + j * b_dim1; r_cnjg(&q__4, &e[*n - 1]); i__4 = *n - 1 + j * x_dim1; q__3.r = q__4.r * x[i__4].r - q__4.i * x[i__4].i, q__3.i = q__4.r * x[i__4].i + q__4.i * x[i__4].r; q__2.r = b[i__3].r + q__3.r, q__2.i = b[i__3].i + q__3.i; i__5 = *n; i__6 = *n + j * x_dim1; q__5.r = d__[i__5] * x[i__6].r, q__5.i = d__[i__5] * x[ i__6].i; q__1.r = q__2.r + q__5.r, q__1.i = q__2.i + q__5.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; i__2 = *n - 1; for (i__ = 2; i__ <= i__2; ++i__) { i__3 = i__ + j * b_dim1; i__4 = i__ + j * b_dim1; r_cnjg(&q__5, &e[i__ - 1]); i__5 = i__ - 1 + j * x_dim1; q__4.r = q__5.r * x[i__5].r - q__5.i * x[i__5].i, q__4.i = q__5.r * x[i__5].i + q__5.i * x[i__5] .r; q__3.r = b[i__4].r + q__4.r, q__3.i = b[i__4].i + q__4.i; i__6 = i__; i__7 = i__ + j * x_dim1; q__6.r = d__[i__6] * x[i__7].r, q__6.i = d__[i__6] * x[i__7].i; q__2.r = q__3.r + q__6.r, q__2.i = q__3.i + q__6.i; i__8 = i__; i__9 = i__ + 1 + j * x_dim1; q__7.r = e[i__8].r * x[i__9].r - e[i__8].i * x[i__9] .i, q__7.i = e[i__8].r * x[i__9].i + e[i__8] .i * x[i__9].r; q__1.r = q__2.r + q__7.r, q__1.i = q__2.i + q__7.i; b[i__3].r = q__1.r, b[i__3].i = q__1.i; /* L50: */ } } /* L60: */ } } else { /* Compute B := B + A*X, where E is the subdiagonal of A. */ i__1 = *nrhs; for (j = 1; j <= i__1; ++j) { if (*n == 1) { i__2 = j * b_dim1 + 1; i__3 = j * b_dim1 + 1; i__4 = j * x_dim1 + 1; q__2.r = d__[1] * x[i__4].r, q__2.i = d__[1] * x[i__4].i; q__1.r = b[i__3].r + q__2.r, q__1.i = b[i__3].i + q__2.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; } else { i__2 = j * b_dim1 + 1; i__3 = j * b_dim1 + 1; i__4 = j * x_dim1 + 1; q__3.r = d__[1] * x[i__4].r, q__3.i = d__[1] * x[i__4].i; q__2.r = b[i__3].r + q__3.r, q__2.i = b[i__3].i + q__3.i; r_cnjg(&q__5, &e[1]); i__5 = j * x_dim1 + 2; q__4.r = q__5.r * x[i__5].r - q__5.i * x[i__5].i, q__4.i = q__5.r * x[i__5].i + q__5.i * x[i__5].r; q__1.r = q__2.r + q__4.r, q__1.i = q__2.i + q__4.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; i__2 = *n + j * b_dim1; i__3 = *n + j * b_dim1; i__4 = *n - 1; i__5 = *n - 1 + j * x_dim1; q__3.r = e[i__4].r * x[i__5].r - e[i__4].i * x[i__5].i, q__3.i = e[i__4].r * x[i__5].i + e[i__4].i * x[ i__5].r; q__2.r = b[i__3].r + q__3.r, q__2.i = b[i__3].i + q__3.i; i__6 = *n; i__7 = *n + j * x_dim1; q__4.r = d__[i__6] * x[i__7].r, q__4.i = d__[i__6] * x[ i__7].i; q__1.r = q__2.r + q__4.r, q__1.i = q__2.i + q__4.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; i__2 = *n - 1; for (i__ = 2; i__ <= i__2; ++i__) { i__3 = i__ + j * b_dim1; i__4 = i__ + j * b_dim1; i__5 = i__ - 1; i__6 = i__ - 1 + j * x_dim1; q__4.r = e[i__5].r * x[i__6].r - e[i__5].i * x[i__6] .i, q__4.i = e[i__5].r * x[i__6].i + e[i__5] .i * x[i__6].r; q__3.r = b[i__4].r + q__4.r, q__3.i = b[i__4].i + q__4.i; i__7 = i__; i__8 = i__ + j * x_dim1; q__5.r = d__[i__7] * x[i__8].r, q__5.i = d__[i__7] * x[i__8].i; q__2.r = q__3.r + q__5.r, q__2.i = q__3.i + q__5.i; r_cnjg(&q__7, &e[i__]); i__9 = i__ + 1 + j * x_dim1; q__6.r = q__7.r * x[i__9].r - q__7.i * x[i__9].i, q__6.i = q__7.r * x[i__9].i + q__7.i * x[i__9] .r; q__1.r = q__2.r + q__6.r, q__1.i = q__2.i + q__6.i; b[i__3].r = q__1.r, b[i__3].i = q__1.i; /* L70: */ } } /* L80: */ } } } else if (*alpha == -1.f) { if (lsame_(uplo, "U")) { /* Compute B := B - A*X, where E is the superdiagonal of A. */ i__1 = *nrhs; for (j = 1; j <= i__1; ++j) { if (*n == 1) { i__2 = j * b_dim1 + 1; i__3 = j * b_dim1 + 1; i__4 = j * x_dim1 + 1; q__2.r = d__[1] * x[i__4].r, q__2.i = d__[1] * x[i__4].i; q__1.r = b[i__3].r - q__2.r, q__1.i = b[i__3].i - q__2.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; } else { i__2 = j * b_dim1 + 1; i__3 = j * b_dim1 + 1; i__4 = j * x_dim1 + 1; q__3.r = d__[1] * x[i__4].r, q__3.i = d__[1] * x[i__4].i; q__2.r = b[i__3].r - q__3.r, q__2.i = b[i__3].i - q__3.i; i__5 = j * x_dim1 + 2; q__4.r = e[1].r * x[i__5].r - e[1].i * x[i__5].i, q__4.i = e[1].r * x[i__5].i + e[1].i * x[i__5].r; q__1.r = q__2.r - q__4.r, q__1.i = q__2.i - q__4.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; i__2 = *n + j * b_dim1; i__3 = *n + j * b_dim1; r_cnjg(&q__4, &e[*n - 1]); i__4 = *n - 1 + j * x_dim1; q__3.r = q__4.r * x[i__4].r - q__4.i * x[i__4].i, q__3.i = q__4.r * x[i__4].i + q__4.i * x[i__4].r; q__2.r = b[i__3].r - q__3.r, q__2.i = b[i__3].i - q__3.i; i__5 = *n; i__6 = *n + j * x_dim1; q__5.r = d__[i__5] * x[i__6].r, q__5.i = d__[i__5] * x[ i__6].i; q__1.r = q__2.r - q__5.r, q__1.i = q__2.i - q__5.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; i__2 = *n - 1; for (i__ = 2; i__ <= i__2; ++i__) { i__3 = i__ + j * b_dim1; i__4 = i__ + j * b_dim1; r_cnjg(&q__5, &e[i__ - 1]); i__5 = i__ - 1 + j * x_dim1; q__4.r = q__5.r * x[i__5].r - q__5.i * x[i__5].i, q__4.i = q__5.r * x[i__5].i + q__5.i * x[i__5] .r; q__3.r = b[i__4].r - q__4.r, q__3.i = b[i__4].i - q__4.i; i__6 = i__; i__7 = i__ + j * x_dim1; q__6.r = d__[i__6] * x[i__7].r, q__6.i = d__[i__6] * x[i__7].i; q__2.r = q__3.r - q__6.r, q__2.i = q__3.i - q__6.i; i__8 = i__; i__9 = i__ + 1 + j * x_dim1; q__7.r = e[i__8].r * x[i__9].r - e[i__8].i * x[i__9] .i, q__7.i = e[i__8].r * x[i__9].i + e[i__8] .i * x[i__9].r; q__1.r = q__2.r - q__7.r, q__1.i = q__2.i - q__7.i; b[i__3].r = q__1.r, b[i__3].i = q__1.i; /* L90: */ } } /* L100: */ } } else { /* Compute B := B - A*X, where E is the subdiagonal of A. */ i__1 = *nrhs; for (j = 1; j <= i__1; ++j) { if (*n == 1) { i__2 = j * b_dim1 + 1; i__3 = j * b_dim1 + 1; i__4 = j * x_dim1 + 1; q__2.r = d__[1] * x[i__4].r, q__2.i = d__[1] * x[i__4].i; q__1.r = b[i__3].r - q__2.r, q__1.i = b[i__3].i - q__2.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; } else { i__2 = j * b_dim1 + 1; i__3 = j * b_dim1 + 1; i__4 = j * x_dim1 + 1; q__3.r = d__[1] * x[i__4].r, q__3.i = d__[1] * x[i__4].i; q__2.r = b[i__3].r - q__3.r, q__2.i = b[i__3].i - q__3.i; r_cnjg(&q__5, &e[1]); i__5 = j * x_dim1 + 2; q__4.r = q__5.r * x[i__5].r - q__5.i * x[i__5].i, q__4.i = q__5.r * x[i__5].i + q__5.i * x[i__5].r; q__1.r = q__2.r - q__4.r, q__1.i = q__2.i - q__4.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; i__2 = *n + j * b_dim1; i__3 = *n + j * b_dim1; i__4 = *n - 1; i__5 = *n - 1 + j * x_dim1; q__3.r = e[i__4].r * x[i__5].r - e[i__4].i * x[i__5].i, q__3.i = e[i__4].r * x[i__5].i + e[i__4].i * x[ i__5].r; q__2.r = b[i__3].r - q__3.r, q__2.i = b[i__3].i - q__3.i; i__6 = *n; i__7 = *n + j * x_dim1; q__4.r = d__[i__6] * x[i__7].r, q__4.i = d__[i__6] * x[ i__7].i; q__1.r = q__2.r - q__4.r, q__1.i = q__2.i - q__4.i; b[i__2].r = q__1.r, b[i__2].i = q__1.i; i__2 = *n - 1; for (i__ = 2; i__ <= i__2; ++i__) { i__3 = i__ + j * b_dim1; i__4 = i__ + j * b_dim1; i__5 = i__ - 1; i__6 = i__ - 1 + j * x_dim1; q__4.r = e[i__5].r * x[i__6].r - e[i__5].i * x[i__6] .i, q__4.i = e[i__5].r * x[i__6].i + e[i__5] .i * x[i__6].r; q__3.r = b[i__4].r - q__4.r, q__3.i = b[i__4].i - q__4.i; i__7 = i__; i__8 = i__ + j * x_dim1; q__5.r = d__[i__7] * x[i__8].r, q__5.i = d__[i__7] * x[i__8].i; q__2.r = q__3.r - q__5.r, q__2.i = q__3.i - q__5.i; r_cnjg(&q__7, &e[i__]); i__9 = i__ + 1 + j * x_dim1; q__6.r = q__7.r * x[i__9].r - q__7.i * x[i__9].i, q__6.i = q__7.r * x[i__9].i + q__7.i * x[i__9] .r; q__1.r = q__2.r - q__6.r, q__1.i = q__2.i - q__6.i; b[i__3].r = q__1.r, b[i__3].i = q__1.i; /* L110: */ } } /* L120: */ } } } return 0; /* End of CLAPTM */ } /* claptm_ */