avoid dumping particle magnetic field to output in Python_PlasmaAcceleration (#893)

Co-authored-by: David Grote <dpgrote@lbl.gov>

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')