Updated ffts

Author: rohanbabbar04

examples/plot_fft.py

@@ -24,7 +24,7 @@
 nfft = 2**10
 d = np.sin(2 * np.pi * f0 * t)
 
-FFTop = pylops.signalprocessing.FFT(dims=nt, nfft=nfft, sampling=dt, engine="numpy")
+FFTop = pylops.FFT(dims=nt, nfft=nfft, sampling=dt, engine="numpy")
 D = FFTop * d
 
 # Adjoint = inverse for FFT
@@ -49,7 +49,7 @@
 # numpy in many cases but it is not inserted in the mandatory requirements of
 # PyLops. If interested to use ``FFTW`` backend, read the `fft routines`
 # section at :ref:`performance`.
-FFTop = pylops.signalprocessing.FFT(dims=nt, nfft=nfft, sampling=dt, engine="fftw")
+FFTop = pylops.FFT(dims=nt, nfft=nfft, sampling=dt, engine="fftw")
 D = FFTop * d
 
 # Adjoint = inverse for FFT
@@ -69,7 +69,7 @@
 ###############################################################################
 # PyLops implements a third engine (``engine='mkl_fft'``) which uses the
 # well-known `mkl_fft <https://github.com/IntelPython/mkl_fft>`_ .
-FFTop = pylops.signalprocessing.FFT(dims=nt, nfft=nfft, sampling=dt, engine="mkl_fft")
+FFTop = pylops.FFT(dims=nt, nfft=nfft, sampling=dt, engine="mkl_fft")
 D = FFTop * d
 
 # Adjoint = inverse for FFT
@@ -129,7 +129,7 @@
 nfft = 2**10
 d = np.outer(np.sin(2 * np.pi * f0 * t), np.arange(nx) + 1)
 
-FFTop = pylops.signalprocessing.FFT2D(
+FFTop = pylops.FFT2D(
     dims=(nt, nx), nffts=(nfft, nfft), sampling=(dt, dx)
 )
 D = FFTop * d.ravel()
@@ -170,7 +170,7 @@
 d = np.outer(np.sin(2 * np.pi * f0 * t), np.arange(nx) + 1)
 d = np.tile(d[:, :, np.newaxis], [1, 1, ny])
 
-FFTop = pylops.signalprocessing.FFTND(
+FFTop = pylops.FFTND(
     dims=(nt, nx, ny), nffts=(nfft, nfftk, nfftk), sampling=(dt, dx, dy)
 )
 D = FFTop * d.ravel()

pylops/__init__.py

@@ -57,6 +57,7 @@
     waveeqprocessing,
 )
 from .torchoperator import *
+from .signalprocessing import *
 from .avo.poststack import *
 from .avo.prestack import *
 from .optimization.basic import *