lll: Use enum definitions from fmpz_lll.h

Author: oscarbenjamin

src/flint/flintlib/fmpz_types.pxd

@@ -55,12 +55,20 @@ cdef extern from "flint/fmpz_factor.h":
 
 
 cdef extern from "flint/fmpz_lll.h":
+
+    cdef enum _rep_type:
+        GRAM
+        Z_BASIS
+    ctypedef _rep_type rep_type
+
+    cdef enum _gram_type:
+        APPROX
+        EXACT
+    ctypedef _gram_type gram_type
+
     ctypedef struct fmpz_lll_struct:
         double delta
         double eta
-        int rt
-        int gt
-
+        rep_type rt
+        gram_type gt
     ctypedef fmpz_lll_struct fmpz_lll_t[1]
-    ctypedef int gram_type
-    ctypedef int rep_type

src/flint/types/fmpz_mat.pyx

@@ -10,7 +10,11 @@ cimport cython
 
 from flint.flintlib.fmpz cimport fmpz_set, fmpz_init, fmpz_clear
 from flint.flintlib.fmpz cimport fmpz_is_zero, fmpz_is_pm1
-from flint.flintlib.fmpz_types cimport fmpz_mat_struct
+from flint.flintlib.fmpz_types cimport (
+    fmpz_mat_struct,
+    rep_type,
+    gram_type,
+)
 from flint.flintlib.fmpz_poly cimport fmpz_poly_init
 from flint.flintlib.fmpz_mat cimport *
 from flint.flintlib.fmpz_lll cimport *
@@ -633,18 +637,18 @@ cdef class fmpz_mat(flint_mat):
         cdef fmpz_mat u, v
         cdef fmpz_lll_t ctx
         cdef long i
-        cdef int rt
-        cdef int gt
+        cdef rep_type rt
+        cdef gram_type gt
         if rep == "zbasis":
-            rt = 1
+            rt = rep_type.Z_BASIS
         elif rep == "gram":
-            rt = 0
+            rt = rep_type.GRAM
         else:
             raise ValueError("rep must be 'zbasis' or 'gram'")
         if gram == "approx":
-            gt = 0
+            gt = gram_type.APPROX
         elif gram == "exact":
-            gt = 1
+            gt = gram_type.EXACT
         else:
             raise ValueError("gram must be 'approx' or 'exact'")
         fmpz_lll_context_init(ctx, delta, eta, rt, gt)