From f9371f0e7c4fd786fb6308395bd55f92882cedde 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

---
 .../full/options/full_reconstruction.py       | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/benchmarks/full/options/full_reconstruction.py b/benchmarks/full/options/full_reconstruction.py
index a247a5b2..19d20d31 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,11 @@ pmtreco = PhotoMultiplierReco("pmtreco",
 #        outputClusterCollection="ForwardRICHClusters")
 
 # Output
-podout.outputCommands = ["keep *",
-        "keep *Hits",
+podout.outputCommands = [
+        "keep *",
+        "drop *Digi",
+        "keep *Reco*",
+        "keep *ClusterHits",
         "keep *Clusters",
         "keep *Layers",
         "drop mcparticles"
-- 
GitLab