vgrid_weights.hpp

Go to the documentation of this file.

#ifndef VGRID_WEIGHTS_HPP
#define VGRID_WEIGHTS_HPP

#include "linear_weights.hpp"

// Struct containing indices and weights on velocity grid
struct VGridWeights{
    int i_vr;
    int i_vz;
    double w_00;
    double w_01;
    double w_10;
    double w_11;

    KOKKOS_INLINE_FUNCTION VGridWeights(const LinearWeights& vr_wt, const LinearWeights& vz_wt)
     : i_vr(vr_wt.i),
       i_vz(vz_wt.i),
       w_00(vr_wt.w[0]*vz_wt.w[0]),
       w_01(vr_wt.w[0]*vz_wt.w[1]),
       w_10(vr_wt.w[1]*vz_wt.w[0]),
       w_11(vr_wt.w[1]*vz_wt.w[1])
    {}

    // Reuse i_vr as indicator of validity
    KOKKOS_INLINE_FUNCTION VGridWeights()
     : i_vr(-1)
    {}

    KOKKOS_INLINE_FUNCTION bool is_valid(){
        return (i_vr==-1) ? false : true;
    }
};

#endif