|
|
import abc |
|
|
import decimal |
|
|
import numbers |
|
|
from collections.abc import Iterator, Mapping, Sequence |
|
|
from typing import ( |
|
|
Any, |
|
|
ClassVar, |
|
|
Final, |
|
|
Generic, |
|
|
Literal, |
|
|
SupportsIndex, |
|
|
TypeAlias, |
|
|
TypeGuard, |
|
|
overload, |
|
|
) |
|
|
|
|
|
import numpy as np |
|
|
import numpy.typing as npt |
|
|
from numpy._typing import ( |
|
|
_FloatLike_co, |
|
|
_NumberLike_co, |
|
|
|
|
|
_ArrayLikeFloat_co, |
|
|
_ArrayLikeComplex_co, |
|
|
) |
|
|
|
|
|
from ._polytypes import ( |
|
|
_AnyInt, |
|
|
_CoefLike_co, |
|
|
|
|
|
_Array2, |
|
|
_Tuple2, |
|
|
|
|
|
_Series, |
|
|
_CoefSeries, |
|
|
|
|
|
_SeriesLikeInt_co, |
|
|
_SeriesLikeCoef_co, |
|
|
|
|
|
_ArrayLikeCoefObject_co, |
|
|
_ArrayLikeCoef_co, |
|
|
) |
|
|
|
|
|
from typing_extensions import LiteralString, TypeVar |
|
|
|
|
|
|
|
|
__all__: Final[Sequence[str]] = ("ABCPolyBase",) |
|
|
|
|
|
|
|
|
_NameCo = TypeVar("_NameCo", bound=LiteralString | None, covariant=True, default=LiteralString | None) |
|
|
_Self = TypeVar("_Self") |
|
|
_Other = TypeVar("_Other", bound=ABCPolyBase) |
|
|
|
|
|
_AnyOther: TypeAlias = ABCPolyBase | _CoefLike_co | _SeriesLikeCoef_co |
|
|
_Hundred: TypeAlias = Literal[100] |
|
|
|
|
|
|
|
|
class ABCPolyBase(Generic[_NameCo], metaclass=abc.ABCMeta): |
|
|
__hash__: ClassVar[None] |
|
|
__array_ufunc__: ClassVar[None] |
|
|
|
|
|
maxpower: ClassVar[_Hundred] |
|
|
_superscript_mapping: ClassVar[Mapping[int, str]] |
|
|
_subscript_mapping: ClassVar[Mapping[int, str]] |
|
|
_use_unicode: ClassVar[bool] |
|
|
|
|
|
basis_name: _NameCo |
|
|
coef: _CoefSeries |
|
|
domain: _Array2[np.inexact[Any] | np.object_] |
|
|
window: _Array2[np.inexact[Any] | np.object_] |
|
|
|
|
|
_symbol: LiteralString |
|
|
@property |
|
|
def symbol(self, /) -> LiteralString: ... |
|
|
|
|
|
def __init__( |
|
|
self, |
|
|
/, |
|
|
coef: _SeriesLikeCoef_co, |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
symbol: str = ..., |
|
|
) -> None: ... |
|
|
|
|
|
@overload |
|
|
def __call__(self, /, arg: _Other) -> _Other: ... |
|
|
|
|
|
|
|
|
@overload |
|
|
def __call__( |
|
|
self, |
|
|
/, |
|
|
arg: _FloatLike_co | decimal.Decimal | numbers.Real | np.object_, |
|
|
) -> np.float64 | np.complex128: ... |
|
|
@overload |
|
|
def __call__( |
|
|
self, |
|
|
/, |
|
|
arg: _NumberLike_co | numbers.Complex, |
|
|
) -> np.complex128: ... |
|
|
@overload |
|
|
def __call__(self, /, arg: _ArrayLikeFloat_co) -> ( |
|
|
npt.NDArray[np.float64] |
|
|
| npt.NDArray[np.complex128] |
|
|
| npt.NDArray[np.object_] |
|
|
): ... |
|
|
@overload |
|
|
def __call__( |
|
|
self, |
|
|
/, |
|
|
arg: _ArrayLikeComplex_co, |
|
|
) -> npt.NDArray[np.complex128] | npt.NDArray[np.object_]: ... |
|
|
@overload |
|
|
def __call__( |
|
|
self, |
|
|
/, |
|
|
arg: _ArrayLikeCoefObject_co, |
|
|
) -> npt.NDArray[np.object_]: ... |
|
|
|
|
|
def __format__(self, fmt_str: str, /) -> str: ... |
|
|
def __eq__(self, x: object, /) -> bool: ... |
|
|
def __ne__(self, x: object, /) -> bool: ... |
|
|
def __neg__(self: _Self, /) -> _Self: ... |
|
|
def __pos__(self: _Self, /) -> _Self: ... |
|
|
def __add__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __sub__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __mul__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __truediv__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __floordiv__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __mod__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __divmod__(self: _Self, x: _AnyOther, /) -> _Tuple2[_Self]: ... |
|
|
def __pow__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __radd__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __rsub__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __rmul__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __rtruediv__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __rfloordiv__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __rmod__(self: _Self, x: _AnyOther, /) -> _Self: ... |
|
|
def __rdivmod__(self: _Self, x: _AnyOther, /) -> _Tuple2[_Self]: ... |
|
|
def __len__(self, /) -> int: ... |
|
|
def __iter__(self, /) -> Iterator[np.inexact[Any] | object]: ... |
|
|
def __getstate__(self, /) -> dict[str, Any]: ... |
|
|
def __setstate__(self, dict: dict[str, Any], /) -> None: ... |
|
|
|
|
|
def has_samecoef(self, /, other: ABCPolyBase) -> bool: ... |
|
|
def has_samedomain(self, /, other: ABCPolyBase) -> bool: ... |
|
|
def has_samewindow(self, /, other: ABCPolyBase) -> bool: ... |
|
|
@overload |
|
|
def has_sametype(self: _Self, /, other: ABCPolyBase) -> TypeGuard[_Self]: ... |
|
|
@overload |
|
|
def has_sametype(self, /, other: object) -> Literal[False]: ... |
|
|
|
|
|
def copy(self: _Self, /) -> _Self: ... |
|
|
def degree(self, /) -> int: ... |
|
|
def cutdeg(self: _Self, /) -> _Self: ... |
|
|
def trim(self: _Self, /, tol: _FloatLike_co = ...) -> _Self: ... |
|
|
def truncate(self: _Self, /, size: _AnyInt) -> _Self: ... |
|
|
|
|
|
@overload |
|
|
def convert( |
|
|
self, |
|
|
domain: None | _SeriesLikeCoef_co, |
|
|
kind: type[_Other], |
|
|
/, |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
) -> _Other: ... |
|
|
@overload |
|
|
def convert( |
|
|
self, |
|
|
/, |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
*, |
|
|
kind: type[_Other], |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
) -> _Other: ... |
|
|
@overload |
|
|
def convert( |
|
|
self: _Self, |
|
|
/, |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
kind: None | type[_Self] = ..., |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
) -> _Self: ... |
|
|
|
|
|
def mapparms(self, /) -> _Tuple2[Any]: ... |
|
|
|
|
|
def integ( |
|
|
self: _Self, /, |
|
|
m: SupportsIndex = ..., |
|
|
k: _CoefLike_co | _SeriesLikeCoef_co = ..., |
|
|
lbnd: None | _CoefLike_co = ..., |
|
|
) -> _Self: ... |
|
|
|
|
|
def deriv(self: _Self, /, m: SupportsIndex = ...) -> _Self: ... |
|
|
|
|
|
def roots(self, /) -> _CoefSeries: ... |
|
|
|
|
|
def linspace( |
|
|
self, /, |
|
|
n: SupportsIndex = ..., |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
) -> _Tuple2[_Series[np.float64 | np.complex128]]: ... |
|
|
|
|
|
@overload |
|
|
@classmethod |
|
|
def fit( |
|
|
cls: type[_Self], /, |
|
|
x: _SeriesLikeCoef_co, |
|
|
y: _SeriesLikeCoef_co, |
|
|
deg: int | _SeriesLikeInt_co, |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
rcond: _FloatLike_co = ..., |
|
|
full: Literal[False] = ..., |
|
|
w: None | _SeriesLikeCoef_co = ..., |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
symbol: str = ..., |
|
|
) -> _Self: ... |
|
|
@overload |
|
|
@classmethod |
|
|
def fit( |
|
|
cls: type[_Self], /, |
|
|
x: _SeriesLikeCoef_co, |
|
|
y: _SeriesLikeCoef_co, |
|
|
deg: int | _SeriesLikeInt_co, |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
rcond: _FloatLike_co = ..., |
|
|
*, |
|
|
full: Literal[True], |
|
|
w: None | _SeriesLikeCoef_co = ..., |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
symbol: str = ..., |
|
|
) -> tuple[_Self, Sequence[np.inexact[Any] | np.int32]]: ... |
|
|
@overload |
|
|
@classmethod |
|
|
def fit( |
|
|
cls: type[_Self], |
|
|
x: _SeriesLikeCoef_co, |
|
|
y: _SeriesLikeCoef_co, |
|
|
deg: int | _SeriesLikeInt_co, |
|
|
domain: None | _SeriesLikeCoef_co, |
|
|
rcond: _FloatLike_co, |
|
|
full: Literal[True], /, |
|
|
w: None | _SeriesLikeCoef_co = ..., |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
symbol: str = ..., |
|
|
) -> tuple[_Self, Sequence[np.inexact[Any] | np.int32]]: ... |
|
|
|
|
|
@classmethod |
|
|
def fromroots( |
|
|
cls: type[_Self], /, |
|
|
roots: _ArrayLikeCoef_co, |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
symbol: str = ..., |
|
|
) -> _Self: ... |
|
|
|
|
|
@classmethod |
|
|
def identity( |
|
|
cls: type[_Self], /, |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
symbol: str = ..., |
|
|
) -> _Self: ... |
|
|
|
|
|
@classmethod |
|
|
def basis( |
|
|
cls: type[_Self], /, |
|
|
deg: _AnyInt, |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
symbol: str = ..., |
|
|
) -> _Self: ... |
|
|
|
|
|
@classmethod |
|
|
def cast( |
|
|
cls: type[_Self], /, |
|
|
series: ABCPolyBase, |
|
|
domain: None | _SeriesLikeCoef_co = ..., |
|
|
window: None | _SeriesLikeCoef_co = ..., |
|
|
) -> _Self: ... |
|
|
|
|
|
@classmethod |
|
|
def _str_term_unicode(cls, /, i: str, arg_str: str) -> str: ... |
|
|
@staticmethod |
|
|
def _str_term_ascii(i: str, arg_str: str) -> str: ... |
|
|
@staticmethod |
|
|
def _repr_latex_term(i: str, arg_str: str, needs_parens: bool) -> str: ... |
|
|
|
