xsuite/xtrack

Python package for tracking simulations in particle accelerators on CPU and GPU.

xtrack

Python package for tracking simulations in particle accelerators on CPU and GPU.

xtrack's Issues

The xfields dependency is not declared

When importing anything from xtrack, the __init__.py is loaded which loads from xtrack.multiline.multiline and this module imports xfields, which is not declared as a dependency and not installed with xtrack.

A simple fix would be to add xfields in the dependencies in setup.py.

pzeta prints a warning

xtrack/xtrack/particles/particles.py

Lines 1273 to 1278 in 8d7a2e7

	@property
	def pzeta(self):
	pzeta = self.ptau / self.beta0
	return self._buffer.context.linked_array_type.from_array(
	pzeta, mode='readonly',
	container=self)

The pzeta calculation should automatically return nan for particles with state==0 without doing the division by beta0==0. Current behaviour prints a warning:

Example:

import xtrack as xt

m = xt.ParticlesMonitor(num_particles=2, start_at_turn=0, stop_at_turn=10)

# Reseting monitor data
with m.data._bypass_linked_vars():
    for tt, nn in m._ParticlesClass.per_particle_vars:
        getattr(m.data, nn)[:] = 0

# Accessing pzeta
m.pzeta

-------------------------------
RuntimeWarning: invalid value encountered in divide
  pzeta = self.ptau / self.beta0