hermite_e.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 hermetrim
  29. __all__ = [
  30. "hermezero",
  31. "hermeone",
  32. "hermex",
  33. "hermedomain",
  34. "hermeline",
  35. "hermeadd",
  36. "hermesub",
  37. "hermemulx",
  38. "hermemul",
  39. "hermediv",
  40. "hermepow",
  41. "hermeval",
  42. "hermeder",
  43. "hermeint",
  44. "herme2poly",
  45. "poly2herme",
  46. "hermefromroots",
  47. "hermevander",
  48. "hermefit",
  49. "hermetrim",
  50. "hermeroots",
  51. "HermiteE",
  52. "hermeval2d",
  53. "hermeval3d",
  54. "hermegrid2d",
  55. "hermegrid3d",
  56. "hermevander2d",
  57. "hermevander3d",
  58. "hermecompanion",
  59. "hermegauss",
  60. "hermeweight",
  61. ]
  62. _ShapeT = TypeVar("_ShapeT", bound=_Shape)
  63. poly2herme: Final[_FuncPoly2Ortho] = ...
  64. herme2poly: Final[_FuncUnOp] = ...
  65. hermedomain: Final[_Array2[np.float64]] = ...
  66. hermezero: Final[_Array1[np.int_]] = ...
  67. hermeone: Final[_Array1[np.int_]] = ...
  68. hermex: Final[_Array2[np.int_]] = ...
  69. hermeline: Final[_FuncLine] = ...
  70. hermefromroots: Final[_FuncFromRoots] = ...
  71. hermeadd: Final[_FuncBinOp] = ...
  72. hermesub: Final[_FuncBinOp] = ...
  73. hermemulx: Final[_FuncUnOp] = ...
  74. hermemul: Final[_FuncBinOp] = ...
  75. hermediv: Final[_FuncBinOp] = ...
  76. hermepow: Final[_FuncPow] = ...
  77. hermeder: Final[_FuncDer] = ...
  78. hermeint: Final[_FuncInteg] = ...
  79. hermeval: Final[_FuncVal] = ...
  80. hermeval2d: Final[_FuncVal2D] = ...
  81. hermeval3d: Final[_FuncVal3D] = ...
  82. hermegrid2d: Final[_FuncVal2D] = ...
  83. hermegrid3d: Final[_FuncVal3D] = ...
  84. hermevander: Final[_FuncVander] = ...
  85. hermevander2d: Final[_FuncVander2D] = ...
  86. hermevander3d: Final[_FuncVander3D] = ...
  87. hermefit: Final[_FuncFit] = ...
  88. hermecompanion: Final[_FuncCompanion] = ...
  89. hermeroots: Final[_FuncRoots] = ...
  90. def _normed_hermite_e_n(x: np.ndarray[_ShapeT, np.dtype[np.float64]], n: int) -> np.ndarray[_ShapeT, np.dtype[np.float64]]: ...
  91. hermegauss: Final[_FuncGauss] = ...
  92. hermeweight: Final[_FuncWeight] = ...
  93. class HermiteE(ABCPolyBase[L["He"]]):
  94. basis_name: ClassVar[L["He"]] = "He" # pyright: ignore[reportIncompatibleMethodOverride]
  95. domain: _Array2[np.float64 | Any] = ... # pyright: ignore[reportIncompatibleMethodOverride]
  96. window: _Array2[np.float64 | Any] = ... # pyright: ignore[reportIncompatibleMethodOverride]