ó
    Eñi,  ã                   ól   • S SK rS SKJr  SSKJr  SSKJr  / SQrS rS r	SS	 jr
SS
 jrSS jrSS jrg)é    N)Únormalize_axis_indexé   )Ú_ni_support)Ú	_nd_image)Úfourier_gaussianÚfourier_uniformÚfourier_ellipsoidÚfourier_shiftc                 óš  • U c¢  UR                   R                  [        R                  [        R                  [        R
                  4;   a+  [        R                  " UR                  UR                   S9n U $ [        R                  " UR                  [        R                  S9n  U $ [        U 5      [        L an  U [        R                  [        R                  [        R
                  [        R                  4;  a  [        S5      e[        R                  " UR                  U S9n U $ U R                  UR                  :w  a  [        S5      eU $ ©N©Údtypezoutput type not supportedzoutput shape not correct)
r   ÚtypeÚnpÚ	complex64Ú
complex128Úfloat32ÚzerosÚshapeÚfloat64ÚRuntimeError©ÚoutputÚinputs     ÚS/home/mande/repo/quber/.venv/lib/python3.13/site-packages/scipy/ndimage/_fourier.pyÚ_get_output_fourierr   (   sê   € Ø�~Ø�;‰;×Ñ¤§¡¬b¯m©m¼R¿Z¹ZÐHÓHÜ—X’X˜eŸk™k°·±Ñ=ˆFð €Mô —X’X˜eŸk™k´·±Ñ<‰Fð €Mô 
ˆf‹œÒ	Øœ"Ÿ,™,¬¯©ÜŸ*™*¤b§j¡jð2ó 2äÐ:Ó;Ð;Ü—’˜%Ÿ+™+¨VÑ4ˆð €Mð 
�‰˜Ÿ™Ó	$ÜÐ5Ó6Ð6Ø€Mó    c                 ó@  • U c“  UR                   R                  [        R                  [        R                  4;   a+  [        R
                  " UR                  UR                   S9n U $ [        R
                  " UR                  [        R                  S9n  U $ [        U 5      [        L aP  U [        R                  [        R                  4;  a  [        S5      e[        R
                  " UR                  U S9n U $ U R                  UR                  :w  a  [        S5      eU $ r   )r   r   r   r   r   r   r   r   r   s     r   Ú_get_output_fourier_complexr   8   sÓ   € Ø�~Ø�;‰;×Ñ¤§¡¬b¯m©mÐ<Ó<Ü—X’X˜eŸk™k°·±Ñ=ˆFð €Mô —X’X˜eŸk™k´·±Ñ?‰Fð €Mô 
ˆf‹œÒ	Øœ"Ÿ,™,¬¯©Ð6Ó6ÜÐ:Ó;Ð;Ü—’˜%Ÿ+™+¨VÑ4ˆð €Mð 
�‰˜Ÿ™Ó	$ÜÐ5Ó6Ð6Ø€Mr   c                 ó€  • [         R                  " U 5      n [        X@5      n[        X0R                  5      n[
        R                  " XR                  5      n[         R                  " U[         R                  S9nUR                  R                  (       d  UR                  5       n[        R                  " XX#US5        U$ )ac  
Multidimensional Gaussian fourier filter.

The array is multiplied with the fourier transform of a Gaussian
kernel.

Parameters
----------
input : array_like
    The input array.
sigma : float or sequence
    The sigma of the Gaussian kernel. If a float, `sigma` is the same for
    all axes. If a sequence, `sigma` has to contain one value for each
    axis.
n : int, optional
    If `n` is negative (default), then the input is assumed to be the
    result of a complex fft.
    If `n` is larger than or equal to zero, the input is assumed to be the
    result of a real fft, and `n` gives the length of the array before
    transformation along the real transform direction.
axis : int, optional
    The axis of the real transform.
output : ndarray, optional
    If given, the result of filtering the input is placed in this array.

Returns
-------
fourier_gaussian : ndarray
    The filtered input.

Examples
--------
>>> from scipy import ndimage, datasets
>>> import numpy.fft
>>> import matplotlib.pyplot as plt
>>> fig, (ax1, ax2) = plt.subplots(1, 2)
>>> plt.gray()  # show the filtered result in grayscale
>>> ascent = datasets.ascent()
>>> input_ = numpy.fft.fft2(ascent)
>>> result = ndimage.fourier_gaussian(input_, sigma=4)
>>> result = numpy.fft.ifft2(result)
>>> ax1.imshow(ascent)
>>> ax2.imshow(result.real)  # the imaginary part is an artifact
>>> plt.show()
r   r   ©r   Úasarrayr   r   Úndimr   Ú_normalize_sequencer   ÚflagsÚ
contiguousÚcopyr   Úfourier_filter)r   ÚsigmaÚnÚaxisr   Úsigmass         r   r   r   G   s„   € ô\ �JŠJ�uÓ€EÜ  Ó/€FÜ §j¡jÓ1€DÜ×,Ò,¨U·J±JÓ?€FÜ�ZŠZ˜¤b§j¡jÑ1€FØ�<‰<×"×"Ø—‘“ˆä×Ò˜U¨A°V¸QÔ?Ø€Mr   c                 ó€  • [         R                  " U 5      n [        X@5      n[        X0R                  5      n[
        R                  " XR                  5      n[         R                  " U[         R                  S9nUR                  R                  (       d  UR                  5       n[        R                  " XX#US5        U$ )ae  
Multidimensional uniform fourier filter.

The array is multiplied with the Fourier transform of a box of given
size.

Parameters
----------
input : array_like
    The input array.
size : float or sequence
    The size of the box used for filtering.
    If a float, `size` is the same for all axes. If a sequence, `size` has
    to contain one value for each axis.
n : int, optional
    If `n` is negative (default), then the input is assumed to be the
    result of a complex fft.
    If `n` is larger than or equal to zero, the input is assumed to be the
    result of a real fft, and `n` gives the length of the array before
    transformation along the real transform direction.
axis : int, optional
    The axis of the real transform.
output : ndarray, optional
    If given, the result of filtering the input is placed in this array.

Returns
-------
fourier_uniform : ndarray
    The filtered input.

Examples
--------
>>> from scipy import ndimage, datasets
>>> import numpy.fft
>>> import matplotlib.pyplot as plt
>>> fig, (ax1, ax2) = plt.subplots(1, 2)
>>> plt.gray()  # show the filtered result in grayscale
>>> ascent = datasets.ascent()
>>> input_ = numpy.fft.fft2(ascent)
>>> result = ndimage.fourier_uniform(input_, size=20)
>>> result = numpy.fft.ifft2(result)
>>> ax1.imshow(ascent)
>>> ax2.imshow(result.real)  # the imaginary part is an artifact
>>> plt.show()
r   r   r!   ©r   Úsizer*   r+   r   Úsizess         r   r   r   �   s„   € ô\ �JŠJ�uÓ€EÜ  Ó/€FÜ §j¡jÓ1€DÜ×+Ò+¨D·*±*Ó=€EÜ�JŠJ�u¤B§J¡JÑ/€EØ�;‰;×!×!Ø—
‘
“ˆÜ×Ò˜U¨1°F¸AÔ>Ø€Mr   c                 óÚ  • [         R                  " U 5      n U R                  S:”  a  [        S5      e[	        X@5      nUR
                  S:X  a  U$ [        X0R                  5      n[        R                  " XR                  5      n[         R                  " U[         R                  S9nUR                  R                  (       d  UR                  5       n[        R                  " XX#US5        U$ )a¼  
Multidimensional ellipsoid Fourier filter.

The array is multiplied with the fourier transform of an ellipsoid of
given sizes.

Parameters
----------
input : array_like
    The input array.
size : float or sequence
    The size of the box used for filtering.
    If a float, `size` is the same for all axes. If a sequence, `size` has
    to contain one value for each axis.
n : int, optional
    If `n` is negative (default), then the input is assumed to be the
    result of a complex fft.
    If `n` is larger than or equal to zero, the input is assumed to be the
    result of a real fft, and `n` gives the length of the array before
    transformation along the real transform direction.
axis : int, optional
    The axis of the real transform.
output : ndarray, optional
    If given, the result of filtering the input is placed in this array.

Returns
-------
fourier_ellipsoid : ndarray
    The filtered input.

Notes
-----
This function is implemented for arrays of rank 1, 2, or 3.

Examples
--------
>>> from scipy import ndimage, datasets
>>> import numpy.fft
>>> import matplotlib.pyplot as plt
>>> fig, (ax1, ax2) = plt.subplots(1, 2)
>>> plt.gray()  # show the filtered result in grayscale
>>> ascent = datasets.ascent()
>>> input_ = numpy.fft.fft2(ascent)
>>> result = ndimage.fourier_ellipsoid(input_, size=20)
>>> result = numpy.fft.ifft2(result)
>>> ax1.imshow(ascent)
>>> ax2.imshow(result.real)  # the imaginary part is an artifact
>>> plt.show()
é   z'Only 1d, 2d and 3d inputs are supportedr   r   é   )r   r"   r#   ÚNotImplementedErrorr   r/   r   r   r$   r   r%   r&   r'   r   r(   r.   s         r   r	   r	   º   s®   € ôd �JŠJ�uÓ€EØ‡z�z�Aƒ~Ü!Ð"KÓLÐLÜ  Ó/€FØ‡{�{�aÓð ˆÜ §j¡jÓ1€DÜ×+Ò+¨D·*±*Ó=€EÜ�JŠJ�u¤B§J¡JÑ/€EØ�;‰;×!×!Ø—
‘
“ˆÜ×Ò˜U¨1°F¸AÔ>Ø€Mr   c                 ó~  • [         R                  " U 5      n [        X@5      n[        X0R                  5      n[
        R                  " XR                  5      n[         R                  " U[         R                  S9nUR                  R                  (       d  UR                  5       n[        R                  " XX#U5        U$ )a`  
Multidimensional Fourier shift filter.

The array is multiplied with the Fourier transform of a shift operation.

Parameters
----------
input : array_like
    The input array.
shift : float or sequence
    The size of the box used for filtering.
    If a float, `shift` is the same for all axes. If a sequence, `shift`
    has to contain one value for each axis.
n : int, optional
    If `n` is negative (default), then the input is assumed to be the
    result of a complex fft.
    If `n` is larger than or equal to zero, the input is assumed to be the
    result of a real fft, and `n` gives the length of the array before
    transformation along the real transform direction.
axis : int, optional
    The axis of the real transform.
output : ndarray, optional
    If given, the result of shifting the input is placed in this array.

Returns
-------
fourier_shift : ndarray
    The shifted input.

Examples
--------
>>> from scipy import ndimage, datasets
>>> import matplotlib.pyplot as plt
>>> import numpy.fft
>>> fig, (ax1, ax2) = plt.subplots(1, 2)
>>> plt.gray()  # show the filtered result in grayscale
>>> ascent = datasets.ascent()
>>> input_ = numpy.fft.fft2(ascent)
>>> result = ndimage.fourier_shift(input_, shift=200)
>>> result = numpy.fft.ifft2(result)
>>> ax1.imshow(ascent)
>>> ax2.imshow(result.real)  # the imaginary part is an artifact
>>> plt.show()
r   )r   r"   r   r   r#   r   r$   r   r%   r&   r'   r   r
   )r   Úshiftr*   r+   r   Úshiftss         r   r
   r
   ý   s‚   € ôZ �JŠJ�uÓ€EÜ(¨Ó7€FÜ §j¡jÓ1€DÜ×,Ò,¨U·J±JÓ?€FÜ�ZŠZ˜¤b§j¡jÑ1€FØ�<‰<×"×"Ø—‘“ˆÜ×Ò˜E¨1°FÔ;Ø€Mr   )éÿÿÿÿr8   N)Únumpyr   Úscipy._lib._utilr   Ú r   r   Ú__all__r   r   r   r   r	   r
   © r   r   Ú<module>r>      s<   ðó> Ý 1Ý Ý ò€òò ô7ôt6ôr@õF5r   