Parallelize 10_streaming_write.py example

Author: franzpoeschel

examples/10_streaming_write.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 import sys
 
+from mpi4py import MPI
 import numpy as np
 import openpmd_api as io
 
@@ -18,9 +19,11 @@
         print("SST engine not available in ADIOS2.")
         sys.exit(0)
 
+    comm = MPI.COMM_WORLD
+
     # create a series and specify some global metadata
     # change the file extension to .json, .h5 or .bp for regular file writing
-    series = io.Series("simData.sst", io.Access_Type.create, config)
+    series = io.Series("simData.bp5", io.Access_Type.create, config)
     series.set_author("Franz Poeschel <f.poeschel@hzdr.de>")
     series.set_software("openPMD-api-python-examples")
 
@@ -52,8 +55,10 @@
                                dtype=np.dtype("double"))
         for dim in ["x", "y", "z"]:
             pos = electronPositions[dim]
-            pos.reset_dataset(io.Dataset(local_data.dtype, [length]))
-            pos[()] = local_data
+            pos.reset_dataset(
+                io.Dataset(local_data.dtype, [comm.size * length])
+            )
+            pos[comm.rank * length : (comm.rank + 1) * length] = local_data
 
         # optionally: flush now to clear buffers
         iteration.series_flush()  # this is a shortcut for `series.flush()`