cbrt.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. // Copyright 2009 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. package math
  5. // The go code is a modified version of the original C code from
  6. // http://www.netlib.org/fdlibm/s_cbrt.c and came with this notice.
  7. //
  8. // ====================================================
  9. // Copyright (C) 1993 by Sun Microsystems, Inc. All rights reserved.
  10. //
  11. // Developed at SunSoft, a Sun Microsystems, Inc. business.
  12. // Permission to use, copy, modify, and distribute this
  13. // software is freely granted, provided that this notice
  14. // is preserved.
  15. // ====================================================
  16. // Cbrt returns the cube root of x.
  17. //
  18. // Special cases are:
  19. // Cbrt(±0) = ±0
  20. // Cbrt(±Inf) = ±Inf
  21. // Cbrt(NaN) = NaN
  22. func Cbrt(x float64) float64 {
  23. return libc_cbrt(x)
  24. }
  25. //extern cbrt
  26. func libc_cbrt(float64) float64
  27. func cbrt(x float64) float64 {
  28. const (
  29. B1 = 715094163 // (682-0.03306235651)*2**20
  30. B2 = 696219795 // (664-0.03306235651)*2**20
  31. C = 5.42857142857142815906e-01 // 19/35 = 0x3FE15F15F15F15F1
  32. D = -7.05306122448979611050e-01 // -864/1225 = 0xBFE691DE2532C834
  33. E = 1.41428571428571436819e+00 // 99/70 = 0x3FF6A0EA0EA0EA0F
  34. F = 1.60714285714285720630e+00 // 45/28 = 0x3FF9B6DB6DB6DB6E
  35. G = 3.57142857142857150787e-01 // 5/14 = 0x3FD6DB6DB6DB6DB7
  36. SmallestNormal = 2.22507385850720138309e-308 // 2**-1022 = 0x0010000000000000
  37. )
  38. // special cases
  39. switch {
  40. case x == 0 || IsNaN(x) || IsInf(x, 0):
  41. return x
  42. }
  43. sign := false
  44. if x < 0 {
  45. x = -x
  46. sign = true
  47. }
  48. // rough cbrt to 5 bits
  49. t := Float64frombits(Float64bits(x)/3 + B1<<32)
  50. if x < SmallestNormal {
  51. // subnormal number
  52. t = float64(1 << 54) // set t= 2**54
  53. t *= x
  54. t = Float64frombits(Float64bits(t)/3 + B2<<32)
  55. }
  56. // new cbrt to 23 bits
  57. r := t * t / x
  58. s := C + r*t
  59. t *= G + F/(s+E+D/s)
  60. // chop to 22 bits, make larger than cbrt(x)
  61. t = Float64frombits(Float64bits(t)&(0xFFFFFFFFC<<28) + 1<<30)
  62. // one step newton iteration to 53 bits with error less than 0.667ulps
  63. s = t * t // t*t is exact
  64. r = x / s
  65. w := t + t
  66. r = (r - t) / (w + r) // r-s is exact
  67. t = t + t*r
  68. // restore the sign bit
  69. if sign {
  70. t = -t
  71. }
  72. return t
  73. }