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.

slasd0.f 9.1 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315
  1. *> \brief \b SLASD0 computes the singular values of a real upper bidiagonal n-by-m matrix B with diagonal d and off-diagonal e. Used by sbdsdc.
  2. *
  3. * =========== DOCUMENTATION ===========
  4. *
  5. * Online html documentation available at
  6. * http://www.netlib.org/lapack/explore-html/
  7. *
  8. *> \htmlonly
  9. *> Download SLASD0 + dependencies
  10. *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.tgz?format=tgz&filename=/lapack/lapack_routine/slasd0.f">
  11. *> [TGZ]</a>
  12. *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.zip?format=zip&filename=/lapack/lapack_routine/slasd0.f">
  13. *> [ZIP]</a>
  14. *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.txt?format=txt&filename=/lapack/lapack_routine/slasd0.f">
  15. *> [TXT]</a>
  16. *> \endhtmlonly
  17. *
  18. * Definition:
  19. * ===========
  20. *
  21. * SUBROUTINE SLASD0( N, SQRE, D, E, U, LDU, VT, LDVT, SMLSIZ, IWORK,
  22. * WORK, INFO )
  23. *
  24. * .. Scalar Arguments ..
  25. * INTEGER INFO, LDU, LDVT, N, SMLSIZ, SQRE
  26. * ..
  27. * .. Array Arguments ..
  28. * INTEGER IWORK( * )
  29. * REAL D( * ), E( * ), U( LDU, * ), VT( LDVT, * ),
  30. * $ WORK( * )
  31. * ..
  32. *
  33. *
  34. *> \par Purpose:
  35. * =============
  36. *>
  37. *> \verbatim
  38. *>
  39. *> Using a divide and conquer approach, SLASD0 computes the singular
  40. *> value decomposition (SVD) of a real upper bidiagonal N-by-M
  41. *> matrix B with diagonal D and offdiagonal E, where M = N + SQRE.
  42. *> The algorithm computes orthogonal matrices U and VT such that
  43. *> B = U * S * VT. The singular values S are overwritten on D.
  44. *>
  45. *> A related subroutine, SLASDA, computes only the singular values,
  46. *> and optionally, the singular vectors in compact form.
  47. *> \endverbatim
  48. *
  49. * Arguments:
  50. * ==========
  51. *
  52. *> \param[in] N
  53. *> \verbatim
  54. *> N is INTEGER
  55. *> On entry, the row dimension of the upper bidiagonal matrix.
  56. *> This is also the dimension of the main diagonal array D.
  57. *> \endverbatim
  58. *>
  59. *> \param[in] SQRE
  60. *> \verbatim
  61. *> SQRE is INTEGER
  62. *> Specifies the column dimension of the bidiagonal matrix.
  63. *> = 0: The bidiagonal matrix has column dimension M = N;
  64. *> = 1: The bidiagonal matrix has column dimension M = N+1;
  65. *> \endverbatim
  66. *>
  67. *> \param[in,out] D
  68. *> \verbatim
  69. *> D is REAL array, dimension (N)
  70. *> On entry D contains the main diagonal of the bidiagonal
  71. *> matrix.
  72. *> On exit D, if INFO = 0, contains its singular values.
  73. *> \endverbatim
  74. *>
  75. *> \param[in,out] E
  76. *> \verbatim
  77. *> E is REAL array, dimension (M-1)
  78. *> Contains the subdiagonal entries of the bidiagonal matrix.
  79. *> On exit, E has been destroyed.
  80. *> \endverbatim
  81. *>
  82. *> \param[in,out] U
  83. *> \verbatim
  84. *> U is REAL array, dimension (LDU, N)
  85. *> On exit, U contains the left singular vectors,
  86. *> if U passed in as (N, N) Identity.
  87. *> \endverbatim
  88. *>
  89. *> \param[in] LDU
  90. *> \verbatim
  91. *> LDU is INTEGER
  92. *> On entry, leading dimension of U.
  93. *> \endverbatim
  94. *>
  95. *> \param[in,out] VT
  96. *> \verbatim
  97. *> VT is REAL array, dimension (LDVT, M)
  98. *> On exit, VT**T contains the right singular vectors,
  99. *> if VT passed in as (M, M) Identity.
  100. *> \endverbatim
  101. *>
  102. *> \param[in] LDVT
  103. *> \verbatim
  104. *> LDVT is INTEGER
  105. *> On entry, leading dimension of VT.
  106. *> \endverbatim
  107. *>
  108. *> \param[in] SMLSIZ
  109. *> \verbatim
  110. *> SMLSIZ is INTEGER
  111. *> On entry, maximum size of the subproblems at the
  112. *> bottom of the computation tree.
  113. *> \endverbatim
  114. *>
  115. *> \param[out] IWORK
  116. *> \verbatim
  117. *> IWORK is INTEGER array, dimension (8*N)
  118. *> \endverbatim
  119. *>
  120. *> \param[out] WORK
  121. *> \verbatim
  122. *> WORK is REAL array, dimension (3*M**2+2*M)
  123. *> \endverbatim
  124. *>
  125. *> \param[out] INFO
  126. *> \verbatim
  127. *> INFO is INTEGER
  128. *> = 0: successful exit.
  129. *> < 0: if INFO = -i, the i-th argument had an illegal value.
  130. *> > 0: if INFO = 1, a singular value did not converge
  131. *> \endverbatim
  132. *
  133. * Authors:
  134. * ========
  135. *
  136. *> \author Univ. of Tennessee
  137. *> \author Univ. of California Berkeley
  138. *> \author Univ. of Colorado Denver
  139. *> \author NAG Ltd.
  140. *
  141. *> \ingroup OTHERauxiliary
  142. *
  143. *> \par Contributors:
  144. * ==================
  145. *>
  146. *> Ming Gu and Huan Ren, Computer Science Division, University of
  147. *> California at Berkeley, USA
  148. *>
  149. * =====================================================================
  150. SUBROUTINE SLASD0( N, SQRE, D, E, U, LDU, VT, LDVT, SMLSIZ, IWORK,
  151. $ WORK, INFO )
  152. *
  153. * -- LAPACK auxiliary routine --
  154. * -- LAPACK is a software package provided by Univ. of Tennessee, --
  155. * -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
  156. *
  157. * .. Scalar Arguments ..
  158. INTEGER INFO, LDU, LDVT, N, SMLSIZ, SQRE
  159. * ..
  160. * .. Array Arguments ..
  161. INTEGER IWORK( * )
  162. REAL D( * ), E( * ), U( LDU, * ), VT( LDVT, * ),
  163. $ WORK( * )
  164. * ..
  165. *
  166. * =====================================================================
  167. *
  168. * .. Local Scalars ..
  169. INTEGER I, I1, IC, IDXQ, IDXQC, IM1, INODE, ITEMP, IWK,
  170. $ J, LF, LL, LVL, M, NCC, ND, NDB1, NDIML, NDIMR,
  171. $ NL, NLF, NLP1, NLVL, NR, NRF, NRP1, SQREI
  172. REAL ALPHA, BETA
  173. * ..
  174. * .. External Subroutines ..
  175. EXTERNAL SLASD1, SLASDQ, SLASDT, XERBLA
  176. * ..
  177. * .. Executable Statements ..
  178. *
  179. * Test the input parameters.
  180. *
  181. INFO = 0
  182. *
  183. IF( N.LT.0 ) THEN
  184. INFO = -1
  185. ELSE IF( ( SQRE.LT.0 ) .OR. ( SQRE.GT.1 ) ) THEN
  186. INFO = -2
  187. END IF
  188. *
  189. M = N + SQRE
  190. *
  191. IF( LDU.LT.N ) THEN
  192. INFO = -6
  193. ELSE IF( LDVT.LT.M ) THEN
  194. INFO = -8
  195. ELSE IF( SMLSIZ.LT.3 ) THEN
  196. INFO = -9
  197. END IF
  198. IF( INFO.NE.0 ) THEN
  199. CALL XERBLA( 'SLASD0', -INFO )
  200. RETURN
  201. END IF
  202. *
  203. * If the input matrix is too small, call SLASDQ to find the SVD.
  204. *
  205. IF( N.LE.SMLSIZ ) THEN
  206. CALL SLASDQ( 'U', SQRE, N, M, N, 0, D, E, VT, LDVT, U, LDU, U,
  207. $ LDU, WORK, INFO )
  208. RETURN
  209. END IF
  210. *
  211. * Set up the computation tree.
  212. *
  213. INODE = 1
  214. NDIML = INODE + N
  215. NDIMR = NDIML + N
  216. IDXQ = NDIMR + N
  217. IWK = IDXQ + N
  218. CALL SLASDT( N, NLVL, ND, IWORK( INODE ), IWORK( NDIML ),
  219. $ IWORK( NDIMR ), SMLSIZ )
  220. *
  221. * For the nodes on bottom level of the tree, solve
  222. * their subproblems by SLASDQ.
  223. *
  224. NDB1 = ( ND+1 ) / 2
  225. NCC = 0
  226. DO 30 I = NDB1, ND
  227. *
  228. * IC : center row of each node
  229. * NL : number of rows of left subproblem
  230. * NR : number of rows of right subproblem
  231. * NLF: starting row of the left subproblem
  232. * NRF: starting row of the right subproblem
  233. *
  234. I1 = I - 1
  235. IC = IWORK( INODE+I1 )
  236. NL = IWORK( NDIML+I1 )
  237. NLP1 = NL + 1
  238. NR = IWORK( NDIMR+I1 )
  239. NRP1 = NR + 1
  240. NLF = IC - NL
  241. NRF = IC + 1
  242. SQREI = 1
  243. CALL SLASDQ( 'U', SQREI, NL, NLP1, NL, NCC, D( NLF ), E( NLF ),
  244. $ VT( NLF, NLF ), LDVT, U( NLF, NLF ), LDU,
  245. $ U( NLF, NLF ), LDU, WORK, INFO )
  246. IF( INFO.NE.0 ) THEN
  247. RETURN
  248. END IF
  249. ITEMP = IDXQ + NLF - 2
  250. DO 10 J = 1, NL
  251. IWORK( ITEMP+J ) = J
  252. 10 CONTINUE
  253. IF( I.EQ.ND ) THEN
  254. SQREI = SQRE
  255. ELSE
  256. SQREI = 1
  257. END IF
  258. NRP1 = NR + SQREI
  259. CALL SLASDQ( 'U', SQREI, NR, NRP1, NR, NCC, D( NRF ), E( NRF ),
  260. $ VT( NRF, NRF ), LDVT, U( NRF, NRF ), LDU,
  261. $ U( NRF, NRF ), LDU, WORK, INFO )
  262. IF( INFO.NE.0 ) THEN
  263. RETURN
  264. END IF
  265. ITEMP = IDXQ + IC
  266. DO 20 J = 1, NR
  267. IWORK( ITEMP+J-1 ) = J
  268. 20 CONTINUE
  269. 30 CONTINUE
  270. *
  271. * Now conquer each subproblem bottom-up.
  272. *
  273. DO 50 LVL = NLVL, 1, -1
  274. *
  275. * Find the first node LF and last node LL on the
  276. * current level LVL.
  277. *
  278. IF( LVL.EQ.1 ) THEN
  279. LF = 1
  280. LL = 1
  281. ELSE
  282. LF = 2**( LVL-1 )
  283. LL = 2*LF - 1
  284. END IF
  285. DO 40 I = LF, LL
  286. IM1 = I - 1
  287. IC = IWORK( INODE+IM1 )
  288. NL = IWORK( NDIML+IM1 )
  289. NR = IWORK( NDIMR+IM1 )
  290. NLF = IC - NL
  291. IF( ( SQRE.EQ.0 ) .AND. ( I.EQ.LL ) ) THEN
  292. SQREI = SQRE
  293. ELSE
  294. SQREI = 1
  295. END IF
  296. IDXQC = IDXQ + NLF - 1
  297. ALPHA = D( IC )
  298. BETA = E( IC )
  299. CALL SLASD1( NL, NR, SQREI, D( NLF ), ALPHA, BETA,
  300. $ U( NLF, NLF ), LDU, VT( NLF, NLF ), LDVT,
  301. $ IWORK( IDXQC ), IWORK( IWK ), WORK, INFO )
  302. *
  303. * Report the possible convergence failure.
  304. *
  305. IF( INFO.NE.0 ) THEN
  306. RETURN
  307. END IF
  308. 40 CONTINUE
  309. 50 CONTINUE
  310. *
  311. RETURN
  312. *
  313. * End of SLASD0
  314. *
  315. END