diff --git a/baseprocessor.py b/baseprocessor.py index 081af3a..9f8b8b1 100644 --- a/baseprocessor.py +++ b/baseprocessor.py @@ -457,5 +457,9 @@ def vect_getter(name, i, dset): return dset[name][:, i] +def oct_vect_getter(name, i, dset): + return dset[name][:, :, i] + + def norm_getter(name, dset): return np.sqrt(np.sum(dset[name] ** 2, axis=1)) diff --git a/postprocessor.py b/postprocessor.py index 61a2882..e9d5943 100644 --- a/postprocessor.py +++ b/postprocessor.py @@ -32,7 +32,14 @@ import pspec_new from units import U -from baseprocessor import HDF5Container, Rule, norm_getter, simple_getter, vect_getter +from baseprocessor import ( + HDF5Container, + Rule, + norm_getter, + simple_getter, + vect_getter, + oct_vect_getter, +) # Getters @@ -1706,6 +1713,7 @@ class PostProcessor(HDF5Container): field + dir, partial(vect_getter, field, i), self.unit_key[field], + oct_getter=partial(oct_vect_getter, field, i), ) # Radial