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