def addBHadronGhostAssociation(jetcollection):
if TruthD3PDFlags.GenParticleAssocLabel() != "":
addToIncludeBlocks(
jetcollection,
[JetTagD3PDKeys.JetBHadronGhostAssocBlockName() + "tmp"])
JetTagD3PDFlags.D3PDPropDict().update({
jetcollection + "_" + JetTagD3PDKeys.JetBHadronGhostAssocBlockName(
) + "_target":
TruthD3PDFlags.GenParticleAssocLabel()
})
def getJetTagPrimaryVertexD3PDObject(level=10, **kw):
if JetTagD3PDFlags.TrackAssocLabel() == "":
TrackD3PDFlags.storeVertexTrackIndexAssociation = False
JetTagPrimaryVertexD3PDObject = \
BuildxAODVertexD3PDObject(_prefix=JetTagD3PDKeys.PrimaryVertexPrefix(),
_label=JetTagD3PDKeys.PrimaryVertexGetterLabel(),
_sgkey=JetTagD3PDFlags.PrimaryVertexSGKey(),
trackTarget=JetTagD3PDFlags.TrackAssocLabel(),
trackPrefix='trk_')
return JetTagPrimaryVertexD3PDObject(level=level, **kw)
def getJetTagPrimaryVertexD3PDObject(level=10, **kw):
if JetTagD3PDFlags.TrackAssocLabel() == "":
TrackD3PDFlags.storeVertexTrackIndexAssociation = False
JetTagPrimaryVertexD3PDObject = BuildVertexD3PDObject(
_prefix=JetTagD3PDKeys.PrimaryVertexPrefix(),
_label=JetTagD3PDKeys.PrimaryVertexGetterLabel(),
_sgkey=JetTagD3PDFlags.PrimaryVertexSGKey(),
#_object_name='JetTagPrimaryVertexD3PDObject',
trackTarget=JetTagD3PDFlags.TrackAssocLabel(),
trackPrefix='trk_',
trackType='Rec::TrackParticleContainer')
return JetTagPrimaryVertexD3PDObject(level=level, **kw)
def addTrackGhostAssociation(jetcollection):
### assoc name is TrackAssoc
### using the default assoc otherwise use setupGhostAssociator to get a custom assoc
if JetTagD3PDFlags.TrackAssocLabel() != "":
JetTagD3PDFlags.D3PDPropDict().update(
{
jetcollection + "_" + JetTagD3PDKeys.JetTrackGhostAssocBlockName(
) + "_target":
JetTagD3PDFlags.TrackAssocLabel()
})
##
## Tau trigger information:
##
if triggerAvailable:
from TrigTauD3PDMaker.TrigTauD3PD import TrigTauD3PDObjects
TrigTauD3PDObjects(d3pdalg, level=1, addNavInfo=True, addConfigInfo=False)
pass
##
## Offline jet information with b-tag info:
##
from JetTagD3PDMaker.JetTagD3PDMakerFlags import JetTagD3PDFlags
from JetTagD3PDMaker.JetTagD3PDMakerFlags import JetTagD3PDKeys
JetTagD3PDFlags.Taggers = ["IP1D", "IP2D", "IP3D", "SV0", "SV1", "SV2"]
jet_includes = [
JetTagD3PDKeys.BTagWeightsBlockName(),
JetTagD3PDKeys.SVInfoPlusBlockName(),
JetTagD3PDKeys.SV0InfoPlusBlockName(), 'JetVertexFraction', 'JVtx'
]
from JetD3PDMaker.JetD3PDObject import getJetD3PDObject
if cfgKeyStore.isInInput("JetCollection", "AntiKt4TopoEMJets"):
JetD3PDObject = getJetD3PDObject()
d3pdalg += JetD3PDObject(
1,
sgkey="AntiKt4TopoEMJets",
prefix="jet_AntiKt4TopoEMJets_",
include=jet_includes,
JetTag_SoftMuonChi2InfoMuonAssoc_target="mu_staco_")
else:
TriggerD3PDStream_msg.warning("Offline jets not found in input")