vgrid_distribution.hpp

Go to the documentation of this file.

#ifndef VGRID_DISTRIBUTION_HPP
#define VGRID_DISTRIBUTION_HPP

#include "space_settings.hpp"
#include "velocity_grid.hpp"
#include "view_arithmetic.hpp"
#include "velocity_grid.hpp"
#include "domain_decomposition.hpp"

template<class Device>
class VGridDistribution{
    public:

    Kokkos::View<double****,Kokkos::LayoutRight, Device> f;

    // Make f private when possible - ALS
    //public:

    double vp_max;  
    double dvp;     
    double smu_max; 
    double dsmu;    
    double inv_mu0_factor;

    VGridDistribution(){}

    VGridDistribution(int nsp, const VelocityGrid& vgrid, const DomainDecomposition<DeviceType>& pol_decomp)
      : f("f", nsp, vgrid.nvr, pol_decomp.nnodes, vgrid.nvz),
        vp_max(vgrid.vp_max),
        dvp(vgrid.dvp),
        smu_max(vgrid.smu_max),
        dsmu(vgrid.dsmu),
        inv_mu0_factor(vgrid.inv_mu0_factor)
    {}

    double& operator()(int isp, int ivr, int inode, int ivz){ return f(isp, ivr, inode, ivz);}
    double  operator()(int isp, int ivr, int inode, int ivz) const { return f(isp, ivr, inode, ivz);}
    
    double* data(){return f.data();}

    void add_ay(const VGridDistribution<Device>& dist2, double scalar){
        host_add_ay_to_x(f, dist2.f, scalar);
    }

    // Access dimensions
    inline int n_species() const{
        return f.extent(0);
    }

    inline int n_vr() const {
        return f.extent(1);
    }

    inline int n_nodes() const{
        return f.extent(2);
    }

    inline int n_vz() const{
        return f.extent(3);
    }

    // Miscellaneous
    inline double get_smu_n(int imu) const{
        return (imu>0 ? dsmu*imu : dsmu*inv_mu0_factor);
    }
};

#endif