_array2.py

_array2.py

import numpy as np

#pythran export getArrayFactor (complex[:], float[:], float[:], float[:,:], float[:,:], float)
#pythran export getArrayFactor (complex[:], float[:], float[:], float, float, float)
def getArrayFactor(W, px, py, theta, phi, L):
    theta = np.array (theta)
    phi = np.array (phi)
    N = px.shape[0]
    AF = np.zeros(theta.shape, dtype=complex)
    kx = np.sin(theta) * np.cos(phi) * 2.0 * np.pi / L
    ky = np.sin(theta) * np.sin(phi) * 2.0 * np.pi / L
    for n in range(N):
        AF += W[n] * np.exp(-1j * (kx * px[n] + ky * py[n]))
    magAF = np.abs(AF)
    return magAF