virtual-casing-jax 0.0.2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- virtual_casing_jax/__init__.py +129 -0
- virtual_casing_jax/functional.py +1053 -0
- virtual_casing_jax/geom/LHD.npz +0 -0
- virtual_casing_jax/geom/Quas3.npz +0 -0
- virtual_casing_jax/geom/W7X.npz +0 -0
- virtual_casing_jax/geom/__init__.py +1 -0
- virtual_casing_jax/integrals.py +1618 -0
- virtual_casing_jax/kernels.py +160 -0
- virtual_casing_jax/simsopt_virtual_casing.py +298 -0
- virtual_casing_jax/singular_quadrature.py +237 -0
- virtual_casing_jax/surface_ops.py +274 -0
- virtual_casing_jax/testdata.py +463 -0
- virtual_casing_jax/utils.py +100 -0
- virtual_casing_jax/virtual_casing.py +1714 -0
- virtual_casing_jax/w7x_coeffs.py +178 -0
- virtual_casing_jax-0.0.2.dist-info/METADATA +138 -0
- virtual_casing_jax-0.0.2.dist-info/RECORD +20 -0
- virtual_casing_jax-0.0.2.dist-info/WHEEL +5 -0
- virtual_casing_jax-0.0.2.dist-info/licenses/LICENSE +201 -0
- virtual_casing_jax-0.0.2.dist-info/top_level.txt +1 -0
|
@@ -0,0 +1,129 @@
|
|
|
1
|
+
"""virtual_casing_jax package."""
|
|
2
|
+
|
|
3
|
+
from . import surface_ops
|
|
4
|
+
from .surface_ops import (
|
|
5
|
+
complete_vec_field,
|
|
6
|
+
rotate_toroidal,
|
|
7
|
+
upsample,
|
|
8
|
+
resample,
|
|
9
|
+
grad2d,
|
|
10
|
+
surf_normal_area_elem,
|
|
11
|
+
dot_prod,
|
|
12
|
+
cross_prod,
|
|
13
|
+
normal_orientation,
|
|
14
|
+
)
|
|
15
|
+
from .kernels import (
|
|
16
|
+
laplace_fx_u,
|
|
17
|
+
laplace_fxd_u,
|
|
18
|
+
laplace_fxd2_u,
|
|
19
|
+
laplace_dx_u,
|
|
20
|
+
biotsavart_fx_u,
|
|
21
|
+
biotsavart_fxd_u,
|
|
22
|
+
)
|
|
23
|
+
from .integrals import (
|
|
24
|
+
laplace_fxd_u_eval,
|
|
25
|
+
laplace_fxd_u_eval_vec,
|
|
26
|
+
laplace_fxd_u_eval_singular,
|
|
27
|
+
laplace_fxd_u_eval_vec_singular,
|
|
28
|
+
laplace_fxd2_u_eval,
|
|
29
|
+
laplace_fxd2_u_eval_vec,
|
|
30
|
+
laplace_fxd2_u_eval_singular,
|
|
31
|
+
laplace_fxd2_u_eval_vec_singular,
|
|
32
|
+
field_period_target_coords,
|
|
33
|
+
biotsavart_fx_u_eval,
|
|
34
|
+
biotsavart_fxd_u_eval,
|
|
35
|
+
laplace_dx_u_eval,
|
|
36
|
+
computeB_offsurface_baseline,
|
|
37
|
+
computeB_offsurface_adaptive,
|
|
38
|
+
computeB_offsurface_adaptive_schedule,
|
|
39
|
+
computeGradB_offsurface_adaptive_schedule,
|
|
40
|
+
laplace_dx_u_eval_singular,
|
|
41
|
+
)
|
|
42
|
+
from .virtual_casing import VirtualCasingJAX
|
|
43
|
+
from .simsopt_virtual_casing import VirtualCasing
|
|
44
|
+
from .utils import build_offsurface_levels
|
|
45
|
+
from . import functional
|
|
46
|
+
from .functional import (
|
|
47
|
+
FunctionalSetup,
|
|
48
|
+
build_surface_coord,
|
|
49
|
+
build_quad_setup,
|
|
50
|
+
build_patch_idx,
|
|
51
|
+
select_patch_dim_from_geom,
|
|
52
|
+
target_surface_normal,
|
|
53
|
+
prepare_functional_setup,
|
|
54
|
+
compute_external_B_functional,
|
|
55
|
+
compute_external_B_normal_functional,
|
|
56
|
+
compute_external_B_jvp_columns_functional,
|
|
57
|
+
compute_external_B_normal_jvp_columns_functional,
|
|
58
|
+
compute_internal_B_functional,
|
|
59
|
+
compute_external_gradB_functional,
|
|
60
|
+
compute_internal_gradB_functional,
|
|
61
|
+
compute_external_B_offsurf_functional,
|
|
62
|
+
compute_external_gradB_offsurf_functional,
|
|
63
|
+
)
|
|
64
|
+
from . import singular_quadrature
|
|
65
|
+
from . import testdata
|
|
66
|
+
from .testdata import SurfType, VirtualCasingTestData, surface_coordinates, magnetic_field_data, magnetic_field_grad_data
|
|
67
|
+
|
|
68
|
+
__all__ = [
|
|
69
|
+
"surface_ops",
|
|
70
|
+
"complete_vec_field",
|
|
71
|
+
"rotate_toroidal",
|
|
72
|
+
"upsample",
|
|
73
|
+
"resample",
|
|
74
|
+
"grad2d",
|
|
75
|
+
"surf_normal_area_elem",
|
|
76
|
+
"dot_prod",
|
|
77
|
+
"cross_prod",
|
|
78
|
+
"normal_orientation",
|
|
79
|
+
"laplace_fx_u",
|
|
80
|
+
"laplace_fxd_u",
|
|
81
|
+
"laplace_fxd2_u",
|
|
82
|
+
"laplace_dx_u",
|
|
83
|
+
"biotsavart_fx_u",
|
|
84
|
+
"biotsavart_fxd_u",
|
|
85
|
+
"laplace_fxd_u_eval",
|
|
86
|
+
"laplace_fxd_u_eval_vec",
|
|
87
|
+
"laplace_fxd_u_eval_singular",
|
|
88
|
+
"laplace_fxd_u_eval_vec_singular",
|
|
89
|
+
"laplace_fxd2_u_eval",
|
|
90
|
+
"laplace_fxd2_u_eval_vec",
|
|
91
|
+
"laplace_fxd2_u_eval_singular",
|
|
92
|
+
"laplace_fxd2_u_eval_vec_singular",
|
|
93
|
+
"field_period_target_coords",
|
|
94
|
+
"biotsavart_fx_u_eval",
|
|
95
|
+
"biotsavart_fxd_u_eval",
|
|
96
|
+
"laplace_dx_u_eval",
|
|
97
|
+
"computeB_offsurface_baseline",
|
|
98
|
+
"computeB_offsurface_adaptive",
|
|
99
|
+
"computeB_offsurface_adaptive_schedule",
|
|
100
|
+
"computeGradB_offsurface_adaptive_schedule",
|
|
101
|
+
"laplace_dx_u_eval_singular",
|
|
102
|
+
"VirtualCasingJAX",
|
|
103
|
+
"VirtualCasing",
|
|
104
|
+
"build_offsurface_levels",
|
|
105
|
+
"functional",
|
|
106
|
+
"FunctionalSetup",
|
|
107
|
+
"build_surface_coord",
|
|
108
|
+
"build_quad_setup",
|
|
109
|
+
"build_patch_idx",
|
|
110
|
+
"select_patch_dim_from_geom",
|
|
111
|
+
"target_surface_normal",
|
|
112
|
+
"prepare_functional_setup",
|
|
113
|
+
"compute_external_B_functional",
|
|
114
|
+
"compute_external_B_normal_functional",
|
|
115
|
+
"compute_external_B_jvp_columns_functional",
|
|
116
|
+
"compute_external_B_normal_jvp_columns_functional",
|
|
117
|
+
"compute_internal_B_functional",
|
|
118
|
+
"compute_external_gradB_functional",
|
|
119
|
+
"compute_internal_gradB_functional",
|
|
120
|
+
"compute_external_B_offsurf_functional",
|
|
121
|
+
"compute_external_gradB_offsurf_functional",
|
|
122
|
+
"singular_quadrature",
|
|
123
|
+
"testdata",
|
|
124
|
+
"SurfType",
|
|
125
|
+
"VirtualCasingTestData",
|
|
126
|
+
"surface_coordinates",
|
|
127
|
+
"magnetic_field_data",
|
|
128
|
+
"magnetic_field_grad_data",
|
|
129
|
+
]
|