Functions
def	DerivationTrackIsoCfg (flags, **jwarg)

def	iso_vars (WP="Nonprompt_All_MaxWeight")

Variables
	flags = initConfigFlags()

	Files

	doWriteESD

	doWriteAOD

	mlog = logging.getLogger("derivationTrackIsolationConfigTest")

	acc = MainServicesCfg(flags)

string	name = "IsolationBuilderNonprompt_All_MaxWeight"+par

	nestLevel

	printDefaults

Function Documentation

◆ DerivationTrackIsoCfg()

def DerivationTrackIsoConfig.DerivationTrackIsoCfg	(		flags,
		**	jwarg
	)

Definition at line 10 of file DerivationTrackIsoConfig.py.

 def DerivationTrackIsoCfg(flags,**jwarg):
  
     mlog = logging.getLogger('DerivationTrackIsoConfig')
     mlog.info('Starting track isolation for derivation')
  
     acc = ComponentAccumulator()
  
     from xAODPrimitives.xAODIso import xAODIso as isoPar
     # dR in decreasing order
     ptcone_list = [ [ isoPar.ptcone40, isoPar.ptcone30, isoPar.ptcone20 ] ]
     trkcor_list = [ [ isoPar.coreTrackPtr ] ]
  
     if 'object_types' not in jwarg:
         object_types = ("Electrons", "Photons", "Muons")
     else:
         object_types = jwarg['object_types']
  
     if 'WP' not in jwarg:
         WP = 'Nonprompt_All_MaxWeight'
     else:
         WP = jwarg['WP']
         # TrackIsolationTool need a specific track-vertex association tool
         #   if WP is not Nonprompt_All_MaxWeight
         from TrackVertexAssociationTool.TrackVertexAssociationToolConfig import isoTTVAToolCfg
         ttvaCA = acc.popToolsAndMerge(
             isoTTVAToolCfg(flags, WorkingPoint = WP))
  
     postfix = ''
     if 'postfix' in jwarg:
         postfix = jwarg['postfix']
  
     if 'ptCuts' not in jwarg:
         ptCuts = (500, 1000)
     else:
         ptCuts = jwarg['ptCuts']
  
     do_egamma = any(x in object_types for x in ("Electrons", "Photons"))
  
     mlog.info('will isolate objects '+" ".join(object_types)+' with TTVA WP = '+WP)
  
     for track_pt in ptCuts:
         for loose_cone in (True, False):
             if loose_cone and not do_egamma:
                 # Loose cone isolation variables only for electrons and photons
                 continue
  
             cone_str = "LooseCone" if loose_cone else ""
  
             # The custom variable suffix
             name = f"{WP}TTVA{cone_str}_pt{track_pt}"
  
             # Build up extra IsolationBuilder kwargs
             kwargs = {}
             if "Electrons" in object_types:
                 kwargs["ElIsoTypes"] = ptcone_list
                 kwargs["ElCorTypes"] = trkcor_list
                 kwargs["ElCorTypesExtra"] = [[]]
                 kwargs["CustomConfigurationNameEl"] = name
             if "Electrons" in object_types and "LRT" in postfix:
                 kwargs["ElectronCollectionContainerName"] = "LRTElectrons"
             if "Photons" in object_types:
                 kwargs["PhIsoTypes"] = ptcone_list
                 kwargs["PhCorTypes"] = trkcor_list
                 kwargs["PhCorTypesExtra"] = [[]]
                 kwargs["CustomConfigurationNamePh"] = name
             if "Muons" in object_types and not loose_cone:
                 kwargs["MuIsoTypes"] = ptcone_list
                 kwargs["MuCorTypes"] = trkcor_list
                 kwargs["MuCorTypesExtra"] = [[]]
                 kwargs["CustomConfigurationNameMu"] = name
             if "Muons" in object_types and not loose_cone and "LRT" in postfix:
                 kwargs["MuonCollectionContainerName"] = "MuonsLRT"
                 
             
             toolkwargs = {}
             # the TTVA tool is not the default choice
             if 'WP' in jwarg:
                 toolkwargs['TTVATool'] = ttvaCA
             # and a track selection tool
  
             from InDetConfig.InDetTrackSelectionToolConfig import (
                 isoTrackSelectionToolCfg )
             toolkwargs['TrackSelectionTool'] = acc.popToolsAndMerge(
                 isoTrackSelectionToolCfg(flags, minPt = track_pt))
  
             if loose_cone:
                 toolkwargs['CoreTrackEtaRange'] = 0.01
  
             from IsolationAlgs.IsoToolsConfig import TrackIsolationToolCfg
             kwargs['TrackIsolationTool'] = acc.popToolsAndMerge(
                 TrackIsolationToolCfg(flags,**toolkwargs))
  
             # The algorithm
             kwargs['name'] = f"IsolationBuilder{WP}{cone_str}{track_pt}{postfix}"
             acc.addEventAlgo(CompFactory.IsolationBuilder(**kwargs))
  
     return acc
  

◆ iso_vars()

def DerivationTrackIsoConfig.iso_vars ( WP = "Nonprompt_All_MaxWeight" )

Definition at line 108 of file DerivationTrackIsoConfig.py.

 def iso_vars(WP = "Nonprompt_All_MaxWeight"):
     # Get the list of isolation variables calculated by these functions
     iso_vars = []
     for track_pt in 500, 1000:
         for cone_str in "", "LooseCone":
             name = f"{WP}TTVA{cone_str}_pt{track_pt}"
             iso_vars += [ "ptconeCorrBitset_"+name,
                              "ptconecoreTrackPtrCorrection_"+name ]
             for cone_size in 20, 30, 40:
                 for var_str in "", "var":
                     iso_vars.append(f"pt{var_str}cone{cone_size}_{name}")
     return iso_vars
  