a few additions

Author: fredrik-johansson

src/acb.pyx

@@ -188,6 +188,9 @@ cdef class acb(flint_scalar):
         other = any_as_acb(other)
         return bool(acb_overlaps((<acb>self).val, (<acb>other).val))
 
+    def contains_integer(self):
+        return bool(acb_contains_int(self.val))
+
     def mid(self):
         """
         Returns an exact *acb* representing the midpoint of *self*:
@@ -630,6 +633,16 @@ cdef class acb(flint_scalar):
             acb_hurwitz_zeta((<acb>u).val, (<acb>s).val, (<acb>a).val, getprec())
             return u
 
+    def dirichlet_l(s, chi):
+        cdef dirichlet_char cchar
+        if isinstance(chi, dirichlet_char):
+            cchar = chi
+        else:
+            cchar = dirichlet_char(chi[0], chi[1])
+        u = acb.__new__(acb)
+        acb_dirichlet_l((<acb>u).val, (<acb>s).val, cchar.G.val, cchar.val, getprec())
+        return u
+
     @staticmethod
     def pi():
         """

src/acb_series.pyx

@@ -447,6 +447,20 @@ cdef class acb_series(flint_series):
         (<acb_series>u).prec = cap
         return u
 
+    def dirichlet_l(s, chi, bint deflate=0):
+        cdef long cap
+        cdef dirichlet_char cchar
+        if isinstance(chi, dirichlet_char):
+            cchar = chi
+        else:
+            cchar = dirichlet_char(chi[0], chi[1])
+        cap = getcap()
+        cap = min(cap, (<acb_series>s).prec)
+        u = acb_series.__new__(acb_series)
+        acb_dirichlet_l_series((<acb_series>u).val, (<acb_series>s).val, cchar.G.val, cchar.val, deflate, cap, getprec())
+        (<acb_series>u).prec = cap
+        return u
+
     @classmethod
     def polylog(cls, s, z):
         cdef long cap
@@ -640,6 +654,22 @@ cdef class acb_series(flint_series):
         (<acb_series>z).prec = cap
         return u, v, w, z
 
+    def modular_theta(self, tau):
+        cdef long cap
+        tau = acb(tau)
+        cap = getcap()
+        cap = min(cap, (<acb_series>self).prec)
+        t1 = acb_series.__new__(acb_series)
+        t2 = acb_series.__new__(acb_series)
+        t3 = acb_series.__new__(acb_series)
+        t4 = acb_series.__new__(acb_series)
+        acb_modular_theta_series((<acb_series>t1).val, (<acb_series>t2).val, (<acb_series>t3).val, (<acb_series>t4).val, (<acb_series>self).val, (<acb>tau).val, cap, getprec())
+        (<acb_series>t1).prec = cap
+        (<acb_series>t2).prec = cap
+        (<acb_series>t3).prec = cap
+        (<acb_series>t4).prec = cap
+        return t1, t2, t3, t4
+
     def coulomb(self, l, eta):
         cdef long cap
         l = acb(l)

src/arb.pyx

@@ -476,6 +476,9 @@ cdef class arb(flint_scalar):
         other = any_as_arb(other)
         return bool(arb_overlaps((<arb>self).val, (<arb>other).val))
 
+    def contains_integer(self):
+        return bool(arb_contains_int(self.val))
+
     def __pos__(self):
         res = arb.__new__(arb)
         arb_set_round((<arb>res).val, (<arb>self).val, getprec())

src/flint.pxd

@@ -903,6 +903,7 @@ cdef extern from "arb.h":
     int arb_contains_nonpositive(const arb_t x)
     int arb_contains_positive(const arb_t x)
     int arb_contains_nonnegative(const arb_t x)
+    int arb_contains_int(const arb_t x)
     void arb_get_mag_lower(mag_t z, const arb_t x)
     void arb_get_mag_lower_nonnegative(mag_t z, const arb_t x)
     void arb_get_mag(mag_t z, const arb_t x)
@@ -1168,6 +1169,7 @@ cdef extern from "acb.h":
     int acb_contains_fmpz(const acb_t x, const fmpz_t y)
     int acb_contains(const acb_t x, const acb_t y)
     int acb_get_unique_fmpz(fmpz_t z, const acb_t x)
+    int acb_contains_int(const acb_t x)
     void acb_set_ui(acb_t z, ulong c)
     void acb_set_si(acb_t z, long c)
     void acb_set_fmpz(acb_t z, const fmpz_t c)
@@ -1910,6 +1912,7 @@ cdef extern from "acb_mat.h":
 cdef extern from "acb_modular.h":
     void acb_modular_theta(acb_t theta1, acb_t theta2, acb_t theta3, acb_t theta4, const acb_t z, const acb_t tau, long prec)
     void acb_modular_theta_jet(acb_ptr theta1, acb_ptr theta2, acb_ptr theta3, acb_ptr theta4, const acb_t z, const acb_t tau, long len, long prec)
+    void acb_modular_theta_series(acb_poly_t theta1, acb_poly_t theta2, acb_poly_t theta3, acb_poly_t theta4, const acb_poly_t z, const acb_t tau, long len, long prec)
     void acb_modular_eta(acb_t r, const acb_t tau, long prec)
     void acb_modular_j(acb_t r, const acb_t tau, long prec)
     void acb_modular_lambda(acb_t r, const acb_t tau, long prec)
@@ -2130,6 +2133,7 @@ cdef extern from "acb_dirichlet.h":
 
     void acb_dirichlet_l(acb_t res, const acb_t s, const dirichlet_group_t G, const dirichlet_char_t chi, long prec)
     void acb_dirichlet_hardy_z(acb_ptr res, const acb_t t, const dirichlet_group_t G, const dirichlet_char_t chi, long len, long prec)
+    void acb_dirichlet_l_series(acb_poly_t res, const acb_poly_t s, const dirichlet_group_t G, const dirichlet_char_t chi, int deflate, long len, long prec)
 
     void acb_dirichlet_stieltjes(acb_t res, const fmpz_t n, const acb_t a, long prec);
 
@@ -2138,6 +2142,7 @@ cdef extern from "acb_dirichlet.h":
     void acb_dirichlet_zeta_nzeros(arb_t res, const arb_t t, long prec)
     void acb_dirichlet_backlund_s(arb_t res, const arb_t t, long prec)
     void acb_dirichlet_zeta_zero(acb_t res, const fmpz_t n, long prec)
+    void acb_dirichlet_zeta_zeros(acb_ptr res, const fmpz_t n, long len, long prec)
 
 cdef extern from "acb_elliptic.h":
     void acb_elliptic_rf(acb_t res, const acb_t x, const acb_t y, const acb_t z, int flags, long prec)