Call init draw when restarting and saving the animation.

jenshnielsen · jenshnielsen · commit f4e52dff1c15 · 2015-12-24T15:28:30.000+01:00
This makes sure that when restarting or saving the animation is cleared. To prevent the
single animation saves _save_seq is always cleared as part of the init function.
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -778,8 +778,7 @@ def save(self, filename, writer=None, fps=None, dpi=None, codec=None,
         with writer.saving(self._fig, filename, dpi):
             for anim in all_anim:
                 # Clear the initial frame
-                if anim._init_func:
-                    anim._init_draw()
+                anim._init_draw()
             for data in zip(*[a.new_saved_frame_seq()
                               for a in all_anim]):
                 for anim, d in zip(all_anim, data):
@@ -998,6 +997,7 @@ def _step(self, *args):
         # back.
         still_going = Animation._step(self, *args)
         if not still_going and self.repeat:
+            self._init_draw()
             self.frame_seq = self.new_frame_seq()
             if self._repeat_delay:
                 self.event_source.remove_callback(self._step)
@@ -1168,11 +1168,13 @@ def _init_draw(self):
         # artists.
         if self._init_func is None:
             self._draw_frame(next(self.new_frame_seq()))
+
         else:
             self._drawn_artists = self._init_func()
             if self._blit:
                 for a in self._drawn_artists:
                     a.set_animated(self._blit)
+        self._save_seq = []
 
     def _draw_frame(self, framedata):
         # Save the data for potential saving of movies.
