Extending codes for ParticleGroup at https://github.com/ChristopherMayes/openPMD-beamphysics/tree/master/pmd_beamphysics - View it on GitHub