ALICE3: use track extension + allow configuring passes via JSON

Author: mpuccio

Detectors/Upgrades/ALICE3/GlobalReconstruction/workflow/README.md

@@ -81,12 +81,10 @@ The tracking configuration is provided via a JSON file that specifies:
     "SaveTimeBenchmarks": false,
     "DoUPCIteration": false,
     "FataliseUponFailure": true,
-    "UseTrackFollower": true,
-    "UseTrackFollowerTop": false,
-    "UseTrackFollowerBot": false,
-    "UseTrackFollowerMix": true,
+    "TrackFollower": "mix",
     "TrackFollowerNSigmaCutZ": 1.0,
     "TrackFollowerNSigmaCutPhi": 1.0,
+    "TrackFollowerBeamWidth": 1,
     "createArtefactLabels": false,
     "PrintMemory": false,
     "DropTFUponFailure": false

Detectors/Upgrades/ALICE3/GlobalReconstruction/workflow/src/TrackerSpec.cxx

@@ -96,6 +96,15 @@ std::vector<o2::its::TrackingParameters> TrackerDPL::createTrackingParamsFromCon
   auto loadTrackingParamsFromJson = [](std::vector<o2::its::TrackingParameters>& trackingParams, const nlohmann::json& paramConfigJson) {
     for (const auto& paramConfig : paramConfigJson) {
       o2::its::TrackingParameters params;
+      auto applyPassFlag = [&](const char* name, o2::its::IterationStep step) {
+        if (paramConfig.contains(name)) {
+          if (paramConfig[name].get<bool>()) {
+            params.PassFlags.set(step);
+          } else {
+            params.PassFlags.reset(step);
+          }
+        }
+      };
 
       if (paramConfig.contains("NLayers")) {
         params.NLayers = paramConfig["NLayers"].get<int>();
@@ -165,6 +174,37 @@ std::vector<o2::its::TrackingParameters> TrackerDPL::createTrackingParamsFromCon
       if (paramConfig.contains("CreateArtefactLabels")) {
         params.CreateArtefactLabels = paramConfig["CreateArtefactLabels"].get<bool>();
       }
+      if (paramConfig.contains("TrackFollower")) {
+        const auto mode = paramConfig["TrackFollower"].get<std::string>();
+        if (mode == "top" || mode == "outward") {
+          params.PassFlags.set(o2::its::IterationStep::TrackFollowerTop);
+        } else if (mode == "bot" || mode == "bottom" || mode == "inward") {
+          params.PassFlags.set(o2::its::IterationStep::TrackFollowerBot);
+        } else if (mode == "mix" || mode == "both") {
+          params.PassFlags.set(o2::its::IterationStep::TrackFollowerTop);
+          params.PassFlags.set(o2::its::IterationStep::TrackFollowerBot);
+        } else if (mode != "off") {
+          LOGP(fatal, "Invalid ALICE3 TRK tracking parameter TrackFollower: {}", mode);
+        }
+      }
+      if (paramConfig.contains("TrackFollowerNSigmaCutZ")) {
+        params.TrackFollowerNSigmaCutZ = paramConfig["TrackFollowerNSigmaCutZ"].get<float>();
+      }
+      if (paramConfig.contains("TrackFollowerNSigmaCutPhi")) {
+        params.TrackFollowerNSigmaCutPhi = paramConfig["TrackFollowerNSigmaCutPhi"].get<float>();
+      }
+      if (paramConfig.contains("TrackFollowerBeamWidth")) {
+        params.TrackFollowerBeamWidth = std::max(1, paramConfig["TrackFollowerBeamWidth"].get<int>());
+      }
+      applyPassFlag("FirstPass", o2::its::IterationStep::FirstPass);
+      applyPassFlag("RebuildClusterLUT", o2::its::IterationStep::RebuildClusterLUT);
+      applyPassFlag("UseUPCMask", o2::its::IterationStep::UseUPCMask);
+      applyPassFlag("SelectUPCVertices", o2::its::IterationStep::SelectUPCVertices);
+      applyPassFlag("ResetVertices", o2::its::IterationStep::ResetVertices);
+      applyPassFlag("SkipROFsAboveThreshold", o2::its::IterationStep::SkipROFsAboveThreshold);
+      applyPassFlag("MarkVerticesAsUPC", o2::its::IterationStep::MarkVerticesAsUPC);
+      applyPassFlag("TrackFollowerTop", o2::its::IterationStep::TrackFollowerTop);
+      applyPassFlag("TrackFollowerBot", o2::its::IterationStep::TrackFollowerBot);
       if (paramConfig.contains("PrintMemory")) {
         params.PrintMemory = paramConfig["PrintMemory"].get<bool>();
       }
@@ -282,6 +322,7 @@ void TrackerDPL::run(ProcessingContext& pc)
     trackerTraits.adoptTimeFrame(static_cast<o2::its::TimeFrame<11>*>(&timeFrame));
     itsTracker.adoptTimeFrame(timeFrame);
     trackerTraits.updateTrackingParameters(trackingParams);
+    timeFrame.initTrackerTopologies(trackingParams, 11);
 
     int nRofs{0};
     if (!mHitRecoConfig.empty()) {