1 files changed, 58 insertions, 1 deletions
diff --git a/Python/pywarpx/WarpX.py b/Python/pywarpx/WarpX.py
index f0a31f3f9..f32dd735c 100644
--- a/Python/pywarpx/WarpX.py
+++ b/Python/pywarpx/WarpX.py
@@ -1,19 +1,65 @@
 from .Bucket import Bucket
+
+from .Constants import constants
+from .Amr import amr
+from .Geometry import geometry
+from .Algo import algo
+from .Langmuirwave import langmuirwave
+from .Interpolation import interpolation
+from .Laser import laser
+from . import Particles
+from .Particles import particles, particles_list
+
+import ctypes
 from ._libwarpx import libwarpx
+from ._libwarpx import amrex_init
 
 class WarpX(Bucket):
     """
     A Python wrapper for the WarpX C++ class
     """
 
+    def create_argv_list(self):
+        argv = []
+        argv += warpx.attrlist()
+        argv += constants.attrlist()
+        argv += amr.attrlist()
+        argv += geometry.attrlist()
+        argv += algo.attrlist()
+        argv += langmuirwave.attrlist()
+        argv += interpolation.attrlist()
+        argv += particles.attrlist()
+        argv += laser.attrlist()
+
+        # --- 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(' '):
+            if pstring in particles_list_names:
+                # --- The species is already included in particles_list
+                continue
+            elif hasattr(Particles, pstring):
+                # --- Add the predefined species to particles_list
+                particles_list.append(getattr(Particles, pstring))
+                particles_list_names.append(pstring)
+            else:
+                raise Exception('Species %s listed in species_names not defined'%pstring)
+
+        for particle in particles_list:
+            argv += particle.attrlist()
+
+        return argv
+
     def init(self):
+        argv = ['warpx'] + self.create_argv_list()
+        amrex_init(argv)
         libwarpx.warpx_init()
 
     def evolve(self, nsteps=-1):
         libwarpx.warpx_evolve(nsteps)
 
-    def finalize(self):
+    def finalize(self, finalize_mpi=1):
         libwarpx.warpx_finalize()
+        libwarpx.amrex_finalize(finalize_mpi)
 
     def getProbLo(self, direction):
         return libwarpx.warpx_getProbLo(direction)
@@ -21,4 +67,15 @@ class WarpX(Bucket):
     def getProbHi(self, direction):
         return libwarpx.warpx_getProbHi(direction)
 
+    def write_inputs(self, filename='inputs', **kw):
+        argv = self.create_argv_list()
+        with open(filename, 'w') as ff:
+
+            for k, v in kw.iteritems():
+                ff.write('{0} = {1}\n'.format(k, v))
+
+            for arg in argv:
+                ff.write('{0}\n'.format(arg))
+
 warpx = WarpX('warpx')
+