desc.particles.VacuumGuidingCenterTrajectory.vf

VacuumGuidingCenterTrajectory.vf(t, x, args)Source

RHS of guiding center trajectories without collisions or slowing down.

vf method corresponds to the vf method in diffrax.AbstractTerm class and must have the same name and signature.

Parameters:

  • t (float) – Time to evaluate RHS at.

  • x (jax.Array, shape(4,)) – Position of particle in phase space [rho, theta, zeta, vpar] or [R, phi, Z, vpar].

  • args (tuple) – Should include the arguments needed by the model, (m, q, mu) as an array, Equilibrium or MagneticField object, params and any additional keyword arguments needed for magnetic field computation, such as iota profile for the Equilibrium, and source_grid for the MagneticField.

Returns:

dx (jax.Array, shape(N,4)) – Velocity of particles in phase space.