bindings: auto generate arb_poly and arb_mat .pxd files

Author: oscarbenjamin

bin/all_rst_to_pxd.sh

@@ -13,6 +13,8 @@ modules="\
     fmpz_lll\
     arf\
     arb\
+    arb_poly\
+    arb_mat\
     "
 
 for module in $modules; do

src/flint/flintlib/arb_mat.pxd

@@ -1,24 +1,15 @@
-from flint.flintlib.arb_types cimport arb_ptr, arb_struct, mag_t, mag_srcptr, arb_t, arb_srcptr
-from flint.flintlib.fmpz_mat cimport fmpz_mat_t
+from flint.flintlib.arb_types cimport arb_mat_t, arb_poly_t, arb_ptr, arb_srcptr, arb_t, mag_srcptr, mag_t
+from flint.flintlib.flint cimport flint_rand_t, slong, ulong
 from flint.flintlib.fmpq_mat cimport fmpq_mat_t
-from flint.flintlib.flint cimport ulong, slong, flint_rand_t
-from flint.flintlib.fmpz cimport fmpz_t
-from flint.flintlib.arb_poly cimport arb_poly_t
+from flint.flintlib.fmpz_types cimport fmpz_mat_t, fmpz_t
 
-cdef extern from "flint/arb_mat.h":
-    ctypedef struct arb_mat_struct:
-        arb_ptr entries
-        long r
-        long c
-        arb_ptr * rows
+# unknown type FILE
 
-    ctypedef arb_mat_struct arb_mat_t[1]
-#macros
-    arb_struct * arb_mat_entry(arb_mat_t mat, long i, long j)
+# .. macro:: arb_mat_entry(mat, i, j)
+# .. macro:: arb_mat_nrows(mat)
+# .. macro:: arb_mat_ncols(mat)
 
-    long arb_mat_nrows(const arb_mat_t x)
-    long arb_mat_ncols(const arb_mat_t x)
-# from here on is parsed
+cdef extern from "flint/arb_mat.h":
     void arb_mat_init(arb_mat_t mat, slong r, slong c)
     void arb_mat_clear(arb_mat_t mat)
     slong arb_mat_allocated_bytes(const arb_mat_t x)
@@ -29,7 +20,10 @@ cdef extern from "flint/arb_mat.h":
     void arb_mat_set_round_fmpz_mat(arb_mat_t dest, const fmpz_mat_t src, slong prec)
     void arb_mat_set_fmpq_mat(arb_mat_t dest, const fmpq_mat_t src, slong prec)
     void arb_mat_randtest(arb_mat_t mat, flint_rand_t state, slong prec, slong mag_bits)
+    void arb_mat_randtest_cho(arb_mat_t mat, flint_rand_t state, slong prec, slong mag_bits)
+    void arb_mat_randtest_spd(arb_mat_t mat, flint_rand_t state, slong prec, slong mag_bits)
     void arb_mat_printd(const arb_mat_t mat, slong digits)
+    # void arb_mat_fprintd(FILE * file, const arb_mat_t mat, slong digits)
     int arb_mat_equal(const arb_mat_t mat1, const arb_mat_t mat2)
     int arb_mat_overlaps(const arb_mat_t mat1, const arb_mat_t mat2)
     int arb_mat_contains(const arb_mat_t mat1, const arb_mat_t mat2)
@@ -80,6 +74,10 @@ cdef extern from "flint/arb_mat.h":
     void arb_mat_scalar_div_si(arb_mat_t B, const arb_mat_t A, slong c, slong prec)
     void arb_mat_scalar_div_fmpz(arb_mat_t B, const arb_mat_t A, const fmpz_t c, slong prec)
     void arb_mat_scalar_div_arb(arb_mat_t B, const arb_mat_t A, const arb_t c, slong prec)
+    void _arb_mat_vector_mul_row(arb_ptr res, arb_srcptr v, const arb_mat_t A, slong prec)
+    void _arb_mat_vector_mul_col(arb_ptr res, const arb_mat_t A, arb_srcptr v, slong prec)
+    void arb_mat_vector_mul_row(arb_ptr res, arb_srcptr v, const arb_mat_t A, slong prec)
+    void arb_mat_vector_mul_col(arb_ptr res, const arb_mat_t A, arb_srcptr v, slong prec)
     int arb_mat_lu_classical(slong * perm, arb_mat_t LU, const arb_mat_t A, slong prec)
     int arb_mat_lu_recursive(slong * perm, arb_mat_t LU, const arb_mat_t A, slong prec)
     int arb_mat_lu(slong * perm, arb_mat_t LU, const arb_mat_t A, slong prec)
@@ -130,3 +128,6 @@ cdef extern from "flint/arb_mat.h":
     slong arb_mat_count_not_is_zero(const arb_mat_t mat)
     void arb_mat_get_mid(arb_mat_t B, const arb_mat_t A)
     void arb_mat_add_error_mag(arb_mat_t mat, const mag_t err)
+    int arb_mat_spd_get_fmpz_mat(fmpz_mat_t B, const arb_mat_t A, slong prec)
+    void arb_mat_spd_lll_reduce(fmpz_mat_t U, const arb_mat_t A, slong prec)
+    int arb_mat_spd_is_lll_reduced(const arb_mat_t A, slong tol_exp, slong prec)

src/flint/flintlib/arb_poly.pxd

@@ -1,23 +1,15 @@
-from flint.flintlib.flint cimport flint_rand_t, ulong, slong
-from flint.flintlib.arb cimport arb_ptr, arb_t, arb_srcptr
-from flint.flintlib.fmpz_poly cimport fmpz_poly_t
+from flint.flintlib.acb cimport acb_srcptr, acb_t
+from flint.flintlib.arb_types cimport arb_poly_t, arb_ptr, arb_srcptr, arb_t, mag_t
+from flint.flintlib.arf_types cimport arf_t
+from flint.flintlib.flint cimport flint_rand_t, slong, ulong
 from flint.flintlib.fmpq_poly cimport fmpq_poly_t
-from flint.flintlib.acb cimport acb_t, acb_srcptr
-from flint.flintlib.mag cimport mag_t
-from flint.flintlib.arf cimport arf_t
+from flint.flintlib.fmpz_types cimport fmpz_poly_t
 
-cdef extern from "flint/arb_poly.h":
-    ctypedef struct arb_poly_struct:
-        arb_ptr coeffs
-        long length
-        long alloc
-
-    ctypedef arb_poly_struct arb_poly_t[1]
-    #macros
+# unknown type FILE
 
-    arb_ptr arb_poly_get_coeff_ptr(arb_poly_t poly, long n)
+# .. macro:: arb_poly_get_coeff_ptr(poly, n)
 
-# from here on is parsed
+cdef extern from "flint/arb_poly.h":
     void arb_poly_init(arb_poly_t poly)
     void arb_poly_clear(arb_poly_t poly)
     void arb_poly_fit_length(arb_poly_t poly, slong len)
@@ -31,7 +23,6 @@ cdef extern from "flint/arb_poly.h":
     int arb_poly_is_x(const arb_poly_t poly)
     void arb_poly_zero(arb_poly_t poly)
     void arb_poly_one(arb_poly_t poly)
-    void arb_poly_set_arb(arb_poly_t poly, const arb_t c)
     void arb_poly_set(arb_poly_t dest, const arb_poly_t src)
     void arb_poly_set_round(arb_poly_t dest, const arb_poly_t src, slong prec)
     void arb_poly_set_trunc(arb_poly_t dest, const arb_poly_t src, slong n)
@@ -49,6 +40,7 @@ cdef extern from "flint/arb_poly.h":
     void arb_poly_set_fmpq_poly(arb_poly_t poly, const fmpq_poly_t src, slong prec)
     void arb_poly_set_si(arb_poly_t poly, slong src)
     void arb_poly_printd(const arb_poly_t poly, slong digits)
+    # void arb_poly_fprintd(FILE * file, const arb_poly_t poly, slong digits)
     void arb_poly_randtest(arb_poly_t poly, flint_rand_t state, slong len, slong prec, slong mag_bits)
     int arb_poly_contains(const arb_poly_t poly1, const arb_poly_t poly2)
     int arb_poly_contains_fmpz_poly(const arb_poly_t poly1, const fmpz_poly_t poly2)
@@ -88,32 +80,12 @@ cdef extern from "flint/arb_poly.h":
     void _arb_poly_divrem(arb_ptr Q, arb_ptr R, arb_srcptr A, slong lenA, arb_srcptr B, slong lenB, slong prec)
     int arb_poly_divrem(arb_poly_t Q, arb_poly_t R, const arb_poly_t A, const arb_poly_t B, slong prec)
     void _arb_poly_div_root(arb_ptr Q, arb_t R, arb_srcptr A, slong len, const arb_t c, slong prec)
-    void _arb_poly_taylor_shift_horner(arb_ptr g, const arb_t c, slong n, slong prec)
-    void arb_poly_taylor_shift_horner(arb_poly_t g, const arb_poly_t f, const arb_t c, slong prec)
-    void _arb_poly_taylor_shift_divconquer(arb_ptr g, const arb_t c, slong n, slong prec)
-    void arb_poly_taylor_shift_divconquer(arb_poly_t g, const arb_poly_t f, const arb_t c, slong prec)
-    void _arb_poly_taylor_shift_convolution(arb_ptr g, const arb_t c, slong n, slong prec)
-    void arb_poly_taylor_shift_convolution(arb_poly_t g, const arb_poly_t f, const arb_t c, slong prec)
     void _arb_poly_taylor_shift(arb_ptr g, const arb_t c, slong n, slong prec)
     void arb_poly_taylor_shift(arb_poly_t g, const arb_poly_t f, const arb_t c, slong prec)
-    void _arb_poly_compose_horner(arb_ptr res, arb_srcptr poly1, slong len1, arb_srcptr poly2, slong len2, slong prec)
-    void arb_poly_compose_horner(arb_poly_t res, const arb_poly_t poly1, const arb_poly_t poly2, slong prec)
-    void _arb_poly_compose_divconquer(arb_ptr res, arb_srcptr poly1, slong len1, arb_srcptr poly2, slong len2, slong prec)
-    void arb_poly_compose_divconquer(arb_poly_t res, const arb_poly_t poly1, const arb_poly_t poly2, slong prec)
     void _arb_poly_compose(arb_ptr res, arb_srcptr poly1, slong len1, arb_srcptr poly2, slong len2, slong prec)
     void arb_poly_compose(arb_poly_t res, const arb_poly_t poly1, const arb_poly_t poly2, slong prec)
-    void _arb_poly_compose_series_horner(arb_ptr res, arb_srcptr poly1, slong len1, arb_srcptr poly2, slong len2, slong n, slong prec)
-    void arb_poly_compose_series_horner(arb_poly_t res, const arb_poly_t poly1, const arb_poly_t poly2, slong n, slong prec)
-    void _arb_poly_compose_series_brent_kung(arb_ptr res, arb_srcptr poly1, slong len1, arb_srcptr poly2, slong len2, slong n, slong prec)
-    void arb_poly_compose_series_brent_kung(arb_poly_t res, const arb_poly_t poly1, const arb_poly_t poly2, slong n, slong prec)
     void _arb_poly_compose_series(arb_ptr res, arb_srcptr poly1, slong len1, arb_srcptr poly2, slong len2, slong n, slong prec)
     void arb_poly_compose_series(arb_poly_t res, const arb_poly_t poly1, const arb_poly_t poly2, slong n, slong prec)
-    void _arb_poly_revert_series_lagrange(arb_ptr h, arb_srcptr f, slong flen, slong n, slong prec)
-    void arb_poly_revert_series_lagrange(arb_poly_t h, const arb_poly_t f, slong n, slong prec)
-    void _arb_poly_revert_series_newton(arb_ptr h, arb_srcptr f, slong flen, slong n, slong prec)
-    void arb_poly_revert_series_newton(arb_poly_t h, const arb_poly_t f, slong n, slong prec)
-    void _arb_poly_revert_series_lagrange_fast(arb_ptr h, arb_srcptr f, slong flen, slong n, slong prec)
-    void arb_poly_revert_series_lagrange_fast(arb_poly_t h, const arb_poly_t f, slong n, slong prec)
     void _arb_poly_revert_series(arb_ptr h, arb_srcptr f, slong flen, slong n, slong prec)
     void arb_poly_revert_series(arb_poly_t h, const arb_poly_t f, slong n, slong prec)
     void _arb_poly_evaluate_horner(arb_t y, arb_srcptr f, slong len, const arb_t x, slong prec)
@@ -162,6 +134,8 @@ cdef extern from "flint/arb_poly.h":
     void arb_poly_interpolate_fast(arb_poly_t poly, arb_srcptr xs, arb_srcptr ys, slong n, slong prec)
     void _arb_poly_derivative(arb_ptr res, arb_srcptr poly, slong len, slong prec)
     void arb_poly_derivative(arb_poly_t res, const arb_poly_t poly, slong prec)
+    void _arb_poly_nth_derivative(arb_ptr res, arb_srcptr poly, ulong n, slong len, slong prec)
+    void arb_poly_nth_derivative(arb_poly_t res, const arb_poly_t poly, ulong n, slong prec)
     void _arb_poly_integral(arb_ptr res, arb_srcptr poly, slong len, slong prec)
     void arb_poly_integral(arb_poly_t res, const arb_poly_t poly, slong prec)
     void _arb_poly_borel_transform(arb_ptr res, arb_srcptr poly, slong len, slong prec)
@@ -175,7 +149,7 @@ cdef extern from "flint/arb_poly.h":
     void _arb_poly_binomial_transform(arb_ptr b, arb_srcptr a, slong alen, slong len, slong prec)
     void arb_poly_binomial_transform(arb_poly_t b, const arb_poly_t a, slong len, slong prec)
     void _arb_poly_graeffe_transform(arb_ptr b, arb_srcptr a, slong len, slong prec)
-    void arb_poly_graeffe_transform(arb_poly_t b, arb_poly_t a, slong prec)
+    void arb_poly_graeffe_transform(arb_poly_t b, const arb_poly_t a, slong prec)
     void _arb_poly_pow_ui_trunc_binexp(arb_ptr res, arb_srcptr f, slong flen, ulong exp, slong len, slong prec)
     void arb_poly_pow_ui_trunc_binexp(arb_poly_t res, const arb_poly_t poly, ulong exp, slong len, slong prec)
     void _arb_poly_pow_ui(arb_ptr res, arb_srcptr f, slong flen, ulong exp, slong prec)
@@ -202,10 +176,6 @@ cdef extern from "flint/arb_poly.h":
     void arb_poly_exp_series_basecase(arb_poly_t f, const arb_poly_t h, slong n, slong prec)
     void _arb_poly_exp_series(arb_ptr f, arb_srcptr h, slong hlen, slong n, slong prec)
     void arb_poly_exp_series(arb_poly_t f, const arb_poly_t h, slong n, slong prec)
-    void _arb_poly_sin_cos_series_basecase(arb_ptr s, arb_ptr c, arb_srcptr h, slong hlen, slong n, slong prec, int times_pi)
-    void arb_poly_sin_cos_series_basecase(arb_poly_t s, arb_poly_t c, const arb_poly_t h, slong n, slong prec, int times_pi)
-    void _arb_poly_sin_cos_series_tangent(arb_ptr s, arb_ptr c, arb_srcptr h, slong hlen, slong n, slong prec, int times_pi)
-    void arb_poly_sin_cos_series_tangent(arb_poly_t s, arb_poly_t c, const arb_poly_t h, slong n, slong prec, int times_pi)
     void _arb_poly_sin_cos_series(arb_ptr s, arb_ptr c, arb_srcptr h, slong hlen, slong n, slong prec)
     void arb_poly_sin_cos_series(arb_poly_t s, arb_poly_t c, const arb_poly_t h, slong n, slong prec)
     void _arb_poly_sin_series(arb_ptr s, arb_srcptr h, slong hlen, slong n, slong prec)

src/flint/flintlib/arb_types.pxd

@@ -18,14 +18,6 @@ cdef extern from "flint/arb_types.h":
     ctypedef arb_struct * arb_ptr
     ctypedef const arb_struct * arb_srcptr
 
-    # Macros
-    cdef arf_ptr arb_midref(const arb_t x)
-    cdef mag_ptr arb_radref(const arb_t x)
-
-    cdef const ulong ARB_STR_MORE
-    cdef const ulong ARB_STR_NO_RADIUS
-    cdef const ulong ARB_STR_CONDENSE
-
     ctypedef struct arb_mat_struct:
         arb_ptr entries
         slong r
@@ -40,3 +32,20 @@ cdef extern from "flint/arb_types.h":
         slong length
 
     ctypedef arb_poly_struct arb_poly_t[1]
+
+cdef extern from "flint/arb_types.h":
+    # Macros
+    cdef arf_ptr arb_midref(const arb_t x)
+    cdef mag_ptr arb_radref(const arb_t x)
+    cdef const ulong ARB_STR_MORE
+    cdef const ulong ARB_STR_NO_RADIUS
+    cdef const ulong ARB_STR_CONDENSE
+
+cdef extern from "flint/arb_mat.h":
+    # Macros
+    arb_ptr arb_mat_entry(arb_mat_t mat, slong i, slong j)
+    slong arb_mat_nrows(arb_mat_t mat)
+    slong arb_mat_ncols(arb_mat_t mat)
+
+cdef extern from "flint/arb_poly.h":
+    void arb_poly_set_arb(arb_poly_t poly, const arb_t c)

src/flint/types/acb.pyx

@@ -18,7 +18,7 @@ from flint.flintlib.mag cimport *
 from flint.flintlib.arb cimport *
 from flint.flintlib.arb_types cimport arb_midref, arb_radref
 from flint.flintlib.arf cimport *
-from flint.flintlib.arf_types cimport arf_rnd_t, ARF_RND_NEAR
+from flint.flintlib.arf_types cimport ARF_RND_NEAR
 from flint.flintlib.acb cimport *
 from flint.flintlib.acb_modular cimport *
 from flint.flintlib.acb_hypgeom cimport *

src/flint/types/acb_mat.pyx

@@ -15,7 +15,14 @@ from flint.flintlib.fmpz_mat cimport fmpz_mat_nrows, fmpz_mat_ncols
 from flint.flintlib.fmpq_mat cimport fmpq_mat_nrows, fmpq_mat_ncols
 from flint.flintlib.mag cimport *
 from flint.flintlib.arb cimport *
-from flint.flintlib.arb_types cimport mag_struct, arb_radref, arb_midref
+from flint.flintlib.arb_types cimport (
+    mag_struct,
+    arb_radref,
+    arb_midref,
+    arb_mat_entry,
+    arb_mat_nrows,
+    arb_mat_ncols,
+)
 from flint.flintlib.arb_mat cimport *
 from flint.flintlib.arf cimport *
 from flint.flintlib.acb cimport *

src/flint/types/arb_mat.pyx

@@ -16,7 +16,13 @@ from flint.flintlib.fmpz_mat cimport fmpz_mat_nrows, fmpz_mat_ncols
 from flint.flintlib.fmpq_mat cimport fmpq_mat_nrows, fmpq_mat_ncols
 from flint.flintlib.arf cimport *
 from flint.flintlib.arb cimport *
-from flint.flintlib.arb_types cimport arb_radref, arb_midref
+from flint.flintlib.arb_types cimport (
+    arb_radref,
+    arb_midref,
+    arb_mat_nrows,
+    arb_mat_ncols,
+    arb_mat_entry,
+)
 from flint.flintlib.arb_mat cimport *
 cimport cython
 

src/flint/types/arb_poly.pyx

@@ -12,7 +12,7 @@ from flint.types.fmpq cimport fmpq
 from flint.types.acb cimport acb
 from flint.types.acb_poly cimport acb_poly
 
-from flint.flintlib.arb_types cimport arb_struct
+from flint.flintlib.arb_types cimport arb_struct, arb_poly_set_arb
 from flint.flintlib.arb cimport *
 from flint.flintlib.arb_poly cimport *
 cimport libc.stdlib