Begin adding fmpz_mod_poly

setup.py

@@ -95,6 +95,7 @@
     ("flint.types.acb_series", ["src/flint/types/acb_series.pyx"]),
     ("flint.types.fmpz_mpoly", ["src/flint/types/fmpz_mpoly.pyx"]),
     ("flint.types.fmpz_mod", ["src/flint/types/fmpz_mod.pyx"]),
+    ("flint.types.fmpz_mod_poly", ["src/flint/types/fmpz_mod_poly.pyx"]),
     ("flint.types.dirichlet", ["src/flint/types/dirichlet.pyx"]),
     ("flint.flint_base.flint_base", ["src/flint/flint_base/flint_base.pyx"]),
     ("flint.flint_base.flint_context", ["src/flint/flint_base/flint_context.pyx"]),

src/flint/__init__.py

@@ -22,6 +22,7 @@
 from .types.acb_series import *
 from .types.fmpz_mpoly import *
 from .types.fmpz_mod import *
+from .types.fmpz_mod_poly import *
 from .types.dirichlet import *
 from .functions.showgood import showgood
 

src/flint/flintlib/fmpz_mod_poly_factor.pxd

@@ -0,0 +1,46 @@
+from flint.flintlib.flint cimport fmpz_struct, slong, flint_rand_t
+from flint.flintlib.fmpz cimport fmpz_t
+from flint.flintlib.fmpz_mod cimport fmpz_mod_ctx_t
+from flint.flintlib.fmpz_factor cimport fmpz_factor_t
+from flint.flintlib.fmpz_mod_poly cimport fmpz_mod_poly_struct, fmpz_mod_poly_t
+
+# unimported types  {'fmpz_mod_poly_factor_t', 'void', 'fmpz_mod_poly_t'}
+
+cdef extern from "flint/fmpz_mod_poly_factor.h":
+    ctypedef struct fmpz_mod_poly_factor_struct:
+        fmpz_mod_poly_struct * poly
+        slong *exp
+        slong num
+        slong alloc
+    ctypedef fmpz_mod_poly_factor_struct fmpz_mod_poly_factor_t[1]
+
+    # Parsed from here
+    void fmpz_mod_poly_factor_init(fmpz_mod_poly_factor_t fac, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_clear(fmpz_mod_poly_factor_t fac, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_realloc(fmpz_mod_poly_factor_t fac, slong alloc, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_fit_length(fmpz_mod_poly_factor_t fac, slong len, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_set(fmpz_mod_poly_factor_t res, const fmpz_mod_poly_factor_t fac, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_print(const fmpz_mod_poly_factor_t fac, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_insert(fmpz_mod_poly_factor_t fac, const fmpz_mod_poly_t poly, slong exp, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_concat(fmpz_mod_poly_factor_t res, const fmpz_mod_poly_factor_t fac, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_pow(fmpz_mod_poly_factor_t fac, slong exp, const fmpz_mod_ctx_t ctx)
+    int fmpz_mod_poly_is_irreducible(const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    int fmpz_mod_poly_is_irreducible_ddf(const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    int fmpz_mod_poly_is_irreducible_rabin(const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    int fmpz_mod_poly_is_irreducible_rabin_f(fmpz_t r, const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    int _fmpz_mod_poly_is_squarefree(const fmpz_struct * f, slong len, const fmpz_mod_ctx_t ctx)
+    int _fmpz_mod_poly_is_squarefree_f(fmpz_t fac, const fmpz_struct * f, slong len, const fmpz_mod_ctx_t ctx)
+    int fmpz_mod_poly_is_squarefree(const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    int fmpz_mod_poly_is_squarefree_f(fmpz_t fac, const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    int fmpz_mod_poly_factor_equal_deg_prob(fmpz_mod_poly_t factor, flint_rand_t state, const fmpz_mod_poly_t pol, slong d, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_equal_deg(fmpz_mod_poly_factor_t factors, const fmpz_mod_poly_t pol, slong d, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_distinct_deg(fmpz_mod_poly_factor_t res, const fmpz_mod_poly_t poly, slong * const *degs, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_distinct_deg_threaded(fmpz_mod_poly_factor_t res, const fmpz_mod_poly_t poly, slong * const *degs, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_squarefree(fmpz_mod_poly_factor_t res, const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor(fmpz_mod_poly_factor_t res, const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_cantor_zassenhaus(fmpz_mod_poly_factor_t res, const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_kaltofen_shoup(fmpz_mod_poly_factor_t res, const fmpz_mod_poly_t poly, const fmpz_mod_ctx_t ctx)
+    void fmpz_mod_poly_factor_berlekamp(fmpz_mod_poly_factor_t factors, const fmpz_mod_poly_t f, const fmpz_mod_ctx_t ctx)
+    void _fmpz_mod_poly_interval_poly_worker(void* arg_ptr)
+    void fmpz_mod_poly_roots(fmpz_mod_poly_factor_t r, const fmpz_mod_poly_t f, int with_multiplicity, const fmpz_mod_ctx_t ctx)
+    int fmpz_mod_poly_roots_factored(fmpz_mod_poly_factor_t r, const fmpz_mod_poly_t f, int with_multiplicity, const fmpz_factor_t n, const fmpz_mod_ctx_t ctx)

src/flint/test/test.py

@@ -1813,7 +1813,17 @@ def test_fmpz_mod_dlog():
             x = g.discrete_log(a)
             assert g**x == a
 
+def test_fmpz_mod_poly():
+    from flint import fmpz_poly, fmpz_mod_poly_ctx
+
+    R = fmpz_mod_poly_ctx(11)
+    f = fmpz_poly([1,2,3])
+    g = R(f)
+    assert str(g) == "3*x^2 + 2*x + 1"
 
+    f = fmpz_poly([12,13,14])
+    g = R(f)
+    assert str(g) == "3*x^2 + 2*x + 1"
 
 all_tests = [
     test_pyflint,
@@ -1834,5 +1844,6 @@ def test_fmpz_mod_dlog():
     test_nmod_mat,
     test_arb,
     test_fmpz_mod,
-    test_fmpz_mod_dlog
+    test_fmpz_mod_dlog,
+    test_fmpz_mod_poly
 ]

src/flint/types/fmpz_mod_poly.pxd

@@ -0,0 +1,14 @@
+from flint.flintlib.fmpz_mod cimport fmpz_mod_ctx_t
+from flint.flintlib.fmpz_mod_poly cimport *
+
+from flint.flint_base.flint_base cimport flint_poly
+from flint.types.fmpz_mod cimport fmpz_mod_ctx
+
+cdef class fmpz_mod_poly_ctx(fmpz_mod_ctx):
+    pass
+
+cdef class fmpz_mod_poly(flint_poly):
+    cdef fmpz_mod_poly_t val
+    cdef fmpz_mod_poly_ctx ctx
+    cpdef long length(self)
+    cpdef long degree(self)

src/flint/types/fmpz_mod_poly.pyx

@@ -0,0 +1,91 @@
+from flint.flintlib.fmpz_mod_poly cimport *
+from flint.flintlib.fmpz cimport(
+    fmpz_equal,
+)
+from flint.types.fmpz cimport fmpz
+from flint.types.fmpz_mod cimport fmpz_mod_ctx, fmpz_mod
+from flint.types.fmpz_poly cimport any_as_fmpz_poly, fmpz_poly
+
+from flint.flint_base.flint_base cimport flint_poly
+from flint.utils.typecheck cimport typecheck
+
+cdef class fmpz_mod_poly_ctx(fmpz_mod_ctx):
+    """
+    NOTE:
+
+    Technically this could just be the same as `fmpz_mod_ctx`,
+    however, the usage of fmpz_mod_ctx allows the creation of 
+    `fmpz_mod` types by calling the context class. For symmetry
+    we allow this to be the case here.
+    """
+    def __eq__(self, other):
+        # Most often, we expect both `fmpz_mod_poly` to be pointing 
+        # to the same ctx, so this seems the fastest way to check
+        if self is other:
+            return True
+
+        # If they're not the same object in memory, they may have the
+        # same modulus, which is good enough
+        if typecheck(other, fmpz_mod_poly_ctx):
+            return fmpz_equal(self.val.n, (<fmpz_mod_poly_ctx>other).val.n)
+        return False
+
+    def __str__(self):
+        return f"Context for fmpz_mod_poly with modulus: {self.modulus()}"
+
+    def __repr__(self):
+        return f"fmpz_mod_poly_ctx({self.modulus()})"
+
+    def __call__(self, val):
+        return fmpz_mod_poly(val, self)
+
+
+cdef class fmpz_mod_poly(flint_poly):
+    """
+    """
+    def __cinit__(self):
+        fmpz_mod_poly_init(self.val, self.ctx.val)
+
+    def __dealloc__(self):
+        fmpz_mod_poly_clear(self.val, self.ctx.val)
+
+    def __init__(self, val, ctx):
+        if not typecheck(ctx, fmpz_mod_ctx):
+            raise TypeError
+        self.ctx = ctx
+
+        val = any_as_fmpz_poly(val)
+        if val is NotImplemented:
+            raise TypeError
+
+        fmpz_mod_poly_set_fmpz_poly(
+            self.val, (<fmpz_poly>val).val, self.ctx.val
+        )
+
+    def __getitem__(self, long i):
+        cdef fmpz x
+        x = fmpz()
+        if i < 0:
+            return x
+        fmpz_mod_poly_get_coeff_fmpz(x.val, self.val, i, self.ctx.val)
+        return x
+
+    def __setitem__(self, long i, x):
+        if i < 0:
+            raise ValueError("cannot assign to index < 0 of polynomial")
+        v = self.ctx.any_as_fmpz_mod(x)
+        if v is NotImplemented:
+            raise TypeError
+        fmpz_mod_poly_set_coeff_fmpz(self.val, i, (<fmpz_mod>v).val, self.ctx.val)
+
+    def repr(self):
+        return "fmpz_mod_poly([%s])" % (", ".join(map(str, self.coeffs())))
+
+    def __len__(self):
+        return fmpz_mod_poly_length(self.val, self.ctx.val)
+
+    cpdef long length(self):
+        return fmpz_mod_poly_length(self.val, self.ctx.val)
+
+    cpdef long degree(self):
+        return fmpz_mod_poly_degree(self.val, self.ctx.val)