3 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
4 from AthenaConfiguration.ComponentFactory
import CompFactory
7 """Configure the seed jet builder"""
9 acc.setPrivateTools(CompFactory.SeedJetBuilder(name,
10 JetCollection = jetCollection
if jetCollection !=
"" else flags.DiTau.SeedJetCollection[0]))
15 """Configure the subjet builder"""
17 acc.setPrivateTools(CompFactory.SubjetBuilder(name,
18 Rsubjet = flags.DiTau.Rsubjet,
19 ptminsubjet = flags.DiTau.PtMinSubjet))
22 def TVAToolCfg(flags, name="TVATool_forDiTaus", **kwargs):
23 """Configure the TVA tool"""
26 kwargs.setdefault(
"TrackParticleContainer",
"InDetTrackParticles")
27 kwargs.setdefault(
"TrackVertexAssociation",
"JetTrackVtxAssoc_forDiTaus")
28 kwargs.setdefault(
"VertexContainer",
"PrimaryVertices")
29 kwargs.setdefault(
"MaxTransverseDistance", 2.5)
30 kwargs.setdefault(
"MaxLongitudinalDistance", 2)
32 acc.setPrivateTools(CompFactory.TrackVertexAssociationTool(name, **kwargs))
35 def JetAlgCfg(flags, name="DiTauRec_JetAlgorithm", **kwargs):
36 """Configure the JetAlgorithm"""
39 tools = [acc.popToolsAndMerge(
TVAToolCfg(flags))]
40 kwargs.setdefault(
"Tools", tools)
42 acc.addEventAlgo(CompFactory.JetAlgorithm(name, **kwargs))
47 """Configure the vertex finder"""
50 kwargs.setdefault(
"PrimVtxContainerName",
"PrimaryVertices")
51 kwargs.setdefault(
"AssociatedTracks",
"GhostTrack")
52 kwargs.setdefault(
"TrackVertexAssociation",
"JetTrackVtxAssoc_forDiTaus")
53 kwargs.setdefault(
"UseTJVA", flags.Tau.doTJVA)
55 acc.setPrivateTools(CompFactory.VertexFinder(name, **kwargs))
59 """Configure the di-tau track finder"""
62 kwargs.setdefault(
"MaxDrJet", 1.0)
63 kwargs.setdefault(
"MaxDrSubjet", 0.2)
64 kwargs.setdefault(
"MaxNTracksSubjet", -1)
65 kwargs.setdefault(
"TrackParticleContainer",
"InDetTrackParticles")
67 if "TrackSelectorTool" not in kwargs:
68 from InDetConfig.InDetTrackSelectorToolConfig
import TauRecInDetTrackSelectorToolCfg
70 acc.addPublicTool(InDetTrackSelectorTool)
71 kwargs.setdefault(
"TrackSelectorTool", InDetTrackSelectorTool)
73 acc.setPrivateTools(CompFactory.DiTauTrackFinder(name, **kwargs))
77 """Configure the cell finder"""
80 CellFinder = CompFactory.CellFinder(name,
81 Rsubjet = flags.DiTau.Rsubjet,)
82 acc.setPrivateTools(CellFinder)
86 """Configure the di-tau constituent finder"""
88 kwargs.setdefault(
"Rsubjet", 0.2)
89 kwargs.setdefault(
"UseRawConstit",
True)
91 acc.setPrivateTools(CompFactory.DiTauConstituentFinder(name, **kwargs))
95 """Configure the IDVarDecorator"""
97 acc.setPrivateTools(CompFactory.DiTauIDVarDecorator(name, **kwargs))
101 """Configure the OnnxScoreCalculator"""
103 kwargs.setdefault(
"onnxModelPath",
"DiTauRec/omni.onnx")
104 kwargs.setdefault(
"maxTracks", 10)
105 acc.setPrivateTools(CompFactory.DiTauOnnxDiscriminantTool(name, **kwargs))