Additional far forward benchmarks

benchmarks/far_forward/options/far_forward_reconstruction.py

@@ -28,39 +28,77 @@ from Configurables import Jug__Reco__TrackFindingAlgorithm as TrackFindingAlgori
 from Configurables import Jug__Reco__ParticlesFromTrackFit as ParticlesFromTrackFit
 from Configurables import Jug__Base__InputCopier_dd4pod__Geant4ParticleCollection_dd4pod__Geant4ParticleCollection_ as MCCopier
 
+from Configurables import Jug__Reco__TrackerHitReconstruction as TrackerHitReconstruction
 
 sim_colls = [
     "mcparticles",
-    "B0TrackerHits"
+    "B0TrackerHits",
+    "ForwardRomanPotHits",
+    "ForwardOffMTrackerHits"
 ]
+
+# list of algorithms
+algorithms = []
+
+# input
 podin = PodioInput("PodioReader", collections=sim_colls)
-podout = PodioOutput("out", filename=output_rec)
+algorithms.append(podin)
 
 ## copiers to get around input --> output copy bug. Note the "2" appended to the output collection.
 mccopier = MCCopier("MCCopier",
         inputCollection="mcparticles",
         outputCollection="mcparticles2")
-
+algorithms.append(mccopier)
+
+## 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)
+
+## 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 = "ForwardOffMTrackerRedHits")
+algorithms.append(ffi_offmtracker_reco)
+
+## B0 tracker
 trk_b0_digi = TrackerDigi("trk_b0_digi",
         inputHitCollection="B0TrackerHits",
         outputHitCollection="B0TrackerRawHits",
         timeResolution=8)
+algorithms.append(trk_b0_digi)
 
 trk_b0_reco = TrackerReco("trk_b_reco",
         inputHitCollection = trk_b0_digi.outputHitCollection,
         outputHitCollection="B0TrackerRecHits")
+algorithms.append(trk_b0_reco)
 
 sourcelinker = TrackerSourcesLinker("trk_srcslnkr",
-        inputHitCollections=["B0TrackerRecHits"],
+        inputHitCollections=[trk_b0_reco.outputHitCollection],
         outputSourceLinks="B0TrackerSourceLinks",
         outputMeasurements="B0TrackerMeasurements",
         OutputLevel=INFO)
+algorithms.append(sourcelinker)
 
 ## Track param init
 truth_trk_init = TrackParamTruthInit("truth_trk_init",
         inputMCParticles="mcparticles",
         outputInitialTrackParameters="InitTrackParams",
         OutputLevel=INFO)
+algorithms.append(truth_trk_init)
 
 # Tracking algorithms
 trk_find_alg = TrackFindingAlgorithm("trk_find_alg",
@@ -69,13 +107,17 @@ trk_find_alg = TrackFindingAlgorithm("trk_find_alg",
         inputInitialTrackParameters= truth_trk_init.outputInitialTrackParameters,
         outputTrajectories="trajectories",
         OutputLevel=INFO)
+algorithms.append(trk_find_alg)
 
 parts_from_fit = ParticlesFromTrackFit("parts_from_fit",
         inputTrajectories=trk_find_alg.outputTrajectories,
         outputParticles="ReconstructedParticles",
         outputTrackParameters="outputTrackParameters",
         OutputLevel=INFO)
+algorithms.append(parts_from_fit)
 
+# output
+podout = PodioOutput("out", filename=output_rec)
 podout.outputCommands = [
     "keep *",
     "drop InitTrackParams",
@@ -84,19 +126,12 @@ podout.outputCommands = [
     "drop outputInitialTrackParameters",
     "drop mcparticles",
 ]
+algorithms.append(podout)
 
 ApplicationMgr(
-    TopAlg = [podin, mccopier,
-              trk_b0_digi,
-              trk_b0_reco,
-              sourcelinker,
-              truth_trk_init,
-              trk_find_alg,
-              parts_from_fit,
-              podout
-              ],
+    TopAlg = algorithms,
     EvtSel = 'NONE',
-    EvtMax   = n_events,
+    EvtMax = n_events,
     ExtSvc = [podioevent, geo_service],
     OutputLevel=WARNING
  )