ENH : add stale flag to *Image* classes

tacaswell · tacaswell · commit 0d7640b32531 · 2015-05-01T21:52:41.000-04:00
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -375,6 +375,7 @@ def draw(self, renderer, *args, **kwargs):
             im._gid = self.get_gid()
             renderer.draw_image(gc, l, b, im)
         gc.restore()
+        self.stale = False
 
     def contains(self, mouseevent):
         """
@@ -437,6 +438,7 @@ def set_data(self, A):
         self._rgbacache = None
         self._oldxslice = None
         self._oldyslice = None
+        self.stale = True
 
     def set_array(self, A):
         """
@@ -481,6 +483,7 @@ def set_interpolation(self, s):
         if s not in self._interpd:
             raise ValueError('Illegal interpolation string')
         self._interpolation = s
+        self.stale = True
 
     def set_resample(self, v):
         """
@@ -491,6 +494,7 @@ def set_resample(self, v):
         if v is None:
             v = rcParams['image.resample']
         self._resample = v
+        self.stale = True
 
     def get_resample(self):
         """Return the image resample boolean"""
@@ -508,6 +512,8 @@ def set_filternorm(self, filternorm):
         else:
             self._filternorm = 0
 
+        self.stale = True
+
     def get_filternorm(self):
         """Return the filternorm setting"""
         return self._filternorm
@@ -523,6 +529,7 @@ def set_filterrad(self, filterrad):
         if r <= 0:
             raise ValueError("The filter radius must be a positive number")
         self._filterrad = r
+        self.stale = True
 
     def get_filterrad(self):
         """return the filterrad setting"""
@@ -671,6 +678,7 @@ def set_extent(self, extent):
             self.axes.set_xlim((xmin, xmax), auto=None)
         if self.axes._autoscaleYon:
             self.axes.set_ylim((ymin, ymax), auto=None)
+        self.stale = True
 
     def get_extent(self):
         """Get the image extent: left, right, bottom, top"""
@@ -778,6 +786,7 @@ def set_data(self, x, y, A):
         # accessed - JDH 3/3/2010
         self._oldxslice = None
         self._oldyslice = None
+        self.stale = True
 
     def set_array(self, *args):
         raise NotImplementedError('Method not supported')
@@ -890,6 +899,7 @@ def draw(self, renderer, *args, **kwargs):
                             round(self.axes.bbox.ymin),
                             im)
         gc.restore()
+        self.stale = False
 
     def set_data(self, x, y, A):
         A = cbook.safe_masked_invalid(A)
@@ -923,6 +933,7 @@ def set_data(self, x, y, A):
         self._Ax = x
         self._Ay = y
         self._rgbacache = None
+        self.stale = True
 
     def set_array(self, *args):
         raise NotImplementedError('Method not supported')
@@ -1046,6 +1057,7 @@ def draw(self, renderer, *args, **kwargs):
         gc.set_alpha(self.get_alpha())
         renderer.draw_image(gc, round(self.ox), round(self.oy), im)
         gc.restore()
+        self.stale = False
 
     def write_png(self, fname):
         """Write the image to png file with fname"""
@@ -1197,6 +1209,7 @@ def draw(self, renderer, *args, **kwargs):
         b = np.min([y0, y1])
         renderer.draw_image(gc, round(l), round(b), im)
         gc.restore()
+        self.stale = True
 
 
 def imread(fname, format=None):
