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) {}
};

void sort_particles_by_triangle(const SortViews<DeviceType> &sort_views, const Grid<DeviceType> &grid, const Species<DeviceType> &species, const SheathParticles<DeviceType> &sheath_particles, bool sort_sheath_ptl, bool sort_phase0);
void sort_particles_by_pid(const 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);
#endif