bindings: auto generate acb_* .pxd files

Author: oscarbenjamin

bin/all_rst_to_pxd.sh

@@ -15,6 +15,9 @@ modules="\
     arb\
     arb_poly\
     arb_mat\
+    acb\
+    acb_poly\
+    acb_mat\
     "
 
 for module in $modules; do

bin/rst_to_pxd.py

@@ -44,6 +44,7 @@
     (re.compile(r"\bfmpz\b"),"fmpz_struct"),
     (re.compile(r"\bfmpq\b"), "fmpq_struct"),
     (re.compile(r"\bin\b"), "in_"),
+    (re.compile(r"\blambda\b"), "lambda_"),
 ]
 # comment out functions which use these types
 comment_types = re.compile(r"(\bFILE\b)|(\bmpz_t\b)|(\bmpq_t\b)")

src/flint/flintlib/acb.pxd

@@ -1,21 +1,16 @@
-from flint.flintlib.flint cimport ulong, slong, flint_rand_t
+from flint.flintlib.acb_types cimport acb_ptr, acb_srcptr, acb_t
+from flint.flintlib.arb_types cimport arb_ptr, arb_srcptr, arb_t, mag_srcptr, mag_t
+from flint.flintlib.arf_types cimport arf_srcptr, arf_t
+from flint.flintlib.flint cimport flint_rand_t, fmpz_struct, slong, ulong
 from flint.flintlib.fmpq cimport fmpq_t
-from flint.flintlib.fmpz cimport fmpz_t, fmpz_struct
-from flint.flintlib.arf_types cimport arf_t, arf_srcptr
-from flint.flintlib.arb_types cimport mag_t, mag_srcptr, arb_struct, arb_t, arb_ptr
+from flint.flintlib.fmpz_types cimport fmpz_t
 
-cdef extern from "flint/acb.h":
-    ctypedef struct acb_struct:
-        arb_struct real
-        arb_struct imag
-
-    ctypedef acb_struct * acb_ptr
-    ctypedef const acb_struct * acb_srcptr
-    ctypedef acb_struct acb_t[1]
+# unknown type FILE
 
-    arb_ptr acb_realref(const acb_t x)
-    arb_ptr acb_imagref(const acb_t x)
+# .. macro:: acb_realref(x)
+# .. macro:: acb_imagref(x)
 
+cdef extern from "flint/acb.h":
     void acb_init(acb_t x)
     void acb_clear(acb_t x)
     acb_ptr _acb_vec_init(slong n)
@@ -46,12 +41,16 @@ cdef extern from "flint/acb.h":
     void acb_add_error_arb(acb_t x, const arb_t err)
     void acb_get_mid(acb_t m, const acb_t x)
     void acb_print(const acb_t x)
+    # void acb_fprint(FILE * file, const acb_t x)
     void acb_printd(const acb_t x, slong digits)
+    # void acb_fprintd(FILE * file, const acb_t x, slong digits)
     void acb_printn(const acb_t x, slong digits, ulong flags)
+    # void acb_fprintn(FILE * file, const acb_t x, slong digits, ulong flags)
     void acb_randtest(acb_t z, flint_rand_t state, slong prec, slong mag_bits)
     void acb_randtest_special(acb_t z, flint_rand_t state, slong prec, slong mag_bits)
     void acb_randtest_precise(acb_t z, flint_rand_t state, slong prec, slong mag_bits)
     void acb_randtest_param(acb_t z, flint_rand_t state, slong prec, slong mag_bits)
+    void acb_urandom(acb_t z, flint_rand_t state, slong prec)
     int acb_is_zero(const acb_t z)
     int acb_is_one(const acb_t z)
     int acb_is_finite(const acb_t z)
@@ -104,6 +103,7 @@ cdef extern from "flint/acb.h":
     void acb_sub(acb_t z, const acb_t x, const acb_t y, slong prec)
     void acb_mul_onei(acb_t z, const acb_t x)
     void acb_div_onei(acb_t z, const acb_t x)
+    void acb_mul_i_pow_si(acb_t z, const acb_t x, slong k)
     void acb_mul_ui(acb_t z, const acb_t x, ulong y, slong prec)
     void acb_mul_si(acb_t z, const acb_t x, slong y, slong prec)
     void acb_mul_fmpz(acb_t z, const acb_t x, const fmpz_t y, slong prec)
@@ -143,6 +143,7 @@ cdef extern from "flint/acb.h":
     void acb_sqrt_analytic(acb_t r, const acb_t z, int analytic, slong prec)
     void acb_rsqrt(acb_t r, const acb_t z, slong prec)
     void acb_rsqrt_analytic(acb_t r, const acb_t z, int analytic, slong prec)
+    void acb_sqrts(acb_t y1, acb_t y2, const acb_t x, slong prec)
     void acb_quadratic_roots_fmpz(acb_t r1, acb_t r2, const fmpz_t a, const fmpz_t b, const fmpz_t c, slong prec)
     void acb_root_ui(acb_t r, const acb_t z, ulong k, slong prec)
     void acb_pow_fmpz(acb_t y, const acb_t b, const fmpz_t e, slong prec)
@@ -226,9 +227,16 @@ cdef extern from "flint/acb.h":
     void _acb_vec_zero(acb_ptr A, slong n)
     int _acb_vec_is_zero(acb_srcptr vec, slong len)
     int _acb_vec_is_real(acb_srcptr v, slong len)
+    int _acb_vec_is_finite(acb_srcptr vec, slong len)
+    int _acb_vec_equal(acb_srcptr vec1, acb_srcptr vec2, slong len)
+    int _acb_vec_overlaps(acb_srcptr vec1, acb_srcptr vec2, slong len)
+    int _acb_vec_contains(acb_srcptr vec1, acb_srcptr vec2, slong len)
     void _acb_vec_set(acb_ptr res, acb_srcptr vec, slong len)
     void _acb_vec_set_round(acb_ptr res, acb_srcptr vec, slong len, slong prec)
     void _acb_vec_swap(acb_ptr vec1, acb_ptr vec2, slong len)
+    void _acb_vec_get_real(arb_ptr re, acb_srcptr vec, slong len)
+    void _acb_vec_get_imag(arb_ptr im, acb_srcptr vec, slong len)
+    void _acb_vec_set_real_imag(acb_ptr vec, arb_srcptr re, arb_srcptr im, slong len)
     void _acb_vec_neg(acb_ptr res, acb_srcptr vec, slong len)
     void _acb_vec_add(acb_ptr res, acb_srcptr vec1, acb_srcptr vec2, slong len, slong prec)
     void _acb_vec_sub(acb_ptr res, acb_srcptr vec1, acb_srcptr vec2, slong len, slong prec)
@@ -244,6 +252,7 @@ cdef extern from "flint/acb.h":
     void _acb_vec_scalar_div_arb(acb_ptr res, acb_srcptr vec, slong len, const arb_t c, slong prec)
     void _acb_vec_scalar_mul_fmpz(acb_ptr res, acb_srcptr vec, slong len, const fmpz_t c, slong prec)
     void _acb_vec_scalar_div_fmpz(acb_ptr res, acb_srcptr vec, slong len, const fmpz_t c, slong prec)
+    void _acb_vec_sqr(acb_ptr res, acb_srcptr vec, slong len, slong prec)
     slong _acb_vec_bits(acb_srcptr vec, slong len)
     void _acb_vec_set_powers(acb_ptr xs, const acb_t x, slong len, slong prec)
     void _acb_vec_unit_roots(acb_ptr z, slong order, slong len, slong prec)
@@ -253,3 +262,5 @@ cdef extern from "flint/acb.h":
     void _acb_vec_trim(acb_ptr res, acb_srcptr vec, slong len)
     int _acb_vec_get_unique_fmpz_vec(fmpz_struct * res,  acb_srcptr vec, slong len)
     void _acb_vec_sort_pretty(acb_ptr vec, slong len)
+    void _acb_vec_printd(acb_srcptr vec, slong len, slong digits)
+    void _acb_vec_printn(acb_srcptr vec, slong len, slong digits, ulong flags)

src/flint/flintlib/acb_dirichlet.pxd

@@ -4,7 +4,7 @@ from flint.flintlib.flint cimport ulong, slong
 from flint.flintlib.acb_poly cimport acb_poly_t
 from flint.flintlib.fmpz cimport fmpz_t
 from flint.flintlib.arb_types cimport mag_t, mag_struct, arb_t, arb_ptr
-from flint.flintlib.acb cimport acb_struct, acb_srcptr
+from flint.flintlib.acb_types cimport acb_struct, acb_srcptr
 from flint.flintlib.fmpq cimport fmpq_t
 from flint.flintlib.arf cimport arf_t
 from flint.flintlib.arb cimport arb_srcptr

src/flint/flintlib/acb_hypgeom.pxd

@@ -1,7 +1,12 @@
-from flint.flintlib.acb cimport acb_t, acb_srcptr, acb_ptr
-from flint.flintlib.acb_poly cimport acb_poly_t, acb_poly_struct
-from flint.flintlib.mag cimport mag_t
 from flint.flintlib.flint cimport ulong, slong
+from flint.flintlib.acb_types cimport (
+    acb_t,
+    acb_srcptr,
+    acb_ptr,
+    acb_poly_t,
+    acb_poly_struct,
+)
+from flint.flintlib.arb_types cimport mag_t
 
 cdef extern from "flint/acb_hypgeom.h":
 # from here on is parsed

src/flint/flintlib/acb_mat.pxd

@@ -1,27 +1,16 @@
-from flint.flintlib.flint cimport ulong, flint_rand_t, slong
-from flint.flintlib.fmpz_mat cimport fmpz_mat_t
+from flint.flintlib.acb_types cimport acb_mat_t, acb_poly_t, acb_ptr, acb_srcptr, acb_t
+from flint.flintlib.arb_types cimport arb_mat_t, arb_t, mag_t
+from flint.flintlib.flint cimport flint_rand_t, slong, ulong
 from flint.flintlib.fmpq_mat cimport fmpq_mat_t
-from flint.flintlib.mag cimport mag_t
-from flint.flintlib.fmpz cimport fmpz_t
-from flint.flintlib.acb_poly cimport acb_poly_t
-from flint.flintlib.arb cimport arb_t
-from flint.flintlib.acb cimport  acb_ptr, acb_struct, acb_t, acb_srcptr
-from flint.flintlib.arb_mat cimport arb_mat_t
+from flint.flintlib.fmpz_types cimport fmpz_mat_t, fmpz_t
 
-cdef extern from "flint/acb_mat.h":
-    ctypedef struct acb_mat_struct:
-        acb_ptr entries
-        long r
-        long c
-        acb_ptr * rows
+# unknown type FILE
 
-    ctypedef acb_mat_struct acb_mat_t[1]
-#macros
-    acb_struct * acb_mat_entry(acb_mat_t mat, long i, long j)
+# .. macro:: acb_mat_entry(mat, i, j)
+# .. macro:: acb_mat_nrows(mat)
+# .. macro:: acb_mat_ncols(mat)
 
-    long acb_mat_nrows(const acb_mat_t x)
-    long acb_mat_ncols(const acb_mat_t x)
-# from here on is parsed
+cdef extern from "flint/acb_mat.h":
     void acb_mat_init(acb_mat_t mat, slong r, slong c)
     void acb_mat_clear(acb_mat_t mat)
     slong acb_mat_allocated_bytes(const acb_mat_t x)
@@ -33,9 +22,13 @@ cdef extern from "flint/acb_mat.h":
     void acb_mat_set_fmpq_mat(acb_mat_t dest, const fmpq_mat_t src, slong prec)
     void acb_mat_set_arb_mat(acb_mat_t dest, const arb_mat_t src)
     void acb_mat_set_round_arb_mat(acb_mat_t dest, const arb_mat_t src, slong prec)
+    void acb_mat_get_real(arb_mat_t re, const arb_mat_t mat)
+    void acb_mat_get_imag(arb_mat_t im, const arb_mat_t mat)
+    void acb_mat_set_real_imag(acb_mat_t mat, const arb_mat_t re, const arb_mat_t im)
     void acb_mat_randtest(acb_mat_t mat, flint_rand_t state, slong prec, slong mag_bits)
     void acb_mat_randtest_eig(acb_mat_t mat, flint_rand_t state, acb_srcptr E, slong prec)
     void acb_mat_printd(const acb_mat_t mat, slong digits)
+    # void acb_mat_fprintd(FILE * file, const acb_mat_t mat, slong digits)
     int acb_mat_equal(const acb_mat_t mat1, const acb_mat_t mat2)
     int acb_mat_overlaps(const acb_mat_t mat1, const acb_mat_t mat2)
     int acb_mat_contains(const acb_mat_t mat1, const acb_mat_t mat2)
@@ -55,13 +48,14 @@ cdef extern from "flint/acb_mat.h":
     void acb_mat_zero(acb_mat_t mat)
     void acb_mat_one(acb_mat_t mat)
     void acb_mat_ones(acb_mat_t mat)
+    void acb_mat_onei(acb_mat_t mat)
     void acb_mat_indeterminate(acb_mat_t mat)
     void acb_mat_dft(acb_mat_t mat, int type, slong prec)
     void acb_mat_transpose(acb_mat_t dest, const acb_mat_t src)
     void acb_mat_conjugate_transpose(acb_mat_t dest, const acb_mat_t src)
     void acb_mat_conjugate(acb_mat_t dest, const acb_mat_t src)
     void acb_mat_bound_inf_norm(mag_t b, const acb_mat_t A)
-    void acb_mat_frobenius_norm(acb_t res, const acb_mat_t A, slong prec)
+    void acb_mat_frobenius_norm(arb_t res, const acb_mat_t A, slong prec)
     void acb_mat_bound_frobenius_norm(mag_t res, const acb_mat_t A)
     void acb_mat_neg(acb_mat_t dest, const acb_mat_t src)
     void acb_mat_add(acb_mat_t res, const acb_mat_t mat1, const acb_mat_t mat2, slong prec)
@@ -88,6 +82,10 @@ cdef extern from "flint/acb_mat.h":
     void acb_mat_scalar_div_fmpz(acb_mat_t B, const acb_mat_t A, const fmpz_t c, slong prec)
     void acb_mat_scalar_div_arb(acb_mat_t B, const acb_mat_t A, const arb_t c, slong prec)
     void acb_mat_scalar_div_acb(acb_mat_t B, const acb_mat_t A, const acb_t c, slong prec)
+    void _acb_mat_vector_mul_row(acb_ptr res, acb_srcptr v, const acb_mat_t A, slong prec)
+    void _acb_mat_vector_mul_col(acb_ptr res, const acb_mat_t A, acb_srcptr v, slong prec)
+    void acb_mat_vector_mul_row(acb_ptr res, acb_srcptr v, const acb_mat_t A, slong prec)
+    void acb_mat_vector_mul_col(acb_ptr res, const acb_mat_t A, acb_srcptr v, slong prec)
     int acb_mat_lu_classical(slong * perm, acb_mat_t LU, const acb_mat_t A, slong prec)
     int acb_mat_lu_recursive(slong * perm, acb_mat_t LU, const acb_mat_t A, slong prec)
     int acb_mat_lu(slong * perm, acb_mat_t LU, const acb_mat_t A, slong prec)
@@ -124,7 +122,7 @@ cdef extern from "flint/acb_mat.h":
     void acb_mat_add_error_mag(acb_mat_t mat, const mag_t err)
     int acb_mat_approx_eig_qr(acb_ptr E, acb_mat_t L, acb_mat_t R, const acb_mat_t A, const mag_t tol, slong maxiter, slong prec)
     void acb_mat_eig_global_enclosure(mag_t eps, const acb_mat_t A, acb_srcptr E, const acb_mat_t R, slong prec)
-    void acb_mat_eig_enclosure_rump(acb_t l, acb_mat_t J, acb_mat_t R, const acb_mat_t A, const acb_t lambda_approx, const acb_mat_t R_approx, slong prec)
+    void acb_mat_eig_enclosure_rump(acb_t lambda_, acb_mat_t J, acb_mat_t R, const acb_mat_t A, const acb_t lambda_approx, const acb_mat_t R_approx, slong prec)
     int acb_mat_eig_simple_rump(acb_ptr E, acb_mat_t L, acb_mat_t R, const acb_mat_t A, acb_srcptr E_approx, const acb_mat_t R_approx, slong prec)
     int acb_mat_eig_simple_vdhoeven_mourrain(acb_ptr E, acb_mat_t L, acb_mat_t R, const acb_mat_t A, acb_srcptr E_approx, const acb_mat_t R_approx, slong prec)
     int acb_mat_eig_simple(acb_ptr E, acb_mat_t L, acb_mat_t R, const acb_mat_t A, acb_srcptr E_approx, const acb_mat_t R_approx, slong prec)