feat: add gr versions of fmpz and fmpq

Author: oscarbenjamin

src/flint/__init__.py

@@ -1,5 +1,7 @@
 from .pyflint import *
 
+from .types.gr import gr_ctx, gr_fmpz_ctx, gr_fmpq_ctx
+
 from .types.fmpz import *
 from .types.fmpz_poly import *
 from .types.fmpz_mat import *

src/flint/flint_base/flint_base.pxd

@@ -1,5 +1,8 @@
 from flint.flintlib.types.mpoly cimport ordering_t
 
+cdef class flint_ctx:
+    pass
+
 cdef class flint_elem:
     pass
 

src/flint/flintlib/functions/gr_domains.pxd

@@ -5,8 +5,9 @@ from flint.flintlib.types.gr cimport gr_ctx_t, truth_t
 from flint.flintlib.types.mpoly cimport ordering_t
 
 
-
-cdef extern from "flint/gr_domains.h":
+# XXX: Need to fix the header from gr_domain.h to gr.h:
+# Needs hard-coding in rst_to_pxd.py
+cdef extern from "flint/gr.h":
     int gr_ctx_cmp_coercion(gr_ctx_t ctx1, gr_ctx_t ctx2)
     truth_t gr_ctx_is_finite(gr_ctx_t ctx)
     truth_t gr_ctx_is_multiplicative_group(gr_ctx_t ctx)

src/flint/flintlib/types/flint.pxd

@@ -61,14 +61,23 @@ cdef extern from "flint/flint.h":
     #define SIZEOF_ULONG sizeof(ulong)
     """
     int SIZEOF_ULONG
+
+    ctypedef struct __FLINT_FILE:
+        pass
+    ctypedef __FLINT_FILE FLINT_FILE
+
     const char * FLINT_VERSION
     const int __FLINT_RELEASE
+
     const int FLINT_BITS
+
     ctypedef void * flint_rand_t
     void flint_rand_init(flint_rand_t state)
     void flint_rand_clear(flint_rand_t state)
+
     void flint_set_num_threads(long)
     long flint_get_num_threads()
+
     void flint_cleanup()
 
     ctypedef struct nmod_t:

src/flint/flintlib/types/gr.pxd

@@ -1,12 +1,25 @@
+from flint.flintlib.types.flint cimport (
+    FLINT_FILE,
+    slong,
+    ulong,
+    flint_bitcnt_t,
+)
+
+
 cdef extern from "flint/gr.h":
 
     cdef enum truth_t_enum:
         T_TRUE
         T_FALSE
-        T_UNKOWN
+        T_UNKNOWN
 
     ctypedef truth_t_enum truth_t
 
+    # Macros
+    cdef int GR_SUCCESS
+    cdef int GR_DOMAIN
+    cdef int GR_UNABLE
+
     ctypedef struct gr_stream_struct:
         FLINT_FILE * fp
         char * s
@@ -15,12 +28,10 @@ cdef extern from "flint/gr.h":
 
     ctypedef gr_stream_struct gr_stream_t[1]
 
-    ctypedef int (*gr_funcptr)(void)
+    ctypedef int (*gr_funcptr)()
 
     cdef const int GR_CTX_STRUCT_DATA_BYTES
 
-    ctypedef int (*gr_funcptr)(void)
-
     cdef struct gr_ctx_struct:
         char data[GR_CTX_STRUCT_DATA_BYTES]
         ulong which_ring

src/flint/types/gr.pxd

@@ -0,0 +1,125 @@
+cimport cython
+
+from flint.flintlib.types.gr cimport (
+    truth_t,
+    T_TRUE,
+    T_FALSE,
+    T_UNKNOWN,
+
+    GR_SUCCESS,
+    GR_DOMAIN,
+    GR_UNABLE,
+
+    gr_ctx_t,
+    gr_ptr,
+)
+from flint.flintlib.functions.gr cimport (
+    gr_heap_init,
+    gr_set_str,
+    gr_get_str,
+
+    gr_neg,
+    gr_add,
+    gr_add_si,
+    gr_sub,
+    gr_sub_si,
+    gr_mul,
+    gr_mul_si,
+    gr_inv,
+    gr_div,
+    gr_div_si,
+    gr_pow_si,
+)
+
+from flint.flint_base.flint_base cimport flint_ctx, flint_scalar
+
+from flint.utils.flint_exceptions import DomainError, UnableError
+
+
+cdef inline truth_to_py(truth_t t):
+    if t == T_TRUE:
+        return True
+    elif t == T_FALSE:
+        return False
+    else:
+        return None
+
+
+@cython.no_gc
+cdef class gr_ctx(flint_ctx):
+    cdef gr_ctx_t ctx_t
+    cdef bint _init
+
+    @cython.final
+    cdef inline gr new_gr(self):
+        cdef gr py_val
+        cdef gr_ptr pval
+        pval = gr_heap_init(self.ctx_t)
+        if pval == NULL:
+            raise MemoryError("Failed to allocate memory for gr object")
+        py_val = gr.__new__(gr)
+        py_val.pval = pval
+        py_val.ctx = self
+        py_val._init = True
+        return py_val
+
+    @cython.final
+    cdef inline gr from_str(self, s: str):
+        cdef gr py_val
+        cdef bytes b
+        b = s.encode('utf-8')
+        py_val = self.new_gr()
+        err = gr_set_str(py_val.pval, b, self.ctx_t)
+        if err != GR_SUCCESS:
+            raise self._error(err, "Failed to parse string")
+        return py_val
+
+    @cython.final
+    cdef inline str to_str(self, val: gr):
+        cdef str py_str
+        cdef char *s
+        err = gr_get_str(&s, val.pval, self.ctx_t)
+        if err != GR_SUCCESS:
+            raise self._error(err, "Failed to convert to string")
+        py_str = (<bytes>s).decode("utf-8")
+        # flint_free(s)
+        return py_str
+
+    @cython.final
+    cdef inline _error(self, int err, str msg):
+        if err & GR_DOMAIN:
+            return DomainError(msg)
+        elif err & GR_UNABLE:
+            return UnableError(msg)
+        else:
+            return AssertionError("Bad error code")
+
+
+@cython.no_gc
+cdef class _gr_fmpz_ctx(gr_ctx):
+
+    @staticmethod
+    cdef _gr_fmpz_ctx _new()
+
+
+@cython.no_gc
+cdef class _gr_fmpq_ctx(gr_ctx):
+
+    @staticmethod
+    cdef _gr_fmpq_ctx _new()
+
+
+# The global contexts for use in cython code:
+cdef _gr_fmpz_ctx gr_fmpz_ctx_c
+cdef _gr_fmpq_ctx gr_fmpq_ctx_c
+
+
+@cython.no_gc
+cdef class gr(flint_scalar):
+    cdef gr_ptr pval
+    cdef gr_ctx ctx
+    cdef bint _init
+
+    @cython.final
+    cdef inline _error(self, int err, str msg):
+        return self.ctx._error(err, msg)