3D LWFA PICMI Test: Add Electron Beam (#2723)

* LWFA PICMI Tests: Add Electron Beam

* Remove Electron Beam from 1D Test

Author: EZoni

Examples/Physics_applications/laser_acceleration/PICMI_inputs_3d.py

@@ -57,6 +57,31 @@
     name = 'electrons',
     initial_distribution = uniform_distribution)
 
+# Particles: beam electrons
+q_tot = 1e-12
+x_m = 0.
+y_m = 0.
+z_m = -28e-06
+x_rms = 0.5e-06
+y_rms = 0.5e-06
+z_rms = 0.5e-06
+ux_m = 0.
+uy_m = 0.
+uz_m = 500.
+ux_th = 2.
+uy_th = 2.
+uz_th = 50.
+gaussian_bunch_distribution = picmi.GaussianBunchDistribution(
+    n_physical_particles = q_tot / q_e,
+    rms_bunch_size = [x_rms, y_rms, z_rms],
+    rms_velocity = [c*ux_th, c*uy_th, c*uz_th],
+    centroid_position = [x_m, y_m, z_m],
+    centroid_velocity = [c*ux_m, c*uy_m, c*uz_m])
+beam = picmi.Species(
+    particle_type = 'electron',
+    name = 'beam',
+    initial_distribution = gaussian_bunch_distribution)
+
 # Laser
 e_max = 16e12
 position_z = 9e-06
@@ -108,6 +133,11 @@
     electrons,
     layout = picmi.GriddedLayout(grid = grid, n_macroparticle_per_cell = [1, 1, 1]))
 
+# Add beam electrons
+sim.add_species(
+    beam,
+    layout = picmi.PseudoRandomLayout(grid = grid, n_macroparticles = 100))
+
 # Add laser
 sim.add_laser(
     laser,

Regression/Checksum/benchmarks_json/Python_LaserAcceleration.json

@@ -1,25 +1,36 @@
 {
+  "beam": {
+    "particle_cpu": 0.0,
+    "particle_id": 5050.0,
+    "particle_momentum_x": 4.707586336874016e-20,
+    "particle_momentum_y": 4.4850722108112576e-20,
+    "particle_momentum_z": 1.36054441043288e-17,
+    "particle_position_x": 4.058764306495361e-05,
+    "particle_position_y": 3.7888695722549883e-05,
+    "particle_position_z": 0.00019656701118308398,
+    "particle_weight": 6241509.074460764
+  },
   "electrons": {
     "particle_cpu": 69212.0,
     "particle_id": 2655287162.0,
-    "particle_momentum_x": 1.79212316501518e-20,
-    "particle_momentum_y": 7.225819832716737e-20,
-    "particle_momentum_z": 4.231725460519779e-20,
-    "particle_position_x": 0.7139122621161638,
-    "particle_position_y": 0.7150340887578206,
-    "particle_position_z": 1.317577060864619,
+    "particle_momentum_x": 1.7921232203945004e-20,
+    "particle_momentum_y": 7.225819894813053e-20,
+    "particle_momentum_z": 4.231725460173154e-20,
+    "particle_position_x": 0.7139122621161993,
+    "particle_position_y": 0.7150340887578637,
+    "particle_position_z": 1.3175770600690966,
     "particle_weight": 12926557617.187498
   },
   "lev=0": {
-    "Bx": 5863879.027613791,
-    "By": 2411.49823974812,
-    "Bz": 116025.4367923822,
-    "Ex": 6267728226590.701,
-    "Ey": 1670763224821434.0,
-    "Ez": 104345981838458.8,
-    "jx": 555687757148559.1,
-    "jy": 1595895515963762.0,
-    "jz": 1045266123023548.0,
-    "rho": 2211742630.95043
+    "Bx": 5866866.85492377,
+    "By": 11177.920546471447,
+    "Bz": 116026.93444649166,
+    "Ex": 8178548880638.266,
+    "Ey": 1671614207789070.8,
+    "Ez": 106548168484665.61,
+    "jx": 555903247963958.4,
+    "jy": 1595974150308405.2,
+    "jz": 1366292284444382.5,
+    "rho": 2212813480.6236324
   }
 }