added mp4 conversion

fusion-energy · Dec 12, 2024 · 33f08cc · 33f08cc
1 parent 6430e22
commit 33f08cc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/examples/spherical_tokamak_from_plasma_animation.py b/examples/spherical_tokamak_from_plasma_animation.py
@@ -1,3 +1,4 @@
+import os
 import paramak
 import cadquery_png_plugin.plugin
 import numpy as np
@@ -166,4 +167,7 @@ def export_reactor_to_png(reactor, file_path):
 for modified_triangularity in [0.55, 0.3667, 0.1833, 0.0, -0.1833, -0.3667, -0.55, -0.3667, -0.1833, 0.0, 0.1833, 0.3667, 0.55]:
     reactor = create_reactor(triangularity=modified_triangularity)
     export_reactor_to_png(reactor, f'spherical_tokamak_frame_{frame:03d}.png')
-    frame += 1
+    frame += 1
+
+
+os.system('ffmpeg -r 10 -i spherical_tokamak_frame_%3d.png -c:v libx264 -r 30 -pix_fmt yuv420p spherical_tokamak_animation.mp4')
diff --git a/examples/tokamak_from_plasma_animation.py b/examples/tokamak_from_plasma_animation.py
@@ -1,5 +1,5 @@
 import paramak
-import imageio
+import io
 import cadquery_png_plugin.plugin
 import numpy as np
 import cadquery as cq
@@ -174,3 +174,5 @@ def export_reactor_to_png(reactor, file_path):
     reactor = create_reactor(triangularity=modified_triangularity)
     export_reactor_to_png(reactor, f'tokamak_frame_{frame:03d}.png')
     frame += 1
+
+os.system('ffmpeg -r 10 -i tokamak_frame_%3d.png -c:v libx264 -r 30 -pix_fmt yuv420p tokamak_animation.mp4')