def _epsn(self, u, up, beta, gamma): index = self.first_particle_index_in_slice stats = np.zeros(self.n_slices) for i in xrange(self.n_slices): k = u[index[i]:index[i + 1]] kp = up[index[i]:index[i + 1]] stats[i] = cp.emittance(k, kp) * gamma * beta * 1e6 return stats
def epsn_y(self): return cp.emittance(self.y, self.yp) * self.gamma * self.beta * 1e6
def epsn_x(self): return cp.emittance(self.x, self.xp) * self.gamma * self.beta * 1e6