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-05-24 14:58:43 -0700
commit24e3acabb41d915b7d5d61f633e676590a0ca15e (patch)
tree5c598a2877c9b665901393cbc60ae7692e48588f /Python/pywarpx/picmi.py
parent78dd050f9d75fafd9b7333ac173df5ed6667a4f5 (diff)
downloadWarpX-24e3acabb41d915b7d5d61f633e676590a0ca15e.tar.gz
WarpX-24e3acabb41d915b7d5d61f633e676590a0ca15e.tar.zst
WarpX-24e3acabb41d915b7d5d61f633e676590a0ca15e.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 f242f8589..151de388d 100644
--- a/Python/pywarpx/picmi.py
+++ b/Python/pywarpx/picmi.py
@@ -256,13 +256,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