QPMS
/
qpms
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Pseudovectorisation of qpms_p.lpy() 

Former-commit-id: f09d71c52ef73453ee1a49f327a5b52548ce978b
This commit is contained in:
Marek Nečada 2017-07-20 17:30:24 +03:00
parent 46c67a548e
commit 5ef938ac11
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
qpms/qpms_p.py
View File

@ -182,8 +182,18 @@ def nelem2lMax(nelem):
#@jit
def lpy(nmax, z): def lpy(nmax, z):
# TODO TRUEVECTORIZE
# TODO DOC
nelem = nmax * (nmax+2)
z = np.array(z, copy=False)
P_y = np.empty(z.shape + (nelem,), dtype=np.float_)
dP_y = np.empty(z.shape + (nelem,), dtype=np.float_)
for i in np.ndindex(z.shape):
P_y[i], dP_y[i] = lpy1(nmax, z[i])
return (P_y, dP_y)
def lpy1(nmax, z):
""" """
Associated legendre function and its derivatative at z in the 'y-indexing'. Associated legendre function and its derivatative at z in the 'y-indexing'.
(Without Condon-Shortley phase AFAIK.) (Without Condon-Shortley phase AFAIK.)
@ -205,8 +215,8 @@ def lpy(nmax, z):
pmn_plus, dpmn_plus = lpmn(nmax, nmax, z) pmn_plus, dpmn_plus = lpmn(nmax, nmax, z)
pmn_minus, dpmn_minus = lpmn(-nmax, nmax, z) pmn_minus, dpmn_minus = lpmn(-nmax, nmax, z)
nelem = nmax * nmax + 2*nmax nelem = nmax * nmax + 2*nmax
P_y = np.empty((nelem), dtype=np.float_) P_y = np.empty((nelem,), dtype=np.float_)
dP_y = np.empty((nelem), dtype=np.float_) dP_y = np.empty((nelem,), dtype=np.float_)
mn_p_y, mn_n_y = get_y_mn_unsigned(nmax) mn_p_y, mn_n_y = get_y_mn_unsigned(nmax)
mn_plus_mask = (mn_p_y >= 0) mn_plus_mask = (mn_p_y >= 0)
mn_minus_mask = (mn_n_y >= 0) mn_minus_mask = (mn_n_y >= 0)