API Brain.save_movie(): more parameters for imageio

Author: christianbrodbeck

surfer/viz.py

@@ -2161,8 +2161,8 @@ def save_montage(self, filename, order=['lat', 'ven', 'med'],
         return out
 
     def save_movie(self, fname, time_dilation=4., tmin=None, tmax=None,
-                   framerate=24, interpolation='quadratic', codec='mpeg4',
-                   bitrate='1M'):
+                   framerate=24, interpolation='quadratic', codec=None,
+                   bitrate=None, **kwargs):
         """Save a movie (for data with a time axis)
 
         .. Warning::
@@ -2188,11 +2188,6 @@ def save_movie(self, fname, time_dilation=4., tmin=None, tmax=None,
             Interpolation method (``scipy.interpolate.interp1d`` parameter,
             one of 'linear' | 'nearest' | 'zero' | 'slinear' | 'quadratic' |
             'cubic', default 'quadratic').
-        codec : str
-            Codec to use with ffmpeg (default 'mpeg4').
-        bitrate : str | float
-            Bitrate to use to encode movie. Can be specified as number (e.g.
-            64000) or string (e.g. '64k'). Default value is 1M
 
         Notes
         -----
@@ -2211,6 +2206,15 @@ def save_movie(self, fname, time_dilation=4., tmin=None, tmax=None,
                               "run\n\n    $ pip install -U "
                               "pysurfer[save_movie]\n")
 
+        # find imageio FFMPEG parameters
+        if 'fps' not in kwargs:
+            kwargs['fps'] = framerate
+        if codec is not None:
+            kwargs['codec'] = codec
+        if bitrate is not None:
+            kwargs['bitrate'] = bitrate
+
+        # find tmin
         if tmin is None:
             tmin = self._times[0]
         elif tmin < self._times[0]:
@@ -2238,8 +2242,7 @@ def save_movie(self, fname, time_dilation=4., tmin=None, tmax=None,
                      % (times, time_idx))
         images = (self.screenshot() for _ in
                   self._iter_time(time_idx, interpolation))
-        imageio.mimwrite(fname, images, fps=framerate, codec=codec,
-                         bitrate=bitrate)
+        imageio.mimwrite(fname, images, **kwargs)
 
     def animate(self, views, n_steps=180., fname=None, use_cache=False,
                 row=-1, col=-1):