usePtCorrectedMass = True,
useCombinedIPNN = useNNcombinedWithIP3D,
calibrationTool=BTagCalibrationBrokerTool
)
ToolSvc += JetFitterCOMBNNTool
if BTaggingFlags.OutputLevel < 3:
print JetFitterCOMBNNTool
from JetTagTools.JetTagToolsConf import Analysis__JetFitterTag
JetFitterCOMBNN = Analysis__JetFitterTag( name = "JetFitterTagCOMBNN",
OutputLevel = BTaggingFlags.OutputLevel,
Runmodus = BTaggingFlags.Runmodus,
jetfitterNtupleWriter = JetFitterNtupleWriterCOMBNN,
jetfitterClassifier = JetFitterCOMBNNTool,
useForcedCalibration = False,
ipinfoTaggerName = "IP3D" if useNNcombinedWithIP3D else "",
SecVxFinderName="JetFitter",
xAODBaseName="JetFitterCombNN"
)
ToolSvc += JetFitterCOMBNN
JetFitterCOMBNN.jetCollectionList = []
for key in BTaggingFlags.Jets:
JetFitterCOMBNN.jetCollectionList += [key]
# if (key == 'Cone4Tower' or key == 'Cone7Tower'):
if BTaggingFlags.OutputLevel < 3:
print JetFitterCOMBNN
NeuralNetworkToHistoTool = NeuralNetworkToHistoNNTool,
useCombinedIPNN = useNNcombinedWithIP3D,
calibrationTool=BTagCalibrationBrokerTool
)
ToolSvc += JetFitterCOMBNNTool
if BTaggingFlags.OutputLevel < 3:
print JetFitterCOMBNNTool
from JetTagTools.JetTagToolsConf import Analysis__JetFitterTag
JetFitterIP3DPosCOMBNN = Analysis__JetFitterTag( name = "JetFitterTagCOMBNNIP3DPos",
OutputLevel = BTaggingFlags.OutputLevel,
Runmodus = BTaggingFlags.Runmodus,
jetfitterNtupleWriter = JetFitterNtupleWriterCOMBNN,
jetfitterClassifier = JetFitterCOMBNNTool,
jetfitterVariablesFactory = JetFitterVariablesFactoryCOMBNNIP3DPos,
useForcedCalibration = False,
supplementalTaggers = ['IP3DPos'] if useNNcombinedWithIP3D else [],
storeOnlyBaseObject = True
)
ToolSvc += JetFitterIP3DPosCOMBNN
JetFitterIP3DPosCOMBNN.jetCollectionList = []
for key in BTaggingFlags.Jets:
JetFitterIP3DPosCOMBNN.jetCollectionList += [key]
# if (key == 'Cone4Tower' or key == 'Cone7Tower'):
if BTaggingFlags.OutputLevel < 3:
print JetFitterIP3DPosCOMBNN
NeuralNetworkToHistoTool = NeuralNetworkToHistoNNTool,
useCombinedIPNN = useNNcombinedWithIP3D,
calibrationTool=BTagCalibrationBrokerTool
)
ToolSvc += JetFitterCOMBNNTool
if BTaggingFlags.OutputLevel < 3:
print JetFitterCOMBNNTool
from JetTagTools.JetTagToolsConf import Analysis__JetFitterTag
JetFitterIP3DNegCOMBNN = Analysis__JetFitterTag( name = "JetFitterTagCOMBNNIP3DNeg",
OutputLevel = BTaggingFlags.OutputLevel,
Runmodus = BTaggingFlags.Runmodus,
jetfitterNtupleWriter = JetFitterNtupleWriterCOMBNN,
jetfitterClassifier = JetFitterCOMBNNTool,
jetfitterVariablesFactory = JetFitterVariablesFactoryCOMBNNIP3DNeg,
useForcedCalibration = False,
supplementalTaggers = ["IP3DNeg"] if useNNcombinedWithIP3D else [],
storeOnlyBaseObject = True
)
ToolSvc += JetFitterIP3DNegCOMBNN
JetFitterIP3DNegCOMBNN.jetCollectionList = []
for key in BTaggingFlags.Jets:
JetFitterIP3DNegCOMBNN.jetCollectionList += [key]
# if (key == 'Cone4Tower' or key == 'Cone7Tower'):
if BTaggingFlags.OutputLevel < 3:
print JetFitterIP3DNegCOMBNN
CalibrationSubDirectory="NeuralNetwork",
OutputLevel=BTaggingFlags.OutputLevel,
NeuralNetworkToHistoTool=NeuralNetworkToHistoNNTool,
useCombinedIPNN=useNNcombinedWithIP3D,
calibrationTool=BTagCalibrationBrokerTool)
ToolSvc += JetFitterCOMBNNTool
if BTaggingFlags.OutputLevel < 3:
print JetFitterCOMBNNTool
from JetTagTools.JetTagToolsConf import Analysis__JetFitterTag
JetFitterIP3DNegCOMBNN = Analysis__JetFitterTag(
name="JetFitterTagCOMBNNIP3DNeg",
OutputLevel=BTaggingFlags.OutputLevel,
Runmodus=BTaggingFlags.Runmodus,
jetfitterNtupleWriter=JetFitterNtupleWriterCOMBNN,
jetfitterClassifier=JetFitterCOMBNNTool,
jetfitterVariablesFactory=JetFitterVariablesFactoryCOMBNNIP3DNeg,
useForcedCalibration=False,
supplementalTaggers=["IP3DNeg"] if useNNcombinedWithIP3D else [],
storeOnlyBaseObject=True)
ToolSvc += JetFitterIP3DNegCOMBNN
JetFitterIP3DNegCOMBNN.jetCollectionList = []
for key in BTaggingFlags.Jets:
JetFitterIP3DNegCOMBNN.jetCollectionList += [key]
# if (key == 'Cone4Tower' or key == 'Cone7Tower'):
if BTaggingFlags.OutputLevel < 3:
print JetFitterIP3DNegCOMBNN
CalibrationSubDirectory="NeuralNetwork",
OutputLevel=BTaggingFlags.OutputLevel,
NeuralNetworkToHistoTool=NeuralNetworkToHistoNNTool,
useCombinedIPNN=useNNcombinedWithIP3D,
calibrationTool=BTagCalibrationBrokerTool)
ToolSvc += JetFitterCOMBNNTool
if BTaggingFlags.OutputLevel < 3:
print JetFitterCOMBNNTool
from JetTagTools.JetTagToolsConf import Analysis__JetFitterTag
JetFitterIP3DPosCOMBNN = Analysis__JetFitterTag(
name="JetFitterTagCOMBNNIP3DPos",
OutputLevel=BTaggingFlags.OutputLevel,
Runmodus=BTaggingFlags.Runmodus,
jetfitterNtupleWriter=JetFitterNtupleWriterCOMBNN,
jetfitterClassifier=JetFitterCOMBNNTool,
jetfitterVariablesFactory=JetFitterVariablesFactoryCOMBNNIP3DPos,
useForcedCalibration=False,
supplementalTaggers=['IP3DPos'] if useNNcombinedWithIP3D else [],
storeOnlyBaseObject=True)
ToolSvc += JetFitterIP3DPosCOMBNN
JetFitterIP3DPosCOMBNN.jetCollectionList = []
for key in BTaggingFlags.Jets:
JetFitterIP3DPosCOMBNN.jetCollectionList += [key]
# if (key == 'Cone4Tower' or key == 'Cone7Tower'):
if BTaggingFlags.OutputLevel < 3:
print JetFitterIP3DPosCOMBNN
def get_jet_fitter_charm(jet_collection_list,
calibration_tool,
output_level=3,
runmodus='analysis',
name_extension='Charm',
vx_finder_name='NewJetFitterVxFinder',
output_threshold=3,
do_neg=False,
):
from AthenaCommon.AppMgr import ToolSvc
from JetTagTools.JetTagToolsConf import Analysis__JetFitterCharmVariablesFactory
VariablesFactory = Analysis__JetFitterCharmVariablesFactory(
name = "JetFitterVariablesFactory" + name_extension,
OutputLevel=output_level,
JetFitterInstance = "JetFitter" + name_extension,
secVxFinderName = vx_finder_name,
usePtCorrectedMass=True,
revertFromPositiveToNegativeTags=do_neg)
ToolSvc += VariablesFactory
if output_level < output_threshold:
print VariablesFactory
from JetTagTools.JetTagToolsConf import Analysis__JetFitterCharmNNTool
JetFitterCharmNNTool = Analysis__JetFitterCharmNNTool(
name = "JetFitter" + name_extension + "NNTool",
calibration_tool = calibration_tool,
calibration_directory = 'JetFitterCharm',
calibration_subdirectory = 'NeuralNetwork'
)
ToolSvc += JetFitterCharmNNTool
if output_level < output_threshold:
print JetFitterCharmNNTool
if do_neg:
supplemental_taggers = ['IP3DNeg','SV1Flip']
else:
supplemental_taggers = ['IP3D','SV1']
from JetTagTools.JetTagToolsConf import Analysis__JetFitterTag
JetFitterCharm = Analysis__JetFitterTag(
name = "JetFitter" + name_extension,
OutputLevel = output_level,
Runmodus = runmodus,
jetfitterClassifier = JetFitterCharmNNTool,
jetfitterVariablesFactory = VariablesFactory,
supplementalTaggers = supplemental_taggers,
useForcedCalibration = False,
svxTaggerName = 'SVInfoPlus',
)
ToolSvc += JetFitterCharm
JetFitterCharm.jetCollectionList = jet_collection_list
if output_level < output_threshold:
print JetFitterCharm
return JetFitterCharm
# ------ BTagging configuration:
redoBtag = True
if not 'BTaggingFlags' in dir():
from BTagging.BTaggingFlags import BTaggingFlags
if redoBtag:
BTaggingFlags.Runmodus = "analysis"
BTaggingFlags.PoolInputType = "AOD"
from ParticleBuilderOptions.AODFlags import AODFlags
AODFlags.TruthParticleJet = False
BTaggingFlags.OutputLevel = INFO
BTaggingFlags.Jets = ['Cone4H1Tower']
BTaggingFlags.JetsWithInfoPlus = ['Cone4H1Tower']
include( "BTagging/BTagging_jobOptions.py" )
from JetTagTools.JetTagToolsConf import Analysis__JetFitterTag
JetFitterTagTool = Analysis__JetFitterTag(name = "JetFitterTag")
JetFitterTagTool.jetCollectionList = ['Cone4H1Tower']
# ------ JetTagAna part:
include("AnalysisExamples/JetTagAna_jobOptions.py")
if redoBtag:
MyJetTagAna.JetContainer = 'Cone4H1TowerJetsAOD'
else:
MyJetTagAna.JetContainer = 'Cone4H1TowerJets'
MyJetTagAna.PrimaryVertexContainer = BTaggingFlags.PrimaryVertexCollectionName
MyJetTagAna.TrackParticleContainer = BTaggingFlags.TrackParticleCollectionName
print MyJetTagAna
# ------ MsgSvc:
ServiceMgr.MessageSvc.OutputLevel = ERROR
ServiceMgr.MessageSvc.defaultLimit = 9999999 # all messages
ServiceMgr.MessageSvc.Format = "% F%50W%S%7W%R%T %0W%M"
def get_gaia(
jet_collection_list,
calibration_tool,
output_level=3,
runmodus='analysis',
output_threshold=3,
do_neg=False,
):
from AthenaCommon.AppMgr import ToolSvc
from JetTagTools.JetTagToolsConf import Analysis__GaiaVariablesFactory
VariablesFactory = Analysis__GaiaVariablesFactory(
name="GaiaVariablesFactory",
OutputLevel=output_level,
GaiaName='GaiaNeg' if do_neg else 'Gaia',
JetFitterCharmName="JetFitterCharmNeg" if do_neg else "JetFitterCharm",
JetFitterName="JetFitterTagNNFlip" if do_neg else "JetFitterTagNN",
)
ToolSvc += VariablesFactory
if output_level < output_threshold:
print VariablesFactory
from JetTagTools.JetTagToolsConf import Analysis__GaiaNNTool
NNTool = Analysis__GaiaNNTool(
name="GaiaNNTool",
OutputLevel=output_level,
# OutputLevel=output_level, # 2 = debug, 3 = info
calibration_tool=calibration_tool,
calibration_directory='Gaia',
calibration_subdirectory='NeuralNetwork')
ToolSvc += NNTool
if output_level < output_threshold:
print NNTool
if do_neg:
supplemental_taggers = ['SV1Flip', 'SV0Flip']
multiweight_taggers = ['IP3DNeg'] # add IPFordG
else:
supplemental_taggers = ['SV1', 'SV0']
multiweight_taggers = ['IP3D'] #,'IPFordG']
ip3d_proxy = [1.0, 0.0, 0.0]
proxy_likelihoods = {'IP3D': ip3d_proxy, 'IP3DNeg': ip3d_proxy}
from JetTagTools.JetTagToolsConf import Analysis__JetFitterTag
Gaia = Analysis__JetFitterTag(
name="Gaia",
OutputLevel=output_level,
Runmodus=runmodus,
jetfitterClassifier=NNTool,
jetfitterVariablesFactory=VariablesFactory,
supplementalTaggers=supplemental_taggers,
multiweightSupplementalTaggers=multiweight_taggers,
useForcedCalibration=False,
svxTaggerName='SVInfoPlus',
proxyLikelihoods=proxy_likelihoods,
storeOnlyBaseObject=True,
)
ToolSvc += Gaia
Gaia.jetCollectionList = jet_collection_list
if output_level < output_threshold:
print Gaia
return Gaia
OutputLevel=BTaggingFlags.OutputLevel,
NeuralNetworkToHistoTool=NeuralNetworkToHistoNNTool,
usePtCorrectedMass=True,
useCombinedIPNN=useNNcombinedWithIP3D,
calibrationTool=BTagCalibrationBrokerTool)
ToolSvc += JetFitterCOMBNNTool
if BTaggingFlags.OutputLevel < 3:
print JetFitterCOMBNNTool
from JetTagTools.JetTagToolsConf import Analysis__JetFitterTag
JetFitterCOMBNN = Analysis__JetFitterTag(
name="JetFitterTagCOMBNN",
OutputLevel=BTaggingFlags.OutputLevel,
Runmodus=BTaggingFlags.Runmodus,
jetfitterNtupleWriter=JetFitterNtupleWriterCOMBNN,
jetfitterClassifier=JetFitterCOMBNNTool,
useForcedCalibration=False,
ipinfoTaggerName="IP3D" if useNNcombinedWithIP3D else "",
SecVxFinderName="JetFitter",
xAODBaseName="JetFitterCombNN")
ToolSvc += JetFitterCOMBNN
JetFitterCOMBNN.jetCollectionList = []
for key in BTaggingFlags.Jets:
JetFitterCOMBNN.jetCollectionList += [key]
# if (key == 'Cone4Tower' or key == 'Cone7Tower'):
if BTaggingFlags.OutputLevel < 3:
print JetFitterCOMBNN
