aboutsummaryrefslogtreecommitdiff
path: root/Examples/Tests/Langmuir/PICMI_inputs_langmuir_rt.py
diff options
context:
space:
mode:
Diffstat (limited to 'Examples/Tests/Langmuir/PICMI_inputs_langmuir_rt.py')
-rw-r--r--Examples/Tests/Langmuir/PICMI_inputs_langmuir_rt.py49
1 files changed, 49 insertions, 0 deletions
diff --git a/Examples/Tests/Langmuir/PICMI_inputs_langmuir_rt.py b/Examples/Tests/Langmuir/PICMI_inputs_langmuir_rt.py
new file mode 100644
index 000000000..ae6731c05
--- /dev/null
+++ b/Examples/Tests/Langmuir/PICMI_inputs_langmuir_rt.py
@@ -0,0 +1,49 @@
+# --- Simple example of Langmuir oscillations in a uniform plasma
+
+import numpy as np
+from pywarpx import picmi
+
+constants = picmi.constants
+
+nx = 64
+ny = 64
+nz = 64
+
+xmin = -20.e-6
+ymin = -20.e-6
+zmin = -20.e-6
+xmax = +20.e-6
+ymax = +20.e-6
+zmax = +20.e-6
+
+uniform_plasma = picmi.UniformDistribution(density = 1.e25,
+ upper_bound = [0., None, None],
+ directed_velocity = [0.1*constants.c, 0., 0.])
+
+electrons = picmi.Species(particle_type='electron', name='electrons', initial_distribution=uniform_plasma)
+
+grid = picmi.Cartesian3DGrid(number_of_cells = [nx, ny, nz],
+ lower_bound = [xmin, ymin, zmin],
+ upper_bound = [xmax, ymax, zmax],
+ lower_boundary_conditions = ['periodic', 'periodic', 'periodic'],
+ upper_boundary_conditions = ['periodic', 'periodic', 'periodic'],
+ moving_window_velocity = [0., 0., 0.],
+ warpx_max_grid_size=32)
+
+solver = picmi.ElectromagneticSolver(grid=grid, cfl=1.)
+
+sim = picmi.Simulation(solver = solver,
+ max_steps = 40,
+ verbose = 1,
+ warpx_plot_int = 40,
+ warpx_current_deposition_algo = 'direct')
+
+sim.add_species(electrons, layout=picmi.GriddedLayout(n_macroparticle_per_cell=[2,2,2], grid=grid))
+
+# 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')
+
+# Alternatively, sim.step will run WarpX, controlling it from Python
+sim.step()
+