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

Update zJn to use new scipy functions for further testing. 

Former-commit-id: 6b301eff9a6cf2f07d265145f54b55c0c74ecbb2
This commit is contained in:
Marek Nečada 2019-02-27 09:58:58 +02:00
parent e77fea8b7c
commit 0add6f9428
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
qpms/qpms_p.py
View File

@ -269,27 +269,27 @@ def vswf_yr(pos_sph,lMax,J=1):
return (M_y, N_y) return (M_y, N_y)
#@jit #@jit
def _sph_zn_1(n,z): def _sph_zn_1(n,z,derivative=False):
return spherical_jn(n,z) return spherical_jn(n,z,derivative)
#@jit #@jit
def _sph_zn_2(n,z): def _sph_zn_2(n,z,derivative=False):
return spherical_yn(n,z) return spherical_yn(n,z,derivative)
#@jit #@jit
def _sph_zn_3(n,z): def _sph_zn_3(n,z,derivative=False):
besj=spherical_jn(n,z) besj=spherical_jn(n,z,derivative)
besy=spherical_yn(n,z) besy=spherical_yn(n,z,derivative)
return (besj[0] + 1j*besy[0],besj[1] + 1j*besy[1]) return besj + 1j*besy
#@jit #@jit
def _sph_zn_4(n,z): def _sph_zn_4(n,z,derivative=False):
besj=spherical_jn(n,z) besj=spherical_jn(n,z,derivative)
besy=spherical_yn(n,z) besy=spherical_yn(n,z,derivative)
return (besj[0] - 1j*besy[0],besj[1] - 1j*besy[1]) return besj + 1j*besy
_sph_zn = [_sph_zn_1,_sph_zn_2,_sph_zn_3,_sph_zn_4] _sph_zn = [_sph_zn_1,_sph_zn_2,_sph_zn_3,_sph_zn_4]
# computes bessel/hankel functions for orders from 0 up to n; # computes bessel/hankel functions for orders from 0 up to n;
#@jit #@jit
def zJn(n, z, J=1): def zJn(n, z, J=1):
return _sph_zn[J-1](n=n,z=z) return (_sph_zn[J-1](n=n,z=z,derivative=False), _sph_zn[J-1](n=n,z=z,derivative=True))