Updated handling of fields_to_plot in picmi interface

1 files changed, 8 insertions, 0 deletions

diff --git a/Python/pywarpx/WarpX.py b/Python/pywarpx/WarpX.py
index 5f7477e58..1b536a55c 100644
--- a/Python/pywarpx/WarpX.py
+++ b/Python/pywarpx/WarpX.py
@@ -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()