ATLAS Offline Software
Loading...
Searching...
No Matches
BPHY24.py
Go to the documentation of this file.
1# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
2
3#====================================================================
4# BPHY24.py
5#====================================================================
6
7from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
8from AthenaConfiguration.ComponentFactory import CompFactory
9
10
11BPHYDerivationName = "BPHY24"
12streamName = "StreamDAOD_BPHY24"
13
14def BPHY24Cfg(flags):
15
16 # Lists for better code organization
17 augsList = [] # List of active augmentation tools
18 skimList = [] # List of active skimming algorithms
19 thinList = [] # List of active thinning algorithms
20 outVtxList = [] # List of reconstructed candidates to store
21 outRePVList = [] # List of candidates holding refitted primary vertices
22 thinTrkVtxList = [] # List of reconstructed candidates to use for the thinning of tracks from vertices
23 thinPassFlagsList = [] # List of pass-flags in the reconstructed candidates to se for the thinning
24 finalCandidateList = []
25
26 from AthenaServices.PartPropSvcConfig import PartPropSvcCfg
27 from DerivationFrameworkBPhys.commonBPHYMethodsCfg import (
28 BPHY_V0ToolCfg, BPHY_InDetDetailedTrackSelectorToolCfg,
29 BPHY_VertexPointEstimatorCfg, BPHY_TrkVKalVrtFitterCfg,
30 AugOriginalCountsCfg)
31 from JpsiUpsilonTools.JpsiUpsilonToolsConfig import PrimaryVertexRefittingToolCfg
32 acc = ComponentAccumulator()
33 PartPropSvcName = acc.getPrimaryAndMerge(PartPropSvcCfg(flags))
34 isSimulation = flags.Input.isMC
35 V0Tools = acc.popToolsAndMerge(BPHY_V0ToolCfg(flags, BPHYDerivationName))
36 vkalvrt = acc.popToolsAndMerge(BPHY_TrkVKalVrtFitterCfg(flags, BPHYDerivationName)) # VKalVrt vertex fitter
37 acc.addPublicTool(vkalvrt)
38 acc.addPublicTool(V0Tools)
39 trackselect = acc.popToolsAndMerge(BPHY_InDetDetailedTrackSelectorToolCfg(flags, BPHYDerivationName))
40 acc.addPublicTool(trackselect)
41 vpest = acc.popToolsAndMerge(BPHY_VertexPointEstimatorCfg(flags, BPHYDerivationName))
42 acc.addPublicTool(vpest)
43 PVrefit = acc.popToolsAndMerge(PrimaryVertexRefittingToolCfg(flags))
44 acc.addPublicTool(PVrefit)
45
46 bSkim = "(count(BPHY24JpsimmKshortCascadeSV1.Bd_mass) + count(BPHY24JpsieeKshortCascadeSV1.Bd_mass)) > 0"
47
48 BPHY24_Skim_Bcandidates = CompFactory.DerivationFramework.xAODStringSkimmingTool( name = "BPHY24_Skim_Bcandidates",
49 expression = bSkim )
50 skimList += [ BPHY24_Skim_Bcandidates ]
51
52 BPHY24_AugOriginalCounts = acc.popToolsAndMerge(
53 AugOriginalCountsCfg(flags, name = "BPHY24_AugOriginalCounts"))
54 augsList += [ BPHY24_AugOriginalCounts ]
55
56 # LRT track merge
57 from DerivationFrameworkInDet.InDetToolsConfig import InDetLRTMergeCfg
58 acc.merge(InDetLRTMergeCfg(flags))
59
60 mainIDInput = "InDetWithLRTTrackParticles"
61 originalTrackCond = ["InDetTrackParticles", "InDetLargeD0TrackParticles","GSFTrackParticles"]
62 BPHY24_Finder_DiMuon = CompFactory.Analysis.JpsiFinder( name = "BPHY24_Finder_DiMuon",
63 muAndMu = True,
64 muAndTrack = False,
65 TrackAndTrack = False,
66 assumeDiMuons = True,
67 muonThresholdPt = 3000.,
68 higherPt = 3500.,
69 invMassUpper = 7000.,
70 invMassLower = 1.,
71 Chi2Cut = 30.,
72 oppChargesOnly = False,
73 allChargeCombinations = True,
74 atLeastOneComb = True,
75 useCombinedMeasurement = False, # Only takes effect if combOnly=True
76 muonCollectionKey = "Muons",
77 TrackParticleCollection = "InDetTrackParticles",
78 V0VertexFitterTool = None,
79 useV0Fitter = False,
80 TrkVertexFitterTool = vkalvrt,
81 TrackSelectorTool = trackselect,
82 VertexPointEstimator = vpest,
83 useMCPCuts = False )
84
85
86 BPHY24_SelectAndWrite_DiMuon = CompFactory.DerivationFramework.Reco_mumu( name = "BPHY24_SelectAndWrite_DiMuon",
87 JpsiFinder = BPHY24_Finder_DiMuon,
88 V0Tools = V0Tools,
89 PVRefitter = PVrefit,
90 OutputVtxContainerName = "BPHY24_DiMuon_Candidates",
91 PVContainerName = "PrimaryVertices",
92 RefPVContainerName = "SHOULDNOTBEUSED", # The container would be created if PV refit was requested (not needed at this point)
93 DoVertexType = 7 ) # Vertex type marking our own reconstruced secondary candidates
94
95 augsList += [ BPHY24_SelectAndWrite_DiMuon ]
96 # Final selection of the di-muon candidates
97 thinTrkVtxList += [ "BPHY24_DiMuon_Candidates" ]
98 outVtxList += [ "BPHY24_DiMuon_Candidates" ]
99 thinPassFlagsList += [ "passed_Jpsi" ] # TODO: is this really needed?
100
101 BPHY24_Select_DiMuons = CompFactory.DerivationFramework.Select_onia2mumu( name = "BPHY24_Select_DiMuons",
102 HypothesisName = "Jpsi",
103 InputVtxContainerName = "BPHY24_DiMuon_Candidates",
104 V0Tools = V0Tools,
105 VtxMassHypo = 3096.916, # used only for pseudo-proper decay time etc. calculations
106 MassMax = 10000., # loose cut to keep selection from BPHY24_Finder_DiMuon
107 MassMin = 0., # loose cut to keep selection from BPHY24_Finder_DiMuon
108 Chi2Max = 1000., # loose cut to keep selection from BPHY24_Finder_DiMuon (chi2, not chi2/NDF)
109 DoVertexType = 7 ) # Vertex type marking our own reconstruced secondary candidates
110
111 augsList += [ BPHY24_Select_DiMuons ]
112
113 ElectronLHSelectorLHvloose = CompFactory.AsgElectronLikelihoodTool("ElectronLHSelectorLHvloose",
114 primaryVertexContainer = "PrimaryVertices",
115 ConfigFile="ElectronPhotonSelectorTools/offline/mc20_20210514/ElectronLikelihoodVeryLooseOfflineConfig2017_Smooth.conf")
116
117
118
119 ElectronLHSelectorLHvloose_nod0 = CompFactory.AsgElectronLikelihoodTool("ElectronLHSelectorLHvloosenod0", primaryVertexContainer = "PrimaryVertices",
120 ConfigFile="ElectronPhotonSelectorTools/offline/mc16_20190328_nod0/ElectronLikelihoodVeryLooseOfflineConfig2017_Smooth_nod0.conf") # Still OK to use in Run3?
121
122
123 # decorate electrons with the output of LH vloose (nod0)
124 from DerivationFrameworkEGamma.EGammaToolsConfig import EGElectronLikelihoodToolWrapperCfg
125 ElectronPassLHvloose = acc.addPublicTool(acc.popToolsAndMerge(EGElectronLikelihoodToolWrapperCfg(flags, name = "ElectronPassLHvloose",
126 EGammaElectronLikelihoodTool = ElectronLHSelectorLHvloose,
127 EGammaFudgeMCTool = "",
128 CutType = "",
129 StoreGateEntryName = "DFCommonElectronsLHVeryLoose",
130 ContainerName = "Electrons",
131 StoreTResult=False)))
132
133 ElectronPassLHvloosenod0 = acc.addPublicTool(acc.popToolsAndMerge(EGElectronLikelihoodToolWrapperCfg(flags, name = "ElectronPassLHvloosenod0",
134 EGammaElectronLikelihoodTool = ElectronLHSelectorLHvloose_nod0,
135 EGammaFudgeMCTool = "",
136 CutType = "",
137 StoreGateEntryName = "DFCommonElectronsLHVeryLoosenod0",
138 ContainerName = "Electrons",
139 StoreTResult=False)))
140 augsList += [ElectronPassLHvloose, ElectronPassLHvloosenod0]
141
142 BPHY24DiElectronFinder = CompFactory.Analysis.JpsiFinder_ee(
143 name = "BPHY24DiElectronFinder",
144 elAndEl = True,
145 elAndTrack = False,
146 TrackAndTrack = False,
147 assumeDiElectrons = True,
148 elThresholdPt = 4000.0,
149 invMassUpper = 7000.0,
150 invMassLower = 200.0,
151 Chi2Cut = 30.,
152 oppChargesOnly = False,
153 allChargeCombinations = True,
154 useElectronTrackMeasurement = True,
155 electronCollectionKey = "Electrons",
156 TrackParticleCollection = "GSFTrackParticles",
157 useEgammaCuts = True,
158 V0VertexFitterTool = None,
159 useV0Fitter = False,
160 TrkVertexFitterTool = vkalvrt,
161 TrackSelectorTool = trackselect,
162 VertexPointEstimator = vpest,
163 ElectronSelection = "d0_or_nod0",
164 PartPropSvc = PartPropSvcName
165 )
166
167 BPHY24_SelectAndWrite_DiElectron = CompFactory.DerivationFramework.Reco_mumu(
168 name = "BPHY24_SelectAndWrite_DiElectron",
169 JpsiFinder = BPHY24DiElectronFinder,
170 V0Tools = V0Tools,
171 PVRefitter = PVrefit,
172 OutputVtxContainerName = "BPHY24_DiElectron_Candidates",
173 PVContainerName = "PrimaryVertices",
174 RefPVContainerName = "SHOULDNOTBEUSED",
175 DoVertexType = 7
176 )
177
178 augsList += [ BPHY24_SelectAndWrite_DiElectron ]
179
180 BPHY24_Select_DiElectrons = CompFactory.DerivationFramework.Select_onia2mumu(
181 name = "BPHY24_Select_DiElectrons",
182 HypothesisName = "Jpsi",
183 InputVtxContainerName = "BPHY24_DiElectron_Candidates",
184 V0Tools = V0Tools,
185 VtxMassHypo = 3096.916,
186 MassMin = 400.0,
187 MassMax = 7000.0,
188 Chi2Max = 30,
189 DoVertexType = 7
190 )
191
192 thinTrkVtxList += [ "BPHY24_DiElectron_Candidates" ]
193 outVtxList += [ "BPHY24_DiElectron_Candidates" ]
194
195 augsList += [ BPHY24_Select_DiElectrons ]
196
197 V0ContainerName = "BPHY24RecoV0Candidates"
198 KshortContainerName = "BPHY24RecoKshortCandidates"
199 LambdaContainerName = "BPHY24RecoLambdaCandidates"
200 LambdabarContainerName = "BPHY24RecoLambdabarCandidates"
201
202 from DerivationFrameworkBPhys.V0ToolConfig import (
203 BPHY_InDetV0FinderToolCfg, BPHY_Reco_V0FinderCfg)
204
205 BPHY24_V0FinderTool = acc.popToolsAndMerge(BPHY_InDetV0FinderToolCfg(
206 flags, BPHYDerivationName,
207 TrackParticleCollection = mainIDInput,
208 V0ContainerName = V0ContainerName,
209 KshortContainerName = KshortContainerName,
210 LambdaContainerName = LambdaContainerName,
211 LambdabarContainerName = LambdabarContainerName,
212 RelinkTracks = originalTrackCond,
213 use_innerPixHits = True,
214 errmass = 99999,
215 uksmin = 400,
216 uksmax = 600,
217 ksmin = 400,
218 ksmax = 600,
219 ulamin = -1,
220 ulamax = -1,
221 lamin = -1,
222 lamax = -1,
223 d0_cut = -1,
224 max_d0_cut = 50,
225 max_z0_cut = 250,
226 ptTRT = 0,
227 minConstrVertProb = 0.000001,
228 useBDT = True,
229 BDTCut = 0.1,
230 use_TrackSelector = False))
231
232 BPHY24_Reco_V0Finder = acc.popToolsAndMerge(BPHY_Reco_V0FinderCfg(
233 flags, derivation = BPHYDerivationName,
234 V0ContainerName = V0ContainerName,
235 KshortContainerName = KshortContainerName,
236 LambdaContainerName = LambdaContainerName,
237 LambdabarContainerName = LambdabarContainerName,
238 VxPrimaryCandidateName = "BPHY24_DiMuon_Candidates",
239 CheckVertexContainers = ['BPHY24_DiMuon_Candidates',
240 'BPHY24_DiElectron_Candidates'],
241 V0FinderTool = BPHY24_V0FinderTool))
242
243 augsList += [BPHY24_Reco_V0Finder]
244 outVtxList += ['BPHY24RecoKshortCandidates']
245 outVtxList += ["BPHY24RecoV0Candidates"]
246 thinTrkVtxList += ['BPHY24RecoKshortCandidates']
247 thinPassFlagsList += [ "" ] # TODO: is this really needed?
248 finalCandidateList += ["BPHY24RecoKshortCandidates"]
249 finalCandidateList += ["BPHY24RecoV0Candidates"]
250 from TrkConfig.TrkVKalVrtFitterConfig import JpsiV0VertexFitCfg
251 JpsiV0VertexFit = acc.popToolsAndMerge(JpsiV0VertexFitCfg(flags))
252 acc.addPublicTool(JpsiV0VertexFit)
253
254 BPHY24JpsimmKshort = CompFactory.DerivationFramework.JpsiPlusV0Cascade(
255 name = "BPHY24mmKshort",
256 V0Tools = V0Tools,
257 HypothesisName = "Bd",
258 TrkVertexFitterTool = JpsiV0VertexFit,
259 PVRefitter = acc.popToolsAndMerge(PrimaryVertexRefittingToolCfg(flags)),
260 V0Hypothesis = 310,
261 JpsiMassLowerCut = 1.,
262 JpsiMassUpperCut = 7000.,
263 ApplyJpsiMassConstraint = False,
264 V0MassLowerCut = 400.,
265 V0MassUpperCut = 600.,
266 MassLowerCut = 4300.,
267 MassUpperCut = 6300.,
268 RefitPV = True,
269 RefPVContainerName = "BPHY24RefittedPrimaryVertices_mm",
270 JpsiVertices = "BPHY24_DiMuon_Candidates",
271 CascadeVertexCollections= ["BPHY24JpsimmKshortCascadeSV2", "BPHY24JpsimmKshortCascadeSV1"],
272 V0Vertices = "BPHY24RecoV0Candidates",
273 V0TrackContainerName = mainIDInput,
274 RelinkTracks = originalTrackCond)
275
276 augsList += [BPHY24JpsimmKshort]
277 outVtxList += BPHY24JpsimmKshort.CascadeVertexCollections
278 outVtxList += ["BPHY24RefittedPrimaryVertices_mm"]
279 thinTrkVtxList += BPHY24JpsimmKshort.CascadeVertexCollections
280 finalCandidateList += BPHY24JpsimmKshort.CascadeVertexCollections
281
282 BPHY24JpsieeKshort = CompFactory.DerivationFramework.JpsiPlusV0Cascade(
283 name = "BPHY24eeKshort",
284 V0Tools = V0Tools,
285 HypothesisName = "Bd",
286 TrkVertexFitterTool = JpsiV0VertexFit,
287 PVRefitter = acc.popToolsAndMerge(PrimaryVertexRefittingToolCfg(flags)),
288 V0Hypothesis = 310,
289 JpsiMassLowerCut = 100.,
290 JpsiMassUpperCut = 7000.,
291 ApplyJpsiMassConstraint = False,
292 V0MassLowerCut = 400.,
293 V0MassUpperCut = 600.,
294 MassLowerCut = 4300.,
295 MassUpperCut = 6300.,
296 JpsiTrackPDGID = 11,
297 JpsiTrackContainerName = "GSFTrackParticles",
298 RefitPV = True,
299 RefPVContainerName = "BPHY24RefittedPrimaryVertices_ee",
300 JpsiVertices = "BPHY24_DiElectron_Candidates",
301 CascadeVertexCollections= ["BPHY24JpsieeKshortCascadeSV2", "BPHY24JpsieeKshortCascadeSV1"],
302 V0Vertices = "BPHY24RecoV0Candidates",
303 V0TrackContainerName = mainIDInput,
304 RelinkTracks = originalTrackCond)
305
306 augsList += [BPHY24JpsieeKshort]
307 finalCandidateList += BPHY24JpsieeKshort.CascadeVertexCollections
308 outVtxList += BPHY24JpsieeKshort.CascadeVertexCollections
309 outVtxList += ["BPHY24RefittedPrimaryVertices_ee"]
310 thinTrkVtxList += BPHY24JpsieeKshort.CascadeVertexCollections
311
312 from TrackVertexAssociationTool.TrackVertexAssociationToolConfig import isoTTVAToolCfg
313 TTVATool = acc.popToolsAndMerge(isoTTVAToolCfg(flags,
314 WorkingPoint = "Custom",
315 d0_cut = -1,
316 d0sig_cut = -1,
317 dzSinTheta_cut = -1,
318 doUsedInFit = False))
319 acc.addPublicTool(TTVATool)
320
321 from InDetConfig.InDetTrackSelectionToolConfig import isoTrackSelectionToolCfg
322 TrackSelTool = acc.popToolsAndMerge(isoTrackSelectionToolCfg(flags,
323 maxZ0SinTheta= 2,
324 minPt= 1000,
325 CutLevel= "Loose"))
326 acc.addPublicTool(TrackSelTool)
327
328 from IsolationAlgs.IsoToolsConfig import TrackIsolationToolCfg
329 TrackIsoTool = acc.popToolsAndMerge(TrackIsolationToolCfg(flags,
330 TrackSelectionTool = TrackSelTool,
331 TTVATool = TTVATool))
332 acc.addPublicTool(TrackIsoTool)
333 BPHY24TrackIsolationDecoratorBtoKee = CompFactory.DerivationFramework.VertexTrackIsolation(
334 name = "BPHY24TrackIsolationDecoratorBtoKee",
335 TrackIsoTool = TrackIsoTool,
336 TrackContainer = "InDetTrackParticles",
337 InputVertexContainer = "BPHY24JpsieeKshortCascadeSV1",
338 FixElecExclusion = True,
339 IncludeV0 = True)
340 BPHY24TrackIsolationDecoratorBtoKmumu = CompFactory.DerivationFramework.VertexTrackIsolation(
341 name = "BPHY24TrackIsolationDecoratorBtoKmumu ",
342 TrackIsoTool = TrackIsoTool,
343 TrackContainer = "InDetTrackParticles",
344 InputVertexContainer = "BPHY24JpsimmKshortCascadeSV1",
345 FixElecExclusion = False,
346 IncludeV0 = True)
347
348 BPHY24TrackIsolationDecoratorJpsiee = CompFactory.DerivationFramework.VertexTrackIsolation(
349 name = "BPHY24TrackIsolationDecoratorJpsiee",
350 TrackIsoTool = TrackIsoTool,
351 TrackContainer = "InDetTrackParticles",
352 InputVertexContainer = "BPHY24_DiElectron_Candidates",
353 FixElecExclusion = True,
354 IncludeV0 = False)
355
356 BPHY24TrackIsolationDecoratorJpsimumu = CompFactory.DerivationFramework.VertexTrackIsolation(
357 name = "BPHY24TrackIsolationDecoratorJpsimumu",
358 TrackIsoTool = TrackIsoTool,
359 TrackContainer = "InDetTrackParticles",
360 InputVertexContainer = "BPHY24_DiMuon_Candidates",
361 FixElecExclusion = False,
362 IncludeV0 = False)
363
364 augsList += [ BPHY24TrackIsolationDecoratorBtoKee,
365 BPHY24TrackIsolationDecoratorBtoKmumu,
366 BPHY24TrackIsolationDecoratorJpsiee,
367 BPHY24TrackIsolationDecoratorJpsimumu]
368
369 trigger_list = [ # Pure muon triggers
370 "HLT_mu11_mu6_bDimu",
371 "HLT_mu11_mu6_bDimu2700",
372 "HLT_mu11_mu6_bDimu_L1LFV-MU11",
373 "HLT_mu11_mu6_bDimu2700_L1LFV-MU11",
374 "HLT_mu11_mu6_bBmumuxv2",
375 "HLT_mu10_mu6_bBmumuxv2",
376 "HLT_mu10_mu6_bBmumuxv2_delayed",
377 "HLT_2mu6_bBmumuxv2",
378 "HLT_2mu6_bBmumuxv2_delayed",
379 "HLT_2mu6_bBmumuxv2_L1LFV-MU6",
380 "HLT_mu6_mu4_bBmumuxv2",
381 "HLT_mu6_mu4_bBmumuxv2_delayed",
382 "HLT_2mu4_bBmumuxv2",
383 "HLT_mu11_mu6_bJpsimumu",
384 "HLT_2mu10_bJpsimumu",
385 "HLT_mu11_mu6_bJpsimumu_L1LFV-MU11",
386 "HLT_2mu6_bJpsimumu_L1BPH-2M9-2MU6_BPH-2DR15-2MU6",
387 "HLT_2mu6_bJpsimumu_delayed_L1BPH-2M9-2MU6_BPH-2DR15-2MU6",
388 "HLT_2mu10_bJpsimumu_noL2",
389 "HLT_mu10_mu6_bJpsimumu",
390 "HLT_mu10_mu6_bJpsimumu_delayed",
391 "HLT_2mu6_bJpsimumu",
392 "HLT_2mu6_bJpsimumu_delayed",
393 "HLT_mu6_mu4_bJpsimumu",
394 "HLT_mu6_mu4_bJpsimumu_Lxy0_L1BPH-2M9-MU6MU4_BPH-0DR15-MU6MU4",
395 "HLT_2mu4_bJpsimumu",
396 "HLT_mu6_2mu4_bJpsi",
397 "HLT_mu6_2mu4_bJpsi_delayed",
398 "HLT_2mu14",
399 "HLT_2mu10",
400 # dielectron triggers
401 "HLT_2e5_lhvloose_nod0_bBeexM6000t", #37,143,877 inb
402 "HLT_e5_lhvloose_nod0_bBeexM6000t", #37,143,877
403 "HLT_e5_lhvloose_nod0_bBeexM6000t_2mu4_nomucomb_L1BPH-0DR3-EM7J15_2MU4", #37,312,506
404 "HLT_e5_lhvloose_nod0_bBeexM6000t_mu6_nomucomb_L1BPH-0DR3-EM7J15_MU6", #27,041,892
405 "HLT_e5_lhvloose_nod0_bBeexM6000_mu6_nomucomb_L1BPH-0DR3-EM7J15_MU6", #149,100
406 "HLT_e9_lhloose_bBeexM2700_2mu4_nomucomb_L1BPH-0DR3-EM7J15_2MU4", #2,681,764
407 "HLT_e9_lhloose_bBeexM2700_mu6_nomucomb_L1BPH-0DR3-EM7J15_MU6", #1,979,362
408 "HLT_e9_lhloose_bBeexM6000_2mu4_nomucomb_L1BPH-0DR3-EM7J15_2MU4", #3,359,105
409 "HLT_e9_lhloose_bBeexM6000_mu6_nomucomb_L1BPH-0DR3-EM7J15_MU6", #2,426,663
410 "HLT_e9_lhloose_e5_lhloose_bBeexM2700_2mu4_nomucomb_L1BPH-0M9-EM7-EM5_2MU4", #2,950,935
411 "HLT_e9_lhloose_e5_lhloose_bBeexM2700_mu6_nomucomb_L1BPH-0M9-EM7-EM5_MU6", #2,928,030
412 "HLT_e9_lhloose_e5_lhloose_bBeexM6000_2mu4_nomucomb_L1BPH-0M9-EM7-EM5_2MU4", #3,647,507
413 "HLT_e9_lhloose_e5_lhloose_bBeexM6000_mu6_nomucomb_L1BPH-0M9-EM7-EM5_MU6", #3,605,371
414 "HLT_e9_lhvloose_nod0_e5_lhvloose_nod0_bBeexM6000t_2mu4_nomucomb_L1BPH-0M9-EM7-EM5_2MU4", #40,169,436
415 "HLT_e9_lhvloose_nod0_e5_lhvloose_nod0_bBeexM6000t_mu6_nomucomb_L1BPH-0M9-EM7-EM5_MU6", #37,312,506
416 "HLT_e9_lhvloose_nod0_e5_lhvloose_nod0_bBeexM6000_mu6_nomucomb_L1BPH-0M9-EM7-EM5_MU6", #677,340
417 ]
418
419 BPHY24TrigSkimmingTool = CompFactory.DerivationFramework.TriggerSkimmingTool( name = "BPHY24TrigSkimmingTool",
420 TriggerListOR = trigger_list,
421 TriggerListORHLTOnly = ["HLT_2e5_lhvloose_nod0_bBeexM6000t","HLT_e5_lhvloose_nod0_bBeexM6000t"] )
422 acc.addPublicTool(BPHY24TrigSkimmingTool)
423 skimList += [BPHY24TrigSkimmingTool]
424
425 # ID tracks
426 BPHY24_Thin_VtxTracks = CompFactory.DerivationFramework.Thin_vtxTrk( name = "BPHY24_Thin_VtxTracks",
427 StreamName = streamName,
428 TrackParticleContainerName = "InDetTrackParticles",
429 VertexContainerNames = finalCandidateList,
430 IgnoreFlags = True )
431 # PassFlags = thinPassFlagsList )
432 thinList += [ BPHY24_Thin_VtxTracks ]
433
434 # LRT ID tracks
435 BPHY24_Thin_VtxTracks_LRT = CompFactory.DerivationFramework.Thin_vtxTrk( name = "BPHY24_Thin_VtxTracks_LRT",
436 StreamName = streamName,
437 TrackParticleContainerName = "InDetLargeD0TrackParticles",
438 VertexContainerNames = finalCandidateList,
439 IgnoreFlags = True )
440 # PassFlags = thinPassFlagsList )
441 thinList += [ BPHY24_Thin_VtxTracks_LRT ]
442
443 # GSF tracks
444 BPHY24_Thin_VtxTracks_GSF = CompFactory.DerivationFramework.Thin_vtxTrk( name = "BPHY24_Thin_VtxTracks_GSF",
445 StreamName = streamName,
446 TrackParticleContainerName = "GSFTrackParticles",
447 VertexContainerNames = finalCandidateList,
448 IgnoreFlags = True )
449 thinList += [ BPHY24_Thin_VtxTracks_GSF ]
450
451 # Muons (TODO: thinning not used muons or something else ?)
452 BPHY24_Thin_Muons = CompFactory.DerivationFramework.MuonTrackParticleThinning( name = "BPHY24_Thin_Muons",
453 MuonKey = "Muons",
454 StreamName = streamName,
455 InDetTrackParticlesKey = "InDetTrackParticles" )
456 thinList += [ BPHY24_Thin_Muons ]
457
458 # Electrons
459 BPHY24_Thin_Egamma = CompFactory.DerivationFramework.EgammaTrackParticleThinning(
460 name = "BPHY24_Thin_Egamma",
461 SGKey = "Electrons",
462 StreamName = streamName,
463 InDetTrackParticlesKey = mainIDInput)
464 thinList += [BPHY24_Thin_Egamma]
465
466 # Primary vertices
467 BPHY24_Thin_PV = CompFactory.DerivationFramework.BPhysPVThinningTool( name = "BPHY24_Thin_PV",
468 CandidateCollections = finalCandidateList,
469 StreamName = streamName,
470 KeepPVTracks = True )
471 thinList += [ BPHY24_Thin_PV ]
472
473 if isSimulation:
474
475 #Decorate Truth Particles with track parameters
476 from InDetPhysValMonitoring.InDetPhysValDecorationConfig import InDetPhysValTruthDecoratorAlgCfg
477 acc.merge(InDetPhysValTruthDecoratorAlgCfg(flags, BPHYDerivationName))
478
479 # Keep all muons and electrons
480 keepParticles = ('abs(TruthParticles.pdgId) == 11 || ' # mu
481 'abs(TruthParticles.pdgId) == 13') # e
482 # Keep only the potentially signal b-hadrons
483
484 keepParticles += (' || '
485 'abs(TruthParticles.pdgId) == 511 || ' # B0
486 'abs(TruthParticles.pdgId) == 513 || ' # B0*
487 'abs(TruthParticles.pdgId) == 515') # B0**
488
489
490 BPHY24_Thin_TruthHadrons = CompFactory.DerivationFramework.GenericTruthThinning( name = "BPHY24_Thin_TruthHadrons",
491 ParticleSelectionString = keepParticles,
492 PreserveDescendants = True,
493 StreamName = streamName,
494 PreserveAncestors = True)
495 thinList += [ BPHY24_Thin_TruthHadrons ]
496
497 # Save also neutrinos and b-quarks, without their decay trees
498 BPHY24_Thin_TruthQuarks = CompFactory.DerivationFramework.GenericTruthThinning( name = "BPHY24_Thin_TruthQuarks",
499 ParticleSelectionString = ('abs(TruthParticles.pdgId) == 5 || '
500 'abs(TruthParticles.pdgId) == 12 || abs(TruthParticles.pdgId) == 14' ),
501 PreserveDescendants = False,
502 StreamName = streamName,
503 PreserveAncestors = False)
504 thinList += [ BPHY24_Thin_TruthQuarks ]
505
506 for t in augsList + skimList + thinList: acc.addPublicTool(t)
507 acc.addEventAlgo(CompFactory.DerivationFramework.DerivationKernel("BPHY24Kernel",
508 AugmentationTools = augsList,
509 #OutputLevel = DEBUG,
510 #Only skim if not MC
511 SkimmingTools = skimList,
512 ThinningTools = thinList))
513
514
515 from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper
516 from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
517 BPHY24SlimmingHelper = SlimmingHelper("BPHY24SlimmingHelper", NamesAndTypes = flags.Input.TypedCollections, flags = flags)
518 from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
519 from AthenaConfiguration.Enums import MetadataCategory
520 from DerivationFrameworkBPhys.commonBPHYMethodsCfg import getDefaultAllVariables
521 AllVariables = getDefaultAllVariables()
522 StaticContent = []
523 ExtraVariables = []
524
525 # Smart collections
526 # What is the difference w.r.t. adding them into AllVariables?
527 # AB Answer: SmarCollections trims commonly unused variables.
528 BPHY24SlimmingHelper.SmartCollections = [ "Electrons", "Muons", "InDetTrackParticles", "InDetLargeD0TrackParticles" ]
529
530 # Full combined muon-ID tracks
531 AllVariables += ["InDetLargeD0TrackParticles"]
532 AllVariables += [ "CombinedMuonTrackParticles" ]
533 AllVariables += [ "ExtrapolatedMuonTrackParticles" ]
534 ExtraVariables += [ "Muons.etaLayer1Hits.etaLayer2Hits.etaLayer3Hits.etaLayer4Hits.phiLayer1Hits.phiLayer2Hits.phiLayer3Hits.phiLayer4Hits",
535 "Muons.numberOfTriggerEtaLayers.numberOfPhiLayers",
536 "InDetTrackParticles.numberOfTRTHits.numberOfTRTHighThresholdHits.vx.vy.vz",
537 "InDetLargeD0TrackParticles.numberOfTRTHits.numberOfTRTHighThresholdHits.vx.vy.vz",
538 "PrimaryVertices.chiSquared.covariance", "Electrons.deltaEta1.DFCommonElectronsLHVeryLoosenod0",
539 "egammaClusters.calE.calEta.calPhi.e_sampl.eta_sampl.etaCalo.phiCalo.ETACALOFRAME.PHICALOFRAME",
540 "HLT_xAOD__ElectronContainer_egamma_ElectronsAuxDyn.charge" ]
541
542 # Include also trigger objects
543 # DONE: Test it works (HLT objects appear/not-present)
544
545 BPHY24SlimmingHelper.IncludeMuonTriggerContent = True
546 BPHY24SlimmingHelper.IncludeEgammaTriggerContent = True
547 BPHY24SlimmingHelper.IncludeBPhysTriggerContent = True
548
549 # Include primary vertices
550 AllVariables += [ "PrimaryVertices" ]
551 print("BPHY24: List of refitted-PV output: ", outRePVList)
552 for i in outRePVList:
553 StaticContent += [ "xAOD::VertexContainer#%s" % i ]
554 StaticContent += [ "xAOD::VertexAuxContainer#%sAux." % i ]
555
556 # B-vertexing output
557 print("BPHY24: List of B-vertexing output: ", outVtxList)
558 for i in outVtxList:
559 StaticContent += [ "xAOD::VertexContainer#%s" % i ]
560 StaticContent += [ "xAOD::VertexAuxContainer#%sAux.-vxTrackAtVertex" % i ]
561
562 print("BPHY24: Full list of B-augmentation: ", StaticContent)
563
564 # Truth information for MC only
565 if isSimulation:
566 AllVariables += [ "TruthEvents",
567 "TruthParticles",
568 "TruthVertices",
569 "MuonTruthParticles" ]
570 AllVariables = list(set(AllVariables)) # remove duplicates
571
572 BPHY24SlimmingHelper.AllVariables = AllVariables
573 BPHY24SlimmingHelper.StaticContent = StaticContent
574 BPHY24SlimmingHelper.ExtraVariables = ExtraVariables
575 BPHY24ItemList = BPHY24SlimmingHelper.GetItemList()
576 acc.merge(OutputStreamCfg(flags, "DAOD_BPHY24", ItemList=BPHY24ItemList, AcceptAlgs=["BPHY24Kernel"]))
577 acc.merge(SetupMetaDataForStreamCfg(flags, "DAOD_BPHY24", AcceptAlgs=["BPHY24Kernel"], createMetadata=[MetadataCategory.CutFlowMetaData]))
578 acc.printConfig(withDetails=True, summariseProps=True, onlyComponents = [], printDefaults=True)
579 return acc
print
void print(char *figname, TCanvas *c1)
Definition TRTCalib_StrawStatusPlots.cxx:26
set
STL class.
BPHY24.BPHY24Cfg
BPHY24Cfg(flags)
Definition BPHY24.py:14
Generated on for ATLAS Offline Software by doxygen 1.14.0