desc.integrals.Bounce2D.compute_theta

static Bounce2D.compute_theta(eq, X=16, Y=32, rho=Array([1.], dtype=float64), iota=None, params=None, profiles=None, tol=1e-07, **kwargs)Source

Return DESC coordinates θ of (α,ζ) Fourier Chebyshev basis nodes.

Parameters:

  • eq (Equilibrium) – Equilibrium to use defining the coordinate mapping.

  • X (int) – Poloidal Fourier grid resolution to interpolate the poloidal coordinate. Preferably rounded down to power of 2.

  • Y (int) – Toroidal Chebyshev grid resolution to interpolate the poloidal coordinate. Preferably rounded down to power of 2.

  • rho (float or jnp.ndarray) – Shape (num rho, ). Flux surfaces labels in [0, 1] on which to compute.

  • iota (float or jnp.ndarray) – Shape (num rho, ). Optional, rotational transform on the flux surfaces to compute on.

  • params (dict[str,jnp.ndarray]) – Parameters from the equilibrium, such as R_lmn, Z_lmn, i_l, p_l, etc Defaults to eq.params_dict.

  • profiles – Optional profiles.

  • tol (float) – Stopping tolerance for root finding. Default is 1e-7.

  • kwargs – Additional parameters to supply to the coordinate mapping function. See desc.equilibrium.Equilibrium.map_coordinates.

Returns:

theta (jnp.ndarray) – Shape (num rho, X, Y). DESC coordinates θ.