VGridDistribution< Device > Class Template Reference

#include <vgrid_distribution.hpp>

Collaboration diagram for VGridDistribution< Device >:

Public Types
using	exec_space = typename Device::execution_space

Public Member Functions
	VGridDistribution ()
	VGridDistribution (int nsp, const VelocityGrid &vgrid, const DomainDecomposition< DeviceType > &pol_decomp)
KOKKOS_INLINE_FUNCTION double &	operator() (int isp, int ivr, int inode, int ivz) const
KOKKOS_INLINE_FUNCTION double &	operator[] (int i) const
double *	data ()
template<typename F >
void	for_all_elements (const std::string label, F lambda_func) const
template<typename F >
void	for_each_element (const std::string label, F lambda_func) const
KOKKOS_INLINE_FUNCTION VGridWeights	get_weights (double mu, double vp, double inv_vgrid_vol) const
KOKKOS_INLINE_FUNCTION void	scatter (int i_node, const VGridWeights &wt, double value) const
KOKKOS_INLINE_FUNCTION double	gather (int i_node, const VGridWeights &wt) const
KOKKOS_INLINE_FUNCTION int	n_species () const
KOKKOS_INLINE_FUNCTION int	n_vr () const
KOKKOS_INLINE_FUNCTION int	n_nodes () const
KOKKOS_INLINE_FUNCTION int	n_vz () const
KOKKOS_INLINE_FUNCTION int	size () const
double	get_smu_n (int imu) const

Public Attributes
View< double ****, CLayout, Device >	f
View< double *, CLayout, Device, Kokkos::MemoryTraits < Kokkos::Unmanaged > >	f_1D
double	vp_max
	max parallel velocity More...
double	dvp
	grid spacing in parallel velocity More...
double	smu_max
	max mu More...
double	dsmu
	grid spacing in mu More...
double	inv_mu0_factor

Member Typedef Documentation

template<class Device>

using VGridDistribution< Device >::exec_space = typename Device::execution_space

Constructor & Destructor Documentation

template<class Device>

VGridDistribution< Device >::VGridDistribution ( )

inline

template<class Device>

VGridDistribution< Device >::VGridDistribution ( int  nsp, const VelocityGrid &  vgrid, const DomainDecomposition< DeviceType > &  pol_decomp  )

inline

Member Function Documentation

template<class Device>

double* VGridDistribution< Device >::data ( )

inline

Here is the caller graph for this function:

template<class Device>

template<typename F >

void VGridDistribution< Device >::for_all_elements ( const std::string  label, F  lambda_func  ) const

inline

Loops over all elements with a 1D index. Enables clean syntax for operations that don't have a velocity or node dependence

Parameters

[in]	label	is the label of the parallel_for used for debugging
[in]	lambda_func	is the lambda function. Must be copy [=] not by reference

Here is the caller graph for this function:

template<class Device>

template<typename F >

void VGridDistribution< Device >::for_each_element ( const std::string  label, F  lambda_func  ) const

inline

Loops over all elements with a 4D index. For use when the velocity, node, or species indices are needed for the operation

Parameters

[in]	label	is the label of the parallel_for used for debugging
[in]	lambda_func	is the lambda function. Must be copy [=] not by reference

template<class Device>

KOKKOS_INLINE_FUNCTION double VGridDistribution< Device >::gather ( int  i_node, const VGridWeights &  wt  ) const

inline

template<class Device>

double VGridDistribution< Device >::get_smu_n ( int  imu ) const

inline

Here is the caller graph for this function:

template<class Device>

KOKKOS_INLINE_FUNCTION VGridWeights VGridDistribution< Device >::get_weights ( double  mu, double  vp, double  inv_vgrid_vol  ) const

inline

template<class Device>

KOKKOS_INLINE_FUNCTION int VGridDistribution< Device >::n_nodes ( ) const

inline

Here is the caller graph for this function:

template<class Device>

KOKKOS_INLINE_FUNCTION int VGridDistribution< Device >::n_species ( ) const

inline

Here is the caller graph for this function:

template<class Device>

KOKKOS_INLINE_FUNCTION int VGridDistribution< Device >::n_vr ( ) const

inline

Here is the caller graph for this function:

template<class Device>

KOKKOS_INLINE_FUNCTION int VGridDistribution< Device >::n_vz ( ) const

inline

Here is the caller graph for this function:

template<class Device>

KOKKOS_INLINE_FUNCTION double& VGridDistribution< Device >::operator() ( int  isp, int  ivr, int  inode, int  ivz  ) const

inline

template<class Device>

KOKKOS_INLINE_FUNCTION double& VGridDistribution< Device >::operator[] ( int  i ) const

inline

template<class Device>

KOKKOS_INLINE_FUNCTION void VGridDistribution< Device >::scatter ( int  i_node, const VGridWeights &  wt, double  value  ) const

inline

template<class Device>

KOKKOS_INLINE_FUNCTION int VGridDistribution< Device >::size ( ) const

inline

Member Data Documentation

template<class Device>

double VGridDistribution< Device >::dsmu

grid spacing in mu

template<class Device>

double VGridDistribution< Device >::dvp

grid spacing in parallel velocity

template<class Device>

View<double****,CLayout, Device> VGridDistribution< Device >::f

template<class Device>

View<double*,CLayout, Device, Kokkos::MemoryTraits<Kokkos::Unmanaged> > VGridDistribution< Device >::f_1D

template<class Device>

double VGridDistribution< Device >::inv_mu0_factor

template<class Device>

double VGridDistribution< Device >::smu_max

max mu

template<class Device>

double VGridDistribution< Device >::vp_max

max parallel velocity

---

The documentation for this class was generated from the following file:

• /u/gitlab-xgc/builds/YGMz2TJ8/1/xgc/XGC-Devel/XGC_core/cpp/vgrid_distribution.hpp