From 4398ec62d712ac34dc9f41d8a86bb1e9cac5382b Mon Sep 17 00:00:00 2001
From: Wouter Deconinck <wouter.deconinck@umanitoba.ca>
Date: Fri, 29 Oct 2021 20:44:55 +0000
Subject: [PATCH] Resolve "Add Far Forward raw hit determination"

---
 options/reconstruction.py     | 51 +++++++++++++++++++++++++++++++++++
 options/reconstruction.raw.py | 24 +++++++++++++++++
 2 files changed, 75 insertions(+)

diff --git a/options/reconstruction.py b/options/reconstruction.py
index 1e9a2f09..49fd05cd 100644
--- a/options/reconstruction.py
+++ b/options/reconstruction.py
@@ -86,6 +86,8 @@ from Configurables import Jug__Digi__PhotoMultiplierDigi as PhotoMultiplierDigi
 from Configurables import Jug__Digi__CalorimeterHitDigi as CalHitDigi
 from Configurables import Jug__Digi__SiliconTrackerDigi as TrackerDigi
 
+from Configurables import Jug__Reco__FarForwardParticles as FarForwardParticles
+
 from Configurables import Jug__Reco__TrackerHitReconstruction as TrackerHitReconstruction
 from Configurables import Jug__Reco__TrackingHitsCollector2 as TrackingHitsCollector
 from Configurables import Jug__Reco__TrackerSourceLinker as TrackerSourceLinker
@@ -121,6 +123,9 @@ from Configurables import Jug__Reco__ParticleCollector as ParticleCollector
 # branches needed from simulation root file
 sim_coll = [
     'mcparticles',
+    'B0TrackerHits',
+    'ForwardRomanPotHits',
+    'ForwardOffMTrackerHits',
     'EcalEndcapNHits',
     'EcalEndcapPHits',
     'EcalBarrelHits',
@@ -161,6 +166,52 @@ truth_incl_kin = InclusiveKinematicsTruth("truth_incl_kin",
 )
 algorithms.append(truth_incl_kin)
 
+## Roman pots
+ffi_romanpot_digi = TrackerDigi("ffi_romanpot_digi",
+        inputHitCollection = "ForwardRomanPotHits",
+        outputHitCollection = "ForwardRomanPotRawHits",
+        timeResolution = 8)
+algorithms.append(ffi_romanpot_digi)
+
+ffi_romanpot_reco = TrackerHitReconstruction("ffi_romanpot_reco",
+        inputHitCollection = ffi_romanpot_digi.outputHitCollection,
+        outputHitCollection = "ForwardRomanPotRecHits")
+algorithms.append(ffi_romanpot_reco)
+
+ffi_romanpot_parts = FarForwardParticles("ffi_romanpot_parts",
+        inputCollection = ffi_romanpot_reco.outputHitCollection,
+        outputCollection = "ForwardRomanPotParticles")
+algorithms.append(ffi_romanpot_parts)
+
+## Off momentum tracker
+ffi_offmtracker_digi = TrackerDigi("ffi_offmtracker_digi",
+        inputHitCollection = "ForwardOffMTrackerHits",
+        outputHitCollection = "ForwardOffMTrackerRawHits",
+        timeResolution = 8)
+algorithms.append(ffi_offmtracker_digi)
+
+ffi_offmtracker_reco = TrackerHitReconstruction("ffi_offmtracker_reco",
+        inputHitCollection = ffi_offmtracker_digi.outputHitCollection,
+        outputHitCollection = "ForwardOffMTrackerRecHits")
+algorithms.append(ffi_offmtracker_reco)
+
+ffi_offmtracker_parts = FarForwardParticles("ffi_offmtracker_parts",
+        inputCollection = ffi_offmtracker_reco.outputHitCollection,
+        outputCollection = "ForwardOffMTrackerParticles")
+algorithms.append(ffi_offmtracker_parts)
+
+## B0 tracker
+trk_b0_digi = TrackerDigi("trk_b0_digi",
+        inputHitCollection="B0TrackerHits",
+        outputHitCollection="B0TrackerRawHits",
+        timeResolution=8)
+algorithms.append(trk_b0_digi)
+
+trk_b0_reco = TrackerHitReconstruction("trk_b0_reco",
+        inputHitCollection = trk_b0_digi.outputHitCollection,
+        outputHitCollection="B0TrackerRecHits")
+algorithms.append(trk_b0_reco)
+
 # Crystal Endcap Ecal
 ce_ecal_daq = dict(
         dynamicRangeADC=5.*units.GeV,
diff --git a/options/reconstruction.raw.py b/options/reconstruction.raw.py
index dc706826..a605e012 100644
--- a/options/reconstruction.raw.py
+++ b/options/reconstruction.raw.py
@@ -49,6 +49,9 @@ from Configurables import Jug__Digi__SiliconTrackerDigi as TrackerDigi
 # branches needed from simulation root file
 sim_coll = [
     'mcparticles',
+    'B0TrackerHits',
+    'ForwardRomanPotHits',
+    'ForwardOffMTrackerHits',
     'EcalEndcapNHits',
     'EcalEndcapPHits',
     'EcalBarrelHits',
@@ -75,6 +78,27 @@ algorithms = []
 podin = PodioInput("PodioReader", collections=sim_coll)
 algorithms.append(podin)
 
+## Roman pots
+ffi_romanpot_digi = TrackerDigi("ffi_romanpot_digi",
+        inputHitCollection = "ForwardRomanPotHits",
+        outputHitCollection = "ForwardRomanPotRawHits",
+        timeResolution = 8)
+algorithms.append(ffi_romanpot_digi)
+
+## Off momentum tracker
+ffi_offmtracker_digi = TrackerDigi("ffi_offmtracker_digi",
+        inputHitCollection = "ForwardOffMTrackerHits",
+        outputHitCollection = "ForwardOffMTrackerRawHits",
+        timeResolution = 8)
+algorithms.append(ffi_offmtracker_digi)
+
+## B0 tracker
+trk_b0_digi = TrackerDigi("trk_b0_digi",
+        inputHitCollection="B0TrackerHits",
+        outputHitCollection="B0TrackerRawHits",
+        timeResolution=8)
+algorithms.append(trk_b0_digi)
+
 # Crystal Endcap Ecal
 ce_ecal_daq = dict(
         dynamicRangeADC=5.*units.GeV,
-- 
GitLab