ENH : add stale flag to Figure

tacaswell · tacaswell · commit 15cf72f2dd92 · 2015-05-01T21:52:41.000-04:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -426,6 +426,7 @@ def set_tight_layout(self, tight):
             tight = rcParams['figure.autolayout']
         self._tight = bool(tight)
         self._tight_parameters = tight if isinstance(tight, dict) else {}
+        self.stale = True
 
     def autofmt_xdate(self, bottom=0.2, rotation=30, ha='right'):
         """
@@ -465,6 +466,7 @@ def autofmt_xdate(self, bottom=0.2, rotation=30, ha='right'):
 
         if allsubplots:
             self.subplots_adjust(bottom=bottom)
+        self.stale = True
 
     def get_children(self):
         'get a list of artists contained in the figure'
@@ -541,6 +543,7 @@ def suptitle(self, t, **kwargs):
             sup.remove()
         else:
             self._suptitle = sup
+        self.stale = True
         return self._suptitle
 
     def set_canvas(self, canvas):
@@ -550,6 +553,7 @@ def set_canvas(self, canvas):
         ACCEPTS: a FigureCanvas instance
         """
         self.canvas = canvas
+        self.stale = True
 
     def hold(self, b=None):
         """
@@ -652,6 +656,7 @@ def figimage(self, X,
             im.set_clim(vmin, vmax)
         self.images.append(im)
         im._remove_method = lambda h: self.images.remove(h)
+        self.stale = True
         return im
 
     def set_size_inches(self, *args, **kwargs):
@@ -693,6 +698,7 @@ def set_size_inches(self, *args, **kwargs):
             manager = getattr(self.canvas, 'manager', None)
             if manager is not None:
                 manager.resize(int(canvasw), int(canvash))
+        self.stale = True
 
     def get_size_inches(self):
         """
@@ -758,6 +764,7 @@ def set_dpi(self, val):
         ACCEPTS: float
         """
         self.dpi = val
+        self.stale = True
 
     def set_figwidth(self, val):
         """
@@ -766,6 +773,7 @@ def set_figwidth(self, val):
         ACCEPTS: float
         """
         self.bbox_inches.x1 = val
+        self.stale = True
 
     def set_figheight(self, val):
         """
@@ -774,6 +782,7 @@ def set_figheight(self, val):
         ACCEPTS: float
         """
         self.bbox_inches.y1 = val
+        self.stale = True
 
     def set_frameon(self, b):
         """
@@ -782,12 +791,14 @@ def set_frameon(self, b):
         ACCEPTS: boolean
         """
         self.frameon = b
+        self.stale = True
 
     def delaxes(self, a):
         'remove a from the figure and update the current axes'
         self._axstack.remove(a)
         for func in self._axobservers:
             func(self)
+        self.stale = True
 
     def _make_key(self, *args, **kwargs):
         'make a hashable key out of args and kwargs'
@@ -900,6 +911,7 @@ def add_axes(self, *args, **kwargs):
 
         self._axstack.add(key, a)
         self.sca(a)
+        self.stale = True
         return a
 
     @docstring.dedent_interpd
@@ -987,6 +999,7 @@ def add_subplot(self, *args, **kwargs):
 
         self._axstack.add(key, a)
         self.sca(a)
+        self.stale = True
         return a
 
     def clf(self, keep_observers=False):
@@ -1016,6 +1029,7 @@ def clf(self, keep_observers=False):
         if not keep_observers:
             self._axobservers = []
         self._suptitle = None
+        self.stale = True
 
     def clear(self):
         """
@@ -1227,6 +1241,7 @@ def legend(self, handles, labels, *args, **kwargs):
         l = Legend(self, handles, labels, *args, **kwargs)
         self.legends.append(l)
         l._remove_method = lambda h: self.legends.remove(h)
+        self.stale = True
         return l
 
     @docstring.dedent_interpd
@@ -1254,6 +1269,7 @@ def text(self, x, y, s, *args, **kwargs):
         self._set_artist_props(t)
         self.texts.append(t)
         t._remove_method = lambda h: self.texts.remove(h)
+        self.stale = True
         return t
 
     def _set_artist_props(self, a):
@@ -1399,6 +1415,7 @@ def make_active(event):
             self.number = num
 
             plt.draw_if_interactive()
+        self.stale = True
 
     def add_axobserver(self, func):
         'whenever the axes state change, ``func(self)`` will be called'
@@ -1541,6 +1558,7 @@ def colorbar(self, mappable, cax=None, ax=None, use_gridspec=True, **kw):
         cb = cbar.colorbar_factory(cax, mappable, **kw)
 
         self.sca(current_ax)
+        self.stale = True
         return cb
 
     def subplots_adjust(self, *args, **kwargs):
@@ -1569,6 +1587,7 @@ def subplots_adjust(self, *args, **kwargs):
             else:
                 ax.update_params()
                 ax.set_position(ax.figbox)
+        self.stale = True
 
     def ginput(self, n=1, timeout=30, show_clicks=True, mouse_add=1,
                mouse_pop=3, mouse_stop=2):
