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.

zlatm2.f 9.6 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325
  1. *> \brief \b ZLATM2
  2. *
  3. * =========== DOCUMENTATION ===========
  4. *
  5. * Online html documentation available at
  6. * http://www.netlib.org/lapack/explore-html/
  7. *
  8. * Definition:
  9. * ===========
  10. *
  11. * COMPLEX*16 FUNCTION ZLATM2( M, N, I, J, KL, KU, IDIST,
  12. * ISEED, D, IGRADE, DL, DR, IPVTNG, IWORK, SPARSE )
  13. *
  14. * .. Scalar Arguments ..
  15. *
  16. * INTEGER I, IDIST, IGRADE, IPVTNG, J, KL, KU, M, N
  17. * DOUBLE PRECISION SPARSE
  18. * ..
  19. *
  20. * .. Array Arguments ..
  21. *
  22. * INTEGER ISEED( 4 ), IWORK( * )
  23. * COMPLEX*16 D( * ), DL( * ), DR( * )
  24. * ..
  25. *
  26. *
  27. *> \par Purpose:
  28. * =============
  29. *>
  30. *> \verbatim
  31. *>
  32. *> ZLATM2 returns the (I,J) entry of a random matrix of dimension
  33. *> (M, N) described by the other parameters. It is called by the
  34. *> ZLATMR routine in order to build random test matrices. No error
  35. *> checking on parameters is done, because this routine is called in
  36. *> a tight loop by ZLATMR which has already checked the parameters.
  37. *>
  38. *> Use of ZLATM2 differs from CLATM3 in the order in which the random
  39. *> number generator is called to fill in random matrix entries.
  40. *> With ZLATM2, the generator is called to fill in the pivoted matrix
  41. *> columnwise. With ZLATM3, the generator is called to fill in the
  42. *> matrix columnwise, after which it is pivoted. Thus, ZLATM3 can
  43. *> be used to construct random matrices which differ only in their
  44. *> order of rows and/or columns. ZLATM2 is used to construct band
  45. *> matrices while avoiding calling the random number generator for
  46. *> entries outside the band (and therefore generating random numbers
  47. *>
  48. *> The matrix whose (I,J) entry is returned is constructed as
  49. *> follows (this routine only computes one entry):
  50. *>
  51. *> If I is outside (1..M) or J is outside (1..N), return zero
  52. *> (this is convenient for generating matrices in band format).
  53. *>
  54. *> Generate a matrix A with random entries of distribution IDIST.
  55. *>
  56. *> Set the diagonal to D.
  57. *>
  58. *> Grade the matrix, if desired, from the left (by DL) and/or
  59. *> from the right (by DR or DL) as specified by IGRADE.
  60. *>
  61. *> Permute, if desired, the rows and/or columns as specified by
  62. *> IPVTNG and IWORK.
  63. *>
  64. *> Band the matrix to have lower bandwidth KL and upper
  65. *> bandwidth KU.
  66. *>
  67. *> Set random entries to zero as specified by SPARSE.
  68. *> \endverbatim
  69. *
  70. * Arguments:
  71. * ==========
  72. *
  73. *> \param[in] M
  74. *> \verbatim
  75. *> M is INTEGER
  76. *> Number of rows of matrix. Not modified.
  77. *> \endverbatim
  78. *>
  79. *> \param[in] N
  80. *> \verbatim
  81. *> N is INTEGER
  82. *> Number of columns of matrix. Not modified.
  83. *> \endverbatim
  84. *>
  85. *> \param[in] I
  86. *> \verbatim
  87. *> I is INTEGER
  88. *> Row of entry to be returned. Not modified.
  89. *> \endverbatim
  90. *>
  91. *> \param[in] J
  92. *> \verbatim
  93. *> J is INTEGER
  94. *> Column of entry to be returned. Not modified.
  95. *> \endverbatim
  96. *>
  97. *> \param[in] KL
  98. *> \verbatim
  99. *> KL is INTEGER
  100. *> Lower bandwidth. Not modified.
  101. *> \endverbatim
  102. *>
  103. *> \param[in] KU
  104. *> \verbatim
  105. *> KU is INTEGER
  106. *> Upper bandwidth. Not modified.
  107. *> \endverbatim
  108. *>
  109. *> \param[in] IDIST
  110. *> \verbatim
  111. *> IDIST is INTEGER
  112. *> On entry, IDIST specifies the type of distribution to be
  113. *> used to generate a random matrix .
  114. *> 1 => real and imaginary parts each UNIFORM( 0, 1 )
  115. *> 2 => real and imaginary parts each UNIFORM( -1, 1 )
  116. *> 3 => real and imaginary parts each NORMAL( 0, 1 )
  117. *> 4 => complex number uniform in DISK( 0 , 1 )
  118. *> Not modified.
  119. *> \endverbatim
  120. *>
  121. *> \param[in,out] ISEED
  122. *> \verbatim
  123. *> ISEED is INTEGER array of dimension ( 4 )
  124. *> Seed for random number generator.
  125. *> Changed on exit.
  126. *> \endverbatim
  127. *>
  128. *> \param[in] D
  129. *> \verbatim
  130. *> D is COMPLEX*16 array of dimension ( MIN( I , J ) )
  131. *> Diagonal entries of matrix. Not modified.
  132. *> \endverbatim
  133. *>
  134. *> \param[in] IGRADE
  135. *> \verbatim
  136. *> IGRADE is INTEGER
  137. *> Specifies grading of matrix as follows:
  138. *> 0 => no grading
  139. *> 1 => matrix premultiplied by diag( DL )
  140. *> 2 => matrix postmultiplied by diag( DR )
  141. *> 3 => matrix premultiplied by diag( DL ) and
  142. *> postmultiplied by diag( DR )
  143. *> 4 => matrix premultiplied by diag( DL ) and
  144. *> postmultiplied by inv( diag( DL ) )
  145. *> 5 => matrix premultiplied by diag( DL ) and
  146. *> postmultiplied by diag( CONJG(DL) )
  147. *> 6 => matrix premultiplied by diag( DL ) and
  148. *> postmultiplied by diag( DL )
  149. *> Not modified.
  150. *> \endverbatim
  151. *>
  152. *> \param[in] DL
  153. *> \verbatim
  154. *> DL is COMPLEX*16 array ( I or J, as appropriate )
  155. *> Left scale factors for grading matrix. Not modified.
  156. *> \endverbatim
  157. *>
  158. *> \param[in] DR
  159. *> \verbatim
  160. *> DR is COMPLEX*16 array ( I or J, as appropriate )
  161. *> Right scale factors for grading matrix. Not modified.
  162. *> \endverbatim
  163. *>
  164. *> \param[in] IPVTNG
  165. *> \verbatim
  166. *> IPVTNG is INTEGER
  167. *> On entry specifies pivoting permutations as follows:
  168. *> 0 => none.
  169. *> 1 => row pivoting.
  170. *> 2 => column pivoting.
  171. *> 3 => full pivoting, i.e., on both sides.
  172. *> Not modified.
  173. *> \endverbatim
  174. *>
  175. *> \param[out] IWORK
  176. *> \verbatim
  177. *> IWORK is INTEGER array ( I or J, as appropriate )
  178. *> This array specifies the permutation used. The
  179. *> row (or column) in position K was originally in
  180. *> position IWORK( K ).
  181. *> This differs from IWORK for ZLATM3. Not modified.
  182. *> \endverbatim
  183. *>
  184. *> \param[in] SPARSE
  185. *> \verbatim
  186. *> SPARSE is DOUBLE PRECISION between 0. and 1.
  187. *> On entry specifies the sparsity of the matrix
  188. *> if sparse matrix is to be generated.
  189. *> SPARSE should lie between 0 and 1.
  190. *> A uniform ( 0, 1 ) random number x is generated and
  191. *> compared to SPARSE; if x is larger the matrix entry
  192. *> is unchanged and if x is smaller the entry is set
  193. *> to zero. Thus on the average a fraction SPARSE of the
  194. *> entries will be set to zero.
  195. *> Not modified.
  196. *> \endverbatim
  197. *
  198. * Authors:
  199. * ========
  200. *
  201. *> \author Univ. of Tennessee
  202. *> \author Univ. of California Berkeley
  203. *> \author Univ. of Colorado Denver
  204. *> \author NAG Ltd.
  205. *
  206. *> \ingroup complex16_matgen
  207. *
  208. * =====================================================================
  209. COMPLEX*16 FUNCTION ZLATM2( M, N, I, J, KL, KU, IDIST,
  210. $ ISEED, D, IGRADE, DL, DR, IPVTNG, IWORK, SPARSE )
  211. *
  212. * -- LAPACK auxiliary routine --
  213. * -- LAPACK is a software package provided by Univ. of Tennessee, --
  214. * -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
  215. *
  216. * .. Scalar Arguments ..
  217. *
  218. INTEGER I, IDIST, IGRADE, IPVTNG, J, KL, KU, M, N
  219. DOUBLE PRECISION SPARSE
  220. * ..
  221. *
  222. * .. Array Arguments ..
  223. *
  224. INTEGER ISEED( 4 ), IWORK( * )
  225. COMPLEX*16 D( * ), DL( * ), DR( * )
  226. * ..
  227. *
  228. * =====================================================================
  229. *
  230. * .. Parameters ..
  231. *
  232. COMPLEX*16 CZERO
  233. PARAMETER ( CZERO = ( 0.0D0, 0.0D0 ) )
  234. DOUBLE PRECISION ZERO
  235. PARAMETER ( ZERO = 0.0D0 )
  236. * ..
  237. *
  238. * .. Local Scalars ..
  239. *
  240. INTEGER ISUB, JSUB
  241. COMPLEX*16 CTEMP
  242. * ..
  243. *
  244. * .. External Functions ..
  245. *
  246. DOUBLE PRECISION DLARAN
  247. COMPLEX*16 ZLARND
  248. EXTERNAL DLARAN, ZLARND
  249. * ..
  250. *
  251. * .. Intrinsic Functions ..
  252. *
  253. INTRINSIC DCONJG
  254. * ..
  255. *
  256. *-----------------------------------------------------------------------
  257. *
  258. * .. Executable Statements ..
  259. *
  260. *
  261. * Check for I and J in range
  262. *
  263. IF( I.LT.1 .OR. I.GT.M .OR. J.LT.1 .OR. J.GT.N ) THEN
  264. ZLATM2 = CZERO
  265. RETURN
  266. END IF
  267. *
  268. * Check for banding
  269. *
  270. IF( J.GT.I+KU .OR. J.LT.I-KL ) THEN
  271. ZLATM2 = CZERO
  272. RETURN
  273. END IF
  274. *
  275. * Check for sparsity
  276. *
  277. IF( SPARSE.GT.ZERO ) THEN
  278. IF( DLARAN( ISEED ).LT.SPARSE ) THEN
  279. ZLATM2 = CZERO
  280. RETURN
  281. END IF
  282. END IF
  283. *
  284. * Compute subscripts depending on IPVTNG
  285. *
  286. IF( IPVTNG.EQ.0 ) THEN
  287. ISUB = I
  288. JSUB = J
  289. ELSE IF( IPVTNG.EQ.1 ) THEN
  290. ISUB = IWORK( I )
  291. JSUB = J
  292. ELSE IF( IPVTNG.EQ.2 ) THEN
  293. ISUB = I
  294. JSUB = IWORK( J )
  295. ELSE IF( IPVTNG.EQ.3 ) THEN
  296. ISUB = IWORK( I )
  297. JSUB = IWORK( J )
  298. END IF
  299. *
  300. * Compute entry and grade it according to IGRADE
  301. *
  302. IF( ISUB.EQ.JSUB ) THEN
  303. CTEMP = D( ISUB )
  304. ELSE
  305. CTEMP = ZLARND( IDIST, ISEED )
  306. END IF
  307. IF( IGRADE.EQ.1 ) THEN
  308. CTEMP = CTEMP*DL( ISUB )
  309. ELSE IF( IGRADE.EQ.2 ) THEN
  310. CTEMP = CTEMP*DR( JSUB )
  311. ELSE IF( IGRADE.EQ.3 ) THEN
  312. CTEMP = CTEMP*DL( ISUB )*DR( JSUB )
  313. ELSE IF( IGRADE.EQ.4 .AND. ISUB.NE.JSUB ) THEN
  314. CTEMP = CTEMP*DL( ISUB ) / DL( JSUB )
  315. ELSE IF( IGRADE.EQ.5 ) THEN
  316. CTEMP = CTEMP*DL( ISUB )*DCONJG( DL( JSUB ) )
  317. ELSE IF( IGRADE.EQ.6 ) THEN
  318. CTEMP = CTEMP*DL( ISUB )*DL( JSUB )
  319. END IF
  320. ZLATM2 = CTEMP
  321. RETURN
  322. *
  323. * End of ZLATM2
  324. *
  325. END