3 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
4 from AthenaConfiguration.ComponentFactory
import CompFactory
5 from TrkConfig.VertexFindingFlags
import VertexSortingSetup
10 if "VertexWeightCalculator" not in kwargs:
11 from TrkConfig.TrkVertexWeightCalculatorsConfig
import (
12 SumPt2VertexWeightCalculatorCfg)
13 kwargs.setdefault(
"VertexWeightCalculator", acc.popToolsAndMerge(
16 kwargs.setdefault(
"decorationName",
"sumPt2")
19 CompFactory.Trk.VertexCollectionSortingTool(name,**kwargs))
25 if "VertexWeightCalculator" not in kwargs:
26 from TrkConfig.TrkVertexWeightCalculatorsConfig
import (
27 SumPtVertexWeightCalculatorCfg)
28 kwargs.setdefault(
"VertexWeightCalculator", acc.popToolsAndMerge(
31 kwargs.setdefault(
"decorationName",
"sumPt")
34 CompFactory.Trk.VertexCollectionSortingTool(name,**kwargs))
41 if "VertexWeightCalculator" not in kwargs:
42 from TrkConfig.TrkVertexWeightCalculatorsConfig
import (
43 JetRestrictedSumPt2VertexWeightCalculatorCfg)
44 kwargs.setdefault(
"VertexWeightCalculator", acc.popToolsAndMerge(
47 kwargs.setdefault(
"decorationName",
"sumPt2")
50 CompFactory.Trk.VertexCollectionSortingTool(name,**kwargs))
56 if "VertexWeightCalculator" not in kwargs:
57 from TrkConfig.TrkVertexWeightCalculatorsConfig
import GNNVertexWeightCalculatorCfg
58 kwargs.setdefault(
"VertexWeightCalculator", acc.popToolsAndMerge(
61 kwargs.setdefault(
"decorationName",
"gnnScore")
64 CompFactory.Trk.VertexCollectionSortingTool(name,**kwargs))
68 if flags.Tracking.PriVertex.sortingSetup == \
69 VertexSortingSetup.SumPt2Sorting:
71 elif flags.Tracking.PriVertex.sortingSetup == \
72 VertexSortingSetup.SumPtSorting:
74 elif flags.Tracking.PriVertex.sortingSetup == \
75 VertexSortingSetup.JetWeightedSumPt2Sorting:
83 if "VertexFitterTool" not in kwargs:
84 from TrkConfig.TrkVertexFittersConfig
import AdaptiveVertexFitterCfg
87 kwargs.setdefault(
"MininumDistance", 5.0)
88 kwargs.setdefault(
"CompatibilityDimension", 2)
91 acc.setPrivateTools(CompFactory.Trk.SecVertexMergingTool(name,**kwargs))