Change of routines for MPI with dz array

Author: Antoniahuber

wakis/routines.py

@@ -248,6 +248,7 @@ def wakesolve(self, wakelength,
         self.v = self.wake.v
         if self.use_mpi:  #E- should it be zmin, zmax instead?
             z = self.Z # use global coords
+            dz = np.diff(self.Z)
             zz = slice(0, self.NZ)
         else:
             z = self.z
@@ -271,13 +272,13 @@ def wakesolve(self, wakelength,
             if self.rank==0:
                 hf = h5py.File(self.Ez_file, 'w')
                 hf['x'], hf['y'], hf['z'] = self.x[xx], self.y[yy], z[zz]
-                hf['dx'], hf['dy'], hf['dz'] = self.grid.dx, self.grid.dy, self.grid.dz
+                hf['dx'], hf['dy'], hf['dz'] = self.grid.dx, self.grid.dy, dz
                 hf['t'] = np.arange(0, Nt*self.dt, self.dt)
 
                 if save_J:
                     hfJ = h5py.File('Jz.h5', 'w')
                     hfJ['x'], hfJ['y'], hfJ['z'] = self.x[xx], self.y[yy], z[zz]
-                    hfJ['dx'], hfJ['dy'], hfJ['dz'] = self.grid.dx, self.grid.dy, self.grid.dz
+                    hfJ['dx'], hfJ['dy'], hfJ['dz'] = self.grid.dx, self.grid.dy, dz
                     hfJ['t'] = np.arange(0, Nt*self.dt, self.dt)
         else:
             hf = h5py.File(self.Ez_file, 'w')