Fix the transfer map of `Drift` and edge focusing of `Dipole`

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

33 stars 13 forks source link

Fix the transfer map of `Drift` and edge focusing of `Dipole` #90

Closed cr-xu closed 1 year ago

cr-xu commented 1 year ago

Fix the R_56 component of the transfer map in Drift;
Updated the test functions, now the difference should be less.
Add R_56 component in VerticalCorrector and HorizontalCorrector
Fix the edge focusing transfer maps for bends (e.g. Dipole RBend and SBend)
Removed the large tolerances for result comparison in test functions

cr-xu commented 1 year ago

@jank324 This probably also helps to reduce the discrepancy between Bmad and cheetah tracking. There's still a problem with the Bend though, I'm trying to look into it.