Commit 4700be14 authored by Sean Leavey's avatar Sean Leavey
Browse files

Fixing bug with distance (variable 'L' not defined)

parent c56b35f6
......@@ -81,7 +81,7 @@ def FFT_scale_propagate(field, grid0, grid1, Lambda, distance, w0, w1, nr):
field = field * np.exp(-1j*k*distance) * np.exp(1j*plD*grid0.fft_ir_squared)
field = np.fft.ifft2(field)
# final scaling
field = field *w0/w1 * np.exp(1j* grid1.r_squared*(z0+L)/(2.0*z0*z0))
field = field *w0/w1 * np.exp(1j* grid1.r_squared*(z0+distance)/(2.0*z0*z0))
return field
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment