Fix animation frames ordering

CExA-project · Jan 21, 2025 · 5b69600 · 5b69600
1 parent eb8dd5b
commit 5b69600
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/projects/01_wave/python/animate.py b/projects/01_wave/python/animate.py
@@ -11,6 +11,7 @@
 from matplotlib import *
 from matplotlib.pyplot import *
 import matplotlib.animation as animation
+import re
 
 # ______________________________________________________________________________
 # RCParams - personalize the figure output
@@ -50,7 +51,8 @@
 
 file_list = glob.glob("{}/grid_*.json".format(file_path))
 
-file_list = sorted(file_list)
+iter_number_re = re.compile(r"{}/grid_(\d+).json".format(file_path))
+file_list = sorted(file_list, key=lambda p: int(iter_number_re.match(p).group(1)))
 
 # ______________________________________________________________________________
 # Animation function

diff --git a/projects/01_wave/python/animate_3d.py b/projects/01_wave/python/animate_3d.py
@@ -12,6 +12,7 @@
 from matplotlib.pyplot import *
 import matplotlib.animation as animation
 from mpl_toolkits.mplot3d import Axes3D
+import re
 
 # ______________________________________________________________________________
 # RCParams - personalize the figure output
@@ -51,7 +52,8 @@
 
 file_list = glob.glob("{}/grid_*.json".format(file_path))
 
-file_list = sorted(file_list)
+iter_number_re = re.compile(r"{}/grid_(\d+).json".format(file_path))
+file_list = sorted(file_list, key=lambda p: int(iter_number_re.match(p).group(1)))
 
 # ______________________________________________________________________________
 # Animation function