Element

Generic element factory

elementary.element.element_factory(vector: Callable[[...], tuple[jax.Array, jax.Array, jax.Array]] | None = None, scalar: Callable[[...], jax.Array] | None = None, curvature: Callable[[...], jax.Array] | None = None, torsion: Callable[[...], jax.Array] | None = None, hamiltonian: Callable[[...], jax.Array] | None = None, beta: float | None = None, gamma: float | None = None, driver: Callable[[...], jax.Array] | None = None, settings: dict | None = None, order: int = 0, iterations: int = 1, autonomous: bool = True, final: bool = True) → Callable[[...], jax.Array][source]

Generate generic element transfer map

Parameters:

  • vector (Optional[Callable[..., tuple[Array, Array, Array]]]) – normalized vector potential

  • scalar (Optional[Callable[..., Array]]) – normalized scalar potential

  • curvature (Optional[Callable[..., Array]]) – curvature

  • torsion (Optional[Callable[..., Array]]) – torsion

  • hamiltonian (Optional[Callable[..., Array]]) – generic hamiltonian (other callables are ignored)

  • beta (Optional[float]) – relativistic beta

  • gamma (Optional[float]) – relativistic gamma

  • driver (Optional[Callable[..., Array]]) – symplectic integrator (tao)

  • settings (Optional[dict]) – configuration settings for integrator

  • order (int, default=0) – yoshida composition order

  • iterations (int, default=1) – number of integration

  • autonomous (bool, default=True) – autonomous flag

  • final (bool, default=True) – flag to return only the final state

Return type:

Callable[…, Array]