diff options
Diffstat (limited to 'Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py')
| -rw-r--r-- | Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py | 13 |
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, |