Skip to content
Snippets Groups Projects
Commit aefaf4cb authored by Adam Bailey's avatar Adam Bailey
Browse files

Added new flag to remove duplicate tau core tracks or not. Tidied a couple of...

Added new flag to remove duplicate tau core tracks or not. Tidied a couple of other flags in TauRecBuilder
parent 0dcef4a6
No related branches found
No related tags found
No related merge requests found
......@@ -56,13 +56,13 @@ class TauRecCoreBuilder ( TauRecConfigured ) :
# run vertex finder only in case vertexing is available. This check can also be done in TauAlgorithmsHolder instead doing it here.
from InDetRecExample.InDetJobProperties import InDetFlags
from tauRec.tauRecFlags import jobproperties
doMVATrackClassification = jobproperties.tauRecFlags.tauRecMVATrackClassification()
doRNNTrackClassification = jobproperties.tauRecFlags.tauRecRNNTrackClassification()
doMVATrackClassification = tauFlags.tauRecMVATrackClassification()
doRNNTrackClassification = tauFlags.tauRecRNNTrackClassification()
if tauFlags.isStandalone() or InDetFlags.doVertexFinding():
tools.append(taualgs.getTauVertexFinder(doUseTJVA=self.do_TJVA))
tools.append(taualgs.getTauAxis())
tools.append(taualgs.getTauTrackFinder(removeDuplicateTracks=(not doMVATrackClassification) ))
tools.append(taualgs.getTauTrackFinder(removeDuplicateTracks=(tauFlags.removeDuplicateCoreTracks() ) ))
if doMVATrackClassification : tools.append(taualgs.getTauTrackClassifier())
if not doMVATrackClassification and doRNNTrackClassification:
tools.append(taualgs.getTauTrackRNNClassifier())
......
......@@ -64,6 +64,13 @@ class useLargeD0Tracks(JobProperty):
allowedTypes=['bool']
StoredValue=False
class removeDuplicateCoreTracks(JobProperty):
"""Ignore core tracks already used in previous tau
"""
statusOn=True
allowedTypes=['bool']
StoredValue=True
class tauRecMVATrackClassification(JobProperty):
"""Run the MVA Track Classifier
"""
......@@ -229,7 +236,7 @@ class tauRecFlags(JobPropertyContainer):
jobproperties.add_Container(tauRecFlags)
# I want always the following flags in the Rec container
_list_tau=[Enabled,doTauRec,isStandalone,tauRecSeedJetCollection,tauRecToolsCVMFSPath,doTJVA,useLargeD0Tracks,tauRecMVATrackClassification,tauRecRNNTrackClassification,tauRecMVATrackClassificationConfig,tauRecRNNTrackClassificationConfig,tauRecDecayModeNNClassifierConfig,tauRecSeedMinPt,tauRecSeedMaxEta,tauRecMaxNTracks,tauRecToolsDevToolList,tauRecToolsDevToolListProcessor,doRunTauDiscriminant,useVertexBasedConvFinder,useNewPIDBasedConvFinder,doPanTau,doPi0,pi0EtCuts,pi0MVACuts_1prong,pi0MVACuts_mprong,shotPtCut_1Photon,shotPtCut_2Photons,useOldVertexFitterAPI,useSubtractedCluster]
_list_tau=[Enabled,doTauRec,isStandalone,tauRecSeedJetCollection,tauRecToolsCVMFSPath,doTJVA,useLargeD0Tracks,removeDuplicateCoreTracks,tauRecMVATrackClassification,tauRecRNNTrackClassification,tauRecMVATrackClassificationConfig,tauRecRNNTrackClassificationConfig,tauRecDecayModeNNClassifierConfig,tauRecSeedMinPt,tauRecSeedMaxEta,tauRecMaxNTracks,tauRecToolsDevToolList,tauRecToolsDevToolListProcessor,doRunTauDiscriminant,useVertexBasedConvFinder,useNewPIDBasedConvFinder,doPanTau,doPi0,pi0EtCuts,pi0MVACuts_1prong,pi0MVACuts_mprong,shotPtCut_1Photon,shotPtCut_2Photons,useOldVertexFitterAPI,useSubtractedCluster]
for j in _list_tau:
jobproperties.tauRecFlags.add_JobProperty(j)
del _list_tau
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment