Annotation of rpl/lapack/lapack/zgelsy.f, revision 1.18
1.9 bertrand 1: *> \brief <b> ZGELSY solves overdetermined or underdetermined systems for GE matrices</b>
2: *
3: * =========== DOCUMENTATION ===========
4: *
1.15 bertrand 5: * Online html documentation available at
6: * http://www.netlib.org/lapack/explore-html/
1.9 bertrand 7: *
8: *> \htmlonly
1.15 bertrand 9: *> Download ZGELSY + dependencies
10: *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.tgz?format=tgz&filename=/lapack/lapack_routine/zgelsy.f">
11: *> [TGZ]</a>
12: *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.zip?format=zip&filename=/lapack/lapack_routine/zgelsy.f">
13: *> [ZIP]</a>
14: *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.txt?format=txt&filename=/lapack/lapack_routine/zgelsy.f">
1.9 bertrand 15: *> [TXT]</a>
1.15 bertrand 16: *> \endhtmlonly
1.9 bertrand 17: *
18: * Definition:
19: * ===========
20: *
21: * SUBROUTINE ZGELSY( M, N, NRHS, A, LDA, B, LDB, JPVT, RCOND, RANK,
22: * WORK, LWORK, RWORK, INFO )
1.15 bertrand 23: *
1.9 bertrand 24: * .. Scalar Arguments ..
25: * INTEGER INFO, LDA, LDB, LWORK, M, N, NRHS, RANK
26: * DOUBLE PRECISION RCOND
27: * ..
28: * .. Array Arguments ..
29: * INTEGER JPVT( * )
30: * DOUBLE PRECISION RWORK( * )
31: * COMPLEX*16 A( LDA, * ), B( LDB, * ), WORK( * )
32: * ..
1.15 bertrand 33: *
1.9 bertrand 34: *
35: *> \par Purpose:
36: * =============
37: *>
38: *> \verbatim
39: *>
40: *> ZGELSY computes the minimum-norm solution to a complex linear least
41: *> squares problem:
42: *> minimize || A * X - B ||
43: *> using a complete orthogonal factorization of A. A is an M-by-N
44: *> matrix which may be rank-deficient.
45: *>
46: *> Several right hand side vectors b and solution vectors x can be
47: *> handled in a single call; they are stored as the columns of the
48: *> M-by-NRHS right hand side matrix B and the N-by-NRHS solution
49: *> matrix X.
50: *>
51: *> The routine first computes a QR factorization with column pivoting:
52: *> A * P = Q * [ R11 R12 ]
53: *> [ 0 R22 ]
54: *> with R11 defined as the largest leading submatrix whose estimated
55: *> condition number is less than 1/RCOND. The order of R11, RANK,
56: *> is the effective rank of A.
57: *>
58: *> Then, R22 is considered to be negligible, and R12 is annihilated
59: *> by unitary transformations from the right, arriving at the
60: *> complete orthogonal factorization:
61: *> A * P = Q * [ T11 0 ] * Z
62: *> [ 0 0 ]
63: *> The minimum-norm solution is then
64: *> X = P * Z**H [ inv(T11)*Q1**H*B ]
65: *> [ 0 ]
66: *> where Q1 consists of the first RANK columns of Q.
67: *>
68: *> This routine is basically identical to the original xGELSX except
69: *> three differences:
70: *> o The permutation of matrix B (the right hand side) is faster and
71: *> more simple.
72: *> o The call to the subroutine xGEQPF has been substituted by the
73: *> the call to the subroutine xGEQP3. This subroutine is a Blas-3
74: *> version of the QR factorization with column pivoting.
75: *> o Matrix B (the right hand side) is updated with Blas-3.
76: *> \endverbatim
77: *
78: * Arguments:
79: * ==========
80: *
81: *> \param[in] M
82: *> \verbatim
83: *> M is INTEGER
84: *> The number of rows of the matrix A. M >= 0.
85: *> \endverbatim
86: *>
87: *> \param[in] N
88: *> \verbatim
89: *> N is INTEGER
90: *> The number of columns of the matrix A. N >= 0.
91: *> \endverbatim
92: *>
93: *> \param[in] NRHS
94: *> \verbatim
95: *> NRHS is INTEGER
96: *> The number of right hand sides, i.e., the number of
97: *> columns of matrices B and X. NRHS >= 0.
98: *> \endverbatim
99: *>
100: *> \param[in,out] A
101: *> \verbatim
102: *> A is COMPLEX*16 array, dimension (LDA,N)
103: *> On entry, the M-by-N matrix A.
104: *> On exit, A has been overwritten by details of its
105: *> complete orthogonal factorization.
106: *> \endverbatim
107: *>
108: *> \param[in] LDA
109: *> \verbatim
110: *> LDA is INTEGER
111: *> The leading dimension of the array A. LDA >= max(1,M).
112: *> \endverbatim
113: *>
114: *> \param[in,out] B
115: *> \verbatim
116: *> B is COMPLEX*16 array, dimension (LDB,NRHS)
117: *> On entry, the M-by-NRHS right hand side matrix B.
118: *> On exit, the N-by-NRHS solution matrix X.
119: *> \endverbatim
120: *>
121: *> \param[in] LDB
122: *> \verbatim
123: *> LDB is INTEGER
124: *> The leading dimension of the array B. LDB >= max(1,M,N).
125: *> \endverbatim
126: *>
127: *> \param[in,out] JPVT
128: *> \verbatim
129: *> JPVT is INTEGER array, dimension (N)
130: *> On entry, if JPVT(i) .ne. 0, the i-th column of A is permuted
131: *> to the front of AP, otherwise column i is a free column.
132: *> On exit, if JPVT(i) = k, then the i-th column of A*P
133: *> was the k-th column of A.
134: *> \endverbatim
135: *>
136: *> \param[in] RCOND
137: *> \verbatim
138: *> RCOND is DOUBLE PRECISION
139: *> RCOND is used to determine the effective rank of A, which
140: *> is defined as the order of the largest leading triangular
141: *> submatrix R11 in the QR factorization with pivoting of A,
142: *> whose estimated condition number < 1/RCOND.
143: *> \endverbatim
144: *>
145: *> \param[out] RANK
146: *> \verbatim
147: *> RANK is INTEGER
148: *> The effective rank of A, i.e., the order of the submatrix
149: *> R11. This is the same as the order of the submatrix T11
150: *> in the complete orthogonal factorization of A.
151: *> \endverbatim
152: *>
153: *> \param[out] WORK
154: *> \verbatim
155: *> WORK is COMPLEX*16 array, dimension (MAX(1,LWORK))
156: *> On exit, if INFO = 0, WORK(1) returns the optimal LWORK.
157: *> \endverbatim
158: *>
159: *> \param[in] LWORK
160: *> \verbatim
161: *> LWORK is INTEGER
162: *> The dimension of the array WORK.
163: *> The unblocked strategy requires that:
164: *> LWORK >= MN + MAX( 2*MN, N+1, MN+NRHS )
165: *> where MN = min(M,N).
166: *> The block algorithm requires that:
167: *> LWORK >= MN + MAX( 2*MN, NB*(N+1), MN+MN*NB, MN+NB*NRHS )
168: *> where NB is an upper bound on the blocksize returned
169: *> by ILAENV for the routines ZGEQP3, ZTZRZF, CTZRQF, ZUNMQR,
170: *> and ZUNMRZ.
171: *>
172: *> If LWORK = -1, then a workspace query is assumed; the routine
173: *> only calculates the optimal size of the WORK array, returns
174: *> this value as the first entry of the WORK array, and no error
175: *> message related to LWORK is issued by XERBLA.
176: *> \endverbatim
177: *>
178: *> \param[out] RWORK
179: *> \verbatim
180: *> RWORK is DOUBLE PRECISION array, dimension (2*N)
181: *> \endverbatim
182: *>
183: *> \param[out] INFO
184: *> \verbatim
185: *> INFO is INTEGER
186: *> = 0: successful exit
187: *> < 0: if INFO = -i, the i-th argument had an illegal value
188: *> \endverbatim
189: *
190: * Authors:
191: * ========
192: *
1.15 bertrand 193: *> \author Univ. of Tennessee
194: *> \author Univ. of California Berkeley
195: *> \author Univ. of Colorado Denver
196: *> \author NAG Ltd.
1.9 bertrand 197: *
198: *> \ingroup complex16GEsolve
199: *
200: *> \par Contributors:
201: * ==================
202: *>
1.15 bertrand 203: *> A. Petitet, Computer Science Dept., Univ. of Tenn., Knoxville, USA \n
1.9 bertrand 204: *> E. Quintana-Orti, Depto. de Informatica, Universidad Jaime I, Spain \n
205: *> G. Quintana-Orti, Depto. de Informatica, Universidad Jaime I, Spain \n
206: *>
207: * =====================================================================
1.1 bertrand 208: SUBROUTINE ZGELSY( M, N, NRHS, A, LDA, B, LDB, JPVT, RCOND, RANK,
209: $ WORK, LWORK, RWORK, INFO )
210: *
1.18 ! bertrand 211: * -- LAPACK driver routine --
1.1 bertrand 212: * -- LAPACK is a software package provided by Univ. of Tennessee, --
213: * -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
214: *
215: * .. Scalar Arguments ..
216: INTEGER INFO, LDA, LDB, LWORK, M, N, NRHS, RANK
217: DOUBLE PRECISION RCOND
218: * ..
219: * .. Array Arguments ..
220: INTEGER JPVT( * )
221: DOUBLE PRECISION RWORK( * )
222: COMPLEX*16 A( LDA, * ), B( LDB, * ), WORK( * )
223: * ..
224: *
225: * =====================================================================
226: *
227: * .. Parameters ..
228: INTEGER IMAX, IMIN
229: PARAMETER ( IMAX = 1, IMIN = 2 )
230: DOUBLE PRECISION ZERO, ONE
231: PARAMETER ( ZERO = 0.0D+0, ONE = 1.0D+0 )
232: COMPLEX*16 CZERO, CONE
233: PARAMETER ( CZERO = ( 0.0D+0, 0.0D+0 ),
234: $ CONE = ( 1.0D+0, 0.0D+0 ) )
235: * ..
236: * .. Local Scalars ..
237: LOGICAL LQUERY
238: INTEGER I, IASCL, IBSCL, ISMAX, ISMIN, J, LWKOPT, MN,
239: $ NB, NB1, NB2, NB3, NB4
240: DOUBLE PRECISION ANRM, BIGNUM, BNRM, SMAX, SMAXPR, SMIN, SMINPR,
241: $ SMLNUM, WSIZE
242: COMPLEX*16 C1, C2, S1, S2
243: * ..
244: * .. External Subroutines ..
245: EXTERNAL DLABAD, XERBLA, ZCOPY, ZGEQP3, ZLAIC1, ZLASCL,
246: $ ZLASET, ZTRSM, ZTZRZF, ZUNMQR, ZUNMRZ
247: * ..
248: * .. External Functions ..
249: INTEGER ILAENV
250: DOUBLE PRECISION DLAMCH, ZLANGE
251: EXTERNAL ILAENV, DLAMCH, ZLANGE
252: * ..
253: * .. Intrinsic Functions ..
254: INTRINSIC ABS, DBLE, DCMPLX, MAX, MIN
255: * ..
256: * .. Executable Statements ..
257: *
258: MN = MIN( M, N )
259: ISMIN = MN + 1
260: ISMAX = 2*MN + 1
261: *
262: * Test the input arguments.
263: *
264: INFO = 0
265: NB1 = ILAENV( 1, 'ZGEQRF', ' ', M, N, -1, -1 )
266: NB2 = ILAENV( 1, 'ZGERQF', ' ', M, N, -1, -1 )
267: NB3 = ILAENV( 1, 'ZUNMQR', ' ', M, N, NRHS, -1 )
268: NB4 = ILAENV( 1, 'ZUNMRQ', ' ', M, N, NRHS, -1 )
269: NB = MAX( NB1, NB2, NB3, NB4 )
270: LWKOPT = MAX( 1, MN+2*N+NB*( N+1 ), 2*MN+NB*NRHS )
271: WORK( 1 ) = DCMPLX( LWKOPT )
272: LQUERY = ( LWORK.EQ.-1 )
273: IF( M.LT.0 ) THEN
274: INFO = -1
275: ELSE IF( N.LT.0 ) THEN
276: INFO = -2
277: ELSE IF( NRHS.LT.0 ) THEN
278: INFO = -3
279: ELSE IF( LDA.LT.MAX( 1, M ) ) THEN
280: INFO = -5
281: ELSE IF( LDB.LT.MAX( 1, M, N ) ) THEN
282: INFO = -7
283: ELSE IF( LWORK.LT.( MN+MAX( 2*MN, N+1, MN+NRHS ) ) .AND. .NOT.
284: $ LQUERY ) THEN
285: INFO = -12
286: END IF
287: *
288: IF( INFO.NE.0 ) THEN
289: CALL XERBLA( 'ZGELSY', -INFO )
290: RETURN
291: ELSE IF( LQUERY ) THEN
292: RETURN
293: END IF
294: *
295: * Quick return if possible
296: *
297: IF( MIN( M, N, NRHS ).EQ.0 ) THEN
298: RANK = 0
299: RETURN
300: END IF
301: *
302: * Get machine parameters
303: *
304: SMLNUM = DLAMCH( 'S' ) / DLAMCH( 'P' )
305: BIGNUM = ONE / SMLNUM
306: CALL DLABAD( SMLNUM, BIGNUM )
307: *
308: * Scale A, B if max entries outside range [SMLNUM,BIGNUM]
309: *
310: ANRM = ZLANGE( 'M', M, N, A, LDA, RWORK )
311: IASCL = 0
312: IF( ANRM.GT.ZERO .AND. ANRM.LT.SMLNUM ) THEN
313: *
314: * Scale matrix norm up to SMLNUM
315: *
316: CALL ZLASCL( 'G', 0, 0, ANRM, SMLNUM, M, N, A, LDA, INFO )
317: IASCL = 1
318: ELSE IF( ANRM.GT.BIGNUM ) THEN
319: *
320: * Scale matrix norm down to BIGNUM
321: *
322: CALL ZLASCL( 'G', 0, 0, ANRM, BIGNUM, M, N, A, LDA, INFO )
323: IASCL = 2
324: ELSE IF( ANRM.EQ.ZERO ) THEN
325: *
326: * Matrix all zero. Return zero solution.
327: *
328: CALL ZLASET( 'F', MAX( M, N ), NRHS, CZERO, CZERO, B, LDB )
329: RANK = 0
330: GO TO 70
331: END IF
332: *
333: BNRM = ZLANGE( 'M', M, NRHS, B, LDB, RWORK )
334: IBSCL = 0
335: IF( BNRM.GT.ZERO .AND. BNRM.LT.SMLNUM ) THEN
336: *
337: * Scale matrix norm up to SMLNUM
338: *
339: CALL ZLASCL( 'G', 0, 0, BNRM, SMLNUM, M, NRHS, B, LDB, INFO )
340: IBSCL = 1
341: ELSE IF( BNRM.GT.BIGNUM ) THEN
342: *
343: * Scale matrix norm down to BIGNUM
344: *
345: CALL ZLASCL( 'G', 0, 0, BNRM, BIGNUM, M, NRHS, B, LDB, INFO )
346: IBSCL = 2
347: END IF
348: *
349: * Compute QR factorization with column pivoting of A:
350: * A * P = Q * R
351: *
352: CALL ZGEQP3( M, N, A, LDA, JPVT, WORK( 1 ), WORK( MN+1 ),
353: $ LWORK-MN, RWORK, INFO )
354: WSIZE = MN + DBLE( WORK( MN+1 ) )
355: *
356: * complex workspace: MN+NB*(N+1). real workspace 2*N.
357: * Details of Householder rotations stored in WORK(1:MN).
358: *
359: * Determine RANK using incremental condition estimation
360: *
361: WORK( ISMIN ) = CONE
362: WORK( ISMAX ) = CONE
363: SMAX = ABS( A( 1, 1 ) )
364: SMIN = SMAX
365: IF( ABS( A( 1, 1 ) ).EQ.ZERO ) THEN
366: RANK = 0
367: CALL ZLASET( 'F', MAX( M, N ), NRHS, CZERO, CZERO, B, LDB )
368: GO TO 70
369: ELSE
370: RANK = 1
371: END IF
372: *
373: 10 CONTINUE
374: IF( RANK.LT.MN ) THEN
375: I = RANK + 1
376: CALL ZLAIC1( IMIN, RANK, WORK( ISMIN ), SMIN, A( 1, I ),
377: $ A( I, I ), SMINPR, S1, C1 )
378: CALL ZLAIC1( IMAX, RANK, WORK( ISMAX ), SMAX, A( 1, I ),
379: $ A( I, I ), SMAXPR, S2, C2 )
380: *
381: IF( SMAXPR*RCOND.LE.SMINPR ) THEN
382: DO 20 I = 1, RANK
383: WORK( ISMIN+I-1 ) = S1*WORK( ISMIN+I-1 )
384: WORK( ISMAX+I-1 ) = S2*WORK( ISMAX+I-1 )
385: 20 CONTINUE
386: WORK( ISMIN+RANK ) = C1
387: WORK( ISMAX+RANK ) = C2
388: SMIN = SMINPR
389: SMAX = SMAXPR
390: RANK = RANK + 1
391: GO TO 10
392: END IF
393: END IF
394: *
395: * complex workspace: 3*MN.
396: *
397: * Logically partition R = [ R11 R12 ]
398: * [ 0 R22 ]
399: * where R11 = R(1:RANK,1:RANK)
400: *
401: * [R11,R12] = [ T11, 0 ] * Y
402: *
403: IF( RANK.LT.N )
404: $ CALL ZTZRZF( RANK, N, A, LDA, WORK( MN+1 ), WORK( 2*MN+1 ),
405: $ LWORK-2*MN, INFO )
406: *
407: * complex workspace: 2*MN.
408: * Details of Householder rotations stored in WORK(MN+1:2*MN)
409: *
1.8 bertrand 410: * B(1:M,1:NRHS) := Q**H * B(1:M,1:NRHS)
1.1 bertrand 411: *
412: CALL ZUNMQR( 'Left', 'Conjugate transpose', M, NRHS, MN, A, LDA,
413: $ WORK( 1 ), B, LDB, WORK( 2*MN+1 ), LWORK-2*MN, INFO )
414: WSIZE = MAX( WSIZE, 2*MN+DBLE( WORK( 2*MN+1 ) ) )
415: *
416: * complex workspace: 2*MN+NB*NRHS.
417: *
418: * B(1:RANK,1:NRHS) := inv(T11) * B(1:RANK,1:NRHS)
419: *
420: CALL ZTRSM( 'Left', 'Upper', 'No transpose', 'Non-unit', RANK,
421: $ NRHS, CONE, A, LDA, B, LDB )
422: *
423: DO 40 J = 1, NRHS
424: DO 30 I = RANK + 1, N
425: B( I, J ) = CZERO
426: 30 CONTINUE
427: 40 CONTINUE
428: *
1.8 bertrand 429: * B(1:N,1:NRHS) := Y**H * B(1:N,1:NRHS)
1.1 bertrand 430: *
431: IF( RANK.LT.N ) THEN
432: CALL ZUNMRZ( 'Left', 'Conjugate transpose', N, NRHS, RANK,
433: $ N-RANK, A, LDA, WORK( MN+1 ), B, LDB,
434: $ WORK( 2*MN+1 ), LWORK-2*MN, INFO )
435: END IF
436: *
437: * complex workspace: 2*MN+NRHS.
438: *
439: * B(1:N,1:NRHS) := P * B(1:N,1:NRHS)
440: *
441: DO 60 J = 1, NRHS
442: DO 50 I = 1, N
443: WORK( JPVT( I ) ) = B( I, J )
444: 50 CONTINUE
445: CALL ZCOPY( N, WORK( 1 ), 1, B( 1, J ), 1 )
446: 60 CONTINUE
447: *
448: * complex workspace: N.
449: *
450: * Undo scaling
451: *
452: IF( IASCL.EQ.1 ) THEN
453: CALL ZLASCL( 'G', 0, 0, ANRM, SMLNUM, N, NRHS, B, LDB, INFO )
454: CALL ZLASCL( 'U', 0, 0, SMLNUM, ANRM, RANK, RANK, A, LDA,
455: $ INFO )
456: ELSE IF( IASCL.EQ.2 ) THEN
457: CALL ZLASCL( 'G', 0, 0, ANRM, BIGNUM, N, NRHS, B, LDB, INFO )
458: CALL ZLASCL( 'U', 0, 0, BIGNUM, ANRM, RANK, RANK, A, LDA,
459: $ INFO )
460: END IF
461: IF( IBSCL.EQ.1 ) THEN
462: CALL ZLASCL( 'G', 0, 0, SMLNUM, BNRM, N, NRHS, B, LDB, INFO )
463: ELSE IF( IBSCL.EQ.2 ) THEN
464: CALL ZLASCL( 'G', 0, 0, BIGNUM, BNRM, N, NRHS, B, LDB, INFO )
465: END IF
466: *
467: 70 CONTINUE
468: WORK( 1 ) = DCMPLX( LWKOPT )
469: *
470: RETURN
471: *
472: * End of ZGELSY
473: *
474: END
CVSweb interface <joel.bertrand@systella.fr>