vmec_jax.vmec_parity

VMEC parity helper kernels.

This module provides small kernels that reproduce specific VMEC discrete conventions used in the reference wout_*.nc outputs, as needed for parity diagnostics.

Functions

internal_odd_from_physical(phys_odd, s, *[, ...])	Convert physical odd-m contribution to VMEC internal odd field.
internal_odd_from_physical_vmec_jlam(*, ...)	VMEC-consistent internal odd field for lambda (jlam).
internal_odd_from_physical_vmec_m1(*, ...[, eps])	VMEC-consistent internal odd field from split odd-m contributions.
signed_maps_from_modes(modes)	Cached SignedModeMaps keyed by ModeTable.
split_rzl_even_odd_lasym(state, basis)	Split real-space fields into even/odd pieces for LASYM=True.
split_rzl_even_odd_m(state, basis, modes_m)	Evaluate real-space fields for even-m and odd-m subsets separately.
vmec_m1_internal_to_physical_signed(*, Rcos, ...)	Undo VMEC's m=1 internal constraint in signed coefficient storage.
vmec_m1_internal_to_physical_signed_host(*, ...)	NumPy host-side version of vmec_m1_internal_to_physical_signed.
vmec_m1_physical_to_internal_signed(*, Rcos, ...)	Apply VMEC's m=1 internal constraint in signed coefficient storage.

Classes

ParityRZL(R_even, R_odd, Z_even, Z_odd, ...)	Real-space R/Z/L fields split into VMEC even/odd-m parity pieces.
SignedModeMaps(mpol, nrange, idx_pos, ...[, ...])	Precomputed index/mask maps for signed <-> (m,n>=0) conversions.