add mask again

fresnel/propagate.py

@@ -24,13 +24,13 @@ def rayleighSommerfeldTF(shape, dperp, k, dz):
 
     f = fftfreqn(shape, dperp)  # spatial frequencies
     f2 = squaresum(f)
-    # mask = f2 < 1 / wl
+    mask = f2 < 1 / wl
 
-    # phasechirp = np.sqrt(1 - wl ** 2 * (mask * f2))
-    phasechirp = np.sqrt(1 - wl ** 2 * f2)
+    phasechirp = np.sqrt(1 - wl ** 2 * (mask * f2))
+    # phasechirp = np.sqrt(1 - wl ** 2 * f2)
 
-    # TF = mask * np.exp(1j * k * dz * phasechirp)
-    TF = np.exp(1j * k * dz * phasechirp)
+    TF = mask * np.exp(1j * k * dz * phasechirp)
+    # TF = np.exp(1j * k * dz * phasechirp)
 
     return TF