48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
from typing import Any, TypeVar, overload
|
|
|
|
from numpy import generic, integer, floating, complexfloating
|
|
from numpy._typing import (
|
|
NDArray,
|
|
ArrayLike,
|
|
_ShapeLike,
|
|
_ArrayLike,
|
|
_ArrayLikeFloat_co,
|
|
_ArrayLikeComplex_co,
|
|
)
|
|
|
|
_SCT = TypeVar("_SCT", bound=generic)
|
|
|
|
__all__: list[str]
|
|
|
|
@overload
|
|
def fftshift(x: _ArrayLike[_SCT], axes: None | _ShapeLike = ...) -> NDArray[_SCT]: ...
|
|
@overload
|
|
def fftshift(x: ArrayLike, axes: None | _ShapeLike = ...) -> NDArray[Any]: ...
|
|
|
|
@overload
|
|
def ifftshift(x: _ArrayLike[_SCT], axes: None | _ShapeLike = ...) -> NDArray[_SCT]: ...
|
|
@overload
|
|
def ifftshift(x: ArrayLike, axes: None | _ShapeLike = ...) -> NDArray[Any]: ...
|
|
|
|
@overload
|
|
def fftfreq(
|
|
n: int | integer[Any],
|
|
d: _ArrayLikeFloat_co = ...,
|
|
) -> NDArray[floating[Any]]: ...
|
|
@overload
|
|
def fftfreq(
|
|
n: int | integer[Any],
|
|
d: _ArrayLikeComplex_co = ...,
|
|
) -> NDArray[complexfloating[Any, Any]]: ...
|
|
|
|
@overload
|
|
def rfftfreq(
|
|
n: int | integer[Any],
|
|
d: _ArrayLikeFloat_co = ...,
|
|
) -> NDArray[floating[Any]]: ...
|
|
@overload
|
|
def rfftfreq(
|
|
n: int | integer[Any],
|
|
d: _ArrayLikeComplex_co = ...,
|
|
) -> NDArray[complexfloating[Any, Any]]: ...
|