def setupTruthJets(myjetfinder, myjetdr, GroomedDicts, theseq):
if rec.doTruth():
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
sortertool = JetSorterTool('myJetSorter',
SortOrder="ByPtDown",
OutputLevel=3)
assoc_tool_truth = JetSubStructure.JetAssociationTool(
name='JetAssocToolTruth',
GroomingTools=myGroomingTools,
UsedSignalState=2,
JetAlgTools=[sortertool],
OutputLevel=3)
myjetgetter_truth = make_StandardJetGetter(myjetfinder,
myjetdr,
'Truth',
seq=theseq,
doCalib=False,
addDefaultMoment=False)
truthcontname = myjetgetter_truth.outputKey()
make_JetMomentGetter(truthcontname, [assoc_tool_truth], theseq)
return [truthcontname, catNames]
else:
return [None, None]
def myjets():
from JetRec.JetMomentGetter import make_JetMomentGetter
from JetMomentTools.SetupJetMomentTools import getJetVertexAssociationTool
jvatool = getJetVertexAssociationTool('AntiKt', 0.4, 'Topo') # parameters are irrelevant, these will work for any jets
#jvatool_lc = getJetVertexAssociationTool('AntiKt', 0.4, 'LCTopo') # parameters are irrelevant, these will work for any jets
make_JetMomentGetter( 'AntiKt4TopoEMJets' , [jvatool] )
def setupTrackJets(myjetfinder, myjetdr, GroomedDicts, theseq):
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
sortertool = JetSorterTool('myJetSorter',
SortOrder="ByPtDown",
OutputLevel=3)
assoc_tool_track = JetSubStructure.JetAssociationTool(
name='JetAssocToolTrack',
GroomingTools=myGroomingTools,
UsedSignalState=2,
JetAlgTools=[sortertool],
OutputLevel=3)
#---- TrackJet ----
from JetSubStructure.JSTrackJets import createJSTrackJets
mygetter = createJSTrackJets(theseq, myjetfinder, myjetdr)
trackjetcontname = mygetter.outputKey()
#---- End TrackJet ----
make_JetMomentGetter(trackjetcontname, [assoc_tool_track], theseq)
return [trackjetcontname, catNames]
def hsg2_redo_jvf_for_d3pd():
from JetRec.JetMomentGetter import make_JetMomentGetter
from JetMomentTools.SetupJetMomentTools import getJetVertexAssociationTool
jvatool = getJetVertexAssociationTool('AntiKt', 0.4, 'Topo')
# optional : rename the moment if you don't want to overwrite the existing one
jvatool.JVFMomentName = "JVF2"
make_JetMomentGetter( 'AntiKt4TopoJets' , [jvatool] )
make_JetMomentGetter( 'AntiKt4TopoEMJets' , [jvatool] )
import JetD3PDMaker
from JetD3PDMaker.JetD3PDObject import JetD3PDObject
JetD3PDObject.defineBlock(1, 'JVF2',
JetD3PDMaker.JetMomentFillerTool,
Moments=['JVF2'])
def jtsetupTruthJets(myjetfinder, myjetdr, GroomedDicts, theseq):
if rec.doTruth():
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
sortertool = JetSorterTool('myJetSorter',
SortOrder="ByPtDown",
OutputLevel=3)
### tagging truth subjets
subjets_names = [x for x in catNames if "Subjets" in x]
btag_tool = []
if subjets_names != '':
from JetRec.JetGetters import getStandardBTaggerTool
from JetRec.JetRecFlags import jetFlags
jetFlags.doBTagging = True
_btag_tool = getStandardBTaggerTool('AntiKt', 0.4, 'Truth')
if len(_btag_tool) > 0:
btag_tool += [
JetSubStructure.JetAlgToolWrapper(
name='JetBTagger_' + ''.join(subjets_names),
JetAlgTool=_btag_tool[0],
GroomedjetsAuthor=subjets_names)
]
jetFlags.doBTagging = False
assoc_tool_truth = JetSubStructure.JetAssociationTool(
name='JetAssocToolTruth',
GroomingTools=myGroomingTools,
UsedSignalState=2,
JetAlgTools=[sortertool] + btag_tool,
OutputLevel=3)
myjetgetter_truth = make_StandardJetGetter(myjetfinder,
myjetdr,
'Truth',
seq=theseq,
doCalib=False,
addDefaultMoment=False)
truthcontname = myjetgetter_truth.outputKey()
make_JetMomentGetter(truthcontname, [assoc_tool_truth], theseq)
return [truthcontname, catNames]
else:
return [None, None]
def setupTruthJets(myjetfinder, myjetdr, GroomedDicts, theseq):
if rec.doTruth():
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
sortertool = JetSorterTool('JS_JetSorter', SortOrder="ByPtDown", OutputLevel=3 )
widthtool = JetWidthTool('JS_JetWidthTool')
assoc_tools = [ JetSubStructure.GhostJetJetAssociatorTool("myTruthGhostJetJetAssociatorTool", JetdR = myjetdr, JetFinder = myjetfinder, JetInputTool = getStandardInputTools(input='Truth')[0],OutputLevel=3) ]
assoc_tool_truth = JetSubStructure.JetAssociationTool(
name = 'JetAssocToolTruth',
AssociatorTools = assoc_tools, Area = True,
GroomingTools = myGroomingTools, UsedSignalState = 2,
JetAlgTools = [sortertool,widthtool], OutputLevel = 3
)
jetsubstructure_tool_truth = JetSubStructure.JetSubstructureTool(
name = 'JetSubstructureToolTruth', UsedSignalState = 2, OutputLevel=3,
SubstructureTools = [JetSubStructure.NSubjettinessTool("myNSubjettinessTool", N=[1,2,3], R0=myjetdr, GroomedjetsAuthor = catNames),
JetSubStructure.KtHadSubstructureTool("myhadtool", GroomedjetsAuthor = catNames),
JetSubStructure.DipolarityTool("myDipolarityTool",GroomedjetsAuthor = catNames),
JetSubStructure.JetVoronoiAreaCalcTool("myVoronoiArea", GroomedjetsAuthor = catNames, JetdR = myjetdr, JetInputTool = getStandardInputTools(input='Truth')[0]) ])
myjetgetter_truth = make_StandardJetGetter(myjetfinder, myjetdr, 'Truth', seq = theseq, doCalib=False, minPt=30*GeV)#,addDefaultMoment=False)
truthcontname = myjetgetter_truth.outputKey()
extra_moments_list = []
if myjetgetter_truth.jetAlgorithmHandle() == None:
if truthcontname == None:
truthcontname = myjetgetter_truth.buildName()
print 'yot: ', truthcontname
extra_moments_list += [widthtool] ## jet moments lost for truth jets in ESD->AOD transition
make_JetMomentGetter(truthcontname , extra_moments_list + [assoc_tool_truth,jetsubstructure_tool_truth], theseq )
return [ truthcontname, catNames ]
else:
return [ None, None ]
def addGhostAssociation2(jetcollName,
assocName,
assocType,
assocKeys="",
finder=None,
input=None,
mainParam=None):
from JetRec.JetMomentGetter import make_JetMomentGetter
t = setupGhostAssociationTool(jetcollName, assocName, assocType, assocKeys,
finder, input, mainParam)
g = make_JetMomentGetter(jetcollName, [t])
return g.getJetMomentCalc()
def setupTrackJets(myjetfinder, myjetdr, GroomedDicts, theseq):
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
sortertool = JetSorterTool('JS_JetSorter', SortOrder="ByPtDown", OutputLevel=3 )
widthtool = JetWidthTool('JS_JetWidthTool')
assoc_tool_track = JetSubStructure.JetAssociationTool(
name = 'JetAssocToolTrack',
AssociatorTools = [],
GroomingTools = myGroomingTools, UsedSignalState = 2,
JetAlgTools = [sortertool, widthtool], OutputLevel = 3
)
jetsubstructure_tool_track = JetSubStructure.JetSubstructureTool(
name = 'JetSubstructureToolTrack', UsedSignalState = 2, OutputLevel=3,
SubstructureTools = [JetSubStructure.NSubjettinessTool("myNSubjettinessTool", N=[1,2,3], R0=myjetdr, GroomedjetsAuthor = catNames),
JetSubStructure.KtHadSubstructureTool("myhadtool", GroomedjetsAuthor = catNames),
JetSubStructure.DipolarityTool("myDipolarityTool",GroomedjetsAuthor = catNames) ])
#---- TrackJet ----
jetFlags.finalMinEt = 30*GeV
from QcdD3PDMaker.JSTrackJets import createJSTrackJets
mygetter = createJSTrackJets(theseq, myjetfinder,myjetdr)
jetFlags.finalMinEt =30*GeV
trackjetcontname = mygetter.outputKey()
add_jetMoments_to_JetAlg(mygetter, [widthtool])
#---- End TrackJet ----
make_JetMomentGetter(trackjetcontname , [assoc_tool_track, jetsubstructure_tool_track], theseq )
return [ trackjetcontname, catNames ]
def addGhostAssociation(jetcollName,
ghostAssoc=[],
finder=None,
input=None,
mainParam=None,
seq=AlgSequence()):
from JetRec.JetMomentGetter import make_JetMomentGetter
t = setupGhostAssociationTool(jetcollName,
ghostAssoc,
finder,
input,
mainParam,
seq=seq)
g = make_JetMomentGetter(jetcollName, [t], seq=seq)
return g.getJetMomentCalc()
atool4.CellCalibrator = cellcalibtool
atool6 = JetBadChanCorrTool("JetBadChanCorrTool_anti06",UpdateEvery=False,ConeDr=0.6,UseCalibScale=False)
atool6.CellCalibrator = cellcalibtool
cqual_tool = JetCaloQualityTool("CaloQualTool")
cqual_tool.doSamplingBasedVariables=True
cqual_tool.doCellBasedVariables=True
BadCellCorrT = JetBadCellCorrTool("JetBadCellCorrTool")
BadCellCorrT.CellCalibrator = cellcalibtool
##printfunc (BadCellCorrT)
##toolist += [BadCellCorrT]
make_JetMomentGetter("AntiKt4TopoJets",[atool4,cqual_tool,BadCellCorrT])
make_JetMomentGetter("AntiKt6TopoJets",[atool6,cqual_tool,BadCellCorrT])
else:
from AthenaCommon.Logging import logging
log = logging.getLogger ('MissingETGoodness')
log.warning ('No cells; not rebuilding jets.')
###############################################
from MissingETGoodness.MissingETGoodnessConf import JetVarTool as ConfiguredJetVarTool
JetVarTool = ConfiguredJetVarTool(
JetCollectionKey = 'AntiKt4TopoJets',
JetTruthCollectionKey = 'Cone4TruthJets',
#make_JetMomentGetter("AntiKt4TruthNewJets", [widthtool, assoc_tool_truth, jetsubstructure_tool_truth], topSequence)
#make_JetMomentGetter("AntiKt6TruthNewJets", [widthtool, assoc_tool_truth, jetsubstructure_tool_truth], topSequence)
# SPLIT variables for R=0.4/0.6 jets.
jetsubstructure_tool = JetSubStructure.JetSubstructureTool(
name='JetSubstructureTool',
UsedSignalState=2,
OutputLevel=3,
SubstructureTools=[
JetSubStructure.KtHadSubstructureTool("myhadtool",
GroomedjetsAuthor=[])
])
#make_JetMomentGetter("AntiKt4TopoEMJets", [jetsubstructure_tool], topSequence)
#make_JetMomentGetter("AntiKt6TopoEMJets", [jetsubstructure_tool], topSequence)
make_JetMomentGetter("AntiKt4LCTopoNewJets", [jetsubstructure_tool],
topSequence) # replacing RetaggedJets with NewJets
#make_JetMomentGetter("AntiKt6LCTopoNewJets", [jetsubstructure_tool], topSequence) # replacing RetaggedJets with NewJets
if rec.doTruth() and CommonD3PDMakerFlags.doRecTruthJet():
make_JetMomentGetter("AntiKt4TruthNewJets", [jetsubstructure_tool],
topSequence)
#make_JetMomentGetter("AntiKt6TruthNewJets", [jetsubstructure_tool], topSequence)
make_JetMomentGetter(
"AntiKt4TruthNewJets",
[widthtool, assoc_tool_truth, jetsubstructure_tool_truth], topSequence)
#make_JetMomentGetter("AntiKt6TruthNewJets", [widthtool, assoc_tool_truth, jetsubstructure_tool_truth], topSequence)
make_JetMomentGetter(
"AntiKt4TrackJets", [jetsubstructure_tool], topSequence
) # there is no TrackZJets but only TrackJets inside CommonJetTrackZClusterTool_jobOptions.py
#make_JetMomentGetter("AntiKt6TrackJets", [jetsubstructure_tool], topSequence)
def setupTruthJets(myjetfinder, myjetdr, GroomedDicts, theseq):
if rec.doTruth():
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
catNames_trimmed = list(x for x in catNames if 'Trimmed' in x)
sortertool = JetSorterTool('myJetSorter',
SortOrder="ByPtDown",
OutputLevel=3)
wrapped_sortertool = JetSubStructure.JetAlgToolWrapper(
"JetSorterWrapper",
JetAlgTool=sortertool,
GroomedjetsAuthor=catNames_trimmed)
widthtool = JetWidthTool('JS_JetWidthTool')
truth_assoc = JetSubStructure.TruthParticleJetAssociatorTool(
"myTruthLeptonsAssoc",
ContainerName="SpclMC",
dR=myjetdr,
OutputLevel=3, #SpclMC for AODs, INav4MomTruthEvent for ESD
ParticlesPdgID=[11, 13],
ParticlesStatus=[1, 1],
ParticlesType=[2, 6],
ParticlesOrigin=[-1, -1])
truth_assoc.mcTruthClassifier.OutputLevel = 3
truth_assoc.mcTruthClassifier.McEventCollection = "GEN_AOD" #"TruthEvent" for ESD
from MCTruthClassifier.MCTruthClassifierBase import exToCalo
truth_assoc.mcTruthClassifier.ExtrapolateToCaloTool = exToCalo
assoc_tool_truth = JetSubStructure.JetAssociationTool(
name='JetAssocToolTruth',
AssociatorTools=[truth_assoc],
GroomingTools=myGroomingTools,
UsedSignalState=2,
JetAlgTools=[wrapped_sortertool, widthtool],
OutputLevel=3)
jetsubstructure_tool_truth = JetSubStructure.JetSubstructureTool(
name='JetSubstructureToolTruth',
UsedSignalState=2,
OutputLevel=3,
SubstructureTools=[
JetSubStructure.KtTruthLeptonSubstructureTool(
"mytruthleptontool", OutputLevel=3, GroomedjetsAuthor=[]),
JetSubStructure.NSubjettinessTool(
"myNSubjettinessTool",
N=[1, 2, 3],
R0=myjetdr,
GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.KtHadSubstructureTool(
"myhadtool", GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.JetPullVectorTool(
"myJetPullVectorTool", GroomedjetsAuthor=catNames_trimmed)
])
#need to use truth collection on disk for smaller R jets for the TrueFlavorComponent block
if myjetdr < 0.9:
contname = myjetfinder + '%dTruthJets' % (myjetdr * 10)
else:
contname = myjetfinder + '%dTruthJSJets' % (myjetdr * 10)
myjetgetter_truth = make_StandardJetGetter(
myjetfinder,
myjetdr,
'Truth',
seq=theseq,
doCalib=False,
outputCollectionName=contname) #,addDefaultMoment=False)
truthcontname = myjetgetter_truth.outputKey()
extra_moments_list = []
if myjetgetter_truth.jetAlgorithmHandle() == None:
if truthcontname == None:
truthcontname = myjetgetter_truth.buildName()
print 'yot: ', truthcontname
extra_moments_list += [
widthtool
] ## jet moments lost for truth jets in ESD->AOD transition
sss = make_JetMomentGetter(
truthcontname, extra_moments_list +
[assoc_tool_truth, jetsubstructure_tool_truth], theseq)
## for lepton assoc
jetcont = truthcontname[:-4] #removing the 'Jets'
for x in catNames_trimmed:
make_JetMomentGetter(jetcont + x + 'Jets', [
JetSubStructure.JetAssociationTool(
'JetAssocToolTruthGr',
AssociatorTools=[truth_assoc],
GroomingTools=[],
UsedSignalState=2,
JetAlgTools=[]),
JetSubStructure.JetSubstructureTool(
'JetSubstructureToolTruthGr',
UsedSignalState=2,
SubstructureTools=[
JetSubStructure.KtTruthLeptonSubstructureTool(
"mytruthleptontool", GroomedjetsAuthor=[])
])
], theseq)
return [truthcontname, catNames]
else:
return [None, None]
from RecExConfig.RecFlags import rec
from JetRec.JetMomentGetter import make_JetMomentGetter
from JetMomentTools.SetupJetMomentTools import getJetVertexAssociationTool
jvatool = getJetVertexAssociationTool('AntiKt', 0.4, 'Topo')
jvatool.StoreQualityVariables = True
# jvatool.JVFMomentName = "JVF2"
make_JetMomentGetter('AntiKt4TopoEMNewJets', [jvatool])
#make_JetMomentGetter( 'AntiKt6TopoEMNewJets' , [jvatool] )
#make_JetMomentGetter( 'AntiKt4TopoJets' , [jvatool] )
#make_JetMomentGetter( 'AntiKt6TopoJets' , [jvatool] )
make_JetMomentGetter('AntiKt4LCTopoJets', [jvatool])
#make_JetMomentGetter( 'AntiKt6LCTopoJets' , [jvatool] )
#make_JetMomentGetter( 'AntiKt4TowerJets' , [jvatool] )
#make_JetMomentGetter( 'AntiKt6TowerJets' , [jvatool] )
from D3PDMakerConfig.CommonD3PDMakerFlags import CommonD3PDMakerFlags
if CommonD3PDMakerFlags.doRecJet():
make_JetMomentGetter('AntiKt4TopoEMNewJets', [jvatool])
#make_JetMomentGetter( 'AntiKt6TopoEMNewJets' , [jvatool] )
#make_JetMomentGetter( 'AntiKt4TopoNewJets' , [jvatool] )
#make_JetMomentGetter( 'AntiKt6TopoNewJets' , [jvatool] )
make_JetMomentGetter('AntiKt4LCTopoNewJets', [jvatool])
#make_JetMomentGetter( 'AntiKt6LCTopoNewJets' , [jvatool] )
if not rec.readAOD():
make_JetMomentGetter('AntiKt4TowerEMNewJets', [jvatool])
#make_JetMomentGetter( 'AntiKt6TowerEMNewJets' , [jvatool] )
make_JetMomentGetter('AntiKt4TowerNewJets', [jvatool])
#make_JetMomentGetter( 'AntiKt6TowerNewJets' , [jvatool] )
def setupTrackJets(myjetfinder, myjetdr, GroomedDicts, theseq):
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
catNames_trimmed = list(x for x in catNames if 'Trimmed' in x)
sortertool = JetSorterTool('myJetSorter',
SortOrder="ByPtDown",
OutputLevel=3)
wrapped_sortertool = JetSubStructure.JetAlgToolWrapper(
"JetSorterWrapper",
JetAlgTool=sortertool,
GroomedjetsAuthor=catNames_trimmed)
widthtool = JetWidthTool('JS_JetWidthTool')
e_assoc = JetSubStructure.ElectronJetAssociatorTool(
"JSEAssocTrack",
ContainerName='ElectronAODCollection',
UseCluster=False,
UseTrack=True,
dR=myjetdr,
OutputLevel=3)
mu_assoc = JetSubStructure.MuonJetAssociatorTool(
"JSMuAssocTrack",
ContainerName='MuidMuonCollection',
UseTrack=True,
dR=myjetdr,
OutputLevel=3)
assoc_tool_track = JetSubStructure.JetAssociationTool(
name='JetAssocToolTrack',
AssociatorTools=[e_assoc, mu_assoc],
GroomingTools=myGroomingTools,
UsedSignalState=2,
JetAlgTools=[wrapped_sortertool, widthtool],
OutputLevel=3)
jetsubstructure_tool_track = JetSubStructure.JetSubstructureTool(
name='JetSubstructureToolTrack',
UsedSignalState=2,
OutputLevel=3,
SubstructureTools=[
JetSubStructure.NSubjettinessTool(
"myNSubjettinessTool",
N=[1, 2, 3],
R0=myjetdr,
GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.KtHadSubstructureTool(
"myhadtool", GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.KtElectronSubstructureTool("myelectool",
GroomedjetsAuthor=[]),
JetSubStructure.KtMuonSubstructureTool("mymuontool",
GroomedjetsAuthor=[]),
JetSubStructure.JetPullVectorTool(
"myJetPullVectorTool", GroomedjetsAuthor=catNames_trimmed)
])
#---- TrackJet ----
#jetFlags.finalMinEt = 5*GeV
from BoostedTopD3PDMaker.JSTrackJets import createJSTrackJets
mygetter = createJSTrackJets(theseq, myjetfinder, myjetdr)
trackjetcontname = mygetter.outputKey()
add_jetMoments_to_JetAlg(mygetter, [widthtool])
#---- End TrackJet ----
sss = make_JetMomentGetter(trackjetcontname,
[assoc_tool_track, jetsubstructure_tool_track],
theseq)
## for lepton assoc
jetcont = trackjetcontname[:-4] #removing the 'Jets'
for x in catNames_trimmed:
make_JetMomentGetter(jetcont + x + 'Jets', [
JetSubStructure.JetAssociationTool(
'JetAssocToolTrackGr',
AssociatorTools=[e_assoc, mu_assoc],
GroomingTools=[],
UsedSignalState=2,
JetAlgTools=[]),
JetSubStructure.JetSubstructureTool(
'JetSubstructureToolTrackGr',
UsedSignalState=2,
SubstructureTools=[
JetSubStructure.KtElectronSubstructureTool(
"myelectool", GroomedjetsAuthor=[]),
JetSubStructure.KtMuonSubstructureTool(
"mymuontool", GroomedjetsAuthor=[])
])
], theseq)
return [trackjetcontname, catNames]
def setupCaloJets(myjetfinder, myjetdr, myjetinput, GroomedDicts, theseq):
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
catNames_trimmed = list(x for x in catNames if 'Trimmed' in x)
sortertool = JetSorterTool('myJetSorter',
SortOrder="ByPtDown",
OutputLevel=3)
wrapped_sortertool = JetSubStructure.JetAlgToolWrapper(
"JetSorterWrapper",
JetAlgTool=sortertool,
GroomedjetsAuthor=catNames_trimmed)
widthtool = JetWidthTool('JS_JetWidthTool')
grjets_calib = []
from JetSubStructure.mySetupJetCalibrators import doEtaMassJESCorrection
from JetCalibTools.MakeCalibSequences import calibTags
calibTags['ETAMASSJES'] = doEtaMassJESCorrection
## Origin correction
mydoCalib = False
mycalibName = ''
if 'LCTopo' in myjetinput:
mydoCalib = False
if (myjetfinder + '%d' %
(myjetdr * 10) == 'AntiKt10') or (myjetfinder + '%d' %
(myjetdr * 10) == 'CamKt12'):
mycalibName = 'LC:ORIGIN_ETAMASSJES'
else:
mycalibName = 'LC:ORIGIN'
stdcaliblist = []
allowedcalib = [
'AntiKt10LCTopoTrimmedJets',
'AntiKt10LCTopoTrimmedPtFrac3SmallR30Jets',
'AntiKt10LCTopoTrimmedPtFrac5SmallR30Jets',
'CamKt12LCTopoTrimmedPtFrac5SmallR30Jets'
]
for x in catNames:
author = myjetfinder + '%d' % (myjetdr *
10) + myjetinput + x + 'Jets'
#if author in allowedcalib :
#grjets_calib += [JetSubStructure.JetAlgToolWrapper(name = 'calibtool_origin_etamassjes_'+x,
# JetAlgTool = getStandardCalibTool(myjetfinder,myjetdr, myjetinput+x, doCalib=True, calibName='LC:ORIGIN_ETAMASSJES')[0],
# GroomedjetsAuthor = [x] )]
#elif 'Trimmed' in x:
# stdcaliblist += [x]
#if stdcaliblist != [] :
#grjets_calib += [JetSubStructure.JetAlgToolWrapper(name = 'calibtool_origin',
# JetAlgTool = getStandardCalibTool(myjetfinder,myjetdr, myjetinput, doCalib=True, calibName='LC:ORIGIN')[0],
# GroomedjetsAuthor = stdcaliblist )]
elif 'Topo' in myjetinput:
mydoCalib = True
mycalibName = 'EM:ORIGIN'
grjets_calib += [
JetSubStructure.JetAlgToolWrapper(
name='calibtool_origin',
JetAlgTool=getStandardCalibTool(myjetfinder,
myjetdr,
myjetinput,
doCalib=True,
calibName='EM:ORIGIN')[0],
GroomedjetsAuthor=catNames_trimmed)
]
truthcontname = 'none'
if rec.doTruth():
if myjetdr < 0.9: #need to use truth collection on disk for smaller R jets for the TrueFlavorComponent block
truthcontname = myjetfinder + '%dTruthJets' % (myjetdr * 10)
else:
truthcontname = myjetfinder + '%dTruthJSJets' % (myjetdr * 10)
trackjetcontname = myjetfinder + '%dTrackZJets' % (myjetdr * 10)
assoc_tools = []
trkj_assoc = JetSubStructure.JetJetAssociatorTool(
"Trk_JetJetAssociatorTool",
ContainerName=trackjetcontname,
AssociationName="TrackJets",
MatchOrigin=True)
assoc_tools += [trkj_assoc]
if rec.doTruth():
truthj_assoc = JetSubStructure.JetJetAssociatorTool(
"Truth_JetJetAssociatorTool",
ContainerName=truthcontname,
AssociationName="TruthJets",
MatchOrigin=False)
assoc_tools += [truthj_assoc]
e_assoc = JetSubStructure.ElectronJetAssociatorTool(
"JSEAssoc",
ContainerName='ElectronAODCollection',
UseCluster=True,
UseTrack=False,
dR=myjetdr,
ResolveCluster=True,
OutputLevel=3)
e_assoc.METClusterResolver.MaximumRadialDistance = 0.2
e_assoc.METClusterResolver.LongitudinalExtension = 1.0
e_assoc.METClusterResolver.RadialExtension = 1.0
e_assoc.METClusterResolver.EnergyFractionThreshold = 0.1
e_assoc.METClusterResolver.OutputLevel = 3
mu_assoc = JetSubStructure.MuonJetAssociatorTool(
"JSMuAssoc",
ContainerName='MuidMuonCollection',
UseTrack=False,
dR=myjetdr,
OutputLevel=3)
assoc_tools += [e_assoc, mu_assoc]
assoc_tool = JetSubStructure.JetAssociationTool(
name='JetAssocTool',
AssociatorTools=assoc_tools,
GroomingTools=myGroomingTools,
UsedSignalState=2,
JetAlgTools=grjets_calib + [wrapped_sortertool, widthtool],
OutputLevel=5)
jetsubstructure_tool = JetSubStructure.JetSubstructureTool(
name='JetSubstructureTool',
UsedSignalState=2,
OutputLevel=3,
SubstructureTools=[
JetSubStructure.NSubjettinessTool(
"myNSubjettinessTool",
N=[1, 2, 3],
R0=myjetdr,
GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.KtHadSubstructureTool(
"myhadtool", GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.KtElectronSubstructureTool("myelectool",
GroomedjetsAuthor=[]),
JetSubStructure.KtMuonSubstructureTool("mymuontool",
GroomedjetsAuthor=[]),
JetSubStructure.JetPullVectorTool(
"myJetPullVectorTool", GroomedjetsAuthor=catNames_trimmed)
])
outkey = myjetfinder + '%d' % (myjetdr * 10) + myjetinput
if myjetinput == 'Topo':
outkey += 'EM'
outkey += 'Jets' #'JSJets' for rebuilding on the fly
myjetgetter = make_StandardJetGetter(myjetfinder,
myjetdr,
myjetinput,
seq=theseq,
doCalib=mydoCalib,
calibName=mycalibName,
addDefaultMoment=False,
outputCollectionName=outkey)
contname = myjetgetter.outputKey()
extra_moments_list = []
if myjetgetter.jetAlgorithmHandle(
) == None: # JetCollection already in AOD
if contname == None:
contname = myjetgetter.buildName()
if contname == 'AntiKt4LCTopoJets':
contname = 'AntiKt4LCTopoJetsReTagged'
else:
extra_moments_list += [widthtool]
# ------------------------------------------------
# Associate a vertex and set the JVF moment. Must be placed AFTER vertex correction
# ------------------------------------------------
from AthenaCommon.DetFlags import DetFlags
from AthenaCommon.AppMgr import ToolSvc
noVertex = jobproperties.Beam.beamType == 'cosmics' or jobproperties.Beam.beamType == 'singlebeam'
if DetFlags.detdescr.ID_on() and jetFlags.doJVF(
) and 'Truth' not in myjetinput and not noVertex and myjetdr >= 0.7:
from JetSubStructure.mySetupJetMomentTools import mygetJetVertexAssociationTool
myjvatool = mygetJetVertexAssociationTool(myjetfinder, myjetdr,
myjetinput,
'Custom%d' % (myjetdr * 10))
#add_jetMoments_to_JetAlg(myjetgetter, [myjvatool])
extra_moments_list += [myjvatool]
make_JetMomentGetter(
contname, extra_moments_list + [assoc_tool, jetsubstructure_tool],
theseq)
## for lepton assoc
jetcont = contname[:-4] #removing the 'Jets'
if contname == 'AntiKt4LCTopoJetsReTagged':
jetcont = 'AntiKt4LCTopo'
for x in catNames_trimmed:
make_JetMomentGetter(jetcont + x + 'Jets', [
JetSubStructure.JetAssociationTool(
'JetAssocToolGr',
AssociatorTools=[e_assoc, mu_assoc],
GroomingTools=[],
UsedSignalState=2,
JetAlgTools=[],
OutputLevel=5),
JetSubStructure.JetSubstructureTool(
'JetSubstructureToolGr',
UsedSignalState=2,
SubstructureTools=[
JetSubStructure.KtElectronSubstructureTool(
"myelectool", GroomedjetsAuthor=[]),
JetSubStructure.KtMuonSubstructureTool(
"mymuontool", GroomedjetsAuthor=[])
])
], theseq)
return [contname, catNames]
JetTagD3PD_AntiKt4MuonNonInteractingTruth = False
JetTagD3PD_AntiKt6MuonNonInteractingTruth = False
JetTagD3PD_AntiKtZ4TrackJets = False
JetTagD3PD_AntiKt4TruthPurePU = False
if JetTagD3PD_rerunJVF or JetTagD3PD_runJVF:
from JetRec.JetMomentGetter import make_JetMomentGetter
from JetMomentTools.SetupJetMomentTools import getJetVertexAssociationTool
if JetTagD3PD_rerunJVF:
# parameters are irrelevant, these will work for any jets
# optional : rename the moment if you don't want to overwrite the existing one
# jvatool.JVFMomentName = "JVF2"
if JetTagD3PD_AntiKt4TopoEM:
myjvatoolakt4em = getJetVertexAssociationTool('AntiKt', 0.4, 'Topo')
make_JetMomentGetter('AntiKt4TopoEMJets', [myjvatoolakt4em])
if JetTagD3PD_AntiKt6TopoEM:
myjvatoolakt6em = getJetVertexAssociationTool('AntiKt', 0.6, 'Topo')
make_JetMomentGetter('AntiKt6TopoEMJets', [myjvatoolakt6em])
if JetTagD3PD_AntiKt4LCTopo:
myjvatoolakt4lc = getJetVertexAssociationTool('AntiKt', 0.4, 'LCTopo')
make_JetMomentGetter('AntiKt4LCTopoJets', [myjvatoolakt4lc])
if JetTagD3PD_AntiKt6LCTopo:
myjvatoolakt6lc = getJetVertexAssociationTool('AntiKt', 0.6, 'LCTopo')
make_JetMomentGetter('AntiKt6LCTopoJets', [myjvatoolakt6lc])
if JetTagD3PD_runJVF:
if rec.doTruth():
from JetMomentTools.JetMomentToolsConf import JetMomentsFromCalib
from RecExConfig.RecFlags import rec
from QcdD3PDMaker.QcdD3PDMakerFlags import QcdD3PDMakerFlags
mt = JetMomentsFromCalib()
from JetCalibTools.MakeCalibSequences import alternateCalibSequence
from JetRec.JetAlgConfiguration import checkAndUpdateOptions
#Request the GSC calib sequence, starting from the jet *final scale* (i.e. ETAJES for TopoEMJets)
d = checkAndUpdateOptions(input='Topo')
StructSeq = alternateCalibSequence("CAL:STRUCT1", d)
mt.Calibrator = StructSeq # link the calibrator to the moment tool.
# redefine StructSeq, and set options to save only the last
# 2 moments of the GSC sequence.
StructSeq = StructSeq.CalibToolSequence[-1]
StructSeq.MomentTag = "GSCFactor"
StructSeq.SimpleFactorCorrection = True
for t in StructSeq.CalibToolSequence[:
-1]: #switch off all moments writing but the last.
t.WriteMoments = False
# schedule the alg computing jet moments :
make_JetMomentGetter("AntiKt4TopoEMJets", [mt])
make_JetMomentGetter("AntiKt6TopoEMJets", [mt])
if QcdD3PDMakerFlags.doRecJet():
if QcdD3PDMakerFlags.doTopoEMJet():
make_JetMomentGetter("AntiKt4TopoEMNewJets", [mt])
make_JetMomentGetter("AntiKt6TopoEMNewJets", [mt])
if QcdD3PDMakerFlags.doLCTopoJet():
make_JetMomentGetter("AntiKt4LCTopoNewJets", [mt])
make_JetMomentGetter("AntiKt5LCTopoNewJets", [mt])
make_JetMomentGetter("AntiKt6LCTopoNewJets", [mt])
def setupCaloJets(myjetfinder, myjetdr, myjetinput, GroomedDicts, theseq):
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
sortertool = JetSorterTool('JS_JetSorter', SortOrder="ByPtDown", OutputLevel=3 )
widthtool = JetWidthTool('JS_JetWidthTool')
grjets_calib = []
## Origin correction
mydoCalib = False
mycalibName = ''
from JetSubStructure.mySetupJetCalibrators import doEtaMassJESCorrection
from JetCalibTools.MakeCalibSequences import calibTags
calibTags['ETAMASSJES'] = doEtaMassJESCorrection
if 'LCTopo' in myjetinput:
#mydoCalib = True
mydoCalib = False
if (myjetfinder + '%d'%(myjetdr*10) == 'AntiKt10') or (myjetfinder + '%d'%(myjetdr*10) == 'CamKt12'):
mycalibName = 'LC:ORIGIN_ETAMASSJES'
else:
mycalibName = 'LC:ORIGIN'
stdcaliblist = []
allowedcalib = ['AntiKt10LCTopoTrimmedJets', 'AntiKt10LCTopoTrimmedPtFrac3SmallR30Jets','AntiKt10LCTopoTrimmedPtFrac5SmallR30Jets','CamKt12LCTopoTrimmedPtFrac5SmallR30Jets']
for x in catNames:
author = myjetfinder + '%d'%(myjetdr*10) + myjetinput + x + 'Jets'
if author in allowedcalib :
#grjets_calib += [JetSubStructure.JetAlgToolWrapper(name = 'calibtool_origin_etamassjes_'+x, JetAlgTool = getStandardCalibTool(myjetfinder,myjetdr, myjetinput+x, doCalib=True, calibName='LC:ORIGIN_ETAMASSJES')[0], GroomedjetsAuthor = stdcaliblist )]
#grjets_calib += [JetSubStructure.JetAlgToolWrapper(name = 'calibtool_origin_etamassjes_'+x, JetAlgTool = getStandardCalibTool(myjetfinder,myjetdr, myjetinput+x, doCalib=False)[0], GroomedjetsAuthor = stdcaliblist) ]
pass
else:
stdcaliblist += [x]
if stdcaliblist != [] :
#grjets_calib += [JetSubStructure.JetAlgToolWrapper(name = 'calibtool_origin',JetAlgTool = getStandardCalibTool(myjetfinder,myjetdr, myjetinput, doCalib=False, calibName='LC:ORIGIN')[0], GroomedjetsAuthor = stdcaliblist )]
pass
elif 'Topo' in myjetinput:
#mydoCalib = True
mydoCalib = False
mycalibName = 'EM:ORIGIN'
#grjets_calib += getStandardCalibTool(myjetfinder,myjetdr, myjetinput, doCalib=True, calibName='EM:ORIGIN')
grjets_calib += getStandardCalibTool(myjetfinder,myjetdr, myjetinput, doCalib=False, calibName='EM:ORIGIN')
truthcontname = 'none'
if rec.doTruth():
truthcontname = myjetfinder + '%dTruthJets' % (myjetdr*10)
trackjetcontname = myjetfinder + '%dTrackZJets' % (myjetdr*10)
assoc_tools = []
trkj_assoc = JetSubStructure.JetJetAssociatorTool("Trk_JetJetAssociatorTool", ContainerName = trackjetcontname,
AssociationName = "TrackJets",
MatchOrigin = True)
assoc_tools += [ trkj_assoc ]
if rec.doTruth():
truthj_assoc = JetSubStructure.JetJetAssociatorTool("Truth_JetJetAssociatorTool", ContainerName = truthcontname,
AssociationName = "TruthJets",
MatchOrigin = False)
assoc_tools += [ truthj_assoc ]
doArea = False
if 'LCTopo' in myjetinput:
doArea = True
assoc_tools += [ JetSubStructure.GhostJetJetAssociatorTool("myGhostJetJetAssociatorTool", JetdR = myjetdr, JetFinder = myjetfinder, JetInputTool = getStandardInputTools(input=myjetinput)[0],OutputLevel=3) ]
assoc_tool = JetSubStructure.JetAssociationTool(
name = 'JetAssocTool',
AssociatorTools = assoc_tools,
GroomingTools = myGroomingTools, UsedSignalState = 2, Area = doArea,
JetAlgTools = grjets_calib + [sortertool, widthtool], OutputLevel = 3
)
jetsubstructure_tool = JetSubStructure.JetSubstructureTool(
name = 'JetSubstructureTool', UsedSignalState = 2, OutputLevel=3,
SubstructureTools = [JetSubStructure.NSubjettinessTool("myNSubjettinessTool", N=[1,2,3], R0=myjetdr, GroomedjetsAuthor = catNames),
JetSubStructure.KtHadSubstructureTool("myhadtool", GroomedjetsAuthor = catNames),
JetSubStructure.DipolarityTool("myDipolarityTool",GroomedjetsAuthor = catNames),
JetSubStructure.JetVoronoiAreaCalcTool("myVoronoiArea", GroomedjetsAuthor = catNames, JetdR = myjetdr, JetInputTool = getStandardInputTools(input=myjetinput)[0]) ])
outkey = myjetfinder + '%d'%(myjetdr*10) + myjetinput
if myjetinput == 'Topo':
outkey += 'EM'
outkey += 'Jets'
myjetgetter = make_StandardJetGetter(myjetfinder, myjetdr, myjetinput, seq = theseq, doCalib=mydoCalib, calibName=mycalibName,addDefaultMoment=False, minPt=30*GeV,
outputCollectionName = outkey)
contname = myjetgetter.outputKey()
extra_moments_list = []
if myjetgetter.jetAlgorithmHandle() == None: # JetCollection already in AOD
if contname == None:
contname=myjetgetter.buildName()
print 'yo2: ', contname
else:
extra_moments_list += [widthtool]
#add_jetMoments_to_JetAlg(myjetgetter, [JetWidthTool()])
# ------------------------------------------------
# Associate a vertex and set the JVF moment. Must be placed AFTER vertex correction
# ------------------------------------------------
from AthenaCommon.DetFlags import DetFlags
from AthenaCommon.AppMgr import ToolSvc
noVertex = jobproperties.Beam.beamType == 'cosmics' or jobproperties.Beam.beamType == 'singlebeam'
if DetFlags.detdescr.ID_on() and jetFlags.doJVF() and 'Truth' not in myjetinput and not noVertex:
from JetSubStructure.mySetupJetMomentTools import mygetJetVertexAssociationTool
myjvatool = mygetJetVertexAssociationTool(myjetfinder, myjetdr,myjetinput,'Custom%d'%(myjetdr*10))
#add_jetMoments_to_JetAlg(myjetgetter, [myjvatool])
extra_moments_list += [myjvatool]
make_JetMomentGetter(contname , extra_moments_list + [assoc_tool, jetsubstructure_tool], theseq )
return [ contname, catNames ]
def setupCaloJets(myjetfinder, myjetdr, myjetinput, GroomedDicts, theseq):
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
#catNames_trimmed = list(x for x in catNames if 'Trimmed' in x)
#catNames_trimmed += list(x for x in catNames if 'Filter' in x)
#catNames_trimmed = catNames
catNames_trimmed = list(x for x in catNames if 'Subjet' not in x)
sortertool = JetSorterTool('myJetSorter',
SortOrder="ByPtDown",
OutputLevel=3)
wrapped_sortertool = JetSubStructure.JetAlgToolWrapper(
"JetSorterWrapper",
JetAlgTool=sortertool,
GroomedjetsAuthor=catNames_trimmed)
widthtool = JetWidthTool('JS_JetWidthTool')
grjets_calib = []
from JetSubStructure.mySetupJetCalibrators import doEtaMassJESCorrection
from JetCalibTools.MakeCalibSequences import calibTags
calibTags['ETAMASSJES'] = doEtaMassJESCorrection
## Origin correction
mydoCalib = False
mycalibName = ''
if 'LCTopo' in myjetinput:
mydoCalib = True
if (myjetfinder + '%d' %
(myjetdr * 10) == 'AntiKt10') or (myjetfinder + '%d' %
(myjetdr * 10) == 'CamKt12'):
mycalibName = 'LC:ORIGIN_ETAMASSJES'
else:
mycalibName = 'LC:ORIGIN'
# overwrite with latest offset substraction
mycalibName = "LC:ApplyAreaOffset"
stdcaliblist = []
#allowedcalib = ['AntiKt10LCTopoTrimmedJets', 'AntiKt10LCTopoTrimmedPtFrac3SmallR30Jets','AntiKt10LCTopoTrimmedPtFrac5SmallR30Jets','CamKt12LCTopoTrimmedPtFrac5SmallR30Jets']
allowedcalib = [] # etamassjes is currently obsolete for these jets
for x in catNames:
author = myjetfinder + '%d' % (myjetdr *
10) + myjetinput + x + 'Jets'
if author in allowedcalib:
grjets_calib += [
JetSubStructure.JetAlgToolWrapper(
name='calibtool_origin_etamassjes_' + x,
JetAlgTool=getStandardCalibTool(
myjetfinder,
myjetdr,
myjetinput + x,
doCalib=True,
calibName='LC:ORIGIN_ETAMASSJES')[0],
GroomedjetsAuthor=[x])
]
#elif 'Trimmed' in x:
else:
stdcaliblist += [x]
if stdcaliblist != []:
grjets_calib += [
JetSubStructure.JetAlgToolWrapper(
name='calibtool_origin',
JetAlgTool=getStandardCalibTool(myjetfinder,
myjetdr,
myjetinput,
doCalib=True,
calibName='LC:ORIGIN')[0],
GroomedjetsAuthor=stdcaliblist)
]
elif 'Topo' in myjetinput:
mydoCalib = True
#mycalibName = 'EM:ORIGIN'
# overwrite with latest offset substraction
mycalibName = "EM:ApplyAreaOffset"
grjets_calib += [
JetSubStructure.JetAlgToolWrapper(
name='calibtool_origin',
JetAlgTool=getStandardCalibTool(myjetfinder,
myjetdr,
myjetinput,
doCalib=True,
calibName='EM:ORIGIN')[0],
GroomedjetsAuthor=catNames_trimmed)
]
#grjets_calib += getStandardCalibTool(myjetfinder,myjetdr, myjetinput, doCalib=True, calibName='EM:ORIGIN')
truthcontname = 'none'
if rec.doTruth():
if myjetdr < 0.9: #need to use truth collection on disk for smaller R jets for the TrueFlavorComponent block
truthcontname = myjetfinder + '%dTruthJets' % (myjetdr * 10)
else:
truthcontname = myjetfinder + '%dTruthJSJets' % (myjetdr * 10)
trackjetcontname = myjetfinder + '%dTrackZJets' % (myjetdr * 10)
assoc_tools = []
trkj_assoc = JetSubStructure.JetJetAssociatorTool(
"Trk_JetJetAssociatorTool",
ContainerName=trackjetcontname,
AssociationName="TrackJets",
MatchOrigin=True)
assoc_tools += [trkj_assoc]
if rec.doTruth():
truthj_assoc = JetSubStructure.JetJetAssociatorTool(
"Truth_JetJetAssociatorTool",
ContainerName=truthcontname,
AssociationName="TruthJets",
MatchOrigin=False)
assoc_tools += [truthj_assoc]
e_assoc = JetSubStructure.ElectronJetAssociatorTool(
"JSEAssoc",
ContainerName='ElectronAODCollection',
UseCluster=True,
UseTrack=False,
dR=myjetdr,
ResolveCluster=True,
OutputLevel=3)
e_assoc.METClusterResolver.MaximumRadialDistance = 0.2
e_assoc.METClusterResolver.LongitudinalExtension = 1.0
e_assoc.METClusterResolver.RadialExtension = 1.0
e_assoc.METClusterResolver.EnergyFractionThreshold = 0.1
e_assoc.METClusterResolver.OutputLevel = 3
mu_assoc = JetSubStructure.MuonJetAssociatorTool(
"JSMuAssoc",
ContainerName='MuidMuonCollection',
UseTrack=False,
dR=myjetdr,
OutputLevel=3)
assoc_tools += [e_assoc, mu_assoc]
assoc_tool_ghost = JetSubStructure.GhostJetJetAssociatorTool(
"myGhostJetJetAssociatorTool",
JetdR=myjetdr,
JetFinder=myjetfinder,
JetInputTool=getStandardInputTools(input=myjetinput)[0],
OutputLevel=3)
## btgging ##
subjets_names = [x for x in catNames if "Subjets" in x]
btag_tool = []
if subjets_names != []:
from JetRec.JetGetters import getStandardBTaggerTool
jetFlags.doBTagging = True
_btag_tool = getStandardBTaggerTool('AntiKt', 0.4, 'LCTopo')
if len(_btag_tool) > 0:
btag_tool += [
JetSubStructure.JetAlgToolWrapper(
name='JetBTagger_' + ''.join(subjets_names),
JetAlgTool=_btag_tool[0],
GroomedjetsAuthor=subjets_names)
]
jetFlags.doBTagging = True
## btagging end ##
assoc_tool = JetSubStructure.JetAssociationTool(
name='JetAssocTool',
AssociatorTools=assoc_tools + [assoc_tool_ghost],
GroomingTools=myGroomingTools,
UsedSignalState=2,
Area=True,
JetAlgTools=grjets_calib + [wrapped_sortertool, widthtool] + btag_tool,
OutputLevel=5)
jetsubstructure_tool = JetSubStructure.JetSubstructureTool(
name='JetSubstructureTool',
UsedSignalState=2,
OutputLevel=3,
SubstructureTools=[
JetSubStructure.NSubjettinessTool(
"myNSubjettinessTool",
N=[1, 2, 3],
R0=myjetdr,
GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.KtHadSubstructureTool(
"myhadtool", GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.KtElectronSubstructureTool("myelectool",
GroomedjetsAuthor=[]),
JetSubStructure.KtMuonSubstructureTool("mymuontool",
GroomedjetsAuthor=[]),
JetSubStructure.DipolarityTool("myDipolarityTool",
GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.JetVoronoiAreaCalcTool(
"myVoronoiArea",
GroomedjetsAuthor=catNames_trimmed,
JetdR=myjetdr,
JetInputTool=getStandardInputTools(input=myjetinput)[0]),
JetSubStructure.JetShapesSubstructureTool(
"myJetShapesTool", GroomedjetsAuthor=catNames_trimmed),
JetSubStructure.JetPullVectorTool(
"myJetPullVectorTool", GroomedjetsAuthor=catNames_trimmed)
])
outkey = myjetfinder + '%d' % (myjetdr * 10) + myjetinput
if myjetinput == 'Topo':
outkey += 'EM'
outkey += 'Jets' #'JSJets' for rebuilding on the fly
myjetgetter = make_StandardJetGetter(myjetfinder,
myjetdr,
myjetinput,
seq=theseq,
doCalib=mydoCalib,
calibName=mycalibName,
addDefaultMoment=False,
outputCollectionName=outkey,
minPt=50 * GeV)
contname = myjetgetter.outputKey()
extra_moments_list = []
if myjetgetter.jetAlgorithmHandle(
) == None: # JetCollection already in AOD
if contname == None:
contname = myjetgetter.buildName()
#if contname == 'AntiKt4LCTopoJets':
# contname = 'AntiKt4LCTopoJetsReTagged'
else:
extra_moments_list += [widthtool]
# ------------------------------------------------
# Associate a vertex and set the JVF moment. Must be placed AFTER vertex correction
# ------------------------------------------------
from AthenaCommon.DetFlags import DetFlags
from AthenaCommon.AppMgr import ToolSvc
noVertex = jobproperties.Beam.beamType == 'cosmics' or jobproperties.Beam.beamType == 'singlebeam'
if DetFlags.detdescr.ID_on() and jetFlags.doJVF(
) and 'Truth' not in myjetinput and not noVertex and myjetdr >= 0.7:
from JetSubStructure.mySetupJetMomentTools import mygetJetVertexAssociationTool
myjvatool = mygetJetVertexAssociationTool(myjetfinder, myjetdr,
myjetinput,
'Custom%d' % (myjetdr * 10))
#add_jetMoments_to_JetAlg(myjetgetter, [myjvatool])
extra_moments_list += [myjvatool]
## Ghost Track Matching
from JetMomentTools.GhostAssociation import setupGhostAssociationTool
t = setupGhostAssociationTool(contname, ["TrackAssoc"])
t.GhostAssociators[0].AssociationName = "TrackAssocTmp"
t.CopyJetFindingMoment = False
extra_moments_list += [t]
make_JetMomentGetter(
contname, extra_moments_list + [assoc_tool, jetsubstructure_tool],
theseq)
## for lepton assoc
jetcont = contname[:-4] #removing the 'Jets'
if contname == 'AntiKt4LCTopoJetsReTagged' or contname == 'AntiKt4LCTopoNewJets':
jetcont = 'AntiKt4LCTopo'
for x in catNames_trimmed:
make_JetMomentGetter(jetcont + x + 'Jets', [
JetSubStructure.JetAssociationTool(
'JetAssocToolGr',
AssociatorTools=[e_assoc, mu_assoc],
GroomingTools=[],
UsedSignalState=2,
Area=True,
JetAlgTools=[],
OutputLevel=5),
JetSubStructure.JetSubstructureTool(
'JetSubstructureToolGr',
UsedSignalState=2,
SubstructureTools=[
JetSubStructure.KtElectronSubstructureTool(
"myelectool", GroomedjetsAuthor=[]),
JetSubStructure.KtMuonSubstructureTool(
"mymuontool", GroomedjetsAuthor=[])
])
], theseq)
jetFlags.doBTagging = True
return [contname, catNames]
def setupCaloJets(myjetfinder, myjetdr, myjetinput, GroomedDicts, theseq):
myGroomingTools, catNames = getGroomingTools(GroomedDicts)
sortertool = JetSorterTool('myJetSorter',
SortOrder="ByPtDown",
OutputLevel=3)
grjets_calib = []
## Origin correction / Eta-Mass-JES hack
from JetSubStructure.mySetupJetCalibrators import doEtaMassJESCorrection
from JetCalibTools.MakeCalibSequences import calibTags
calibTags['ETAMASSJES'] = doEtaMassJESCorrection
mydoCalib = False
mycalibName = ''
if 'LCTopo' in myjetinput:
mydoCalib = True
if (myjetfinder + '%d' %
(myjetdr * 10) == 'AntiKt10') or (myjetfinder + '%d' %
(myjetdr * 10) == 'CamKt12'):
mycalibName = 'LC:ORIGIN_ETAMASSJES'
else:
mycalibName = 'LC:ORIGIN'
stdcaliblist = []
allowedcalib = [
'AntiKt10LCTopoTrimmedJets',
'AntiKt10LCTopoTrimmedPtFrac3SmallR30Jets',
'AntiKt10LCTopoTrimmedPtFrac5SmallR30Jets',
'CamKt12LCTopoTrimmedPtFrac5SmallR30Jets'
]
for x in catNames:
author = myjetfinder + '%d' % (myjetdr *
10) + myjetinput + x + 'Jets'
if author in allowedcalib:
grjets_calib += [
JetSubStructure.JetAlgToolWrapper(
name='calibtool_origin_etamassjes_' + x,
JetAlgTool=getStandardCalibTool(
myjetfinder,
myjetdr,
myjetinput + x,
doCalib=True,
calibName='LC:ORIGIN_ETAMASSJES')[0],
GroomedjetsAuthor=[x])
]
else:
stdcaliblist += [x]
if stdcaliblist != []:
grjets_calib += [
JetSubStructure.JetAlgToolWrapper(
name='calibtool_origin',
JetAlgTool=getStandardCalibTool(myjetfinder,
myjetdr,
myjetinput,
doCalib=True,
calibName='LC:ORIGIN')[0],
GroomedjetsAuthor=stdcaliblist)
]
elif 'Topo' in myjetinput:
mydoCalib = True
mycalibName = 'EM:ORIGIN'
grjets_calib += getStandardCalibTool(myjetfinder,
myjetdr,
myjetinput,
doCalib=True,
calibName='EM:ORIGIN')
assoc_tools = []
if rec.doTruth():
truthcontname = myjetfinder + '%dTruthJets' % (myjetdr * 10)
if hasattr(theseq, truthcontname):
truthj_assoc = JetSubStructure.JetJetAssociatorTool(
"Truth_JetJetAssociatorTool",
ContainerName=truthcontname,
AssociationName="TruthJets",
MatchOrigin=False)
trackjetcontname = myjetfinder + '%dTrackZJets' % (myjetdr * 10)
if hasattr(theseq, trackjetcontname):
trkj_assoc = JetSubStructure.JetJetAssociatorTool(
"Trk_JetJetAssociatorTool",
ContainerName=trackjetcontname,
AssociationName="TrackJets",
MatchOrigin=True)
## batgging ##
subjets_names = [x for x in catNames if "Subjets" in x]
btag_tool = []
if subjets_names != '':
from JetRec.JetGetters import getStandardBTaggerTool
from JetRec.JetRecFlags import jetFlags
jetFlags.doBTagging = True
_btag_tool = getStandardBTaggerTool('AntiKt', 0.4, 'LCTopo')
if len(_btag_tool) > 0:
btag_tool += [
JetSubStructure.JetAlgToolWrapper(
name='JetBTagger_' + ''.join(subjets_names),
JetAlgTool=_btag_tool[0],
GroomedjetsAuthor=subjets_names)
]
jetFlags.doBTagging = False
## btagging end ##
assoc_tool = JetSubStructure.JetAssociationTool(
name='JetAssocTool',
AssociatorTools=assoc_tools,
GroomingTools=myGroomingTools,
UsedSignalState=2,
JetAlgTools=grjets_calib + [sortertool] + btag_tool,
OutputLevel=3)
myjetgetter = make_StandardJetGetter(myjetfinder,
myjetdr,
myjetinput,
seq=theseq,
doCalib=mydoCalib,
calibName=mycalibName,
addDefaultMoment=False)
contname = myjetgetter.outputKey()
# ------------------------------------------------
# Associate a vertex and set the JVF moment. Must be placed AFTER vertex correction
# ------------------------------------------------
from AthenaCommon.DetFlags import DetFlags
from AthenaCommon.AppMgr import ToolSvc
noVertex = jobproperties.Beam.beamType == 'cosmics' or jobproperties.Beam.beamType == 'singlebeam'
if DetFlags.detdescr.ID_on() and jetFlags.doJVF(
) and 'Truth' not in myjetinput and not noVertex:
from JetSubStructure.mySetupJetMomentTools import mygetJetVertexAssociationTool
myjvatool = mygetJetVertexAssociationTool(myjetfinder, myjetdr,
myjetinput,
'Custom%d' % (myjetdr * 10))
add_jetMoments_to_JetAlg(myjetgetter, [myjvatool])
make_JetMomentGetter(contname, [assoc_tool], theseq)
return [contname, catNames]
def JetVarToolConfig(seq=AlgSequence):
from RecExConfig.ObjKeyStore import cfgKeyStore
if (cfgKeyStore.isInInput('CaloCellContainer', 'AllCalo')
and cfgKeyStore.isInInput('CaloCellLinkContainer',
'CaloCalTopoCluster_Link')):
from JetRec.JetMomentGetter import make_JetMomentGetter
from JetRec.JetRecFlags import jetFlags
from CaloClusterCorrection.StandardCellWeightCalib import getCellWeightTool
cellcalibtool = getCellWeightTool('Kt',
0.4,
'Topo',
onlyCellWeight=True)
jetFlags.doJVF = False ##the following jet getter seems to fail if we don't deactivate the jet vertex finder.
#make_StandardJetGetter('AntiKt',0.4,'Topo',disable=False, doCalib=True, calibName='H1NumInv', outputCollectionName='AntiKt4TopoJets',seq=seq)
#make_StandardJetGetter('AntiKt',0.6,'Topo',disable=False, doCalib=True, calibName='H1NumInv', outputCollectionName='AntiKt6TopoJets',seq=seq)
# MB: April 19, 2011, Suggested fix from SS
make_StandardJetGetter('AntiKt',
0.4,
'Topo',
disable=False,
doCalib=True,
calibName='EM:H1NumInv',
outputCollectionName='AntiKt4TopoJets',
seq=seq)
make_StandardJetGetter('AntiKt',
0.6,
'Topo',
disable=False,
doCalib=True,
calibName='EM:H1NumInv',
outputCollectionName='AntiKt6TopoJets',
seq=seq)
# MB: April 7, 2011 : Suggested fix from SS
from JetMomentTools.JetMomentToolsConf import \
JetBadChanCorrTool, JetBadCellCorrTool, JetCaloQualityTool
atool4 = JetBadChanCorrTool("JetBadChanCorrTool_anti04",
ConeDr=0.4,
UseCalibScale=False)
atool4.CellCalibrator = cellcalibtool
atool6 = JetBadChanCorrTool("JetBadChanCorrTool_anti06",
ConeDr=0.6,
UseCalibScale=False)
atool6.CellCalibrator = cellcalibtool
#atool4 = JetBadChanCorrTool("JetBadChanCorrTool_anti04",UpdateEvery=False,ConeDr=0.4,UseCalibScale=False)
#atool4.CellCalibrator = cellcalibtool
#atool6 = JetBadChanCorrTool("JetBadChanCorrTool_anti06",UpdateEvery=False,ConeDr=0.6,UseCalibScale=False)
#atool6.CellCalibrator = cellcalibtool
cqual_tool = JetCaloQualityTool("CaloQualTool")
cqual_tool.doSamplingBasedVariables = True
cqual_tool.doCellBasedVariables = True
BadCellCorrT = JetBadCellCorrTool("JetBadCellCorrTool")
BadCellCorrT.CellCalibrator = cellcalibtool
##print BadCellCorrT
##toolist += [BadCellCorrT]
make_JetMomentGetter("AntiKt4TopoJets",
[atool4, cqual_tool, BadCellCorrT],
seq=seq)
make_JetMomentGetter("AntiKt6TopoJets",
[atool6, cqual_tool, BadCellCorrT],
seq=seq)
else:
from AthenaCommon.Logging import logging
log = logging.getLogger('MissingETGoodness')
log.warning('No cells; not rebuilding jets.')
###############################################
jet_collection_key = 'AntiKt4TopoJets'
if cfgKeyStore.isInInput('JetCollection', 'AntiKt4TopoEMJets'):
jet_collection_key = 'AntiKt4TopoEMJets'
truth_jet_collection_key = 'Cone4TruthJets'
if cfgKeyStore.isInInput('JetCollection', 'AntiKt4TruthJets'):
truth_jet_collection_key = 'AntiKt4TruthJets'
from MissingETGoodness.MissingETGoodnessConf import JetVarTool as ConfiguredJetVarTool
JetVarTool = ConfiguredJetVarTool(
JetCollectionKey=jet_collection_key,
JetTruthCollectionKey=truth_jet_collection_key,
TrackParticleContainerKey='TrackParticleCandidate',
)
from AthenaCommon.AppMgr import ToolSvc
ToolSvc += JetVarTool
print JetVarTool
def addGhostAssociation2(jetcollName, assocName, assocType, assocKeys="", finder = None,input=None, mainParam=None):
from JetRec.JetMomentGetter import make_JetMomentGetter
t = setupGhostAssociationTool( jetcollName, assocName, assocType, assocKeys, finder ,input, mainParam)
g=make_JetMomentGetter(jetcollName,[t])
return g.getJetMomentCalc()
def addGhostAssociation(jetcollName, ghostAssoc = [], finder = None,input=None, mainParam=None, seq=AlgSequence()):
from JetRec.JetMomentGetter import make_JetMomentGetter
t = setupGhostAssociationTool( jetcollName, ghostAssoc, finder ,input, mainParam, seq = seq)
g=make_JetMomentGetter(jetcollName,[t],seq=seq)
return g.getJetMomentCalc()
def myjets():
from JetRec.JetMomentGetter import make_JetMomentGetter
from JetMomentTools.SetupJetMomentTools import getJetVertexAssociationTool
jvatool = getJetVertexAssociationTool('AntiKt', 0.4, 'Topo')
make_JetMomentGetter('AntiKt4TopoEMJets', [jvatool])
make_JetMomentGetter('AntiKt6TopoEMJets', [jvatool])
