pyoculus.toybox.cylindrical_toybox.psi_squared

pyoculus.toybox.cylindrical_toybox.psi_squared(rr: array, R: float, Z: float) → float64

\(\psi\) flux function for the squared circle equilibrium field. The squared circle equilibrium field is defined as:

\[\psi(r, z) = (r - R)^2 + (z - Z)^2.\]

Parameters:

• rr (array) – Position vector in cylindrical coordinates.

• R (float) – R coordinate of the center of the squared circle.

• Z (float) – Z coordinate of the center of the squared circle.

Returns:

\(\psi\) flux at position rr.

Return type:

float