[postprocessor] Added oct_vect_getter in generic rules

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))

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