matplotlib Graphics() wrapper

jasongrout commented 15 years ago

This provides an easy way to make a matplotlib image and combine it with other Graphics() objects:

class Matplotlib_Primitive(GraphicPrimitive):
    """
    Primitive class that initializes the
    matrix_plot graphics type 
    """
    def __init__(self, artist, options=None):
        self.artist = artist
        GraphicPrimitive.__init__(self, options)        

    def get_minmax_data(self):
        """
        Returns a dictionary with the bounding box data.

        EXAMPLES:
            sage: m = matrix_plot(matrix([[1,3,5,1],[2,4,5,6],[1,3,5,7]]))[0]
            sage: list(sorted(m.get_minmax_data().items()))
            [('xmax', 4), ('xmin', 0), ('ymax', 3), ('ymin', 0)]

        """
        return dict(zip(['xmin', 'xmax', 'ymax', 'ymin'], self.artist.get_extent()))

    def _allowed_options(self):
        return {}

    def _repr_(self):
        return "MatrixPlot defined by a %s x %s data grid"%(self.xy_array_row, self.xy_array_col)

    def _render_on_subplot(self, subplot):
        subplot.add_artist(self.artist)

def matplotlib_plot(mat):
    from sage.plot.plot import Graphics
    g = Graphics()
    g.add_primitive(Matplotlib_Primitive(mat))
    return g

Example use:

A=random_matrix(RDF,100)
A.numpy()
import pylab
import numpy
B=A.numpy().astype(float)
im = pylab.imshow(B/numpy.max(B),  origin='upper',alpha=0.6)
matplotlib_plot(im)+polygon([[0,10],[10,0],[20,40]])

This just needs to be put in a file in the plot/ directory and an entry added to all.py.

CC: @nilesjohnson

Component: graphics

Keywords: matplotlib

Issue created by migration from https://trac.sagemath.org/ticket/5128

jasongrout commented 15 years ago

comment:1

And the documentation should be updated, of course.

jasongrout commented 15 years ago

comment:2

Something doesn't work in the above patch. I need to figure out how to get the extents of any matplotlib image passed in. Is there an easy way to that information?

jasongrout commented 15 years ago

comment:3

Attachment: trac_5128-matplotlib-plot.patch.gz

This makes things a little better, but not much. Currently, you can add an axes object. Note that the doctests involving hist are wrong.

kcrisman commented 14 years ago

comment:4

What still needs to be done on this? It would be really good to get this functionality in now that #5448 is merged, so that other tickets could use it. Am I correct that currently there is no way to create a Sage graphics object from a mpl one - the process is one direction only?