From 268bdd7edbef7166daee58bb634c4b9d9f2615a3 Mon Sep 17 00:00:00 2001
From: Alexis Gamelin <alexis.gamelin@synchrotron-soleil.fr>
Date: Tue, 23 Apr 2024 18:37:22 +0200
Subject: [PATCH] [Fix] BeamLoadingEquilibrium
Change location of update_rho()
---
mbtrack2/utilities/beamloading.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mbtrack2/utilities/beamloading.py b/mbtrack2/utilities/beamloading.py
index d8d64b1..041741b 100644
--- a/mbtrack2/utilities/beamloading.py
+++ b/mbtrack2/utilities/beamloading.py
@@ -107,7 +107,6 @@ class BeamLoadingEquilibrium():
def center_of_mass(self):
"""Return center of mass position in [s]"""
- self.update_rho()
CM = np.average(self.z0, weights=self.rho0)
return CM / c
@@ -166,6 +165,7 @@ class BeamLoadingEquilibrium():
else:
self.F = x[::2]
self.PHI = x[1::2]
+ self.update_rho()
# Compute system
if self.auto_set_MC_theta:
--
GitLab