added resize parameter to plot 2d-arrays using figimage

mehdimu · mehdimu · commit ab8e5cf52141 · 2015-04-05T22:09:49.000-04:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -568,6 +568,7 @@ def hold(self, b=None):
             self._hold = b
 
     def figimage(self, X,
+                 resize=False,
                  xo=0,
                  yo=0,
                  alpha=None,
@@ -603,6 +604,8 @@ def figimage(self, X,
           =========   =========================================================
           Keyword     Description
           =========   =========================================================
+          resize      a boolean, True or False. If "True", then re-size the
+                      Figure to match the given image size.
           xo or yo    An integer, the *x* and *y* image offset in pixels
           cmap        a :class:`matplotlib.colors.Colormap` instance, e.g.,
                       cm.jet. If *None*, default to the rc ``image.cmap``
@@ -637,6 +640,11 @@ def figimage(self, X,
         if not self._hold:
             self.clf()
 
+        if resize:
+            dpi = self.get_dpi()
+            figsize = [x / float(dpi) for x in (X.shape[1], X.shape[0])]
+            self.set_size_inches(figsize, forward=True)
+
         im = FigureImage(self, cmap, norm, xo, yo, origin, **kwargs)
         im.set_array(X)
         im.set_alpha(alpha)
