DerivationTrackIsoConfig Namespace Reference

Functions
def	DerivationTrackIsoCfg (flags, **jwarg)
def	iso_vars (WP="Nonprompt_All_MaxWeight")

Variables
string	__doc__
	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 11 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 109 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
 

Variable Documentation

__doc__

string DerivationTrackIsoConfig.__doc__

private

Initial value:

=  """
          Instantiate more custom track isolation in derivation
          """

Definition at line 3 of file DerivationTrackIsoConfig.py.

acc

DerivationTrackIsoConfig.acc = MainServicesCfg(flags)

Definition at line 136 of file DerivationTrackIsoConfig.py.

doWriteAOD

DerivationTrackIsoConfig.doWriteAOD

Definition at line 130 of file DerivationTrackIsoConfig.py.

doWriteESD

DerivationTrackIsoConfig.doWriteESD

Definition at line 129 of file DerivationTrackIsoConfig.py.

Files

DerivationTrackIsoConfig.Files

Definition at line 128 of file DerivationTrackIsoConfig.py.

flags

DerivationTrackIsoConfig.flags = initConfigFlags()

Definition at line 127 of file DerivationTrackIsoConfig.py.

mlog

DerivationTrackIsoConfig.mlog = logging.getLogger("derivationTrackIsolationConfigTest")

Definition at line 133 of file DerivationTrackIsoConfig.py.

name

string DerivationTrackIsoConfig.name = "IsolationBuilderNonprompt_All_MaxWeight"+par

Definition at line 141 of file DerivationTrackIsoConfig.py.

nestLevel

DerivationTrackIsoConfig.nestLevel

Definition at line 144 of file DerivationTrackIsoConfig.py.

printDefaults

DerivationTrackIsoConfig.printDefaults

Definition at line 145 of file DerivationTrackIsoConfig.py.