sort.hpp

Go to the documentation of this file.

#ifndef SORT_HPP
#define SORT_HPP
 
#include "domain_decomposition.hpp"
#include "grid.hpp"
#include "sheath.hpp"
 
template<class Device>
struct SortViews{
    Kokkos::View<double**,Kokkos::LayoutRight,Device> tmp_const; 
    Kokkos::View<int*,Kokkos::LayoutRight,Device> iperm; 
    Kokkos::View<int*,Kokkos::LayoutRight,Device> key; 
 
    // Permutation
    Kokkos::View<int*,Kokkos::LayoutRight,Device> count;     
    Kokkos::View<unsigned int*,Kokkos::LayoutRight,Device> place;     
 
    SortViews(int n_ptl, int n_bins)
      : tmp_const(Kokkos::ViewAllocateWithoutInitializing("tmp_const"), 4,n_ptl),
        iperm(Kokkos::ViewAllocateWithoutInitializing("iperm"), n_ptl),
        key(Kokkos::ViewAllocateWithoutInitializing("key"), n_ptl),
        count(Kokkos::ViewAllocateWithoutInitializing("count"), n_bins),
        place(Kokkos::ViewAllocateWithoutInitializing("place"), n_bins) {}
 
    inline void increase_key_max(int n_ptl){
        // No-op if the new scratch value is smaller
        if(n_ptl<iperm.extent(0)) return;
 
        // Resize to zero first to avoid temporary double allocation
        Kokkos::resize(tmp_const,4,0);
        Kokkos::resize(iperm,0);
        Kokkos::resize(key,0);
        
        // Reallocate
        Kokkos::resize(tmp_const,4,n_ptl);
        Kokkos::resize(iperm,n_ptl);
        Kokkos::resize(key,n_ptl);
    }
 
    inline void increase_bin_max(int n_bins){
        // No-op if the new scratch value is smaller
        if(n_bins<count.extent(0)) return;
 
        // Resize to zero first to avoid temporary double allocation
        Kokkos::resize(count,0);
        Kokkos::resize(place,0);
        
        // Reallocate
        Kokkos::resize(count,n_bins);
        Kokkos::resize(place,n_bins);
    }
};
 
void sort_particles_by_triangle(SortViews<DeviceType> &sort_views, const Grid<DeviceType> &grid, const MagneticField<DeviceType> &magnetic_field, const Species<DeviceType> &species, const SheathParticles<DeviceType> &sheath_particles, bool sort_sheath_ptl);
 
void sort_particles_by_pid(SortViews<DeviceType> &sort_views, const Grid<DeviceType> &grid, const MagneticField<DeviceType> &magnetic_field, const DomainDecomposition<DeviceType> &pol_decomp, const Species<DeviceType> &species, bool plane_sort_only);
 
void sort_particles_by_key(SortViews<DeviceType> &sort_views, const Species<DeviceType> &species, int n_bins, Cabana::AoSoA<PhaseDataTypes,DeviceType,VEC_LEN>& phase0, Cabana::AoSoA<PhaseDataTypes,DeviceType,VEC_LEN>& dy_sum, bool sort_intermediate_values);
 
int sort_particles_by_field_id(SortViews<DeviceType> &sort_views, const Grid<DeviceType> &grid, const MagneticField<DeviceType> &magnetic_field, const DomainDecomposition<DeviceType> &pol_decomp, const Species<DeviceType> &species, Cabana::AoSoA<PhaseDataTypes,DeviceType,VEC_LEN>& phase0, Cabana::AoSoA<PhaseDataTypes,DeviceType,VEC_LEN>& dy_sum, bool sort_intermediate_values);
#endif