Merge pull request #18901 from anntzer/repeat_delay

QuLogic · web-flow · commit a129e6ec2c7e · 2020-11-09T23:13:56.000-05:00
Simplify repeat_delay and fix support for it when using iterable frames.
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -1445,27 +1445,12 @@ def _step(self, *args):
         if not still_going and self.repeat:
             self._init_draw()
             self.frame_seq = self.new_frame_seq()
-            self.event_source.remove_callback(self._step)
-            self.event_source.add_callback(self._loop_delay)
             self.event_source.interval = self._repeat_delay
             return True
         else:
+            self.event_source.interval = self._interval
             return still_going
 
-    def _stop(self, *args):
-        # If we stop in the middle of a loop delay (which is relatively likely
-        # given the potential pause here), remove the loop_delay callback as
-        # well.
-        self.event_source.remove_callback(self._loop_delay)
-        super()._stop()
-
-    def _loop_delay(self, *args):
-        # Reset the interval and change callbacks after the delay.
-        self.event_source.remove_callback(self._loop_delay)
-        self.event_source.interval = self._interval
-        self.event_source.add_callback(self._step)
-        Animation._step(self)
-
 
 class ArtistAnimation(TimedAnimation):
     """
@@ -1657,10 +1642,10 @@ def __init__(self, fig, func, frames=None, init_func=None, fargs=None,
             self._iter_gen = frames
         elif np.iterable(frames):
             if kwargs.get('repeat', True):
+                self._tee_from = frames
                 def iter_frames(frames=frames):
-                    while True:
-                        this, frames = itertools.tee(frames, 2)
-                        yield from this
+                    this, self._tee_from = itertools.tee(self._tee_from, 2)
+                    yield from this
                 self._iter_gen = iter_frames
             else:
                 self._iter_gen = lambda: iter(frames)
