aboutsummaryrefslogtreecommitdiff
path: root/Examples
diff options
context:
space:
mode:
authorGravatar MaxThevenet <mthevenet@lbl.gov> 2020-04-03 11:17:27 -0700
committerGravatar GitHub <noreply@github.com> 2020-04-03 11:17:27 -0700
commitde2071e04be17ea47e46a0a1e7c1f28b308b43e5 (patch)
tree7ce289731915d01fc8c3c2bc7bd0c5ff76d3c7df /Examples
parent9381c4c6d12fb2800485a11123c3e3c968b51a6d (diff)
downloadWarpX-de2071e04be17ea47e46a0a1e7c1f28b308b43e5.tar.gz
WarpX-de2071e04be17ea47e46a0a1e7c1f28b308b43e5.tar.zst
WarpX-de2071e04be17ea47e46a0a1e7c1f28b308b43e5.zip
avoid dumping particle magnetic field to output in Python_PlasmaAcceleration (#893)
Co-authored-by: David Grote <dpgrote@lbl.gov>
Diffstat (limited to 'Examples')
-rw-r--r--Examples/Physics_applications/plasma_acceleration/PICMI_inputs_plasma_acceleration.py19
1 files changed, 17 insertions, 2 deletions
diff --git a/Examples/Physics_applications/plasma_acceleration/PICMI_inputs_plasma_acceleration.py b/Examples/Physics_applications/plasma_acceleration/PICMI_inputs_plasma_acceleration.py
index 890d089a7..a1a8a0ba5 100644
--- a/Examples/Physics_applications/plasma_acceleration/PICMI_inputs_plasma_acceleration.py
+++ b/Examples/Physics_applications/plasma_acceleration/PICMI_inputs_plasma_acceleration.py
@@ -18,6 +18,8 @@ moving_window_velocity = [0., 0., constants.c]
number_per_cell_each_dim = [2, 2, 1]
+max_steps = 10
+
grid = picmi.Cartesian3DGrid(number_of_cells = [nx, ny, nz],
lower_bound = [xmin, ymin, zmin],
upper_bound = [xmax, ymax, zmax],
@@ -42,14 +44,27 @@ beam = picmi.Species(particle_type='electron', name='beam', initial_distribution
plasma = picmi.Species(particle_type='electron', name='plasma', initial_distribution=plasma_distribution)
sim = picmi.Simulation(solver = solver,
- max_steps = 10,
+ max_steps = max_steps,
verbose = 1,
- warpx_plot_int = 2,
+ warpx_plot_int = max_steps,
warpx_current_deposition_algo = 'esirkepov')
sim.add_species(beam, layout=picmi.GriddedLayout(grid=grid, n_macroparticle_per_cell=number_per_cell_each_dim))
sim.add_species(plasma, layout=picmi.GriddedLayout(grid=grid, n_macroparticle_per_cell=number_per_cell_each_dim))
+field_diag = picmi.FieldDiagnostic(grid = grid,
+ period = max_steps,
+ data_list = ['Ex', 'Ey', 'Ez', 'Jx', 'Jy', 'Jz', 'part_per_cell'],
+ write_dir = 'diags')
+
+part_diag = picmi.ParticleDiagnostic(period = max_steps,
+ species = [beam, plasma],
+ data_list = ['ux', 'uy', 'uz', 'weighting', 'Ex', 'Ey', 'Ez'],
+ write_dir = 'diags')
+
+sim.add_diagnostic(field_diag)
+sim.add_diagnostic(part_diag)
+
# write_inputs will create an inputs file that can be used to run
# with the compiled version.
#sim.write_input_file(file_name = 'inputs_from_PICMI')