ParticleJetToolsConfig Namespace Reference

Functions
def	isAnalysisRelease ()
def	getMCTruthClassifier ()
def	getCopyTruthLabelParticles (truthtype)
def	getCopyTruthJetParticles (modspec, cflags)
def	getJetQuarkLabel ()
def	getJetConeLabeling ()
def	_getCommonLabelNames (prefix)
def	getJetDeltaRFlavorLabelTool (name='jetdrlabeler', jet_pt_min=5000)
def	getJetDeltaRLabelTool (jetdef, modspec)
def	getJetGhostFlavorLabelTool (name="jetghostlabeler")
def	getJetGhostLabelTool (jetdef, modspec)
def	getJetTruthLabelTool (jetdef, modspec)
def	getJetPileupLabelTool (jetdef, modspec)

Variables
	jrtlog = Logging.logging.getLogger('ParticleJetToolsConfig')
dictionary	truthpartoptions

Function Documentation

_getCommonLabelNames()

def ParticleJetToolsConfig._getCommonLabelNames (   prefix )

private

Internal unlity to name labels

Returns a dictionary to configure labeling tools. Takes one
argument which is prefixed to each label.

Definition at line 103 of file ParticleJetToolsConfig.py.

def _getCommonLabelNames(prefix):
    """Internal unlity to name labels
 
    Returns a dictionary to configure labeling tools. Takes one
    argument which is prefixed to each label.
    """
    return dict(
        LabelName=f"{prefix}TruthLabelID",
        DoubleLabelName=f"{prefix}ExtendedTruthLabelID",
        LabelPtName=f"{prefix}TruthLabelPt",
        LabelLxyName=f"{prefix}TruthLabelLxy",
        LabelDRName=f"{prefix}TruthLabelDR",
        LabelPdgIdName=f"{prefix}TruthLabelPdgId",
        LabelPositionDPhiName=f"{prefix}TruthLabelPositionDPhi",
        LabelPositionDEtaName=f"{prefix}TruthLabelPositionDEta",
        LabelBarcodeName=f"{prefix}TruthLabelBarcode",
        ChildLxyName=f"{prefix}TruthLabelChildLxy",
        ChildPtName=f"{prefix}TruthLabelChildPt",
        ChildPdgIdName=f"{prefix}TruthLabelChildPdgId",
        ChildPositionDPhiName=f"{prefix}TruthLabelChildPositionDPhi",
        ChildPositionDEtaName=f"{prefix}TruthLabelChildPositionDEta",
    )
 
 

getCopyTruthJetParticles()

def ParticleJetToolsConfig.getCopyTruthJetParticles	(		modspec,
			cflags
	)

Definition at line 63 of file ParticleJetToolsConfig.py.

def getCopyTruthJetParticles(modspec, cflags):
    truthclassif = getMCTruthClassifier()
 
    truthpartcopy = CompFactory.CopyTruthJetParticles(
        "truthpartcopy"+modspec,
        OutputName="JetInputTruthParticles"+modspec,
        MCTruthClassifier=truthclassif)
    if modspec=="NoWZ":
        truthpartcopy.IncludePromptLeptons=False
        truthpartcopy.IncludePromptPhotons=False
        truthpartcopy.IncludeMuons=True
        truthpartcopy.IncludeNeutrinos=True
    if modspec=="DressedWZ":
        truthpartcopy.IncludePromptLeptons=False
        truthpartcopy.IncludePromptPhotons=True
        truthpartcopy.IncludeMuons=True
        truthpartcopy.IncludeNeutrinos=True
        truthpartcopy.DressingDecorationName='dressedPhoton'
        
        truthpartcopy.ExtraInputs = {( 'xAOD::TruthParticleContainer' , 'StoreGateSvc+TruthParticles.dressedPhoton' )}
    if modspec=="Charged":
        truthpartcopy.ChargedParticlesOnly=True
    return truthpartcopy
 

getCopyTruthLabelParticles()

def ParticleJetToolsConfig.getCopyTruthLabelParticles

(

truthtype

)

Definition at line 43 of file ParticleJetToolsConfig.py.

def getCopyTruthLabelParticles(truthtype):
    toolProperties = {}
    if truthtype == "Partons":
        truthcategory = "Partons"
    elif truthtype in ["WBosons", "ZBosons", "HBosons", "TQuarksFinal"]:
        truthcategory = "BosonTop"
        toolProperties['ParticleType'] = truthtype
    else:
        truthcategory = "FlavourLabel"
        toolProperties['ParticleType'] = truthtype
 
    tooltype = truthpartoptions[truthcategory]["ToolType"]
    toolProperties.update( PtMin = truthpartoptions[truthcategory]["ptmin"],
                           OutputName = "TruthLabel"+truthtype)
    ctp = tooltype("truthpartcopy_"+truthtype,
                   **toolProperties
                   )
    return ctp
 
# Generates input truth particle containers for truth jets

getJetConeLabeling()

def ParticleJetToolsConfig.getJetConeLabeling

(

)

Definition at line 94 of file ParticleJetToolsConfig.py.

def getJetConeLabeling():
    jetquarklabel = getJetQuarkLabel()
    truthpartonlabel = CompFactory.Analysis.JetConeLabeling(
        "truthpartondr",
        JetTruthMatchTool = jetquarklabel
        )
    return truthpartonlabel
 
 

getJetDeltaRFlavorLabelTool()

def ParticleJetToolsConfig.getJetDeltaRFlavorLabelTool	(		name = 'jetdrlabeler',
			jet_pt_min = 5000
	)

Get the standard flavor tagging delta-R labeling tool

Uses cone matching to B, C and tau truth particles.

Definition at line 127 of file ParticleJetToolsConfig.py.

def getJetDeltaRFlavorLabelTool(name='jetdrlabeler', jet_pt_min=5000):
    """Get the standard flavor tagging delta-R labeling tool
 
    Uses cone matching to B, C and tau truth particles.
    """
    return CompFactory.ParticleJetDeltaRLabelTool(
        name,
        **_getCommonLabelNames("HadronConeExcl"),
        BLabelName = "ConeExclBHadronsFinal",
        CLabelName = "ConeExclCHadronsFinal",
        TauLabelName = "ConeExclTausFinal",
        BParticleCollection = "TruthLabelBHadronsFinal",
        CParticleCollection = "TruthLabelCHadronsFinal",
        TauParticleCollection = "TruthLabelTausFinal",
        PartPtMin = 5000.,
        DRMax = 0.3,
        MatchMode = "MinDR",
        JetPtMin = jet_pt_min,
        )
 
 

getJetDeltaRLabelTool()

def ParticleJetToolsConfig.getJetDeltaRLabelTool	(		jetdef,
			modspec
	)

returns a ParticleJetDeltaRLabelTool
Cone matching for B, C and tau truth for all but track jets.

This function is meant to be used as callback from JetRecConfig where
it is called as func(jetdef, modspec). Hence the jetdef argument even if not used in this case.

Definition at line 148 of file ParticleJetToolsConfig.py.

def getJetDeltaRLabelTool(jetdef, modspec):
    """returns a ParticleJetDeltaRLabelTool
    Cone matching for B, C and tau truth for all but track jets.
 
    This function is meant to be used as callback from JetRecConfig where
    it is called as func(jetdef, modspec). Hence the jetdef argument even if not used in this case.
    """
    jetptmin = float(modspec)
    name = "jetdrlabeler_jetpt{0}GeV".format(int(jetptmin/1000))
    return getJetDeltaRFlavorLabelTool(name, jetptmin)
 
 

getJetGhostFlavorLabelTool()

def ParticleJetToolsConfig.getJetGhostFlavorLabelTool

(

name = "jetghostlabeler"

)

Definition at line 160 of file ParticleJetToolsConfig.py.

def getJetGhostFlavorLabelTool(name="jetghostlabeler"):
    return CompFactory.ParticleJetGhostLabelTool(
        name,
        **_getCommonLabelNames('HadronGhost'),
        GhostBName = "GhostBHadronsFinal",
        GhostCName = "GhostCHadronsFinal",
        GhostTauName = "GhostTausFinal",
        PartPtMin = 5000.0
    )
 
 

getJetGhostLabelTool()

def ParticleJetToolsConfig.getJetGhostLabelTool	(		jetdef,
			modspec
	)

get ghost-based flavor tagging labeling

This is a wrapper for JetRecConfig where it's called as
func(jetdef, modspec)

Definition at line 171 of file ParticleJetToolsConfig.py.

def getJetGhostLabelTool(jetdef, modspec):
    """get ghost-based flavor tagging labeling
 
    This is a wrapper for JetRecConfig where it's called as
    func(jetdef, modspec)
    """
    return getJetGhostFlavorLabelTool()
 
 

getJetPileupLabelTool()

def ParticleJetToolsConfig.getJetPileupLabelTool	(		jetdef,
			modspec
	)

Definition at line 196 of file ParticleJetToolsConfig.py.

def getJetPileupLabelTool(jetdef, modspec):
 
    jetPileupLabelTool = CompFactory.JetPileupLabelingTool('pileuplabeler',
                                                           RecoJetContainer = jetdef.fullname(),
                                                           TruthJetContainer= "AntiKt4TruthDressedWZJets")
 
    return jetPileupLabelTool

getJetQuarkLabel()

def ParticleJetToolsConfig.getJetQuarkLabel

(

)

Definition at line 87 of file ParticleJetToolsConfig.py.

def getJetQuarkLabel():
    jetquarklabel = CompFactory.Analysis.JetQuarkLabel(
        "jetquarklabel",
        McEventCollection = "TruthEvents"
        )
    return jetquarklabel
 

getJetTruthLabelTool()

def ParticleJetToolsConfig.getJetTruthLabelTool	(		jetdef,
			modspec
	)

Definition at line 180 of file ParticleJetToolsConfig.py.

def getJetTruthLabelTool(jetdef, modspec):
 
    isTruthJet = 'Truth' in jetdef.fullname()
 
    if not isinstance(modspec, str):
        raise ValueError("JetTruthLabelingTool can only be scheduled with str as modspec")
    else:
        truthLabel = str(modspec)
 
    jetTruthLabelTool = CompFactory.JetTruthLabelingTool('truthlabeler_{0}'.format(truthLabel),
                                                         RecoJetContainer = jetdef.fullname(),
                                                         IsTruthJetCollection = isTruthJet,
                                                         TruthLabelName = truthLabel)
 
    return jetTruthLabelTool
 

getMCTruthClassifier()

def ParticleJetToolsConfig.getMCTruthClassifier

(

)

Definition at line 23 of file ParticleJetToolsConfig.py.

def getMCTruthClassifier():
    # Assume mc15 value
    truthclassif = CompFactory.MCTruthClassifier(
        "JetMCTruthClassifier"
        )
    if not isAnalysisRelease() :
        truthclassif.xAODTruthLinkVector= ""
    # Config neessary only for Athena releases
    import os
    if "AtlasProject" in os.environ.keys():
        if os.environ["AtlasProject"] in ["Athena","AthDerivation"]:
            truthclassif.ParticleCaloExtensionTool=""
    return truthclassif
 
# Generates truth particle containers for truth labeling

isAnalysisRelease()

def ParticleJetToolsConfig.isAnalysisRelease

(

)

Definition at line 19 of file ParticleJetToolsConfig.py.

    def isAnalysisRelease():
        return True
 
# Putting MCTruthClassifier here as we needn't stick jet configs in really foreign packages

Variable Documentation

jrtlog

ParticleJetToolsConfig.jrtlog = Logging.logging.getLogger('ParticleJetToolsConfig')

Definition at line 12 of file ParticleJetToolsConfig.py.

truthpartoptions

dictionary ParticleJetToolsConfig.truthpartoptions

Initial value:

=  {
    "Partons":{"ToolType":CompFactory.CopyTruthPartons,"ptmin":5000},
    "BosonTop":{"ToolType":CompFactory.CopyBosonTopLabelTruthParticles,"ptmin":100000},
    "FlavourLabel":{"ToolType":CompFactory.CopyFlavorLabelTruthParticles,"ptmin":5000},
}

Definition at line 38 of file ParticleJetToolsConfig.py.