Particles

Parthenon provides a data framework for particle methods that allows for Kokkos accelerated parallel dispatch of compute operations. Particle memory is allocated as a separate pool for each variable in the particle.

Swarms

A Swarm contains all the particle data for all particles of a given species. It owns a set of ParticleVariables, one for each value of each particle. For example, the spatial positions x, y, and z of the particles in a swarm are three separate ParticleVariables. ParticleVariables can be either Real- or int-valued, which is specified by the metadata values Metadata::Real and Metadata::Integer. ParticleVariables should also contain the Metadata::Particle flag. By default, ParticleVariables provide one scalar quantity per particle, but up to 2D data per particle is currently supported, by passing std::vector<int>{N1, N2} as the second argument to the ParticleVariable Metadata. All Swarms by default contain x, y, and z ParticleVariables; additional fields can be added as:

Swarm.Add(name, metadata)

For a given species, each MeshBlock contains its own Swarm that holds the particles of that species that are spatially contained by that MeshBlock. The MeshBlock is pointed to by Swarm::pmy_block.

The Swarm is a host-side object, but some of its data members are required for device- side compution. To access this data, a SwarmDeviceContext object is created via Swarm::GetDeviceContext(). This object can then be passed by copy into Kokkos lambdas. Hereafter we refer to it as swarm_d.

To add particles to a Swarm, one calls

ParArray1D<bool> new_particles_mask = swarm->AddEmptyParticles(num_to_add, new_indices)

This call automatically resizes the memory pools as necessary and returns a ParArray1D<bool> mask indicating which indices in the ParticleVariables are newly available. new_indices is a reference to a ParArrayND<int> of size num_to_add which contains the indices of each newly added particle.

To remove particles from a Swarm, one first calls

swarm_d.MarkParticleForRemoval(index_to_remove)

inside device code. This only indicates that this particle should be removed from the pool, it does not actually update any data. To remove all particles so marked, one then calls

swarm.RemoveMarkedParticles()

in host code. This updates the swarm such that the marked particles are seen as free slots in the memory pool.

Parallel Dispatch

Parallel computations on particle data can be performed with the usual MeshBlock par_for calls. Typically one loops over the entire range of active indices and uses a mask variable to only perform computations on currently active particles:

auto &x = swarm.Get("x").Get();
swarm.pmy_block->par_for("Simple loop", 0, swarm.GetMaxActiveIndex(),
  KOKKOS_LAMBDA(const int n) {
    if (swarm_d.IsActive(n)) {
      x(n) += 1.0;
    }
  });

Sorting

By default, particles are stored in per-meshblock pools of memory. However, one frequently wants convenient access to all the particles in each computational cell separately. To facilitate this, the Swarm provides the method SortParticlesByCell (and the SwarmContainer provides the matching task SortParticlesByCell). Calling this function populates internal data structures that map from per-cell indices to the per-meshblock data array. These are accessed by the SwarmDeviceContext member functions GetParticleCountPerCell and GetFullIndex. See examples/particles for example usage.

Defragmenting

Because one typically loops over particles from 0 to max_active_index, if only a small fraction of particles in that range are active, significant effort will be wasted. To clean up these situations, Swarm provides a Defrag method which, when called, will copy all active particles to be contiguous starting from the 0 index. Defrag is not fully parallelized so should be called only sparingly.

SwarmContainer

A SwarmContainer contains a set of related Swarms, such as the different stages used by a higher order time integrator. This feature is currently not exercised in detail.

particles Example

An example showing how to create a Parthenon application that defines a Swarm and creates, destroys, and transports particles is available in parthenon/examples/particles.

Communication

Communication of particles across MeshBlocks, including across MPI processors, is supported. Particle communication is currently handled via paired asynchronous/synchronous tasking regions on each MPI processor. The asynchronous tasks include transporting particles and SwarmContainer::Send and SwarmContainer::Receive calls. The synchronous task checks every MeshBlock on that MPI processor for whether the Swarms are finished transporting. This set of tasks must be repeated in the driver’s evolution function until all particles are completed. See the particles example for further details. Note that this pattern is blocking, and may be replaced in the future.

AMR is currently not supported, but support will be added in the future.

Variable Packing

Similarly to grid variables, particle swarms support ParticleVariable packing, by the function Swarm::PackVariables. This also supports FlatIdx for indexing; see the particle_leapfrog example for usage.

Boundary conditions

Particle boundary conditions are not applied in separate kernel calls; instead, inherited classes containing boundary condition functions for updating particles or removing them when they are in boundary regions are allocated depending on the boundary flags specified in the input file. Swarm boundary conditions are set via the swarm_*_bc variable in the parthenon/mesh input block in the input file, where * is ix1, ox2, etc.

Currently, outflow and periodic boundaries are supported natively. You can register a new boundary condition by calling ApplicationInput::RegisterSwarmBoundaryCondition and assigning the function that allocates the device-side boundary condition objects. Then the parthenon/mesh/swarm_*_bv variable can be assigned to the name you registered. An example is given in the particles example.

Outputs

Outputs for swarms can be set in an output block, just like any other variable. The user must specify a comma separated list denoting which swarms are marked for output:

swarms = swarm1, swarm2, ...

By default every swarm is initialized with x, y, and z position variables. These are automatically output.

To specify additional outputs, one may add an additional comma separated list:

swarmname_variables = var1, var2, ...

Here swarmname is the name of the swarm in question, and var1, var2, etc., are the variables to output for that particular swarm. You may still specify x, y, and z, but specifying them is superfluous as they are automatically output for any swarm that is output.

Alternatively, you may provide the

swarm_variables = var1, var2, ...

input as a comma separated list. This will output each variable in the swarm_variables list for every swarm. This is most useful if all the swarms contain similar variable structure, or if you only have one swarm to output. The per-swarm lists can be composed with the swarm_variables field. Every swarm will output the vars in swarm_variables but then additionally the variables in a per-swarm list will be output for that swarm.

Note

Some visualization tools, like Visit and Paraview, prefer to have access to an id field for each particle, however it’s not clear that a unique ID is required for each particle in general. Therefore, swarms do not automatically contain an ID swarm variable. However, when Parthenon outputs a swarm, it automatically generates an ID variable even if one is not present or requested. If a variable named id'' is available **and** the user requests it be output, Parthenon will use it. Otherwise, Parthenon will generate an ``id variable just for output and write it to file.

Warning

The automatically generted id is unique for a snapshot in time, but not guaranteed to be time invariant. Indeed it is likely not the same between dumps.

Putting it all together, you might have an output block that looks like this:

<parthenon/output1>
file_type = hdf5
dt = 1.0
swarms = swarm1, swarm2
swarm_variables = shared_var
swarm1_variables = per_swarm_var
swarm2_variables = id

The result would be that both swarm1 and swarm2 output the variables x, y, z, and shared_var. But only swarm1 outputs per_swarm_var. Both swarm1 and swarm2 will output an id field. But the id field for swarm1 will be automatically generated, but the id field for swarm2 will use the user-initialized value if such a quantity is available.