making it possible to use writer.setup kwargs from animation.save

fredrik-1 · fredrik-1 · commit dc30bb4436d2 · 2018-08-16T11:20:25.000+02:00
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -1005,7 +1005,7 @@ def _stop(self, *args):
 
     def save(self, filename, writer=None, fps=None, dpi=None, codec=None,
              bitrate=None, extra_args=None, metadata=None, extra_anim=None,
-             savefig_kwargs=None):
+             savefig_kwargs=None, **kwargs):
         '''Saves a movie file by drawing every frame.
 
         Parameters
@@ -1060,6 +1060,9 @@ class to use, such as 'ffmpeg'. If ``None``, defaults to
            on to the `savefig` command which is called repeatedly to
            save the individual frames.
 
+        **kwargs :
+            Additional kwargs sendt to the writer's setup method.
+
         Notes
         -----
         fps, codec, bitrate, extra_args, metadata are used to
@@ -1155,7 +1158,7 @@ class to use, such as 'ffmpeg'. If ``None``, defaults to
                           "frame size to vary, which is inappropriate for "
                           "animation.")
                 rcParams['savefig.bbox'] = None
-            with writer.saving(self._fig, filename, dpi):
+            with writer.saving(self._fig, filename, dpi, **kwargs):
                 for anim in all_anim:
                     # Clear the initial frame
                     anim._init_draw()
