diff --git a/mbtrack2/tracking/element.py b/mbtrack2/tracking/element.py index 752c044c71322585f056a3d4e467a3e23274715c..a46c6b889f0f9f05660d5826dc548fed8ca3434a 100644 --- a/mbtrack2/tracking/element.py +++ b/mbtrack2/tracking/element.py @@ -166,20 +166,21 @@ class SynchrotronRadiation(Element): ---------- bunch : Bunch or Beam object """ + N = len(bunch) if self.switch[0] == True: - rand = np.random.normal(size=len(bunch)) + rand = np.random.standard_normal(size=N) bunch["delta"] = (1 - 2 * self.ring.T0 / self.ring.tau[2]) * bunch[ "delta"] + 2 * self.ring.sigma_delta * ( self.ring.T0 / self.ring.tau[2])**0.5 * rand if self.switch[1] == True: - rand = np.random.normal(size=len(bunch)) + rand = np.random.standard_normal(size=N) bunch["xp"] = (1 - 2 * self.ring.T0 / self.ring.tau[0] ) * bunch["xp"] + 2 * self.ring.sigma()[1] * ( self.ring.T0 / self.ring.tau[0])**0.5 * rand if self.switch[2] == True: - rand = np.random.normal(size=len(bunch)) + rand = np.random.standard_normal(size=N) bunch["yp"] = (1 - 2 * self.ring.T0 / self.ring.tau[1] ) * bunch["yp"] + 2 * self.ring.sigma()[3] * ( self.ring.T0 / self.ring.tau[1])**0.5 * rand