diff options
| author |  Dave Grote <grote1@llnl.gov>
| 2019-05-24 14:58:43 -0700 |
|---|---|---|
| committer | Dave Grote <grote1@llnl.gov>
| 2019-05-24 14:58:43 -0700 |
| commit | 24e3acabb41d915b7d5d61f633e676590a0ca15e (patch) | |
| tree | 5c598a2877c9b665901393cbc60ae7692e48588f /Python/pywarpx/PGroup.py | |
| parent | 78dd050f9d75fafd9b7333ac173df5ed6667a4f5 (diff) | |
| download | WarpX-24e3acabb41d915b7d5d61f633e676590a0ca15e.tar.gz WarpX-24e3acabb41d915b7d5d61f633e676590a0ca15e.tar.zst WarpX-24e3acabb41d915b7d5d61f633e676590a0ca15e.zip | |
Update Python wrapper to handle RZ version
Diffstat (limited to 'Python/pywarpx/PGroup.py')
| -rw-r--r-- | Python/pywarpx/PGroup.py | 8 |
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 |