ATLAS Offline Software
Loading...
Searching...
No Matches
TCAL1.py
Go to the documentation of this file.
1# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
2
3from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
4from AthenaConfiguration.ComponentFactory import CompFactory
5from AthenaConfiguration.Enums import BeamType, MetadataCategory
6from AthenaCommon.SystemOfUnits import GeV
7
8
9def TCAL1TrackToolsCfg(flags, **kwargs):
10 """ Configure the TrackTools tool """
11
12 acc = ComponentAccumulator()
13
14 from TrackToCalo.TrackToCaloConfig import ParticleCaloExtensionToolCfg
15 kwargs.setdefault('ParticleCaloExtensionTool', acc.popToolsAndMerge(
16 ParticleCaloExtensionToolCfg(flags)))
17
18 kwargs.setdefault('IsCollision', flags.Beam.Type is BeamType.Collisions)
19
20 TrackTools = CompFactory.TileCal.TrackTools
21 acc.setPrivateTools(TrackTools(**kwargs))
22
23 return acc
24
25
26def TCAL1TileCellsDecoratorCfg(flags, Prefix="TCAL1_", **kwargs):
27 """ Configure the Tile Cells decorator tool """
28
29 acc = ComponentAccumulator()
30 kwargs.setdefault("CellsEnergy", Prefix + "cells_energy")
31 kwargs.setdefault("CellsEt", Prefix + "cells_et")
32 kwargs.setdefault("CellsEta", Prefix + "cells_eta")
33 kwargs.setdefault("CellsPhi", Prefix + "cells_phi")
34 kwargs.setdefault("CellsGain", Prefix + "cells_gain")
35 kwargs.setdefault("CellsBad", Prefix + "cells_bad")
36 kwargs.setdefault("CellsSampling", Prefix + "cells_sampling")
37 kwargs.setdefault("CellsTime", Prefix + "cells_time")
38 kwargs.setdefault("CellsQuality", Prefix + "cells_quality")
39 kwargs.setdefault("CellsSinTh", Prefix + "cells_sinTh")
40 kwargs.setdefault("CellsCosTh", Prefix + "cells_cosTh")
41 kwargs.setdefault("CellsCotTh", Prefix + "cells_cotTh")
42 kwargs.setdefault("CellsX", Prefix + "cells_x")
43 kwargs.setdefault("CellsY", Prefix + "cells_y")
44 kwargs.setdefault("CellsZ", Prefix + "cells_z")
45 kwargs.setdefault("CellsR", Prefix + "cells_r")
46 kwargs.setdefault("CellsDx", Prefix + "cells_dx")
47 kwargs.setdefault("CellsDy", Prefix + "cells_dy")
48 kwargs.setdefault("CellsDz", Prefix + "cells_dz")
49 kwargs.setdefault("CellsDr", Prefix + "cells_dr")
50 kwargs.setdefault("CellsVolume", Prefix + "cells_volume")
51 kwargs.setdefault("CellsDeta", Prefix + "cells_deta")
52 kwargs.setdefault("CellsDphi", Prefix + "cells_dphi")
53 kwargs.setdefault("CellsSide", Prefix + "cells_side")
54 kwargs.setdefault("CellsSection", Prefix + "cells_section")
55 kwargs.setdefault("CellsModule", Prefix + "cells_module")
56 kwargs.setdefault("CellsTower", Prefix + "cells_tower")
57 kwargs.setdefault("CellsSample", Prefix + "cells_sample")
58 kwargs.setdefault("CellsPmt1Ros", Prefix + "cells_pmt1_ros")
59 kwargs.setdefault("CellsPmt2Ros", Prefix + "cells_pmt2_ros")
60 kwargs.setdefault("CellsPmt1Drawer", Prefix + "cells_pmt1_drawer")
61 kwargs.setdefault("CellsPmt2Drawer", Prefix + "cells_pmt2_drawer")
62 kwargs.setdefault("CellsPmt1Channel", Prefix + "cells_pmt1_channel")
63 kwargs.setdefault("CellsPmt2Channel", Prefix + "cells_pmt2_channel")
64 kwargs.setdefault("CellsPmt1Energy", Prefix + "cells_pmt1_energy")
65 kwargs.setdefault("CellsPmt2Energy", Prefix + "cells_pmt2_energy")
66 kwargs.setdefault("CellsPmt1Time", Prefix + "cells_pmt1_time")
67 kwargs.setdefault("CellsPmt2Time", Prefix + "cells_pmt2_time")
68 kwargs.setdefault("CellsPmt1Quality", Prefix + "cells_pmt1_quality")
69 kwargs.setdefault("CellsPmt2Quality", Prefix + "cells_pmt2_quality")
70 kwargs.setdefault("CellsPmt1Qbit", Prefix + "cells_pmt1_qbit")
71 kwargs.setdefault("CellsPmt2Qbit", Prefix + "cells_pmt2_qbit")
72 kwargs.setdefault("CellsPmt1Bad", Prefix + "cells_pmt1_bad")
73 kwargs.setdefault("CellsPmt2Bad", Prefix + "cells_pmt2_bad")
74 kwargs.setdefault("CellsPmt1Gain", Prefix + "cells_pmt1_gain")
75 kwargs.setdefault("CellsPmt2Gain", Prefix + "cells_pmt2_gain")
76 acc.setPrivateTools(CompFactory.DerivationFramework.TileCellsDecorator(name="TileCellsDecorator", **kwargs))
77 return acc
78
79
80def TCAL1TileCellsMuonDecoratorCfg(flags, **kwargs):
81 """ Configure the Tile Cells Muon decorator augmentation tool """
82
83 acc = ComponentAccumulator()
84
85 from TileGeoModel.TileGMConfig import TileGMCfg
86 acc.merge(TileGMCfg(flags))
87
88 from TileConditions.TileCablingSvcConfig import TileCablingSvcCfg
89 acc.merge(TileCablingSvcCfg(flags))
90
91 kwargs.setdefault('name', 'TCAL1TileCellsMuonDecorator')
92 prefix = kwargs.pop('Prefix', 'TCAL1_')
93 kwargs.setdefault('IsoCone', 0.4)
94 kwargs.setdefault('DeltaRCones', [0.2, 0.4])
95 kwargs.setdefault('MuonContainer', 'Muons')
96 kwargs.setdefault('ClusterContainer', "CaloCalTopoClusters")
97 # Moved logic of setting WriteDecorHandleKeys to python to make configuration clearer
98 kwargs.setdefault("SelectedMuon", prefix + "SelectedMuon")
99 kwargs.setdefault("Etrkcone", prefix + "etrkcone" + str(int(kwargs['IsoCone'] * 100)))
100 kwargs.setdefault("CellsMuonX", prefix + "cells_muon_x")
101 kwargs.setdefault("CellsMuonY", prefix + "cells_muon_y")
102 kwargs.setdefault( "CellsMuonZ", prefix + "cells_muon_z")
103 kwargs.setdefault("CellsMuonEta", prefix + "cells_muon_eta")
104 kwargs.setdefault("CellsMuonPhi", prefix + "cells_muon_phi")
105 kwargs.setdefault("CellsToMuonDx", prefix + "cells_to_muon_dx")
106 kwargs.setdefault("CellsToMuonDy", prefix + "cells_to_muon_dy")
107 kwargs.setdefault("CellsToMuonDz", prefix + "cells_to_muon_dz")
108 kwargs.setdefault("CellsToMuonDeta", prefix + "cells_to_muon_deta")
109 kwargs.setdefault("CellsToMuonDphi", prefix + "cells_to_muon_dphi")
110 kwargs.setdefault("CellsMuonDx", prefix + "cells_muon_dx")
111 kwargs.setdefault("CellsMuonDeDx", prefix + "cells_muon_dedx")
112 if len(kwargs['ClusterContainer']) > 0:
113 kwargs.setdefault("LArEnergyInCone", [f"{prefix}elarcone{str(int(x*100))}" for x in kwargs['DeltaRCones']])
114
115 kwargs.setdefault('SelectMuons', flags.Beam.Type is BeamType.Collisions)
116 kwargs.setdefault('MinMuonPt', 10 * GeV)
117 kwargs.setdefault('MaxAbsMuonEta', 1.7)
118 kwargs.setdefault('MaxRelETrkInIsoCone', 100000)
119
120 kwargs.setdefault('TrackTools', acc.popToolsAndMerge(TCAL1TrackToolsCfg(flags)) )
121
122 kwargs.setdefault('CellsDecorator', acc.popToolsAndMerge(TCAL1TileCellsDecoratorCfg(flags,Prefix=prefix)))
123
124 kwargs.setdefault('TracksInConeTool', CompFactory.xAOD.TrackParticlesInConeTool())
125
126 TileCellsMuonDecorator = CompFactory.DerivationFramework.TileCellsMuonDecorator
127 acc.addPublicTool(TileCellsMuonDecorator(**kwargs), primary = True)
128
129 return acc
130
131
132def TCAL1StringSkimmingToolCfg(flags, **kwargs):
133 """Configure TCAL1 derivation framework skimming tool"""
134
135 prefix = kwargs.pop('Prefix', 'TCAL1_')
136
137 selectionExpression = ""
138 if flags.Beam.Type is BeamType.Collisions:
139 selectionExpression = f'(Muons.ptvarcone30_Nonprompt_All_MaxWeightTTVA_pt500 + 0.4 * Muons.neflowisol20) / Muons.pt < 0.18 && Muons.{prefix}SelectedMuon'
140 else:
141 selectionExpression = 'abs(Muons.eta) < 1.7'
142
143 skimmingExpression = f'count({selectionExpression}) > 0'
144
145 kwargs.setdefault('name', 'TCAL1StringSkimmingTool')
146 kwargs.setdefault('expression', skimmingExpression)
147
148 from DerivationFrameworkTools.DerivationFrameworkToolsConfig import (
149 xAODStringSkimmingToolCfg)
150 return xAODStringSkimmingToolCfg(flags, **kwargs)
151
152
153def TCAL1MuonTPThinningToolCfg(flags, streamName, **kwargs):
154 """Configure TCAL1 derivation framework thinning tool"""
155
156 acc = ComponentAccumulator()
157
158 kwargs['StreamName'] = streamName
159 kwargs.setdefault('name', 'TCAL1MuonTPThinningTool')
160 kwargs.setdefault('MuonKey', 'Muons')
161 kwargs.setdefault('InDetTrackParticlesKey', 'InDetTrackParticles')
162
163 from DerivationFrameworkInDet.InDetToolsConfig import MuonTrackParticleThinningCfg
164 muonTPThinningTool = acc.getPrimaryAndMerge(MuonTrackParticleThinningCfg(flags, **kwargs))
165
166 acc.addPublicTool(muonTPThinningTool, primary = True)
167
168 return acc
169
170
171def TCAL1KernelCfg(flags, name='TCAL1Kernel', **kwargs):
172 """Configure the TCAL1 derivation framework driving algorithm (kernel)"""
173
174 acc = ComponentAccumulator()
175
176 prefix = kwargs.pop('Prefix', 'TCAL1_')
177 streamName = kwargs.pop('StreamName', 'StreamDAOD_TCAL1')
178
179 # Common augmentations
180 triggerListsHelper = kwargs.pop('TriggerListsHelper', 'TriggerListsHelper')
181 from DerivationFrameworkPhys.PhysCommonConfig import PhysCommonAugmentationsCfg
182 acc.merge(PhysCommonAugmentationsCfg(flags, TriggerListsHelper = triggerListsHelper))
183
184 deltaRCones = kwargs.pop('DeltaRCones', [0.2, 0.4])
185 cellsMuonDecorator = acc.getPrimaryAndMerge( TCAL1TileCellsMuonDecoratorCfg(flags, Prefix=prefix, DeltaRCones=deltaRCones) )
186 kwargs.setdefault('AugmentationTools', [cellsMuonDecorator])
187
188 skimmingTool = acc.getPrimaryAndMerge(TCAL1StringSkimmingToolCfg(flags, Prefix=prefix))
189 kwargs.setdefault('SkimmingTools', [skimmingTool])
190
191 thinningTool = acc.getPrimaryAndMerge(TCAL1MuonTPThinningToolCfg(flags, streamName=streamName))
192 kwargs.setdefault('ThinningTools', [thinningTool])
193
194 DerivationKernel = CompFactory.DerivationFramework.DerivationKernel
195 acc.addEventAlgo(DerivationKernel(name, **kwargs))
196
197 return acc
198
199
200def TCAL1Cfg(flags):
201 """Configure the TCAL1 derivation framework"""
202
203 TCAL1Prefix = 'TCAL1_'
204 from DerivationFrameworkPhys.TriggerListsHelper import TriggerListsHelper
205 TCAL1TriggerListsHelper = TriggerListsHelper(flags)
206
207 deltaRCones = [0.2, 0.4]
208 acc = ComponentAccumulator()
209 acc.merge(TCAL1KernelCfg(flags, name="TCAL1Kernel", StreamName="StreamDAOD_TCAL1", Prefix=TCAL1Prefix, TriggerListsHelper=TCAL1TriggerListsHelper, DeltaRCones=deltaRCones))
210
211 from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
212 from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
213 from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
214 TCAL1SlimmingHelper = SlimmingHelper("TCAL1SlimmingHelper", NamesAndTypes = flags.Input.TypedCollections, flags = flags)
215 TCAL1SlimmingHelper.SmartCollections = ['EventInfo', 'Muons', 'AntiKt4EMTopoJets', 'AntiKt4EMPFlowJets', 'MET_Baseline_AntiKt4EMTopo', 'MET_Baseline_AntiKt4EMPFlow', 'PrimaryVertices']
216
217 TCAL1ExtraVariables = f'Muons.{TCAL1Prefix}etrkcone40'
218
219 TCAL1ExtraVariables += f'.{TCAL1Prefix}cells_'.join(['', 'energy', 'et', 'eta', 'phi', 'gain', 'bad', 'time', 'quality'])
220 TCAL1ExtraVariables += f'.{TCAL1Prefix}cells_'.join(['', 'sampling', 'sinTh', 'cosTh', 'cotTh', 'x', 'y', 'z'])
221
222 TCAL1ExtraVariables += f'.{TCAL1Prefix}cells_'.join(['', 'side', 'section', 'module', 'tower', 'sample'])
223 TCAL1ExtraVariables += f'.{TCAL1Prefix}cells_'.join(['', 'r', 'dx', 'dy', 'dz', 'dr', 'dphi', 'deta', 'volume'])
224
225 TCAL1ExtraVariables += f'.{TCAL1Prefix}cells_muon_'.join(['', 'dx', 'dedx', 'x', 'y', 'z', 'eta', 'phi'])
226 TCAL1ExtraVariables += f'.{TCAL1Prefix}cells_to_muon_'.join(['', 'dx', 'dy', 'dz', 'deta', 'dphi'])
227
228 for pmt in ['pmt1', 'pmt2']:
229 TCAL1ExtraVariables += f'.{TCAL1Prefix}cells_{pmt}_'.join(['', 'ros', 'drawer', 'channel', 'energy', 'time', 'quality', 'qbit', 'bad', 'gain'])
230
231 for drCone in deltaRCones:
232 TCAL1ExtraVariables += f'.{TCAL1Prefix}elarcone{int(drCone*100)}'
233
234 TCAL1SlimmingHelper.ExtraVariables = [TCAL1ExtraVariables]
235
236 # Trigger matching
237 # Run 2
238 if flags.Trigger.EDMVersion == 2:
239 from DerivationFrameworkPhys.TriggerMatchingCommonConfig import AddRun2TriggerMatchingToSlimmingHelper
240 AddRun2TriggerMatchingToSlimmingHelper(SlimmingHelper = TCAL1SlimmingHelper,
241 OutputContainerPrefix = "TrigMatch_",
242 TriggerList = TCAL1TriggerListsHelper.Run2TriggerNamesTau)
243 AddRun2TriggerMatchingToSlimmingHelper(SlimmingHelper = TCAL1SlimmingHelper,
244 OutputContainerPrefix = "TrigMatch_",
245 TriggerList = TCAL1TriggerListsHelper.Run2TriggerNamesNoTau)
246 # Run 3, or Run 2 with navigation conversion
247 if flags.Trigger.EDMVersion == 3 or (flags.Trigger.EDMVersion == 2 and flags.Trigger.doEDMVersionConversion):
248 from TrigNavSlimmingMT.TrigNavSlimmingMTConfig import AddRun3TrigNavSlimmingCollectionsToSlimmingHelper
249 AddRun3TrigNavSlimmingCollectionsToSlimmingHelper(TCAL1SlimmingHelper)
250
251 TCAL1ItemList = TCAL1SlimmingHelper.GetItemList()
252 acc.merge(OutputStreamCfg(flags, "DAOD_TCAL1", ItemList=TCAL1ItemList, AcceptAlgs=["TCAL1Kernel"]))
253 acc.merge(SetupMetaDataForStreamCfg(flags, "DAOD_TCAL1", AcceptAlgs=["TCAL1Kernel"], createMetadata=[MetadataCategory.CutFlowMetaData]))
254
255 return acc
TCAL1.TCAL1Cfg
TCAL1Cfg(flags)
Definition TCAL1.py:200
TCAL1.TCAL1KernelCfg
TCAL1KernelCfg(flags, name='TCAL1Kernel', **kwargs)
Definition TCAL1.py:171
TCAL1.TCAL1StringSkimmingToolCfg
TCAL1StringSkimmingToolCfg(flags, **kwargs)
Definition TCAL1.py:132
TCAL1.TCAL1TileCellsMuonDecoratorCfg
TCAL1TileCellsMuonDecoratorCfg(flags, **kwargs)
Definition TCAL1.py:80
TCAL1.TCAL1TrackToolsCfg
TCAL1TrackToolsCfg(flags, **kwargs)
Definition TCAL1.py:9
TCAL1.TCAL1MuonTPThinningToolCfg
TCAL1MuonTPThinningToolCfg(flags, streamName, **kwargs)
Definition TCAL1.py:153
TCAL1.TCAL1TileCellsDecoratorCfg
TCAL1TileCellsDecoratorCfg(flags, Prefix="TCAL1_", **kwargs)
Definition TCAL1.py:26
Generated on for ATLAS Offline Software by doxygen 1.14.0