Ticket #9542: trac_9542-znpoly_timing.patch

sage/misc/all.py

@@ -13 +13 @@
 
 from html import html
 
-from sage_timeit_class import timeit
+from sage_timeit_class import timeit, test_timing
 
 from edit_module import edit, set_edit_template
 

sage/misc/sage_timeit_class.pyx

@@ -82 +82 @@
         print self.eval(code, globals, preparse=preparse, **kwds)
 
 timeit = SageTimeit()
+
+
+
+
+from misc import cputime, walltime
+import preparser
+def test_timing(code, max_cputime, max_walltime=0, loops=1):
+    G = globals()
+    code0 = code; code = preparser.preparse(code)
+    t = cputime(); w = walltime()
+    for i in range(loops):
+        eval(code, G)
+    if max_cputime>0 and cputime(t) > max_cputime:
+        print "CPU time (%s) exceeded evaluating '%s' %s times"%(max_cputime, code0, loops)
+    if max_walltime>0 and walltime(w)  > max_walltime:
+        print "WALL time (%s) exceeded evaluating '%s' %s times"%(max_walltime, code0, loops)
+    
         
 

sage/rings/polynomial/polynomial_template.pxi

@@ -198 +198 @@
         cdef cparent _parent = get_cparent((<Polynomial_template>self)._parent)
         return [self[i] for i in range(celement_len(&self.x, _parent))]
 
-    def __dealloc__(self):
+    def __del__(self):
         """
         EXAMPLE::
 

sage/rings/polynomial/polynomial_zmod_flint.pyx

@@ -37 +37 @@
 from sage.structure.factorization import Factorization
 from sage.structure.element import coerce_binop
 
+from sage.structure.parent cimport Parent
+
 # We need to define this stuff before including the templating stuff
 # to make sure the function get_cparent is found since it is used in
 # 'polynomial_template.pxi'.
@@ -99 +101 @@
                 pass
         Polynomial_template.__init__(self, parent, x, check, is_gen, construct)
 
+    cpdef RingElement _mul_(self, RingElement right):
+        """
+        TESTS::
+
+            sage: R.<x> = Integers(2^6)[]
+            sage: f = R([1,2,5,-9]); g = R([1,2,3,4])
+            sage: f*g
+            4060*x^6 + 4089*x^5 + 5*x^4 + 11*x^3 + 12*x^2 + 4*x + 1
+            sage: test_timing('f*g', 1)      # optional -- timing
+            True
+        """
+        cdef Polynomial_zmod_flint r = PY_NEW(Polynomial_zmod_flint)
+        r._parent = self._parent
+        zmod_poly_mul(&r.x, &(<Polynomial_template>self).x, &(<Polynomial_template>right).x)
+        return r
+
+    cpdef ModuleElement _add_(self, ModuleElement right):
+        cdef Polynomial_zmod_flint r = PY_NEW(Polynomial_zmod_flint)
+        r._parent = self._parent
+        zmod_poly_add(&r.x, &(<Polynomial_template>self).x, &(<Polynomial_template>right).x)
+        return r
+    
+    cpdef ModuleElement _sub_(self, ModuleElement right):
+        cdef Polynomial_zmod_flint r = PY_NEW(Polynomial_zmod_flint)
+        r._parent = self._parent
+        zmod_poly_sub(&r.x, &(<Polynomial_template>self).x, &(<Polynomial_template>right).x)
+        return r
+
+    def __dealloc__(self):
+        zmod_poly_clear(&self.x)
+
     cdef Polynomial_template _new(self):
         """
         EXAMPLES::