beam_calculation.envelope_1d.transfer_matrix_factory module

Provide an easy way to generate TransferMatrix.

class TransferMatrixFactoryEnvelope1D(is_3d: bool)

Bases: TransferMatrixFactory

Provide a method for easy creation of TransferMatrix.

_abc_impl = <_abc._abc_data object>

_preprocess(single_elts_results: list[dict[str, Any]]) → ndarray

Preprocess the data given by the BeamCalculator.

run(first_cumulated_transfer_matrix: ndarray, single_elts_results: list[dict[str, Any]], element_to_index: Callable) → TransferMatrix

Create the transfer matrix from a simulation.

Parameters:

• first_cumulated_transfer_matrix (np.ndarray) – Cumulated transfer matrix at beginning of ListOfElements under study.

• single_elts_results (list[dict[str, Any]]) – Results of the solver.

Returns:

Holds all cumulated transfer matrices in all the planes.

Return type:

TransferMatrix