|
5164
|
1 SUBROUTINE ZGTTRS( TRANS, N, NRHS, DL, D, DU, DU2, IPIV, B, LDB, |
|
|
2 $ INFO ) |
|
|
3 * |
|
7034
|
4 * -- LAPACK routine (version 3.1) -- |
|
|
5 * Univ. of Tennessee, Univ. of California Berkeley and NAG Ltd.. |
|
|
6 * November 2006 |
|
5164
|
7 * |
|
|
8 * .. Scalar Arguments .. |
|
|
9 CHARACTER TRANS |
|
|
10 INTEGER INFO, LDB, N, NRHS |
|
|
11 * .. |
|
|
12 * .. Array Arguments .. |
|
|
13 INTEGER IPIV( * ) |
|
|
14 COMPLEX*16 B( LDB, * ), D( * ), DL( * ), DU( * ), DU2( * ) |
|
|
15 * .. |
|
|
16 * |
|
|
17 * Purpose |
|
|
18 * ======= |
|
|
19 * |
|
|
20 * ZGTTRS solves one of the systems of equations |
|
|
21 * A * X = B, A**T * X = B, or A**H * X = B, |
|
|
22 * with a tridiagonal matrix A using the LU factorization computed |
|
|
23 * by ZGTTRF. |
|
|
24 * |
|
|
25 * Arguments |
|
|
26 * ========= |
|
|
27 * |
|
7034
|
28 * TRANS (input) CHARACTER*1 |
|
|
29 * Specifies the form of the system of equations. |
|
5164
|
30 * = 'N': A * X = B (No transpose) |
|
|
31 * = 'T': A**T * X = B (Transpose) |
|
|
32 * = 'C': A**H * X = B (Conjugate transpose) |
|
|
33 * |
|
|
34 * N (input) INTEGER |
|
7034
|
35 * The order of the matrix A. |
|
5164
|
36 * |
|
|
37 * NRHS (input) INTEGER |
|
|
38 * The number of right hand sides, i.e., the number of columns |
|
|
39 * of the matrix B. NRHS >= 0. |
|
|
40 * |
|
|
41 * DL (input) COMPLEX*16 array, dimension (N-1) |
|
|
42 * The (n-1) multipliers that define the matrix L from the |
|
|
43 * LU factorization of A. |
|
|
44 * |
|
|
45 * D (input) COMPLEX*16 array, dimension (N) |
|
|
46 * The n diagonal elements of the upper triangular matrix U from |
|
|
47 * the LU factorization of A. |
|
|
48 * |
|
|
49 * DU (input) COMPLEX*16 array, dimension (N-1) |
|
7034
|
50 * The (n-1) elements of the first super-diagonal of U. |
|
5164
|
51 * |
|
|
52 * DU2 (input) COMPLEX*16 array, dimension (N-2) |
|
7034
|
53 * The (n-2) elements of the second super-diagonal of U. |
|
5164
|
54 * |
|
|
55 * IPIV (input) INTEGER array, dimension (N) |
|
|
56 * The pivot indices; for 1 <= i <= n, row i of the matrix was |
|
|
57 * interchanged with row IPIV(i). IPIV(i) will always be either |
|
|
58 * i or i+1; IPIV(i) = i indicates a row interchange was not |
|
|
59 * required. |
|
|
60 * |
|
|
61 * B (input/output) COMPLEX*16 array, dimension (LDB,NRHS) |
|
7034
|
62 * On entry, the matrix of right hand side vectors B. |
|
|
63 * On exit, B is overwritten by the solution vectors X. |
|
5164
|
64 * |
|
|
65 * LDB (input) INTEGER |
|
|
66 * The leading dimension of the array B. LDB >= max(1,N). |
|
|
67 * |
|
|
68 * INFO (output) INTEGER |
|
|
69 * = 0: successful exit |
|
7034
|
70 * < 0: if INFO = -k, the k-th argument had an illegal value |
|
5164
|
71 * |
|
|
72 * ===================================================================== |
|
|
73 * |
|
|
74 * .. Local Scalars .. |
|
|
75 LOGICAL NOTRAN |
|
7034
|
76 INTEGER ITRANS, J, JB, NB |
|
5164
|
77 * .. |
|
|
78 * .. External Functions .. |
|
7034
|
79 INTEGER ILAENV |
|
|
80 EXTERNAL ILAENV |
|
5164
|
81 * .. |
|
|
82 * .. External Subroutines .. |
|
7034
|
83 EXTERNAL XERBLA, ZGTTS2 |
|
5164
|
84 * .. |
|
|
85 * .. Intrinsic Functions .. |
|
7034
|
86 INTRINSIC MAX, MIN |
|
5164
|
87 * .. |
|
|
88 * .. Executable Statements .. |
|
|
89 * |
|
|
90 INFO = 0 |
|
7034
|
91 NOTRAN = ( TRANS.EQ.'N' .OR. TRANS.EQ.'n' ) |
|
|
92 IF( .NOT.NOTRAN .AND. .NOT.( TRANS.EQ.'T' .OR. TRANS.EQ. |
|
|
93 $ 't' ) .AND. .NOT.( TRANS.EQ.'C' .OR. TRANS.EQ.'c' ) ) THEN |
|
5164
|
94 INFO = -1 |
|
|
95 ELSE IF( N.LT.0 ) THEN |
|
|
96 INFO = -2 |
|
|
97 ELSE IF( NRHS.LT.0 ) THEN |
|
|
98 INFO = -3 |
|
|
99 ELSE IF( LDB.LT.MAX( N, 1 ) ) THEN |
|
|
100 INFO = -10 |
|
|
101 END IF |
|
|
102 IF( INFO.NE.0 ) THEN |
|
|
103 CALL XERBLA( 'ZGTTRS', -INFO ) |
|
|
104 RETURN |
|
|
105 END IF |
|
|
106 * |
|
|
107 * Quick return if possible |
|
|
108 * |
|
|
109 IF( N.EQ.0 .OR. NRHS.EQ.0 ) |
|
|
110 $ RETURN |
|
|
111 * |
|
7034
|
112 * Decode TRANS |
|
5164
|
113 * |
|
7034
|
114 IF( NOTRAN ) THEN |
|
|
115 ITRANS = 0 |
|
|
116 ELSE IF( TRANS.EQ.'T' .OR. TRANS.EQ.'t' ) THEN |
|
|
117 ITRANS = 1 |
|
|
118 ELSE |
|
|
119 ITRANS = 2 |
|
|
120 END IF |
|
5164
|
121 * |
|
7034
|
122 * Determine the number of right-hand sides to solve at a time. |
|
5164
|
123 * |
|
7034
|
124 IF( NRHS.EQ.1 ) THEN |
|
|
125 NB = 1 |
|
5164
|
126 ELSE |
|
7034
|
127 NB = MAX( 1, ILAENV( 1, 'ZGTTRS', TRANS, N, NRHS, -1, -1 ) ) |
|
|
128 END IF |
|
5164
|
129 * |
|
7034
|
130 IF( NB.GE.NRHS ) THEN |
|
|
131 CALL ZGTTS2( ITRANS, N, NRHS, DL, D, DU, DU2, IPIV, B, LDB ) |
|
|
132 ELSE |
|
|
133 DO 10 J = 1, NRHS, NB |
|
|
134 JB = MIN( NRHS-J+1, NB ) |
|
|
135 CALL ZGTTS2( ITRANS, N, JB, DL, D, DU, DU2, IPIV, B( 1, J ), |
|
|
136 $ LDB ) |
|
|
137 10 CONTINUE |
|
5164
|
138 END IF |
|
|
139 * |
|
|
140 * End of ZGTTRS |
|
|
141 * |
|
|
142 END |
