From a4c2b99651cbac22ccbf2b0f3e1441c61e639009 Mon Sep 17 00:00:00 2001
From: Roelof Groenewald <40245517+roelof-groenewald@users.noreply.github.com>
Date: Fri, 27 Aug 2021 11:39:37 -0700
Subject: Functionality to save particle positions from the previous step
 (#2206)

* added functionality to save particle positions from the previous step
* copied WarpX variable to local variable to fix issue with GPU and DPC++ compilation
* switched to using a species attribute to toggle whether previous positions are saved so it can be turned on for only a subset of species if desired
* changed variable name to be more verbose
* added CI test of saving the previous particle positions
* start of a table in the documentation to describe commonly used runtime attributes
* generate test benchmark data from results obtained with a 2 processor run - the same as what is done during the test
* relaxed tolerance on test
* regenerate CI test reference data with USE_PSATD=TRUE
* Update Docs/source/developers/particles.rst

Co-authored-by: Phil Miller <unmobile+gh@gmail.com>
Co-authored-by: Axel Huebl <axel.huebl@plasma.ninja>
---
 .../ParticleDataPython/PICMI_inputs_prev_pos_2d.py | 125 +++++++++++++++++++++
 1 file changed, 125 insertions(+)
 create mode 100644 Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py

(limited to 'Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py')

diff --git a/Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py b/Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py
new file mode 100644
index 000000000..c25b522af
--- /dev/null
+++ b/Examples/Tests/ParticleDataPython/PICMI_inputs_prev_pos_2d.py
@@ -0,0 +1,125 @@
+# --- Input file to test the saving of old particle positions
+
+import numpy as np
+from pywarpx import picmi
+
+constants = picmi.constants
+
+##########################
+# numerics parameters
+##########################
+
+dt = 7.5e-10
+
+# --- Nb time steps
+
+max_steps = 10
+
+# --- grid
+
+nx = 64
+nz = 64
+
+xmin = 0
+xmax = 0.03
+zmin = 0
+zmax = 0.03
+
+
+##########################
+# numerics components
+##########################
+
+grid = picmi.Cartesian2DGrid(
+    number_of_cells = [nx, nz],
+    lower_bound = [xmin, zmin],
+    upper_bound = [xmax, zmax],
+    lower_boundary_conditions = ['dirichlet', 'periodic'],
+    upper_boundary_conditions = ['dirichlet', 'periodic'],
+    lower_boundary_conditions_particles = ['absorbing', 'periodic'],
+    upper_boundary_conditions_particles = ['absorbing', 'periodic'],
+    moving_window_velocity = None,
+    warpx_max_grid_size = 32
+)
+
+solver = picmi.ElectrostaticSolver(
+    grid=grid, method='Multigrid', required_precision=1e-6,
+    warpx_self_fields_verbosity=0
+)
+
+##########################
+# physics components
+##########################
+
+uniform_plasma_elec = picmi.UniformDistribution(
+    density = 1e15,
+    upper_bound = [None] * 3,
+    rms_velocity = [np.sqrt(constants.kb * 1e3 / constants.m_e)] * 3,
+    directed_velocity = [0.] * 3
+)
+
+electrons = picmi.Species(
+    particle_type='electron', name='electrons',
+    initial_distribution=uniform_plasma_elec,
+    warpx_save_previous_position=True
+)
+
+##########################
+# diagnostics
+##########################
+
+field_diag = picmi.ParticleDiagnostic(
+    species=electrons,
+    name = 'diag1',
+    data_list=['previous_positions'],
+    period = 10,
+    write_dir = '.',
+    warpx_file_prefix = 'Python_prev_positions_plt'
+)
+
+##########################
+# simulation setup
+##########################
+
+sim = picmi.Simulation(
+    solver = solver,
+    time_step_size = dt,
+    max_steps = max_steps,
+    verbose = 1
+)
+
+sim.add_species(
+    electrons,
+    layout = picmi.GriddedLayout(
+        n_macroparticle_per_cell=[1, 1], grid=grid
+    )
+)
+sim.add_diagnostic(field_diag)
+
+##########################
+# simulation run
+##########################
+
+sim.step(max_steps - 1)
+
+##########################
+# check that the new PIDs
+# exist
+##########################
+
+from pywarpx import _libwarpx
+
+assert (_libwarpx.get_particle_comp_index('electrons', 'prev_x') > 0)
+assert (_libwarpx.get_particle_comp_index('electrons', 'prev_z') > 0)
+
+prev_z_vals = _libwarpx.get_particle_arrays(
+    'electrons', 'prev_z', 0
+)
+for z_vals in prev_z_vals:
+    assert np.all(z_vals < zmax)
+
+##########################
+# take the final sim step
+##########################
+
+sim.step(1)
-- 
cgit v1.2.3