legendre.pyi 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. from typing import Any, ClassVar, Final, Literal as L
  2. import numpy as np
  3. from ._polybase import ABCPolyBase
  4. from ._polytypes import (
  5. _Array1,
  6. _Array2,
  7. _FuncBinOp,
  8. _FuncCompanion,
  9. _FuncDer,
  10. _FuncFit,
  11. _FuncFromRoots,
  12. _FuncGauss,
  13. _FuncInteg,
  14. _FuncLine,
  15. _FuncPoly2Ortho,
  16. _FuncPow,
  17. _FuncRoots,
  18. _FuncUnOp,
  19. _FuncVal,
  20. _FuncVal2D,
  21. _FuncVal3D,
  22. _FuncVander,
  23. _FuncVander2D,
  24. _FuncVander3D,
  25. _FuncWeight,
  26. )
  27. from .polyutils import trimcoef as legtrim
  28. __all__ = [
  29. "legzero",
  30. "legone",
  31. "legx",
  32. "legdomain",
  33. "legline",
  34. "legadd",
  35. "legsub",
  36. "legmulx",
  37. "legmul",
  38. "legdiv",
  39. "legpow",
  40. "legval",
  41. "legder",
  42. "legint",
  43. "leg2poly",
  44. "poly2leg",
  45. "legfromroots",
  46. "legvander",
  47. "legfit",
  48. "legtrim",
  49. "legroots",
  50. "Legendre",
  51. "legval2d",
  52. "legval3d",
  53. "leggrid2d",
  54. "leggrid3d",
  55. "legvander2d",
  56. "legvander3d",
  57. "legcompanion",
  58. "leggauss",
  59. "legweight",
  60. ]
  61. poly2leg: Final[_FuncPoly2Ortho] = ...
  62. leg2poly: Final[_FuncUnOp] = ...
  63. legdomain: Final[_Array2[np.float64]] = ...
  64. legzero: Final[_Array1[np.int_]] = ...
  65. legone: Final[_Array1[np.int_]] = ...
  66. legx: Final[_Array2[np.int_]] = ...
  67. legline: Final[_FuncLine] = ...
  68. legfromroots: Final[_FuncFromRoots] = ...
  69. legadd: Final[_FuncBinOp] = ...
  70. legsub: Final[_FuncBinOp] = ...
  71. legmulx: Final[_FuncUnOp] = ...
  72. legmul: Final[_FuncBinOp] = ...
  73. legdiv: Final[_FuncBinOp] = ...
  74. legpow: Final[_FuncPow] = ...
  75. legder: Final[_FuncDer] = ...
  76. legint: Final[_FuncInteg] = ...
  77. legval: Final[_FuncVal] = ...
  78. legval2d: Final[_FuncVal2D] = ...
  79. legval3d: Final[_FuncVal3D] = ...
  80. leggrid2d: Final[_FuncVal2D] = ...
  81. leggrid3d: Final[_FuncVal3D] = ...
  82. legvander: Final[_FuncVander] = ...
  83. legvander2d: Final[_FuncVander2D] = ...
  84. legvander3d: Final[_FuncVander3D] = ...
  85. legfit: Final[_FuncFit] = ...
  86. legcompanion: Final[_FuncCompanion] = ...
  87. legroots: Final[_FuncRoots] = ...
  88. leggauss: Final[_FuncGauss] = ...
  89. legweight: Final[_FuncWeight] = ...
  90. class Legendre(ABCPolyBase[L["P"]]):
  91. basis_name: ClassVar[L["P"]] = "P" # pyright: ignore[reportIncompatibleMethodOverride]
  92. domain: _Array2[np.float64 | Any] = ... # pyright: ignore[reportIncompatibleMethodOverride]
  93. window: _Array2[np.float64 | Any] = ... # pyright: ignore[reportIncompatibleMethodOverride]