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 += 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")
centerofmassshapes = ToolSvc.centerofmassshapes
jetrec2.JetModifiers += [centerofmassshapes]
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import EnergyCorrelatorTool
ToolSvc += EnergyCorrelatorTool("energycorrelator")
energycorrelator = ToolSvc.energycorrelator
#energycorrelator.Beta = 0.5
#energycorrelator.IncludeECF4 = True
jetrec2.JetModifiers += [energycorrelator]
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import EnergyCorrelatorRatiosTool
ToolSvc += EnergyCorrelatorRatiosTool("energycorrelatorratios")
energycorrelatorratios = ToolSvc.energycorrelatorratios
#energycorrelatorratios.IncludeECF4 = True
jetrec2.JetModifiers += [energycorrelatorratios]
from JetSubStructureMomentTools.JetSubStructureMomentToolsConf import EnergyCorrelatorGeneralizedTool
ToolSvc += EnergyCorrelatorGeneralizedTool("energycorrelatorgeneralized")
energycorrelatorgeneralized = ToolSvc.energycorrelatorgeneralized
jetrec2.JetModifiers += [energycorrelatorgeneralized]
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)
# Generalized energy correlations
jtm += EnergyCorrelatorGeneralizedTool("energycorrelatorgeneralized")
# ... & their ratios
jtm += EnergyCorrelatorGeneralizedRatiosTool(
"energycorrelatorgeneralizedratios")
# COM shapes.
jtm += CenterOfMassShapesTool("comshapes")
# Jet pull
jtm += JetPullTool("pull", UseEtaInsteadOfY=False, IncludeTensorMoments=True)
# Jet charge