6 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
7 from AthenaConfiguration.ComponentFactory
import CompFactory
14 cfg.setPrivateTools(CompFactory.InDet.InDetAlignModuleTool(name, **kwargs))
21 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
24 kwargs.setdefault(
"AlignmentLevel", flags.InDet.Align.pixelAlignmentLevel)
25 kwargs.setdefault(
"AlignmentLevelBarrel", flags.InDet.Align.pixelAlignmentLevelBarrel)
26 kwargs.setdefault(
"AlignmentLevelEndcaps", flags.InDet.Align.pixelAlignmentLevelEndcaps)
28 kwargs.setdefault(
"AlignBarrelBowX",
True)
29 kwargs.setdefault(
"AlignEndcapRotX",
False)
30 kwargs.setdefault(
"AlignEndcapRotY",
False)
31 kwargs.setdefault(
"AlignEndcapZ",
False)
33 if kwargs[
"AlignmentLevel"] == 16:
34 kwargs.setdefault(
"AlignBarrelRotX",
False)
35 kwargs.setdefault(
"AlignBarrelRotY",
False)
36 kwargs.setdefault(
"AlignBarrelRotZ",
False)
37 kwargs.setdefault(
"AlignBarrelX",
False)
38 kwargs.setdefault(
"AlignBarrelY",
False)
39 kwargs.setdefault(
"AlignBarrelZ",
False)
40 kwargs.setdefault(
"AlignEndcaps",
False)
43 kwargs.setdefault(
"SetSoftCutEndcapX", 0.02)
44 kwargs.setdefault(
"SetSoftCutEndcapY", 0.02)
45 kwargs.setdefault(
"SetSoftCutEndcapZ", 0.02)
46 kwargs.setdefault(
"SetSoftCutEndcapRotX", 0.05)
47 kwargs.setdefault(
"SetSoftCutEndcapRotY", 0.05)
48 kwargs.setdefault(
"SetSoftCutEndcapRotZ", 0.05)
50 kwargs.setdefault(
"SetSoftCutBarrelX", 0.02)
51 kwargs.setdefault(
"SetSoftCutBarrelY", 0.02)
52 kwargs.setdefault(
"SetSoftCutBarrelZ", 0.02)
53 kwargs.setdefault(
"SetSoftCutBarrelRotX", 0.05)
54 kwargs.setdefault(
"SetSoftCutBarrelRotY", 0.05)
55 kwargs.setdefault(
"SetSoftCutBarrelRotZ", 0.05)
58 CompFactory.InDet.PixelGeometryManagerTool(name, **kwargs))
65 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
68 kwargs.setdefault(
"AlignmentLevel", flags.InDet.Align.SCTAlignmentLevel)
69 kwargs.setdefault(
"AlignmentLevelBarrel", flags.InDet.Align.SCTAlignmentLevelBarrel)
70 kwargs.setdefault(
"AlignmentLevelEndcaps", flags.InDet.Align.SCTAlignmentLevelEndcaps)
72 kwargs.setdefault(
"AlignBarrelRotX",
False)
73 kwargs.setdefault(
"AlignBarrelRotY",
False)
74 kwargs.setdefault(
"AlignBarrelRotZ",
False)
75 kwargs.setdefault(
"AlignBarrelX",
False)
76 kwargs.setdefault(
"AlignBarrelY",
False)
77 kwargs.setdefault(
"AlignBarrelZ",
False)
78 kwargs.setdefault(
"AlignEndcapZ",
False)
80 kwargs.setdefault(
"SetSoftCutBarrelX", 0.05)
81 kwargs.setdefault(
"SetSoftCutBarrelY", 0.05)
82 kwargs.setdefault(
"SetSoftCutBarrelZ", 0.05)
83 kwargs.setdefault(
"SetSoftCutBarrelRotX", 0.05)
84 kwargs.setdefault(
"SetSoftCutBarrelRotY", 0.05)
85 kwargs.setdefault(
"SetSoftCutBarrelRotZ", 0.05)
86 kwargs.setdefault(
"SetSoftCutEndcapX", 0.05)
87 kwargs.setdefault(
"SetSoftCutEndcapY", 0.05)
88 kwargs.setdefault(
"SetSoftCutEndcapZ", 0.005)
89 kwargs.setdefault(
"SetSoftCutEndcapRotX", 0.005)
90 kwargs.setdefault(
"SetSoftCutEndcapRotY", 0.05)
91 kwargs.setdefault(
"SetSoftCutEndcapRotZ", 0.05)
94 CompFactory.InDet.SCTGeometryManagerTool(name, **kwargs))
101 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
104 kwargs.setdefault(
"AlignPixel", flags.InDet.Align.alignPixel)
105 kwargs.setdefault(
"AlignSCT", flags.InDet.Align.alignSCT)
106 kwargs.setdefault(
"AlignmentLevel", -1)
107 kwargs.setdefault(
"ModuleSelection", [])
109 kwargs.setdefault(
"PixelGeometryManager", cfg.addPublicTool(
112 kwargs.setdefault(
"SCTGeometryManager", cfg.addPublicTool(
115 cfg.setPrivateTools(CompFactory.InDet.SiGeometryManagerTool(name, **kwargs))
122 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
125 kwargs.setdefault(
"AlignmentLevel", flags.InDet.Align.TRTAlignmentLevel)
126 kwargs.setdefault(
"AlignmentLevelBarrel", flags.InDet.Align.TRTAlignmentLevelBarrel)
127 kwargs.setdefault(
"AlignmentLevelEndcaps", flags.InDet.Align.TRTAlignmentLevelEndcaps)
129 kwargs.setdefault(
"AlignEndcapZ",
False)
131 kwargs.setdefault(
"SetSoftCutBarrelX", 0.1)
132 kwargs.setdefault(
"SetSoftCutBarrelY", 0.1)
133 kwargs.setdefault(
"SetSoftCutBarrelZ", 0.1)
134 kwargs.setdefault(
"SetSoftCutBarrelRotX", 0.05)
135 kwargs.setdefault(
"SetSoftCutBarrelRotY", 0.05)
136 kwargs.setdefault(
"SetSoftCutBarrelRotZ", 0.05)
137 kwargs.setdefault(
"SetSoftCutEndcapX", 0.1)
138 kwargs.setdefault(
"SetSoftCutEndcapY", 0.1)
139 kwargs.setdefault(
"SetSoftCutEndcapZ", 0.001)
140 kwargs.setdefault(
"SetSoftCutEndcapRotX", 0.05)
141 kwargs.setdefault(
"SetSoftCutEndcapRotY", 0.05)
142 kwargs.setdefault(
"SetSoftCutEndcapRotZ", 0.05)
145 CompFactory.InDet.TRTGeometryManagerTool(name, **kwargs))
150 flags, name="InDetGeometryManagerTool", **kwargs):
153 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
156 kwargs.setdefault(
"AlignSilicon", flags.InDet.Align.alignPixel
or flags.InDet.Align.alignSCT)
157 kwargs.setdefault(
"AlignTRT", flags.InDet.Align.alignTRT)
158 kwargs.setdefault(
"AlignmentLevel", -1)
160 kwargs.setdefault(
"SiGeometryManager", cfg.addPublicTool(
163 kwargs.setdefault(
"TRTGeometryManager", cfg.addPublicTool(
167 CompFactory.InDet.InDetGeometryManagerTool(name, **kwargs))
177 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
180 kwargs.setdefault(
"SiGeometryManager", cfg.addPublicTool(
183 kwargs.setdefault(
"PixelGeometryManager", cfg.addPublicTool(
186 kwargs.setdefault(
"SCTGeometryManager", cfg.addPublicTool(
189 from InDetAlignGenTools.InDetAlignGenToolsConfig
import InDetAlignDBTool
190 kwargs.setdefault(
"IDAlignDBTool", cfg.addPublicTool(cfg.popToolsAndMerge(
InDetAlignDBTool(flags))))
192 kwargs.setdefault(
"WriteOldConstants",
not flags.InDet.Align.accumulate)
193 kwargs.setdefault(
"UpdateConstants",
not flags.InDet.Align.accumulate)
195 cfg.setPrivateTools(CompFactory.InDet.SiTrkAlignDBTool(name, **kwargs))
201 kwargs.setdefault(
"TrtAlignDbSvc", cfg.addService(CompFactory.TRT_AlignDbSvc(name)))
202 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
205 kwargs.setdefault(
"TRTGeometryManager", cfg.addPublicTool(
208 kwargs.setdefault(
"WriteOldConstants",
not flags.InDet.Align.accumulate)
209 kwargs.setdefault(
"UpdateConstants",
not flags.InDet.Align.accumulate)
211 cfg.setPrivateTools(CompFactory.InDet.TRTTrkAlignDBTool(name, **kwargs))
218 kwargs.setdefault(
"SiTrkAlignDBTool", cfg.addPublicTool(
220 kwargs.setdefault(
"TRTTrkAlignDBTool", cfg.addPublicTool(
223 cfg.setPrivateTools(CompFactory.InDet.InDetTrkAlignDBTool(name, **kwargs))
228 if flags.InDet.Align.alignInDet:
230 elif flags.InDet.Align.alignSilicon:
232 elif flags.InDet.Align.alignPixel:
234 elif flags.InDet.Align.alignSCT:
236 elif flags.InDet.Align.alignTRT:
239 print(
"Configuration Error - No Geometry Manager Configured!")
244 if flags.InDet.Align.alignInDet:
246 elif (flags.InDet.Align.alignSilicon
or flags.InDet.Align.alignPixel
or flags.InDet.Align.alignSCT):
248 elif flags.InDet.Align.alignTRT:
251 print(
"Configuration Error - No AlignDBTool Configured!")
259 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
262 kwargs.setdefault(
"PathBinName", f
"{flags.InDet.Align.baseDir}/Accumulate/")
263 kwargs.setdefault(
"PathTxtName", f
"{flags.InDet.Align.baseDir}/Accumulate/")
264 kwargs.setdefault(
"InputTFiles", flags.InDet.Align.inputTFiles)
265 kwargs.setdefault(
"SolveOption", 0
if flags.InDet.Align.accumulate
else 1)
266 kwargs.setdefault(
"MinNumHitsPerModule", 10)
267 kwargs.setdefault(
"AlignIBLbutNotPixel", flags.InDet.Align.pixelAlignmentLevel == 16)
268 kwargs.setdefault(
"Remove_IBL_Rz", flags.InDet.Align.pixelAlignmentLevel == 11)
269 kwargs.setdefault(
"RunLocalMethod",
False)
270 kwargs.setdefault(
"ReadTFile",
not flags.InDet.Align.accumulate)
271 kwargs.setdefault(
"ScaleMatrix",
True)
272 kwargs.setdefault(
"WriteEigenMat",
False)
273 kwargs.setdefault(
"WriteEigenMatTxt",
False)
274 kwargs.setdefault(
"WriteMat",
False)
275 kwargs.setdefault(
"WriteTFile",
True)
277 cfg.setPrivateTools(CompFactory.Trk.MatrixTool(name, **kwargs))
284 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
287 kwargs.setdefault(
"MatrixTool", cfg.popToolsAndMerge(
MatrixToolCfg(flags)))
288 kwargs.setdefault(
"SecondDerivativeCut", 0)
290 cfg.setPrivateTools(CompFactory.Trk.GlobalChi2AlignTool(name, **kwargs))
299 if "ResidualPullCalculator" not in kwargs:
300 from TrkConfig.TrkResidualPullCalculatorConfig
import (
301 ResidualPullCalculatorCfg)
302 kwargs.setdefault(
"ResidualPullCalculator", cfg.popToolsAndMerge(
305 if "UpdatorTool" not in kwargs:
306 from TrkConfig.TrkMeasurementUpdatorConfig
import KalmanUpdatorCfg
307 kwargs.setdefault(
"UpdatorTool", cfg.popToolsAndMerge(
310 kwargs.setdefault(
"ResidualType", 0)
312 cfg.setPrivateTools(CompFactory.Trk.AlignResidualCalculator(name, **kwargs))
319 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
322 kwargs.setdefault(
"ResidualCalculator", cfg.popToolsAndMerge(
325 kwargs.setdefault(
"IncludeScatterers",
False)
326 kwargs.setdefault(
"RemoveATSOSNotInAlignModule",
False)
328 cfg.setPrivateTools(CompFactory.Trk.AlignTrackCreator(name, **kwargs))
335 flags, name="BeamspotVertexPreProcessor", **kwargs):
338 kwargs.setdefault(
"AlignModuleTool", cfg.addPublicTool(
341 if "TrackFitter" not in kwargs:
342 from TrkConfig.CommonTrackFitterConfig
import InDetStandaloneTrackFitterCfg
343 kwargs.setdefault(
"TrackFitter", cfg.popToolsAndMerge(
346 if "TrackToVertexIPEstimatorTool" not in kwargs:
347 from TrkConfig.TrkVertexFitterUtilsConfig
import (
348 TrackToVertexIPEstimatorCfg)
349 kwargs.setdefault(
"TrackToVertexIPEstimatorTool", cfg.popToolsAndMerge(
352 if "BSConstraintTrackSelector" not in kwargs:
353 from InDetConfig.InDetTrackSelectionToolConfig
import (
354 Align_InDetTrackSelectionToolCfg)
355 kwargs.setdefault(
"BSConstraintTrackSelector", cfg.popToolsAndMerge(
358 if "TrackSelector" not in kwargs:
359 from InDetConfig.InDetTrackSelectionToolConfig
import (
360 Align_InDetTrackSelectionToolCfg)
361 kwargs.setdefault(
"TrackSelector", cfg.popToolsAndMerge(
364 if "Extrapolator" not in kwargs:
365 from TrkConfig.AtlasExtrapolatorConfig
import InDetExtrapolatorCfg
366 kwargs.setdefault(
"Extrapolator", cfg.popToolsAndMerge(
369 kwargs.setdefault(
"UseSingleFitter",
True)
370 kwargs.setdefault(
"RunOutlierRemoval",
True)
371 kwargs.setdefault(
"DoBSConstraint",
False)
372 kwargs.setdefault(
"DoAssociatedToPVSelection",
False)
373 kwargs.setdefault(
"RunOutlierRemoval",
True)
376 CompFactory.Trk.BeamspotVertexPreProcessor(name, **kwargs))