From e4695ded23b0b1d51619e8de5a0e2cc3ffb65578 Mon Sep 17 00:00:00 2001
From: giadarol <giovanni.iadarola@cern.ch>
Date: Fri, 23 Feb 2024 21:38:22 +0100
Subject: [PATCH] Good turn count

---
 examples/coasting/002_try_to_simplify.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/examples/coasting/002_try_to_simplify.py b/examples/coasting/002_try_to_simplify.py
index 485c6291f..073e1eaba 100644
--- a/examples/coasting/002_try_to_simplify.py
+++ b/examples/coasting/002_try_to_simplify.py
@@ -42,7 +42,6 @@ def track(self, particles):
 
         # Update zeta for particles that are stopped
         particles.zeta[mask_stop] += beta0_beta1 * self.circumference
-        particles.at_turn[mask_stop] += 1
         particles.pdg_id[mask_stop] += 1 # HACK!!!!!
 
         # Stop particles
@@ -123,8 +122,8 @@ def y_mean_hist(line, particles):
 plt.xlabel('Turn')
 
 plt.figure(2)
-plt.plot(p.delta, p.pdg_id, '.')
-plt.ylabel('Missing turns')
+plt.plot(p.delta, p.at_turn, '.')
+plt.ylabel('Number of turns')
 plt.xlabel(r'$\delta$')
 
 plt.figure(3)