From 9c4890d885f237dc0122d77c0071924f3f4ba9bc Mon Sep 17 00:00:00 2001
From: Gamelin Alexis <gamelin@synchrotron-soleil.fr>
Date: Mon, 26 Oct 2020 16:32:04 +0100
Subject: [PATCH] Bugfix in SynchrotronRadiation Element
---
tracking/element.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/tracking/element.py b/tracking/element.py
index 519190e..0ef071c 100644
--- a/tracking/element.py
+++ b/tracking/element.py
@@ -129,15 +129,15 @@ class SynchrotronRadiation(Element):
if (self.switch[1] == True):
rand = np.random.normal(size=(len(bunch),2))
- bunch["x"] += self.ring.sigma[0]*(2*self.ring.T0/self.ring.tau[0])**0.5*rand[:,0]
+ bunch["x"] += self.ring.sigma()[0]*(2*self.ring.T0/self.ring.tau[0])**0.5*rand[:,0]
bunch["xp"] = (1 + bunch["delta"])/(1 + bunch["delta"] + bunch.energy_change)*bunch["xp"]
- bunch["xp"] += self.ring.sigma[1]*(2*self.ring.T0/self.ring.tau[0])**0.5*rand[:,1]
+ bunch["xp"] += self.ring.sigma()[1]*(2*self.ring.T0/self.ring.tau[0])**0.5*rand[:,1]
if (self.switch[2] == True):
rand = np.random.normal(size=(len(bunch),2))
- bunch["y"] += self.ring.sigma[2]*(2*self.ring.T0/self.ring.tau[1])**0.5*rand[:,0]
+ bunch["y"] += self.ring.sigma()[2]*(2*self.ring.T0/self.ring.tau[1])**0.5*rand[:,0]
bunch["yp"] = (1 + bunch["delta"])/(1 + bunch["delta"] + bunch.energy_change)*bunch["yp"]
- bunch["yp"] += self.ring.sigma[3]*(2*self.ring.T0/self.ring.tau[1])**0.5*rand[:,1]
+ bunch["yp"] += self.ring.sigma()[3]*(2*self.ring.T0/self.ring.tau[1])**0.5*rand[:,1]
# Reset energy change to 0 for next turn
bunch.energy_change = 0
--
GitLab