root / src / lapack / double / dormrz.f @ 2
Historique | Voir | Annoter | Télécharger (8,26 ko)
1 | 1 | equemene | SUBROUTINE DORMRZ( SIDE, TRANS, M, N, K, L, A, LDA, TAU, C, LDC, |
---|---|---|---|
2 | 1 | equemene | $ WORK, LWORK, INFO ) |
3 | 1 | equemene | * |
4 | 1 | equemene | * -- LAPACK routine (version 3.2) -- |
5 | 1 | equemene | * -- LAPACK is a software package provided by Univ. of Tennessee, -- |
6 | 1 | equemene | * -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..-- |
7 | 1 | equemene | * January 2007 |
8 | 1 | equemene | * |
9 | 1 | equemene | * .. Scalar Arguments .. |
10 | 1 | equemene | CHARACTER SIDE, TRANS |
11 | 1 | equemene | INTEGER INFO, K, L, LDA, LDC, LWORK, M, N |
12 | 1 | equemene | * .. |
13 | 1 | equemene | * .. Array Arguments .. |
14 | 1 | equemene | DOUBLE PRECISION A( LDA, * ), C( LDC, * ), TAU( * ), WORK( * ) |
15 | 1 | equemene | * .. |
16 | 1 | equemene | * |
17 | 1 | equemene | * Purpose |
18 | 1 | equemene | * ======= |
19 | 1 | equemene | * |
20 | 1 | equemene | * DORMRZ overwrites the general real M-by-N matrix C with |
21 | 1 | equemene | * |
22 | 1 | equemene | * SIDE = 'L' SIDE = 'R' |
23 | 1 | equemene | * TRANS = 'N': Q * C C * Q |
24 | 1 | equemene | * TRANS = 'T': Q**T * C C * Q**T |
25 | 1 | equemene | * |
26 | 1 | equemene | * where Q is a real orthogonal matrix defined as the product of k |
27 | 1 | equemene | * elementary reflectors |
28 | 1 | equemene | * |
29 | 1 | equemene | * Q = H(1) H(2) . . . H(k) |
30 | 1 | equemene | * |
31 | 1 | equemene | * as returned by DTZRZF. Q is of order M if SIDE = 'L' and of order N |
32 | 1 | equemene | * if SIDE = 'R'. |
33 | 1 | equemene | * |
34 | 1 | equemene | * Arguments |
35 | 1 | equemene | * ========= |
36 | 1 | equemene | * |
37 | 1 | equemene | * SIDE (input) CHARACTER*1 |
38 | 1 | equemene | * = 'L': apply Q or Q**T from the Left; |
39 | 1 | equemene | * = 'R': apply Q or Q**T from the Right. |
40 | 1 | equemene | * |
41 | 1 | equemene | * TRANS (input) CHARACTER*1 |
42 | 1 | equemene | * = 'N': No transpose, apply Q; |
43 | 1 | equemene | * = 'T': Transpose, apply Q**T. |
44 | 1 | equemene | * |
45 | 1 | equemene | * M (input) INTEGER |
46 | 1 | equemene | * The number of rows of the matrix C. M >= 0. |
47 | 1 | equemene | * |
48 | 1 | equemene | * N (input) INTEGER |
49 | 1 | equemene | * The number of columns of the matrix C. N >= 0. |
50 | 1 | equemene | * |
51 | 1 | equemene | * K (input) INTEGER |
52 | 1 | equemene | * The number of elementary reflectors whose product defines |
53 | 1 | equemene | * the matrix Q. |
54 | 1 | equemene | * If SIDE = 'L', M >= K >= 0; |
55 | 1 | equemene | * if SIDE = 'R', N >= K >= 0. |
56 | 1 | equemene | * |
57 | 1 | equemene | * L (input) INTEGER |
58 | 1 | equemene | * The number of columns of the matrix A containing |
59 | 1 | equemene | * the meaningful part of the Householder reflectors. |
60 | 1 | equemene | * If SIDE = 'L', M >= L >= 0, if SIDE = 'R', N >= L >= 0. |
61 | 1 | equemene | * |
62 | 1 | equemene | * A (input) DOUBLE PRECISION array, dimension |
63 | 1 | equemene | * (LDA,M) if SIDE = 'L', |
64 | 1 | equemene | * (LDA,N) if SIDE = 'R' |
65 | 1 | equemene | * The i-th row must contain the vector which defines the |
66 | 1 | equemene | * elementary reflector H(i), for i = 1,2,...,k, as returned by |
67 | 1 | equemene | * DTZRZF in the last k rows of its array argument A. |
68 | 1 | equemene | * A is modified by the routine but restored on exit. |
69 | 1 | equemene | * |
70 | 1 | equemene | * LDA (input) INTEGER |
71 | 1 | equemene | * The leading dimension of the array A. LDA >= max(1,K). |
72 | 1 | equemene | * |
73 | 1 | equemene | * TAU (input) DOUBLE PRECISION array, dimension (K) |
74 | 1 | equemene | * TAU(i) must contain the scalar factor of the elementary |
75 | 1 | equemene | * reflector H(i), as returned by DTZRZF. |
76 | 1 | equemene | * |
77 | 1 | equemene | * C (input/output) DOUBLE PRECISION array, dimension (LDC,N) |
78 | 1 | equemene | * On entry, the M-by-N matrix C. |
79 | 1 | equemene | * On exit, C is overwritten by Q*C or Q**H*C or C*Q**H or C*Q. |
80 | 1 | equemene | * |
81 | 1 | equemene | * LDC (input) INTEGER |
82 | 1 | equemene | * The leading dimension of the array C. LDC >= max(1,M). |
83 | 1 | equemene | * |
84 | 1 | equemene | * WORK (workspace/output) DOUBLE PRECISION array, dimension (MAX(1,LWORK)) |
85 | 1 | equemene | * On exit, if INFO = 0, WORK(1) returns the optimal LWORK. |
86 | 1 | equemene | * |
87 | 1 | equemene | * LWORK (input) INTEGER |
88 | 1 | equemene | * The dimension of the array WORK. |
89 | 1 | equemene | * If SIDE = 'L', LWORK >= max(1,N); |
90 | 1 | equemene | * if SIDE = 'R', LWORK >= max(1,M). |
91 | 1 | equemene | * For optimum performance LWORK >= N*NB if SIDE = 'L', and |
92 | 1 | equemene | * LWORK >= M*NB if SIDE = 'R', where NB is the optimal |
93 | 1 | equemene | * blocksize. |
94 | 1 | equemene | * |
95 | 1 | equemene | * If LWORK = -1, then a workspace query is assumed; the routine |
96 | 1 | equemene | * only calculates the optimal size of the WORK array, returns |
97 | 1 | equemene | * this value as the first entry of the WORK array, and no error |
98 | 1 | equemene | * message related to LWORK is issued by XERBLA. |
99 | 1 | equemene | * |
100 | 1 | equemene | * INFO (output) INTEGER |
101 | 1 | equemene | * = 0: successful exit |
102 | 1 | equemene | * < 0: if INFO = -i, the i-th argument had an illegal value |
103 | 1 | equemene | * |
104 | 1 | equemene | * Further Details |
105 | 1 | equemene | * =============== |
106 | 1 | equemene | * |
107 | 1 | equemene | * Based on contributions by |
108 | 1 | equemene | * A. Petitet, Computer Science Dept., Univ. of Tenn., Knoxville, USA |
109 | 1 | equemene | * |
110 | 1 | equemene | * ===================================================================== |
111 | 1 | equemene | * |
112 | 1 | equemene | * .. Parameters .. |
113 | 1 | equemene | INTEGER NBMAX, LDT |
114 | 1 | equemene | PARAMETER ( NBMAX = 64, LDT = NBMAX+1 ) |
115 | 1 | equemene | * .. |
116 | 1 | equemene | * .. Local Scalars .. |
117 | 1 | equemene | LOGICAL LEFT, LQUERY, NOTRAN |
118 | 1 | equemene | CHARACTER TRANST |
119 | 1 | equemene | INTEGER I, I1, I2, I3, IB, IC, IINFO, IWS, JA, JC, |
120 | 1 | equemene | $ LDWORK, LWKOPT, MI, NB, NBMIN, NI, NQ, NW |
121 | 1 | equemene | * .. |
122 | 1 | equemene | * .. Local Arrays .. |
123 | 1 | equemene | DOUBLE PRECISION T( LDT, NBMAX ) |
124 | 1 | equemene | * .. |
125 | 1 | equemene | * .. External Functions .. |
126 | 1 | equemene | LOGICAL LSAME |
127 | 1 | equemene | INTEGER ILAENV |
128 | 1 | equemene | EXTERNAL LSAME, ILAENV |
129 | 1 | equemene | * .. |
130 | 1 | equemene | * .. External Subroutines .. |
131 | 1 | equemene | EXTERNAL DLARZB, DLARZT, DORMR3, XERBLA |
132 | 1 | equemene | * .. |
133 | 1 | equemene | * .. Intrinsic Functions .. |
134 | 1 | equemene | INTRINSIC MAX, MIN |
135 | 1 | equemene | * .. |
136 | 1 | equemene | * .. Executable Statements .. |
137 | 1 | equemene | * |
138 | 1 | equemene | * Test the input arguments |
139 | 1 | equemene | * |
140 | 1 | equemene | INFO = 0 |
141 | 1 | equemene | LEFT = LSAME( SIDE, 'L' ) |
142 | 1 | equemene | NOTRAN = LSAME( TRANS, 'N' ) |
143 | 1 | equemene | LQUERY = ( LWORK.EQ.-1 ) |
144 | 1 | equemene | * |
145 | 1 | equemene | * NQ is the order of Q and NW is the minimum dimension of WORK |
146 | 1 | equemene | * |
147 | 1 | equemene | IF( LEFT ) THEN |
148 | 1 | equemene | NQ = M |
149 | 1 | equemene | NW = MAX( 1, N ) |
150 | 1 | equemene | ELSE |
151 | 1 | equemene | NQ = N |
152 | 1 | equemene | NW = MAX( 1, M ) |
153 | 1 | equemene | END IF |
154 | 1 | equemene | IF( .NOT.LEFT .AND. .NOT.LSAME( SIDE, 'R' ) ) THEN |
155 | 1 | equemene | INFO = -1 |
156 | 1 | equemene | ELSE IF( .NOT.NOTRAN .AND. .NOT.LSAME( TRANS, 'T' ) ) THEN |
157 | 1 | equemene | INFO = -2 |
158 | 1 | equemene | ELSE IF( M.LT.0 ) THEN |
159 | 1 | equemene | INFO = -3 |
160 | 1 | equemene | ELSE IF( N.LT.0 ) THEN |
161 | 1 | equemene | INFO = -4 |
162 | 1 | equemene | ELSE IF( K.LT.0 .OR. K.GT.NQ ) THEN |
163 | 1 | equemene | INFO = -5 |
164 | 1 | equemene | ELSE IF( L.LT.0 .OR. ( LEFT .AND. ( L.GT.M ) ) .OR. |
165 | 1 | equemene | $ ( .NOT.LEFT .AND. ( L.GT.N ) ) ) THEN |
166 | 1 | equemene | INFO = -6 |
167 | 1 | equemene | ELSE IF( LDA.LT.MAX( 1, K ) ) THEN |
168 | 1 | equemene | INFO = -8 |
169 | 1 | equemene | ELSE IF( LDC.LT.MAX( 1, M ) ) THEN |
170 | 1 | equemene | INFO = -11 |
171 | 1 | equemene | END IF |
172 | 1 | equemene | * |
173 | 1 | equemene | IF( INFO.EQ.0 ) THEN |
174 | 1 | equemene | IF( M.EQ.0 .OR. N.EQ.0 ) THEN |
175 | 1 | equemene | LWKOPT = 1 |
176 | 1 | equemene | ELSE |
177 | 1 | equemene | * |
178 | 1 | equemene | * Determine the block size. NB may be at most NBMAX, where |
179 | 1 | equemene | * NBMAX is used to define the local array T. |
180 | 1 | equemene | * |
181 | 1 | equemene | NB = MIN( NBMAX, ILAENV( 1, 'DORMRQ', SIDE // TRANS, M, N, |
182 | 1 | equemene | $ K, -1 ) ) |
183 | 1 | equemene | LWKOPT = NW*NB |
184 | 1 | equemene | END IF |
185 | 1 | equemene | WORK( 1 ) = LWKOPT |
186 | 1 | equemene | * |
187 | 1 | equemene | IF( LWORK.LT.MAX( 1, NW ) .AND. .NOT.LQUERY ) THEN |
188 | 1 | equemene | INFO = -13 |
189 | 1 | equemene | END IF |
190 | 1 | equemene | END IF |
191 | 1 | equemene | * |
192 | 1 | equemene | IF( INFO.NE.0 ) THEN |
193 | 1 | equemene | CALL XERBLA( 'DORMRZ', -INFO ) |
194 | 1 | equemene | RETURN |
195 | 1 | equemene | ELSE IF( LQUERY ) THEN |
196 | 1 | equemene | RETURN |
197 | 1 | equemene | END IF |
198 | 1 | equemene | * |
199 | 1 | equemene | * Quick return if possible |
200 | 1 | equemene | * |
201 | 1 | equemene | IF( M.EQ.0 .OR. N.EQ.0 ) THEN |
202 | 1 | equemene | WORK( 1 ) = 1 |
203 | 1 | equemene | RETURN |
204 | 1 | equemene | END IF |
205 | 1 | equemene | * |
206 | 1 | equemene | NBMIN = 2 |
207 | 1 | equemene | LDWORK = NW |
208 | 1 | equemene | IF( NB.GT.1 .AND. NB.LT.K ) THEN |
209 | 1 | equemene | IWS = NW*NB |
210 | 1 | equemene | IF( LWORK.LT.IWS ) THEN |
211 | 1 | equemene | NB = LWORK / LDWORK |
212 | 1 | equemene | NBMIN = MAX( 2, ILAENV( 2, 'DORMRQ', SIDE // TRANS, M, N, K, |
213 | 1 | equemene | $ -1 ) ) |
214 | 1 | equemene | END IF |
215 | 1 | equemene | ELSE |
216 | 1 | equemene | IWS = NW |
217 | 1 | equemene | END IF |
218 | 1 | equemene | * |
219 | 1 | equemene | IF( NB.LT.NBMIN .OR. NB.GE.K ) THEN |
220 | 1 | equemene | * |
221 | 1 | equemene | * Use unblocked code |
222 | 1 | equemene | * |
223 | 1 | equemene | CALL DORMR3( SIDE, TRANS, M, N, K, L, A, LDA, TAU, C, LDC, |
224 | 1 | equemene | $ WORK, IINFO ) |
225 | 1 | equemene | ELSE |
226 | 1 | equemene | * |
227 | 1 | equemene | * Use blocked code |
228 | 1 | equemene | * |
229 | 1 | equemene | IF( ( LEFT .AND. .NOT.NOTRAN ) .OR. |
230 | 1 | equemene | $ ( .NOT.LEFT .AND. NOTRAN ) ) THEN |
231 | 1 | equemene | I1 = 1 |
232 | 1 | equemene | I2 = K |
233 | 1 | equemene | I3 = NB |
234 | 1 | equemene | ELSE |
235 | 1 | equemene | I1 = ( ( K-1 ) / NB )*NB + 1 |
236 | 1 | equemene | I2 = 1 |
237 | 1 | equemene | I3 = -NB |
238 | 1 | equemene | END IF |
239 | 1 | equemene | * |
240 | 1 | equemene | IF( LEFT ) THEN |
241 | 1 | equemene | NI = N |
242 | 1 | equemene | JC = 1 |
243 | 1 | equemene | JA = M - L + 1 |
244 | 1 | equemene | ELSE |
245 | 1 | equemene | MI = M |
246 | 1 | equemene | IC = 1 |
247 | 1 | equemene | JA = N - L + 1 |
248 | 1 | equemene | END IF |
249 | 1 | equemene | * |
250 | 1 | equemene | IF( NOTRAN ) THEN |
251 | 1 | equemene | TRANST = 'T' |
252 | 1 | equemene | ELSE |
253 | 1 | equemene | TRANST = 'N' |
254 | 1 | equemene | END IF |
255 | 1 | equemene | * |
256 | 1 | equemene | DO 10 I = I1, I2, I3 |
257 | 1 | equemene | IB = MIN( NB, K-I+1 ) |
258 | 1 | equemene | * |
259 | 1 | equemene | * Form the triangular factor of the block reflector |
260 | 1 | equemene | * H = H(i+ib-1) . . . H(i+1) H(i) |
261 | 1 | equemene | * |
262 | 1 | equemene | CALL DLARZT( 'Backward', 'Rowwise', L, IB, A( I, JA ), LDA, |
263 | 1 | equemene | $ TAU( I ), T, LDT ) |
264 | 1 | equemene | * |
265 | 1 | equemene | IF( LEFT ) THEN |
266 | 1 | equemene | * |
267 | 1 | equemene | * H or H' is applied to C(i:m,1:n) |
268 | 1 | equemene | * |
269 | 1 | equemene | MI = M - I + 1 |
270 | 1 | equemene | IC = I |
271 | 1 | equemene | ELSE |
272 | 1 | equemene | * |
273 | 1 | equemene | * H or H' is applied to C(1:m,i:n) |
274 | 1 | equemene | * |
275 | 1 | equemene | NI = N - I + 1 |
276 | 1 | equemene | JC = I |
277 | 1 | equemene | END IF |
278 | 1 | equemene | * |
279 | 1 | equemene | * Apply H or H' |
280 | 1 | equemene | * |
281 | 1 | equemene | CALL DLARZB( SIDE, TRANST, 'Backward', 'Rowwise', MI, NI, |
282 | 1 | equemene | $ IB, L, A( I, JA ), LDA, T, LDT, C( IC, JC ), |
283 | 1 | equemene | $ LDC, WORK, LDWORK ) |
284 | 1 | equemene | 10 CONTINUE |
285 | 1 | equemene | * |
286 | 1 | equemene | END IF |
287 | 1 | equemene | * |
288 | 1 | equemene | WORK( 1 ) = LWKOPT |
289 | 1 | equemene | * |
290 | 1 | equemene | RETURN |
291 | 1 | equemene | * |
292 | 1 | equemene | * End of DORMRZ |
293 | 1 | equemene | * |
294 | 1 | equemene | END |