|
@@ -21,28 +21,13 @@
|
|
|
#
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
-# distutils: language = c++
|
|
|
-# distutils: sources = nmie-wrapper.cc
|
|
|
+
|
|
|
from __future__ import division
|
|
|
import numpy as np
|
|
|
cimport numpy as np
|
|
|
from libcpp.vector cimport vector
|
|
|
from libcpp.vector cimport complex
|
|
|
|
|
|
-# cdef extern from "<vector>" namespace "std":
|
|
|
-# cdef cppclass vector[T]:
|
|
|
-# cppclass iterator:
|
|
|
-# T operator*()
|
|
|
-# iterator operator++()
|
|
|
-# bint operator==(iterator)
|
|
|
-# bint operator!=(iterator)
|
|
|
-# vector()
|
|
|
-# void push_back(T&)
|
|
|
-# T& operator[](int)
|
|
|
-# T& at(int)
|
|
|
-# iterator begin()
|
|
|
-# iterator end()
|
|
|
-
|
|
|
cdef inline double *npy2c(np.ndarray a):
|
|
|
assert a.dtype == np.float64
|
|
|
|
|
@@ -51,112 +36,12 @@ cdef inline double *npy2c(np.ndarray a):
|
|
|
|
|
|
# Return data pointer
|
|
|
return <double *>(a.data)
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-
|
|
|
-# cdef extern from "py_nmie.h":
|
|
|
-# cdef int nMie(int L, int pl, vector[double] x, vector[double complex] m, int nTheta, vector[double] Theta, int nmax, double *Qext, double *Qsca, double *Qabs, double *Qbk, double *Qpr, double *g, double *Albedo, double S1r[], double S1i[], double S2r[], double S2i[])
|
|
|
-# cdef int nField(int L, int pl, vector[double] x, vector[double complex] m, int nmax, int nCoords, vector[double] Xp, vector[double] Yp, vector[double] Zp, double Erx[], double Ery[], double Erz[], double Eix[], double Eiy[], double Eiz[], double Hrx[], double Hry[], double Hrz[], double Hix[], double Hiy[], double Hiz[])
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-
|
|
|
-cdef extern from "nmie-wrapper.h" namespace "nmie":
|
|
|
- cdef int nMie_wrapper(int L, const vector[double] x, const vector[double complex] m , int nTheta, const vector[double] Theta, double *qext, double *Qsca, double *Qabs, double *Qbk, double *Qpr, double *g, double *Albedo, vector[double complex] S1, vector[double complex] S2);
|
|
|
-
|
|
|
-
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-# def scattnlay(np.ndarray[np.float64_t, ndim = 2] x, np.ndarray[np.complex128_t, ndim = 2] m, np.ndarray[np.float64_t, ndim = 1] theta = np.zeros(0, dtype = np.float64), np.int_t pl = -1, np.int_t nmax = -1):
|
|
|
-# cdef Py_ssize_t i
|
|
|
-
|
|
|
-# cdef np.ndarray[np.int_t, ndim = 1] terms = np.zeros(x.shape[0], dtype = np.int)
|
|
|
-
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Qext = np.zeros(x.shape[0], dtype = np.float64)
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Qabs = np.zeros(x.shape[0], dtype = np.float64)
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Qsca = np.zeros(x.shape[0], dtype = np.float64)
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Qbk = np.zeros(x.shape[0], dtype = np.float64)
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Qpr = np.zeros(x.shape[0], dtype = np.float64)
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] g = np.zeros(x.shape[0], dtype = np.float64)
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Albedo = np.zeros(x.shape[0], dtype = np.float64)
|
|
|
-
|
|
|
-# cdef np.ndarray[np.complex128_t, ndim = 2] S1 = np.zeros((x.shape[0], theta.shape[0]), dtype = np.complex128)
|
|
|
-# cdef np.ndarray[np.complex128_t, ndim = 2] S2 = np.zeros((x.shape[0], theta.shape[0]), dtype = np.complex128)
|
|
|
-
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] S1r
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] S1i
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] S2r
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] S2i
|
|
|
-
|
|
|
-# for i in range(x.shape[0]):
|
|
|
-# S1r = np.zeros(theta.shape[0], dtype = np.float64)
|
|
|
-# S1i = np.zeros(theta.shape[0], dtype = np.float64)
|
|
|
-# S2r = np.zeros(theta.shape[0], dtype = np.float64)
|
|
|
-# S2i = np.zeros(theta.shape[0], dtype = np.float64)
|
|
|
-
|
|
|
-# terms[i] = nMie(x.shape[1], pl, x[i].copy('C'), m[i].copy('C'), theta.shape[0], theta.copy('C'), nmax, &Qext[i], &Qsca[i], &Qabs[i], &Qbk[i], &Qpr[i], &g[i], &Albedo[i], npy2c(S1r), npy2c(S1i), npy2c(S2r), npy2c(S2i))
|
|
|
-
|
|
|
-# S1[i] = S1r.copy('C') + 1.0j*S1i.copy('C')
|
|
|
-# S2[i] = S2r.copy('C') + 1.0j*S2i.copy('C')
|
|
|
-
|
|
|
-# return terms, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2
|
|
|
-# ##############################################################################
|
|
|
-# ##############################################################################
|
|
|
-# ##############################################################################
|
|
|
-# ##############################################################################
|
|
|
-
|
|
|
-# #def fieldnlay(np.ndarray[np.float64_t, ndim = 2] x, np.ndarray[np.complex128_t, ndim = 2] m, np.ndarray[np.float64_t, ndim = 2] coords = np.zeros((0, 3), dtype = np.float64), np.int_t pl = 0, np.int_t nmax = 0):
|
|
|
-# def fieldnlay(np.ndarray[np.float64_t, ndim = 2] x, np.ndarray[np.complex128_t, ndim = 2] m, np.ndarray[np.float64_t, ndim = 2] coords, np.int_t pl = 0, np.int_t nmax = 0):
|
|
|
-# cdef Py_ssize_t i
|
|
|
-
|
|
|
-# cdef np.ndarray[np.int_t, ndim = 1] terms = np.zeros(x.shape[0], dtype = np.int)
|
|
|
-
|
|
|
-# cdef np.ndarray[np.complex128_t, ndim = 3] E = np.zeros((x.shape[0], coords.shape[0], 3), dtype = np.complex128)
|
|
|
-# cdef np.ndarray[np.complex128_t, ndim = 3] H = np.zeros((x.shape[0], coords.shape[0], 3), dtype = np.complex128)
|
|
|
-
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Erx
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Ery
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Erz
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Eix
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Eiy
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Eiz
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Hrx
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Hry
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Hrz
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Hix
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Hiy
|
|
|
-# cdef np.ndarray[np.float64_t, ndim = 1] Hiz
|
|
|
-
|
|
|
-# for i in range(x.shape[0]):
|
|
|
-# Erx = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Ery = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Erz = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Eix = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Eiy = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Eiz = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Hrx = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Hry = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Hrz = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Hix = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Hiy = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-# Hiz = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
-
|
|
|
-# terms[i] = nField(x.shape[1], pl, x[i].copy('C'), m[i].copy('C'), nmax, coords.shape[0], coords[:, 0].copy('C'), coords[:, 1].copy('C'), coords[:, 2].copy('C'), npy2c(Erx), npy2c(Ery), npy2c(Erz), npy2c(Eix), npy2c(Eiy), npy2c(Eiz), npy2c(Hrx), npy2c(Hry), npy2c(Hrz), npy2c(Hix), npy2c(Hiy), npy2c(Hiz))
|
|
|
-
|
|
|
-# E[i] = np.vstack((Erx.copy('C') + 1.0j*Eix.copy('C'), Ery.copy('C') + 1.0j*Eiy.copy('C'), Erz.copy('C') + 1.0j*Eiz.copy('C'))).transpose()
|
|
|
-# H[i] = np.vstack((Hrx.copy('C') + 1.0j*Hix.copy('C'), Hry.copy('C') + 1.0j*Hiy.copy('C'), Hrz.copy('C') + 1.0j*Hiz.copy('C'))).transpose()
|
|
|
-
|
|
|
-# return terms, E, H
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-##############################################################################
|
|
|
-def scattnlay_wrapper(np.ndarray[np.float64_t, ndim = 2] x, np.ndarray[np.complex128_t, ndim = 2] m, np.ndarray[np.float64_t, ndim = 1] theta = np.zeros(0, dtype = np.float64), np.int_t pl = -1, np.int_t nmax = -1):
|
|
|
+
|
|
|
+cdef extern from "py_nmie.h":
|
|
|
+ cdef int nMie(int L, int pl, vector[double] x, vector[complex] m, int nTheta, vector[double] Theta, int nmax, double *Qext, double *Qsca, double *Qabs, double *Qbk, double *Qpr, double *g, double *Albedo, double S1r[], double S1i[], double S2r[], double S2i[])
|
|
|
+ cdef int nField(int L, int pl, vector[double] x, vector[complex] m, int nmax, int nCoords, vector[double] Xp, vector[double] Yp, vector[double] Zp, double Erx[], double Ery[], double Erz[], double Eix[], double Eiy[], double Eiz[], double Hrx[], double Hry[], double Hrz[], double Hix[], double Hiy[], double Hiz[])
|
|
|
+
|
|
|
+def scattnlay(np.ndarray[np.float64_t, ndim = 2] x, np.ndarray[np.complex128_t, ndim = 2] m, np.ndarray[np.float64_t, ndim = 1] theta = np.zeros(0, dtype = np.float64), np.int_t pl = -1, np.int_t nmax = -1):
|
|
|
cdef Py_ssize_t i
|
|
|
|
|
|
cdef np.ndarray[np.int_t, ndim = 1] terms = np.zeros(x.shape[0], dtype = np.int)
|
|
@@ -183,15 +68,53 @@ def scattnlay_wrapper(np.ndarray[np.float64_t, ndim = 2] x, np.ndarray[np.comple
|
|
|
S2r = np.zeros(theta.shape[0], dtype = np.float64)
|
|
|
S2i = np.zeros(theta.shape[0], dtype = np.float64)
|
|
|
|
|
|
- terms[i] = nMie_wrapper(x.shape[1], x[i].copy('C'),
|
|
|
- m[i].copy('C'), theta.shape[0],
|
|
|
- theta.copy('C'), &Qext[i], &Qsca[i],
|
|
|
- &Qabs[i], &Qbk[i], &Qpr[i], &g[i],
|
|
|
- &Albedo[i],
|
|
|
- S1r, S2r)
|
|
|
+ terms[i] = nMie(x.shape[1], pl, x[i].copy('C'), m[i].copy('C'), theta.shape[0], theta.copy('C'), nmax, &Qext[i], &Qsca[i], &Qabs[i], &Qbk[i], &Qpr[i], &g[i], &Albedo[i], npy2c(S1r), npy2c(S1i), npy2c(S2r), npy2c(S2i))
|
|
|
|
|
|
- S1[i] = S1r.copy('C')
|
|
|
- S2[i] = S2r.copy('C')
|
|
|
+ S1[i] = S1r.copy('C') + 1.0j*S1i.copy('C')
|
|
|
+ S2[i] = S2r.copy('C') + 1.0j*S2i.copy('C')
|
|
|
|
|
|
return terms, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2
|
|
|
|
|
|
+#def fieldnlay(np.ndarray[np.float64_t, ndim = 2] x, np.ndarray[np.complex128_t, ndim = 2] m, np.ndarray[np.float64_t, ndim = 2] coords = np.zeros((0, 3), dtype = np.float64), np.int_t pl = 0, np.int_t nmax = 0):
|
|
|
+def fieldnlay(np.ndarray[np.float64_t, ndim = 2] x, np.ndarray[np.complex128_t, ndim = 2] m, np.ndarray[np.float64_t, ndim = 2] coords, np.int_t pl = 0, np.int_t nmax = 0):
|
|
|
+ cdef Py_ssize_t i
|
|
|
+
|
|
|
+ cdef np.ndarray[np.int_t, ndim = 1] terms = np.zeros(x.shape[0], dtype = np.int)
|
|
|
+
|
|
|
+ cdef np.ndarray[np.complex128_t, ndim = 3] E = np.zeros((x.shape[0], coords.shape[0], 3), dtype = np.complex128)
|
|
|
+ cdef np.ndarray[np.complex128_t, ndim = 3] H = np.zeros((x.shape[0], coords.shape[0], 3), dtype = np.complex128)
|
|
|
+
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Erx
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Ery
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Erz
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Eix
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Eiy
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Eiz
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Hrx
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Hry
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Hrz
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Hix
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Hiy
|
|
|
+ cdef np.ndarray[np.float64_t, ndim = 1] Hiz
|
|
|
+
|
|
|
+ for i in range(x.shape[0]):
|
|
|
+ Erx = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Ery = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Erz = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Eix = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Eiy = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Eiz = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Hrx = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Hry = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Hrz = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Hix = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Hiy = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+ Hiz = np.zeros(coords.shape[0], dtype = np.float64)
|
|
|
+
|
|
|
+ terms[i] = nField(x.shape[1], pl, x[i].copy('C'), m[i].copy('C'), nmax, coords.shape[0], coords[:, 0].copy('C'), coords[:, 1].copy('C'), coords[:, 2].copy('C'), npy2c(Erx), npy2c(Ery), npy2c(Erz), npy2c(Eix), npy2c(Eiy), npy2c(Eiz), npy2c(Hrx), npy2c(Hry), npy2c(Hrz), npy2c(Hix), npy2c(Hiy), npy2c(Hiz))
|
|
|
+
|
|
|
+ E[i] = np.vstack((Erx.copy('C') + 1.0j*Eix.copy('C'), Ery.copy('C') + 1.0j*Eiy.copy('C'), Erz.copy('C') + 1.0j*Eiz.copy('C'))).transpose()
|
|
|
+ H[i] = np.vstack((Hrx.copy('C') + 1.0j*Hix.copy('C'), Hry.copy('C') + 1.0j*Hiy.copy('C'), Hrz.copy('C') + 1.0j*Hiz.copy('C'))).transpose()
|
|
|
+
|
|
|
+ return terms, E, H
|
|
|
+
|