Add init_printing

Author: isuruf

symengine/__init__.py

@@ -19,6 +19,7 @@
 )
 from .utilities import var, symbols
 from .functions import *
+from .printing import init_printing
 
 if have_mpfr:
     from .lib.symengine_wrapper import RealMPFR

symengine/lib/symengine_wrapper.pyx

@@ -774,6 +774,8 @@ cdef list vec_pair_to_list(symengine.vec_pair& vec):
     return result
 
 
+repr_latex=[False]
+
 cdef class Basic(object):
 
     def __str__(self):
@@ -783,9 +785,9 @@ cdef class Basic(object):
         return self.__str__()
 
     def _repr_latex_(self):
-        try:
+        if repr_latex[0]:
             return "${}$".format(latex(self))
-        except:
+        else:
             return None
 
     def __hash__(self):

symengine/printing.py

@@ -1,4 +1,5 @@
 from symengine.lib.symengine_wrapper import ccode, sympify, Basic
+import symengine.lib.symengine_wrapper
 
 class CCodePrinter:
 
@@ -22,3 +23,12 @@ def doprint(self, expr, assign_to=None):
             code_line = '{}[{}] = {};'.format(assign_to, i, element)
             code_lines.append(code_line)
         return '\n'.join(code_lines)
+
+
+def init_printing(pretty_print=True, use_latex=True):
+    if pretty_print:
+        if not use_latex:
+            raise RuntimeError("Only latex is supported for pretty printing")
+        symengine.lib.symengine_wrapper.repr_latex[0] = True
+    else:
+        symengine.lib.symengine_wrapper.repr_latex[0] = False