Skip to content

Commit

Permalink
Update Stub Files
Browse files Browse the repository at this point in the history
  • Loading branch information
ax3l authored and github-actions[bot] committed Feb 11, 2025
1 parent cdf408b commit 6f46573
Show file tree
Hide file tree
Showing 2 changed files with 49 additions and 0 deletions.
1 change: 1 addition & 0 deletions src/python/impactx/impactx_pybind/__init__.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -724,6 +724,7 @@ def push(
| elements.SoftSolenoid
| elements.SoftQuadrupole
| elements.Sol
| elements.Source
| elements.TaperedPL
| elements.ThinDipole,
step: int = 0,
Expand Down
48 changes: 48 additions & 0 deletions src/python/impactx/impactx_pybind/elements/__init__.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@ __all__ = [
"SoftQuadrupole",
"SoftSolenoid",
"Sol",
"Source",
"TaperedPL",
"ThinDipole",
"mixin",
Expand Down Expand Up @@ -882,6 +883,7 @@ class KnownElementsList:
| SoftSolenoid
| SoftQuadrupole
| Sol
| Source
| TaperedPL
| ThinDipole,
) -> None: ...
Expand Down Expand Up @@ -919,6 +921,7 @@ class KnownElementsList:
| SoftSolenoid
| SoftQuadrupole
| Sol
| Source
| TaperedPL
| ThinDipole
]: ...
Expand Down Expand Up @@ -957,6 +960,7 @@ class KnownElementsList:
| SoftSolenoid
| SoftQuadrupole
| Sol
| Source
| TaperedPL
| ThinDipole,
) -> None:
Expand Down Expand Up @@ -1710,6 +1714,50 @@ class Sol(mixin.Named, mixin.Thick, mixin.Alignment, mixin.PipeAperture):
@ks.setter
def ks(self, arg1: float) -> None: ...

class Source(mixin.Named, mixin.Thin):
@staticmethod
def _pybind11_conduit_v1_(*args, **kwargs): ...
def __init__(
self, distribution: str, openpmd_path: str, name: str | None = None
) -> None:
"""
A particle source.
"""
def __repr__(self) -> str: ...
@typing.overload
def push(
self,
pc: impactx.impactx_pybind.ImpactXParticleContainer,
step: int = 0,
period: int = 0,
) -> None:
"""
Push first the reference particle, then all other particles.
"""
@typing.overload
def push(
self,
cm: amrex.space3d.amrex_3d_pybind.SmallMatrix_6x6_F_SI1_double,
ref: impactx.impactx_pybind.RefPart,
) -> None:
"""
Linear push of the covariance matrix through an element. Expects that the reference particle was advanced first.
"""
@property
def distribution(self) -> str:
"""
Distribution type of particles in the source
"""
@distribution.setter
def distribution(self, arg1: str) -> None: ...
@property
def series_name(self) -> str:
"""
Path to openPMD series as accepted by openPMD_api.Series
"""
@series_name.setter
def series_name(self, arg1: str) -> None: ...

class TaperedPL(mixin.Named, mixin.Thin, mixin.Alignment):
@staticmethod
def _pybind11_conduit_v1_(*args, **kwargs): ...
Expand Down

0 comments on commit 6f46573

Please sign in to comment.