Skip to content
Snippets Groups Projects

Update tagging array qmtest including mctruth infos

Merged Davide Fazzini requested to merge dfazzini_update_tagging_test into master
@@ -61,6 +61,8 @@ def main(options: Options):
MCTRUTH_pions = lambda func: F.MAP_INPUT(Functor=func, Relations=mctruth_pions.MCAssocTable)
#make collection of functors
#N.B: a default value has to be defined for functors returning an integer (or an array
#of integers) in case the output is empty: e.g. "TRUEID" or "TagTr_TRUEKEY[nTags]".
variables_B = FC({
'THOR_MASS':
F.MASS,
@@ -77,12 +79,15 @@ def main(options: Options):
F.MAP_INPUT_ARRAY(Functor=F.PT, Relations=tagAlg_rels),
"TagTr_PHI[nTags]":
F.MAP_INPUT_ARRAY(Functor=F.PHI, Relations=tagAlg_rels),
#N.B.: an additional default value has to be added in the definition of
#MAP_INPUT_ARRAY in case of internal functors built via lambda functions
#in order to ensure that a valid output is always defined, e.g. MCTRUTH_pions.
"TagTr_TRUEID[nTags]":
F.VALUE_OR(0) @ F.MAP_INPUT_ARRAY(
F.VALUE_OR([0]) @ F.MAP_INPUT_ARRAY(
Functor=F.VALUE_OR(0) @ MCTRUTH_pions(F.PARTICLE_ID),
Relations=tagAlg_rels),
"TagTr_TRUEKEY[nTags]":
F.VALUE_OR(-1) @ F.MAP_INPUT_ARRAY(
F.VALUE_OR([-1]) @ F.MAP_INPUT_ARRAY(
Functor=F.VALUE_OR(-1) @ MCTRUTH_pions(F.OBJECT_KEY),
Relations=tagAlg_rels),
"TagTr_TRUEP[nTags]":
Loading