Skip to content

Commit

Permalink
Update Stub Files
Browse files Browse the repository at this point in the history
  • Loading branch information
cemitch99 authored and github-actions[bot] committed Oct 29, 2024
1 parent bcbdd53 commit 58b6a32
Show file tree
Hide file tree
Showing 2 changed files with 35 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 @@ -697,6 +697,7 @@ def push(
| elements.Marker
| elements.Multipole
| elements.NonlinearLens
| elements.PlaneXYRot
| elements.Programmable
| elements.PRot
| elements.Quad
Expand Down
34 changes: 34 additions & 0 deletions src/python/impactx/impactx_pybind/elements.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ __all__ = [
"Named",
"NonlinearLens",
"PRot",
"PlaneXYRot",
"Programmable",
"Quad",
"RFCavity",
Expand Down Expand Up @@ -688,6 +689,7 @@ class KnownElementsList:
| Marker
| Multipole
| NonlinearLens
| PlaneXYRot
| Programmable
| PRot
| Quad
Expand Down Expand Up @@ -723,6 +725,7 @@ class KnownElementsList:
| Marker
| Multipole
| NonlinearLens
| PlaneXYRot
| Programmable
| PRot
| Quad
Expand Down Expand Up @@ -759,6 +762,7 @@ class KnownElementsList:
| Marker
| Multipole
| NonlinearLens
| PlaneXYRot
| Programmable
| PRot
| Quad
Expand Down Expand Up @@ -935,6 +939,36 @@ class PRot(Named, Thin):
@phi_out.setter
def phi_out(self, arg1: float) -> None: ...

class PlaneXYRot(Named, Thin, Alignment):
def __init__(
self,
angle: float,
dx: float = 0,
dy: float = 0,
rotation: float = 0,
name: str | None = None,
) -> None:
"""
A rotation in the x-y plane.
"""
def __repr__(self) -> str: ...
def push(
self,
pc: impactx.impactx_pybind.ImpactXParticleContainer,
step: int = 0,
period: int = 0,
) -> None:
"""
Push first the reference particle, then all other particles.
"""
@property
def angle(self) -> float:
"""
Rotation angle (rad).
"""
@angle.setter
def angle(self, arg1: float) -> None: ...

class Programmable(Named):
ds: float
nslice: int
Expand Down

0 comments on commit 58b6a32

Please sign in to comment.