Skip to content

En/review eg/fields surfaces objective functions mod #40

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
eduardolneto merged 8 commits into from
Jul 7, 2026
+469 −464
Merged

En/review eg/fields surfaces objective functions mod #40

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
ab1fac8
Small changes on fields.py class near_axis to work as a dependecy of …
eduardolneto Jun 27, 2026
eb72e45
Modified objective_functions.py to have the B dot N losses up to date…
eduardolneto Jul 2, 2026
8a7074d
Correncting timestep input on some of the particle objectives and als…
eduardolneto Jul 2, 2026
04685b8
Adding block_size to loretnz force objective
eduardolneto Jul 2, 2026
7097381
Changing coils optimization example slightly
eduardolneto Jul 2, 2026
2059260
Added example for coil-coild, surface-coil distance, likingnumber and…
eduardolneto Jul 2, 2026
8f118bc
Addressing some remaining bugs
eduardolneto Jul 3, 2026
51db1de
Merge branch 'eg/analysis' into en/review_eg/fields_surfaces_objectiv…
eduardolneto Jul 7, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions essos/fields.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -390,8 +390,8 @@ def to_xyz(self, points):
return jnp.array([X, Y, Z])

class near_axis():
def __init__(self, rc=jnp.array([1, 0.1]), zs=jnp.array([0, 0.1]), etabar=1.0,
B0=1, sigma0=0, I2=0, nphi=31, spsi=1, sG=1, nfp=2, order='r1', B2c=0, p2=0):
def __init__(self, rc=jnp.array([1., 0.1]), zs=jnp.array([0., 0.1]), etabar=1.0,
B0=1., sigma0=0., I2=0., nphi=31, spsi=1., sG=1., nfp=2, order='r1', B2c=0., p2=0.):
assert nphi % 2 == 1, 'nphi must be odd'
self.rc = jnp.array(rc)
self.zs = jnp.array(zs)
Expand Down Expand Up @@ -424,7 +424,8 @@ def dofs(self):

@dofs.setter
def dofs(self, new_dofs):
self._dofs = jnp.array(new_dofs)
# Ensure dofs are always float for JAX autodiff compatibility
self._dofs = jnp.array(new_dofs, dtype=float)
Comment thread
EstevaoMGomes marked this conversation as resolved.
self.rc = self._dofs[:self.nfourier]
self.zs = self._dofs[self.nfourier:2*self.nfourier]
self.etabar = self._dofs[-1]
Expand Down Expand Up @@ -954,4 +955,4 @@ def to_vtk(self, filename, r=0.1, ntheta=40, nphi=120, ntheta_fourier=20, extra_

tree_util.register_pytree_node(near_axis,
near_axis._tree_flatten,
near_axis._tree_unflatten)
near_axis._tree_unflatten)
Loading