Source code for monalisa.fourier._idf1

"""Port of src/fourier1/bmIDF1.m."""

import numpy as np
import scipy.fft

from monalisa._fft import fftshift, ifftshift
from monalisa._mat import mixed, msize, reshape
from monalisa.array._block_reshape import block_reshape




def idf1(x, N_u, dK_u):
    """Centered 1D inverse Fourier transform scaled by prod(N_u)*prod(dK_u)."""
    argSize = msize(x)
    x = block_reshape(x, N_u)
    for n in range(1):
        x = fftshift(scipy.fft.ifft(ifftshift(x, axes=n), axis=n, workers=-1, overwrite_x=True), axes=n)
    F = np.prod(np.ravel(N_u)) * np.prod(np.ravel(dK_u))
    x = mixed(np.multiply, x, F)
    return reshape(x, argSize)

↑