desc.integrals.Bounce1D.check_points

Bounce1D.check_points(points, pitch_inv, *, plot=True, **kwargs)Source

Check that bounce points are computed correctly.

Parameters:

  • points (tuple[jnp.ndarray]) – Shape (num ρ, num α, num pitch, num well). Output of method self.points. Tuple of length two (z1, z2) that stores ζ coordinates of bounce points. The points are ordered and grouped such that the straight line path between z1 and z2 resides in the epigraph of B.

  • pitch_inv (jnp.ndarray) – Shape (num ρ, num pitch). 1/λ values to compute the bounce integrals. 1/λ(ρ) is specified by pitch_inv[ρ] where in the latter the labels are interpreted as the indices that correspond to that field line.

  • plot (bool) – Whether to plot the field lines and bounce points of the given pitch angles.

  • kwargs – Keyword arguments into desc/integrals/_bounce_utils.py::plot_ppoly.

Returns:

plots (list) – Matplotlib (fig, ax) tuples for the 1D plot of each field line.