ENH : add stale flag to EventCollection

tacaswell · tacaswell · commit 895a3f592f60 · 2015-05-01T21:52:41.000-04:00
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -1006,7 +1006,6 @@ class LineCollection(Collection):
     number of segments.
     """
 
-
     def __init__(self, segments,     # Can be None.
                  linewidths=None,
                  colors=None,
@@ -1118,6 +1117,7 @@ def set_segments(self, segments):
             _segments = self._add_offsets(_segments)
 
         self._paths = [mpath.Path(_seg) for _seg in _segments]
+        self.stale = True
 
     set_verts = set_segments  # for compatibility with PolyCollection
     set_paths = set_segments
@@ -1155,9 +1155,11 @@ def set_color(self, c):
         ACCEPTS: matplotlib color arg or sequence of rgba tuples
         """
         self.set_edgecolor(c)
+        self.stale = True
 
     def get_color(self):
         return self._edgecolors
+
     get_colors = get_color  # for compatibility with old versions
 
 
@@ -1332,6 +1334,7 @@ def switch_orientation(self):
             segments[i] = np.fliplr(segment)
         self.set_segments(segments)
         self._is_horizontal = not self.is_horizontal()
+        self.stale = True
 
     def set_orientation(self, orientation=None):
         '''
