field_weights.hpp

Go to the documentation of this file.

#ifndef FIELD_WEIGHTS_HPP
#define FIELD_WEIGHTS_HPP

#include "globals.hpp"
#include "linear_weights.hpp"
#include "grid.hpp"
#include "gyro_radius.hpp"

// Pack of LinearWeights
// Contains the linear weights needed for a particular templated configuration
// Constructor populates the weights
template<KinType T, PhiInterpType PIT>
struct FieldWeights;

// Weights for gyrokinetic species in XGC1
#  ifdef NEWGYROMATRIX
template<>
struct FieldWeights<GyroKin, PhiInterpType::Planes>
{
    LinearWeights phi;
    LinearWeights rho[2];

    template<class Device>
    KOKKOS_INLINE_FUNCTION FieldWeights(const Grid<Device>& grid, double phi_in, int i_simd, SimdGyroRadius<GyroKin>& rho_in, double inv_drho, int nrho)
    {
        phi.set(phi_in, grid.inv_delta_phi, grid.nplanes);
        rho[0].set(rho_in[0][i_simd], inv_drho, nrho);
        rho[1].set(rho_in[1][i_simd], inv_drho, nrho);
    }
};
#  else
template<>
struct FieldWeights<GyroKin, PhiInterpType::Planes>
{
    LinearWeights phi;
    LinearWeights rho;

    template<class Device>
    KOKKOS_INLINE_FUNCTION FieldWeights(const Grid<Device>& grid, double phi_in, int i_simd, SimdGyroRadius<GyroKin>& rho_in, double inv_drho, int nrho)
    {
        phi.set(phi_in, grid.inv_delta_phi, grid.nplanes);
        rho.set(rho_in[i_simd], inv_drho, nrho);
    }
};
#  endif

// Weights for drift kinetic species in XGC1
template<>
struct FieldWeights<DriftKin, PhiInterpType::Planes>
{
    LinearWeights phi;

    template<class Device>
    KOKKOS_INLINE_FUNCTION FieldWeights(const Grid<Device>& grid, double phi_in, int i_simd, SimdGyroRadius<DriftKin>& rho_in, double inv_drho, int nrho)
    {
        phi.set(phi_in, grid.inv_delta_phi, grid.nplanes);
    }
};

// Weights for XGCa
template<>
struct FieldWeights<GyroKin, PhiInterpType::None>
{
    LinearWeights rho;

    template<class Device>
    KOKKOS_INLINE_FUNCTION FieldWeights(const Grid<Device>& grid, double phi_in, int i_simd, SimdGyroRadius<GyroKin>& rho_in, double inv_drho, int nrho)
    {
        rho.set(rho_in[i_simd], inv_drho, nrho);
    }
};

template<>
struct FieldWeights<DriftKin, PhiInterpType::None>
{
    // Empty
    template<class Device>
    KOKKOS_INLINE_FUNCTION FieldWeights(const Grid<Device>& grid, double phi_in, int i_simd, SimdGyroRadius<DriftKin>& rho_in, double inv_drho, int nrho) {}
};


#endif