added image capability

Author: lionel42

images/lenna.jpg

@@ -28,6 +28,7 @@
     ...
     plt.savefig("figure.xyz")
 """
+import numpy as np
 from matplotlib.transforms import Affine2D
 import pygame
 from matplotlib._pylab_helpers import Gcf
@@ -143,7 +144,10 @@ def draw_path(self, gc, path, transform, rgbFace=None):
 #         pass
 
     def draw_image(self, gc, x, y, im):
-        pass
+        print(type(im))
+        print(x, y)
+        img_surf = pygame.image.frombuffer(np.ascontiguousarray(np.flip(im, axis=0)), (im.shape[1], im.shape[0]), 'RGBA')
+        self.surface.blit(img_surf, (x, self.surface.get_height() - y - im.shape[0]))
 
     def draw_text(self, gc, x, y, s, prop, angle, ismath=False, mtext=None):
         # make sure font module is initialized
@@ -167,7 +171,7 @@ def draw_text(self, gc, x, y, s, prop, angle, ismath=False, mtext=None):
 
     def flipy(self):
         # docstring inherited
-        return True
+        return False
 
     def get_canvas_width_height(self):
         # docstring inherited
@@ -327,7 +331,10 @@ class FigureManagerTemplate(FigureManagerBase):
     def show(self):
         # do something to display the GUI
         pygame.init()
-        main_display = pygame.display.set_mode(self.canvas.figure.get_size())
+        main_display = pygame.display.set_mode(
+            self.canvas.figure.get_size(),  # Size matches figure size
+            pygame.RESIZABLE   # Allow resizing
+        )
 
         FPS = 144
         FramePerSec = pygame.time.Clock()

images/long_dog.jpg

@@ -0,0 +1,19 @@
+import sys
+import os
+from numpy.ma.core import size
+import pygame
+
+import matplotlib
+#matplotlib.use('module://pygame_matplotlib.backend_pygame')
+matplotlib.use('Qt4Agg')
+
+import matplotlib.pyplot as plt
+import matplotlib.figure as fg
+
+
+plt.plot([1,2], [1,2], color='green')
+plt.text(1.5, 1.5, '2', size=50)
+plt.xlabel('swag')
+
+
+plt.show()

images/qt_agg.png

@@ -5,16 +5,19 @@
 
 import matplotlib
 matplotlib.use('module://pygame_matplotlib.backend_pygame')
-#matplotlib.use('Qt4Agg')
+# matplotlib.use('Qt4Agg')
 
 import matplotlib.pyplot as plt
 import matplotlib.figure as fg
 
-fig, ax = plt.subplots(figsize=(16, 9))
+import matplotlib.image as mpimg
+
+fig, axes = plt.subplots(2,2,figsize=(16, 9))
 print(type(fig))
-ax.plot([1,2], [1,2], color='green')
-ax.text(1.5, 1.5, '2', size=50)
-ax.set_xlabel('swag')
+axes[0, 0].plot([1,2], [1,2], color='green')
+axes[0, 1].text(0.5, 0.5, '2', size=50)
+axes[1, 0].set_xlabel('swag')
+axes[1, 1].imshow(mpimg.imread('images' + os.sep + 'long_dog.jpg'))
 
 
 plt.show()