LSST Applications g042eb84c57+730a74494b,g04e9c324dd+8c5ae1fdc5,g134cb467dc+1f1e3e7524,g199a45376c+0ba108daf9,g1fd858c14a+fa7d31856b,g210f2d0738+f66ac109ec,g262e1987ae+83a3acc0e5,g29ae962dfc+d856a2cb1f,g2cef7863aa+aef1011c0b,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+a1e0c9f713,g47891489e3+0d594cb711,g4d44eb3520+c57ec8f3ed,g4d7b6aa1c5+f66ac109ec,g53246c7159+8c5ae1fdc5,g56a1a4eaf3+fd7ad03fde,g64539dfbff+f66ac109ec,g67b6fd64d1+0d594cb711,g67fd3c3899+f66ac109ec,g6985122a63+0d594cb711,g74acd417e5+3098891321,g786e29fd12+668abc6043,g81db2e9a8d+98e2ab9f28,g87389fa792+8856018cbb,g89139ef638+0d594cb711,g8d7436a09f+80fda9ce03,g8ea07a8fe4+760ca7c3fc,g90f42f885a+033b1d468d,g97be763408+a8a29bda4b,g99822b682c+e3ec3c61f9,g9d5c6a246b+0d5dac0c3d,ga41d0fce20+9243b26dd2,gbf99507273+8c5ae1fdc5,gd7ef33dd92+0d594cb711,gdab6d2f7ff+3098891321,ge410e46f29+0d594cb711,geaed405ab2+c4bbc419c6,gf9a733ac38+8c5ae1fdc5,w.2025.38
LSST Data Management Base Package
|
Classes | |
class | Fourier |
Functions | |
np.ndarray | centered (np.ndarray arr, Sequence[int] newshape) |
np.ndarray | fast_zero_pad (np.ndarray arr, Sequence[Sequence[int]] pad_width) |
np.ndarray | _pad (np.ndarray arr, Sequence[int] newshape, int|Sequence[int]|None axes=None, str mode="constant", float constant_values=0) |
tuple | get_fft_shape (np.ndarray|Sequence[int] im_or_shape1, np.ndarray|Sequence[int] im_or_shape2, int padding=3, int|Sequence[int]|None axes=None, bool use_max=False) |
Fourier | _kspace_operation (Fourier image1, Fourier image2, int padding, Callable op, Sequence[int] shape, int|Sequence[int] axes) |
Fourier|np.ndarray | match_kernel (np.ndarray|Fourier kernel1, np.ndarray|Fourier kernel2, int padding=3, int|Sequence[int] axes=(-2, -1), bool return_fourier=True, bool normalize=False) |
np.ndarray|Fourier | convolve (np.ndarray|Fourier image, np.ndarray|Fourier kernel, int padding=3, int|Sequence[int] axes=(-2, -1), bool return_fourier=True, bool normalize=False) |
|
protected |
Combine two images in k-space using a given `operator` Parameters ---------- image1: The LHS of the equation. image2: The RHS of the equation. padding: The amount of padding to add before transforming into k-space. op: The operator used to combine the two images. This is either ``operator.mul`` for a convolution or ``operator.truediv`` for deconvolution. shape: The shape of the output image. axes: The dimension(s) of the array that will be transformed.
Definition at line 374 of file fft.py.
|
protected |
Pad an array to fit into newshape Pad `arr` with zeros to fit into newshape, which uses the `np.fft.fftshift` convention of moving the center pixel of `arr` (if `arr.shape` is odd) to the center-right pixel in an even shaped `newshape`. Parameters ---------- arr: The arrray to pad. newshape: The new shape of the array. axes: The axes that are being reshaped. mode: The numpy mode used to pad the array. In other words, how to fill the new padded elements. See ``numpy.pad`` for details. constant_values: If `mode` == "constant" then this is the value to set all of the new padded elements to.
Definition at line 98 of file fft.py.
np.ndarray lsst.scarlet.lite.fft.centered | ( | np.ndarray | arr, |
Sequence[int] | newshape ) |
Return the central newshape portion of the array. Parameters ---------- arr: The array to center. newshape: The new shape of the array. Notes ----- If the array shape is odd and the target is even, the center of `arr` is shifted to the center-right pixel position. This is slightly different than the scipy implementation, which uses the center-left pixel for the array center. The reason for the difference is that we have adopted the convention of `np.fft.fftshift` in order to make sure that changing back and forth from fft standard order (0 frequency and position is in the bottom left) to 0 position in the center.
Definition at line 34 of file fft.py.
np.ndarray | Fourier lsst.scarlet.lite.fft.convolve | ( | np.ndarray | Fourier | image, |
np.ndarray | Fourier | kernel, | ||
int | padding = 3, | ||
int | Sequence[int] | axes = (-2, -1), | ||
bool | return_fourier = True, | ||
bool | normalize = False ) |
Convolve image with a kernel Parameters ---------- image: Image either as array or as `Fourier` object kernel: Convolution kernel either as array or as `Fourier` object padding: Additional padding to use when generating the FFT to suppress artifacts. axes: Axes that contain the spatial information for the PSFs. return_fourier: Whether to return `Fourier` or array normalize: Whether or not to normalize the input kernels. Returns ------- result: The convolution of the image with the kernel.
Definition at line 481 of file fft.py.
np.ndarray lsst.scarlet.lite.fft.fast_zero_pad | ( | np.ndarray | arr, |
Sequence[Sequence[int]] | pad_width ) |
Fast version of numpy.pad when `mode="constant"` Executing `numpy.pad` with zeros is ~1000 times slower because it doesn't make use of the `zeros` method for padding. Parameters --------- arr: The array to pad pad_width: Number of values padded to the edges of each axis. See numpy.pad docs for more. Returns ------- result: np.ndarray The array padded with `constant_values`
Definition at line 71 of file fft.py.
tuple lsst.scarlet.lite.fft.get_fft_shape | ( | np.ndarray | Sequence[int] | im_or_shape1, |
np.ndarray | Sequence[int] | im_or_shape2, | ||
int | padding = 3, | ||
int | Sequence[int] | None | axes = None, | ||
bool | use_max = False ) |
Return the fast fft shapes for each spatial axis Calculate the fast fft shape for each dimension in axes. Parameters ---------- im_or_shape1: The left image or shape of an image. im_or_shape2: The right image or shape of an image. padding: Any additional padding to add to the final shape. axes: The axes that are being transformed. use_max: Whether or not to use the maximum of the two shapes, or the sum of the two shapes. Returns ------- shape: Tuple of the shape to use when the two images are transformed into k-space.
Definition at line 152 of file fft.py.
Fourier | np.ndarray lsst.scarlet.lite.fft.match_kernel | ( | np.ndarray | Fourier | kernel1, |
np.ndarray | Fourier | kernel2, | ||
int | padding = 3, | ||
int | Sequence[int] | axes = (-2, -1), | ||
bool | return_fourier = True, | ||
bool | normalize = False ) |
Calculate the difference kernel to match kernel1 to kernel2 Parameters ---------- kernel1: The first kernel, either as array or as `Fourier` object kernel2: The second kernel, either as array or as `Fourier` object padding: Additional padding to use when generating the FFT to supress artifacts. axes: Axes that contain the spatial information for the kernels. return_fourier: Whether to return `Fourier` or array normalize: Whether or not to normalize the input kernels. Returns ------- result: The difference kernel to go from `kernel1` to `kernel2`.
Definition at line 433 of file fft.py.