Skip to content

DRAFT: Spin tracking2 #164

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

Open
Eirhorlyk wants to merge 4 commits into
base: main
Choose a base branch
from
Open

DRAFT: Spin tracking2 #164

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
291b16f
Added spin polarization to abel/classes/source/source.py.
Eirhorlyk Nov 4, 2025
119dd6a
Added new stage class StageSpin.
Eirhorlyk Nov 4, 2025
baf2dbe
Added new physics model for spin tracking.
Eirhorlyk Nov 4, 2025
76c9793
Added StageSpin to abel/__init__.py.
Eirhorlyk Nov 4, 2025
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
3 changes: 3 additions & 0 deletions abel/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,9 @@
from .classes.source.impl.source_flattop import SourceFlatTop
from .classes.source.impl.source_capsule import SourceCapsule
from .classes.stage.impl.stage_basic import StageBasic

from .classes.stage.impl.stage_spin import StageSpin

from .classes.stage.impl.stage_hipace import StageHipace
from .classes.stage.impl.stage_wake_t import StageWakeT
from .classes.stage.impl.stage_quasistatic_2d import StageQuasistatic2d
Expand Down
10 changes: 5 additions & 5 deletions abel/classes/source/source.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ class Source(Trackable, CostModeled):
#TODO: Why is accel_gradient needed??

@abstractmethod
def __init__(self, length=0, charge=None, energy=None, accel_gradient=None, wallplug_efficiency=1, x_offset=0, y_offset=0, x_angle=0, y_angle=0, norm_jitter_emittance_x=None, norm_jitter_emittance_y=None, waist_shift_x=0, waist_shift_y=0, rep_rate_trains=None, num_bunches_in_train=None, bunch_separation=None):
def __init__(self, length=0, charge=None, energy=None, accel_gradient=None, wallplug_efficiency=1, x_offset=0, y_offset=0, x_angle=0, y_angle=0, norm_jitter_emittance_x=None, norm_jitter_emittance_y=None, waist_shift_x=0, waist_shift_y=0, rep_rate_trains=None, num_bunches_in_train=None, bunch_separation=None, spin_polarization=0.0, spin_polarization_direction='z'):

super().__init__(num_bunches_in_train=num_bunches_in_train, bunch_separation=bunch_separation, rep_rate_trains=rep_rate_trains)

Expand All @@ -83,6 +83,9 @@ def __init__(self, length=0, charge=None, energy=None, accel_gradient=None, wall
self.charge = charge
self.accel_gradient = accel_gradient
self.wallplug_efficiency = wallplug_efficiency

self.spin_polarization = spin_polarization
self.spin_polarization_direction = spin_polarization_direction

self.x_offset = x_offset
self.y_offset = y_offset
Expand All @@ -104,10 +107,7 @@ def __init__(self, length=0, charge=None, energy=None, accel_gradient=None, wall
self.jitter.yp = 0
self.jitter.E = 0

self.spin_polarization = 0
self.spin_polarization_direction = 'z'

self.is_polarized = False #TODO shouldn't this rather be a function that checks whether spin_polarization > 0?
self.is_polarized = False # TODO write a function instead to check this.


@abstractmethod
Expand Down
Loading