From 0e04f5ca77dac78b9403374e796adb9a845a5764 Mon Sep 17 00:00:00 2001 From: Jan Petykiewicz Date: Thu, 11 Jun 2020 19:29:25 -0700 Subject: [PATCH] zero k=0 values --- meanas/fdfd/farfield.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/meanas/fdfd/farfield.py b/meanas/fdfd/farfield.py index ea11224..5e6f98e 100644 --- a/meanas/fdfd/farfield.py +++ b/meanas/fdfd/farfield.py @@ -188,6 +188,11 @@ def far_to_nearfield(E_far: fdfield_t, sin_th[numpy.logical_and(kx == 0, ky == 0)] = 0 cos_th[numpy.logical_and(kx == 0, ky == 0)] = 1 + theta = numpy.arctan2(ky, kx) + phi = numpy.arccos(cos_phi) + theta[numpy.logical_and(kx == 0, ky == 0)] = 0 + phi[numpy.logical_and(kx == 0, ky == 0)] = 0 + # Zero fields beyond valid (phi, theta) invalid_ind = kxy2 >= k * k theta[invalid_ind] = 0