lemoer / vswf Goto Github PK

 def coord_s2c(r, theta, phi):
     x = r * np.sin(theta) * np.cos(phi)
     y = r * np.sin(theta) * np.sin(phi)
@@ -274,7 +266,7 @@
     elif c == 4:
         rl = spherical_hn2

-    _, vsh_lm_theta, vsh_lm_phi = vec_sph_harm(tau, l, m, theta, phi)
+    _, vsh_lm_theta, vsh_lm_phi = vec_sph_harm(tau, l, m, theta, phi) # This is either called Xlm or Zlm (depending on tau)

     if tau == 1:
         rl_res = rl(l, k*r)
@@ -284,18 +276,18 @@
         f_r = 0
     elif tau == 2:
         rl_intermed = rl(l, k*r)/(k*r)
-        rl_res = rl_intermed + rl(l, k*r, derivative=True)
+        rl_res = rl_intermed + rl(l, k*r, derivative=True) #rl(l, k*r) # rl(l, k*r, derivative=True) #

-        f_theta = 1j*rl_res * vsh_lm_theta
-        f_phi = 1j*rl_res * vsh_lm_phi
-        f_r = -rl_intermed*np.sqrt(l*(l+1))*my_sph_harm(l, m, theta, phi)
+        f_theta = rl_res * vsh_lm_theta
+        f_phi = rl_res * vsh_lm_phi
+        f_r = rl_intermed*np.sqrt(l*(l+1))*my_sph_harm(l, m, theta, phi)
     else:
         raise NotImplemented('NIY')

     return f_r, f_theta, f_phi