def declareDefaultTools():
from JetRecConfig.JetRecStandardToolManager import jtm
try:
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import ShowerDeconstructionTool
jtm.haveShowerDeconstructionTool = True
except ImportError:
jtm.haveShowerDeconstructionTool = False
#--------------------------------------------------------------
# Substructure moment builders.
#--------------------------------------------------------------
# Nsubjettiness
jtm += NSubjettinessTool(
"nsubjettiness",
Alpha = 1.0
)
# KtDR
jtm += KtDeltaRTool(
"ktdr",
JetRadius = 0.4
)
# Kt-splitter
jtm += KTSplittingScaleTool("ktsplitter")
# Angularity.
jtm += AngularityTool("angularity")
# Dipolarity.
jtm += DipolarityTool("dipolarity", SubJetRadius = 0.3)
# Planar flow.
jtm += PlanarFlowTool("planarflow")
# Kt mass drop.
jtm += KtMassDropTool("ktmassdrop")
# Energy correlations.
jtm += EnergyCorrelatorTool("encorr", Beta = 1.0)
# COM shapes.
jtm += CenterOfMassShapesTool("comshapes")
# Jet pull
jtm += JetPullTool(
"pull",
UseEtaInsteadOfY = False,
IncludeTensorMoments = True
)
# Jet charge
jtm += JetChargeTool("charge", K=1.0)
# Shower deconstruction.
if jtm.haveShowerDeconstructionTool:
jtm += ShowerDeconstructionTool("showerdec")
ToolSvc += KTSplittingScaleTool("ktsplittingscale")
ktsplittingscale = ToolSvc.ktsplittingscale
jetrec2.JetModifiers += [ktsplittingscale]
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import DipolarityTool
ToolSvc += DipolarityTool("dipolarity")
dipolarity = ToolSvc.dipolarity
jetrec2.JetModifiers += [dipolarity]
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import AngularityTool
ToolSvc += AngularityTool("angularity")
angularity = ToolSvc.angularity
jetrec2.JetModifiers += [angularity]
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import KtDeltaRTool
ToolSvc += KtDeltaRTool("ktdr")
ktdr = ToolSvc.ktdr
jetrec2.JetModifiers += [ktdr]
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import KtMassDropTool
ToolSvc += KtMassDropTool("ktmassdrop")
ktmassdrop = ToolSvc.ktmassdrop
jetrec2.JetModifiers += [ktmassdrop]
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import PlanarFlowTool
ToolSvc += PlanarFlowTool("planarflow")
planarflow = ToolSvc.planarflow
jetrec2.JetModifiers += [planarflow]
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import CenterOfMassShapesTool
ToolSvc += CenterOfMassShapesTool("centerofmassshapes")
apply_origin_correction=HIJetFlags.ApplyOriginCorrection())
]
###
#subtracted algorithms
#make main jets from unsubtr collections w/ same R, add modifiers for subtraction
unsubtr_suffix = HIJetFlags.UnsubtractedSuffix()
for k in jtm.jetrecs:
if unsubtr_suffix in k.name():
in_name = k.OutputContainer
out_name = in_name.replace("_%s" % unsubtr_suffix, "")
#>slight tweak in case R=1.0 jets are requestd, add some substructure tools
modifiers = GetHIModifierList(out_name, hi_tools)
if '10HIJets' in k.name():
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import KtDeltaRTool
jtm += KtDeltaRTool('ktdr10', JetRadius=1.0)
modifiers += [
jtm.ktdr10, jtm.ktsplitter, jtm.pull, jtm.angularity,
jtm.planarflow, jtm.ktmassdrop
]
copier = jtm.addJetCopier(out_name, in_name, modifiers, shallow=False)
AddToOutputList(out_name)
jtm.HIJetRecs += [copier]
AppendOutputList(jetFlags.jetAODList)
JetAlgFromTools(jtm.HIJetRecs, suffix="HI", persistify=True)
HIJetFlags.IteratedEventShapeKey = cell_level_shape_key
# code cloned from BTagging_jobOptions.py
# to allow b-tagging over HI jets
jtm += JetConstitFourMomTool("constitfourmom_pflow",
JetScaleNames=["DetectorEtaPhi"],
AltConstitColls=[""],
AltConstitScales=[0],
AltJetScales=["JetConstitScaleMomentum"])
#--------------------------------------------------------------
# Substructure moment builders.
#--------------------------------------------------------------
# Nsubjettiness
jtm += NSubjettinessTool("nsubjettiness", Alpha=1.0)
# KtDR
jtm += KtDeltaRTool("ktdr", JetRadius=0.4)
# Kt-splitter
jtm += KTSplittingScaleTool("ktsplitter")
# Angularity.
jtm += AngularityTool("angularity")
# Dipolarity.
jtm += DipolarityTool("dipolarity", SubJetRadius=0.3)
# Planar flow.
jtm += PlanarFlowTool("planarflow")
# Kt mass drop.
jtm += KtMassDropTool("ktmassdrop")