10 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
11 from AthenaConfiguration.ComponentFactory
import CompFactory
12 from DerivationFrameworkMCTruth.HFDSIDList
import DSIDList
17 """Configure and add the tools to compute and add the HF classifier in the derivation"""
31 JetMatchingTool = CompFactory.DerivationFramework.JetMatchingTool(name=
'DFCommonJetMatchingTool')
33 JetMatchingTool.jetPtCut = 15000.
34 JetMatchingTool.jetEtaCut = 2.5
35 JetMatchingTool.drCut = 0.3
39 acc.addPublicTool(JetMatchingTool, primary=
True)
49 HadronOriginClassifierTool = CompFactory.DerivationFramework.HadronOriginClassifier(name=
'DFCommonHadronOriginClassifier')
50 HadronOriginClassifierTool.DSID = mc_channel_number
54 acc.addPublicTool(HadronOriginClassifierTool, primary=
True)
67 ClassifyAndCalculateHFTool = CompFactory.DerivationFramework.ClassifyAndCalculateHFTool(name=
'DFCommonClassifyAndCalculateHFTool')
69 ClassifyAndCalculateHFTool.jetPtCut = 15000.
70 ClassifyAndCalculateHFTool.jetEtaCut = 2.5
71 ClassifyAndCalculateHFTool.leadingHadronPtCut = 5000.
72 ClassifyAndCalculateHFTool.leadingHadronPtRatioCut = -1
76 acc.addPublicTool(ClassifyAndCalculateHFTool, primary=
True)
93 ClassifyAndCalculateHFAugmentation = CompFactory.DerivationFramework.ClassifyAndCalculateHFAugmentation(name =
"DFCommonClassifyAndCalculateHFAugmentation")
95 ClassifyAndCalculateHFAugmentation.jetCollectionName =
"AntiKt4TruthDressedWZJets"
96 ClassifyAndCalculateHFAugmentation.TruthParticleContainerName =
"TruthParticles"
97 ClassifyAndCalculateHFAugmentation.hfDecorationName =
"HF_Classification"
98 ClassifyAndCalculateHFAugmentation.SimplehfDecorationName =
"HF_SimpleClassification"
99 ClassifyAndCalculateHFAugmentation.HadronOriginIDDecorationName =
"HFHadronOriginID"
101 ClassifyAndCalculateHFAugmentation.ClassifyAndComputeHFtool = ClassifyAndCalculateHFTool
102 ClassifyAndCalculateHFAugmentation.HadronOriginClassifierTool = HadronOriginClassifierTool
103 ClassifyAndCalculateHFAugmentation.JetMatchingTool = JetMatchingTool
107 acc.addPublicTool(ClassifyAndCalculateHFAugmentation, primary =
True)
117 """HF Classification configuration."""
123 mc_channel_number =
int(flags.Input.MCChannelNumber)
125 if mc_channel_number > 0:
127 if mc_channel_number
in DSIDList:
134 CommonAugmentation = CompFactory.DerivationFramework.CommonAugmentation
135 acc.addEventAlgo(CommonAugmentation(name =
"HFClassificationCommonKernel",
136 AugmentationTools = [DFCommonClassifyAndCalculateHFAugmentation]))