clear method for Track Container

acts-project · Jul 10, 2023 · da37cec · da37cec
1 parent d0ad664
commit da37cec
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 7 deletions.
diff --git a/Core/include/Acts/EventData/TrackContainer.hpp b/Core/include/Acts/EventData/TrackContainer.hpp
@@ -236,6 +236,13 @@ class TrackContainer {
     container().ensureDynamicColumns_impl(other.container());
   }
 
+  /// Clear the content of the track container  
+  template <bool RO = ReadOnly, typename = std::enable_if_t<!RO>>
+  void clear() {
+    m_container->clear_impl();
+    m_traj->clear_impl();
+  }
+
  protected:
   template <typename T, HashedString key, bool RO = ReadOnly,
             typename = std::enable_if_t<!RO>>

diff --git a/Core/include/Acts/EventData/VectorTrackContainer.hpp b/Core/include/Acts/EventData/VectorTrackContainer.hpp
@@ -234,13 +234,13 @@ class VectorTrackContainer final : public detail_vtc::VectorTrackContainerBase {
       const detail_vtc::VectorTrackContainerBase& other);
 
   void reserve(IndexType size);
-  void clear();
-
+  void clear_impl();
+  
   void setReferenceSurface_impl(IndexType itrack,
                                 std::shared_ptr<const Surface> surface) {
     m_referenceSurfaces[itrack] = std::move(surface);
   }
-
+  
   // END INTERFACE
 };
 

diff --git a/Core/src/EventData/VectorTrackContainer.cpp b/Core/src/EventData/VectorTrackContainer.cpp
@@ -132,7 +132,7 @@ void VectorTrackContainer::reserve(IndexType size) {
   }
 }
 
-void VectorTrackContainer::clear() {
+void VectorTrackContainer::clear_impl() {
   m_tipIndex.clear();
 
   m_params.clear();

diff --git a/Examples/Algorithms/TrackFinding/src/TrackFindingAlgorithm.cpp b/Examples/Algorithms/TrackFinding/src/TrackFindingAlgorithm.cpp
@@ -123,7 +123,7 @@ ActsExamples::ProcessCode ActsExamples::TrackFindingAlgorithm::execute(
   auto trackContainerTemp = std::make_shared<Acts::VectorTrackContainer>();
   auto trackStateContainerTemp =
       std::make_shared<Acts::VectorMultiTrajectory>();
-
+ 
   TrackContainer tracks(trackContainer, trackStateContainer);
   TrackContainer tracksTemp(trackContainerTemp, trackStateContainerTemp);
 
@@ -134,8 +134,8 @@ ActsExamples::ProcessCode ActsExamples::TrackFindingAlgorithm::execute(
   unsigned int nSeed = 0;
 
   for (std::size_t iseed = 0; iseed < initialParameters.size(); ++iseed) {
-    trackContainerTemp->clear();
-    trackStateContainerTemp->clear();
+    tracksTemp.clear();
+
     auto result =
         (*m_cfg.findTracks)(initialParameters.at(iseed), options, tracksTemp);
     m_nTotalSeeds++;