aboutsummaryrefslogtreecommitdiff
path: root/Python/pywarpx/picmi.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/picmi.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/picmi.py')
-rw-r--r--Python/pywarpx/picmi.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/Python/pywarpx/picmi.py b/Python/pywarpx/picmi.py
index f6dc47fec..80d8e01e7 100644
--- a/Python/pywarpx/picmi.py
+++ b/Python/pywarpx/picmi.py
@@ -259,13 +259,13 @@ class CylindricalGrid(picmistandard.PICMI_CylindricalGrid):
assert self.lower_bound[0] >= 0., Exception('Lower radial boundary must be >= 0.')
assert self.bc_rmin != 'periodic' and self.bc_rmax != 'periodic', Exception('Radial boundaries can not be periodic')
- assert self.n_azimuthal_modes is None or self.n_azimuthal_modes == 1, Exception('Only one azimuthal mode supported')
# Geometry
pywarpx.geometry.coord_sys = 1 # RZ
pywarpx.geometry.is_periodic = '0 %d'%(self.bc_zmin=='periodic') # Is periodic?
pywarpx.geometry.prob_lo = self.lower_bound # physical domain
pywarpx.geometry.prob_hi = self.upper_bound
+ pywarpx.warpx.nmodes = self.n_azimuthal_modes
if self.moving_window_velocity is not None and np.any(np.not_equal(self.moving_window_velocity, 0.)):
pywarpx.warpx.do_moving_window = 1