def test_script_full():
# initialize the graph edges (intial and final state)
initial_state = [("Y", [-1, 1])]
final_state = ["D0", "D0bar", "pi0", "pi0"]
# because the amount of solutions is too big we change the default domains
formalism_type = 'canonical-helicity'
int_settings = create_default_interaction_settings(formalism_type)
change_qn_domain(int_settings[InteractionTypes.Strong],
InteractionQuantumNumberNames.L,
create_spin_domain([0, 1, 2, 3], True))
change_qn_domain(int_settings[InteractionTypes.Strong],
InteractionQuantumNumberNames.S,
create_spin_domain([0, 1, 2], True))
tbd_manager = StateTransitionManager(
initial_state,
final_state, ['D*'],
interaction_type_settings=int_settings,
formalism_type=formalism_type)
tbd_manager.set_allowed_interaction_types([InteractionTypes.Strong])
tbd_manager.add_final_state_grouping([['D0', 'pi0'], ['D0bar', 'pi0']])
tbd_manager.number_of_threads = 2
graph_node_setting_pairs = tbd_manager.prepare_graphs()
(solutions,
violated_rules) = tbd_manager.find_solutions(graph_node_setting_pairs)
print("found " + str(len(solutions)) + " solutions!")
canonical_xml_generator = CanonicalDecayAmplitudeGeneratorXML()
canonical_xml_generator.generate(solutions)
# because the amount of solutions is too big we change the default domains
formalism_type = 'helicity'
int_settings = create_default_interaction_settings(formalism_type)
change_qn_domain(int_settings[InteractionTypes.Strong],
InteractionQuantumNumberNames.L,
create_spin_domain([0, 1, 2, 3], True))
change_qn_domain(int_settings[InteractionTypes.Strong],
InteractionQuantumNumberNames.S,
create_spin_domain([0, 1, 2], True))
tbd_manager = StateTransitionManager(
initial_state,
final_state, ['D*'],
interaction_type_settings=int_settings,
formalism_type=formalism_type)
tbd_manager.set_allowed_interaction_types([InteractionTypes.Strong])
tbd_manager.add_final_state_grouping([['D0', 'pi0'], ['D0bar', 'pi0']])
tbd_manager.number_of_threads = 2
graph_node_setting_pairs = tbd_manager.prepare_graphs()
(solutions,
violated_rules) = tbd_manager.find_solutions(graph_node_setting_pairs)
print("found " + str(len(solutions)) + " solutions!")
helicity_xml_generator = HelicityDecayAmplitudeGeneratorXML()
helicity_xml_generator.generate(solutions)
#print(helicity_xml_generator.fit_parameters,
# canonical_xml_generator.fit_parameters)
assert (len(helicity_xml_generator.fit_parameters) == len(
canonical_xml_generator.fit_parameters))
def test_script_full():
# initialize the graph edges (intial and final state)
initial_state = [("Y", [-1, 1])]
final_state = ["D0", "D0bar", "pi0", "pi0"]
# because the amount of solutions is too big we change the default domains
formalism_type = 'canonical-helicity'
int_settings = create_default_interaction_settings(formalism_type)
change_qn_domain(int_settings[InteractionTypes.Strong],
InteractionQuantumNumberNames.L,
create_spin_domain([0, 1, 2, 3], True)
)
change_qn_domain(int_settings[InteractionTypes.Strong],
InteractionQuantumNumberNames.S,
create_spin_domain([0, 1, 2], True)
)
tbd_manager = StateTransitionManager(initial_state, final_state, ['D*'],
interaction_type_settings=int_settings,
formalism_type=formalism_type)
tbd_manager.set_allowed_interaction_types([InteractionTypes.Strong])
tbd_manager.add_final_state_grouping([['D0', 'pi0'], ['D0bar', 'pi0']])
tbd_manager.number_of_threads = 2
graph_node_setting_pairs = tbd_manager.prepare_graphs()
(solutions, violated_rules) = tbd_manager.find_solutions(
graph_node_setting_pairs)
print("found " + str(len(solutions)) + " solutions!")
canonical_xml_generator = CanonicalAmplitudeGeneratorXML()
canonical_xml_generator.generate(solutions)
# because the amount of solutions is too big we change the default domains
formalism_type = 'helicity'
int_settings = create_default_interaction_settings(formalism_type)
change_qn_domain(int_settings[InteractionTypes.Strong],
InteractionQuantumNumberNames.L,
create_spin_domain([0, 1, 2, 3], True)
)
change_qn_domain(int_settings[InteractionTypes.Strong],
InteractionQuantumNumberNames.S,
create_spin_domain([0, 1, 2], True)
)
tbd_manager = StateTransitionManager(initial_state, final_state, ['D*'],
interaction_type_settings=int_settings,
formalism_type=formalism_type)
tbd_manager.set_allowed_interaction_types([InteractionTypes.Strong])
tbd_manager.add_final_state_grouping([['D0', 'pi0'], ['D0bar', 'pi0']])
tbd_manager.number_of_threads = 2
graph_node_setting_pairs = tbd_manager.prepare_graphs()
(solutions, violated_rules) = tbd_manager.find_solutions(
graph_node_setting_pairs)
print("found " + str(len(solutions)) + " solutions!")
helicity_xml_generator = HelicityAmplitudeGeneratorXML()
helicity_xml_generator.generate(solutions)
assert (len(helicity_xml_generator.get_fit_parameters()) ==
len(canonical_xml_generator.get_fit_parameters()))
