8 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
9 from AthenaConfiguration.ComponentFactory
import CompFactory
12 """Main config for jet reconstruction and decorations"""
17 if "McEventCollection#GEN_EVENT" not in ConfigFlags.Input.TypedCollections:
27 """Jet reconstruction needed for PHYS/PHYSLITE"""
29 from JetRecConfig.StandardSmallRJets
import AntiKt4EMTopo,AntiKt4EMPFlow,AntiKtVR30Rmax4Rmin02PV0Track
30 from JetRecConfig.StandardLargeRJets
import AntiKt10UFOCSSKSoftDrop
31 from JetRecConfig.JetRecConfig
import JetRecCfg
35 AntiKt4EMTopo_deriv = AntiKt4EMTopo.clone(
36 modifiers = AntiKt4EMTopo.modifiers+(
"JetPtAssociation",
"QGTagging")
39 AntiKt4EMPFlow_deriv = AntiKt4EMPFlow.clone(
40 modifiers = AntiKt4EMPFlow.modifiers+(
"JetPtAssociation",
"QGTagging",
"fJVT",
"NNJVT",
"CaloEnergiesClus",
"JetPileupLabel",
"qgtransformer")
43 AntiKt10UFOCSSKSoftDrop_deriv = AntiKt10UFOCSSKSoftDrop.clone(
44 modifiers = AntiKt10UFOCSSKSoftDrop.modifiers+(
"toptransformer",)
47 jetList = [AntiKt4EMTopo_deriv, AntiKt4EMPFlow_deriv,
48 AntiKtVR30Rmax4Rmin02PV0Track,
49 AntiKt10UFOCSSKSoftDrop_deriv]
57 """Add bad batman decoration for events with large EMEC-IW noise"""
61 CommonAugmentation = CompFactory.DerivationFramework.CommonAugmentation
62 from DerivationFrameworkJetEtMiss.JetToolConfig
import BadBatmanToolCfg
64 acc.addEventAlgo(CommonAugmentation(
"BadBatmanAugmentation", AugmentationTools = [badBatmanTool]))
69 """Add distance in train information to EventInfo"""
70 from DerivationFrameworkJetEtMiss.JetToolConfig
import DistanceInTrainToolCfg
74 CommonAugmentation = CompFactory.DerivationFramework.CommonAugmentation
76 acc.addEventAlgo(CommonAugmentation(
"DistanceInTrainAugmentation", AugmentationTools = [distanceInTrainTool]))
81 """Special rho definitions for PFlow jets"""
82 from JetRecConfig.JetRecConfig
import getInputAlgs,getConstitPJGAlg,reOrderAlgs
83 from JetRecConfig.StandardJetConstits
import stdConstitDic
as cst
84 from JetRecConfig.JetInputConfig
import buildEventShapeAlg
88 constit_algs =
getInputAlgs(cst.GPFlow, flags=ConfigFlags)
89 constit_algs, ca =
reOrderAlgs( [a
for a
in constit_algs
if a
is not None])
92 for a
in constit_algs:
108 from JetJvtEfficiency.JetJvtEfficiencyToolConfig
import getJvtEffToolCfg
110 passJvtTool = acc.popToolsAndMerge(
getJvtEffToolCfg(ConfigFlags, jetContainer))
111 passJvtTool.PassJVTKey =
"{}Jets.DFCommonJets_passJvt".
format(jetContainer)
112 passJvtTool.SuppressOutputDependence =
False
113 kwargs.setdefault(
"Decorators", [passJvtTool])
114 kwargs.setdefault(
"JetContainer",
"{}Jets".
format(jetContainer))
115 acc.addEventAlgo(CompFactory.JetDecorationAlg(algName, **kwargs), primary =
True)
119 """Add event cleaning flags"""
122 acc.merge(
AddJvtDecorationAlgCfg(ConfigFlags, algName=
"JvtPassDecorAlg_EMTopo", jetContainer=
'AntiKt4EMTopo'))
125 from DerivationFrameworkTau.TauCommonConfig
import AddTauAugmentationCfg
127 acc.addSequence(CompFactory.AthSequencer(
'EventCleanSeq', Sequential=
True))
130 from AssociationUtils.AssociationUtilsConfig
import OverlapRemovalToolCfg
131 inputLabel_legacy =
'selected_eventClean_EMTopo'
132 outputLabel_legacy =
'DFCommonJets_passOR_EMTopo'
134 tauLabel =
'DFTauRNNLoose'
135 orTool_legacy = acc.popToolsAndMerge(
OverlapRemovalToolCfg(ConfigFlags,inputLabel=inputLabel_legacy,outputLabel=outputLabel_legacy,bJetLabel=bJetLabel))
136 algOR_legacy = CompFactory.OverlapRemovalGenUseAlg(
'OverlapRemovalGenUseAlg_EMTopo',
137 JetKey=
"AntiKt4EMTopoJets",
138 SelectionLabel=inputLabel_legacy,
139 OverlapLabel=outputLabel_legacy,
140 OverlapRemovalTool=orTool_legacy,
144 acc.addEventAlgo(algOR_legacy,
'EventCleanSeq')
147 inputLabel =
'selected_eventClean_EMPFlow'
148 outputLabel =
'DFCommonJets_passOR_EMPFlow'
149 orTool = acc.popToolsAndMerge(
OverlapRemovalToolCfg(ConfigFlags,inputLabel=inputLabel,outputLabel=outputLabel,bJetLabel=bJetLabel))
150 algOR = CompFactory.OverlapRemovalGenUseAlg(
'OverlapRemovalGenUseAlg',
151 SelectionLabel=inputLabel,
152 OverlapLabel=outputLabel,
153 OverlapRemovalTool=orTool,
156 acc.addEventAlgo(algOR,
'EventCleanSeq')
158 CommonAugmentation = CompFactory.DerivationFramework.CommonAugmentation
159 from DerivationFrameworkMuons.MuonsToolsConfig
import MuonJetDrToolCfg
160 muonJetDrTool = acc.getPrimaryAndMerge(
MuonJetDrToolCfg(ConfigFlags,
"MuonJetDrTool"))
161 acc.addEventAlgo(CommonAugmentation(
"DFCommonMuonsKernel2", AugmentationTools = [muonJetDrTool]),
'EventCleanSeq')
163 from JetSelectorTools.JetSelectorToolsConfig
import EventCleaningToolCfg,JetCleaningToolCfg
165 supportedWPs = [
'Loose',
'Tight',
'LooseLLP',
'VeryLooseLLP',
'SuperLooseLLP']
166 prefix =
"DFCommonJets_"
167 evt_lvl_suppWPs_PFlow = [
'LooseBad',
'TightBad']
169 for wp
in workingPoints:
170 if wp
not in supportedWPs:
173 cleaningLevel = wp +
'Bad'
176 cleaningLevel = wp.replace(
'LLP',
'BadLLP')
180 for evt_swp
in evt_lvl_suppWPs_PFlow:
181 if evt_swp == cleaningLevel:
186 if 'Loose' in cleaningLevel:
192 ConfigFlags,
'JetCleaningTool_'+cleaningLevel+
'_EMTopo',
193 'AntiKt4EMTopo', cleaningLevel,
False))
194 acc.addPublicTool(jetCleaningTool_legacy)
196 ConfigFlags,
'EventCleaningTool_'+wp+
'_EMTopo', cleaningLevel))
197 ecTool_legacy.JetCleanPrefix = prefix
198 ecTool_legacy.OrDecorator =
"passOR_EMTopo"
199 ecTool_legacy.JetContainer =
"AntiKt4EMTopoJets"
200 ecTool_legacy.JetCleaningTool = jetCleaningTool_legacy
201 acc.addPublicTool(ecTool_legacy)
203 eventCleanAlg_legacy = CompFactory.EventCleaningTestAlg(
'EventCleaningTestAlg_'+wp+
'_EMTopo',
204 EventCleaningTool=ecTool_legacy,
205 JetCollectionName=
"AntiKt4EMTopoJets",
206 EventCleanPrefix=prefix,
207 CleaningLevel=cleaningLevel,
209 acc.addEventAlgo(eventCleanAlg_legacy,
'EventCleanSeq')
214 ConfigFlags,
'JetCleaningTool_'+cleaningLevel,
215 'AntiKt4EMPFlowJets', cleaningLevel,
False))
216 acc.addPublicTool(jetCleaningTool)
218 ecTool = acc.popToolsAndMerge(
EventCleaningToolCfg(ConfigFlags,
'EventCleaningTool_' + wp, cleaningLevel))
219 ecTool.JetCleanPrefix = prefix
220 ecTool.OrDecorator =
"passOR_EMPFlow"
221 ecTool.JetContainer =
"AntiKt4EMPFlowJets"
222 ecTool.JetCleaningTool = jetCleaningTool
223 acc.addPublicTool(ecTool)
225 eventCleanAlg = CompFactory.EventCleaningTestAlg(
'EventCleaningTestAlg_'+wp,
226 EventCleaningTool=ecTool,
227 JetCollectionName=
"AntiKt4EMPFlowJets",
228 EventCleanPrefix=prefix,
229 CleaningLevel=cleaningLevel,
231 acc.addEventAlgo(eventCleanAlg,
'EventCleanSeq')
237 for item
in contentlist:
238 if item
not in slimhelper.AppendToDictionary:
239 slimhelper.AppendToDictionary.update({item:
'xAOD::JetContainer',
240 item+
"Aux":
'xAOD::JetAuxContainer'})
241 if item
in smartlist:
242 slimhelper.SmartCollections.append(item)
244 slimhelper.AllVariables.append(item)
252 slimhelper.ExtraVariables.append(
'CaloCalTopoClusters.calE.calEta.calPhi.calM')
255 if "LCOriginTopoClusters" not in slimhelper.AppendToDictionary:
256 slimhelper.AppendToDictionary.update({
"LCOriginTopoClusters":
'xAOD::CaloClusterContainer',
257 "LCOriginTopoClustersAux":
'xAOD::ShallowAuxContainer'})
258 slimhelper.ExtraVariables.append(
'LCOriginTopoClusters.calEta.calPhi.originalObjectLink')
261 if "EMOriginTopoClusters" not in slimhelper.AppendToDictionary:
262 slimhelper.AppendToDictionary.update({
"EMOriginTopoClusters":
'xAOD::CaloClusterContainer',
263 "EMOriginTopoClustersAux":
'xAOD::ShallowAuxContainer'})
264 slimhelper.ExtraVariables.append(
'EMOriginTopoClusters.calE.calEta.calPhi.originalObjectLink')