Add more lattice elements in cheetah

desy-ml / cheetah

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.

https://cheetah-accelerator.readthedocs.io

GNU General Public License v3.0

28 stars 13 forks source link

Closed cr-xu closed 1 year ago

cr-xu commented 1 year ago

Added the following lattice elements:

Dipole (Sector, Rectangular)
Aperture (Rectangular, Elliptical)
- blocks the beam with a large offset if tracking with ParticleBeam, otherwise ignored
- needs to be activated by self.is_active = True
Solenoid

Added tilt for Quadrupole , set tilt to pi/4 for skew-quadrupole

Moved some repeated transfer map calculations into a separate file: track_method.py

Addresses #42

cr-xu commented 1 year ago

Now added tests for Dipole (w & w/o fringe field), Solenoid, and Aperture

jank324 commented 1 year ago

Looks good! Did testing actually expose bugs? 😄

cr-xu commented 1 year ago

Looks good! Did testing actually expose bugs? 😄

Actually yes 😂 so indeed having proper tests is important.