add an option to save all important files to path_out

snapshotprocessor.py

@@ -9,6 +9,8 @@ import pandas as pd
 from skimage.morphology import medial_axis
 
 import os
+from distutils.file_util import copy_file
+
 from functools import partial
 from scipy.stats import linregress
 
@@ -343,8 +345,17 @@ class SnapshotProcessor(HDF5Container):
 
         self.info = selector.info[self.run][self.num]
         self.namelist = selector.namelist[self.run]
-        self.lbox = self.info["boxlen"]
 
+        # Save important info files
+        if self.params.out.copy_info:
+            info_src = f"{self.path}/output_{self.num:05}/info_{self.num:05}.txt"
+            info_dest = f"{self.path_out}/info/info_{self.num:05}.txt"
+            if os.path.exists(info_src):
+                os.makedirs(os.path.dirname(info_dest), exist_ok=True)
+                copy_file(info_src, info_dest, update=1)
+
+        # Get box length
+        self.lbox = self.info["boxlen"]
         # Get time
         self.time = self.info["time"]
 
@@ -352,7 +363,7 @@ class SnapshotProcessor(HDF5Container):
         self.open()
         self.save.root._v_attrs.dir = os.path.dirname(path)
         self.save.root._v_attrs.run = os.path.basename(path)
-        self.save.root._v_attrs.num = num
+        self.save.root._v_attrs.num = self.num
         self.save.root._v_attrs.lbox = self.lbox
         self.save.root._v_attrs.unit_length = self.info["unit_length"]
         self.save.root._v_attrs.time = self.time