ATLAS Offline Software
TrkRIO_OnTrackCreatorConfig.py
Go to the documentation of this file.
1 # Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
2 
3 # Configuration of TrkRIO_OnTrackCreator package
4 # RIO (Reconstruction Input Objects) On Track (ROT)
5 # Represent clusters, drift circles (PrepRawData)
6 # with additonal detector specific calibration corrections.
7 
8 from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
9 from AthenaConfiguration.ComponentFactory import CompFactory
10 from AthenaConfiguration.Enums import LHCPeriod
11 
12 
13 def InDetRotCreatorCfg(flags, name='InDetRotCreator', **kwargs):
14  if flags.Detector.GeometryITk:
15  name = name.replace("InDet", "ITk")
16  return ITkRotCreatorCfg(flags, name, **kwargs)
17 
18  acc = ComponentAccumulator()
19 
20  if 'ToolPixelCluster' not in kwargs:
21  if flags.Tracking.useBroadPixClusterErrors:
22  from InDetConfig.SiClusterOnTrackTool_PixelConfig import (
23  InDetBroadPixelClusterOnTrackToolCfg)
24  ToolPixelCluster = acc.popToolsAndMerge(
25  InDetBroadPixelClusterOnTrackToolCfg(flags))
26  else:
27  from InDetConfig.SiClusterOnTrackTool_PixelConfig import (
28  InDetPixelClusterOnTrackToolCfg)
29  ToolPixelCluster = acc.popToolsAndMerge(
30  InDetPixelClusterOnTrackToolCfg(flags))
31  kwargs.setdefault("ToolPixelCluster", ToolPixelCluster)
32 
33  if 'ToolSCT_Cluster' not in kwargs:
34  if flags.Tracking.useBroadSCTClusterErrors:
35  from InDetConfig.SiClusterOnTrackTool_SCTStripConfig import (
36  InDetBroadSCT_ClusterOnTrackToolCfg)
37  ToolSCT_Cluster = acc.popToolsAndMerge(
38  InDetBroadSCT_ClusterOnTrackToolCfg(flags))
39  else:
40  from InDetConfig.SiClusterOnTrackTool_SCTStripConfig import (
41  InDetSCT_ClusterOnTrackToolCfg)
42  ToolSCT_Cluster = acc.popToolsAndMerge(
43  InDetSCT_ClusterOnTrackToolCfg(flags))
44  kwargs.setdefault("ToolSCT_Cluster", ToolSCT_Cluster)
45 
46  if 'ToolTRT_DriftCircle' not in kwargs:
47  from InDetConfig.TRT_DriftCircleOnTrackToolConfig import (
48  TRT_DriftCircleOnTrackToolCfg)
49  kwargs.setdefault("ToolTRT_DriftCircle", acc.popToolsAndMerge(
50  TRT_DriftCircleOnTrackToolCfg(flags)))
51 
52  kwargs.setdefault('Mode', 'indet')
53 
54  acc.setPrivateTools(CompFactory.Trk.RIO_OnTrackCreator(name, **kwargs))
55  return acc
56 
57 
58 def InDetBroadRotCreatorCfg(flags, name='InDetBroadInDetRotCreator', **kwargs):
59  acc = ComponentAccumulator()
60 
61  if 'ToolPixelCluster' not in kwargs:
62  from InDetConfig.SiClusterOnTrackTool_PixelConfig import (
63  InDetBroadPixelClusterOnTrackToolCfg)
64  kwargs.setdefault('ToolPixelCluster', acc.popToolsAndMerge(
65  InDetBroadPixelClusterOnTrackToolCfg(flags)))
66 
67  if 'ToolSCT_Cluster' not in kwargs:
68  from InDetConfig.SiClusterOnTrackTool_SCTStripConfig import (
69  InDetBroadSCT_ClusterOnTrackToolCfg)
70  kwargs.setdefault('ToolSCT_Cluster', acc.popToolsAndMerge(
71  InDetBroadSCT_ClusterOnTrackToolCfg(flags)))
72 
73  if flags.Detector.EnableTRT and 'ToolTRT_DriftCircle' not in kwargs:
74  from InDetConfig.TRT_DriftCircleOnTrackToolConfig import (
75  TRT_DriftCircleOnTrackNoDriftTimeToolCfg)
76  kwargs.setdefault('ToolTRT_DriftCircle', acc.popToolsAndMerge(
77  TRT_DriftCircleOnTrackNoDriftTimeToolCfg(flags)))
78 
79  acc.setPrivateTools(acc.popToolsAndMerge(
80  InDetRotCreatorCfg(flags, name, **kwargs)))
81  return acc
82 
83 
84 def InDetRefitRotCreatorCfg(flags, name='InDetRefitRotCreator', **kwargs):
85  acc = ComponentAccumulator()
86 
87  if flags.Detector.EnableTRT:
88  if 'ToolTRT_DriftCircle' not in kwargs:
89  from InDetConfig.TRT_DriftCircleOnTrackToolConfig import (
90  TRT_DriftCircleOnTrackUniversalToolCfg)
91  kwargs.setdefault("ToolTRT_DriftCircle", acc.popToolsAndMerge(
92  TRT_DriftCircleOnTrackUniversalToolCfg(flags)))
93 
94  acc.setPrivateTools(acc.popToolsAndMerge(
95  InDetRotCreatorCfg(flags, name, **kwargs)))
96  return acc
97 
98 
99 def InDetRotCreatorDigitalCfg(flags, name='InDetRotCreatorDigital', **kwargs):
100  acc = ComponentAccumulator()
101 
102  if 'ToolPixelCluster' not in kwargs:
103  from InDetConfig.SiClusterOnTrackTool_PixelConfig import (
104  InDetPixelClusterOnTrackToolDigitalCfg)
105  kwargs.setdefault('ToolPixelCluster', acc.popToolsAndMerge(
106  InDetPixelClusterOnTrackToolDigitalCfg(flags)))
107 
108  acc.setPrivateTools(acc.popToolsAndMerge(
109  InDetRotCreatorCfg(flags, name, **kwargs)))
110  return acc
111 
112 
113 def TrigRotCreatorCfg(flags, name='InDetTrigRotCreator', **kwargs):
114  acc = ComponentAccumulator()
115 
116  if 'ToolPixelCluster' not in kwargs:
117  from InDetConfig.SiClusterOnTrackTool_PixelConfig import (
118  TrigPixelClusterOnTrackToolBaseCfg)
119  kwargs.setdefault("ToolPixelCluster", acc.popToolsAndMerge(
120  TrigPixelClusterOnTrackToolBaseCfg(flags)))
121 
122  if 'ToolSCT_Cluster' not in kwargs:
123  from InDetConfig.SiClusterOnTrackTool_SCTStripConfig import (
124  InDetSCT_ClusterOnTrackToolCfg)
125  kwargs.setdefault("ToolSCT_Cluster", acc.popToolsAndMerge(
126  InDetSCT_ClusterOnTrackToolCfg(flags)))
127 
128  kwargs.setdefault('Mode', 'indet')
129 
130  acc.setPrivateTools(CompFactory.Trk.RIO_OnTrackCreator(name, **kwargs))
131  return acc
132 
133 
134 def ITkRotCreatorCfg(flags, name='ITkRotCreator', **kwargs):
135  acc = ComponentAccumulator()
136 
137  if 'ToolPixelCluster' not in kwargs:
138  from InDetConfig.SiClusterOnTrackTool_PixelConfig import (
139  ITkPixelClusterOnTrackToolCfg)
140  kwargs.setdefault("ToolPixelCluster", acc.popToolsAndMerge(
141  ITkPixelClusterOnTrackToolCfg(flags)))
142 
143  if 'ToolSCT_Cluster' not in kwargs:
144  from InDetConfig.SiClusterOnTrackTool_SCTStripConfig import (
145  ITkStripClusterOnTrackToolCfg)
146  kwargs.setdefault("ToolSCT_Cluster", acc.popToolsAndMerge(
147  ITkStripClusterOnTrackToolCfg(flags)))
148 
149  kwargs.setdefault("ToolTRT_DriftCircle", None)
150  kwargs.setdefault('Mode', 'indet')
151 
152  acc.setPrivateTools(CompFactory.Trk.RIO_OnTrackCreator(name, **kwargs))
153  return acc
154 
155 
156 def ITkBroadRotCreatorCfg(flags, name='ITkBroadRotCreator', **kwargs):
157  acc = ComponentAccumulator()
158 
159  if 'ToolPixelCluster' not in kwargs:
160  from InDetConfig.SiClusterOnTrackTool_PixelConfig import (
161  ITkBroadPixelClusterOnTrackToolCfg)
162  kwargs.setdefault('ToolPixelCluster', acc.popToolsAndMerge(
163  ITkBroadPixelClusterOnTrackToolCfg(flags)))
164 
165  if 'ToolSCT_Cluster' not in kwargs:
166  from InDetConfig.SiClusterOnTrackTool_SCTStripConfig import (
167  ITkBroadStripClusterOnTrackToolCfg)
168  kwargs.setdefault('ToolSCT_Cluster', acc.popToolsAndMerge(
169  ITkBroadStripClusterOnTrackToolCfg(flags)))
170 
171  acc.setPrivateTools(acc.popToolsAndMerge(
172  ITkRotCreatorCfg(flags, name, **kwargs)))
173  return acc
174 
175 
176 def MuonRotCreatorCfg(flags, name="MuonRotCreator", **kwargs):
177  result = ComponentAccumulator()
178 
179  from MuonConfig.MuonRIO_OnTrackCreatorToolConfig import (
180  MdtDriftCircleOnTrackCreatorCfg, MuonClusterOnTrackCreatorCfg)
181  mdt_rot_creator = result.popToolsAndMerge(
182  MdtDriftCircleOnTrackCreatorCfg(flags))
183  cluster_rot_creator = result.popToolsAndMerge(
184  MuonClusterOnTrackCreatorCfg(flags))
185 
186  kwargs.setdefault("ToolMuonDriftCircle", mdt_rot_creator)
187  kwargs.setdefault("ToolMuonCluster", cluster_rot_creator)
188  kwargs.setdefault("ToolPixelCluster", None)
189  kwargs.setdefault("ToolSCT_Cluster", None)
190  kwargs.setdefault("ToolTRT_DriftCircle", None)
191  kwargs.setdefault("Mode", 'muon')
192 
193  result.setPrivateTools(CompFactory.Trk.RIO_OnTrackCreator(name, **kwargs))
194  return result
195 
196 
197 def CombinedRotCreatorCfg(flags, name="TrkRotCreator", **kwargs):
198  result = ComponentAccumulator()
199 
200  if flags.Detector.GeometryITk:
201  from InDetConfig.SiClusterOnTrackTool_PixelConfig import ITkPixelClusterOnTrackToolCfg
202  kwargs.setdefault("ToolPixelCluster", result.popToolsAndMerge(ITkPixelClusterOnTrackToolCfg(flags)))
203  from InDetConfig.SiClusterOnTrackTool_SCTStripConfig import ITkStripClusterOnTrackToolCfg
204  kwargs.setdefault("ToolSCT_Cluster", result.popToolsAndMerge(ITkStripClusterOnTrackToolCfg(flags)))
205  elif flags.Detector.GeometryID:
206  if flags.Tracking.useBroadPixClusterErrors:
207  from InDetConfig.SiClusterOnTrackTool_PixelConfig import InDetBroadPixelClusterOnTrackToolCfg
208  kwargs.setdefault("ToolPixelCluster",
209  result.popToolsAndMerge(InDetBroadPixelClusterOnTrackToolCfg(flags)))
210  else:
211  from InDetConfig.SiClusterOnTrackTool_PixelConfig import InDetPixelClusterOnTrackToolCfg
212  kwargs.setdefault("ToolPixelCluster",
213  result.popToolsAndMerge(InDetPixelClusterOnTrackToolCfg(flags)))
214  if flags.Tracking.useBroadSCTClusterErrors:
215  from InDetConfig.SiClusterOnTrackTool_SCTStripConfig import InDetBroadSCT_ClusterOnTrackToolCfg
216  kwargs.setdefault("ToolSCT_Cluster",
217  result.popToolsAndMerge(InDetBroadSCT_ClusterOnTrackToolCfg(flags)))
218  else:
219  from InDetConfig.SiClusterOnTrackTool_SCTStripConfig import InDetSCT_ClusterOnTrackToolCfg
220  kwargs.setdefault("ToolSCT_Cluster", result.popToolsAnsMerge(InDetSCT_ClusterOnTrackToolCfg(flags)))
221  else:
222  result.setPrivateTools(result.popToolsAndMerge(MuonRotCreatorCfg(flags, name=name, **kwargs)))
223  return result
224 
225  if flags.Detector.GeometryTRT:
226  from InDetConfig.TRT_DriftCircleOnTrackToolConfig import TRT_DriftCircleOnTrackToolCfg
227  kwargs.setdefault("ToolTRT_DriftCircle", result.popToolsAndMerge(TRT_DriftCircleOnTrackToolCfg(flags)))
228 
229  if flags.Detector.GeometryMuon:
230  from MuonConfig.MuonRIO_OnTrackCreatorToolConfig import MdtDriftCircleOnTrackCreatorCfg
231  from MuonConfig.MuonRIO_OnTrackCreatorToolConfig import MuonClusterOnTrackCreatorCfg
232  kwargs.setdefault("ToolMuonDriftCircle", result.popToolsAndMerge(MdtDriftCircleOnTrackCreatorCfg(flags)))
233  kwargs.setdefault("ToolMuonCluster", result.popToolsAndMerge(MuonClusterOnTrackCreatorCfg(flags)))
234  else:
235  result.setPrivateTools(result.popToolsAndMerge(InDetRotCreatorCfg(flags,name=name, **kwargs)))
236  return result
237  result.setPrivateTools(CompFactory.Trk.RIO_OnTrackCreator(name, **kwargs))
238  return result
239 
240 def RIO_OnTrackErrorScalingCondAlgCfg(flags,
241  name='RIO_OnTrackErrorScalingCondAlg',
242  **kwargs):
243  acc = ComponentAccumulator()
244 
245  if flags.GeoModel.Run is LHCPeriod.Run1:
246  error_scaling_type = ["PixelRIO_OnTrackErrorScalingRun1"]
247  error_scaling_outkey = ["/Indet/TrkErrorScalingPixel"]
248  else: # Run 2 and 3
249  error_scaling_type = ["PixelRIO_OnTrackErrorScaling"]
250  error_scaling_outkey = ["/Indet/TrkErrorScalingPixel"]
251 
252  error_scaling_type += ["SCTRIO_OnTrackErrorScaling"]
253  error_scaling_outkey += ["/Indet/TrkErrorScalingSCT"]
254 
255  # TODO: cover Run 4 and beyond
256 
257  if flags.GeoModel.Run <= LHCPeriod.Run3:
258  error_scaling_type += ["TRTRIO_OnTrackErrorScaling"]
259  error_scaling_outkey += ["/Indet/TrkErrorScalingTRT"]
260 
261  from IOVDbSvc.IOVDbSvcConfig import addFoldersSplitOnline
262  acc.merge(addFoldersSplitOnline(flags,
263  'INDET',
264  '/Indet/Onl/TrkErrorScaling',
265  '/Indet/TrkErrorScaling',
266  className="CondAttrListCollection"))
267 
268  kwargs.setdefault("ReadKey", "/Indet/TrkErrorScaling")
269  kwargs.setdefault("ErrorScalingType", error_scaling_type)
270  kwargs.setdefault("OutKeys", error_scaling_outkey)
271 
272  acc.addCondAlgo(CompFactory.RIO_OnTrackErrorScalingCondAlg(name, **kwargs))
273  return acc
python.TRT_DriftCircleOnTrackToolConfig.TRT_DriftCircleOnTrackToolCfg
def TRT_DriftCircleOnTrackToolCfg(flags, name='TRT_DriftCircleOnTrackTool', **kwargs)
Definition: TRT_DriftCircleOnTrackToolConfig.py:12
python.TrkRIO_OnTrackCreatorConfig.ITkRotCreatorCfg
def ITkRotCreatorCfg(flags, name='ITkRotCreator', **kwargs)
Definition: TrkRIO_OnTrackCreatorConfig.py:134
python.TrkRIO_OnTrackCreatorConfig.InDetBroadRotCreatorCfg
def InDetBroadRotCreatorCfg(flags, name='InDetBroadInDetRotCreator', **kwargs)
Definition: TrkRIO_OnTrackCreatorConfig.py:58
python.TrkRIO_OnTrackCreatorConfig.TrigRotCreatorCfg
def TrigRotCreatorCfg(flags, name='InDetTrigRotCreator', **kwargs)
Definition: TrkRIO_OnTrackCreatorConfig.py:113
python.SiClusterOnTrackTool_SCTStripConfig.InDetBroadSCT_ClusterOnTrackToolCfg
def InDetBroadSCT_ClusterOnTrackToolCfg(flags, name='InDetBroadSCT_ClusterOnTrackTool', **kwargs)
Definition: SiClusterOnTrackTool_SCTStripConfig.py:21
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
python.TrkRIO_OnTrackCreatorConfig.InDetRotCreatorCfg
def InDetRotCreatorCfg(flags, name='InDetRotCreator', **kwargs)
Definition: TrkRIO_OnTrackCreatorConfig.py:13
python.SiClusterOnTrackTool_PixelConfig.ITkBroadPixelClusterOnTrackToolCfg
def ITkBroadPixelClusterOnTrackToolCfg(flags, name='ITkBroadPixelClusterOnTrackTool', **kwargs)
Definition: SiClusterOnTrackTool_PixelConfig.py:183
python.TrkRIO_OnTrackCreatorConfig.CombinedRotCreatorCfg
def CombinedRotCreatorCfg(flags, name="TrkRotCreator", **kwargs)
ROT creator for combined ID & Muon fit useage.
Definition: TrkRIO_OnTrackCreatorConfig.py:197
python.TRT_DriftCircleOnTrackToolConfig.TRT_DriftCircleOnTrackUniversalToolCfg
def TRT_DriftCircleOnTrackUniversalToolCfg(flags, name='InDetTRT_RefitRotCreator', **kwargs)
Definition: TRT_DriftCircleOnTrackToolConfig.py:29
python.SiClusterOnTrackTool_SCTStripConfig.ITkStripClusterOnTrackToolCfg
def ITkStripClusterOnTrackToolCfg(flags, name='ITkStrip_ClusterOnTrackTool', **kwargs)
Definition: SiClusterOnTrackTool_SCTStripConfig.py:25
python.TrkRIO_OnTrackCreatorConfig.InDetRefitRotCreatorCfg
def InDetRefitRotCreatorCfg(flags, name='InDetRefitRotCreator', **kwargs)
Definition: TrkRIO_OnTrackCreatorConfig.py:84
python.TrkRIO_OnTrackCreatorConfig.InDetRotCreatorDigitalCfg
def InDetRotCreatorDigitalCfg(flags, name='InDetRotCreatorDigital', **kwargs)
Definition: TrkRIO_OnTrackCreatorConfig.py:99
python.SiClusterOnTrackTool_PixelConfig.InDetPixelClusterOnTrackToolCfg
def InDetPixelClusterOnTrackToolCfg(flags, name="InDetPixelClusterOnTrackTool", **kwargs)
Definition: SiClusterOnTrackTool_PixelConfig.py:47
python.SiClusterOnTrackTool_PixelConfig.InDetPixelClusterOnTrackToolDigitalCfg
def InDetPixelClusterOnTrackToolDigitalCfg(flags, name="InDetPixelClusterOnTrackToolDigital", **kwargs)
Definition: SiClusterOnTrackTool_PixelConfig.py:41
python.TRT_DriftCircleOnTrackToolConfig.TRT_DriftCircleOnTrackNoDriftTimeToolCfg
def TRT_DriftCircleOnTrackNoDriftTimeToolCfg(flags, name='TRT_DriftCircleOnTrackNoDriftTimeTool', **kwargs)
Definition: TRT_DriftCircleOnTrackToolConfig.py:7
python.IOVDbSvcConfig.addFoldersSplitOnline
def addFoldersSplitOnline(flags, detDb, onlineFolders, offlineFolders, className=None, extensible=False, addMCString='_OFL', splitMC=False, tag=None, forceDb=None, modifiers='')
Definition: IOVDbSvcConfig.py:167
MuonRIO_OnTrackCreatorToolConfig.MuonClusterOnTrackCreatorCfg
def MuonClusterOnTrackCreatorCfg(flags, name="MuonClusterOnTrackCreator", **kwargs)
Definition: MuonRIO_OnTrackCreatorToolConfig.py:96
python.TrkRIO_OnTrackCreatorConfig.MuonRotCreatorCfg
def MuonRotCreatorCfg(flags, name="MuonRotCreator", **kwargs)
Definition: TrkRIO_OnTrackCreatorConfig.py:176
MuonRIO_OnTrackCreatorToolConfig.MdtDriftCircleOnTrackCreatorCfg
def MdtDriftCircleOnTrackCreatorCfg(flags, name="MdtDriftCircleOnTrackCreator", **kwargs)
Definition: MuonRIO_OnTrackCreatorToolConfig.py:73
python.SiClusterOnTrackTool_SCTStripConfig.InDetSCT_ClusterOnTrackToolCfg
def InDetSCT_ClusterOnTrackToolCfg(flags, name='InDetSCT_ClusterOnTrackTool', **kwargs)
Definition: SiClusterOnTrackTool_SCTStripConfig.py:6
python.SiClusterOnTrackTool_PixelConfig.InDetBroadPixelClusterOnTrackToolCfg
def InDetBroadPixelClusterOnTrackToolCfg(flags, name='InDetBroadPixelClusterOnTrackTool', **kwargs)
Definition: SiClusterOnTrackTool_PixelConfig.py:82
python.SiClusterOnTrackTool_SCTStripConfig.ITkBroadStripClusterOnTrackToolCfg
def ITkBroadStripClusterOnTrackToolCfg(flags, name='ITkBroadStripClusterOnTrackTool', **kwargs)
Definition: SiClusterOnTrackTool_SCTStripConfig.py:34
python.TrkRIO_OnTrackCreatorConfig.RIO_OnTrackErrorScalingCondAlgCfg
def RIO_OnTrackErrorScalingCondAlgCfg(flags, name='RIO_OnTrackErrorScalingCondAlg', **kwargs)
Definition: TrkRIO_OnTrackCreatorConfig.py:240
python.SiClusterOnTrackTool_PixelConfig.ITkPixelClusterOnTrackToolCfg
def ITkPixelClusterOnTrackToolCfg(flags, name='ITkPixelClusterOnTrackTool', **kwargs)
Definition: SiClusterOnTrackTool_PixelConfig.py:154
python.SiClusterOnTrackTool_PixelConfig.TrigPixelClusterOnTrackToolBaseCfg
def TrigPixelClusterOnTrackToolBaseCfg(flags, name="InDetTrigPixelClusterOnTrackTool", **kwargs)
InDet PixelClusterOnTrackTool trigger ###.
Definition: SiClusterOnTrackTool_PixelConfig.py:92
python.TrkRIO_OnTrackCreatorConfig.ITkBroadRotCreatorCfg
def ITkBroadRotCreatorCfg(flags, name='ITkBroadRotCreator', **kwargs)
Definition: TrkRIO_OnTrackCreatorConfig.py:156
Generated on Mon Sep 1 2025 21:22:40 for ATLAS Offline Software by   doxygen 1.8.18