aboutsummaryrefslogtreecommitdiff
path: root/Python/pywarpx/PGroup.py
diff options
context:
space:
mode:
authorGravatar Dave Grote <grote1@llnl.gov> 2019-05-24 14:58:43 -0700
committerGravatar Dave Grote <grote1@llnl.gov> 2019-06-14 16:22:32 -0700
commit2f09a9943892cb620fff2e745cb414687bfa1854 (patch)
tree0305ee27a6aad090e86480d2ad39093db4c2ea1b /Python/pywarpx/PGroup.py
parent5e5855854d91ab328976d1d5c782826191185c3b (diff)
downloadWarpX-2f09a9943892cb620fff2e745cb414687bfa1854.tar.gz
WarpX-2f09a9943892cb620fff2e745cb414687bfa1854.tar.zst
WarpX-2f09a9943892cb620fff2e745cb414687bfa1854.zip
Update Python wrapper to handle RZ version
Diffstat (limited to 'Python/pywarpx/PGroup.py')
-rw-r--r--Python/pywarpx/PGroup.py8
1 files changed, 8 insertions, 0 deletions
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