ATLAS Offline Software
DiTauBuilderConfig.py
Go to the documentation of this file.
1 # Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
2 from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
3 from AthenaConfiguration.ComponentFactory import CompFactory
4 from DiTauRec.DiTauToolsConfig import (
5  SeedJetBuilderCfg,
6  SubjetBuilderCfg,
7  JetAlgCfg,
8  VertexFinderCfg,
9  DiTauTrackFinderCfg,
10  CellFinderCfg,
11  DiTauConstituentFinderCfg,
12  DiTauIDVarDecoratorCfg,
13  DiTauOnnxScoreCalculatorCfg
14 )
15 
16 def DiTauBuilderCfg(flags, name="DiTauBuilder", doLowPt=False):
17  acc = ComponentAccumulator()
18 
19  tools = [
20  acc.popToolsAndMerge(SeedJetBuilderCfg(flags)),
21  acc.popToolsAndMerge(SubjetBuilderCfg(flags))
22  ]
23 
24  if flags.Tracking.doVertexFinding: # Simplified wrt old config
25  acc.merge(JetAlgCfg(flags)) # To run TVA tool for VertexFinder
26  tools.append(acc.popToolsAndMerge(VertexFinderCfg(flags)))
27 
28  tools.append(acc.popToolsAndMerge(DiTauTrackFinderCfg(flags)))
29  if doLowPt:
30  tools.append(acc.popToolsAndMerge(DiTauConstituentFinderCfg(flags, UseRawConstit=True)))
31  else:
32  tools.append(acc.popToolsAndMerge(CellFinderCfg(flags)))
33 
34  if flags.DiTau.doRunDiTauDiscriminant:
35  tools.append(acc.popToolsAndMerge(DiTauIDVarDecoratorCfg(flags)))
36  tools.append(acc.popToolsAndMerge(DiTauOnnxScoreCalculatorCfg(flags)))
37 
38  acc.addEventAlgo(CompFactory.DiTauBuilder(name,
39  DiTauContainer = flags.DiTau.DiTauContainer[1] if doLowPt else flags.DiTau.DiTauContainer[0],
40  minPt = flags.DiTau.JetSeedPt[1] if doLowPt else flags.DiTau.JetSeedPt[0],
41  Tools = tools,
42  SeedJetName = flags.DiTau.SeedJetCollection[0],
43  maxEta = flags.DiTau.MaxEta,
44  Rjet = flags.DiTau.Rjet,
45  Rsubjet = flags.DiTau.Rsubjet,
46  Rcore = flags.DiTau.Rcore))
47  return acc
48 
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
DiTauToolsConfig.DiTauTrackFinderCfg
def DiTauTrackFinderCfg(flags, name="DiTauRec_DiTauTrackFinder", **kwargs)
Definition: DiTauToolsConfig.py:58
DiTauToolsConfig.SubjetBuilderCfg
def SubjetBuilderCfg(flags, name="DiTauRec_SubjetBuilder")
Definition: DiTauToolsConfig.py:14
DiTauToolsConfig.JetAlgCfg
def JetAlgCfg(flags, name="DiTauRec_JetAlgorithm", **kwargs)
Definition: DiTauToolsConfig.py:35
DiTauToolsConfig.VertexFinderCfg
def VertexFinderCfg(flags, name="DiTauRec_VertexFinder", **kwargs)
Definition: DiTauToolsConfig.py:46
DiTauToolsConfig.DiTauConstituentFinderCfg
def DiTauConstituentFinderCfg(flags, name="DiTauRec_DiTauConstituentFinder", **kwargs)
Definition: DiTauToolsConfig.py:85
DiTauToolsConfig.CellFinderCfg
def CellFinderCfg(flags, name="DiTauRec_CellFinder")
Definition: DiTauToolsConfig.py:76
DiTauBuilderConfig.DiTauBuilderCfg
def DiTauBuilderCfg(flags, name="DiTauBuilder", doLowPt=False)
Definition: DiTauBuilderConfig.py:16
DiTauToolsConfig.SeedJetBuilderCfg
def SeedJetBuilderCfg(flags, name="DiTauRec_SeedJetBuilder", jetCollection="")
Definition: DiTauToolsConfig.py:6
DiTauToolsConfig.DiTauIDVarDecoratorCfg
def DiTauIDVarDecoratorCfg(flags, name="DiTauRec_IDVarDecorator", **kwargs)
Definition: DiTauToolsConfig.py:94
DiTauToolsConfig.DiTauOnnxScoreCalculatorCfg
def DiTauOnnxScoreCalculatorCfg(flags, name="DiTauRec_OnnxScoreCalculator", **kwargs)
Definition: DiTauToolsConfig.py:100