rayleigh_sommerfeld_conv¶

Computes Fresnel diffraction integral using convolution method:

\[\begin{split}U'(x,y)&=-\frac{1}{2\pi}\iint U(u,v)\pfrac{}{d}\frac{\e^{\i kr_d}}{r_d}\d u\d v\\ &=-\frac{1}{2\pi}U(x,y)\ast\pfrac{}{d}\frac{\e^{\i kr}}{r}\\ &=U(x,y)\ast\frac{z}{2\pi r^2}(\frac{1}{r}-\i k)\e^{\i kr}\end{split}\]

where \(U\) and \(U'\) are the complex amplitude on source and target plane, respectively, \(d\) is the distance between two planes, \(r=\sqrt{x^2+y^2+d^2}\), \(\lambda\) is wavelength, \(k=2\pi/\lambda\) and \(\ast\) represents convolution.

Note that some intermediate computational steps don’t depend on pupil and pre-computing their results can avoid repeated computation if this function is expected to be called multiple times, as long as the arguments other than pupil don’t change. This can be done by calling init_rayleigh_sommerfeld_conv() and passing its return value as intermediate argument. In this case, wl, distance, dx and dy cannot be passed. If all of them are given, intermediate will be ignored anyway.

Parameters:

pupil (Tensor) – Pupil function \(U\). A tensor of shape \((\cdots,N_d,N_\lambda,H,W)\).
wl (float, Sequence[float] or Tensor) – Wavelengths \(\lambda\). A scalar or a tensor of shape \((N_\lambda,)\).
distance (float, Sequence[float] or Tensor) – Propagation distance \(d\). A scalar or a tensor of shape \((N_d,)\).
dx (float or Tensor) – Grid spacing in horizontal direction to ensure correct scale for DFT. A scalar or a tensor of shape \((\cdots,N_d,N_\lambda)\). Default: ignored.
dy (float or Tensor) – Grid spacing in vertical direction, similar to dx. Default: same as dx. Note that dy will be also ignored if dx is None even though dy is given.

kwargs –

Optional keyword arguments:

Parameter	Type	Description
ksize	int \| tuple[int, int]	Spatial dimension of the kernel function in vertical and horizontal directions. Default: identical to that of `pupil`.
padding	int \| str	See `dconv()`. Default: `linear`.
dft_scale	bool	Whether to apply DFT-scale. See Fourier transform. Default: `True`.
short_wl	bool	Whether to ignore \(\frac{1}{r}\) term in kernel function. Default: `True`.
intermediate	dict	Cached intermediate results returned by `init_rayleigh_sommerfeld_conv()`. This can be used to speed up computation. Default: omitted.

Returns:

Diffracted complex amplitude \(U'\). A tensor of shape \((\cdots,N_d,N_\lambda,H,W)\).

Return type:

Tensor