use repr_png and repr_jpeg hooks rather than repr_html

these will work in the ipython qtconsole as well!
This commit is contained in:
wilsaj
2012-07-20 16:44:05 -05:00
parent 70cf6cfba0
commit 32d3e1b161
+7 -3
View File
@@ -56,11 +56,15 @@ class Image(np.ndarray):
object_state[2] = (object_state[2], subclass_state)
return tuple(object_state)
def _repr_html_(self):
def _repr_png_(self):
str_buffer = StringIO.StringIO()
imsave(str_buffer, self, format_str='png')
base64_str = base64.b64encode(str_buffer.getvalue())
return '<img src="data:image/png;base64,%s">' % base64_str
return str_buffer.getvalue()
def _repr_jpeg_(self):
str_buffer = StringIO.StringIO()
imsave(str_buffer, self, format_str='jpeg')
return str_buffer.getvalue()
def __setstate__(self, state):
nd_state, subclass_state = state