diff options
author | 2020-01-14 13:13:03 -0800 | |
---|---|---|
committer | 2020-01-14 13:13:03 -0800 | |
commit | ca070b083c4fa5cc2db2428da801b3b55501d60c (patch) | |
tree | 9ad474eb05655cb6ffff78cd73ddbbcf3105b49b /Python/pywarpx/WarpX.py | |
parent | 8e4a42b96fdc3c27730262b49db387205dd4c3af (diff) | |
parent | 727ac478c1e5a7e2cc1217ed64e4605228b95795 (diff) | |
download | WarpX-ca070b083c4fa5cc2db2428da801b3b55501d60c.tar.gz WarpX-ca070b083c4fa5cc2db2428da801b3b55501d60c.tar.zst WarpX-ca070b083c4fa5cc2db2428da801b3b55501d60c.zip |
Merge branch 'dev' of https://github.com/ECP-WarpX/WarpX into ParticlesEBParser
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 5f7477e58..2ef3d99c1 100644 --- a/Python/pywarpx/WarpX.py +++ b/Python/pywarpx/WarpX.py @@ -29,7 +29,7 @@ class WarpX(Bucket): # --- Search through species_names and add any predefined particle objects in the list. particles_list_names = [p.instancename for p in particles_list] - for pstring in particles.species_names.split(' '): + for pstring in particles.species_names: if pstring in particles_list_names: # --- The species is already included in particles_list continue @@ -48,6 +48,14 @@ class WarpX(Bucket): return argv + def add_field_to_plot(self, field): + try: + # Check if the field was already added to the string + if field not in self.fields_to_plot: + self.fields_to_plot.append(field) + except AttributeError: + self.fields_to_plot = [field] + def init(self): from . import wx argv = ['warpx'] + self.create_argv_list() |