From 2720ecab04d01a276633c9e3164de63a5280700f Mon Sep 17 00:00:00 2001
From: Maarten Van Veghel <maarten.vanveghel@cern.ch>
Date: Thu, 5 Oct 2023 20:44:39 +0200
Subject: [PATCH] ghostprob config possibility for PbPb

---
 Hlt/RecoConf/python/RecoConf/hlt2_tracking.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/Hlt/RecoConf/python/RecoConf/hlt2_tracking.py b/Hlt/RecoConf/python/RecoConf/hlt2_tracking.py
index 05f7fbb0b54..9bac0229af9 100644
--- a/Hlt/RecoConf/python/RecoConf/hlt2_tracking.py
+++ b/Hlt/RecoConf/python/RecoConf/hlt2_tracking.py
@@ -589,7 +589,8 @@ def make_PrKalmanFilter_Seed_tracks(
 
 @configurable
 def get_UpgradeGhostId_tool(velo_hits=make_VPClus_hits,
-                            ut_hits=make_PrStorePrUTHits_hits):
+                            ut_hits=make_PrStorePrUTHits_hits,
+                            for_PbPb=False):
     """Returns instance of UpgradeGhostId given VP and UT hits.
 
     Args:
@@ -600,12 +601,15 @@ def get_UpgradeGhostId_tool(velo_hits=make_VPClus_hits,
         Instance of UpgradeGhostId
     """
     return UpgradeGhostId(
-        VPClusterLocation=velo_hits(), UTClusterLocation=ut_hits())
+        VPClusterLocation=velo_hits(),
+        UTClusterLocation=ut_hits(),
+        ForPbPb=for_PbPb)
 
 
 @configurable
 def get_UpgradeGhostId_tool_no_UT(velo_hits=make_VPClus_hits,
-                                  ut_hits=make_PrStorePrUTHits_empty_hits):
+                                  ut_hits=make_PrStorePrUTHits_empty_hits,
+                                  for_PbPb=False):
     """Returns instance of UpgradeGhostId given VP hits.
 
     Args:
@@ -616,7 +620,9 @@ def get_UpgradeGhostId_tool_no_UT(velo_hits=make_VPClus_hits,
         Instance of UpgradeGhostId
     """
     return UpgradeGhostId(
-        VPClusterLocation=velo_hits(), UTClusterLocation=ut_hits())
+        VPClusterLocation=velo_hits(),
+        UTClusterLocation=ut_hits(),
+        ForPbPb=for_PbPb)
 
 
 def get_default_hlt2_tracks():
-- 
GitLab