def _unequal_spacing_conv_core(psf1, psf2):
'''Interpolates psf2 before using fft-based convolution
Args:
psf1 (prysm.PSF): PSF. This one defines the sampling of the output.
psf2 (prysm.PSF): PSF. This one will have its frequency response
truncated.
Returns:
PSF: a new `PSF` that is the convolution of psf1 and psf2.
'''
# map psf1 into the fourier domain
ft1 = fft2(fftshift(psf1.data))
unit1x = forward_ft_unit(psf1.sample_spacing, psf1.samples_x)
unit1y = forward_ft_unit(psf1.sample_spacing, psf1.samples_y)
# map psf2 into the fourier domain
ft2 = fft2(fftshift(psf2.data))
unit2x = forward_ft_unit(psf2.sample_spacing, psf2.samples_x)
unit2y = forward_ft_unit(psf2.sample_spacing, psf2.samples_y)
ft3 = ifftshift(resample_2d_complex(fftshift(ft2), (unit2y, unit2x), (unit1y, unit1x)))
psf3 = PSF(data=abs(ifftshift(ifft2(ft1 * ft3))),
sample_spacing=psf1.sample_spacing)
return psf3._renorm()
def test_resample_2d_complex_does_not_distort(data_2d_complex):
x, y, dat = data_2d_complex
resampled = coordinates.resample_2d_complex(dat, (x, y), (x, y))
assert np.allclose(dat, resampled)