From 81b103968cd303996df2c718e0616e6db316980d Mon Sep 17 00:00:00 2001
From: Leonhard Reichenbach <leonhard.reichenbach@cern.ch>
Date: Tue, 24 Oct 2023 17:19:20 +0200
Subject: [PATCH] add --trackingOnly flag

---
 CLDConfig/CLDReconstruction.py | 45 +++++++++++++++++++++-------------
 test/CMakeLists.txt            |  6 +++++
 2 files changed, 34 insertions(+), 17 deletions(-)

diff --git a/CLDConfig/CLDReconstruction.py b/CLDConfig/CLDReconstruction.py
index aae9bd8..a35bb0a 100644
--- a/CLDConfig/CLDReconstruction.py
+++ b/CLDConfig/CLDReconstruction.py
@@ -62,6 +62,7 @@
 from k4FWCore.parseArgs import parser
 parser.add_argument("--inputFiles", action="extend", nargs="+", metavar=("file1", "file2"), help="One or multiple input files")
 parser.add_argument("--outputBasename", help="Basename of the output file(s)", default=output_basename)
+parser.add_argument("--trackingOnly", action="store_true", help="Run only track reconstruction", default=False)
 my_opts = parser.parse_known_args()[0]
 
 output_basename = my_opts.outputBasename
@@ -1098,8 +1099,10 @@
                           }
 
 # TODO: put this somewhere else, needs to be in front of the output for now :(
+# setup AIDA histogramming and add eventual background overlay
 algList.append(MyAIDAProcessor)
 algList.append(Overlay[CONFIG["Overlay"]])
+# tracker hit digitisation
 algList.append(VXDBarrelDigitiser)
 algList.append(VXDEndcapDigitiser)
 algList.append(InnerPlanarDigiProcessor)
@@ -1107,6 +1110,7 @@
 algList.append(OuterPlanarDigiProcessor)
 algList.append(OuterEndcapPlanarDigiProcessor)
 
+# tracking
 if CONFIG["Tracking"] == "Truth":
     algList.append(MyTruthTrackFinder)
 elif CONFIG["Tracking"] == "Conformal":
@@ -1114,26 +1118,33 @@
     algList.append(ClonesAndSplitTracksFinder)
 
 algList.append(Refit)
-algList.append(MyDDCaloDigi[CONFIG["CalorimeterIntegrationTimeWindow"]])
-algList.append(MyDDSimpleMuonDigi)
-algList.append(MyDDMarlinPandora[CONFIG["CalorimeterIntegrationTimeWindow"]])
-algList.append(LumiCalReco)
+
+# calorimeter digitization and pandora
+if not my_opts.trackingOnly:
+    algList.append(MyDDCaloDigi[CONFIG["CalorimeterIntegrationTimeWindow"]])
+    algList.append(MyDDSimpleMuonDigi)
+    algList.append(MyDDMarlinPandora[CONFIG["CalorimeterIntegrationTimeWindow"]])
+    algList.append(LumiCalReco)
+# monitoring and Reco to MCTruth linking
 algList.append(MyClicEfficiencyCalculator)
 algList.append(MyRecoMCTruthLinker)
 algList.append(MyTrackChecker)
-algList.append(MyCLICPfoSelectorDefault)
-algList.append(MyCLICPfoSelectorLoose)
-algList.append(MyCLICPfoSelectorTight)
-
-if CONFIG["Overlay"] == "False":
-    algList.append(RenameCollection)
-else:
-    algList.append(MyFastJetProcessor)
-
-algList.append(VertexFinder)
-algList.append(JetClusteringAndRefiner)
-if CONFIG["VertexUnconstrained"] == "True":
-    algList.append(VertexFinderUnconstrained)
+# pfo selector (might need re-optimisation)
+if not my_opts.trackingOnly:
+    algList.append(MyCLICPfoSelectorDefault)
+    algList.append(MyCLICPfoSelectorLoose)
+    algList.append(MyCLICPfoSelectorTight)
+# misc.
+    if CONFIG["Overlay"] == "False":
+        algList.append(RenameCollection)
+    else:
+        algList.append(MyFastJetProcessor)
+
+    algList.append(VertexFinder)
+    algList.append(JetClusteringAndRefiner)
+    if CONFIG["VertexUnconstrained"] == "True":
+        algList.append(VertexFinderUnconstrained)
+# event number processor, down here to attach the conversion back to edm4hep to it
 algList.append(EventNumber)
 
 if CONFIG["OutputMode"] == "LCIO":
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 3bd9e04..7fe5c19 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -43,3 +43,9 @@ add_test(NAME edm4hep_input
          COMMAND k4run --inputFiles=test.edm4hep.root --outputBasename=rec_test_edm4hep CLDReconstruction.py --GeoSvc.detectors=${DETECTOR}
 )
 set_property(TEST edm4hep_input APPEND PROPERTY DEPENDS ddsim_edm4hep)
+
+add_test(NAME trkOnly
+         WORKING_DIRECTORY ${CLDConfig_DIR}
+         COMMAND k4run --trackingOnly --inputFiles=test.edm4hep.root --outputBasename=trkOnly_test_edm4hep CLDReconstruction.py --GeoSvc.detectors=${DETECTOR}
+)
+set_property(TEST trkOnly APPEND PROPERTY DEPENDS ddsim_edm4hep)