You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

chbgvx.f 17 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531
  1. *> \brief \b CHBGVX
  2. *
  3. * =========== DOCUMENTATION ===========
  4. *
  5. * Online html documentation available at
  6. * http://www.netlib.org/lapack/explore-html/
  7. *
  8. *> \htmlonly
  9. *> Download CHBGVX + dependencies
  10. *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.tgz?format=tgz&filename=/lapack/lapack_routine/chbgvx.f">
  11. *> [TGZ]</a>
  12. *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.zip?format=zip&filename=/lapack/lapack_routine/chbgvx.f">
  13. *> [ZIP]</a>
  14. *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.txt?format=txt&filename=/lapack/lapack_routine/chbgvx.f">
  15. *> [TXT]</a>
  16. *> \endhtmlonly
  17. *
  18. * Definition:
  19. * ===========
  20. *
  21. * SUBROUTINE CHBGVX( JOBZ, RANGE, UPLO, N, KA, KB, AB, LDAB, BB,
  22. * LDBB, Q, LDQ, VL, VU, IL, IU, ABSTOL, M, W, Z,
  23. * LDZ, WORK, RWORK, IWORK, IFAIL, INFO )
  24. *
  25. * .. Scalar Arguments ..
  26. * CHARACTER JOBZ, RANGE, UPLO
  27. * INTEGER IL, INFO, IU, KA, KB, LDAB, LDBB, LDQ, LDZ, M,
  28. * $ N
  29. * REAL ABSTOL, VL, VU
  30. * ..
  31. * .. Array Arguments ..
  32. * INTEGER IFAIL( * ), IWORK( * )
  33. * REAL RWORK( * ), W( * )
  34. * COMPLEX AB( LDAB, * ), BB( LDBB, * ), Q( LDQ, * ),
  35. * $ WORK( * ), Z( LDZ, * )
  36. * ..
  37. *
  38. *
  39. *> \par Purpose:
  40. * =============
  41. *>
  42. *> \verbatim
  43. *>
  44. *> CHBGVX computes all the eigenvalues, and optionally, the eigenvectors
  45. *> of a complex generalized Hermitian-definite banded eigenproblem, of
  46. *> the form A*x=(lambda)*B*x. Here A and B are assumed to be Hermitian
  47. *> and banded, and B is also positive definite. Eigenvalues and
  48. *> eigenvectors can be selected by specifying either all eigenvalues,
  49. *> a range of values or a range of indices for the desired eigenvalues.
  50. *> \endverbatim
  51. *
  52. * Arguments:
  53. * ==========
  54. *
  55. *> \param[in] JOBZ
  56. *> \verbatim
  57. *> JOBZ is CHARACTER*1
  58. *> = 'N': Compute eigenvalues only;
  59. *> = 'V': Compute eigenvalues and eigenvectors.
  60. *> \endverbatim
  61. *>
  62. *> \param[in] RANGE
  63. *> \verbatim
  64. *> RANGE is CHARACTER*1
  65. *> = 'A': all eigenvalues will be found;
  66. *> = 'V': all eigenvalues in the half-open interval (VL,VU]
  67. *> will be found;
  68. *> = 'I': the IL-th through IU-th eigenvalues will be found.
  69. *> \endverbatim
  70. *>
  71. *> \param[in] UPLO
  72. *> \verbatim
  73. *> UPLO is CHARACTER*1
  74. *> = 'U': Upper triangles of A and B are stored;
  75. *> = 'L': Lower triangles of A and B are stored.
  76. *> \endverbatim
  77. *>
  78. *> \param[in] N
  79. *> \verbatim
  80. *> N is INTEGER
  81. *> The order of the matrices A and B. N >= 0.
  82. *> \endverbatim
  83. *>
  84. *> \param[in] KA
  85. *> \verbatim
  86. *> KA is INTEGER
  87. *> The number of superdiagonals of the matrix A if UPLO = 'U',
  88. *> or the number of subdiagonals if UPLO = 'L'. KA >= 0.
  89. *> \endverbatim
  90. *>
  91. *> \param[in] KB
  92. *> \verbatim
  93. *> KB is INTEGER
  94. *> The number of superdiagonals of the matrix B if UPLO = 'U',
  95. *> or the number of subdiagonals if UPLO = 'L'. KB >= 0.
  96. *> \endverbatim
  97. *>
  98. *> \param[in,out] AB
  99. *> \verbatim
  100. *> AB is COMPLEX array, dimension (LDAB, N)
  101. *> On entry, the upper or lower triangle of the Hermitian band
  102. *> matrix A, stored in the first ka+1 rows of the array. The
  103. *> j-th column of A is stored in the j-th column of the array AB
  104. *> as follows:
  105. *> if UPLO = 'U', AB(ka+1+i-j,j) = A(i,j) for max(1,j-ka)<=i<=j;
  106. *> if UPLO = 'L', AB(1+i-j,j) = A(i,j) for j<=i<=min(n,j+ka).
  107. *>
  108. *> On exit, the contents of AB are destroyed.
  109. *> \endverbatim
  110. *>
  111. *> \param[in] LDAB
  112. *> \verbatim
  113. *> LDAB is INTEGER
  114. *> The leading dimension of the array AB. LDAB >= KA+1.
  115. *> \endverbatim
  116. *>
  117. *> \param[in,out] BB
  118. *> \verbatim
  119. *> BB is COMPLEX array, dimension (LDBB, N)
  120. *> On entry, the upper or lower triangle of the Hermitian band
  121. *> matrix B, stored in the first kb+1 rows of the array. The
  122. *> j-th column of B is stored in the j-th column of the array BB
  123. *> as follows:
  124. *> if UPLO = 'U', BB(kb+1+i-j,j) = B(i,j) for max(1,j-kb)<=i<=j;
  125. *> if UPLO = 'L', BB(1+i-j,j) = B(i,j) for j<=i<=min(n,j+kb).
  126. *>
  127. *> On exit, the factor S from the split Cholesky factorization
  128. *> B = S**H*S, as returned by CPBSTF.
  129. *> \endverbatim
  130. *>
  131. *> \param[in] LDBB
  132. *> \verbatim
  133. *> LDBB is INTEGER
  134. *> The leading dimension of the array BB. LDBB >= KB+1.
  135. *> \endverbatim
  136. *>
  137. *> \param[out] Q
  138. *> \verbatim
  139. *> Q is COMPLEX array, dimension (LDQ, N)
  140. *> If JOBZ = 'V', the n-by-n matrix used in the reduction of
  141. *> A*x = (lambda)*B*x to standard form, i.e. C*x = (lambda)*x,
  142. *> and consequently C to tridiagonal form.
  143. *> If JOBZ = 'N', the array Q is not referenced.
  144. *> \endverbatim
  145. *>
  146. *> \param[in] LDQ
  147. *> \verbatim
  148. *> LDQ is INTEGER
  149. *> The leading dimension of the array Q. If JOBZ = 'N',
  150. *> LDQ >= 1. If JOBZ = 'V', LDQ >= max(1,N).
  151. *> \endverbatim
  152. *>
  153. *> \param[in] VL
  154. *> \verbatim
  155. *> VL is REAL
  156. *>
  157. *> If RANGE='V', the lower bound of the interval to
  158. *> be searched for eigenvalues. VL < VU.
  159. *> Not referenced if RANGE = 'A' or 'I'.
  160. *> \endverbatim
  161. *>
  162. *> \param[in] VU
  163. *> \verbatim
  164. *> VU is REAL
  165. *>
  166. *> If RANGE='V', the upper bound of the interval to
  167. *> be searched for eigenvalues. VL < VU.
  168. *> Not referenced if RANGE = 'A' or 'I'.
  169. *> \endverbatim
  170. *>
  171. *> \param[in] IL
  172. *> \verbatim
  173. *> IL is INTEGER
  174. *>
  175. *> If RANGE='I', the index of the
  176. *> smallest eigenvalue to be returned.
  177. *> 1 <= IL <= IU <= N, if N > 0; IL = 1 and IU = 0 if N = 0.
  178. *> Not referenced if RANGE = 'A' or 'V'.
  179. *> \endverbatim
  180. *>
  181. *> \param[in] IU
  182. *> \verbatim
  183. *> IU is INTEGER
  184. *>
  185. *> If RANGE='I', the index of the
  186. *> largest eigenvalue to be returned.
  187. *> 1 <= IL <= IU <= N, if N > 0; IL = 1 and IU = 0 if N = 0.
  188. *> Not referenced if RANGE = 'A' or 'V'.
  189. *> \endverbatim
  190. *>
  191. *> \param[in] ABSTOL
  192. *> \verbatim
  193. *> ABSTOL is REAL
  194. *> The absolute error tolerance for the eigenvalues.
  195. *> An approximate eigenvalue is accepted as converged
  196. *> when it is determined to lie in an interval [a,b]
  197. *> of width less than or equal to
  198. *>
  199. *> ABSTOL + EPS * max( |a|,|b| ) ,
  200. *>
  201. *> where EPS is the machine precision. If ABSTOL is less than
  202. *> or equal to zero, then EPS*|T| will be used in its place,
  203. *> where |T| is the 1-norm of the tridiagonal matrix obtained
  204. *> by reducing AP to tridiagonal form.
  205. *>
  206. *> Eigenvalues will be computed most accurately when ABSTOL is
  207. *> set to twice the underflow threshold 2*SLAMCH('S'), not zero.
  208. *> If this routine returns with INFO>0, indicating that some
  209. *> eigenvectors did not converge, try setting ABSTOL to
  210. *> 2*SLAMCH('S').
  211. *> \endverbatim
  212. *>
  213. *> \param[out] M
  214. *> \verbatim
  215. *> M is INTEGER
  216. *> The total number of eigenvalues found. 0 <= M <= N.
  217. *> If RANGE = 'A', M = N, and if RANGE = 'I', M = IU-IL+1.
  218. *> \endverbatim
  219. *>
  220. *> \param[out] W
  221. *> \verbatim
  222. *> W is REAL array, dimension (N)
  223. *> If INFO = 0, the eigenvalues in ascending order.
  224. *> \endverbatim
  225. *>
  226. *> \param[out] Z
  227. *> \verbatim
  228. *> Z is COMPLEX array, dimension (LDZ, N)
  229. *> If JOBZ = 'V', then if INFO = 0, Z contains the matrix Z of
  230. *> eigenvectors, with the i-th column of Z holding the
  231. *> eigenvector associated with W(i). The eigenvectors are
  232. *> normalized so that Z**H*B*Z = I.
  233. *> If JOBZ = 'N', then Z is not referenced.
  234. *> \endverbatim
  235. *>
  236. *> \param[in] LDZ
  237. *> \verbatim
  238. *> LDZ is INTEGER
  239. *> The leading dimension of the array Z. LDZ >= 1, and if
  240. *> JOBZ = 'V', LDZ >= N.
  241. *> \endverbatim
  242. *>
  243. *> \param[out] WORK
  244. *> \verbatim
  245. *> WORK is COMPLEX array, dimension (N)
  246. *> \endverbatim
  247. *>
  248. *> \param[out] RWORK
  249. *> \verbatim
  250. *> RWORK is REAL array, dimension (7*N)
  251. *> \endverbatim
  252. *>
  253. *> \param[out] IWORK
  254. *> \verbatim
  255. *> IWORK is INTEGER array, dimension (5*N)
  256. *> \endverbatim
  257. *>
  258. *> \param[out] IFAIL
  259. *> \verbatim
  260. *> IFAIL is INTEGER array, dimension (N)
  261. *> If JOBZ = 'V', then if INFO = 0, the first M elements of
  262. *> IFAIL are zero. If INFO > 0, then IFAIL contains the
  263. *> indices of the eigenvectors that failed to converge.
  264. *> If JOBZ = 'N', then IFAIL is not referenced.
  265. *> \endverbatim
  266. *>
  267. *> \param[out] INFO
  268. *> \verbatim
  269. *> INFO is INTEGER
  270. *> = 0: successful exit
  271. *> < 0: if INFO = -i, the i-th argument had an illegal value
  272. *> > 0: if INFO = i, and i is:
  273. *> <= N: then i eigenvectors failed to converge. Their
  274. *> indices are stored in array IFAIL.
  275. *> > N: if INFO = N + i, for 1 <= i <= N, then CPBSTF
  276. *> returned INFO = i: B is not positive definite.
  277. *> The factorization of B could not be completed and
  278. *> no eigenvalues or eigenvectors were computed.
  279. *> \endverbatim
  280. *
  281. * Authors:
  282. * ========
  283. *
  284. *> \author Univ. of Tennessee
  285. *> \author Univ. of California Berkeley
  286. *> \author Univ. of Colorado Denver
  287. *> \author NAG Ltd.
  288. *
  289. *> \ingroup complexOTHEReigen
  290. *
  291. *> \par Contributors:
  292. * ==================
  293. *>
  294. *> Mark Fahey, Department of Mathematics, Univ. of Kentucky, USA
  295. *
  296. * =====================================================================
  297. SUBROUTINE CHBGVX( JOBZ, RANGE, UPLO, N, KA, KB, AB, LDAB, BB,
  298. $ LDBB, Q, LDQ, VL, VU, IL, IU, ABSTOL, M, W, Z,
  299. $ LDZ, WORK, RWORK, IWORK, IFAIL, INFO )
  300. *
  301. * -- LAPACK driver routine --
  302. * -- LAPACK is a software package provided by Univ. of Tennessee, --
  303. * -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
  304. *
  305. * .. Scalar Arguments ..
  306. CHARACTER JOBZ, RANGE, UPLO
  307. INTEGER IL, INFO, IU, KA, KB, LDAB, LDBB, LDQ, LDZ, M,
  308. $ N
  309. REAL ABSTOL, VL, VU
  310. * ..
  311. * .. Array Arguments ..
  312. INTEGER IFAIL( * ), IWORK( * )
  313. REAL RWORK( * ), W( * )
  314. COMPLEX AB( LDAB, * ), BB( LDBB, * ), Q( LDQ, * ),
  315. $ WORK( * ), Z( LDZ, * )
  316. * ..
  317. *
  318. * =====================================================================
  319. *
  320. * .. Parameters ..
  321. REAL ZERO
  322. PARAMETER ( ZERO = 0.0E+0 )
  323. COMPLEX CZERO, CONE
  324. PARAMETER ( CZERO = ( 0.0E+0, 0.0E+0 ),
  325. $ CONE = ( 1.0E+0, 0.0E+0 ) )
  326. * ..
  327. * .. Local Scalars ..
  328. LOGICAL ALLEIG, INDEIG, TEST, UPPER, VALEIG, WANTZ
  329. CHARACTER ORDER, VECT
  330. INTEGER I, IINFO, INDD, INDE, INDEE, INDISP,
  331. $ INDIWK, INDRWK, INDWRK, ITMP1, J, JJ, NSPLIT
  332. REAL TMP1
  333. * ..
  334. * .. External Functions ..
  335. LOGICAL LSAME
  336. EXTERNAL LSAME
  337. * ..
  338. * .. External Subroutines ..
  339. EXTERNAL CCOPY, CGEMV, CHBGST, CHBTRD, CLACPY, CPBSTF,
  340. $ CSTEIN, CSTEQR, CSWAP, SCOPY, SSTEBZ, SSTERF,
  341. $ XERBLA
  342. * ..
  343. * .. Intrinsic Functions ..
  344. INTRINSIC MIN
  345. * ..
  346. * .. Executable Statements ..
  347. *
  348. * Test the input parameters.
  349. *
  350. WANTZ = LSAME( JOBZ, 'V' )
  351. UPPER = LSAME( UPLO, 'U' )
  352. ALLEIG = LSAME( RANGE, 'A' )
  353. VALEIG = LSAME( RANGE, 'V' )
  354. INDEIG = LSAME( RANGE, 'I' )
  355. *
  356. INFO = 0
  357. IF( .NOT.( WANTZ .OR. LSAME( JOBZ, 'N' ) ) ) THEN
  358. INFO = -1
  359. ELSE IF( .NOT.( ALLEIG .OR. VALEIG .OR. INDEIG ) ) THEN
  360. INFO = -2
  361. ELSE IF( .NOT.( UPPER .OR. LSAME( UPLO, 'L' ) ) ) THEN
  362. INFO = -3
  363. ELSE IF( N.LT.0 ) THEN
  364. INFO = -4
  365. ELSE IF( KA.LT.0 ) THEN
  366. INFO = -5
  367. ELSE IF( KB.LT.0 .OR. KB.GT.KA ) THEN
  368. INFO = -6
  369. ELSE IF( LDAB.LT.KA+1 ) THEN
  370. INFO = -8
  371. ELSE IF( LDBB.LT.KB+1 ) THEN
  372. INFO = -10
  373. ELSE IF( LDQ.LT.1 .OR. ( WANTZ .AND. LDQ.LT.N ) ) THEN
  374. INFO = -12
  375. ELSE
  376. IF( VALEIG ) THEN
  377. IF( N.GT.0 .AND. VU.LE.VL )
  378. $ INFO = -14
  379. ELSE IF( INDEIG ) THEN
  380. IF( IL.LT.1 .OR. IL.GT.MAX( 1, N ) ) THEN
  381. INFO = -15
  382. ELSE IF ( IU.LT.MIN( N, IL ) .OR. IU.GT.N ) THEN
  383. INFO = -16
  384. END IF
  385. END IF
  386. END IF
  387. IF( INFO.EQ.0) THEN
  388. IF( LDZ.LT.1 .OR. ( WANTZ .AND. LDZ.LT.N ) ) THEN
  389. INFO = -21
  390. END IF
  391. END IF
  392. *
  393. IF( INFO.NE.0 ) THEN
  394. CALL XERBLA( 'CHBGVX', -INFO )
  395. RETURN
  396. END IF
  397. *
  398. * Quick return if possible
  399. *
  400. M = 0
  401. IF( N.EQ.0 )
  402. $ RETURN
  403. *
  404. * Form a split Cholesky factorization of B.
  405. *
  406. CALL CPBSTF( UPLO, N, KB, BB, LDBB, INFO )
  407. IF( INFO.NE.0 ) THEN
  408. INFO = N + INFO
  409. RETURN
  410. END IF
  411. *
  412. * Transform problem to standard eigenvalue problem.
  413. *
  414. CALL CHBGST( JOBZ, UPLO, N, KA, KB, AB, LDAB, BB, LDBB, Q, LDQ,
  415. $ WORK, RWORK, IINFO )
  416. *
  417. * Solve the standard eigenvalue problem.
  418. * Reduce Hermitian band matrix to tridiagonal form.
  419. *
  420. INDD = 1
  421. INDE = INDD + N
  422. INDRWK = INDE + N
  423. INDWRK = 1
  424. IF( WANTZ ) THEN
  425. VECT = 'U'
  426. ELSE
  427. VECT = 'N'
  428. END IF
  429. CALL CHBTRD( VECT, UPLO, N, KA, AB, LDAB, RWORK( INDD ),
  430. $ RWORK( INDE ), Q, LDQ, WORK( INDWRK ), IINFO )
  431. *
  432. * If all eigenvalues are desired and ABSTOL is less than or equal
  433. * to zero, then call SSTERF or CSTEQR. If this fails for some
  434. * eigenvalue, then try SSTEBZ.
  435. *
  436. TEST = .FALSE.
  437. IF( INDEIG ) THEN
  438. IF( IL.EQ.1 .AND. IU.EQ.N ) THEN
  439. TEST = .TRUE.
  440. END IF
  441. END IF
  442. IF( ( ALLEIG .OR. TEST ) .AND. ( ABSTOL.LE.ZERO ) ) THEN
  443. CALL SCOPY( N, RWORK( INDD ), 1, W, 1 )
  444. INDEE = INDRWK + 2*N
  445. CALL SCOPY( N-1, RWORK( INDE ), 1, RWORK( INDEE ), 1 )
  446. IF( .NOT.WANTZ ) THEN
  447. CALL SSTERF( N, W, RWORK( INDEE ), INFO )
  448. ELSE
  449. CALL CLACPY( 'A', N, N, Q, LDQ, Z, LDZ )
  450. CALL CSTEQR( JOBZ, N, W, RWORK( INDEE ), Z, LDZ,
  451. $ RWORK( INDRWK ), INFO )
  452. IF( INFO.EQ.0 ) THEN
  453. DO 10 I = 1, N
  454. IFAIL( I ) = 0
  455. 10 CONTINUE
  456. END IF
  457. END IF
  458. IF( INFO.EQ.0 ) THEN
  459. M = N
  460. GO TO 30
  461. END IF
  462. INFO = 0
  463. END IF
  464. *
  465. * Otherwise, call SSTEBZ and, if eigenvectors are desired,
  466. * call CSTEIN.
  467. *
  468. IF( WANTZ ) THEN
  469. ORDER = 'B'
  470. ELSE
  471. ORDER = 'E'
  472. END IF
  473. INDISP = 1 + N
  474. INDIWK = INDISP + N
  475. CALL SSTEBZ( RANGE, ORDER, N, VL, VU, IL, IU, ABSTOL,
  476. $ RWORK( INDD ), RWORK( INDE ), M, NSPLIT, W,
  477. $ IWORK( 1 ), IWORK( INDISP ), RWORK( INDRWK ),
  478. $ IWORK( INDIWK ), INFO )
  479. *
  480. IF( WANTZ ) THEN
  481. CALL CSTEIN( N, RWORK( INDD ), RWORK( INDE ), M, W,
  482. $ IWORK( 1 ), IWORK( INDISP ), Z, LDZ,
  483. $ RWORK( INDRWK ), IWORK( INDIWK ), IFAIL, INFO )
  484. *
  485. * Apply unitary matrix used in reduction to tridiagonal
  486. * form to eigenvectors returned by CSTEIN.
  487. *
  488. DO 20 J = 1, M
  489. CALL CCOPY( N, Z( 1, J ), 1, WORK( 1 ), 1 )
  490. CALL CGEMV( 'N', N, N, CONE, Q, LDQ, WORK, 1, CZERO,
  491. $ Z( 1, J ), 1 )
  492. 20 CONTINUE
  493. END IF
  494. *
  495. 30 CONTINUE
  496. *
  497. * If eigenvalues are not in order, then sort them, along with
  498. * eigenvectors.
  499. *
  500. IF( WANTZ ) THEN
  501. DO 50 J = 1, M - 1
  502. I = 0
  503. TMP1 = W( J )
  504. DO 40 JJ = J + 1, M
  505. IF( W( JJ ).LT.TMP1 ) THEN
  506. I = JJ
  507. TMP1 = W( JJ )
  508. END IF
  509. 40 CONTINUE
  510. *
  511. IF( I.NE.0 ) THEN
  512. ITMP1 = IWORK( 1 + I-1 )
  513. W( I ) = W( J )
  514. IWORK( 1 + I-1 ) = IWORK( 1 + J-1 )
  515. W( J ) = TMP1
  516. IWORK( 1 + J-1 ) = ITMP1
  517. CALL CSWAP( N, Z( 1, I ), 1, Z( 1, J ), 1 )
  518. IF( INFO.NE.0 ) THEN
  519. ITMP1 = IFAIL( I )
  520. IFAIL( I ) = IFAIL( J )
  521. IFAIL( J ) = ITMP1
  522. END IF
  523. END IF
  524. 50 CONTINUE
  525. END IF
  526. *
  527. RETURN
  528. *
  529. * End of CHBGVX
  530. *
  531. END