scatter.hpp

Go to the documentation of this file.

#ifndef SCATTER_HPP
#define SCATTER_HPP
 
#include "globals.hpp"
#include "sml.hpp"
#include "magnetic_field.hpp"
#include "grid.hpp"
#include "particles.hpp"
#include "plasma.hpp"
#include "get_current_triangles.hpp"
#include "charge.hpp"
#include "domain_decomposition.hpp"
 
template<KinType KT>
void calculate_charge(const Simulation<DeviceType> &sml, const Grid<DeviceType> &grid,
        const MagneticField<DeviceType> &magnetic_field, ElectricField<DeviceType> &electric_field,
        Charge<DeviceType, KT> &charge, Plasma& plasma,
        const VelocityGrid& vgrid, const DomainDecomposition<DeviceType>& pol_decomp, bool fullf_only);
 
void charge_offset_fullf_setup(NLReader::NamelistReader& nlr, const Simulation<DeviceType> &sml, const Grid<DeviceType> &grid,
        const MagneticField<DeviceType> &magnetic_field, ElectricField<DeviceType> &electric_field, Plasma& plasma,
        const VelocityGrid& vgrid, const DomainDecomposition<DeviceType>& pol_decomp, Charges& charges);
 
template<KinType KT, PhiInterpType PIT>
void scatter(const Grid<DeviceType> &grid,
        const MagneticField<DeviceType> &magnetic_field, const GridField<DeviceType, VarType::Scalar,PIT, TorType::OnePlane, KT>& Ah_gf,
        const Species<DeviceType> &species, bool exclude_private, bool do_cv_adjustment, bool cv_full_weight, const Charge<DeviceType,KT> &charge);
 
 
// Used in regression tests
template<KinType KT>
inline void var_transpose_in_place(const View<double***,CLayout,DeviceType>& dest_view, const GridField<DeviceType, VarType::Scalar,PIT_GLOBAL, TorType::OnePlane, KT, SCATTER_TYPE_GLOBAL>& src_field){
    int nnode = dest_view.extent(2);
    int nrhop1 = dest_view.extent(0);
    int nphi = dest_view.extent(1);
 
    // Cast input as an unmanaged view of doubles
    View<double***,CLayout,DeviceType,Kokkos::MemoryTraits<Kokkos::Unmanaged>> src_view((double*)src_field.data(),nnode, nrhop1, nphi);
 
    // Copy to a temporary view so there are no conflicts in the tranpose
    View<double***,CLayout,DeviceType> view_tmp(NoInit("view_tmp"),src_view.layout());
    Kokkos::deep_copy(view_tmp, src_view);
 
    // parallellize on node index
    Kokkos::parallel_for("var_transpose_in_place", Kokkos::RangePolicy<ExSpace>( 0, nnode), KOKKOS_LAMBDA(const int inode){
        for(int irho=0; irho<nrhop1; irho++){
            for(int iphi=0; iphi<nphi; iphi++){
                dest_view(irho, iphi, inode) = view_tmp(inode, irho, iphi);
            }
        }
    });
    Kokkos::fence();
}
 
 
 
#endif