SPecs

Author: njacazio

Detectors/Upgrades/ALICE3/GlobalReconstruction/reconstruction/CMakeLists.txt

@@ -41,5 +41,6 @@ o2_add_library(ALICE3GlobalReconstruction
                        TBB::tbb)
 
 if(Acts_FOUND)
+  target_sources(${targetName} PRIVATE src/TrackerACTS.cxx)
   target_compile_definitions(${targetName} PUBLIC O2_WITH_ACTS)
 endif()

Detectors/Upgrades/ALICE3/GlobalReconstruction/reconstruction/include/ALICE3GlobalReconstruction/TrackerACTS.h

@@ -21,7 +21,8 @@
 
 #include <Acts/Definitions/Units.hpp>
 #include <Framework/Logger.h>
-#include <DataFormatsITSMFT/CompCluster.h>
+#include <array>
+#include <vector>
 #include <ITStracking/TimeFrame.h>
 #include <TH2F.h>
 
@@ -104,10 +105,9 @@ class TrackerACTS
 
   /// Adopt a TimeFrame for processing
   void adoptTimeFrame(o2::its::TimeFrame<nLayers>& tf);
-
-  void adoptClusters(gsl::span<const o2::itsmft::CompClusterExt> clusters)
+  void adoptSensorIDs(const std::array<std::vector<int>, nLayers>& sensorIDs)
   {
-    mInputClusters = clusters;
+    mSensorIDs = &sensorIDs;
   }
 
   /// Main tracking entry point: convert clusters to tracks
@@ -136,9 +136,6 @@ class TrackerACTS
   /// Create seeds (triplets) from space points using ACTS SeedFinder
   void createSeeds();
 
-  /// Estimate track parameters from a seed using ACTS
-  bool estimateTrackParams(const SeedACTS& seed, o2::its::TrackITSExt& track) const;
-
   /// Run track finding from seeds
   void findTracks();
 
@@ -166,7 +163,7 @@ class TrackerACTS
   float mBz{0.5f}; ///< Magnetic field in Tesla
   unsigned int mTimeFrameCounter{0};
   double mTotalTime{0.};
-  gsl::span<const o2::itsmft::CompClusterExt> mInputClusters;
+  const std::array<std::vector<int>, nLayers>* mSensorIDs = nullptr;
 
   // Tracking states for logging
   enum State {

Detectors/Upgrades/ALICE3/GlobalReconstruction/workflow/include/ALICE3GlobalReconstructionWorkflow/TrackerSpecImpl.h

@@ -22,6 +22,9 @@
 #include "Field/MagneticField.h"
 #include "Framework/ControlService.h"
 #include "ITStracking/Tracker.h"
+#ifdef O2_WITH_ACTS
+#include "ALICE3GlobalReconstruction/TrackerACTS.h"
+#endif
 #include "SimulationDataFormat/MCCompLabel.h"
 #include "SimulationDataFormat/MCEventHeader.h"
 #include "SimulationDataFormat/MCTruthContainer.h"
@@ -114,8 +117,22 @@ void TrackerDPL::runTracking(framework::ProcessingContext& pc, TimeFrameT& timeF
     nRofs = timeFrame.loadROFrameData(layerROFs, layerClusters, layerPatterns, mIsMC ? &layerLabels : nullptr, yPlaneMLOT);
     timeFrame.addTruthSeedingVertices();
   }
-
   const auto trackingLoopStart = std::chrono::steady_clock::now();
+#ifdef O2_WITH_ACTS
+  if (mUseACTS) {
+    LOG(info) << "Running ACTS tracking";
+    o2::trk::TrackerACTS<11> actsTracker;
+    actsTracker.adoptTimeFrame(timeFrame);
+    actsTracker.adoptSensorIDs(timeFrame.getClusterSensorIDs());
+
+    const auto& config = mHitRecoConfig.empty() ? mClusterRecoConfig : mHitRecoConfig;
+    // The input magnetic field is in kG; ACTS TrackerACTS expects Tesla.
+    actsTracker.setBz(config["geometry"]["bz"].get<float>() * 0.1f);
+    actsTracker.clustersToTracks();
+    actsTracker.printSummary();
+  } else
+#endif
+  {
   for (size_t iter{0}; iter < mTrackingParams.size(); ++iter) {
     LOGP(info, "{}", mTrackingParams[iter].asString());
     trackerTraits.initialiseTimeFrame(iter);