testy

1 files changed, 7 insertions, 2 deletions

diff --git a/test-mogrify.py b/test-mogrify.py
index 2c636d9..61f9899 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -104,9 +104,14 @@ if __name__ == '__main__':
 
   def render_video():
     print(opt.render_dir)
-    video_fn = tag + "_mogrify.mp4"
+    if opt.render_frames:
+      frame_fn = "frame_%05d.png"
+      video_fn = tag + "_mog_frame.mp4"
+    else
+      frame_fn = "ren_%05d.png"
+      video_fn = tag + "_mogrify.mp4"
 
-    cmd = ("ffmpeg", "-i", render_dir + "ren_%05d.png", "-y", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", "-s", "456x256", render_dir + video_fn)
+    cmd = ("ffmpeg", "-i", render_dir + frame_fn, "-y", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", "-s", "456x256", render_dir + video_fn)
     process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
     output, error = process.communicate()