Skip to content
Snippets Groups Projects
Commit 33788256 authored by amathad's avatar amathad
Browse files

test to array of ints also

parent d2098d5b
No related branches found
No related tags found
1 merge request!895Follow-up from https://gitlab.cern.ch/lhcb/Analysis/-/merge_requests/977
Pipeline #5515910 passed
......@@ -94,7 +94,8 @@ def main(options: Options):
long_electrons = make_long_electrons_with_brem()
relation_table_match_by_veloid = VeloIDOverlapRelationTable(
MatchFrom=basic_particles_dz_to_kpi, MatchTo=long_electrons)
MatchFrom=basic_particles_dz_to_kpi,
MatchTo=long_electrons).OutputRelations
# make electrons with some cuts
with reconstruction.bind(from_file=True, spruce=True):
......@@ -103,7 +104,8 @@ def main(options: Options):
Cut=F.FILTER(F.require_all(F.PT > 500. * MeV, F.PID_E > 1.)))
# make a test relation table
relation_table_match_by_veloid_tight_cuts = VeloIDOverlapRelationTable(
MatchFrom=basic_particles_dz_to_kpi, MatchTo=long_electrons_tight_cuts)
MatchFrom=basic_particles_dz_to_kpi,
MatchTo=long_electrons_tight_cuts).OutputRelations
my_filter = create_lines_filter(
name="HDRFilter_D0Kpi", lines=[f"{line_name}"])
......@@ -111,15 +113,16 @@ def main(options: Options):
variables["piplus"] += FunctorCollection({
'Matched_to_Electron_P[nMatchTracks]':
F.MAP_INPUT_ARRAY(
Functor=F.P,
Relations=relation_table_match_by_veloid.OutputRelations),
Functor=F.P, Relations=relation_table_match_by_veloid),
"Matched_to_Electron_Weight[nMatchTracks]":
F.MAP_WEIGHT(Relations=relation_table_match_by_veloid.OutputRelations),
F.MAP_WEIGHT(Relations=relation_table_match_by_veloid),
'Matched_to_Electron_P_tight_cuts[nMatchTracks_tight_cuts]':
F.MAP_INPUT_ARRAY(
Functor=F.P,
Relations=relation_table_match_by_veloid_tight_cuts.
OutputRelations),
Functor=F.P, Relations=relation_table_match_by_veloid_tight_cuts),
'Matched_to_Electron_ID_tight_cuts[nMatchTracks_tight_cuts]':
F.MAP_INPUT_ARRAY(
Functor=F.PARTICLE_ID,
Relations=relation_table_match_by_veloid_tight_cuts),
})
evt_variables = FC.SelectionInfo(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment