Fix missing index arrays

Former-commit-id: 17499c3583715bf5365771c60c66ba58f175a67a
This commit is contained in:
Marek Nečada 2017-05-10 13:44:36 +03:00
parent faa01fcceb
commit 7a4887101a
1 changed files with 2 additions and 1 deletions

View File

@ -951,7 +951,8 @@ def symz_indexarrays(lMax, npart = 1):
TEč, TMč : (npart * 2 * nelem)-shaped bool ndarray
Mask arrays corresponding to the 'TE' and 'TM' modes, respectively.
"""
nelem = lMax * (lMax + 2)
my, ny = get_mn_y(lMax)
nelem = len(my)
ž = np.arange(2*nelem) # single particle spherical wave indices
= ž // nelem # tž == 0: electric waves, tž == 1: magnetic waves
= my[ž%nelem]