Merge pull request #28 from kpj/feature-rich_display

Enable IPython rich display feature

Author: ponnhide

patchworklib/patchworklib.py

@@ -2283,6 +2283,16 @@ def __sub__(self, other):
             else:
                 return vstack(other, self, adjust_width=False)
 
+    def _repr_png_(self):
+        buf = io.BytesIO()
+        self.savefig(buf, "png")
+        return buf.getvalue()
+
+    def _repr_pdf_(self):
+        buf = io.BytesIO()
+        self.savefig(buf, "pdf")
+        return buf.getvalue()
+
 class Brick(axes.Axes): 
     """
 
@@ -2874,6 +2884,16 @@ def __sub__(self, other):
             else:
                 return vstack(other, self, adjust_width=False)
 
+    def _repr_png_(self):
+        buf = io.BytesIO()
+        self.savefig(buf, "png")
+        return buf.getvalue()
+
+    def _repr_pdf_(self):
+        buf = io.BytesIO()
+        self.savefig(buf, "pdf")
+        return buf.getvalue()
+
 class cBrick(matplotlib.projections.polar.PolarAxes): 
     """
 
@@ -3464,6 +3484,16 @@ def __sub__(self, other):
             else:
                 return vstack(other, self, adjust_width=False)
 
+    def _repr_png_(self):
+        buf = io.BytesIO()
+        self.savefig(buf, "png")
+        return buf.getvalue()
+
+    def _repr_pdf_(self):
+        buf = io.BytesIO()
+        self.savefig(buf, "pdf")
+        return buf.getvalue()
+
 class spacer():
     def __init__(self, brick=None, value=1.0):
         self.target = brick