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 May 6, 2024
1 parent 5b8c191 commit 2b11f49
Show file tree
Hide file tree
Showing 2 changed files with 26 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 @@ -739,6 +739,7 @@ def push(
| elements.SoftSolenoid
| elements.SoftQuadrupole
| elements.Sol
| elements.TaperedPL
| elements.ThinDipole
),
step: int = 0,
Expand Down
25 changes: 25 additions & 0 deletions src/python/impactx/impactx_pybind/elements.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ __all__ = [
"SoftQuadrupole",
"SoftSolenoid",
"Sol",
"TaperedPL",
"Thick",
"Thin",
"ThinDipole",
Expand Down Expand Up @@ -570,6 +571,7 @@ class KnownElementsList:
| SoftSolenoid
| SoftQuadrupole
| Sol
| TaperedPL
| ThinDipole
),
) -> None: ...
Expand Down Expand Up @@ -610,6 +612,7 @@ class KnownElementsList:
| SoftSolenoid
| SoftQuadrupole
| Sol
| TaperedPL
| ThinDipole
),
) -> None:
Expand Down Expand Up @@ -923,6 +926,28 @@ class Sol(Thick, Alignment):
Push first the reference particle, then all other particles.
"""

class TaperedPL(Thin, Alignment):
def __init__(
self,
k: float,
taper: float,
units: int = 0,
dx: float = 0,
dy: float = 0,
rotation: float = 0,
) -> None:
"""
A thin nonlinear plasma lens with transverse taper.
"""

def __repr__(self) -> str: ...
def push(
self, pc: impactx.impactx_pybind.ImpactXParticleContainer, step: int = 0
) -> None:
"""
Push first the reference particle, then all other particles.
"""

class Thick:
def __init__(self, ds: float, nslice: float = 1) -> None:
"""
Expand Down

0 comments on commit 2b11f49

Please sign in to comment.