aboutsummaryrefslogtreecommitdiff
path: root/Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py
diff options
context:
space:
mode:
authorGravatar Axel Huebl <axel.huebl@plasma.ninja> 2021-10-16 00:25:20 -0700
committerGravatar GitHub <noreply@github.com> 2021-10-16 07:25:20 +0000
commitd76891e864f199715fe0a27830ae2021d01a6eba (patch)
treebc0fa1a102a4b0a625480c7e17156b03e3134a51 /Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py
parent9cbb0ebc81cc44f15ff789995fc4074b5d51bffd (diff)
downloadWarpX-d76891e864f199715fe0a27830ae2021d01a6eba.tar.gz
WarpX-d76891e864f199715fe0a27830ae2021d01a6eba.tar.zst
WarpX-d76891e864f199715fe0a27830ae2021d01a6eba.zip
Add Test: openPMD PICMI (#2438)
* Gaussian Beam PICMI: Add openPMD Argument Add an `--diagformat` option to switch between plotfiles and openPMD for the Gaussian Beam PICMI example. * CI: openPMD + PICMI on Windows * Regression Test: Add openPMD PICMI * format -> warpx_format
Diffstat (limited to 'Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py')
-rw-r--r--Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py13
1 files changed, 12 insertions, 1 deletions
diff --git a/Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py b/Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py
index 4d3e00687..710b78bd5 100644
--- a/Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py
+++ b/Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py
@@ -1,5 +1,14 @@
from pywarpx import picmi
#from warp import picmi
+import argparse
+
+
+parser = argparse.ArgumentParser(description="Gaussian beam PICMI example")
+
+parser.add_argument('--diagformat', type=str,
+ help='Format of the full diagnostics (plotfile, openpmd, ascent, sensei, ...)',
+ default='plotfile')
+args = parser.parse_args()
constants = picmi.constants
@@ -49,13 +58,15 @@ field_diag1 = picmi.FieldDiagnostic(name = 'diag1',
grid = grid,
period = 10,
data_list = ['E', 'B', 'J', 'part_per_cell'],
+ warpx_format = args.diagformat,
write_dir = '.',
warpx_file_prefix = 'Python_gaussian_beam_plt')
part_diag1 = picmi.ParticleDiagnostic(name = 'diag1',
period = 10,
species = [electrons, protons],
- data_list = ['weighting', 'momentum'])
+ data_list = ['weighting', 'momentum'],
+ warpx_format = args.diagformat)
sim = picmi.Simulation(solver = solver,
max_steps = 10,