def toolNewJetFitterVariablesFactory(name,
useBTagFlagsDefaults=True,
**options):
"""Sets up a NewJetFitterVariablesFactory tool and returns it.
The following options have BTaggingFlags defaults:
OutputLevel default: BTaggingFlags.OutputLevel
JetFitterInstance default: "JetFitterTag"
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,
'JetFitterInstance': 'JetFitterTag'
}
# 'usePtCorrectedMass' : True }
for option in defaults:
options.setdefault(option, defaults[option])
options['name'] = name
from JetTagTools.JetTagToolsConf import Analysis__JetFitterVariablesFactory
return Analysis__JetFitterVariablesFactory(**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)
ToolSvc += NewJetFitterLikelihood
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)
ToolSvc += NeuralNetworkToHistoCOMBNNTool
if BTaggingFlags.OutputLevel < 3:
Mode3dFinder=mode3dfinder,
SequentialVertexFitter=sequentialVtxFitter,
VxPrimaryContainer="PrimaryVertices",
# VxPrimaryContainer=InDetKeys.PrimaryVertices(),
VertexEdmFactory=InDetVxIntEdmCnv,
OutputLevel = BTaggingFlags.OutputLevel,
MaxNumDeleteIterations = 30,
VertexProbCut = 0.001,
MaxClusteringIterations = 30,
VertexClusteringProbabilityCut = 0.005)
ToolSvc += NewJetFitterVxFinder
if BTaggingFlags.OutputLevel < 3:
print NewJetFitterVxFinder
#JetFitterVxFinder.MaxTracksToFitAtOnce = 3000
#to be added in bugfix version
##and the factory for BTagSecVertexing
from JetTagTools.JetTagToolsConf import Analysis__JetFitterVariablesFactory
JetFitterVariablesFactory = Analysis__JetFitterVariablesFactory( name = "JetFitterVariablesFactory",
OutputLevel = BTaggingFlags.OutputLevel,
secVxFinderName = "JetFitterVxFinder",
JetFitterInstance = "JetFitterTag")
ToolSvc += JetFitterVariablesFactory
if BTaggingFlags.OutputLevel < 3:
print JetFitterVariablesFactory
from AthenaCommon.AppMgr import ToolSvc
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
NewJetFitterVariablesFactory = Analysis__JetFitterVariablesFactory(
name="NewJetFitterVariablesFactory",
OutputLevel=BTaggingFlags.OutputLevel,
JetFitterInstance="JetFitterTag",
usePtCorrectedMass=True)
ToolSvc += NewJetFitterVariablesFactory
if BTaggingFlags.OutputLevel < 3:
print NewJetFitterVariablesFactory
from JetTagTools.JetTagToolsConf import Analysis__JetFitterLikelihood
NewJetFitterLikelihood = Analysis__JetFitterLikelihood(
name="NewJetFitterLikelihood",
CalibrationDirectory="JetFitter",
CalibrationSubDirectory="RooFitLikelihood",
calibrationTool=BTagCalibrationBrokerTool,
OutputLevel=BTaggingFlags.OutputLevel)
ToolSvc += NewJetFitterLikelihood
if BTaggingFlags.OutputLevel < 3:
print NewJetFitterLikelihood
