Merge pull request #32 from chowland/plane_mean_read

Add reading of mean planes to afidtools

Makefile

@@ -1,6 +1,6 @@
 # Choose the machine being used
 # Options: PC_GNU, PC_INTEL, (i)SNELLIUS, IRENE, MARENOSTRUM, SUPERMUC
-MACHINE=IRENE_SKL
+MACHINE=PC_GNU
 # Modules required for each HPC system as follows:
 # SNELLIUS: 2021 foss/2021a HDF5/1.10.7-gompi-2021a
 # iSNELLIUS: 2021 intel/2021a FFTW/3.3.9-intel-2021a HDF5/1.10.7-iimpi-2021a

tools/afidtools/afidtools.py

@@ -148,6 +148,18 @@ def read_cut(folder, var, idx, plane):
         A = np.array(f[varname][()])
     return A
 
+def read_plane_mean(folder, var, idx, plane):
+    """
+    Returns a 2D plane of sample number `idx` of the variable `var`.
+    `plane` can be 'y' or 'z' and specifies which dimension
+    is averaged over to obtain the plane. `folder` specifies the root directory
+    of the simulation.
+    """
+    varname = var+"/"+"%05d" % idx
+    with h5py.File(folder+"/outputdir/flowmov/movie_"+plane+"mean.h5","r") as f:
+        A = np.array(f[varname][()])
+    return A
+
 def continua_master_from_input(folder, time=0.0):
     """
     This function generates the file `continua_master.h5` needed to run