From b4579c6a78b1b24ba7d233cd2e9dff937cc359cd Mon Sep 17 00:00:00 2001
From: Wouter Deconinck <wouter.deconinck@umanitoba.ca>
Date: Tue, 17 Aug 2021 23:18:11 +0000
Subject: [PATCH] Add MRICH digi/reco

---
 .../full/options/full_reconstruction.py       | 34 +++++++++++++++----
 1 file changed, 27 insertions(+), 7 deletions(-)

diff --git a/benchmarks/full/options/full_reconstruction.py b/benchmarks/full/options/full_reconstruction.py
index fdc81f03..89a2e5fd 100644
--- a/benchmarks/full/options/full_reconstruction.py
+++ b/benchmarks/full/options/full_reconstruction.py
@@ -98,6 +98,7 @@ sim_coll = [
     "VertexBarrelHits",
     "VertexEndcapHits",
     "DRICHHits"
+    "MRICHHits"
 ]
 
 # list of algorithms
@@ -535,18 +536,38 @@ parts_from_fit = ParticlesFromTrackFit("parts_from_fit",
         outputTrackParameters = "outputTrackParameters")
 algorithms.append(parts_from_fit)
 
-pmtdigi = PhotoMultiplierDigi("pmtdigi",
+# DRICH
+drich_digi = PhotoMultiplierDigi("drich_digi",
         inputHitCollection="DRICHHits",
         outputHitCollection="DRICHRawHits",
         quantumEfficiency=[(a*units.eV, b) for a, b in qe_data])
-algorithms.append(pmtdigi)
+algorithms.append(drich_digi)
 
-pmtreco = PhotoMultiplierReco("pmtreco",
-        inputHitCollection="DRICHHitsDigi",
-        outputHitCollection="DRICHHitsReco")
+drich_reco = PhotoMultiplierReco("drich_reco",
+        inputHitCollection=drich_digi.outputHitCollection,
+        outputHitCollection="DRICHRecHits")
+algorithms.append(drich_reco)
 
 # FIXME
-#richcluster = PhotoRingClusters("richcluster",
+#drich_cluster = PhotoRingClusters("drich_cluster",
+#        inputHitCollection=pmtreco.outputHitCollection,
+#        #inputTrackCollection="ReconstructedParticles",
+#        outputClusterCollection="ForwardRICHClusters")
+
+# MRICH
+mrich_digi = PhotoMultiplierDigi("mrich_digi",
+        inputHitCollection="MRICHHits",
+        outputHitCollection="MRICHRawHits",
+        quantumEfficiency=[(a*units.eV, b) for a, b in qe_data])
+algorithms.append(mrich_digi)
+
+mrich_reco = PhotoMultiplierReco("mrich_reco",
+        inputHitCollection=mrich_digi.outputHitCollection,
+        outputHitCollection="MRICHRecHits")
+algorithms.append(mrich_reco)
+
+# FIXME
+#mrich_cluster = PhotoRingClusters("drich_cluster",
 #        inputHitCollection=pmtreco.outputHitCollection,
 #        #inputTrackCollection="ReconstructedParticles",
 #        outputClusterCollection="ForwardRICHClusters")
@@ -561,7 +582,6 @@ podout.outputCommands = [
         "keep *Clusters",
         "keep *Layers",
         "drop InitTrackParams",
-        "drop mcparticles"
         ]
 algorithms.append(podout)
 
-- 
GitLab