Use time directly if available

xsuite · Jan 2, 2024 · 8f63723 · 8f63723
1 parent 0147cf3
commit 8f63723
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/xplt/particles.py b/xplt/particles.py
@@ -130,6 +130,15 @@ def frev(self, particles=None):
 
     def _particle_time(self, turn, zeta, particles=None):
         """Particle arrival time (t = at_turn / frev - zeta / beta / c0)"""
+
+        # use time directly (if available)
+        if particles is not None:
+            try:
+                return get(particles, "t")
+            except AttributeError:
+                pass
+
+        # determine time from longitudinal coordinates
         beta = self.beta(particles)
         if beta is None:
             raise ValueError(