python.BTaggingContent Namespace Reference

Functions
def	_getBtagging (jetcol)
def	_isRun4 (ConfigFlags)
def	_getVariableList (collection, aux_list)
def	_getVars (name, extra_flavours=None, flip_modes=None)
def	_getTruthVars ()
def	BTaggingExpertContent (jetcol, ConfigFlags=None)
def	BTaggingStandardContent (jetcol, ConfigFlags=None)
def	BTaggingXbbContent (jetcol, ConfigFlags=None)

Variables
list	fold_hashes = ['jetFoldHash', 'jetFoldHash_noHits']
list	JetStandardAux
list	JetExtendedAux
list	BTaggingRun3Aux = ["SV1_NGTinSvx", "SV1_masssvx",]
	flip_modes
	extra_flavours
list	BTaggingRun4Aux
list	BTaggingHighLevelAux

Detailed Description

Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration

Define sets of standard variables to save in output files.
The variable lists returned by these functions are used by the smart slimming
service to determine which variables to save in derivations.

Function Documentation

_getBtagging()

def python.BTaggingContent._getBtagging (   jetcol )

private

Convenience function for getting btagging names

Definition at line 15 of file BTaggingContent.py.

def _getBtagging(jetcol):
    """Convenience function for getting btagging names"""
    btaggingtmp = "BTagging_" + jetcol.split('Jets')[0]
    if 'BTagging' in jetcol:
         stamp = jetcol.split('BTagging')[1]
         btaggingtmp += '_'+stamp
    # deal with name mismatch between PV0TrackJets and BTagging_Track
    btagging = btaggingtmp.replace("PV0Track", "Track")
    return btagging
 

_getTruthVars()

def python.BTaggingContent._getTruthVars ( )

private

Definition at line 43 of file BTaggingContent.py.

def _getTruthVars():
    vals = ['ID', 'Pt', 'Lxy', 'DR', 'PdgId', 'Barcode']
    algs = ['HadronConeExcl', 'HadronGhost']
    base = [f'{a}TruthLabel{v}' for v in vals for a in algs + ['PartonTruthLabel'] ]
    extended = [f'{a}ExtendedTruthLabelID' for a in algs]
    return base + extended
 
# ---------------------------------------------------------------------
# Variable lists
# ---------------------------------------------------------------------
# some jet variables we always want to save

_getVariableList()

def python.BTaggingContent._getVariableList (   collection,   aux_list  )

private

Convenience function for getting variable list

Definition at line 29 of file BTaggingContent.py.

def _getVariableList(collection, aux_list):
    """Convenience function for getting variable list"""
    return [collection] + [".".join( [ collection + "Aux" ] + aux_list )]
 

_getVars()

def python.BTaggingContent._getVars (   name,   extra_flavours = None,   flip_modes = None  )

private

Convenience function for getting output variable names

Definition at line 33 of file BTaggingContent.py.

def _getVars(name, extra_flavours=None, flip_modes=None):
    """Convenience function for getting output variable names"""
    if extra_flavours is None:
        extra_flavours = []
    if flip_modes is None:
        flip_modes = [""]
    flavors = list("cub") + extra_flavours
    variants = [""] + flip_modes
    return [f'{name}{v}_p{f}' for v in variants for f in flavors]
 

_isRun4()

def python.BTaggingContent._isRun4 (   ConfigFlags )

private

Convenience function for checking if we are in Run4

Definition at line 25 of file BTaggingContent.py.

def _isRun4(ConfigFlags):
    """Convenience function for checking if we are in Run4"""
    return ConfigFlags is not None and ConfigFlags.GeoModel.Run >= LHCPeriod.Run4
 

BTaggingExpertContent()

def python.BTaggingContent.BTaggingExpertContent	(		jetcol,
			ConfigFlags = None
	)

Definition at line 153 of file BTaggingContent.py.

def BTaggingExpertContent(jetcol, ConfigFlags = None):
    btagging = _getBtagging(jetcol)
 
    # jet variables
    jetcontent = _getVariableList(jetcol, JetStandardAux + JetExtendedAux)
 
    # b-tagging variables
    isRun4 = _isRun4(ConfigFlags)
    aux = BTaggingRun4Aux if isRun4 else BTaggingRun3Aux
    aux += BTaggingHighLevelAux
    btagcontent = _getVariableList(btagging, aux)
 
    return jetcontent + btagcontent
 
 

BTaggingStandardContent()

def python.BTaggingContent.BTaggingStandardContent	(		jetcol,
			ConfigFlags = None
	)

Definition at line 168 of file BTaggingContent.py.

def BTaggingStandardContent(jetcol, ConfigFlags = None):
    btagging = _getBtagging(jetcol)
 
    # jet variables
    jetcontent = _getVariableList(jetcol, JetStandardAux)
 
    # b-tagging variables
    isRun4 = _isRun4(ConfigFlags)
    aux = BTaggingRun4Aux if isRun4 else BTaggingRun3Aux
    btagcontent = _getVariableList(btagging, aux)
 
    return jetcontent + btagcontent
 
 

BTaggingXbbContent()

def python.BTaggingContent.BTaggingXbbContent	(		jetcol,
			ConfigFlags = None
	)

Definition at line 182 of file BTaggingContent.py.

def BTaggingXbbContent(jetcol, ConfigFlags = None):
    btagging = _getBtagging(jetcol)
 
    # jet variables
    jetAllAux = JetStandardAux + JetExtendedAux
    jetcontent = _getVariableList(jetcol, jetAllAux)
 
    # b-tagging variables
    isRun4 = _isRun4(ConfigFlags)
    aux = BTaggingRun4Aux if isRun4 else BTaggingRun3Aux
    aux += BTaggingHighLevelAux
    btagcontent = _getVariableList(btagging, aux)
 
    return jetcontent + btagcontent

Variable Documentation

BTaggingHighLevelAux

list python.BTaggingContent.BTaggingHighLevelAux

Definition at line 98 of file BTaggingContent.py.

BTaggingRun3Aux

list python.BTaggingContent.BTaggingRun3Aux = ["SV1_NGTinSvx", "SV1_masssvx",]

Definition at line 78 of file BTaggingContent.py.

BTaggingRun4Aux

list python.BTaggingContent.BTaggingRun4Aux

Initial value:

=  [
    "SV1_NGTinSvx",
    "SV1_masssvx",
    "dipsrun420221008_pu",
    "dipsrun420221008_pc",
    "dipsrun420221008_pb",
    "DL1drun420221017_pu",
    "DL1drun420221017_pc",
    "DL1drun420221017_pb",
    "GN1run420221010_pu",
    "GN1run420221010_pc",
    "GN1run420221010_pb"
]

Definition at line 83 of file BTaggingContent.py.

extra_flavours

python.BTaggingContent.extra_flavours

Definition at line 80 of file BTaggingContent.py.

flip_modes

python.BTaggingContent.flip_modes

Definition at line 79 of file BTaggingContent.py.

fold_hashes

list python.BTaggingContent.fold_hashes = ['jetFoldHash', 'jetFoldHash_noHits']

Definition at line 54 of file BTaggingContent.py.

JetExtendedAux

list python.BTaggingContent.JetExtendedAux

Initial value:

=  [
    "GhostBHadronsFinalCount",
    "GhostBHadronsFinalPt",
    "GhostCHadronsFinalCount",
    "GhostCHadronsFinalPt",
    "GhostTausFinalCount",
    "GhostTausFinalPt",
    "PartonTruthLabelEnergy",
]

Definition at line 67 of file BTaggingContent.py.

JetStandardAux

list python.BTaggingContent.JetStandardAux

Initial value:

=  fold_hashes + [
    "pt",
    "eta",
    "btaggingLink",
    "GhostTrack",
    "jetRank",
    "ConeExclBHadronsFinal",
    "ConeExclCHadronsFinal",
    "PartonTruthLabelID",
    *_getTruthVars(),
]

Definition at line 55 of file BTaggingContent.py.