shift.hpp

Go to the documentation of this file.

#ifndef SHIFT_HPP
#define SHIFT_HPP
 
#include "plasma.hpp"
#include "domain_decomposition.hpp"
#include "grid.hpp"
#include "sort.hpp"
 
struct ShiftOptions{
    bool plane_sort_only; // Skip the triangle search; send particles to the MPI process with the same plane rank on the correct plane
    bool evenly_distribute_ptl; // Ignore poloidal domain decomposition; distribute particles evenly on the plane.
 
    ShiftOptions()
      : plane_sort_only(false),
        evenly_distribute_ptl(false)
    {}
};
 
#ifdef USE_MPI
template<class DataType>
void transpose_and_shift(Kokkos::View<int*,HostType>& sendcnt, Kokkos::View<int*,HostType>& sdispls, Kokkos::View<int*,HostType>& recvcnt, Kokkos::View<int*,HostType>& rdispls, Cabana::AoSoA<DataType,DeviceType,VEC_LEN>& local_particles, int n_staying, int n_leaving, int n_arriving, const MyMPI& mpi);
#endif
 
void shift(Species<DeviceType> &species, SortViews<DeviceType> &sort_views, const Grid<DeviceType> &grid, const MagneticField<DeviceType> &magnetic_field, const DomainDecomposition<DeviceType> &pol_decomp, ShiftOptions shift_options = ShiftOptions());
 
void shift_all_species(Plasma& plasma, const Grid<DeviceType>& grid, const MagneticField<DeviceType>& magnetic_field, const DomainDecomposition<DeviceType>& pol_decomp, bool evenly_distribute_ptl=false);
 
#endif