From ea4e37368e93cf41c7166f074feb1795f958e39c Mon Sep 17 00:00:00 2001
From: Sagar Addepalli <addepalli.sagar@cern.ch>
Date: Mon, 17 Jul 2023 17:09:43 +0200
Subject: [PATCH] Extra vertexing option in LLP1

Extra vertexing option in LLP1
---
 .../DerivationFrameworkLLP/python/LLP1.py     | 27 ++++++++++++++-----
 1 file changed, 21 insertions(+), 6 deletions(-)

diff --git a/PhysicsAnalysis/DerivationFramework/DerivationFrameworkLLP/python/LLP1.py b/PhysicsAnalysis/DerivationFramework/DerivationFrameworkLLP/python/LLP1.py
index 845d9865bf53..de8b1ff239c6 100644
--- a/PhysicsAnalysis/DerivationFramework/DerivationFrameworkLLP/python/LLP1.py
+++ b/PhysicsAnalysis/DerivationFramework/DerivationFrameworkLLP/python/LLP1.py
@@ -157,27 +157,42 @@ def LLP1KernelCfg(ConfigFlags, name='LLP1Kernel', **kwargs):
                             TrackFilterToolSTD      = LLP1TrackFilterToolSTD
                           ))
         acc.merge(VrtSecInclusiveCfg(ConfigFlags,
-                                     name = f"VrtSecInclusive_{TrackSystSuffix}",
+                                     name = f"VrtSecInclusive{TrackSystSuffix}",
                                      AugmentingVersionString  = TrackSystSuffix,
                                      FillIntermediateVertices = False,
                                      TrackLocation            = f"{MergedTrackCollection}{TrackSystSuffix}"))
         LLP1VrtSecInclusiveSuffixes.append(TrackSystSuffix)
 
     # leptons-only VSI
-    LeptonsModSuffix = "_LeptonsMod_LRTR3_1p0"
+    LeptonsSuffix = "_Leptons"
     acc.merge(VrtSecInclusiveCfg(ConfigFlags,
-                                 name = "VrtSecInclusive_InDet_"+LeptonsModSuffix,
-                                 AugmentingVersionString     = LeptonsModSuffix,
+                                 name = "VrtSecInclusive_InDet"+LeptonsSuffix,
+                                 AugmentingVersionString     = LeptonsSuffix,
                                  FillIntermediateVertices    = False,
                                  TrackLocation               = MergedTrackCollection,
                                  twoTrkVtxFormingD0Cut       = 1.0,
-                                 doSelectTracksWithLRTCuts   = True,
                                  doSelectTracksFromMuons     = True,
                                  doRemoveCaloTaggedMuons     = True,
                                  doSelectTracksFromElectrons = True,
                                  MuonLocation                = MergedMuonContainer,
                                  ElectronLocation            = MergedElectronContainer))
-    LLP1VrtSecInclusiveSuffixes.append(LeptonsModSuffix)
+    LLP1VrtSecInclusiveSuffixes.append(LeptonsSuffix)
+
+    # track VSI
+    LepTrackSuffix = "_LepTrack"
+    acc.merge(VrtSecInclusiveCfg(ConfigFlags,
+                                 name = "VrtSecInclusive_InDet"+LepTrackSuffix,
+                                 AugmentingVersionString     = LepTrackSuffix,
+                                 FillIntermediateVertices    = False,
+                                 TrackLocation               = MergedTrackCollection,
+                                 MuonLocation                = MergedMuonContainer,
+                                 ElectronLocation            = MergedElectronContainer,
+                                 twoTrkVtxFormingD0Cut       = 1.0,
+                                 doSelectIDAndGSFTracks      = True,
+                                 doRemoveCaloTaggedMuons     = True,
+                                 doRemoveNonLeptonVertices   = True,
+                                 doAssociateNonSelectedTracks= False))
+    LLP1VrtSecInclusiveSuffixes.append(LepTrackSuffix)
 
     # Thinning tools...
     from DerivationFrameworkInDet.InDetToolsConfig import TrackParticleThinningCfg, MuonTrackParticleThinningCfg, TauTrackParticleThinningCfg, DiTauTrackParticleThinningCfg, TauJetLepRMParticleThinningCfg
-- 
GitLab