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 if flags.Tracking.ActiveConfig.name ==
'minBiasPixel':
128 kwargs.setdefault(
"minSiClusters", flags.Tracking.ActiveConfig.nClustersMin)
129 kwargs.setdefault(
"minPt", flags.Tracking.ActiveConfig.minPT)
130 kwargs.setdefault(
"useAmbigFcn",
True)
131 kwargs.setdefault(
"useTRT_AmbigFcn",
False)
132 kwargs.setdefault(
"maxZImp", flags.Tracking.ActiveConfig.maxZImpact)
133 kwargs.setdefault(
"maxRPhiImp", flags.Tracking.ActiveConfig.maxPrimaryImpact)
134 kwargs.setdefault(
"maxEta", flags.Tracking.ActiveConfig.maxEta)
135 kwargs.setdefault(
"maxSCTHoles", flags.Tracking.ActiveConfig.maxSCTHoles)
136 kwargs.setdefault(
"maxSiHoles", flags.Tracking.ActiveConfig.maxSiHoles)
137 kwargs.setdefault(
"usePixel", flags.Tracking.ActiveConfig.usePixel)
138 kwargs.setdefault(
"useSCT", flags.Tracking.ActiveConfig.useSCT)
139 kwargs.setdefault(
"doEmCaloSeed",
False)
140 kwargs.setdefault(
"EMROIPhiRZContainer",
"")
141 kwargs.setdefault(
"minTRTonTrk", 0)
142 kwargs.setdefault(
"minTRTPrecisionFraction", 0)
145 acc.setPrivateTools(ambiScoringFactory(
146 name=name+flags.Tracking.ActiveConfig.input_name,
155 kwargs.setdefault(
"nWeightedClustersMin",
156 flags.Tracking.ActiveConfig.nWeightedClustersMin)
157 kwargs.setdefault(
"minTRTHits", 0)
159 acc.setPrivateTools(CompFactory.InDet.InDetCosmicScoringTool(
160 name+flags.Tracking.ActiveConfig.extension, **kwargs))
165 flags, name='InDetCosmicExtenScoringTool', **kwargs):
166 kwargs.setdefault(
"nWeightedClustersMin", 0)
167 kwargs.setdefault(
"minTRTHits", flags.Tracking.ActiveConfig.minTRTonTrk)
172 flags, name='InDetCosmicScoringTool_TRT', **kwargs):
173 kwargs.setdefault(
"minTRTHits", flags.Tracking.BackTracking.minTRT)
180 if flags.Tracking.doCaloSeededBrem:
181 from InDetConfig.InDetCaloClusterROISelectorConfig
import (
182 CaloClusterROIPhiRZContainerMakerCfg)
184 kwargs.setdefault(
"EMROIPhiRZContainer",
185 "InDetCaloClusterROIPhiRZ5GeV")
187 if "Extrapolator" not in kwargs:
188 from TrkConfig.AtlasExtrapolatorConfig
import InDetExtrapolatorCfg
189 kwargs.setdefault(
"Extrapolator", acc.popToolsAndMerge(
192 if 'DriftCircleCutTool' not in kwargs:
193 from InDetConfig.InDetTrackSelectorToolConfig
import (
194 InDetTRTDriftCircleCutToolCfg)
195 kwargs.setdefault(
"DriftCircleCutTool", acc.popToolsAndMerge(
198 kwargs.setdefault(
"nnCutConfig",
199 "dev/TrackingCP/LRTAmbiNetwork/20200727_225401/nn-config.json")
200 kwargs.setdefault(
"nnCutThreshold",
201 flags.Tracking.nnCutLargeD0Threshold)
203 kwargs.setdefault(
"useAmbigFcn",
True)
204 kwargs.setdefault(
"useTRT_AmbigFcn",
False)
205 kwargs.setdefault(
"maxZImp", flags.Tracking.ActiveConfig.maxZImpact)
206 kwargs.setdefault(
"maxEta", flags.Tracking.ActiveConfig.maxEta)
207 kwargs.setdefault(
"usePixel", flags.Tracking.ActiveConfig.usePixel)
208 kwargs.setdefault(
"useSCT", flags.Tracking.ActiveConfig.useSCT)
209 kwargs.setdefault(
"doEmCaloSeed", flags.Tracking.doCaloSeededBrem)
211 acc.setPrivateTools(CompFactory.InDet.InDetNNScoringTool(name, **kwargs))
216 kwargs.setdefault(
"minTRTonTrk", 0)
217 kwargs.setdefault(
"minTRTPrecisionFraction", 0)
218 kwargs.setdefault(
"minPt", flags.Tracking.ActiveConfig.minPT)
219 kwargs.setdefault(
"maxRPhiImp",
220 flags.Tracking.ActiveConfig.maxPrimaryImpact)
221 kwargs.setdefault(
"maxRPhiImpEM", flags.Tracking.ActiveConfig.maxEMImpact)
222 kwargs.setdefault(
"minSiClusters", flags.Tracking.ActiveConfig.minClusters)
223 kwargs.setdefault(
"minPixel", flags.Tracking.ActiveConfig.minPixel)
224 kwargs.setdefault(
"maxSiHoles", flags.Tracking.ActiveConfig.maxHoles)
225 kwargs.setdefault(
"maxPixelHoles",
226 flags.Tracking.ActiveConfig.maxPixelHoles)
227 kwargs.setdefault(
"maxSCTHoles", flags.Tracking.ActiveConfig.maxSctHoles)
228 kwargs.setdefault(
"maxDoubleHoles",
229 flags.Tracking.ActiveConfig.maxDoubleHoles)
232 flags, name+flags.Tracking.ActiveConfig.extension, **kwargs)
236 kwargs.setdefault(
'DriftCircleCutTool',
None)
246 if "DriftCircleCutTool" not in kwargs:
247 from InDetConfig.InDetTrackSelectorToolConfig
import (
248 InDetTRTDriftCircleCutToolCfg)
249 kwargs.setdefault(
"DriftCircleCutTool", acc.popToolsAndMerge(
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))