Wrap cse

Author: isuruf

symengine/lib/symengine.pxd

@@ -219,6 +219,7 @@ cdef extern from "<symengine/basic.h>" namespace "SymEngine":
     ctypedef unordered_map[int, rcp_const_basic].iterator umap_int_basic_iterator "SymEngine::umap_int_basic::iterator"
     ctypedef unordered_map[rcp_const_basic, rcp_const_number] umap_basic_num "SymEngine::umap_basic_num"
     ctypedef unordered_map[rcp_const_basic, rcp_const_number].iterator umap_basic_num_iterator "SymEngine::umap_basic_num::iterator"
+    ctypedef vector[pair[rcp_const_basic, rcp_const_basic]] vec_pair "SymEngine::vec_pair"
 
     bool eq(const Basic &a, const Basic &b) nogil except +
     bool neq(const Basic &a, const Basic &b) nogil except +
@@ -312,6 +313,7 @@ cdef extern from "<symengine/basic.h>" namespace "SymEngine":
     bool is_a_Xor "SymEngine::is_a<SymEngine::Xor>"(const Basic &b) nogil
     RCP[const Basic] expand(RCP[const Basic] &o, bool deep) nogil except +
     void as_numer_denom(RCP[const Basic] &x, const Ptr[RCP[Basic]] &numer, const Ptr[RCP[Basic]] &denom) nogil
+    void cse(vec_pair &replacements, vec_basic &reduced_exprs, const vec_basic &exprs) nogil except +
 
 cdef extern from "<symengine/subs.h>" namespace "SymEngine":
     RCP[const Basic] msubs (RCP[const Basic] &x, const map_basic_basic &x) nogil

symengine/lib/symengine_wrapper.pyx

@@ -1,6 +1,9 @@
 from cython.operator cimport dereference as deref, preincrement as inc
 cimport symengine
-from symengine cimport RCP, pair, map_basic_basic, umap_int_basic, umap_int_basic_iterator, umap_basic_num, umap_basic_num_iterator, rcp_const_basic, std_pair_short_rcp_const_basic, rcp_const_seriescoeffinterface
+from symengine cimport (RCP, pair, map_basic_basic, umap_int_basic,
+    umap_int_basic_iterator, umap_basic_num, umap_basic_num_iterator,
+    rcp_const_basic, std_pair_short_rcp_const_basic,
+    rcp_const_seriescoeffinterface)
 from libcpp cimport bool as cppbool
 from libcpp.string cimport string
 from libcpp.vector cimport vector
@@ -751,10 +754,24 @@ def get_dict(*args):
 
 
 cdef tuple vec_basic_to_tuple(symengine.vec_basic& vec):
+    return tuple(vec_basic_to_list(vec))
+
+
+cdef list vec_basic_to_list(symengine.vec_basic& vec):
     result = []
     for i in range(vec.size()):
         result.append(c2py(<RCP[const symengine.Basic]>(vec[i])))
-    return tuple(result)
+    return result
+
+
+cdef list vec_pair_to_list(symengine.vec_pair& vec):
+    result = []
+    cdef RCP[const symengine.Basic] a, b
+    for i in range(vec.size()):
+        a = <RCP[const symengine.Basic]>vec[i].first
+        b = <RCP[const symengine.Basic]>vec[i].second
+        result.append((c2py(a), c2py(b)))
+    return result
 
 
 cdef class Basic(object):
@@ -4789,5 +4806,17 @@ def solve(f, sym, domain=None):
     return c2py(<RCP[const symengine.Basic]>(symengine.solve(f_.thisptr, x, d)))
 
 
+def cse(exprs):
+    cdef symengine.vec_basic vec
+    cdef symengine.vec_pair replacements
+    cdef symengine.vec_basic reduced_exprs
+    cdef Basic b
+    for expr in exprs:
+        b = sympify(expr)
+        vec.push_back(b.thisptr)
+    symengine.cse(replacements, reduced_exprs, vec)
+    return (vec_pair_to_list(replacements), vec_basic_to_list(reduced_exprs))
+
+
 # Turn on nice stacktraces:
 symengine.print_stack_on_segfault()