ATLAS Offline Software
Functions
DiTauToolsConfig Namespace Reference

Functions

def SeedJetBuilderCfg (flags, name="DiTauRec_SeedJetBuilder", jetCollection="")
 
def SubjetBuilderCfg (flags, name="DiTauRec_SubjetBuilder")
 
def TVAToolCfg (flags, name="TVATool_forDiTaus", **kwargs)
 
def JetAlgCfg (flags, name="DiTauRec_JetAlgorithm", **kwargs)
 
def VertexFinderCfg (flags, name="DiTauRec_VertexFinder", **kwargs)
 
def DiTauTrackFinderCfg (flags, name="DiTauRec_DiTauTrackFinder", **kwargs)
 
def CellFinderCfg (flags, name="DiTauRec_CellFinder")
 
def DiTauConstituentFinderCfg (flags, name="DiTauRec_DiTauConstituentFinder", **kwargs)
 
def DiTauIDVarDecoratorCfg (flags, name="DiTauRec_IDVarDecorator", **kwargs)
 
def DiTauOnnxScoreCalculatorCfg (flags, name="DiTauRec_OnnxScoreCalculator", **kwargs)
 

Function Documentation

◆ CellFinderCfg()

def DiTauToolsConfig.CellFinderCfg (   flags,
  name = "DiTauRec_CellFinder" 
)
Configure the cell finder

Definition at line 76 of file DiTauToolsConfig.py.

76 def CellFinderCfg(flags, name="DiTauRec_CellFinder"):
77  """Configure the cell finder"""
78  acc = ComponentAccumulator()
79 
80  CellFinder = CompFactory.CellFinder(name,
81  Rsubjet = flags.DiTau.Rsubjet,)
82  acc.setPrivateTools(CellFinder)
83  return acc
84 

◆ DiTauConstituentFinderCfg()

def DiTauToolsConfig.DiTauConstituentFinderCfg (   flags,
  name = "DiTauRec_DiTauConstituentFinder",
**  kwargs 
)
Configure the di-tau constituent finder

Definition at line 85 of file DiTauToolsConfig.py.

85 def DiTauConstituentFinderCfg(flags, name="DiTauRec_DiTauConstituentFinder", **kwargs):
86  """Configure the di-tau constituent finder"""
87  acc = ComponentAccumulator()
88  kwargs.setdefault("Rsubjet", 0.2)
89  kwargs.setdefault("UseRawConstit", True)
90 
91  acc.setPrivateTools(CompFactory.DiTauConstituentFinder(name, **kwargs))
92  return acc
93 

◆ DiTauIDVarDecoratorCfg()

def DiTauToolsConfig.DiTauIDVarDecoratorCfg (   flags,
  name = "DiTauRec_IDVarDecorator",
**  kwargs 
)
Configure the IDVarDecorator

Definition at line 94 of file DiTauToolsConfig.py.

94 def DiTauIDVarDecoratorCfg(flags, name="DiTauRec_IDVarDecorator", **kwargs):
95  """Configure the IDVarDecorator"""
96  acc = ComponentAccumulator()
97  acc.setPrivateTools(CompFactory.DiTauIDVarDecorator(name, **kwargs))
98  return acc
99 

◆ DiTauOnnxScoreCalculatorCfg()

def DiTauToolsConfig.DiTauOnnxScoreCalculatorCfg (   flags,
  name = "DiTauRec_OnnxScoreCalculator",
**  kwargs 
)
Configure the OnnxScoreCalculator

Definition at line 100 of file DiTauToolsConfig.py.

100 def DiTauOnnxScoreCalculatorCfg(flags, name="DiTauRec_OnnxScoreCalculator", **kwargs):
101  """Configure the OnnxScoreCalculator"""
102  acc = ComponentAccumulator()
103  kwargs.setdefault("onnxModelPath", "DiTauRec/omni.onnx")
104  kwargs.setdefault("maxTracks", 10)
105  acc.setPrivateTools(CompFactory.DiTauOnnxDiscriminantTool(name, **kwargs))
106  return acc

◆ DiTauTrackFinderCfg()

def DiTauToolsConfig.DiTauTrackFinderCfg (   flags,
  name = "DiTauRec_DiTauTrackFinder",
**  kwargs 
)
Configure the di-tau track finder

Definition at line 58 of file DiTauToolsConfig.py.

58 def DiTauTrackFinderCfg(flags, name="DiTauRec_DiTauTrackFinder", **kwargs):
59  """Configure the di-tau track finder"""
60  acc = ComponentAccumulator()
61 
62  kwargs.setdefault("MaxDrJet", 1.0)
63  kwargs.setdefault("MaxDrSubjet", 0.2)
64  kwargs.setdefault("MaxNTracksSubjet", -1)
65  kwargs.setdefault("TrackParticleContainer", "InDetTrackParticles")
66 
67  if "TrackSelectorTool" not in kwargs:
68  from InDetConfig.InDetTrackSelectorToolConfig import TauRecInDetTrackSelectorToolCfg
69  InDetTrackSelectorTool = acc.popToolsAndMerge(TauRecInDetTrackSelectorToolCfg(flags))
70  acc.addPublicTool(InDetTrackSelectorTool)
71  kwargs.setdefault("TrackSelectorTool", InDetTrackSelectorTool)
72 
73  acc.setPrivateTools(CompFactory.DiTauTrackFinder(name, **kwargs))
74  return acc
75 

◆ JetAlgCfg()

def DiTauToolsConfig.JetAlgCfg (   flags,
  name = "DiTauRec_JetAlgorithm",
**  kwargs 
)
Configure the JetAlgorithm

Definition at line 35 of file DiTauToolsConfig.py.

35 def JetAlgCfg(flags, name="DiTauRec_JetAlgorithm", **kwargs): # Name changed wrt legacy config DiTauRec_TVATool
36  """Configure the JetAlgorithm"""
37  acc = ComponentAccumulator()
38 
39  tools = [acc.popToolsAndMerge(TVAToolCfg(flags))]
40  kwargs.setdefault("Tools", tools)
41 
42  acc.addEventAlgo(CompFactory.JetAlgorithm(name, **kwargs))
43  return acc
44 
45 # require TrackVertexAssociation to be produced by TVA tool - see above

◆ SeedJetBuilderCfg()

def DiTauToolsConfig.SeedJetBuilderCfg (   flags,
  name = "DiTauRec_SeedJetBuilder",
  jetCollection = "" 
)
Configure the seed jet builder

Definition at line 6 of file DiTauToolsConfig.py.

6 def SeedJetBuilderCfg(flags, name="DiTauRec_SeedJetBuilder", jetCollection=""):
7  """Configure the seed jet builder"""
9  acc.setPrivateTools(CompFactory.SeedJetBuilder(name,
10  JetCollection = jetCollection if jetCollection != "" else flags.DiTau.SeedJetCollection[0]))
11  return acc
12 
13 

◆ SubjetBuilderCfg()

def DiTauToolsConfig.SubjetBuilderCfg (   flags,
  name = "DiTauRec_SubjetBuilder" 
)
Configure the subjet builder

Definition at line 14 of file DiTauToolsConfig.py.

14 def SubjetBuilderCfg(flags, name="DiTauRec_SubjetBuilder"):
15  """Configure the subjet builder"""
16  acc = ComponentAccumulator()
17  acc.setPrivateTools(CompFactory.SubjetBuilder(name,
18  Rsubjet = flags.DiTau.Rsubjet,
19  ptminsubjet = flags.DiTau.PtMinSubjet))
20  return acc
21 

◆ TVAToolCfg()

def DiTauToolsConfig.TVAToolCfg (   flags,
  name = "TVATool_forDiTaus",
**  kwargs 
)
Configure the TVA tool

Definition at line 22 of file DiTauToolsConfig.py.

22 def TVAToolCfg(flags, name="TVATool_forDiTaus", **kwargs):
23  """Configure the TVA tool"""
24  acc = ComponentAccumulator()
25 
26  kwargs.setdefault("TrackParticleContainer", "InDetTrackParticles")
27  kwargs.setdefault("TrackVertexAssociation", "JetTrackVtxAssoc_forDiTaus")
28  kwargs.setdefault("VertexContainer", "PrimaryVertices")
29  kwargs.setdefault("MaxTransverseDistance", 2.5) # in mm
30  kwargs.setdefault("MaxLongitudinalDistance", 2) # in mm
31 
32  acc.setPrivateTools(CompFactory.TrackVertexAssociationTool(name, **kwargs))
33  return acc
34 

◆ VertexFinderCfg()

def DiTauToolsConfig.VertexFinderCfg (   flags,
  name = "DiTauRec_VertexFinder",
**  kwargs 
)
Configure the vertex finder

Definition at line 46 of file DiTauToolsConfig.py.

46 def VertexFinderCfg(flags, name="DiTauRec_VertexFinder", **kwargs):
47  """Configure the vertex finder"""
48  acc = ComponentAccumulator()
49 
50  kwargs.setdefault("PrimVtxContainerName", "PrimaryVertices")
51  kwargs.setdefault("AssociatedTracks", "GhostTrack")
52  kwargs.setdefault("TrackVertexAssociation", "JetTrackVtxAssoc_forDiTaus")
53  kwargs.setdefault("UseTJVA", flags.Tau.doTJVA)
54 
55  acc.setPrivateTools(CompFactory.VertexFinder(name, **kwargs))
56  return acc
57 
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
python.InDetTrackSelectorToolConfig.TauRecInDetTrackSelectorToolCfg
def TauRecInDetTrackSelectorToolCfg(flags, name='tauRec_InDetTrackSelectorTool', **kwargs)
Definition: InDetTrackSelectorToolConfig.py:197
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
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
DiTauToolsConfig.TVAToolCfg
def TVAToolCfg(flags, name="TVATool_forDiTaus", **kwargs)
Definition: DiTauToolsConfig.py:22