ATLAS Offline Software
Loading...
Searching...
No Matches
FullCPAlgorithmsTest.py
Go to the documentation of this file.
1# Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
2#
3# @author Nils Krumnack
4
5from AnaAlgorithm.AlgSequence import AlgSequence
6from AnalysisAlgorithmsConfig.ConfigSequence import ConfigSequence
7from AnalysisAlgorithmsConfig.ConfigAccumulator import ConfigAccumulator, DataType
8from AthenaConfiguration.Enums import LHCPeriod
9from AthenaCommon.SystemOfUnits import GeV
10
11# Config:
12triggerChainsPerYear = {
13 2015: ['HLT_e24_lhmedium_L1EM20VH || HLT_e60_lhmedium || HLT_e120_lhloose', 'HLT_mu20_iloose_L1MU15 || HLT_mu40', 'HLT_2g20_tight'],
14 2016: ['HLT_e26_lhtight_nod0_ivarloose || HLT_e60_lhmedium_nod0 || HLT_e140_lhloose_nod0', 'HLT_mu26_ivarmedium || HLT_mu50', 'HLT_g35_loose_g25_loose'],
15 2017: ['HLT_e26_lhtight_nod0_ivarloose || HLT_e60_lhmedium_nod0 || HLT_e140_lhloose_nod0', 'HLT_2g22_tight_L12EM15VHI', 'HLT_mu50'],
16 2018: ['HLT_e26_lhtight_nod0_ivarloose || HLT_e60_lhmedium_nod0 || HLT_e140_lhloose_nod0', 'HLT_g35_medium_g25_medium_L12EM20VH', 'HLT_mu26_ivarmedium', 'HLT_2mu14'],
17 # '2022': ['HLT_e26_lhtight_ivarloose_L1EM22VHI || HLT_e60_lhmedium_L1EM22VHI || HLT_e140_lhloose_L1EM22VHI'],
18 # '2023': ['HLT_e26_lhtight_ivarloose_L1EM22VHI || HLT_e60_lhmedium_L1EM22VHI || HLT_e140_lhloose_L1EM22VHI'],
19}
20triggerMatchingChainsPerYear = {
21 2015: ['HLT_e24_lhmedium_L1EM20VH || HLT_e60_lhmedium || HLT_e120_lhloose', 'HLT_mu20_iloose_L1MU15 || HLT_mu40'],
22 2016: ['HLT_e26_lhtight_nod0_ivarloose || HLT_e60_lhmedium_nod0 || HLT_e140_lhloose_nod0', 'HLT_mu26_ivarmedium || HLT_mu50'],
23 2017: ['HLT_e26_lhtight_nod0_ivarloose || HLT_e60_lhmedium_nod0 || HLT_e140_lhloose_nod0', 'HLT_mu50'],
24 2018: ['HLT_e26_lhtight_nod0_ivarloose || HLT_e60_lhmedium_nod0 || HLT_e140_lhloose_nod0', 'HLT_mu26_ivarmedium'],
25}
26tauTriggerChainsSF = {
27 2015: ['HLT_tau25_medium1_tracktwo', 'HLT_tau35_medium1_tracktwo'],
28 2016: ['HLT_tau25_medium1_tracktwo', 'HLT_tau35_medium1_tracktwo'],
29 2017: ['HLT_tau25_medium1_tracktwo', 'HLT_tau35_medium1_tracktwo'],
30 2018: ['HLT_tau25_medium1_tracktwoEF', 'HLT_tau35_medium1_tracktwoEF'],
31 2022: ['HLT_tau25_mediumRNN_tracktwoMVA', 'HLT_tau35_mediumRNN_tracktwoMVA'],
32 2023: ['HLT_tau25_mediumRNN_tracktwoMVA', 'HLT_tau35_mediumRNN_tracktwoMVA'],
33}
34bjetTriggerChainsPerYear = {
35 2022: ['HLT_2j45_0eta290_020jvt_bdl1d60_2j45_pf_ftf_presel2j25XX2j25b85_L14J15p0ETA25'],
36 2023: ['HLT_2j45_0eta290_020jvt_bgn160_2j45_pf_ftf_presel2j25XX2j25b85_L14J15p0ETA25'],
37 2024: ['HLT_2j45_0eta290_020jvt_bgn260_2j45_pf_ftf_presel2j25XX2j25bgtwo85_L14jJ40p0ETA25'],
38}
39
40# Example cuts used for event selection algorithm test
41exampleSelectionCuts = {
42 'SUBcommon': """JET_N_BTAG >= 2
43JET_N 25000 >= 4
44MET >= 20000
45SAVE
46""",
47 'ejets': """IMPORT SUBcommon
48EL_N 5000 == 1
49MU_N 3000 == 0
50MWT < 170000
51MET+MWT > 40000
52SAVE
53""",
54 'mujets': """IMPORT SUBcommon
55EL_N 5000 == 0
56MU_N medium 25000 > 0
57SAVE
58"""
59}
60
61electronMinPt = 10*GeV
62photonMinPt = 10*GeV
63
64
65def makeTestSequenceBlocks (dataType, algSeq, isPhyslite,
66 geometry=None, autoconfigFromFlags=None, noSystematics=None,
67 onlyNominalOR=False, forceEGammaFullSimConfig=False,
68 returnConfigSeq=False,
69 bleedingEdge=False # Enabled for CI tests running on new derivations from derivation CI output
70 ) :
71
72 largeRJets = True
73
74 if autoconfigFromFlags is not None:
75 if geometry is None:
76 geometry = autoconfigFromFlags.GeoModel.Run
77
78 configSeq = ConfigSequence ()
79
80 outputContainers = {'mu_' : 'OutMuons',
81 'el_' : 'OutElectrons',
82 'ph_' : 'OutPhotons',
83 'tau_': 'OutTauJets',
84 'jet_': 'OutJets',
85 'met_': 'AnaMET',
86 '' : 'EventInfo'}
87 outputContainersForMC = {'truth_mu_' : 'OutTruthMuons',
88 'truth_el_' : 'OutTruthElectrons',
89 'truth_ph_' : 'OutTruthPhotons',
90 'truth_tau_': 'OutTruthTaus',
91 'truth_jet_': 'OutTruthJets',
92 'truth_met_': 'TruthMET'}
93
94 # create factory object to build block configurations
95 from AnalysisAlgorithmsConfig.ConfigFactory import ConfigFactory
96 config = ConfigFactory()
97
98 configSeq += config.makeConfig('CommonServices')
99 configSeq.setOptionValue('.systematicsHistogram', 'systematicsList')
100 configSeq.setOptionValue('.enableExpertMode', True)
101
102 configSeq += config.makeConfig('PileupReweighting')
103
104 # Skip events with no primary vertex,
105 # and perform loose jet cleaning
106 configSeq += config.makeConfig ('EventCleaning')
107 configSeq.setOptionValue ('.runEventCleaning', True)
108
109 # disabling comparisons for triggers, because the config blocks do a
110 # lot more than the sequences. Also disabling for Run 3+4, as there is no SF yet
111 if geometry is LHCPeriod.Run2:
112 # Include, and then set up the trigger analysis sequence:
113 configSeq += config.makeConfig( 'Trigger' )
114 configSeq.setOptionValue ('.triggerChainsPerYear', triggerChainsPerYear )
115 configSeq.setOptionValue ('.noFilter', True )
116 configSeq.setOptionValue ('.electrons', 'AnaElectrons.loose' )
117 configSeq.setOptionValue ('.photons', 'AnaPhotons.tight' )
118 configSeq.setOptionValue ('.muons', 'AnaMuons.medium' )
119 configSeq.setOptionValue ('.taus', 'AnaTauJets.tight' )
120 configSeq.setOptionValue ('.electronID', 'Tight' )
121 configSeq.setOptionValue ('.electronIsol', 'Tight_VarRad')
122 configSeq.setOptionValue ('.photonIsol', 'TightCaloOnly')
123 configSeq.setOptionValue ('.muonID', 'Tight')
124 configSeq.setOptionValue ('.triggerMatchingChainsPerYear', triggerMatchingChainsPerYear)
125
126 # Include, and then set up the jet analysis algorithm sequence:
127 configSeq += config.makeConfig( 'Jets',
128 containerName='AnaJets',
129 jetCollection='AntiKt4EMPFlowJets')
130 configSeq.setOptionValue ('.runJvtUpdate', False )
131 configSeq.setOptionValue ('.runNNJvtUpdate', True )
132 configSeq.setOptionValue ('.recalibratePhyslite', False)
133
134 configSeq += config.makeConfig( 'Jets.JVT' )
135 configSeq.setOptionValue ('.containerName', 'AnaJets')
136
137 configSeq += config.makeConfig( 'Jets.FTagTriggerMatching' )
138 configSeq.setOptionValue('.containerName', 'AnaJets')
139 configSeq.setOptionValue('.triggerChainsPerYear', bjetTriggerChainsPerYear)
140
141 # disabling flavor tagging for Run 4, as the configuration just
142 # refuses to work on that
143 if geometry is not LHCPeriod.Run4:
144
145 btagger = "GN2v01"
146 btagWP = "FixedCutBEff_65"
147 configSeq += config.makeConfig( 'Jets.FlavourTagging' )
148 configSeq.setOptionValue ('.containerName', 'AnaJets')
149 configSeq.setOptionValue ('.selectionName', 'ftag')
150 configSeq.setOptionValue ('.btagger', btagger)
151 configSeq.setOptionValue ('.btagWP', btagWP)
152 configSeq.setOptionValue ('.saveScores', 'All')
153
154 configSeq += config.makeConfig( 'Jets.FlavourTaggingEventSF' )
155 configSeq.setOptionValue ('.containerName', 'AnaJets.baselineJvt')
156 configSeq.setOptionValue ('.btagger', btagger)
157
158 if largeRJets :
159 configSeq += config.makeConfig( 'Jets',
160 containerName='AnaLargeRJets',
161 jetCollection='AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets' )
162 outputContainers['larger_jet_'] = 'OutLargeRJets'
163 configSeq.setOptionValue ('.recalibratePhyslite', False)
164
165
166 # Include, and then set up the electron analysis algorithm sequence:
167 likelihood = True
168 configSeq += config.makeConfig ('Electrons')
169 configSeq.setOptionValue ('.containerName', 'AnaElectrons')
170 configSeq.setOptionValue ('.decorateTruth', True)
171 configSeq.setOptionValue ('.decorateCaloClusterEta', True)
172 configSeq.setOptionValue ('.writeTrackD0Z0', True)
173 configSeq.setOptionValue ('.forceFullSimConfigForIso', forceEGammaFullSimConfig)
174 configSeq.setOptionValue ('.recalibratePhyslite', False)
175 configSeq.setOptionValue ('.minPt', electronMinPt)
176 configSeq += config.makeConfig ('Electrons.WorkingPoint')
177 configSeq.setOptionValue ('.containerName', 'AnaElectrons')
178 configSeq.setOptionValue ('.selectionName', 'loose')
179 configSeq.setOptionValue ('.forceFullSimConfig', forceEGammaFullSimConfig)
180 if likelihood:
181 configSeq.setOptionValue ('.identificationWP', 'LooseBLayerLH')
182 else:
183 configSeq.setOptionValue ('.identificationWP', 'LooseDNN')
184 configSeq.setOptionValue ('.isolationWP', 'Tight_VarRad')
185 configSeq.setOptionValue ('.chargeIDSelectionRun2', True)
186 configSeq.setOptionValue ('.addChargeMisIDSF', geometry is LHCPeriod.Run2)
187 configSeq += config.makeConfig('Electrons.TriggerSF')
188 configSeq.setOptionValue('.containerName', 'AnaElectrons')
189 configSeq.setOptionValue('.electronID', 'LooseBLayerLH')
190 configSeq.setOptionValue('.electronIsol', 'Tight_VarRad')
191 configSeq.setOptionValue('.triggerChainsPerYear', triggerChainsPerYear)
192
193 configSeq += config.makeConfig ('Electrons.IFFClassification')
194 configSeq.setOptionValue ('.containerName', 'AnaElectrons')
195 configSeq += config.makeConfig ('Electrons.MCTCClassification')
196 configSeq.setOptionValue ('.containerName', 'AnaElectrons')
197 configSeq.setOptionValue ('.prefix', 'truth_')
198
199 configSeq += config.makeConfig ('Electrons.PtEtaSelection')
200 configSeq.setOptionValue ('.containerName', 'AnaElectrons')
201 configSeq.setOptionValue ('.minPt', electronMinPt)
202
203
204 # Include, and then set up the photon analysis algorithm sequence:
205 configSeq += config.makeConfig ('Photons')
206 configSeq.setOptionValue ('.containerName', 'AnaPhotons')
207 configSeq.setOptionValue ('.decorateTruth', True)
208 configSeq.setOptionValue ('.forceFullSimConfigForIso', forceEGammaFullSimConfig)
209 configSeq.setOptionValue ('.recomputeIsEM', False)
210 configSeq.setOptionValue ('.recalibratePhyslite', False)
211 configSeq += config.makeConfig ('Photons.WorkingPoint')
212 configSeq.setOptionValue ('.containerName', 'AnaPhotons')
213 configSeq.setOptionValue ('.selectionName', 'tight')
214 configSeq.setOptionValue ('.forceFullSimConfigForID', forceEGammaFullSimConfig)
215 configSeq.setOptionValue ('.forceFullSimConfigForIso', forceEGammaFullSimConfig)
216 configSeq.setOptionValue ('.qualityWP', 'Tight')
217 configSeq.setOptionValue ('.isolationWP', 'FixedCutTight')
218 configSeq.setOptionValue ('.recomputeIsEM', False)
219
220 configSeq += config.makeConfig ('Photons.PtEtaSelection')
221 configSeq.setOptionValue ('.containerName', 'AnaPhotons')
222 configSeq.setOptionValue ('.minPt', photonMinPt)
223
224
225 # set up the muon analysis algorithm sequence:
226 configSeq += config.makeConfig ('Muons')
227 configSeq.setOptionValue ('.containerName', 'AnaMuons')
228 configSeq.setOptionValue ('.decorateTruth', True)
229 configSeq.setOptionValue ('.writeTrackD0Z0', True)
230 configSeq.setOptionValue ('.recalibratePhyslite', False)
231
232 configSeq += config.makeConfig ('Muons.WorkingPoint')
233 configSeq.setOptionValue ('.containerName', 'AnaMuons')
234 configSeq.setOptionValue ('.selectionName', 'medium')
235 configSeq.setOptionValue ('.quality', 'Medium')
236 configSeq.setOptionValue ('.isolation', 'Loose_VarRad')
237
238 configSeq += config.makeConfig ('Muons.IFFClassification')
239 configSeq.setOptionValue ('.containerName', 'AnaMuons')
240 configSeq += config.makeConfig ('Muons.MCTCClassification')
241 configSeq.setOptionValue ('.containerName', 'AnaMuons')
242 configSeq.setOptionValue ('.prefix', 'truth_')
243
244
245 # TODO: MCP should restore this when the recommendations for Tight WP exist in R23
246 # configSeq += config.makeConfig ('Muons.Selection', 'AnaMuons.tight')
247 # configSeq.setOptionValue ('.quality', 'Tight')
248 # configSeq.setOptionValue ('.isolation', 'Loose_VarRad')
249
250 # Include, and then set up the tau analysis algorithm sequence:
251 configSeq += config.makeConfig ('TauJets')
252 configSeq.setOptionValue ('.containerName', 'AnaTauJets')
253 configSeq.setOptionValue ('.decorateTruth', True)
254 configSeq += config.makeConfig ('TauJets.WorkingPoint')
255 configSeq.setOptionValue ('.containerName', 'AnaTauJets')
256 configSeq.setOptionValue ('.selectionName', 'tight')
257 configSeq.setOptionValue ('.quality', 'Tight')
258
259 configSeq += config.makeConfig('TauJets.TriggerSF')
260 configSeq.setOptionValue('.containerName', 'AnaTauJets')
261 configSeq.setOptionValue('.tauID', 'Tight')
262 configSeq.setOptionValue('.triggerChainsPerYear', tauTriggerChainsSF)
263
264 configSeq += config.makeConfig ('TauJets.MCTCClassification')
265 configSeq.setOptionValue ('.containerName', 'AnaTauJets')
266 configSeq.setOptionValue ('.prefix', 'truth_')
267
268
269 # Add systematic object links
270 configSeq += config.makeConfig('SystObjectLink')
271 configSeq.setOptionValue ('.containerName', 'AnaJets')
272 if largeRJets:
273 configSeq += config.makeConfig('SystObjectLink')
274 configSeq.setOptionValue ('.containerName', 'AnaLargeRJets')
275 configSeq += config.makeConfig('SystObjectLink')
276 configSeq.setOptionValue ('.containerName', 'AnaElectrons')
277 configSeq += config.makeConfig('SystObjectLink')
278 configSeq.setOptionValue ('.containerName', 'AnaPhotons')
279 configSeq += config.makeConfig('SystObjectLink')
280 configSeq.setOptionValue ('.containerName', 'AnaMuons')
281 configSeq += config.makeConfig('SystObjectLink')
282 configSeq.setOptionValue ('.containerName', 'AnaTauJets')
283
284
285 # Particle-level objects
286 configSeq += config.makeConfig ('PL_Electrons')
287 configSeq.setOptionValue ('.containerName', 'TruthElectrons')
288 configSeq += config.makeConfig ('PL_Electrons.MCTCClassification')
289 configSeq.setOptionValue ('.containerName', 'TruthElectrons')
290 configSeq.setOptionValue ('.prefix', '')
291 configSeq += config.makeConfig ('PL_Electrons.PtEtaSelection')
292 configSeq.setOptionValue ('.containerName', 'TruthElectrons')
293 configSeq.setOptionValue ('.skipOnData', True)
294 configSeq.setOptionValue ('.useDressedProperties', True)
295 configSeq.setOptionValue ('.minPt', 20e3)
296
297 configSeq += config.makeConfig ('PL_Muons')
298 configSeq.setOptionValue ('.containerName', 'TruthMuons')
299 configSeq += config.makeConfig ('PL_Muons.MCTCClassification')
300 configSeq.setOptionValue ('.containerName', 'TruthMuons')
301 configSeq.setOptionValue ('.prefix', '')
302 configSeq += config.makeConfig ('PL_Muons.PtEtaSelection')
303 configSeq.setOptionValue ('.containerName', 'TruthMuons')
304 configSeq.setOptionValue ('.skipOnData', True)
305 configSeq.setOptionValue ('.useDressedProperties', True)
306 configSeq.setOptionValue ('.minPt', 20e3)
307
308 configSeq += config.makeConfig ('PL_Neutrinos')
309 configSeq.setOptionValue ('.skipOnData', True)
310
311 configSeq += config.makeConfig ('PL_Jets')
312 configSeq.setOptionValue ('.containerName', 'AntiKt4TruthDressedWZJets')
313 configSeq += config.makeConfig ('PL_Jets.PtEtaSelection')
314 configSeq.setOptionValue ('.containerName', 'AntiKt4TruthDressedWZJets')
315 configSeq.setOptionValue ('.skipOnData', True)
316 configSeq.setOptionValue ('.minPt', 20e3)
317
318 configSeq += config.makeConfig ('PL_Taus')
319 configSeq.setOptionValue ('.containerName', 'TruthTaus')
320 configSeq += config.makeConfig ('PL_Taus.MCTCClassification')
321 configSeq.setOptionValue ('.containerName', 'TruthTaus')
322 configSeq.setOptionValue ('.prefix', '')
323 configSeq += config.makeConfig ('PL_Taus.PtEtaSelection')
324 configSeq.setOptionValue ('.containerName', 'TruthTaus')
325 configSeq.setOptionValue ('.skipOnData', True)
326 configSeq.setOptionValue ('.minPt', 20e3)
327
328 configSeq += config.makeConfig ('PL_Photons')
329 configSeq.setOptionValue ('.containerName', 'TruthPhotons')
330 configSeq += config.makeConfig ('PL_Photons.PtEtaSelection')
331 configSeq.setOptionValue ('.containerName', 'TruthPhotons')
332 configSeq.setOptionValue ('.skipOnData', True)
333 configSeq.setOptionValue ('.minPt', 20e3)
334
335 configSeq += config.makeConfig ('PL_MissingET')
336 configSeq.setOptionValue ('.skipOnData', True)
337
338 configSeq += config.makeConfig ('PL_OverlapRemoval')
339 configSeq.setOptionValue ('.skipOnData', True)
340 configSeq.setOptionValue ('.electrons', 'TruthElectrons')
341 configSeq.setOptionValue ('.muons', 'TruthMuons')
342 configSeq.setOptionValue ('.photons', 'TruthPhotons')
343 configSeq.setOptionValue ('.jets', 'AntiKt4TruthDressedWZJets')
344 configSeq.setOptionValue ('.useRapidityForDeltaR', False)
345
346
347 if dataType is not DataType.Data :
348 # Include, and then set up the generator analysis sequence:
349 configSeq += config.makeConfig( 'GeneratorLevelAnalysis')
350
351
352 # Include, and then set up the met analysis algorithm config:
353 configSeq += config.makeConfig ('MissingET')
354 configSeq.setOptionValue ('.containerName', 'AnaMET')
355 configSeq.setOptionValue ('.jets', 'AnaJets')
356 configSeq.setOptionValue ('.taus', 'AnaTauJets.tight')
357 configSeq.setOptionValue ('.electrons', 'AnaElectrons.loose')
358 configSeq.setOptionValue ('.photons', 'AnaPhotons.tight')
359 # Note that the configuration for the muons is not what you'd
360 # normally do. This is specifically here because this is a unit
361 # test and I wanted to make sure that selection expressions work.
362 # For an actual analysis that would just be `AnaMuons.medium`, but
363 # since `tight` is a strict subset of `medium` it doesn't matter
364 # if we do an "or" of the two.
365 # TODO: MCP should restore this when the recommendations for Tight WP exist in R23
366 # configSeq.setOptionValue ('.muons', 'AnaMuons.medium||tight')
367 configSeq.setOptionValue ('.muons', 'AnaMuons.medium')
368
369
370 # Include, and then set up the overlap analysis algorithm config:
371 configSeq += config.makeConfig( 'OverlapRemoval' )
372 configSeq.setOptionValue ('.electrons', 'AnaElectrons.loose')
373 configSeq.setOptionValue ('.photons', 'AnaPhotons.tight')
374 # TODO: MCP should restore this when the recommendations for Tight WP exist in R23
375 # configSeq.setOptionValue ('.muons', 'AnaMuons.medium||tight')
376 configSeq.setOptionValue ('.muons', 'AnaMuons.medium')
377 configSeq.setOptionValue ('.jets', 'AnaJets.baselineJvt')
378 configSeq.setOptionValue ('.taus', 'AnaTauJets.tight')
379 configSeq.setOptionValue ('.inputLabel', 'preselectOR')
380 configSeq.setOptionValue ('.outputLabel', 'passesOR' )
381 configSeq.setOptionValue ('.nominalOnly', onlyNominalOR )
382
383
384 # ObjectCutFlow blocks
385 configSeq += config.makeConfig ('ObjectCutFlow')
386 configSeq.setOptionValue ('.containerName', 'AnaJets')
387 configSeq.setOptionValue ('.selectionName', 'jvt')
388 configSeq += config.makeConfig ('ObjectCutFlow')
389 configSeq.setOptionValue ('.containerName', 'AnaElectrons')
390 configSeq.setOptionValue ('.selectionName', 'loose')
391 configSeq += config.makeConfig ('ObjectCutFlow')
392 configSeq.setOptionValue ('.containerName', 'AnaPhotons')
393 configSeq.setOptionValue ('.selectionName', 'tight')
394 configSeq += config.makeConfig ('ObjectCutFlow')
395 configSeq.setOptionValue ('.containerName', 'AnaMuons')
396 configSeq.setOptionValue ('.selectionName', 'medium')
397 configSeq += config.makeConfig ('ObjectCutFlow')
398 configSeq.setOptionValue ('.containerName', 'AnaTauJets')
399 configSeq.setOptionValue ('.selectionName', 'tight')
400
401 # Include and set up a basic run of the event selection algorithm config:
402 if geometry is not LHCPeriod.Run4:
403 # configSeq += config.makeConfig( 'EventSelection', None )
404 # configSeq.setOptionValue ('.electrons', 'AnaElectrons.loose')
405 # configSeq.setOptionValue ('.muons', 'AnaMuons.medium')
406 # configSeq.setOptionValue ('.jets', 'AnaJets')
407 # configSeq.setOptionValue ('.met', 'AnaMET')
408 # configSeq.setOptionValue ('.selectionCutsDict', exampleSelectionCuts)
409 from EventSelectionAlgorithms.EventSelectionConfig import makeMultipleEventSelectionConfigs
410 makeMultipleEventSelectionConfigs(configSeq, electrons = 'AnaElectrons.loose', muons = 'AnaMuons.medium', jets = 'AnaJets.baselineJvt',
411 met = 'AnaMET', btagDecoration = 'ftag_select_ftag',
412 selectionCutsDict = exampleSelectionCuts, noFilter = True,
413 cutFlowHistograms = True)
414
415 configSeq += config.makeConfig ('Bootstraps')
416 configSeq.setOptionValue ('.nReplicas', 2000 )
417 configSeq.setOptionValue ('.skipOnMC', False)
418
419 # per-event lepton SF
420 configSeq += config.makeConfig ('LeptonSF')
421 if geometry is not LHCPeriod.Run2:
422 configSeq.setOptionValue ('.electrons', 'AnaElectrons.loose')
423 configSeq.setOptionValue ('.muons', 'AnaMuons.medium')
424 configSeq.setOptionValue ('.photons', 'AnaPhotons.tight')
425 configSeq.setOptionValue ('.lepton_postfix', 'nominal')
426
427 # Thinning blocks
428 configSeq += config.makeConfig ('Thinning')
429 configSeq.setOptionValue ('.containerName', 'AnaElectrons')
430 configSeq.setOptionValue ('.selectionName', 'loose')
431 configSeq.setOptionValue ('.outputName', 'OutElectrons')
432 configSeq += config.makeConfig ('Thinning')
433 configSeq.setOptionValue ('.containerName', 'AnaPhotons')
434 configSeq.setOptionValue ('.selectionName', 'tight')
435 configSeq.setOptionValue ('.outputName', 'OutPhotons')
436 configSeq += config.makeConfig ('Thinning')
437 configSeq.setOptionValue ('.containerName', 'AnaMuons')
438 configSeq.setOptionValue ('.selectionName', 'medium')
439 configSeq.setOptionValue ('.outputName', 'OutMuons')
440 configSeq += config.makeConfig ('Thinning')
441 configSeq.setOptionValue ('.containerName', 'AnaTauJets')
442 configSeq.setOptionValue ('.selectionName', 'tight')
443 configSeq.setOptionValue ('.outputName', 'OutTauJets')
444 configSeq += config.makeConfig ('Thinning')
445 configSeq.setOptionValue ('.containerName', 'AnaJets')
446 configSeq.setOptionValue ('.outputName', 'OutJets')
447 if largeRJets :
448 configSeq += config.makeConfig ('Thinning')
449 configSeq.setOptionValue ('.containerName', 'AnaLargeRJets')
450 configSeq.setOptionValue ('.outputName', 'OutLargeRJets')
451
452 configSeq += config.makeConfig ('Thinning')
453 configSeq.setOptionValue ('.containerName', 'TruthElectrons')
454 configSeq.setOptionValue ('.skipOnData', True)
455 configSeq.setOptionValue ('.outputName', 'OutTruthElectrons')
456 configSeq += config.makeConfig ('Thinning')
457 configSeq.setOptionValue ('.containerName', 'TruthPhotons')
458 configSeq.setOptionValue ('.skipOnData', True)
459 configSeq.setOptionValue ('.outputName', 'OutTruthPhotons')
460 configSeq += config.makeConfig ('Thinning')
461 configSeq.setOptionValue ('.containerName', 'TruthMuons')
462 configSeq.setOptionValue ('.skipOnData', True)
463 configSeq.setOptionValue ('.outputName', 'OutTruthMuons')
464 configSeq += config.makeConfig ('Thinning')
465 configSeq.setOptionValue ('.containerName', 'TruthTaus')
466 configSeq.setOptionValue ('.skipOnData', True)
467 configSeq.setOptionValue ('.outputName', 'OutTruthTaus')
468 configSeq += config.makeConfig ('Thinning')
469 configSeq.setOptionValue ('.containerName', 'AntiKt4TruthDressedWZJets')
470 configSeq.setOptionValue ('.outputName', 'OutTruthJets')
471 configSeq.setOptionValue ('.skipOnData', True)
472
473 configSeq += config.makeConfig ('Output')
474 configSeq.setOptionValue ('.treeName', 'analysis')
475 configSeq.setOptionValue ('.vars', [
476 'EventInfo.actualInteractionsPerCrossing -> actualMuScaled',
477 ])
478 configSeq.setOptionValue ('.metVars', [
479 'AnaMET_%SYS%.met -> met_%SYS%',
480 ])
481 configSeq.setOptionValue ('.truthMetVars', [
482 'TruthMET_NOSYS.met -> truth_met',
483 ])
484 configSeq.setOptionValue ('.containers', outputContainers)
485 configSeq.setOptionValue ('.containersOnlyForMC', outputContainersForMC)
486 configSeq.setOptionValue ('.commands', [
487 'disable actualInteractionsPerCrossing',
488 ])
489
490 # save the tool configuration to a txt file
491 configSeq += config.makeConfig ('PrintConfiguration')
492
493 # return configSeq for unit test
494 if returnConfigSeq:
495 return configSeq
496
497 configAccumulator = ConfigAccumulator (algSeq=algSeq, flags=autoconfigFromFlags, noSystematics=noSystematics)
498 configSeq.fullConfigure (configAccumulator)
499
500 # order can change during fullConfigure
501 configSeq.printOptions()
502
503 from AnaAlgorithm.DualUseConfig import isAthena, useComponentAccumulator
504 if isAthena and useComponentAccumulator:
505 return configAccumulator.CA
506 else:
507 return None
508
509
510
511def printSequenceAlgs (sequence) :
512 """print the algorithms in the sequence without the sequence structure
513
514 This is mostly meant for easy comparison of different sequences
515 during configuration, particularly the sequences resulting from
516 the old sequence configuration and the new block configuration.
517 Those have different sequence structures in the output, but the
518 algorithms should essentially be configured the same way."""
519 if isinstance (sequence, AlgSequence) :
520 for alg in sequence :
521 printSequenceAlgs (alg)
522 else :
523 # assume this is an algorithm then
524 print (sequence)
525
526
527def makeSequence (dataType, noSystematics,
528 yamlPath=None,
529 isPhyslite = False, geometry = None,
530 autoconfigFromFlags = None, onlyNominalOR = False,
531 forceEGammaFullSimConfig = False,
532 bleedingEdge = False) :
533
534 algSeq = AlgSequence('AnalysisSequence')
535
536 ca = None
537 if not yamlPath:
538 ca = makeTestSequenceBlocks (dataType, algSeq,
539 isPhyslite=isPhyslite,
540 geometry=geometry, onlyNominalOR=onlyNominalOR,
541 autoconfigFromFlags=autoconfigFromFlags,
542 noSystematics=noSystematics,
543 forceEGammaFullSimConfig=forceEGammaFullSimConfig,
544 bleedingEdge=bleedingEdge)
545 else:
546 from AnalysisAlgorithmsConfig.ConfigText import makeSequence as makeSequenceText
547 ca = makeSequenceText(yamlPath, algSeq=algSeq, flags=autoconfigFromFlags,
548 noSystematics=noSystematics)
549
550 if ca is not None:
551 return ca
552 else:
553 return algSeq
python.FullCPAlgorithmsTest.makeTestSequenceBlocks
makeTestSequenceBlocks(dataType, algSeq, isPhyslite, geometry=None, autoconfigFromFlags=None, noSystematics=None, onlyNominalOR=False, forceEGammaFullSimConfig=False, returnConfigSeq=False, bleedingEdge=False # Enabled for CI tests running on new derivations from derivation CI output)
Definition FullCPAlgorithmsTest.py:70
python.FullCPAlgorithmsTest.makeSequence
makeSequence(dataType, noSystematics, yamlPath=None, isPhyslite=False, geometry=None, autoconfigFromFlags=None, onlyNominalOR=False, forceEGammaFullSimConfig=False, bleedingEdge=False)
Definition FullCPAlgorithmsTest.py:532
Generated on for ATLAS Offline Software by doxygen 1.14.0