4 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
5 from AthenaConfiguration.ComponentFactory
import CompFactory
6 from AthenaConfiguration.Enums
import LHCPeriod
7 from TrkConfig.TrkConfigFlags
import PrimaryPassConfig
16 if "Extrapolator" not in kwargs:
17 from TrkConfig.AtlasExtrapolatorConfig
import InDetExtrapolatorCfg
18 kwargs.setdefault(
"Extrapolator", acc.popToolsAndMerge(
21 if 'DriftCircleCutTool' not in kwargs:
22 from InDetConfig.InDetTrackSelectorToolConfig
import (
23 InDetTRTDriftCircleCutToolCfg)
24 kwargs.setdefault(
"DriftCircleCutTool", acc.popToolsAndMerge(
27 have_calo_rois = (flags.Tracking.doCaloSeededBrem
and
28 kwargs.get(
"doEmCaloSeed",
True))
31 from InDetConfig.InDetCaloClusterROISelectorConfig
import (
32 CaloClusterROIPhiRZContainerMakerCfg)
34 kwargs.setdefault(
"EMROIPhiRZContainer",
35 "InDetCaloClusterROIPhiRZ5GeV")
37 kwargs.setdefault(
"doEmCaloSeed", have_calo_rois)
38 kwargs.setdefault(
"useAmbigFcn",
True)
39 kwargs.setdefault(
"useTRT_AmbigFcn",
False)
40 kwargs.setdefault(
"maxZImp", flags.Tracking.ActiveConfig.maxZImpact)
41 kwargs.setdefault(
"maxEta", flags.Tracking.ActiveConfig.maxEta)
42 kwargs.setdefault(
"usePixel", flags.Tracking.ActiveConfig.usePixel)
43 kwargs.setdefault(
"useSCT", flags.Tracking.ActiveConfig.useSCT)
45 acc.setPrivateTools(CompFactory.InDet.InDetAmbiScoringTool(name, **kwargs))
50 kwargs.setdefault(
"minTRTonTrk", 0)
51 kwargs.setdefault(
"minTRTPrecisionFraction", 0)
52 kwargs.setdefault(
"minPt", flags.Tracking.ActiveConfig.minPT)
53 kwargs.setdefault(
"maxRPhiImp",
54 flags.Tracking.ActiveConfig.maxPrimaryImpact)
55 kwargs.setdefault(
"maxRPhiImpEM", flags.Tracking.ActiveConfig.maxEMImpact)
56 kwargs.setdefault(
"minSiClusters", flags.Tracking.ActiveConfig.minClusters)
57 kwargs.setdefault(
"minPixel", flags.Tracking.ActiveConfig.minPixel)
58 kwargs.setdefault(
"maxSiHoles", flags.Tracking.ActiveConfig.maxHoles)
59 kwargs.setdefault(
"maxPixelHoles",
60 flags.Tracking.ActiveConfig.maxPixelHoles)
61 kwargs.setdefault(
"maxSCTHoles", flags.Tracking.ActiveConfig.maxSctHoles)
62 kwargs.setdefault(
"maxDoubleHoles",
63 flags.Tracking.ActiveConfig.maxDoubleHoles)
66 flags, name + flags.Tracking.ActiveConfig.extension, **kwargs)
70 kwargs.setdefault(
'DriftCircleCutTool',
None)
75 kwargs.setdefault(
"minTRTonTrk",
76 flags.Tracking.ActiveConfig.minTRTonTrk)
77 kwargs.setdefault(
"minTRTPrecisionFraction",
78 flags.Tracking.ActiveConfig.minTRTPrecFrac)
83 flags, name='InDetTRT_SeededScoringTool', **kwargs):
84 kwargs.setdefault(
"useAmbigFcn",
False)
85 kwargs.setdefault(
"useTRT_AmbigFcn",
True)
86 kwargs.setdefault(
"minTRTonTrk", flags.Tracking.BackTracking.minTRT)
87 kwargs.setdefault(
"minTRTPrecisionFraction",
88 flags.Tracking.BackTracking.minTRTPrecFrac)
89 kwargs.setdefault(
"minPt", flags.Tracking.BackTracking.minPt)
90 kwargs.setdefault(
"maxRPhiImp", flags.Tracking.BackTracking.maxSecondaryImpact)
91 kwargs.setdefault(
"minSiClusters", flags.Tracking.BackTracking.minClusters)
92 kwargs.setdefault(
"maxSiHoles", flags.Tracking.BackTracking.maxHoles)
93 kwargs.setdefault(
"maxPixelHoles", flags.Tracking.BackTracking.maxPixelHoles)
94 kwargs.setdefault(
"maxSCTHoles", flags.Tracking.BackTracking.maxSCTHoles)
95 kwargs.setdefault(
"maxDoubleHoles", flags.Tracking.BackTracking.maxDoubleHoles)
102 name='TrigAmbiguityScoringTool', **kwargs):
106 if flags.Tracking.ActiveConfig.input_name ==
"cosmics":
107 ambiScoringFactory = CompFactory.InDet.InDetCosmicScoringTool
108 kwargs.setdefault(
"nWeightedClustersMin",
109 flags.Tracking.ActiveConfig.nWeightedClustersMin)
110 kwargs.setdefault(
"minTRTHits", 0)
113 ambiScoringFactory = CompFactory.InDet.InDetAmbiScoringTool
115 if "Extrapolator" not in kwargs:
117 from TrkConfig.AtlasExtrapolatorConfig
import InDetExtrapolatorCfg
118 kwargs.setdefault(
"Extrapolator", acc.popToolsAndMerge(
121 if "DriftCircleCutTool" not in kwargs:
122 from InDetConfig.InDetTrackSelectorToolConfig
import (
123 InDetTrigTRTDriftCircleCutToolCfg)
124 kwargs.setdefault(
"DriftCircleCutTool", acc.popToolsAndMerge(
127 kwargs.setdefault(
"minPt", flags.Tracking.ActiveConfig.minPT)
128 kwargs.setdefault(
"useAmbigFcn",
True)
129 kwargs.setdefault(
"useTRT_AmbigFcn",
False)
130 kwargs.setdefault(
"maxZImp", flags.Tracking.ActiveConfig.maxZImpact)
131 kwargs.setdefault(
"maxRPhiImp", flags.Tracking.ActiveConfig.maxRPhiImpact)
132 kwargs.setdefault(
"maxEta", flags.Tracking.ActiveConfig.maxEta)
133 kwargs.setdefault(
"maxSCTHoles", flags.Tracking.ActiveConfig.maxSCTHoles)
134 kwargs.setdefault(
"maxSiHoles", flags.Tracking.ActiveConfig.maxSiHoles)
135 kwargs.setdefault(
"usePixel", flags.Tracking.ActiveConfig.usePixel)
136 kwargs.setdefault(
"useSCT", flags.Tracking.ActiveConfig.useSCT)
137 kwargs.setdefault(
"doEmCaloSeed",
False)
138 kwargs.setdefault(
"EMROIPhiRZContainer",
"")
139 kwargs.setdefault(
"minTRTonTrk", 0)
140 kwargs.setdefault(
"minTRTPrecisionFraction", 0)
143 acc.setPrivateTools(ambiScoringFactory(
144 name=name+flags.Tracking.ActiveConfig.input_name,
153 kwargs.setdefault(
"nWeightedClustersMin",
154 flags.Tracking.ActiveConfig.nWeightedClustersMin)
155 kwargs.setdefault(
"minTRTHits", 0)
157 acc.setPrivateTools(CompFactory.InDet.InDetCosmicScoringTool(
158 name+flags.Tracking.ActiveConfig.extension, **kwargs))
163 flags, name='InDetCosmicExtenScoringTool', **kwargs):
164 kwargs.setdefault(
"nWeightedClustersMin", 0)
165 kwargs.setdefault(
"minTRTHits", flags.Tracking.ActiveConfig.minTRTonTrk)
170 flags, name='InDetCosmicScoringTool_TRT', **kwargs):
171 kwargs.setdefault(
"minTRTHits", flags.Tracking.BackTracking.minTRT)
178 if flags.Tracking.doCaloSeededBrem:
179 from InDetConfig.InDetCaloClusterROISelectorConfig
import (
180 CaloClusterROIPhiRZContainerMakerCfg)
182 kwargs.setdefault(
"EMROIPhiRZContainer",
183 "InDetCaloClusterROIPhiRZ5GeV")
185 if "Extrapolator" not in kwargs:
186 from TrkConfig.AtlasExtrapolatorConfig
import InDetExtrapolatorCfg
187 kwargs.setdefault(
"Extrapolator", acc.popToolsAndMerge(
190 if 'DriftCircleCutTool' not in kwargs:
191 from InDetConfig.InDetTrackSelectorToolConfig
import (
192 InDetTRTDriftCircleCutToolCfg)
193 kwargs.setdefault(
"DriftCircleCutTool", acc.popToolsAndMerge(
196 kwargs.setdefault(
"nnCutConfig",
197 "dev/TrackingCP/LRTAmbiNetwork/20200727_225401/nn-config.json")
198 kwargs.setdefault(
"nnCutThreshold",
199 flags.Tracking.nnCutLargeD0Threshold)
201 kwargs.setdefault(
"useAmbigFcn",
True)
202 kwargs.setdefault(
"useTRT_AmbigFcn",
False)
203 kwargs.setdefault(
"maxZImp", flags.Tracking.ActiveConfig.maxZImpact)
204 kwargs.setdefault(
"maxEta", flags.Tracking.ActiveConfig.maxEta)
205 kwargs.setdefault(
"usePixel", flags.Tracking.ActiveConfig.usePixel)
206 kwargs.setdefault(
"useSCT", flags.Tracking.ActiveConfig.useSCT)
207 kwargs.setdefault(
"doEmCaloSeed", flags.Tracking.doCaloSeededBrem)
209 acc.setPrivateTools(CompFactory.InDet.InDetNNScoringTool(name, **kwargs))
214 kwargs.setdefault(
"minTRTonTrk", 0)
215 kwargs.setdefault(
"minTRTPrecisionFraction", 0)
216 kwargs.setdefault(
"minPt", flags.Tracking.ActiveConfig.minPT)
217 kwargs.setdefault(
"maxRPhiImp",
218 flags.Tracking.ActiveConfig.maxPrimaryImpact)
219 kwargs.setdefault(
"maxRPhiImpEM", flags.Tracking.ActiveConfig.maxEMImpact)
220 kwargs.setdefault(
"minSiClusters", flags.Tracking.ActiveConfig.minClusters)
221 kwargs.setdefault(
"minPixel", flags.Tracking.ActiveConfig.minPixel)
222 kwargs.setdefault(
"maxSiHoles", flags.Tracking.ActiveConfig.maxHoles)
223 kwargs.setdefault(
"maxPixelHoles",
224 flags.Tracking.ActiveConfig.maxPixelHoles)
225 kwargs.setdefault(
"maxSCTHoles", flags.Tracking.ActiveConfig.maxSctHoles)
226 kwargs.setdefault(
"maxDoubleHoles",
227 flags.Tracking.ActiveConfig.maxDoubleHoles)
230 flags, name+flags.Tracking.ActiveConfig.extension, **kwargs)
234 kwargs.setdefault(
'DriftCircleCutTool',
None)
244 if "DriftCircleCutTool" not in kwargs:
245 from InDetConfig.InDetTrackSelectorToolConfig
import (
246 InDetTRTDriftCircleCutToolCfg)
247 InDetTRTDriftCircleCut = acc.popToolsAndMerge(
249 acc.addPublicTool(InDetTRTDriftCircleCut)
250 kwargs.setdefault(
"DriftCircleCutTool", InDetTRTDriftCircleCut)
252 kwargs.setdefault(
"useAmbigFcn",
True)
253 kwargs.setdefault(
"useSigmaChi2",
False)
254 kwargs.setdefault(
"PtMin", flags.Tracking.TRTStandalone.minPt)
255 kwargs.setdefault(
"minTRTonTrk", flags.Tracking.TRTStandalone.minTRT)
256 kwargs.setdefault(
"maxEta", 2.1)
257 kwargs.setdefault(
"OldTransitionLogic", (
258 flags.Tracking.PrimaryPassConfig
is PrimaryPassConfig.MinBias))
259 kwargs.setdefault(
"minTRTPrecisionFraction",
260 flags.Tracking.TRTStandalone.minTRTPrecFrac)
262 kwargs.setdefault(
"TRTTrksEtaBins",
263 [0.7, 0.8, 0.9, 1.2, 1.3, 1.6, 1.7, 1.8, 1.9, 999])
265 kwargs.setdefault(
"TRTTrksMinTRTHitsThresholds",
266 [25, 18, 18, 18, 26, 28, 26, 24, 22, 0]
267 if flags.GeoModel.Run
is LHCPeriod.Run3
else
268 [27, 18, 18, 18, 26, 28, 26, 24, 22, 0])
270 kwargs.setdefault(
"TRTTrksMinTRTHitsMuDependencies",
271 [0.2, 0.05, 0.05, 0.05,
272 0.15, 0.15, 0.15, 0.15, 0.15, 0.])
275 CompFactory.InDet.InDetTrtTrackScoringTool(name, **kwargs))
280 kwargs.setdefault(
"PtMin", flags.Tracking.ActiveConfig.minPT)
291 if 'Extrapolator' not in kwargs:
292 from TrkConfig.AtlasExtrapolatorConfig
import AtlasExtrapolatorCfg
293 kwargs.setdefault(
"Extrapolator", acc.popToolsAndMerge(
296 if 'InDetEtaDependentCutsSvc' not in kwargs:
297 from InDetConfig.InDetEtaDependentCutsConfig
import (
298 ITkEtaDependentCutsSvcCfg)
300 kwargs.setdefault(
"InDetEtaDependentCutsSvc", acc.getService(
301 "ITkEtaDependentCutsSvc"+flags.Tracking.ActiveConfig.extension))
303 if flags.Tracking.ActiveConfig.doCaloSeededBremSi:
304 from InDetConfig.InDetCaloClusterROISelectorConfig
import (
305 ITkCaloClusterROIPhiRZContainerMakerCfg)
307 kwargs.setdefault(
"EMROIPhiRZContainer",
"ITkCaloClusterROIPhiRZ5GeV")
309 kwargs.setdefault(
"DriftCircleCutTool",
None)
310 kwargs.setdefault(
"useAmbigFcn",
True)
311 kwargs.setdefault(
"useTRT_AmbigFcn",
False)
312 kwargs.setdefault(
"maxEta", flags.Tracking.ActiveConfig.maxEta)
313 kwargs.setdefault(
"usePixel", flags.Tracking.ActiveConfig.useITkPixel)
314 kwargs.setdefault(
"useSCT", flags.Tracking.ActiveConfig.useITkStrip)
315 kwargs.setdefault(
"doEmCaloSeed", flags.Tracking.ActiveConfig.doCaloSeededBremSi)
316 kwargs.setdefault(
"useITkAmbigFcn",
True)
317 kwargs.setdefault(
"minTRTonTrk", 0)
318 kwargs.setdefault(
"minTRTPrecisionFraction", 0)
320 acc.setPrivateTools(CompFactory.InDet.InDetAmbiScoringTool(
321 name + flags.Tracking.ActiveConfig.extension, **kwargs))
328 kwargs.setdefault(
"nWeightedClustersMin",
329 flags.Tracking.ActiveConfig.nWeightedClustersMin)
330 kwargs.setdefault(
"minTRTHits", 0)
332 acc.setPrivateTools(CompFactory.InDet.InDetCosmicScoringTool(
333 name+flags.ITk.Tracking.ActiveConfig.extension, **kwargs))