1 #ifndef MAGNETIC_FIELD_HPP
2 #define MAGNETIC_FIELD_HPP
12 template<
class Device>
18 KOKKOS_INLINE_FUNCTION
void derivs(
const double (&x)[2],
double phi,
double (&dx)[2])
const;
20 #ifndef STELLARATOR_TRICUB
21 KOKKOS_INLINE_FUNCTION
double I_value(
double psi_in,
int rgn3)
const;
22 KOKKOS_INLINE_FUNCTION
double I_deriv(
double psi_in,
int rgn3)
const;
28 #ifndef STELLARATOR_TRICUB
50 template<
class Device2>
63 #ifndef STELLARATOR_TRICUB
67 m.psi_tricub = psi_tricub.template mirror<Device2>();
68 m.Br_tricub = Br_tricub.template mirror<Device2>();
69 m.Bz_tricub = Bz_tricub.template mirror<Device2>();
70 m.Bphi_tricub = Bphi_tricub.template mirror<Device2>();
76 #ifndef STELLARATOR_TRICUB
81 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)
88 bounds(0.5, 2.5, -1.0, 1.0),
96 int eq_mr = (eq_mr_in==-1 ? 100 : eq_mr_in);
100 int eq_mz = (eq_mz_in==-1 ? eq_mr : eq_mz_in);
111 int eq_mpsi = (eq_mpsi_in==-1 ? eq_mr : eq_mpsi_in);
124 KOKKOS_INLINE_FUNCTION
double psi_norm()
const;
130 KOKKOS_INLINE_FUNCTION
double get_psi(
double r,
double z,
double phi)
const;
132 KOKKOS_INLINE_FUNCTION
void get_psi_and_derivs(
double r,
double z,
double phi,
double& psi,
double& dpsidr,
double& dpsidz,
double& dpsidphi)
const;
136 KOKKOS_INLINE_FUNCTION
double geometry_r(
double r)
const;
139 KOKKOS_INLINE_FUNCTION
bool is_in_region_1(
double r,
double z,
double psi)
const;
146 KOKKOS_INLINE_FUNCTION
void bvec_interpol(
double r,
double z,
double phi,
double &br,
double &bz,
double &bphi)
const;
Definition: tricub.hpp:13
double inpsi
Boundary condition used in a few spots.
Definition: magnetic_field.hpp:43
KOKKOS_INLINE_FUNCTION bool is_in_region_1(double r, double z, double psi) const
Definition: magnetic_field.tpp:73
CubInterp< Device > I_interp
The object for interpolating I (for deriving toroidal magnetic field)
Definition: magnetic_field.hpp:30
KOKKOS_INLINE_FUNCTION void get_psi_and_derivs(double r, double z, double phi, double &psi, double &dpsidr, double &dpsidz, double &dpsidphi) const
Definition: magnetic_field.tpp:121
KOKKOS_INLINE_FUNCTION void bvec_interpol(double r, double z, double phi, double &br, double &bz, double &bphi) const
Definition: magnetic_field.tpp:321
Definition: magnetic_equil_files.hpp:16
PsiOption
Definition: equil.hpp:11
KOKKOS_INLINE_FUNCTION void get_psi(const SimdVector &v, Simd< double > &psi_out) const
Definition: magnetic_field.tpp:105
KOKKOS_INLINE_FUNCTION void follow_field(const SimdVector2D &x_org, const Simd< double > &phi_org, const Simd< double > &phi_dest, SimdVector2D &x_dest) const
Definition: magnetic_field.tpp:403
KOKKOS_INLINE_FUNCTION void check_boundaries(const SimdVector2D &x, Simd< bool > &rz_outside) const
Definition: magnetic_field.tpp:78
KOKKOS_INLINE_FUNCTION void get_theta(const SimdVector2D &x, Simd< double > &theta) const
Definition: magnetic_field.tpp:83
KOKKOS_INLINE_FUNCTION double geometry_r(double r) const
Definition: magnetic_field.tpp:183
Definition: rz_bounds.hpp:4
int ff_order
Order of RK scheme used for field following. Can be 1, 2, or 4.
Definition: magnetic_field.hpp:26
View< double *, HostType > create_range_view(std::string name, double x_min, double x_max, int n)
Definition: range_view.hpp:6
Definition: NamelistReader.hpp:193
Definition: magnetic_field.hpp:13
KOKKOS_INLINE_FUNCTION double I_deriv(double psi_in, int rgn3) const
Definition: magnetic_field.tpp:167
RZBounds bounds
Simulation boundary.
Definition: magnetic_field.hpp:42
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)
Definition: magnetic_field.hpp:81
Equilibrium equil
The object containing information about the magnetic equilibrium.
Definition: magnetic_field.hpp:45
KOKKOS_INLINE_FUNCTION void bmag_interpol(const SimdVector &v, Simd< double > &bmag) const
Definition: magnetic_field.tpp:372
Definition: cub_interp.hpp:12
MagneticField()
Definition: magnetic_field.hpp:122
KOKKOS_INLINE_FUNCTION double I_value(double psi_in, int rgn3) const
Definition: magnetic_field.tpp:156
KOKKOS_INLINE_FUNCTION double psi_norm() const
Definition: magnetic_field.tpp:62
KOKKOS_INLINE_FUNCTION void field(const SimdVector &v, SimdVector &bvec, SimdVector(&jacb)[3], Simd< double > &psivec, SimdVector2D &gradpsi, SimdVector &tdb, Simd< bool > &rz_outside) const
Definition: magnetic_field.tpp:197
double outpsi
Boundary condition used in a few spots.
Definition: magnetic_field.hpp:44
double max_z
Definition: rz_bounds.hpp:8
double bt_sign
Whether toroidal field is reversed?
Definition: magnetic_field.hpp:40
double min_r
Definition: rz_bounds.hpp:5
RZBounds bounds
Min and max for r and z.
Definition: equil.hpp:77
double bp_sign
Whether poloidal field is reversed?
Definition: magnetic_field.hpp:41
KOKKOS_INLINE_FUNCTION bool is_in_region_1_or_2(double r, double z, double psi) const
Definition: magnetic_field.tpp:68
double min_z
Definition: rz_bounds.hpp:7
Bicub< Device > psi_bicub
The object for interpolating psi (magnetic flux surfaces)
Definition: magnetic_field.hpp:29
int ff_step
Number of steps taken when projecting the particle location onto the midplane.
Definition: magnetic_field.hpp:25
KOKKOS_INLINE_FUNCTION void get_psi_unit_vec(const SimdVector2D &x, double phi, Simd< double > &cosa, Simd< double > &sina) const
Definition: magnetic_field.tpp:140
KOKKOS_INLINE_FUNCTION void derivs(const double(&x)[2], double phi, double(&dx)[2]) const
Definition: magnetic_field.tpp:517
MagneticField< Device2 > mirror() const
Definition: magnetic_field.hpp:51
double max_r
Definition: rz_bounds.hpp:6