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 resize_key_max(int n_ptl){
        // 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 resize_bin_max(int n_bins){
        // 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, bool sort_phase0);

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 sort_phase0, 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