From 24e3acabb41d915b7d5d61f633e676590a0ca15e Mon Sep 17 00:00:00 2001 From: Dave Grote Date: Fri, 24 May 2019 14:58:43 -0700 Subject: Update Python wrapper to handle RZ version --- Python/pywarpx/PGroup.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'Python/pywarpx/PGroup.py') diff --git a/Python/pywarpx/PGroup.py b/Python/pywarpx/PGroup.py index 68b37740d..48e68ceb5 100644 --- a/Python/pywarpx/PGroup.py +++ b/Python/pywarpx/PGroup.py @@ -83,6 +83,10 @@ class PGroup(object): return _libwarpx.get_particle_y(self.ispecie)[self.igroup] yp = property(getyp) + def getrp(self): + return _libwarpx.get_particle_r(self.ispecie)[self.igroup] + rp = property(getrp) + def getzp(self): return _libwarpx.get_particle_z(self.ispecie)[self.igroup] zp = property(getzp) @@ -136,6 +140,10 @@ class PGroup(object): return _libwarpx.get_particle_Bz(self.ispecie)[self.igroup] bz = property(getbz) + def gettheta(self): + return _libwarpx.get_particle_theta(self.ispecie)[self.igroup] + theta = property(gettheta) + class PGroups(object): def __init__(self, ispecie=0): self.ispecie = ispecie -- cgit v1.2.3