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