Model.plot_movie(**kwargs): Pass to png_to_gif()

adtzlr · May 3, 2023 · dd1adea · dd1adea
1 parent 23badbd
commit dd1adea
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/src/trusspy/model.py b/src/trusspy/model.py
@@ -708,6 +708,7 @@ def plot_movie(
         nodesize=10,
         cbar_limits="auto",
         incs="all",
+        **kwargs,
     ):
         p_movie(
             self,
@@ -719,6 +720,7 @@ def plot_movie(
             nodesize,
             cbar_limits,
             incs,
+            **kwargs,
         )
 
     def plot_history(

diff --git a/src/trusspy/tools/movie_generator.py b/src/trusspy/tools/movie_generator.py
@@ -11,7 +11,7 @@
 import imageio
 
 
-def png_to_gif(workdir=r"figures/", png_subdir=r"png/", gif_subdir=r"gif/"):
+def png_to_gif(workdir=r"figures/", png_subdir=r"png/", gif_subdir=r"gif/", **kwargs):
     png_dir = workdir + png_subdir
     gif_dir = workdir + gif_subdir
     images = []
@@ -25,5 +25,5 @@ def png_to_gif(workdir=r"figures/", png_subdir=r"png/", gif_subdir=r"gif/"):
             file_path = os.path.join(png_dir, file_name)
             images.append(imageio.imread(file_path))
     imageio.mimwrite(
-        gif_dir + r"movie.gif", images, duration=5 / 1000, subrectangles=True
+        gif_dir + r"movie.gif", images, subrectangles=True, **kwargs
     )
diff --git a/src/trusspy/tools/plot_utilities.py b/src/trusspy/tools/plot_utilities.py
@@ -218,6 +218,7 @@ def p_movie(
     nodesize=10,
     cbar_limits="auto",
     incs="all",
+    **kwargs,
 ):
     if incs == "all":
         if self.Settings.nsteps > 1:
@@ -244,7 +245,7 @@ def p_movie(
         plt.savefig("figures/png/fig_{:03d}.png".format(i), dpi=200)
         plt.close("all")
 
-    png_to_gif()
+    png_to_gif(**kwargs)
 
 
 def p_path(self, nodepath, increment=-1, Y="Displacement X", fig=None, ax=None):