ATLAS Offline Software
ParticleLevelMuonsConfig.py
Go to the documentation of this file.
1 # Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
2 
3 from AnalysisAlgorithmsConfig.ConfigBlock import ConfigBlock
4 
5 
6 class ParticleLevelMuonsBlock(ConfigBlock):
7  """ConfigBlock for particle-level truth muons"""
8 
9  def __init__(self):
10  super(ParticleLevelMuonsBlock, self).__init__()
11  self.addOption('containerName', 'TruthMuons', type=str,
12  info='the name of the input truth muons container')
13  self.addOption('selectionName', '', type=str,
14  info='the name of the selection to create. The default is "",'
15  ' which applies the selection to all truth muons.')
16  self.addOption('isolated', True, type=bool,
17  info='select only truth muons that are isolated.')
18  self.addOption('notFromTau', True, type=bool,
19  info='select only truth muons that did not orginate '
20  'from a tau decay.')
21  # Always skip on data
22  self.setOptionValue('skipOnData', True)
23 
24  def makeAlgs(self, config):
25  config.setSourceName (self.containerName, self.containerName)
26 
27  # decorate the charge so we can save it later
28  alg = config.createAlgorithm('CP::ParticleLevelChargeDecoratorAlg', 'ParticleLevelChargeDecoratorMuons' + self.selectionName)
29  alg.particles = self.containerName
30 
31  # check for prompt isolation and possible origin from tau decays
32  alg = config.createAlgorithm('CP::ParticleLevelIsolationAlg', 'ParticleLevelIsolationMuons' + self.selectionName)
33  alg.particles = self.containerName
34  alg.isolation = 'isIsolated' + self.selectionName if self.isolated else 'isIsolatedButNotRequired' + self.selectionName
35  alg.notTauOrigin = 'notFromTau' + self.selectionName if self.notFromTau else 'notFromTauButNotRequired' + self.selectionName
36  alg.checkType = 'IsoMuon'
37 
38  if self.isolated:
39  config.addSelection (self.containerName, self.selectionName, alg.isolation+',as_char')
40  if self.notFromTau:
41  config.addSelection (self.containerName, self.selectionName, alg.notTauOrigin+',as_char')
42 
43  # output branches to be scheduled only once
44  if ParticleLevelMuonsBlock.get_instance_count() == 1:
45  outputVars = [
46  ['pt_dressed', 'pt'],
47  ['eta_dressed', 'eta'],
48  ['phi_dressed', 'phi'],
49  ['e_dressed', 'e'],
50  ['charge', 'charge'],
51  ['classifierParticleType', 'type'],
52  ['classifierParticleOrigin', 'origin'],
53  ]
54  for decoration, branch in outputVars:
55  config.addOutputVar (self.containerName, decoration, branch, noSys=True)
56 
57 
ParticleLevelMuonsConfig.ParticleLevelMuonsBlock
Definition: ParticleLevelMuonsConfig.py:6
ParticleLevelMuonsConfig.ParticleLevelMuonsBlock.makeAlgs
def makeAlgs(self, config)
Definition: ParticleLevelMuonsConfig.py:24
ParticleLevelMuonsConfig.ParticleLevelMuonsBlock.__init__
def __init__(self)
Definition: ParticleLevelMuonsConfig.py:9