Index

A

ang_dim (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) AxisymCircleNarrow (virtual_casing_jax.testdata.SurfType attribute)

AxisymCircleWide (virtual_casing_jax.testdata.SurfType attribute) AxisymNarrow (virtual_casing_jax.testdata.SurfType attribute) AxisymWide (virtual_casing_jax.testdata.SurfType attribute)

B

biotsavart_fx_u() (in module virtual_casing_jax.kernels) biotsavart_fx_u_eval() (in module virtual_casing_jax.integrals) biotsavart_fxd_u() (in module virtual_casing_jax.kernels)

biotsavart_fxd_u_eval() (in module virtual_casing_jax.integrals) build_patch_idx() (in module virtual_casing_jax.functional) build_quad_setup() (in module virtual_casing_jax.functional) build_surface_coord() (in module virtual_casing_jax.functional)

C

complete_vec_field() (in module virtual_casing_jax.surface_ops) compute_external_B() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_autodiff() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_batch() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_functional() (in module virtual_casing_jax.functional) compute_external_B_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_jvp_columns_functional() (in module virtual_casing_jax.functional) compute_external_B_normal_functional() (in module virtual_casing_jax.functional) compute_external_B_normal_jvp_columns_functional() (in module virtual_casing_jax.functional) compute_external_B_offsurf() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_offsurf_functional() (in module virtual_casing_jax.functional) compute_external_B_offsurf_schedule() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_B_offsurf_schedule_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB_batch() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB_functional() (in module virtual_casing_jax.functional) compute_external_gradB_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB_offsurf() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_external_gradB_offsurf_functional() (in module virtual_casing_jax.functional) compute_external_gradB_offsurf_schedule() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method)

compute_external_gradB_offsurf_schedule_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_batch() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_functional() (in module virtual_casing_jax.functional) compute_internal_B_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_offsurf() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_offsurf_schedule() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_B_offsurf_schedule_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_batch() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_functional() (in module virtual_casing_jax.functional) compute_internal_gradB_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_offsurf() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_offsurf_schedule() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) compute_internal_gradB_offsurf_schedule_jit() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) computeB_offsurface_adaptive() (in module virtual_casing_jax.integrals) computeB_offsurface_adaptive_schedule() (in module virtual_casing_jax.integrals) computeB_offsurface_baseline() (in module virtual_casing_jax.integrals) computeGradB_offsurface_adaptive_schedule() (in module virtual_casing_jax.integrals) cross_prod() (in module virtual_casing_jax.surface_ops)

D

dot_prod() (in module virtual_casing_jax.surface_ops)

Drand48 (class in virtual_casing_jax.testdata) dX (virtual_casing_jax.virtual_casing.QuadSetup attribute)

F

fft_c2r() (in module virtual_casing_jax.surface_ops) fft_r2c() (in module virtual_casing_jax.surface_ops)

field_period_target_coords() (in module virtual_casing_jax.integrals) from_vmec() (virtual_casing_jax.simsopt_virtual_casing.VirtualCasing class method) FunctionalSetup (class in virtual_casing_jax.functional)

G

Gpou (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

grad2d() (in module virtual_casing_jax.surface_ops)

H

half_period (virtual_casing_jax.functional.FunctionalSetup attribute)

hedgehog_order (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) hedgehog_wts (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

I

I_G2P (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

interp_idx (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

L

laplace_dx_u() (in module virtual_casing_jax.kernels) laplace_dx_u_eval() (in module virtual_casing_jax.integrals) laplace_dx_u_eval_singular() (in module virtual_casing_jax.integrals) laplace_fx_u() (in module virtual_casing_jax.kernels) laplace_fxd2_u() (in module virtual_casing_jax.kernels) laplace_fxd2_u_eval() (in module virtual_casing_jax.integrals) laplace_fxd2_u_eval_singular() (in module virtual_casing_jax.integrals) laplace_fxd2_u_eval_vec() (in module virtual_casing_jax.integrals)

laplace_fxd2_u_eval_vec_singular() (in module virtual_casing_jax.integrals) laplace_fxd_u() (in module virtual_casing_jax.kernels) laplace_fxd_u_eval() (in module virtual_casing_jax.integrals) laplace_fxd_u_eval_singular() (in module virtual_casing_jax.integrals) laplace_fxd_u_eval_vec() (in module virtual_casing_jax.integrals) laplace_fxd_u_eval_vec_singular() (in module virtual_casing_jax.integrals) LHD (virtual_casing_jax.testdata.SurfType attribute) load() (virtual_casing_jax.simsopt_virtual_casing.VirtualCasing class method)

M

M_G2P (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) magnetic_field_data() (in module virtual_casing_jax.testdata) (virtual_casing_jax.testdata.VirtualCasingTestData static method) magnetic_field_data_offsurf() (in module virtual_casing_jax.testdata) (virtual_casing_jax.testdata.VirtualCasingTestData static method) magnetic_field_grad_data() (in module virtual_casing_jax.testdata) (virtual_casing_jax.testdata.VirtualCasingTestData static method) module virtual_casing_jax.functional virtual_casing_jax.integrals virtual_casing_jax.kernels virtual_casing_jax.simsopt_virtual_casing virtual_casing_jax.singular_quadrature virtual_casing_jax.surface_ops virtual_casing_jax.testdata virtual_casing_jax.virtual_casing

N

nfp (virtual_casing_jax.functional.FunctionalSetup attribute) nfp_eff (virtual_casing_jax.functional.FunctionalSetup attribute) ngrid (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

NoneType (virtual_casing_jax.testdata.SurfType attribute) normal (virtual_casing_jax.virtual_casing.QuadSetup attribute) normal_orientation() (in module virtual_casing_jax.surface_ops) npolar (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

O

orient (virtual_casing_jax.functional.FunctionalSetup attribute) (virtual_casing_jax.virtual_casing.QuadSetup attribute)

P

patch_dim (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) patch_dim0 (virtual_casing_jax.functional.FunctionalSetup attribute) (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) patch_idx (virtual_casing_jax.functional.FunctionalSetup attribute)

patch_idx_cache (virtual_casing_jax.virtual_casing.QuadSetup attribute) plot() (virtual_casing_jax.simsopt_virtual_casing.VirtualCasing method) Ppou (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) precompute_singular() (in module virtual_casing_jax.singular_quadrature) prepare_functional_setup() (in module virtual_casing_jax.functional)

Q

quad_coord (virtual_casing_jax.virtual_casing.QuadSetup attribute) quad_np (virtual_casing_jax.functional.FunctionalSetup attribute) (virtual_casing_jax.virtual_casing.QuadSetup attribute) quad_nt (virtual_casing_jax.functional.FunctionalSetup attribute) (virtual_casing_jax.virtual_casing.QuadSetup attribute)

QuadSetup (class in virtual_casing_jax.virtual_casing) Quas3 (virtual_casing_jax.testdata.SurfType attribute) qw (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) qx (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute)

R

rad_dim (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) rad_dim_base (virtual_casing_jax.singular_quadrature.SingularPrecomp attribute) rand() (virtual_casing_jax.testdata.Drand48 method)

resample() (in module virtual_casing_jax.surface_ops) rotate_toroidal() (in module virtual_casing_jax.surface_ops) RotatingEllipseNarrow (virtual_casing_jax.testdata.SurfType attribute) RotatingEllipseWide (virtual_casing_jax.testdata.SurfType attribute)

S

save() (virtual_casing_jax.simsopt_virtual_casing.VirtualCasing method) select_patch_dim() (in module virtual_casing_jax.singular_quadrature) select_patch_dim_from_geom() (in module virtual_casing_jax.functional) setup() (virtual_casing_jax.virtual_casing.VirtualCasingJAX method) SingularPrecomp (class in virtual_casing_jax.singular_quadrature) src_np (virtual_casing_jax.functional.FunctionalSetup attribute) src_nt (virtual_casing_jax.functional.FunctionalSetup attribute)

state (virtual_casing_jax.testdata.Drand48 attribute) Stell (virtual_casing_jax.testdata.SurfType attribute) surf_normal_area_elem() (in module virtual_casing_jax.surface_ops) surf_np (virtual_casing_jax.functional.FunctionalSetup attribute) surf_nt (virtual_casing_jax.functional.FunctionalSetup attribute) surface_coordinates() (in module virtual_casing_jax.testdata) (virtual_casing_jax.testdata.VirtualCasingTestData static method) SurfType (class in virtual_casing_jax.testdata)

T

target_surface_normal() (in module virtual_casing_jax.functional)

trg_np (virtual_casing_jax.functional.FunctionalSetup attribute) trg_nt (virtual_casing_jax.functional.FunctionalSetup attribute)

U

upsample() (in module virtual_casing_jax.surface_ops)

V

virtual_casing_jax.functional module virtual_casing_jax.integrals module virtual_casing_jax.kernels module virtual_casing_jax.simsopt_virtual_casing module virtual_casing_jax.singular_quadrature module

virtual_casing_jax.surface_ops module virtual_casing_jax.testdata module virtual_casing_jax.virtual_casing module VirtualCasing (class in virtual_casing_jax.simsopt_virtual_casing) VirtualCasingJAX (class in virtual_casing_jax.virtual_casing) VirtualCasingTestData (class in virtual_casing_jax.testdata)

W

W7X (virtual_casing_jax.testdata.SurfType attribute)

W7X_ (virtual_casing_jax.testdata.SurfType attribute)