From 28bcdb0b56487cd7f8eb2f3a18e52cef7d5f5544 Mon Sep 17 00:00:00 2001
From: Wouter Deconinck <wouter.deconinck@umanitoba.ca>
Date: Tue, 17 Aug 2021 19:32:45 +0000
Subject: [PATCH] Add gem to tracking

---
 benchmarks/full/options/full_reconstruction.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/benchmarks/full/options/full_reconstruction.py b/benchmarks/full/options/full_reconstruction.py
index 91c9b24a..31a9a3b7 100644
--- a/benchmarks/full/options/full_reconstruction.py
+++ b/benchmarks/full/options/full_reconstruction.py
@@ -446,9 +446,19 @@ gem_ec_reco = TrackerHitReconstruction("gem_ec_digi",
         outputHitCollection="GEMTrackerEndcapRecHits",
         timeResolution=10)
 
+# Tracking hit collector
+trk_hit_col = TrackingHitsCollector("trk_hit_col",
+        inputTrackingHits=[
+            str(trk_b_reco.outputHitCollection),
+            str(trk_ec_reco.outputHitCollection),
+            str(vtx_b_reco.outputHitCollection),
+            str(vtx_ec_reco.outputHitCollection),
+            str(gem_ec_reco.outputHitCollection) ],
+        trackingHits="trackingHits")
+
 # Hit Source linker
 sourcelinker = TrackerSourcesLinker("trk_srcslnkr",
-        inputHitCollections = ["VertexBarrelRecHits", "TrackerBarrelRecHits"],
+        inputHitCollections = trk_hit_col.trackingHits,
         outputSourceLinks = "TrackerSourceLinks",
         outputMeasurements = "TrackerMeasurements")
 
@@ -524,8 +534,9 @@ pmtreco = PhotoMultiplierReco("pmtreco",
 #        outputClusterCollection="ForwardRICHClusters")
 
 # Output
-podout.outputCommands = ["keep *",
-        "keep *Digi",
+podout.outputCommands = [
+        "keep *",
+        "drop *Digi",
         "keep *Reco*",
         "keep *ClusterHits",
         "keep *Clusters",
-- 
GitLab