3 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
4 from AthenaConfiguration.ComponentFactory
import CompFactory
5 from AthenaConfiguration.Enums
import BeamType, LHCPeriod
13 from CaloTools.CaloNoiseCondAlgConfig
import CaloNoiseCondAlgCfg
19 import tauRec.TauToolHolder
as tauTools
22 tools.append( result.popToolsAndMerge(tauTools.JetSeedBuilderCfg(flags)) )
25 if getattr(flags.Tau.ActiveConfig,
'inTauEleRM',
False):
26 tools.append( result.popToolsAndMerge(tauTools.TauEleOverlapChecker(flags)) )
29 if flags.Tau.isStandalone
or flags.Tracking.doVertexFinding:
30 tools.append( result.popToolsAndMerge(tauTools.TauVertexFinderCfg(flags)) )
32 tools.append( result.popToolsAndMerge(tauTools.TauAxisCfg(flags)) )
33 tools.append( result.popToolsAndMerge(tauTools.TauTrackFinderCfg(flags)) )
35 tools.append( result.popToolsAndMerge(tauTools.TauClusterFinderCfg(flags)) )
36 tools.append( result.popToolsAndMerge(tauTools.TauVertexedClusterDecoratorCfg(flags)) )
38 if flags.Beam.Type
is not BeamType.Cosmics:
39 if flags.Tau.doRNNTrackClass:
40 tools.append( result.popToolsAndMerge(tauTools.TauTrackRNNClassifierCfg(flags)) )
41 tools.append( result.popToolsAndMerge(tauTools.EnergyCalibrationLCCfg(flags)) )
43 tools.append( result.popToolsAndMerge(tauTools.CellVariablesCfg(flags)) )
44 tools.append( result.popToolsAndMerge(tauTools.ElectronVetoVarsCfg(flags)) )
45 tools.append( result.popToolsAndMerge(tauTools.TauShotFinderCfg(flags)) )
47 if flags.Tau.doPi0Clus:
48 tools.append( result.popToolsAndMerge(tauTools.Pi0ClusterFinderCfg(flags)) )
51 TauProcessorAlg = CompFactory.getComp(
"TauProcessorAlg")
52 BuildAlg =
TauProcessorAlg(name = flags.Tau.ActiveConfig.prefix+
"TauCoreBuilderAlg",
53 Key_jetInputContainer = flags.Tau.ActiveConfig.SeedJetCollection,
54 Key_tauOutputContainer = flags.Tau.ActiveConfig.TauJets_tmp,
55 Key_tauTrackOutputContainer = flags.Tau.ActiveConfig.TauTracks,
56 Key_tauShotClusOutputContainer = flags.Tau.ActiveConfig.TauShotClusters,
57 Key_tauShotClusLinkContainer = flags.Tau.ActiveConfig.TauShotClustersLinks,
58 Key_tauShotPFOOutputContainer = flags.Tau.ActiveConfig.TauShotPFOs,
59 Key_tauPi0CellOutputContainer = flags.Tau.ActiveConfig.TauCommonPi0Cells,
60 MaxEta = flags.Tau.SeedMaxEta,
61 MinPt = flags.Tau.SeedMinPt,
62 MaxNTracks = flags.Tau.MaxNTracks,
64 CellMakerTool = result.popToolsAndMerge(tauTools.TauCellFinalizerCfg(flags)))
66 if flags.GeoModel.Run
is LHCPeriod.Run4:
67 BuildAlg.PixelDetEleCollKey=
"ITkPixelDetectorElementCollection"
68 BuildAlg.SCTDetEleCollKey=
"ITkStripDetectorElementCollection"
69 BuildAlg.TRTDetEleContKey=
""
71 result.addEventAlgo(BuildAlg)
80 from CaloTools.CaloNoiseCondAlgConfig
import CaloNoiseCondAlgCfg
85 from CaloRec.CaloTopoClusterConfig
import caloTopoCoolFolderCfg
88 from LArBadChannelTool.LArBadChannelConfig
import LArBadChannelCfg
91 from TileConditions.TileBadChannelsConfig
import TileBadChannelsCondAlgCfg
95 import tauRec.TauToolHolder
as tauTools
97 CaloClusterMaker = CompFactory.getComp(
"CaloClusterMaker")
98 CaloTopoForTausMaker = CaloClusterMaker (flags.Tau.ActiveConfig.prefix+
"TauPi0SubtractedClusterMaker")
99 CaloTopoForTausMaker.ClustersOutputName = flags.Tau.ActiveConfig.TauPi0Clusters_tmp
100 CaloTopoForTausMaker.ClusterMakerTools = [result.popToolsAndMerge(tauTools.TauCaloTopoClusterMakerCfg(flags)),
101 result.popToolsAndMerge(tauTools.TauCaloTopoClusterSplitterCfg(flags))]
103 CaloTopoForTausMaker.ClusterCorrectionTools += [result.popToolsAndMerge(tauTools.TauCaloClusterBadChannelCfg(flags))]
104 CaloTopoForTausMaker.ClusterCorrectionTools += [result.popToolsAndMerge(tauTools.TauCaloClusterMomentsMakerCfg(flags))]
106 if flags.Calo.TopoCluster.doTopoClusterLocalCalib:
107 CaloTopoForTausMaker.ClusterCorrectionTools += [result.popToolsAndMerge(tauTools.TauCaloClusterLocalCalibCfg(flags)),
108 result.popToolsAndMerge(tauTools.TauCaloOOCCalibCfg(flags)),
109 result.popToolsAndMerge(tauTools.TauCaloOOCPi0CalibCfg(flags)),
110 result.popToolsAndMerge(tauTools.TauCaloDMCalibCfg(flags))]
112 result.addEventAlgo(CaloTopoForTausMaker)
114 relinkAlg = CompFactory.ClusterCellRelinkAlg(name = flags.Tau.ActiveConfig.prefix+
'ClusterCellRelinkAlg',
116 ClustersInput = flags.Tau.ActiveConfig.TauPi0Clusters_tmp,
117 ClustersOutput = flags.Tau.ActiveConfig.TauPi0Clusters,
118 CellLinksOutput = flags.Tau.ActiveConfig.TauPi0ClustersLinks)
119 result.addEventAlgo(relinkAlg)
128 import tauRec.TauToolHolder
as tauTools
132 tools.append( result.popToolsAndMerge(tauTools.Pi0ClusterCreatorCfg(flags)) )
133 tools.append( result.popToolsAndMerge(tauTools.Pi0ClusterScalerCfg(flags)) )
134 tools.append( result.popToolsAndMerge(tauTools.Pi0ScoreCalculatorCfg(flags)) )
135 tools.append( result.popToolsAndMerge(tauTools.Pi0SelectorCfg(flags)) )
137 if flags.Tau.doPanTau:
138 import PanTauAlgs.JobOptions_Main_PanTau
as pantau
139 tools.append( result.popToolsAndMerge(pantau.PanTauCfg(flags)) )
141 tools.append(result.popToolsAndMerge(tauTools.TauCombinedTESCfg(flags)) )
143 if flags.Beam.Type
is not BeamType.Cosmics:
144 tools.append( result.popToolsAndMerge(tauTools.MvaTESVariableDecoratorCfg(flags)) )
145 tools.append( result.popToolsAndMerge(tauTools.MvaTESEvaluatorCfg(flags)) )
148 tools.append( result.popToolsAndMerge(tauTools.TauAODSelectorCfg(flags)) )
151 if flags.Tau.isStandalone
or flags.Tracking.doVertexFinding:
152 tools.append(result.popToolsAndMerge(tauTools.TauVertexVariablesCfg(flags)) )
153 tools.append( result.popToolsAndMerge(tauTools.TauCommonCalcVarsCfg(flags)) )
154 tools.append( result.popToolsAndMerge(tauTools.TauSubstructureCfg(flags)) )
156 if flags.Tau.doTauDiscriminant:
157 tools.append( result.popToolsAndMerge(tauTools.TauIDVarCalculatorCfg(flags)) )
158 tools.append( result.popToolsAndMerge(tauTools.TauJetRNNEvaluatorCfg(flags)) )
159 tools.append( result.popToolsAndMerge(tauTools.TauWPDecoratorJetRNNCfg(flags)) )
160 tools.append( result.popToolsAndMerge(tauTools.TauEleRNNEvaluatorCfg(flags)) )
161 tools.append( result.popToolsAndMerge(tauTools.TauWPDecoratorEleRNNCfg(flags)) )
162 tools.append( result.popToolsAndMerge(tauTools.TauDecayModeNNClassifierCfg(flags)) )
164 if not flags.Tau.ActiveConfig.inTauEleRM:
166 tools.append( result.popToolsAndMerge(tauTools.TauGNNEvaluatorCfg(flags, version=0, applyTightTrackSel=
True)) )
167 tools.append( result.popToolsAndMerge(tauTools.TauWPDecoratorGNNCfg(flags, version=0, tauContainerName=flags.Tau.ActiveConfig.TauJets)) )
169 TauRunnerAlg = CompFactory.getComp(
"TauRunnerAlg")
170 RunnerAlg =
TauRunnerAlg(name = flags.Tau.ActiveConfig.prefix+
"TauRecRunnerAlg",
171 Key_tauInputContainer = flags.Tau.ActiveConfig.TauJets_tmp,
172 Key_Pi0ClusterInputContainer = flags.Tau.ActiveConfig.TauPi0Clusters,
173 Key_tauOutputContainer = flags.Tau.ActiveConfig.TauJets,
174 Key_neutralPFOOutputContainer = flags.Tau.ActiveConfig.TauNeutralPFOs,
175 Key_hadronicPFOOutputContainer = flags.Tau.ActiveConfig.TauHadronicPFOs,
176 Key_chargedPFOOutputContainer = flags.Tau.ActiveConfig.TauChargedPFOs,
177 Key_vertexOutputContainer = flags.Tau.ActiveConfig.TauSecondaryVertices,
178 Key_pi0Container = flags.Tau.ActiveConfig.TauFinalPi0s,
181 result.addEventAlgo(RunnerAlg)
187 from OutputStreamAthenaPool.OutputStreamConfig
import addToESD,addToAOD
192 TauAODList += [ f
"xAOD::TauJetContainer#{flags.Tau.ActiveConfig.TauJets}" ]
193 TauAODList += [ f
"xAOD::TauTrackContainer#{flags.Tau.ActiveConfig.TauTracks}" ]
194 TauAODList += [ f
"xAOD::TauTrackAuxContainer#{flags.Tau.ActiveConfig.TauTracks}Aux." ]
195 TauAODList += [ f
"xAOD::VertexContainer#{flags.Tau.ActiveConfig.TauSecondaryVertices}" ]
196 TauAODList += [ f
"xAOD::VertexAuxContainer#{flags.Tau.ActiveConfig.TauSecondaryVertices}Aux.-vxTrackAtVertex" ]
197 TauAODList += [ f
"xAOD::CaloClusterContainer#{flags.Tau.ActiveConfig.TauPi0Clusters}" ]
198 TauAODList += [ f
"xAOD::CaloClusterAuxContainer#{flags.Tau.ActiveConfig.TauPi0Clusters}Aux." ]
199 TauAODList += [ f
"CaloClusterCellLinkContainer#{flags.Tau.ActiveConfig.TauPi0Clusters}_links" ]
200 TauAODList += [ f
"xAOD::CaloClusterContainer#{flags.Tau.ActiveConfig.TauShotClusters}" ]
201 TauAODList += [ f
"xAOD::CaloClusterAuxContainer#{flags.Tau.ActiveConfig.TauShotClusters}Aux." ]
202 TauAODList += [ f
"CaloClusterCellLinkContainer#{flags.Tau.ActiveConfig.TauShotClusters}_links" ]
203 TauAODList += [ f
"xAOD::ParticleContainer#{flags.Tau.ActiveConfig.TauFinalPi0s}" ]
204 TauAODList += [ f
"xAOD::ParticleAuxContainer#{flags.Tau.ActiveConfig.TauFinalPi0s}Aux." ]
205 TauAODList += [ f
"xAOD::PFOContainer#{flags.Tau.ActiveConfig.TauShotPFOs}" ]
206 TauAODList += [ f
"xAOD::PFOAuxContainer#{flags.Tau.ActiveConfig.TauShotPFOs}Aux." ]
207 TauAODList += [ f
"xAOD::PFOContainer#{flags.Tau.ActiveConfig.TauNeutralPFOs}" ]
208 TauAODList += [ f
"xAOD::PFOAuxContainer#{flags.Tau.ActiveConfig.TauNeutralPFOs}Aux." ]
209 TauAODList += [ f
"xAOD::PFOContainer#{flags.Tau.ActiveConfig.TauHadronicPFOs}" ]
210 TauAODList += [ f
"xAOD::PFOAuxContainer#{flags.Tau.ActiveConfig.TauHadronicPFOs}Aux." ]
213 TauESDList =
list(TauAODList)
217 removeAODvars =
"-VertexedClusters.-mu.-nVtxPU.-ABS_ETA_LEAD_TRACK.-TAU_ABSDELTAPHI.-TAU_ABSDELTAETA.-absipSigLeadTrk.-passThinning.-chargedGlobalFELinks.-neutralGlobalFELinks"
218 if not flags.Tau.ActiveConfig.inTauEleRM:
219 removeAODvars += f
".-{flags.Tau.GNTauScoreName[0]}.-{flags.Tau.GNTauTransScoreName[0]}.-{flags.Tau.GNTauDecorWPNames[0][0]}.-{flags.Tau.GNTauDecorWPNames[0][1]}.-{flags.Tau.GNTauDecorWPNames[0][2]}.-{flags.Tau.GNTauDecorWPNames[0][3]}.-GNTauProbTau.-GNTauProbJet"
220 TauAODList += [
"xAOD::TauJetAuxContainer#{}Aux.{}".
format(flags.Tau.ActiveConfig.TauJets, removeAODvars) ]
223 removeESDvars =
"-VertexedClusters.-chargedGlobalFELinks.-neutralGlobalFELinks"
224 if not flags.Tau.ActiveConfig.inTauEleRM:
225 removeESDvars += f
".-{flags.Tau.GNTauScoreName[0]}.-{flags.Tau.GNTauTransScoreName[0]}.-{flags.Tau.GNTauDecorWPNames[0][0]}.-{flags.Tau.GNTauDecorWPNames[0][1]}.-{flags.Tau.GNTauDecorWPNames[0][2]}.-{flags.Tau.GNTauDecorWPNames[0][3]}.-GNTauProbTau.-GNTauProbJet"
226 TauESDList += [
"xAOD::TauJetAuxContainer#{}Aux.{}".
format(flags.Tau.ActiveConfig.TauJets, removeESDvars) ]
227 TauESDList += [
"xAOD::PFOContainer#{}" .
format(flags.Tau.ActiveConfig.TauChargedPFOs) ]
228 TauESDList += [
"xAOD::PFOAuxContainer#{}Aux." .
format(flags.Tau.ActiveConfig.TauChargedPFOs) ]
230 result.merge(
addToESD(flags,TauESDList))
231 result.merge(
addToAOD(flags,TauAODList))
237 from OutputStreamAthenaPool.OutputStreamConfig
import addToESD,addToAOD
240 DiTauOutputList = [
"xAOD::DiTauJetContainer#DiTauJets" ]
241 DiTauOutputList += [
"xAOD::DiTauJetAuxContainer#DiTauJetsAux." ]
243 result.merge(
addToESD(flags,DiTauOutputList))
244 result.merge(
addToAOD(flags,DiTauOutputList))
252 tauThinAlg = CompFactory.TauThinningAlg(name = flags.Tau.ActiveConfig.prefix+
"TauThinningAlg",
253 Taus = flags.Tau.ActiveConfig.TauJets,
254 TauTracks = flags.Tau.ActiveConfig.TauTracks,
255 TauNeutralPFOs = flags.Tau.ActiveConfig.TauNeutralPFOs,
256 TauPi0Clusters = flags.Tau.ActiveConfig.TauPi0Clusters,
257 TauPi0CellLinks = flags.Tau.ActiveConfig.TauPi0ClustersLinks,
258 TauFinalPi0s = flags.Tau.ActiveConfig.TauFinalPi0s,
259 TauShotPFOs = flags.Tau.ActiveConfig.TauShotPFOs,
260 TauShotClusters = flags.Tau.ActiveConfig.TauShotClusters,
261 TauShotCellLinks = flags.Tau.ActiveConfig.TauShotClustersLinks,
262 TauHadronicPFOs = flags.Tau.ActiveConfig.TauHadronicPFOs,
263 TauSecondaryVertices = flags.Tau.ActiveConfig.TauSecondaryVertices)
264 result.addEventAlgo(tauThinAlg)
273 from JetRecConfig.JetRecConfig
import JetRecCfg
274 from JetRecConfig.StandardJetConstits
import stdConstitDic
as cst
275 from JetRecConfig.JetDefinition
import JetDefinition
276 from JetRecConfig.StandardSmallRJets
import flavourghosts, calibmods_noCut, standardmods, truthmods
277 minimalghosts = [
"Track",
"MuonSegment",
"Truth"]
280 if flags.Tau.TauRec.SeedJetCollection ==
"AntiKt4EMPFlow10GeVCutTauSeedJets":
281 AntiKt4EMPFlow10GeVCutTauSeed = JetDefinition(
"AntiKt",0.4,cst.GPFlow,
282 infix =
"10GeVCutTauSeed",
283 ghostdefs = minimalghosts+flavourghosts,
284 modifiers = calibmods_noCut+(
"Filter:1",)+truthmods+standardmods+(
"JetPtAssociation",
"CaloEnergiesClus"),
287 result.merge(
JetRecCfg(flags, AntiKt4EMPFlow10GeVCutTauSeed))
288 if flags.Tau.TauRec.SeedJetCollection ==
"AntiKt4EMPFlow5GeVCutTauSeedJets":
289 AntiKt4EMPFlow5GeVCutTauSeed = JetDefinition(
"AntiKt",0.4,cst.GPFlow,
290 infix =
"5GeVCutTauSeed",
291 ghostdefs = minimalghosts+flavourghosts,
292 modifiers = calibmods_noCut+(
"Filter:1",)+truthmods+standardmods+(
"JetPtAssociation",
"CaloEnergiesClus"),
295 result.merge(
JetRecCfg(flags, AntiKt4EMPFlow5GeVCutTauSeed))
296 if flags.Tau.TauRec.SeedJetCollection ==
"AntiKt4EMPFlowNoPtCutTauSeedJets":
297 AntiKt4EMPFlowNoPtCutTauSeed = JetDefinition(
"AntiKt",0.4,cst.GPFlow,
298 infix =
"NoPtCutTauSeed",
299 ghostdefs = minimalghosts+flavourghosts,
300 modifiers = calibmods_noCut+(
"Filter:1",)+truthmods+standardmods+(
"JetPtAssociation",
"CaloEnergiesClus"),
303 result.merge(
JetRecCfg(flags, AntiKt4EMPFlowNoPtCutTauSeed))
308 flags_TauRec = flags.cloneAndReplace(
"Tau.ActiveConfig",
"Tau.TauRec")
316 if (flags.Output.doWriteESD
or flags.Output.doWriteAOD):
319 if (flags.Output.doWriteAOD
and flags.Tau.ThinTaus):
323 if flags.Tau.doTauEleRMRec:
325 flags_TauEleRM = flags.cloneAndReplace(
"Tau.ActiveConfig",
"Tau.TauEleRM")
330 from JetRecConfig.JetRecConfig
import JetRecCfg
331 if 'PFlow' in flags.Tau.TauRec.SeedJetCollection:
332 from JetRecConfig.JetRecConfig
import JetRecCfg
333 from JetRecConfig.StandardSmallRJets
import AntiKt4EMPFlow_tauSeedEleRM
334 result.merge(
JetRecCfg(flags_TauEleRM,AntiKt4EMPFlow_tauSeedEleRM ))
336 from JetRecConfig.StandardSmallRJets
import AntiKt4LCTopo
337 AntiKt4LCTopo_EleRM = AntiKt4LCTopo.clone(suffix=
"_EleRM")
338 AntiKt4LCTopo_EleRM.inputdef.name = flags_TauEleRM.Tau.ActiveConfig.LCTopoOrigin_EleRM
339 AntiKt4LCTopo_EleRM.inputdef.inputname = flags_TauEleRM.Tau.ActiveConfig.CaloCalTopoClusters_EleRM
340 AntiKt4LCTopo_EleRM.inputdef.containername = flags_TauEleRM.Tau.ActiveConfig.LCOriginTopoClusters_EleRM
341 AntiKt4LCTopo_EleRM.standardRecoMode =
True
342 AntiKt4LCTopo_EleRM.context =
"EleRM"
343 result.merge(
JetRecCfg(flags_TauEleRM, AntiKt4LCTopo_EleRM))
351 if (flags.Output.doWriteESD
or flags.Output.doWriteAOD):
354 if (flags.Output.doWriteAOD
and flags.Tau.ThinTaus):
358 if flags.Tau.doDiTauRec:
359 from DiTauRec.DiTauBuilderConfig
import DiTauBuilderCfg
362 if (flags.Output.doWriteESD
or flags.Output.doWriteAOD):
372 from ElectronPhotonSelectorTools.AsgElectronLikelihoodToolsConfig
import AsgElectronLikelihoodToolCfg
373 from ElectronPhotonSelectorTools.LikelihoodEnums
import LikeEnum
374 from ElectronPhotonSelectorTools.ElectronLikelihoodToolMapping
import electronLHmenu
375 ElectronLHSelectorEleRM = result.popToolsAndMerge(
378 name = flags.Tau.ActiveConfig.prefix+
"ElectronLHSelector",
379 quality = getattr(LikeEnum, flags.Tau.ActiveConfig.EleRM_ElectronWorkingPoint),
380 menu = electronLHmenu.offlineMC21,
384 tauElecSubtractAlg = CompFactory.TauElecSubtractAlg(
385 name = flags.Tau.ActiveConfig.prefix+
"TauElecSubtractAlg",
386 Key_ElectronsInput =
'Electrons',
387 Key_ClustersInput =
'CaloCalTopoClusters',
388 Key_ClustersOutput = flags.Tau.ActiveConfig.CaloCalTopoClusters_EleRM,
389 Key_IDTracksInput =
'InDetTrackParticles',
390 Key_IDTracksOutput = flags.Tau.ActiveConfig.TrackCollection,
391 Key_RemovedClustersOutput = flags.Tau.ActiveConfig.RemovedElectronClusters,
392 Key_RemovedTracksOutput = flags.Tau.ActiveConfig.RemovedElectronTracks,
393 ElectronLHTool = ElectronLHSelectorEleRM,
396 result.addEventAlgo(tauElecSubtractAlg)
403 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
404 from AthenaConfiguration.TestDefaults
import defaultTestFiles, defaultConditionsTags
408 flags.Input.Files = defaultTestFiles.RDO_RUN3
409 flags.IOVDb.GlobalTag = defaultConditionsTags.RUN3_MC
411 flags.Output.AODFileName =
"AOD.pool.root"
412 flags.Exec.MaxEvents = 50
414 flags.Scheduler.ShowDataDeps =
True
415 flags.Scheduler.ShowDataFlow =
True
416 flags.Scheduler.ShowControlFlow =
True
418 flags.Concurrency.NumThreads = 1
419 flags.Concurrency.NumConcurrentEvents = 1
421 from tauRec.ConfigurationHelpers
import StandaloneTauRecoFlags
426 from RecJobTransforms.RecoSteering
import RecoSteering
429 from RecJobTransforms.RecoConfigFlags
import printRecoFlags
435 if __name__==
"__main__":