From f1e7dea9148231c55dac796ec4647df93d19e221 Mon Sep 17 00:00:00 2001
From: Gamelin Alexis <alexis.gamelin@synchrotron-soleil.fr>
Date: Fri, 21 Apr 2023 23:08:53 +0200
Subject: [PATCH] [Fix] Bug in BeamLoadingEquilibrium class
Fix a hard coded value of 0.5 A ...
---
mbtrack2/utilities/beamloading.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/mbtrack2/utilities/beamloading.py b/mbtrack2/utilities/beamloading.py
index 3dbc7d5..84a23a1 100644
--- a/mbtrack2/utilities/beamloading.py
+++ b/mbtrack2/utilities/beamloading.py
@@ -138,7 +138,7 @@ class BeamLoadingEquilibrium():
cavity = self.cavity_list[i]
if cavity.m == 1:
cavity.theta = x[-1]
- cavity.set_generator(0.5)
+ cavity.set_generator(self.I0)
self.update_potentials()
else:
self.F = x[::2]
@@ -180,8 +180,8 @@ class BeamLoadingEquilibrium():
if z2 is None:
z2 = self.B2
z0 = np.linspace(z1, z2, 1000)
- plt.plot(z0, self.rho(z0))
- plt.xlabel("z [m]")
+ plt.plot(z0/c*1e12, self.rho(z0))
+ plt.xlabel(r"$\tau$ [ps]")
plt.title("Equilibrium bunch profile")
def voltage(self, z):
--
GitLab