def toolNeuralNetworkToHistoToolNN(name, useBTagFlagsDefaults=True, **options):
"""Sets up a NeuralNetworkToHistoToolNN 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 TrkNeuralNetworkUtils.TrkNeuralNetworkUtilsConf import Trk__NeuralNetworkToHistoTool
return Trk__NeuralNetworkToHistoTool(**options)
if hasattr(ToolSvc, 'InDetRotCreator'):
egRotCreator = ToolSvc.InDetRotCreator
else:
#Setup e/gamma offline RotCreator if one is not present
if InDetFlags.doPixelClusterSplitting(
) and InDetFlags.pixelClusterSplittingType() == 'NeuralNet':
# --- temp: read calib file
from AthenaCommon.AppMgr import ServiceMgr as svcMgr
if not hasattr(svcMgr, 'THistSvc'):
from GaudiSvc.GaudiSvcConf import THistSvc
svcMgr += THistSvc()
# --- neutral network tools
import sys, os
from TrkNeuralNetworkUtils.TrkNeuralNetworkUtilsConf import Trk__NeuralNetworkToHistoTool
egNeuralNetworkToHistoTool = Trk__NeuralNetworkToHistoTool(
name="egNeuralNetworkToHistoTool")
ToolSvc += egNeuralNetworkToHistoTool
#--- new NN factor
# COOL binding
from IOVDbSvc.CondDB import conddb
conddb.addFolder("PIXEL_OFL",
"/PIXEL/PixelClustering/PixelClusNNCalib")
# --- Select the necessary settings when running on run 1 data/MC
# --- since a correction is needed to fix biases when running on new run 2 compatible calibation
from SiClusterizationTool.SiClusterizationToolConf import InDet__NnClusterizationFactory
if not "R2" in globalflags.DetDescrVersion(
) and not "IBL3D25" in globalflags.DetDescrVersion():
egNnClusterizationFactory = InDet__NnClusterizationFactory(
name="egNnClusterizationFactory",
useNNcombinedWithIP3D=False
from AthenaCommon.AppMgr import ToolSvc
from JetTagTools.JetTagToolsConf import Analysis__JetFitterNtupleWriter
JetFitterNtupleWriterNN = Analysis__JetFitterNtupleWriter( name = "JetFitterNtupleWriterNN",
OutputLevel = BTaggingFlags.OutputLevel)
ToolSvc += JetFitterNtupleWriterNN
if BTaggingFlags.OutputLevel < 3:
print JetFitterNtupleWriterNN
from TrkNeuralNetworkUtils.TrkNeuralNetworkUtilsConf import Trk__NeuralNetworkToHistoTool
NeuralNetworkToHistoNNTool = Trk__NeuralNetworkToHistoTool ( name = "NeuralNetworkToHistoToolNN",
OutputLevel = BTaggingFlags.OutputLevel)
ToolSvc += NeuralNetworkToHistoNNTool
if BTaggingFlags.OutputLevel < 3:
print NeuralNetworkToHistoNNTool
from JetTagTools.JetTagToolsConf import Analysis__JetFitterNNTool
JetFitterNNTool = Analysis__JetFitterNNTool( name = "JetFitterNNTool",
CalibrationDirectory="JetFitter",
CalibrationSubDirectory="NeuralNetwork",
OutputLevel = BTaggingFlags.OutputLevel,
NeuralNetworkToHistoTool = NeuralNetworkToHistoNNTool,
useCombinedIPNN = useNNcombinedWithIP3D,
usePtCorrectedMass = True,
calibrationTool=BTagCalibrationBrokerTool
)
