zsqrt.f 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. *DECK ZSQRT
  2. SUBROUTINE ZSQRT (AR, AI, BR, BI)
  3. C***BEGIN PROLOGUE ZSQRT
  4. C***SUBSIDIARY
  5. C***PURPOSE Subsidiary to ZBESH, ZBESI, ZBESJ, ZBESK, ZBESY, ZAIRY and
  6. C ZBIRY
  7. C***LIBRARY SLATEC
  8. C***TYPE ALL (ZSQRT-A)
  9. C***AUTHOR Amos, D. E., (SNL)
  10. C***DESCRIPTION
  11. C
  12. C DOUBLE PRECISION COMPLEX SQUARE ROOT, B=CSQRT(A)
  13. C
  14. C***SEE ALSO ZAIRY, ZBESH, ZBESI, ZBESJ, ZBESK, ZBESY, ZBIRY
  15. C***ROUTINES CALLED ZABS
  16. C***REVISION HISTORY (YYMMDD)
  17. C 830501 DATE WRITTEN
  18. C 910415 Prologue converted to Version 4.0 format. (BAB)
  19. C***END PROLOGUE ZSQRT
  20. DOUBLE PRECISION AR, AI, BR, BI, ZM, DTHETA, DPI, DRT
  21. DOUBLE PRECISION ZABS
  22. EXTERNAL ZABS
  23. DATA DRT , DPI / 7.071067811865475244008443621D-1,
  24. 1 3.141592653589793238462643383D+0/
  25. C***FIRST EXECUTABLE STATEMENT ZSQRT
  26. ZM = ZABS(AR,AI)
  27. ZM = SQRT(ZM)
  28. IF (AR.EQ.0.0D+0) GO TO 10
  29. IF (AI.EQ.0.0D+0) GO TO 20
  30. DTHETA = DATAN(AI/AR)
  31. IF (DTHETA.LE.0.0D+0) GO TO 40
  32. IF (AR.LT.0.0D+0) DTHETA = DTHETA - DPI
  33. GO TO 50
  34. 10 IF (AI.GT.0.0D+0) GO TO 60
  35. IF (AI.LT.0.0D+0) GO TO 70
  36. BR = 0.0D+0
  37. BI = 0.0D+0
  38. RETURN
  39. 20 IF (AR.GT.0.0D+0) GO TO 30
  40. BR = 0.0D+0
  41. BI = SQRT(ABS(AR))
  42. RETURN
  43. 30 BR = SQRT(AR)
  44. BI = 0.0D+0
  45. RETURN
  46. 40 IF (AR.LT.0.0D+0) DTHETA = DTHETA + DPI
  47. 50 DTHETA = DTHETA*0.5D+0
  48. BR = ZM*COS(DTHETA)
  49. BI = ZM*SIN(DTHETA)
  50. RETURN
  51. 60 BR = ZM*DRT
  52. BI = ZM*DRT
  53. RETURN
  54. 70 BR = ZM*DRT
  55. BI = -ZM*DRT
  56. RETURN
  57. END