def toolJetFitterNtupleWriter(name, useBTagFlagsDefaults = True, **options):
"""Sets up a JetFitterNtupleWriter tool and returns it.
The following options have BTaggingFlags defaults:
OutputLevel default: BTaggingFlags.OutputLevel
input: name: The name of the tool (should be unique).
useBTagFlagsDefaults : Whether to use BTaggingFlags defaults for options that are not specified.
**options: Python dictionary with options for the tool.
output: The actual tool, which can then by added to ToolSvc via ToolSvc += output."""
if useBTagFlagsDefaults:
defaults = { 'OutputLevel' : BTaggingFlags.OutputLevel }
for option in defaults:
options.setdefault(option, defaults[option])
options['name'] = name
from JetTagTools.JetTagToolsConf import Analysis__JetFitterNtupleWriter
return Analysis__JetFitterNtupleWriter(**options)
from AthenaCommon.AppMgr import ToolSvc
if (not BTaggingFlags.JetFitterTag):
from JetTagTools.JetTagToolsConf import Analysis__JetFitterNtupleWriter
NewJetFitterNtupleWriter = Analysis__JetFitterNtupleWriter(
name="NewJetFitterNtupleWriter", OutputLevel=BTaggingFlags.OutputLevel)
ToolSvc += NewJetFitterNtupleWriter
if BTaggingFlags.OutputLevel < 3:
print NewJetFitterNtupleWriter
from JetTagTools.JetTagToolsConf import Analysis__JetFitterVariablesFactory
NewJetFitterVariablesFactoryFlip = Analysis__JetFitterVariablesFactory(
name="NewJetFitterVariablesFactoryFlip",
OutputLevel=BTaggingFlags.OutputLevel,
JetFitterInstance="JetFitterTagFlip",
usePtCorrectedMass=True,
revertFromPositiveToNegativeTags=True)
ToolSvc += NewJetFitterVariablesFactoryFlip
if BTaggingFlags.OutputLevel < 3:
print NewJetFitterVariablesFactoryFlip
if (not BTaggingFlags.JetFitterTag):
from JetTagTools.JetTagToolsConf import Analysis__JetFitterLikelihood
NewJetFitterLikelihood = Analysis__JetFitterLikelihood(
name="NewJetFitterLikelihood",
CalibrationDirectory="JetFitter",
CalibrationSubDirectory="RooFitLikelihood",
calibrationTool=BTagCalibrationBrokerTool,
OutputLevel=BTaggingFlags.OutputLevel)
useNNcombinedWithIP3D = True
from AthenaCommon.AppMgr import ToolSvc
if not BTaggingFlags.JetFitterNN:
from JetTagTools.JetTagToolsConf import Analysis__JetFitterNtupleWriter
JetFitterNtupleWriterCOMBNN = Analysis__JetFitterNtupleWriter(
name="JetFitterNtupleWriterCOMBNN",
OutputLevel=BTaggingFlags.OutputLevel)
ToolSvc += JetFitterNtupleWriterCOMBNN
if BTaggingFlags.OutputLevel < 3:
print JetFitterNtupleWriterCOMBNN
from JetTagTools.JetTagToolsConf import Analysis__JetFitterVariablesFactory
JetFitterVariablesFactoryCOMBNNIP3DNeg = Analysis__JetFitterVariablesFactory(
name="JetFitterVariablesFactoryCOMBNNIP3DNeg",
OutputLevel=BTaggingFlags.OutputLevel,
JetFitterInstance="JetFitterCOMBNNIP3DNeg",
secVxFinderName="NewJetFitterVxFinderFlip",
usePtCorrectedMass=True,
revertFromPositiveToNegativeTags=True)
ToolSvc += JetFitterVariablesFactoryCOMBNNIP3DNeg
if BTaggingFlags.OutputLevel < 3:
print JetFitterVariablesFactoryCOMBNNIP3DNeg
if not BTaggingFlags.JetFitterNN:
from TrkNeuralNetworkUtils.TrkNeuralNetworkUtilsConf import Trk__NeuralNetworkToHistoTool
NeuralNetworkToHistoCOMBNNTool = Trk__NeuralNetworkToHistoTool(
name="NeuralNetworkToHistoToolCOMBNN",
OutputLevel=BTaggingFlags.OutputLevel)
