[ALICE3] TRK: allow ACTS clustering

Author: njacazio

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

@@ -9,10 +9,16 @@
 # granted to it by virtue of its status as an Intergovernmental Organization
 # or submit itself to any jurisdiction.
 
+if(Acts_FOUND)
+  set(actsTarget Acts::Core)
+endif()
+
 o2_add_library(TRKReconstruction
                TARGETVARNAME targetName
                SOURCES src/TimeFrame.cxx
                        src/Clusterer.cxx
+                       $<$<BOOL:${Acts_FOUND}>:src/ClustererACTS.cxx>
+                       $<$<BOOL:${Acts_FOUND}>:src/TrackerACTS.cxx>
                PUBLIC_LINK_LIBRARIES
                        O2::ITStracking
                        O2::GPUCommon
@@ -27,11 +33,23 @@ o2_add_library(TRKReconstruction
                        O2::DataFormatsITS
                        O2::TRKSimulation
                        nlohmann_json::nlohmann_json
+                       ${actsTarget}
                PRIVATE_LINK_LIBRARIES
                        O2::Steer
                        TBB::tbb)
 
+if(Acts_FOUND)
+  target_compile_definitions(${targetName} PUBLIC O2_WITH_ACTS)
+endif()
+
+set(dictHeaders include/TRKReconstruction/TimeFrame.h
+                include/TRKReconstruction/Clusterer.h)
+
+if(Acts_FOUND)
+  list(APPEND dictHeaders include/TRKReconstruction/ClustererACTS.h
+                          include/TRKReconstruction/TrackerACTS.h)
+endif()
+
 o2_target_root_dictionary(TRKReconstruction
-                          HEADERS include/TRKReconstruction/TimeFrame.h
-                                  include/TRKReconstruction/Clusterer.h
+                          HEADERS ${dictHeaders}
                           LINKDEF src/TRKReconstructionLinkDef.h)

Detectors/Upgrades/ALICE3/TRK/reconstruction/include/TRKReconstruction/Clusterer.h

@@ -161,17 +161,17 @@ class Clusterer
   };
   //----------------------------------------------
 
-  void process(gsl::span<const Digit> digits,
-               gsl::span<const DigROFRecord> digitROFs,
-               std::vector<o2::trk::Cluster>& clusters,
-               std::vector<unsigned char>& patterns,
-               std::vector<o2::trk::ROFRecord>& clusterROFs,
-               const ConstDigitTruth* digitLabels = nullptr,
-               ClusterTruth* clusterLabels = nullptr,
-               gsl::span<const DigMC2ROFRecord> digMC2ROFs = {},
-               std::vector<o2::trk::MC2ROFRecord>* clusterMC2ROFs = nullptr);
-
- private:
+  virtual void process(gsl::span<const Digit> digits,
+                       gsl::span<const DigROFRecord> digitROFs,
+                       std::vector<o2::trk::Cluster>& clusters,
+                       std::vector<unsigned char>& patterns,
+                       std::vector<o2::trk::ROFRecord>& clusterROFs,
+                       const ConstDigitTruth* digitLabels = nullptr,
+                       ClusterTruth* clusterLabels = nullptr,
+                       gsl::span<const DigMC2ROFRecord> digMC2ROFs = {},
+                       std::vector<o2::trk::MC2ROFRecord>* clusterMC2ROFs = nullptr);
+
+ protected:
   int mNHugeClus = 0;
   std::unique_ptr<ClustererThread> mThread;
   std::vector<int> mSortIdx; ///< reusable per-ROF sort buffer

Detectors/Upgrades/ALICE3/TRK/reconstruction/src/TRKReconstructionLinkDef.h

@@ -17,5 +17,9 @@
 
 #pragma link C++ class o2::trk::TimeFrame < 11> + ;
 #pragma link C++ class o2::trk::Clusterer + ;
+#ifdef O2_WITH_ACTS
+#pragma link C++ class o2::trk::ClustererACTS + ;
+
+#endif
 
 #endif

Detectors/Upgrades/ALICE3/TRK/workflow/include/TRKWorkflow/ClustererSpec.h

@@ -15,6 +15,9 @@
 #include "Framework/DataProcessorSpec.h"
 #include "Framework/Task.h"
 #include "TRKReconstruction/Clusterer.h"
+#ifdef O2_WITH_ACTS
+#include "TRKReconstruction/ClustererACTS.h"
+#endif
 
 namespace o2::trk
 {
@@ -29,7 +32,13 @@ class ClustererDPL : public o2::framework::Task
  private:
   bool mUseMC = true;
   int mNThreads = 1;
+#ifdef O2_WITH_ACTS
+  bool mUseACTS = false;
+#endif
   o2::trk::Clusterer mClusterer;
+#ifdef O2_WITH_ACTS
+  o2::trk::ClustererACTS mClustererACTS;
+#endif
 };
 
 o2::framework::DataProcessorSpec getClustererSpec(bool useMC);

Detectors/Upgrades/ALICE3/TRK/workflow/src/ClustererSpec.cxx

@@ -23,6 +23,9 @@ namespace o2::trk
 void ClustererDPL::init(o2::framework::InitContext& ic)
 {
   mNThreads = std::max(1, ic.options().get<int>("nthreads"));
+#ifdef O2_WITH_ACTS
+  mUseACTS = ic.options().get<bool>("useACTS");
+#endif
 }
 
 void ClustererDPL::run(o2::framework::ProcessingContext& pc)
@@ -48,15 +51,32 @@ void ClustererDPL::run(o2::framework::ProcessingContext& pc)
   }
   o2::base::GeometryManager::loadGeometry("o2sim_geometry.root", false, true);
 
-  mClusterer.process(digits,
-                     rofs,
-                     clusters,
-                     patterns,
-                     clusterROFs,
-                     mUseMC ? &labels : nullptr,
-                     clusterLabels.get(),
-                     mc2rofs,
-                     mUseMC ? &clusterMC2ROFs : nullptr);
+#ifdef O2_WITH_ACTS
+  if (mUseACTS) {
+    LOG(info) << "Running TRKClusterer with ACTS";
+    mClustererACTS.process(digits,
+                           rofs,
+                           clusters,
+                           patterns,
+                           clusterROFs,
+                           mUseMC ? &labels : nullptr,
+                           clusterLabels.get(),
+                           mc2rofs,
+                           mUseMC ? &clusterMC2ROFs : nullptr);
+  } else
+#endif
+  {
+    LOG(info) << "Running TRKClusterer";
+    mClusterer.process(digits,
+                       rofs,
+                       clusters,
+                       patterns,
+                       clusterROFs,
+                       mUseMC ? &labels : nullptr,
+                       clusterLabels.get(),
+                       mc2rofs,
+                       mUseMC ? &clusterMC2ROFs : nullptr);
+  }
 
   pc.outputs().snapshot(o2::framework::Output{"TRK", "COMPCLUSTERS", 0}, clusters);
   pc.outputs().snapshot(o2::framework::Output{"TRK", "PATTERNS", 0}, patterns);
@@ -93,7 +113,12 @@ o2::framework::DataProcessorSpec getClustererSpec(bool useMC)
     inputs,
     outputs,
     o2::framework::AlgorithmSpec{o2::framework::adaptFromTask<o2::trk::ClustererDPL>(useMC)},
-    o2::framework::Options{{"nthreads", o2::framework::VariantType::Int, 1, {"Number of clustering threads"}}}};
+    o2::framework::Options{{"nthreads", o2::framework::VariantType::Int, 1, {"Number of clustering threads"}}
+#ifdef O2_WITH_ACTS
+                           ,
+                           {"useACTS", o2::framework::VariantType::Bool, false, {"Use ACTS for clustering"}}
+#endif
+    }};
 }
 
 } // namespace o2::trk