Fixing Matrix/overbar printing for sympy.

IPython/extensions/sympy_printing.py

@@ -1,60 +1,65 b''
 """A print function that pretty prints sympy Basic objects.
 
 Authors:
 * Brian Granger
 """
 #-----------------------------------------------------------------------------
 #  Copyright (C) 2008-2011  The IPython Development Team
 #
 #  Distributed under the terms of the BSD License.  The full license is in
 #  the file COPYING, distributed as part of this software.
 #-----------------------------------------------------------------------------
 
 #-----------------------------------------------------------------------------
 # Imports
 #-----------------------------------------------------------------------------
 
 from IPython.lib.latextools import latex_to_png
 
 from sympy import pretty, latex
 
 #-----------------------------------------------------------------------------
 # Definitions of magic functions for use with IPython
 #-----------------------------------------------------------------------------
 
 def print_basic_unicode(o, p, cycle):
     """A function to pretty print sympy Basic objects."""
     if cycle:
         return p.text('Basic(...)')
     out = pretty(o, use_unicode=True)
     if '\n' in out:
         p.text(u'\n')
     p.text(out)
 
 
 def print_png(o):
     """A funciton to display sympy expression using LaTex -> PNG."""
     s = latex(o, mode='inline')
-    # mathtext does not understand \\operatorname to we remove it so functions
-    # like sin, cos can print. We should possible replace it with mathrm.
+    # mathtext does not understand certain latex flags, so we try to replace
+    # them with suitable subs.
     s = s.replace('\\operatorname','')
+    s = s.replace('\\overline', '\\bar')
     png = latex_to_png(s, encode=True)
     return png
 
 _loaded = False
 
 
 def load_ipython_extension(ip):
     """Load the extension in IPython."""
     global _loaded
     if not _loaded:
         plaintext_formatter = ip.display_formatter.formatters['text/plain']
         plaintext_formatter.for_type_by_name(
             'sympy.core.basic', 'Basic', print_basic_unicode
         )
+        plaintext_formatter.for_type_by_name(
+            'sympy.matrices.matrices', 'Matrix', print_basic_unicode
+        )
+
         png_formatter = ip.display_formatter.formatters['image/png']
         png_formatter.for_type_by_name(
             'sympy.core.basic', 'Basic', print_png
         )
         _loaded = True