Fix mistakenly removed cos

desy-ml · Sep 11, 2023 · 940dbcc · 940dbcc
1 parent 249aa21
commit 940dbcc
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 36 deletions.
diff --git a/cheetah/accelerator.py b/cheetah/accelerator.py
@@ -927,7 +927,7 @@ def _track_beam(self, incoming: ParticleBeam) -> ParticleBeam:
     def _cavity_rmatrix(self, energy: float) -> torch.Tensor:
         """Produces an R-matrix for a cavity when it is on, i.e. voltage > 0.0."""
         phi = torch.deg2rad(torch.tensor(self.phase))
-        delta_energy = torch.tensor(self.voltage) * phi
+        delta_energy = torch.tensor(self.voltage) * torch.cos(phi)
         # Comment from Ocelot: Pure pi-standing-wave case
         eta = torch.tensor(1)
         Ei = energy / electron_mass_eV

diff --git a/test_bmad.ipynb b/test_bmad.ipynb