#include <magnetic_field.hpp>

Collaboration diagram for MagneticField< Device >:

Public Member Functions
	MagneticField (NLReader::NamelistReader &nlr, const DomainDecomposition< DeviceType > &pol_decomp, const MagneticEquilFiles::Ptr &equil_files, const View< Equil::XPoint , HostType > &xpts, const RZPair &axis_in, const RZPair &psi_ref_point, const View< double , HostType > &psiN)

void	write () const

template<class Device2 >
MagneticField< Device2 >	mirror () const

	MagneticField (PsiOption psi_opt, double safety_factor_coeff=1.0, int eq_mr_in=-1, int eq_mz_in=-1, int eq_mpsi_in=-1)

	MagneticField ()

KOKKOS_INLINE_FUNCTION double	psi_norm () const

KOKKOS_INLINE_FUNCTION void	get_psi (const SimdVector &v, Simd< double > &psi_out) const

KOKKOS_INLINE_FUNCTION void	get_psi (const SimdVector2D &x, const Simd< double > &phi, Simd< double > &psi_out) const

KOKKOS_INLINE_FUNCTION double	get_psi (double r, double z, double phi) const

KOKKOS_INLINE_FUNCTION void	get_psi_and_derivs (double r, double z, double phi, double &psi, double &dpsidr, double &dpsidz, double &dpsidphi) const

KOKKOS_INLINE_FUNCTION void	get_psi_unit_vec (const SimdVector2D &x, double phi, Simd< double > &cosa, Simd< double > &sina) const

KOKKOS_INLINE_FUNCTION double	geometry_r (double r) const

KOKKOS_INLINE_FUNCTION bool	is_below_xpt_tangent (double r, double z) const

KOKKOS_INLINE_FUNCTION bool	is_above_xpt2_tangent (double r, double z) const

KOKKOS_INLINE_FUNCTION bool	is_in_region_3a_or_sep_leg (double r, double z, double psi) const

KOKKOS_INLINE_FUNCTION bool	is_in_region_1_or_2 (double r, double z, double psi) const

KOKKOS_INLINE_FUNCTION bool	is_in_region_1 (double r, double z, double psi) const

KOKKOS_INLINE_FUNCTION void	check_boundaries (const SimdVector2D &x, Simd< bool > &rz_outside) const

KOKKOS_INLINE_FUNCTION void	get_theta (const SimdVector2D &x, Simd< double > &theta) const

KOKKOS_INLINE_FUNCTION RZPair	get_xpoint (int ixpt) const

KOKKOS_INLINE_FUNCTION RZPair	get_axis () const

KOKKOS_INLINE_FUNCTION void	field (const SimdVector &v, SimdVector &bvec, SimdVector(&jacb)[3], Simd< double > &psivec, SimdVector2D &gradpsi, Simd< bool > &rz_outside) const

KOKKOS_INLINE_FUNCTION void	bvec_interpol (double r, double z, double phi, double &br, double &bz, double &bphi) const

KOKKOS_INLINE_FUNCTION void	bmag_interpol (const SimdVector &v, Simd< double > &bmag) const

KOKKOS_INLINE_FUNCTION void	bmag_interpol (const SimdVector2D &x, const Simd< double > &phi, Simd< double > &bmag) const

KOKKOS_INLINE_FUNCTION void	follow_field (const SimdVector2D &x_org, const Simd< double > &phi_org, const Simd< double > &phi_dest, SimdVector2D &x_dest) const

template<>
void	write () const

Public Attributes
double	bt_sign
	Whether toroidal field is reversed? More...

double	bp_sign
	Whether poloidal field is reversed? More...

RZBounds	bounds
	Simulation boundary. More...

double	inpsi
	Boundary condition used in a few spots. More...

double	outpsi
	Boundary condition used in a few spots. More...

Equilibrium	equil
	The object containing information about the magnetic equilibrium. More...

Private Member Functions
KOKKOS_INLINE_FUNCTION void	derivs (const double(&x)[2], double phi, double(&dx)[2]) const

KOKKOS_INLINE_FUNCTION double	I_value (double psi_in, int rgn3) const

KOKKOS_INLINE_FUNCTION double	I_deriv (double psi_in, int rgn3) const

template<>
	MagneticField (NLReader::NamelistReader &nlr, const DomainDecomposition< DeviceType > &pol_decomp, const MagneticEquilFiles::Ptr &equil_files_ptr, const View< Equil::XPoint , HostType > &xpts, const RZPair &axis_in, const RZPair &psi_ref_point, const View< double , HostType > &psiN)

template<>
	MagneticField (PsiOption psi_opt, double safety_factor_coeff, int eq_mr_in, int eq_mz_in, int eq_mpsi_in)

Private Attributes
int	ff_step
	Number of steps taken when projecting the particle location onto the midplane. More...

int	ff_order
	Order of RK scheme used for field following. Can be 1, 2, or 4. More...

Bicub< Device >	psi_bicub
	The object for interpolating psi (magnetic flux surfaces) More...

CubInterp< Device >	I_interp
	The object for interpolating I (for deriving toroidal magnetic field) More...

Friends
template<class Device2 >
class	MagneticField

Constructor & Destructor Documentation

template<class Device>

MagneticField< Device >::MagneticField	(	NLReader::NamelistReader &	nlr,
		const DomainDecomposition< DeviceType > &	pol_decomp,
		const MagneticEquilFiles::Ptr &	equil_files,
		const View< Equil::XPoint *, HostType > &	xpts,
		const RZPair &	axis_in,
		const RZPair &	psi_ref_point,
		const View< double *, HostType > &	psiN
	)

template<class Device>

MagneticField< Device >::MagneticField	(	PsiOption	psi_opt,
		double	safety_factor_coeff = `1.0`,
		int	eq_mr_in = `-1`,
		int	eq_mz_in = `-1`,
		int	eq_mpsi_in = `-1`
	)

template<class Device>

MagneticField< Device >::MagneticField ( )

inline

template<>

MagneticField< HostType >::MagneticField	(	NLReader::NamelistReader &	nlr,
		const DomainDecomposition< DeviceType > &	pol_decomp,
		const MagneticEquilFiles::Ptr &	equil_files_ptr,
		const View< Equil::XPoint *, HostType > &	xpts,
		const RZPair &	axis_in,
		const RZPair &	psi_ref_point,
		const View< double *, HostType > &	psiN
	)

private

Constructor for magnetic field

Here is the call graph for this function:

template<>

MagneticField< HostType >::MagneticField	(	PsiOption	psi_opt,
		double	safety_factor_coeff,
		int	eq_mr_in,
		int	eq_mz_in,
		int	eq_mpsi_in
	)

private

Here is the call graph for this function:

Member Function Documentation

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::bmag_interpol	(	const SimdVector &	v,
		Simd< double > &	bmag
	)		const

For vector of coordinates, get the B-field magnitude

Parameters

[in]	v	Vector of {r,z,phi} coordinates
[out]	bmag	Vector of magnetic field magnitudes

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::bmag_interpol	(	const SimdVector2D &	x,
		const Simd< double > &	phi,
		Simd< double > &	bmag
	)		const

For vector of coordinates, get the B-field magnitude

Parameters

[in]	x	Vector of {r,z} coordinates
[out]	bmag	Vector of magnetic field magnitudes

Here is the call graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::bvec_interpol	(	double	r,
		double	z,
		double	phi,
		double &	br,
		double &	bz,
		double &	bphi
	)		const

Get the magnetic field components at single location

Parameters

[in]	r	r coordinate
[in]	z	z coordinate
[in]	phi	phi coordinate
[out]	br	r component of B
[out]	bz	z component of B
[out]	bphi	phi component of B

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::check_boundaries	(	const SimdVector2D &	x,
		Simd< bool > &	rz_outside
	)		const

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::derivs	(	const double(&)	x[2],
		double	phi,
		double(&)	dx[2]
	)		const

private

Get the derivatives of the magnetic field to see where the particle should move to as it traces the field

Parameters

[in]	x	(r,z) coordinates
[in]	phi	phi coordinate
[out]	dx	change in (r,z)

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::field	(	const SimdVector &	v,
		SimdVector &	bvec,
		SimdVector(&)	jacb[3],
		Simd< double > &	psivec,
		SimdVector2D &	gradpsi,
		Simd< bool > &	rz_outside
	)		const

Get complete B-field info

Parameters

[in]	v	Vector of {r,z, phi} coordinates
[out]	bvec	Vector of magnetic field components (r,z,phi)
[out]	jacb	Vector of jacobians (3x3xSIMD_SIZE)
[out]	psivec	Vector of psi coordinates
[out]	gradpsi	Vector of psi gradients (r,z)
[out]	rz_outside	Vector of whether the particles are outside the simulation RZ bounds

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::follow_field	(	const SimdVector2D &	x_org,
		const Simd< double > &	phi_org,
		const Simd< double > &	phi_dest,
		SimdVector2D &	x_dest
	)		const

For a vector of locations in (r,z,phi), follow the equilibrium magnetic field back to a phi midplane to do particle-grid operations. The field is followed with a Runge Kutta scheme, which can be order 1, 2, or 4. Splitting this up to avoid branching could improve vectorization.

Parameters

[in]	x_org	Vector of (r,z) coordinates
[in]	phi_org	Vector of phi coordinates
[in]	phi_dest	Vector of phi coordinates of the target midplanes
[out]	x_dest	Vector of the (r,z) coordinates when the particles are projected along the field onto the specified plane

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION double MagneticField< Device >::geometry_r ( double r ) const

Returns the first input argument if using normal toroidal geometry, and the r coordinate of the magnetic axis if cylindrical

Parameters

[in] r is the actual r coordinate

Returns: double the r suitable for the geometry

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION RZPair MagneticField< Device >::get_axis ( ) const

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::get_psi	(	const SimdVector &	v,
		Simd< double > &	psi_out
	)		const

Get psi

Parameters

[in]	r_in	Vector of r coordinates
[in]	z_in	Vector of z coordinates
[out]	psi_out	Vector of psi coordinates

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::get_psi	(	const SimdVector2D &	x,
		const Simd< double > &	phi,
		Simd< double > &	psi_out
	)		const

Get psi

Parameters

[in]	r_in	Vector of r coordinates
[in]	z_in	Vector of z coordinates
[out]	psi_out	Vector of psi coordinates

Here is the call graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION double MagneticField< Device >::get_psi	(	double	r,
		double	z,
		double	phi
	)		const

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::get_psi_and_derivs	(	double	r,
		double	z,
		double	phi,
		double &	psi,
		double &	dpsidr,
		double &	dpsidz,
		double &	dpsidphi
	)		const

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::get_psi_unit_vec	(	const SimdVector2D &	x,
		double	phi,
		Simd< double > &	cosa,
		Simd< double > &	sina
	)		const

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION void MagneticField< Device >::get_theta	(	const SimdVector2D &	x,
		Simd< double > &	theta
	)		const

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION RZPair MagneticField< Device >::get_xpoint ( int ixpt ) const

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION double MagneticField< Device >::I_deriv	(	double	psi_in,
		int	rgn3
	)		const

private

Interpolate the phi component of the magnetic field or its derivative at a single point

Parameters

[in]	psi_in	Poloidal magnetic flux at that point
[in]	rgn3	Whether the location is in region 3

Returns: The phi component of the magnetic field or its derivative

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION double MagneticField< Device >::I_value	(	double	psi_in,
		int	rgn3
	)		const

private

Interpolate the phi component of the magnetic field or its derivative at a single point

Parameters

[in]	psi_in	Poloidal magnetic flux at that point
[in]	rgn3	Whether the location is in region 3

Returns: The phi component of the magnetic field or its derivative

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION bool MagneticField< Device >::is_above_xpt2_tangent	(	double	r,
		double	z
	)		const

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION bool MagneticField< Device >::is_below_xpt_tangent	(	double	r,
		double	z
	)		const

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION bool MagneticField< Device >::is_in_region_1	(	double	r,
		double	z,
		double	psi
	)		const

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION bool MagneticField< Device >::is_in_region_1_or_2	(	double	r,
		double	z,
		double	psi
	)		const

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device >

KOKKOS_INLINE_FUNCTION bool MagneticField< Device >::is_in_region_3a_or_sep_leg	(	double	r,
		double	z,
		double	psi
	)		const

Here is the call graph for this function:

Here is the caller graph for this function:

template<class Device>

template<class Device2 >

MagneticField<Device2> MagneticField< Device >::mirror ( ) const

inline

template<class Device >

KOKKOS_INLINE_FUNCTION double MagneticField< Device >::psi_norm ( ) const

Here is the caller graph for this function:

template<>

void MagneticField< HostType >::write ( ) const

Here is the call graph for this function:

template<class Device>

void MagneticField< Device >::write ( ) const

Here is the caller graph for this function:

Friends And Related Function Documentation

template<class Device>

template<class Device2 >

friend class MagneticField

friend

Member Data Documentation

template<class Device>

RZBounds MagneticField< Device >::bounds

Simulation boundary.

template<class Device>

double MagneticField< Device >::bp_sign

Whether poloidal field is reversed?

template<class Device>

double MagneticField< Device >::bt_sign

Whether toroidal field is reversed?

template<class Device>

Equilibrium MagneticField< Device >::equil

The object containing information about the magnetic equilibrium.

template<class Device>

int MagneticField< Device >::ff_order

private

Order of RK scheme used for field following. Can be 1, 2, or 4.

template<class Device>

int MagneticField< Device >::ff_step

private

Number of steps taken when projecting the particle location onto the midplane.

template<class Device>

CubInterp<Device> MagneticField< Device >::I_interp

private

The object for interpolating I (for deriving toroidal magnetic field)

template<class Device>

double MagneticField< Device >::inpsi

Boundary condition used in a few spots.

template<class Device>

double MagneticField< Device >::outpsi

Boundary condition used in a few spots.

template<class Device>

Bicub<Device> MagneticField< Device >::psi_bicub

private

The object for interpolating psi (magnetic flux surfaces)

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

/u/gitlab-xgc/builds/YGMz2TJ8/0/xgc/XGC-Devel/XGC_core/cpp/magnetic_field.hpp
/u/gitlab-xgc/builds/YGMz2TJ8/0/xgc/XGC-Devel/XGC_core/cpp/magnetic_field.tpp

Public Member Functions

Public Attributes

Private Member Functions

Private Attributes

Friends

Constructor & Destructor Documentation

Member Function Documentation

Friends And Related Function Documentation

Member Data Documentation