LSST Applications g04e9c324dd+8c5ae1fdc5,g134cb467dc+b203dec576,g18429d2f64+358861cd2c,g199a45376c+0ba108daf9,g1fd858c14a+dd066899e3,g262e1987ae+ebfced1d55,g29ae962dfc+72fd90588e,g2cef7863aa+aef1011c0b,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+b668f15bc5,g4595892280+3897dae354,g47891489e3+abcf9c3559,g4d44eb3520+fb4ddce128,g53246c7159+8c5ae1fdc5,g67b6fd64d1+abcf9c3559,g67fd3c3899+1f72b5a9f7,g74acd417e5+cb6b47f07b,g786e29fd12+668abc6043,g87389fa792+8856018cbb,g89139ef638+abcf9c3559,g8d7436a09f+bcf525d20c,g8ea07a8fe4+9f5ccc88ac,g90f42f885a+6054cc57f1,g97be763408+06f794da49,g9dd6db0277+1f72b5a9f7,ga681d05dcb+7e36ad54cd,gabf8522325+735880ea63,gac2eed3f23+abcf9c3559,gb89ab40317+abcf9c3559,gbf99507273+8c5ae1fdc5,gd8ff7fe66e+1f72b5a9f7,gdab6d2f7ff+cb6b47f07b,gdc713202bf+1f72b5a9f7,gdfd2d52018+8225f2b331,ge365c994fd+375fc21c71,ge410e46f29+abcf9c3559,geaed405ab2+562b3308c0,gf9a733ac38+8c5ae1fdc5,w.2025.35
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.