Alignment
Alignment transformations factory
- elementary.alignment.alignment_factory(beta: float | None = None, gamma: float | None = None, flag: bool = False) tuple[Callable[[...], jax.Array], Callable[[...], jax.Array]] [source]
Generate entrance and exit alignment error transformations
- Parameters:
beta (Optional[float]) – beta
gamma (Optional[float]) – gamma
flag (bool, default=False) – non-zero layout angle flag
- Return type:
tuple[Callable[…, Array], Callable[…, Array]]
- elementary.alignment.rx(qsps: jax.Array, wx: jax.Array, beta: float = 1.0, constant: float = 0.0) jax.Array [source]
RX rotation (sign matches MADX)
- Parameters:
qsps (Array) – initial state
wx (Array) – q_x rotation angle
beta (float, default=1.0) – beta
constant (float, default=0.0) – 1/(beta*gamma)**2
- Return type:
Array
- elementary.alignment.ry(qsps: jax.Array, wy: jax.Array, beta: float = 1.0, constant: float = 0.0) jax.Array [source]
RY rotation (sign matches MADX)
- Parameters:
qsps (Array) – initial state
wy (Array) – q_y rotation angle
beta (float, default=1.0) – beta
constant (float, default=0.0) – 1/(beta*gamma)**2
- Return type:
Array
- elementary.alignment.rz(qsps: jax.Array, wz: jax.Array) jax.Array [source]
RZ rotation (sign matches MADX)
- Parameters:
qsps (Array) – initial state
wz (Array) – q_z rotation angle
- Return type:
Array
- elementary.alignment.tx(qsps: jax.Array, dx: jax.Array, *args: jax.Array) jax.Array [source]
TX translation (sign matches MADX)
- Parameters:
qsps (Array) – initial state
dx (Array) – q_x translation error
- Return type:
Array
- elementary.alignment.ty(qsps: jax.Array, dy: jax.Array, *args: jax.Array) jax.Array [source]
TY translation (sign matches MADX)
- Parameters:
qsps (Array) – initial state
dy (Array) – q_y translation error
- Return type:
Array
- elementary.alignment.tz(qsps: jax.Array, dz: jax.Array, beta: float = 1.0, constant: float = 0.0) jax.Array [source]
TZ translation (sign matches MADX)
- Parameters:
qsps (Array) – initial state
dz (Array) – q_s translation error
beta (float, default=1.0) – beta
constant (float, default=0.0) – 1/(beta*gamma)**2
- Return type:
Array