Add polygamma function (#60)

* Fix parameter bug in polylog * Reorder function definitions for consistency * Add polygamma * Adjust parameter naming convention * Export polygamma
JeffreySarnoff · Jul 23, 2023 · 3ae7f58 · 3ae7f58
1 parent 2cd87dd
commit 3ae7f58
Show file tree

Hide file tree

Showing 2 changed files with 60 additions and 36 deletions.
diff --git a/src/ArbNumerics.jl b/src/ArbNumerics.jl
@@ -67,7 +67,7 @@ export ArbNumber,
  elliptic_rf, elliptic_rg, elliptic_rj,
  weierstrass_p, weierstrass_p_prime, weierstrass_p_jet, weierstrass_invariants, weierstrass_roots, weierstrass_p_series, weierstrass_inv_p, weierstrass_zeta, weierstrass_sigma,
  zeta, eta, xi, # Reimann
- lambertw, polylog,
+ lambertw, polylog, polygamma,
  π, ℯ, γ, φ, catalan,
 
  # fft

diff --git a/src/float/otherspecial.jl b/src/float/otherspecial.jl
@@ -86,43 +86,8 @@ for (A,F) in ((:agm, :arb_agm), )
  end
 end
 
-function polylog(s::ArbComplex{P}, z::ArbComplex{P}, prec::Int=P) where {P}
- w = ArbComplex{P}()
- ccall(@libarb(acb_polylog), Cvoid, (Ref{ArbComplex}, Ref{ArbComplex}, Ref{ArbComplex}, Cint), w, s, z, P)
- return w
-end
-
-function polylog(s::ArbReal{P}, z::ArbReal{P}, prec::Int=P) where {P}
- sc = ArbComplex(s)
- zc = ArbComplex(z)
- wc = polylog(sc, zc, prec)
- return wc
-end
-
-function polylog(s::ArbFloat{P}, z::ArbFloat{P}, prec::Int=P) where {P}
- sc = ArbComplex(s)
- zc = ArbComplex(z)
- wc = polylog(sc, zc, prec)
- return wc
-end
-
-function polylog(s::Int, z::ArbComplex{P}, prec::Int=P) where {P}
- w = ArbComplex{P}()
- ccall(@libarb(acb_polylog_si), Cvoid, (Ref{ArbComplex}, Cint, Ref{ArbComplex}, Cint), w, s, z, P)
- return w
-end
 
-function polylog(s::Int, z::ArbReal{P}, prec::Int=P) where {P}
- zc = ArbComplex(z)
- wc = polylog(s, zc, prec)
- return wc
-end
 
-function polylog(s::Int, z::ArbFloat{P}, prec::Int=P) where {P}
- zc = ArbComplex(z)
- wc = polylog(s, zc, prec)
- return wc
-end
 
 #=
 for (A,F) in ((:ellipticp, :acb_elliptic_p), (:ellipticpi, :acb_elliptic_pi),
@@ -213,6 +178,65 @@ function agm(x::ArbFloat{P}, y::ArbFloat{P}) where {P}
 end
 
 
+function polylog(s::ArbComplex{P}, z::ArbComplex{P}, prec::Int=P) where {P}
+ w = ArbComplex{P}()
+ ccall(@libarb(acb_polylog), Cvoid, (Ref{ArbComplex}, Ref{ArbComplex}, Ref{ArbComplex}, Cint), w, s, z, prec)
+ return w
+end
+
+function polylog(s::ArbReal{P}, z::ArbReal{P}, prec::Int=P) where {P}
+ sc = ArbComplex(s)
+ zc = ArbComplex(z)
+ wc = polylog(sc, zc, prec)
+ return wc
+end
+
+function polylog(s::ArbFloat{P}, z::ArbFloat{P}, prec::Int=P) where {P}
+ sc = ArbComplex(s)
+ zc = ArbComplex(z)
+ wc = polylog(sc, zc, prec)
+ return wc
+end
+
+function polylog(s::Int, z::ArbComplex{P}, prec::Int=P) where {P}
+ w = ArbComplex{P}()
+ ccall(@libarb(acb_polylog_si), Cvoid, (Ref{ArbComplex}, Cint, Ref{ArbComplex}, Cint), w, s, z, prec)
+ return w
+end
+
+function polylog(s::Int, z::ArbReal{P}, prec::Int=P) where {P}
+ zc = ArbComplex(z)
+ wc = polylog(s, zc, prec)
+ return wc
+end
+
+function polylog(s::Int, z::ArbFloat{P}, prec::Int=P) where {P}
+ zc = ArbComplex(z)
+ wc = polylog(s, zc, prec)
+ return wc
+end
+
+function polygamma(s::ArbComplex{P}, z::ArbComplex{P}, prec::Int=P) where {P}
+ res = ArbComplex{P}()
+ ccall(@libarb(acb_polygamma), Cvoid, (Ref{ArbComplex}, Ref{ArbComplex}, Ref{ArbComplex}, Cint), res, s, z, prec)
+ return res
+end
+
+function polygamma(s::ArbReal{P}, z::ArbReal{P}, prec::Int=P) where {P}
+ sc = ArbComplex(s)
+ zc = ArbComplex(z)
+ res = polygamma(sc, zc, prec)
+ return res
+end
+
+function polygamma(s::ArbFloat{P}, z::ArbFloat{P}, prec::Int=P) where {P}
+ sc = ArbComplex(s)
+ zc = ArbComplex(z)
+ res = polygamma(sc, zc, prec)
+ return res
+end
+
+
 #=
 function erfcx(z::ArbComplex{P}) where {P}
  hiprec = round(Int, P*1.25)