diff options
Diffstat (limited to 'Python/pywarpx/WarpX.py')
-rw-r--r-- | Python/pywarpx/WarpX.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/Python/pywarpx/WarpX.py b/Python/pywarpx/WarpX.py index 175917ab6..517c74d42 100644 --- a/Python/pywarpx/WarpX.py +++ b/Python/pywarpx/WarpX.py @@ -14,7 +14,7 @@ from .Boundary import boundary from .Bucket import Bucket from .Collisions import collisions, collisions_list from .Constants import my_constants -from .Diagnostics import diagnostics +from .Diagnostics import diagnostics, reduced_diagnostics from .EB2 import eb2 from .Geometry import geometry from .Interpolation import interpolation @@ -76,6 +76,14 @@ class WarpX(Bucket): for species_diagnostic in diagnostic._species_dict.values(): argv += species_diagnostic.attrlist() + reduced_diagnostics.reduced_diags_names = reduced_diagnostics._diagnostics_dict.keys() + argv += reduced_diagnostics.attrlist() + for diagnostic in reduced_diagnostics._diagnostics_dict.values(): + diagnostic.species = diagnostic._species_dict.keys() + argv += diagnostic.attrlist() + for species_diagnostic in diagnostic._species_dict.values(): + argv += species_diagnostic.attrlist() + return argv def init(self, mpi_comm=None): |