hermite.pyi 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. from typing import Any, ClassVar, Final, Literal as L, TypeVar
  2. import numpy as np
  3. from numpy._typing import _Shape
  4. from ._polybase import ABCPolyBase
  5. from ._polytypes import (
  6. _Array1,
  7. _Array2,
  8. _FuncBinOp,
  9. _FuncCompanion,
  10. _FuncDer,
  11. _FuncFit,
  12. _FuncFromRoots,
  13. _FuncGauss,
  14. _FuncInteg,
  15. _FuncLine,
  16. _FuncPoly2Ortho,
  17. _FuncPow,
  18. _FuncRoots,
  19. _FuncUnOp,
  20. _FuncVal,
  21. _FuncVal2D,
  22. _FuncVal3D,
  23. _FuncVander,
  24. _FuncVander2D,
  25. _FuncVander3D,
  26. _FuncWeight,
  27. )
  28. from .polyutils import trimcoef as hermtrim
  29. __all__ = [
  30. "hermzero",
  31. "hermone",
  32. "hermx",
  33. "hermdomain",
  34. "hermline",
  35. "hermadd",
  36. "hermsub",
  37. "hermmulx",
  38. "hermmul",
  39. "hermdiv",
  40. "hermpow",
  41. "hermval",
  42. "hermder",
  43. "hermint",
  44. "herm2poly",
  45. "poly2herm",
  46. "hermfromroots",
  47. "hermvander",
  48. "hermfit",
  49. "hermtrim",
  50. "hermroots",
  51. "Hermite",
  52. "hermval2d",
  53. "hermval3d",
  54. "hermgrid2d",
  55. "hermgrid3d",
  56. "hermvander2d",
  57. "hermvander3d",
  58. "hermcompanion",
  59. "hermgauss",
  60. "hermweight",
  61. ]
  62. _ShapeT = TypeVar("_ShapeT", bound=_Shape)
  63. poly2herm: Final[_FuncPoly2Ortho] = ...
  64. herm2poly: Final[_FuncUnOp] = ...
  65. hermdomain: Final[_Array2[np.float64]] = ...
  66. hermzero: Final[_Array1[np.int_]] = ...
  67. hermone: Final[_Array1[np.int_]] = ...
  68. hermx: Final[_Array2[np.int_]] = ...
  69. hermline: Final[_FuncLine] = ...
  70. hermfromroots: Final[_FuncFromRoots] = ...
  71. hermadd: Final[_FuncBinOp] = ...
  72. hermsub: Final[_FuncBinOp] = ...
  73. hermmulx: Final[_FuncUnOp] = ...
  74. hermmul: Final[_FuncBinOp] = ...
  75. hermdiv: Final[_FuncBinOp] = ...
  76. hermpow: Final[_FuncPow] = ...
  77. hermder: Final[_FuncDer] = ...
  78. hermint: Final[_FuncInteg] = ...
  79. hermval: Final[_FuncVal] = ...
  80. hermval2d: Final[_FuncVal2D] = ...
  81. hermval3d: Final[_FuncVal3D] = ...
  82. hermgrid2d: Final[_FuncVal2D] = ...
  83. hermgrid3d: Final[_FuncVal3D] = ...
  84. hermvander: Final[_FuncVander] = ...
  85. hermvander2d: Final[_FuncVander2D] = ...
  86. hermvander3d: Final[_FuncVander3D] = ...
  87. hermfit: Final[_FuncFit] = ...
  88. hermcompanion: Final[_FuncCompanion] = ...
  89. hermroots: Final[_FuncRoots] = ...
  90. def _normed_hermite_n(x: np.ndarray[_ShapeT, np.dtype[np.float64]], n: int) -> np.ndarray[_ShapeT, np.dtype[np.float64]]: ...
  91. hermgauss: Final[_FuncGauss] = ...
  92. hermweight: Final[_FuncWeight] = ...
  93. class Hermite(ABCPolyBase[L["H"]]):
  94. basis_name: ClassVar[L["H"]] = "H" # pyright: ignore[reportIncompatibleMethodOverride]
  95. domain: _Array2[np.float64 | Any] = ... # pyright: ignore[reportIncompatibleMethodOverride]
  96. window: _Array2[np.float64 | Any] = ... # pyright: ignore[reportIncompatibleMethodOverride]