psi.f 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. *DECK PSI
  2. FUNCTION PSI (X)
  3. C***BEGIN PROLOGUE PSI
  4. C***PURPOSE Compute the Psi (or Digamma) function.
  5. C***LIBRARY SLATEC (FNLIB)
  6. C***CATEGORY C7C
  7. C***TYPE SINGLE PRECISION (PSI-S, DPSI-D, CPSI-C)
  8. C***KEYWORDS DIGAMMA FUNCTION, FNLIB, PSI FUNCTION, SPECIAL FUNCTIONS
  9. C***AUTHOR Fullerton, W., (LANL)
  10. C***DESCRIPTION
  11. C
  12. C PSI(X) calculates the psi (or digamma) function for real argument X.
  13. C PSI(X) is the logarithmic derivative of the gamma function of X.
  14. C
  15. C Series for PSI on the interval 0. to 1.00000D+00
  16. C with weighted error 2.03E-17
  17. C log weighted error 16.69
  18. C significant figures required 16.39
  19. C decimal places required 17.37
  20. C
  21. C Series for APSI on the interval 0. to 2.50000D-01
  22. C with weighted error 5.54E-17
  23. C log weighted error 16.26
  24. C significant figures required 14.42
  25. C decimal places required 16.86
  26. C
  27. C***REFERENCES (NONE)
  28. C***ROUTINES CALLED COT, CSEVL, INITS, R1MACH, XERMSG
  29. C***REVISION HISTORY (YYMMDD)
  30. C 770401 DATE WRITTEN
  31. C 890531 Changed all specific intrinsics to generic. (WRB)
  32. C 890531 REVISION DATE from Version 3.2
  33. C 891214 Prologue converted to Version 4.0 format. (BAB)
  34. C 900315 CALLs to XERROR changed to CALLs to XERMSG. (THJ)
  35. C 900727 Added EXTERNAL statement. (WRB)
  36. C 920618 Removed space from variable names. (RWC, WRB)
  37. C***END PROLOGUE PSI
  38. DIMENSION PSICS(23), APSICS(16)
  39. LOGICAL FIRST
  40. EXTERNAL COT
  41. SAVE PSICS, APSICS, PI, NTPSI, NTAPSI, XBIG, DXREL, FIRST
  42. DATA PSICS( 1) / -.0380570808 35217922E0 /
  43. DATA PSICS( 2) / .4914153930 2938713E0 /
  44. DATA PSICS( 3) / -.0568157478 21244730E0 /
  45. DATA PSICS( 4) / .0083578212 25914313E0 /
  46. DATA PSICS( 5) / -.0013332328 57994342E0 /
  47. DATA PSICS( 6) / .0002203132 87069308E0 /
  48. DATA PSICS( 7) / -.0000370402 38178456E0 /
  49. DATA PSICS( 8) / .0000062837 93654854E0 /
  50. DATA PSICS( 9) / -.0000010712 63908506E0 /
  51. DATA PSICS(10) / .0000001831 28394654E0 /
  52. DATA PSICS(11) / -.0000000313 53509361E0 /
  53. DATA PSICS(12) / .0000000053 72808776E0 /
  54. DATA PSICS(13) / -.0000000009 21168141E0 /
  55. DATA PSICS(14) / .0000000001 57981265E0 /
  56. DATA PSICS(15) / -.0000000000 27098646E0 /
  57. DATA PSICS(16) / .0000000000 04648722E0 /
  58. DATA PSICS(17) / -.0000000000 00797527E0 /
  59. DATA PSICS(18) / .0000000000 00136827E0 /
  60. DATA PSICS(19) / -.0000000000 00023475E0 /
  61. DATA PSICS(20) / .0000000000 00004027E0 /
  62. DATA PSICS(21) / -.0000000000 00000691E0 /
  63. DATA PSICS(22) / .0000000000 00000118E0 /
  64. DATA PSICS(23) / -.0000000000 00000020E0 /
  65. DATA APSICS( 1) / -.0204749044 678185E0 /
  66. DATA APSICS( 2) / -.0101801271 534859E0 /
  67. DATA APSICS( 3) / .0000559718 725387E0 /
  68. DATA APSICS( 4) / -.0000012917 176570E0 /
  69. DATA APSICS( 5) / .0000000572 858606E0 /
  70. DATA APSICS( 6) / -.0000000038 213539E0 /
  71. DATA APSICS( 7) / .0000000003 397434E0 /
  72. DATA APSICS( 8) / -.0000000000 374838E0 /
  73. DATA APSICS( 9) / .0000000000 048990E0 /
  74. DATA APSICS(10) / -.0000000000 007344E0 /
  75. DATA APSICS(11) / .0000000000 001233E0 /
  76. DATA APSICS(12) / -.0000000000 000228E0 /
  77. DATA APSICS(13) / .0000000000 000045E0 /
  78. DATA APSICS(14) / -.0000000000 000009E0 /
  79. DATA APSICS(15) / .0000000000 000002E0 /
  80. DATA APSICS(16) / -.0000000000 000000E0 /
  81. DATA PI / 3.1415926535 8979324E0/
  82. DATA FIRST /.TRUE./
  83. C***FIRST EXECUTABLE STATEMENT PSI
  84. IF (FIRST) THEN
  85. NTPSI = INITS (PSICS, 23, 0.1*R1MACH(3))
  86. NTAPSI = INITS (APSICS, 16, 0.1*R1MACH(3))
  87. C
  88. XBIG = 1.0/SQRT(R1MACH(3))
  89. DXREL = SQRT (R1MACH(4))
  90. ENDIF
  91. FIRST = .FALSE.
  92. C
  93. Y = ABS(X)
  94. IF (Y.GE.2.0) GO TO 30
  95. C
  96. C PSI(X) FOR -2. .LT. X .LT. 2.
  97. C
  98. N = X
  99. IF (X.LT.0.) N = N - 1
  100. Y = X - N
  101. N = N - 1
  102. PSI = CSEVL (2.*Y-1., PSICS, NTPSI)
  103. IF (N.EQ.0) RETURN
  104. C
  105. N = -N
  106. IF (X .EQ. 0.) CALL XERMSG ('SLATEC', 'PSI', 'X IS 0', 2, 2)
  107. IF (X .LT. 0. .AND. X+N-2 .EQ. 0.) CALL XERMSG ('SLATEC', 'PSI',
  108. + 'X IS A NEGATIVE INTEGER', 3, 2)
  109. IF (X .LT. (-0.5) .AND. ABS((X-AINT(X-0.5))/X) .LT. DXREL)
  110. + CALL XERMSG ('SLATEC', 'PSI',
  111. + 'ANSWER LT HALF PRECISION BECAUSE X TOO NEAR NEGATIVE INTEGER',
  112. + 1, 1)
  113. C
  114. DO 20 I=1,N
  115. PSI = PSI - 1.0/(X+I-1)
  116. 20 CONTINUE
  117. RETURN
  118. C
  119. C PSI(X) FOR ABS(X) .GE. 2.
  120. C
  121. 30 AUX = 0.
  122. IF (Y.LT.XBIG) AUX = CSEVL (8./Y**2-1., APSICS, NTAPSI)
  123. IF (X.LT.0.) PSI = LOG(ABS(X)) - 0.5/X + AUX - PI*COT(PI*X)
  124. IF (X.GT.0.) PSI = LOG(X) - 0.5/X + AUX
  125. RETURN
  126. C
  127. END