ATLAS Offline Software
Loading...
Searching...
No Matches
MuonD3PDObject.py
Go to the documentation of this file.
1# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
2
3
4from D3PDMakerCoreComps.D3PDObject import make_SGDataVector_D3PDObject
5from D3PDMakerCoreComps.SimpleAssociation import SimpleAssociation
6from D3PDMakerConfig.D3PDMakerFlags import D3PDMakerFlags
7from AthenaConfiguration.ComponentFactory import CompFactory
8
9D3PD = CompFactory.D3PD
10
11
12MuonD3PDObject = \
13 make_SGDataVector_D3PDObject ('DataVector<xAOD::Muon_v1>',
14 D3PDMakerFlags.MuonSGKey,
15 'mu_', 'MuonD3PDObject')
16
17MuonD3PDObject.defineBlock (0, 'Kinematics',
18 D3PD.FourMomFillerTool,
19 WriteRect = True,
20 WriteE = True,
21 WriteM = True)
22MuonD3PDObject.defineBlock (
23 0, 'AllAuthor',
24 D3PD.AuxDataFillerTool,
25 Vars = ['allauthor = allAuthors'])
26
27MuonD3PDObject.defineBlock (
28 1, 'Authors',
29 D3PD.AuxDataFillerTool,
30 Vars = ['author'])
31MuonD3PDObject.defineBlock (1, 'NumberOfSegments',
32 D3PD.MuonNumberOfSegmentsFillerTool)
33
34MuonD3PDObject.defineBlock (
35 1, 'Isolation',
36 D3PD.AuxDataFillerTool,
37 Vars = ['etcone20 < float: 0',
38 'etcone30 < float: 0',
39 'etcone40 < float: 0',
40 'ptcone20 < float: 0',
41 'ptcone30 < float: 0',
42 'ptcone40 < float: 0',
43 'scatteringCurvatureSignificance < float: 0',
44 'scatteringNeighbourSignificance < float: 0',
45 'momentumBalanceSignificance < float: 0'])
46
47MuonD3PDObject.defineBlock (
48 1, 'CaloEnergyLoss',
49 D3PD.AuxDataFillerTool,
50 Vars = ['energyLossType'])
51MuonD3PDObject.defineBlock (
52 1, 'MuonType',
53 D3PD.AuxDataFillerTool,
54 Vars = ['muonType'])
55
56MuonD3PDObject.defineBlock (2, 'MuonHitSummary',
57 D3PD.MuonTrkHitFillerTool)
58
59MuonD3PDObject.defineBlock (
60 2, 'MuonSpectrometerHitSummary',
61 D3PD.AuxDataFillerTool,
62 Vars = ['primarySector',
63 'secondarySector',
64 'innerSmallHits',
65 'innerLargeHits',
66 'middleSmallHits',
67 'middleLargeHits',
68 'outerSmallHits',
69 'outerLargeHits',
70 'extendedSmallHits',
71 'extendedLargeHits',
72 'innerSmallHoles',
73 'innerLargeHoles',
74 'middleSmallHoles',
75 'middleLargeHoles',
76 'outerSmallHoles',
77 'outerLargeHoles',
78 'extendedSmallHoles',
79 'extendedLargeHoles',
80 'phiLayer1Hits',
81 'phiLayer2Hits',
82 'phiLayer3Hits',
83 'phiLayer4Hits',
84 'etaLayer1Hits',
85 'etaLayer2Hits',
86 'etaLayer3Hits',
87 'etaLayer4Hits',
88 'phiLayer1Holes',
89 'phiLayer2Holes',
90 'phiLayer3Holes',
91 'phiLayer4Holes',
92 'etaLayer1Holes',
93 'etaLayer2Holes',
94 'etaLayer3Holes',
95 'etaLayer4Holes',
96 ])
97
98MuonD3PDObject.defineBlock (
99 4, 'MuonSpectrometerFieldIntegral',
100 D3PD.AuxDataFillerTool,
101 Vars = ['MSFieldIntegral = spectrometerFieldIntegral < float: 0'])
102
103
104
107
108
109MuonTPAssoc = SimpleAssociation \
110 (MuonD3PDObject,
111 D3PD.MuonTrackParticleAssociationTool,
112 matched = 'hastrack',
113 blockname = 'TrkInfo')
114MuonTPAssoc.defineBlock (
115 0, 'Charge',
116 # TrackD3PDMaker
117 D3PD.TrackParticleChargeFillerTool)
118TrackParticlePerigeeAssoc = SimpleAssociation \
119 (MuonTPAssoc,
120 # TrackD3PDMaker
121 D3PD.TrackParticlePerigeeAtOOAssociationTool,
122 prefix = 'track',
123 blockname = 'TrackParticlePerigeeAssoc')
124TrackParticlePerigeeAssoc.defineBlock (1, 'TrkParameters',
125 # TrackD3PDMaker
126 D3PD.PerigeeFillerTool)
127
128TrackParticleCovarAssoc = SimpleAssociation (TrackParticlePerigeeAssoc,
129 # TrackD3PDMaker
130 D3PD.PerigeeCovarianceAssociationTool,
131 blockname = 'TrackParticleCovarAssoc')
132TrackParticleCovarAssoc.defineBlock (1, 'TrkCovDiag',
133 # TrackD3PDMaker
134 D3PD.CovarianceFillerTool,
135 IsTrackPerigee = True,
136 Error = False,
137 DiagCovariance = True)
138TrackParticleCovarAssoc.defineBlock (2, 'TrkCovOffDiag',
139 # TrackD3PDMaker
140 D3PD.CovarianceFillerTool,
141 IsTrackPerigee = True,
142 Error = False,
143 OffDiagCovariance = True)
144
145MuonTPAssoc.defineBlock (
146 2, 'TrkFitQuality',
147 D3PD.AuxDataFillerTool,
148 Vars = ['chi2 = chiSquared',
149 'ndof = numberDoF'],
150 prefix = 'trackfit')
151
152
153MuonIDTPAssoc = SimpleAssociation \
154 (MuonD3PDObject,
155 D3PD.MuonTrackParticleAssociationTool,
156 blockname = 'TrkInfoInDet',
157 Type = 'InDet')
158
159if not D3PDMakerFlags.Muons.doSingleMuons:
160 #
161 # Impact parameter variables. Use the ID track for this.
162 # These are likely not very useful for standalone muons anyway.
163 #
164 from TrackD3PDMaker.TrackParticleImpactParameters \
165 import TrackParticleImpactParameters
166 TrackParticleImpactParameters (MuonIDTPAssoc)
167
168
169MuonPVPerigeeAssoc = SimpleAssociation (
170 MuonTPAssoc,
171 # TrackD3PDMaker
172 D3PD.TrackParticlePerigeeAtPVAssociationTool,
173 suffix = '_exPV',
174 blockname = 'MuonPVPerigeeAssoc')
175def _trackToVertexHook (c, flags, acc, *args, **kw):
176 from TrackToVertex.TrackToVertexConfig import InDetTrackToVertexCfg
177 c.Associator.TrackToVertexTool = acc.popToolsAndMerge (InDetTrackToVertexCfg (flags))
178 return
179MuonPVPerigeeAssoc.defineHook (_trackToVertexHook)
180MuonPVPerigeeAssoc.defineBlock (
181 1, 'ParametersAtPV',
182 # TrackD3PDMaker
183 D3PD.PerigeeFillerTool)
184MuonPVCovarAssoc = SimpleAssociation (
185 MuonPVPerigeeAssoc,
186 # TrackD3PDMaker
187 D3PD.PerigeeCovarianceAssociationTool,
188 blockname = 'MuonPVCovarAssoc')
189MuonPVCovarAssoc.defineBlock (
190 1, 'PrimaryTrackDiagonalCovarianceAtPV',
191 # TrackD3PDMaker
192 D3PD.CovarianceFillerTool,
193 IsTrackPerigee = True,
194 Error = True)
195MuonPVCovarAssoc.defineBlock (
196 1, 'PrimaryTrackOffDiagonalCovarianceAtPV',
197 # TrackD3PDMaker
198 D3PD.CovarianceFillerTool,
199 IsTrackPerigee = True,
200 Error = False,
201 OffDiagCovariance = True)
202
203MuonCBTPAssoc = SimpleAssociation \
204 (MuonD3PDObject,
205 D3PD.MuonTrackParticleAssociationTool,
206 Type = 'Combined',
207 prefix = 'cb_',
208 blockname = 'CBTrkInfo')
209MuonCBPVPerigeeAssoc = SimpleAssociation (
210 MuonCBTPAssoc,
211 # TrackD3PDMaker
212 D3PD.TrackParticlePerigeeAtPVAssociationTool,
213 suffix = '_exPV',
214 blockname = 'MuonCBPVPerigeeAssoc')
215MuonCBPVPerigeeAssoc.defineHook (_trackToVertexHook)
216MuonCBPVPerigeeAssoc.defineBlock (
217 1, 'CBParametersAtPV',
218 # TrackD3PDMaker
219 D3PD.PerigeeFillerTool)
220
221MuonIDTPAssoc = SimpleAssociation \
222 (MuonD3PDObject,
223 D3PD.MuonTrackParticleAssociationTool,
224 Type = 'InDet',
225 prefix = 'id_',
226 blockname = 'IDTrkInfo')
227MuonIDPerigeeAssoc = SimpleAssociation (
228 MuonIDTPAssoc,
229 # TrackD3PDMaker
230 D3PD.TrackParticlePerigeeAtOOAssociationTool,
231 blockname = 'MuonIDPerigeeAssoc')
232MuonIDPerigeeAssoc.defineBlock (
233 1, 'IDParameters',
234 # TrackD3PDMaker
235 D3PD.PerigeeFillerTool)
236MuonIDPVPerigeeAssoc = SimpleAssociation (
237 MuonIDTPAssoc,
238 # TrackD3PDMaker
239 D3PD.TrackParticlePerigeeAtPVAssociationTool,
240 suffix = '_exPV',
241 blockname = 'MuonIDPVPerigeeAssoc')
242MuonIDPVPerigeeAssoc.defineHook (_trackToVertexHook)
243MuonIDPVPerigeeAssoc.defineBlock (
244 1, 'IDParametersAtPV',
245 # TrackD3PDMaker
246 D3PD.PerigeeFillerTool)
247
248MuonIDPVCovarAssoc = SimpleAssociation (
249 MuonIDPVPerigeeAssoc,
250 # TrackD3PDMaker
251 D3PD.PerigeeCovarianceAssociationTool,
252 blockname = 'MuonIDPVCovarAssoc')
253MuonIDPVCovarAssoc.defineBlock (
254 2, 'IDTrackDiagonalCovarianceAtPV',
255 # TrackD3PDMaker
256 D3PD.CovarianceFillerTool,
257 IsTrackPerigee = True,
258 Error = True)
259MuonIDPVCovarAssoc.defineBlock (
260 3, 'IDTrackOffDiagonalCovarianceAtPV',
261 # TrackD3PDMaker
262 D3PD.CovarianceFillerTool,
263 IsTrackPerigee = True,
264 Error = False,
265 OffDiagCovariance = True)
266
267
268MuonMETPAssoc = SimpleAssociation \
269 (MuonD3PDObject,
270 D3PD.MuonTrackParticleAssociationTool,
271 Type = 'MuonSpectrometer',
272 prefix = 'me_',
273 blockname = 'METrkInfo')
274MuonMEPerigeeAssoc = SimpleAssociation (
275 MuonMETPAssoc,
276 # TrackD3PDMaker
277 D3PD.TrackParticlePerigeeAtOOAssociationTool)
278MuonMEPerigeeAssoc.defineBlock (
279 1, 'MEParametersAt',
280 # TrackD3PDMaker
281 D3PD.PerigeeFillerTool)
282MuonMEPVPerigeeAssoc = SimpleAssociation (
283 MuonMETPAssoc,
284 # TrackD3PDMaker
285 D3PD.TrackParticlePerigeeAtPVAssociationTool,
286 suffix = '_exPV',
287 blockname = 'MuonMEPVPerigeeAssoc')
288MuonMEPVPerigeeAssoc.defineHook (_trackToVertexHook)
289MuonMEPVPerigeeAssoc.defineBlock (
290 1, 'MEParametersAtPV',
291 # TrackD3PDMaker
292 D3PD.PerigeeFillerTool)
293
294MuonMEPVCovarAssoc = SimpleAssociation (
295 MuonMEPVPerigeeAssoc,
296 # TrackD3PDMaker
297 D3PD.PerigeeCovarianceAssociationTool,
298 blockname = 'MuonMEPVCovarAssoc')
299MuonMEPVCovarAssoc.defineBlock (
300 2, 'METrackDiagonalCovarianceAtPV',
301 # TrackD3PDMaker
302 D3PD.CovarianceFillerTool,
303 IsTrackPerigee = True,
304 Error = True)
305MuonMEPVCovarAssoc.defineBlock (
306 3, 'METrackOffDiagonalCovarianceAtPV',
307 # TrackD3PDMaker
308 D3PD.CovarianceFillerTool,
309 IsTrackPerigee = True,
310 Error = False,
311 OffDiagCovariance = True)
312
313
314
317if D3PDMakerFlags.DoTruth:
318 truthClassification = \
319 MuonD3PDObject.defineBlock (1, 'TruthClassification',
320 D3PD.MuonTruthClassificationFillerTool)
321 def _truthClassificationHook (c, flags, acc, *args, **kw):
322 from TruthD3PDMaker.MCTruthClassifierConfig \
323 import D3PDMCTruthClassifierCfg
324 acc.merge (D3PDMCTruthClassifierCfg (flags))
325 c.Classifier = acc.getPublicTool ('D3PDMCTruthClassifier')
326 return
327 truthClassification.defineHook (_truthClassificationHook)
328
329 MuonTruthPartAssoc = SimpleAssociation \
330 (MuonD3PDObject,
331 D3PD.MuonGenParticleAssociationTool,
332 blockname = 'TruthAssoc',
333 DRVar = 'dr')
334 def _truthClassificationAssocHook (c, flags, acc, *args, **kw):
335 from TruthD3PDMaker.MCTruthClassifierConfig \
336 import D3PDMCTruthClassifierCfg
337 acc.merge (D3PDMCTruthClassifierCfg (flags))
338 c.Associator.Classifier = acc.getPublicTool ('D3PDMCTruthClassifier')
339 return
340 MuonTruthPartAssoc.defineHook (_truthClassificationAssocHook)
341 MuonTruthPartAssoc.defineBlock (0, 'TruthKin',
342 D3PD.FourMomFillerTool,
343 WriteE = True,
344 WriteM = False,
345 prefix = 'truth_')
346 MuonTruthPartAssoc.defineBlock (0, 'Truth',
347 D3PD.TruthParticleFillerTool,
348 PDGIDVariable = 'type',
349 prefix = 'truth_')
350 MuonTruthPartMotherAssoc = SimpleAssociation \
351 (MuonTruthPartAssoc,
352 D3PD.FirstAssociationTool,
353 # TruthD3PDMaker
354 Associator = D3PD.TruthParticleParentAssociationTool
355 ('MuonTruthPartMotherAssoc2'),
356 blockname = 'MuonTruthPartMotherAssoc',
357 prefix = 'truth_mother')
358 MuonTruthPartMotherAssoc.defineBlock (0, 'MotherTruth',
359 # TruthD3PDMaker
360 D3PD.TruthParticleFillerTool,
361 PDGIDVariable = 'type')
362 MuonTruthPartAssoc.defineBlock (0, 'TruthAssocIndex',
363 D3PD.IndexFillerTool,
364 prefix = 'mc_',
365 Target = 'mc_')
366
367
D3PD::AuxDataFillerTool
Copy aux data to D3PD.
Definition AuxDataFillerTool.h:35
D3PD::CovarianceFillerTool
Block filler tool for error and covariance matrix.
Definition CovarianceFillerTool.h:31
D3PD::FirstAssociationTool
Choose the first result from a multiple association.
Definition FirstAssociationTool.h:39
D3PD::FourMomFillerTool
Block filler tool for a four-momentum.
Definition FourMomFillerTool.h:56
D3PD::IndexFillerTool
Fill an index of an object within a container.
Definition IndexFillerTool.h:32
D3PD::MuonGenParticleAssociationTool
Associate from an electron/photon to a matched GenParticle via classification tool.
Definition MuonGenParticleAssociationTool.h:37
D3PD::MuonTrackParticleAssociationTool
Associate from an Muon to its primary TrackParticle.
Definition MuonTrackParticleAssociationTool.h:34
D3PD::PerigeeCovarianceAssociationTool
Associate from a Perigee to its ErrorMatrix.
Definition PerigeeCovarianceAssociationTool.h:28
D3PD::TrackParticlePerigeeAtOOAssociationTool
Associate from a TrackParticle to its default Perigee.
Definition TrackParticlePerigeeAtOOAssociationTool.h:32
D3PD::TrackParticlePerigeeAtPVAssociationTool
Associate from a VxCandidate to its perigee at primary vertex.
Definition TrackParticlePerigeeAtPVAssociationTool.h:33
python.MuonD3PDObject._truthClassificationHook
_truthClassificationHook(c, flags, acc, *args, **kw)
Definition MuonD3PDObject.py:321
python.MuonD3PDObject._truthClassificationAssocHook
_truthClassificationAssocHook(c, flags, acc, *args, **kw)
Definition MuonD3PDObject.py:334
python.MuonD3PDObject._trackToVertexHook
_trackToVertexHook(c, flags, acc, *args, **kw)
Definition MuonD3PDObject.py:175
Generated on for ATLAS Offline Software by doxygen 1.14.0