lightwin package

Automatically compensate cavity failures in linacs.

Subpackages

• beam_calculation package
  • Subpackages
    • envelope_1d package
      • Submodules
        • beam_parameters_factory module
        • element_envelope1d_parameters module
        • element_envelope1d_parameters_factory module
        • envelope_1d module
        • simulation_output_factory module
        • transfer_matrices_c module
        • transfer_matrices_p module
        • transfer_matrix_factory module
    • envelope_3d package
      • Submodules
        • beam_parameters_factory module
        • element_envelope3d_parameters module
        • element_envelope3d_parameters_factory module
        • envelope_3d module
        • simulation_output_factory module
        • transfer_matrices_p module
        • transfer_matrix_factory module
    • integrators package
      • Submodules
        • rk4 module
    • parameters package
      • Submodules
        • element_parameters module
        • factory module
    • simulation_output package
      • Submodules
        • factory module
        • simulation_output module
    • tracewin package
      • Submodules
        • beam_parameters_factory module
        • element_tracewin_parameters module
        • element_tracewin_parameters_factory module
        • simulation_output_factory module
        • tracewin module
        • transfer_matrix_factory module
  • Submodules
    • beam_calculator module
      • BeamCalculator
        • BeamCalculator._ids
        • BeamCalculator.__init__()
        • BeamCalculator._set_up_common_factories()
        • BeamCalculator._set_up_specific_factories()
        • BeamCalculator.run()
        • BeamCalculator.run_with_this()
        • BeamCalculator.post_optimisation_run_with_this()
        • BeamCalculator.init_solver_parameters()
        • BeamCalculator._generate_simulation_output()
        • BeamCalculator.reference_phase
        • BeamCalculator.is_a_multiparticle_simulation
        • BeamCalculator.is_a_3d_simulation
        • BeamCalculator.compute()
        • BeamCalculator.cavity_settings_factory
        • BeamCalculator._abc_impl
    • factory module
      • BEAM_CALCULATORS
      • BeamCalculatorsFactory
        • BeamCalculatorsFactory.__init__()
        • BeamCalculatorsFactory._set_out_folders()
        • BeamCalculatorsFactory._patch_to_remove_misunderstood_key()
        • BeamCalculatorsFactory.run()
        • BeamCalculatorsFactory.run_all()
        • BeamCalculatorsFactory._check_consistency_absolute_phases()
• config package
  • Submodules
    • beam module
      • test()
      • edit_configuration_dict_in_place()
    • beam_calculator module
      • IMPLEMENTED_BEAM_CALCULATORS
      • IMPLEMENTED_ENVELOPE1D_METHODS
      • IMPLEMENTED_ENVELOPE3D_METHODS
      • test()
      • edit_configuration_dict_in_place()
      • _test_envelope1d()
      • _edit_configuration_dict_in_place_envelope1d()
      • _test_tracewin()
      • _test_tracewin_executable()
      • _edit_configuration_dict_in_place_tracewin()
      • _find_file()
      • _test_envelope3d()
      • _edit_configuration_dict_in_place_envelope3d()
    • design_space module
      • test()
      • _test_from_file()
      • _test_not_from_file()
      • edit_configuration_dict_in_place()
      • _edit_configuration_dict_in_place_from_file()
      • _edit_configuration_dict_in_place_not_from_file()
    • evaluators module
      • IMPLEMENTED_EVALUATORS
      • test()
    • files module
      • test()
      • edit_configuration_dict_in_place()
      • _create_project_folders()
    • helper module
      • check_type()
      • dict_for_pretty_output()
      • find_file()
    • plots module
      • IMPLEMENTED_PLOTS
      • TO_UPDATE_OR_FIX
      • test()
    • wtf module
      • IMPLEMENTED_STRATEGIES
      • IMPLEMENTED_OBJECTIVE_PRESETS
      • IMPLEMENTED_OPTIMISATION_ALGORITHMS
      • test()
      • _test_cavity_selection()
      • _test_k_out_of_n()
      • _test_manual()
      • _test_l_neighboring_lattices()
      • _test_global()
      • _test_global_downstream()
• core package
  • Subpackages
    • accelerator package
      • Submodules
        • accelerator module
        • factory module
    • beam_parameters package
      • Subpackages
        • phase_space package
      • Submodules
        • beam_parameters module
        • factory module
        • helper module
        • initial_beam_parameters module
    • commands package
      • Submodules
        • adjust module
        • command module
        • dummy_command module
        • end module
        • error module
        • factory module
        • field_map_path module
        • freq module
        • helper module
        • lattice module
        • marker module
        • repeat_ele module
        • set_adv module
        • set_sync_phase module
        • shift module
        • steerer module
        • superpose_map module
    • elements package
      • Subpackages
        • field_maps package
      • Submodules
        • aperture module
        • bend module
        • diagnostic module
        • drift module
        • dummy module
        • edge module
        • element module
        • factory module
        • helper module
        • quad module
        • solenoid module
        • superposed_field_map module
        • thin_steering module
    • list_of_elements package
      • Submodules
        • factory module
        • helper module
        • list_of_elements module
    • transfer_matrix package
      • Submodules
        • factory module
        • transfer_matrix module
  • Submodules
    • electric_field module
      • compute_param_cav()
      • RfField
        • RfField.__init__()
        • RfField.has()
        • RfField.get()
        • RfField.set_rf_freq()
        • RfField.set_e_spat()
        • RfField.update_phi_0_abs_to_adapt_to_new_ref_phase()
      • NewRfField
        • NewRfField.__init__()
        • NewRfField.has()
        • NewRfField.get()
        • NewRfField.set_e_spat()
    • instruction module
      • Instruction
        • Instruction.n_attributes
        • Instruction.is_implemented
        • Instruction.__init__()
        • Instruction.line
        • Instruction.idx
        • Instruction._assert_correct_number_of_args()
        • Instruction.__str__()
        • Instruction.reinsert_optional_commands_in_line()
        • Instruction.name
        • Instruction.to_line()
        • Instruction.increment_dat_position()
        • Instruction.insert_line()
        • Instruction.insert_object()
        • Instruction._abc_impl
      • Dummy
        • Dummy.is_implemented
        • Dummy.__init__()
        • Dummy._abc_impl
      • Comment
        • Comment.__init__()
        • Comment._abc_impl
      • LineJump
        • LineJump._abc_impl
    • instructions_factory module
      • InstructionsFactory
        • InstructionsFactory.__init__()
        • InstructionsFactory.run()
        • InstructionsFactory._call_proper_factory()
        • InstructionsFactory._handle_lattice_and_section()
        • InstructionsFactory._check_every_elt_has_lattice_and_section()
        • InstructionsFactory._check_last_lattice_of_every_lattice_is_complete()
        • InstructionsFactory._filter_out_elements_to_dump()
    • particle module
      • ParticleInitialState
        • ParticleInitialState.w_kin
        • ParticleInitialState.phi_abs
        • ParticleInitialState.z_in
        • ParticleInitialState.synchronous
        • ParticleInitialState.tracewin_command
        • ParticleInitialState.__init__()
      • ParticleFullTrajectory
        • ParticleFullTrajectory.w_kin
        • ParticleFullTrajectory.phi_abs
        • ParticleFullTrajectory.synchronous
        • ParticleFullTrajectory.__post_init__()
        • ParticleFullTrajectory.__str__()
        • ParticleFullTrajectory.tracewin_command
        • ParticleFullTrajectory.compute_complementary_data()
        • ParticleFullTrajectory.has()
        • ParticleFullTrajectory.get()
        • ParticleFullTrajectory.__init__()
• evaluator package
  • Subpackages
    • simulation_output package
      • Submodules
        • factory module
        • presets module
        • simulation_output_evaluator module
  • Submodules
    • list_of_simulation_output_evaluators module
      • ListOfSimulationOutputEvaluators
        • ListOfSimulationOutputEvaluators.__init__()
        • ListOfSimulationOutputEvaluators.run()
        • ListOfSimulationOutputEvaluators._unpack_other_evals()
        • ListOfSimulationOutputEvaluators._set_indexes()
        • ListOfSimulationOutputEvaluators._set_columns()
        • ListOfSimulationOutputEvaluators._get_evaluations()
      • FaultScenarioSimulationOutputEvaluators
        • FaultScenarioSimulationOutputEvaluators.__init__()
        • FaultScenarioSimulationOutputEvaluators._set_evaluation_elements()
        • FaultScenarioSimulationOutputEvaluators._create_simulation_output_evaluators()
        • FaultScenarioSimulationOutputEvaluators.run()
        • FaultScenarioSimulationOutputEvaluators._to_pandas_dataframe()
        • FaultScenarioSimulationOutputEvaluators._format_evaluations()
        • FaultScenarioSimulationOutputEvaluators._output()
    • post_treaters module
      • do_nothing()
      • set_first_value_to()
      • difference()
      • relative_difference()
      • rms_error()
      • absolute()
      • scale_by()
      • maximum()
      • minimum()
      • take_last()
      • sum()
    • testers module
      • value_is_within_limits()
      • value_is_above()
      • value_is_below()
      • value_is()
    • types module
• failures package
  • Submodules
    • fault module
      • Fault
        • Fault.__init__()
        • Fault.fix()
        • Fault.update_elements_status()
        • Fault.pickle()
        • Fault.from_pickle()
    • fault_scenario module
      • FaultScenario
        • FaultScenario.__init__()
        • FaultScenario._set_faults()
        • FaultScenario._get_reference()
        • FaultScenario._set_cavities_to_rephase()
        • FaultScenario._set_optimisation_algorithms()
        • FaultScenario._break_cavities()
        • FaultScenario.fix_all()
        • FaultScenario._wrap_single_fix()
        • FaultScenario._update_rephased_cavities_status()
        • FaultScenario._transfer_phi0_from_ref_to_broken()
        • FaultScenario._reference_phase
        • FaultScenario._evaluate_fit_quality()
        • FaultScenario._set_evaluation_elements()
        • FaultScenario._simulations_that_should_be_compared()
        • FaultScenario.pickle()
        • FaultScenario.from_pickle()
      • fault_scenario_factory()
      • _force_element_to_index_method_creation()
    • helper module
      • _distance_to_ref()
      • _penalty()
      • sort_by_position()
      • remove_lists_with_less_than_n_elements()
      • gather()
      • nested_containing_desired()
    • set_of_cavity_settings module
      • SetOfCavitySettings
        • SetOfCavitySettings.__init__()
        • SetOfCavitySettings.from_cavity_settings()
        • SetOfCavitySettings.from_incomplete_set()
        • SetOfCavitySettings.re_set_elements_index_to_absolute_value()
        • SetOfCavitySettings._order()
      • _settings_getter()
    • strategy module
      • failed_and_compensating()
      • k_out_of_n()
      • l_neighboring_lattices()
      • manual()
      • global_compensation()
      • global_downstream()
      • COMPENSATING_SELECTOR
• optimisation package
  • Subpackages
    • algorithms package
      • Submodules
        • algorithm module
        • differential_evolution module
        • downhill_simplex module
        • downhill_simplex_penalty module
        • explorator module
        • factory module
        • least_squares module
        • least_squares_penalty module
        • nsga module
        • pso module
    • design_space package
      • Submodules
        • constraint module
        • design_space module
        • design_space_parameter module
        • factory module
        • helper module
        • variable module
    • objective package
      • Submodules
        • factory module
        • minimize_difference_with_ref module
        • minimize_mismatch module
        • objective module
        • position module
        • quantity_is_between module
• scripts package
  • Submodules
    • combine_solutions module
      • combine_bests()
      • _infer_an_output_folder()
      • _select_best_simulations()
      • _reconstruct_folder_names()
      • _concat_evaluations_files()
      • _gather_best_simulations_in_same_place()
      • _copy_or_create_symlink()
      • _load_evaluation()
      • main()
    • compare_beam_calculators module
      • output_comparison()
      • compare_beam_calculators()
    • compare_runs module
      • concatenate_evaluation_files()
      • _compare_one_quantity_all_simulations()
      • compare_simulations()
    • compute_lost_power_per_meter module
      • compute()
      • _treat_single()
      • _add_linear_losses()
      • _remove_doublons()
      • _filter_in_range_only()
      • _running_mean()
      • _meter_per_meter()
      • _treat_full_project()
      • _get_partran1_filepath()
      • get_partran1_paths()
      • _check_validity()
      • _plot_single()
      • _plot_several()
      • main()
    • generate_design_space_files module
      • generate_design_space_files()
    • reorder_output_figures module
      • _create_output_folder()
      • _move_single_simulation_output_figures()
      • reorder_output_figures()
      • main()
    • save_cavity_settings module
      • save_cavity_settings()
      • _settings_as_df()
      • _phases_in_deg()
      • _k_e()
    • scripts_shorthands module
      • compute_beam()
      • compute_beams()
• tracewin_utils package
  • Submodules
    • dat_files module
      • dat_filecontent_from_smaller_list_of_elements()
      • _is_needed_element()
      • _is_useful_command()
      • export_dat_filecontent()
    • electromagnetic_fields module
      • FIELD_GEOMETRIES
      • FIELD_TYPES
      • LOADABLE
      • load_electromagnetic_fields()
      • _load_electromagnetic_fields_for_cython()
      • _geom_to_field_map_type()
      • _get_filemaps_extensions()
      • _get_filemap_extensions()
      • _get_field_nature()
      • _get_type()
      • _get_field_components()
      • _load_field_map_file()
      • _is_a_valid_electric_field()
      • _rescale()
      • output_data_in_tw_fashion()
    • interface module
      • variables_to_command()
      • beam_calculator_to_command()
      • list_of_elements_to_command()
      • beam_parameters_to_command()
      • particle_initial_state_to_command()
      • set_of_cavity_settings_to_command()
      • _cavity_settings_to_command()
      • ARGS_POSITIONS
      • _alter_element()
      • _proper_type()
    • load module
      • load_dat_file()
      • _strip_comments()
      • _split_named_elements()
      • _split_weighted_elements()
      • slice_dat_line()
      • table_structure_file()
      • results()
      • electric_field_1d()
      • _get_number_of_cells()
      • transfer_matrices()
      • _transfer_matrix()
      • FIELD_MAP_LOADERS
• util package
  • Submodules
    • converters module
      • position()
      • energy()
      • longitudinal()
      • emittance()
      • twiss()
    • debug module
      • compute_error_transfer_matrix()
      • load_phase_space()
      • output_cavities()
      • _create_output_fit_dicts()
      • output_fit()
      • output_fit_progress()
    • dicts_output module
    • helper module
      • recursive_items()
      • recursive_getter()
      • flatten()
      • chunks()
      • remove_duplicates()
      • printc()
      • pd_output()
      • pascal_case()
      • get_constructor()
      • get_constructors()
      • resample()
      • range_vals()
      • range_vals_object()
      • save_energy_phase_tm()
      • save_vcav_and_phis()
    • log_manager module
      • LogFormatter
        • LogFormatter.COLOR_CODES
        • LogFormatter.RESET_CODE
        • LogFormatter.__init__()
        • LogFormatter.format()
      • main()
    • pandas_helper module
      • to_csv()
    • pass_beauty module
      • _cavity_settings_to_adjust()
      • set_of_cavity_settings_to_adjust()
      • elements_to_diagnostics()
      • _pass_beauty_instructions()
      • insert_pass_beauty_instructions()
      • _is_adapted_to_pass_beauty()
    • phases module
      • diff_angle()
      • phi_0_abs_to_rel()
      • phi_0_rel_to_abs()
      • phi_bunch_to_phi_rf()
      • phi_rf_to_phi_bunch()
    • pickling module
      • MyPickler
        • MyPickler.pickle()
        • MyPickler.unpickle()
        • MyPickler._abc_impl
      • MyCloudPickler
        • MyCloudPickler.__init__()
        • MyCloudPickler.pickle()
        • MyCloudPickler.unpickle()
        • MyCloudPickler._abc_impl
    • synchronous_phases module
      • phi_s_legacy()
      • phi_s_lagniel()
      • phi_s_from_tracewin_file()
      • SYNCHRONOUS_PHASE_FUNCTIONS
• visualization package
  • Submodules
    • anim module
      • AnimatedScatterDesign
        • AnimatedScatterDesign.__init__()
        • AnimatedScatterDesign.setup_plot()
        • AnimatedScatterDesign.update()
        • AnimatedScatterDesign.data_stream()
    • plot module
      • factory()
      • _plot_preset()
      • _proper_kwargs()
      • _keep_colors()
      • _y_label()
      • _single_simulation_data()
      • _single_simulation_all_data()
      • _single_accelerator_all_simulations_data()
      • _all_accelerators_data()
      • _avoid_similar_labels()
      • _error_calculation_function()
      • _compute_error()
      • _make_a_subplot()
      • create_fig_if_not_exists()
      • clean_figure()
      • clean_axes()
      • remove_artists()
      • _autoscale_based_on()
      • _savefig()
      • plot_pty_with_data_tags()
      • _plot_structure()
      • _plot_aperture()
      • _plot_bend()
      • _plot_drift()
      • _plot_field_map()
      • _plot_edge()
      • _plot_quad()
      • _plot_section()
      • _compute_ellipse_parameters()
      • plot_ellipse()
      • plot_ellipse_emittance()
      • plot_fit_progress()

Submodules

• config_manager module
  • MANDATORY_CONFIG_ENTRIES
  • OPTIONAL_CONFIG_ENTRIES
  • process_config()
  • _load_correct_toml_entries()
  • _override_some_toml_entries()
  • _process_config()
  • _make_global()
• constants module