Halves the memory allocations in the TPCITS matcher

ktf · Oct 10, 2023 · 0ed28b5 · 0ed28b5
1 parent facbf10
commit 0ed28b5
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 6 deletions.
diff --git a/Detectors/GlobalTracking/include/GlobalTracking/MatchTPCITS.h b/Detectors/GlobalTracking/include/GlobalTracking/MatchTPCITS.h
@@ -368,12 +368,42 @@ class MatchTPCITS
   void printCandidatesTPC() const;
   void printCandidatesITS() const;
 
-  const std::vector<o2::dataformats::TrackTPCITS>& getMatchedTracks() const { return mMatchedTracks; }
-  const MCLabContTr& getMatchLabels() const { return mOutLabels; }
-  const MCLabContTr& getABTrackletLabels() const { return mABTrackletLabels; }
-  const std::vector<int>& getABTrackletClusterIDs() const { return mABTrackletClusterIDs; }
-  const std::vector<o2::itsmft::TrkClusRef>& getABTrackletRefs() const { return mABTrackletRefs; }
-  const std::vector<VDTriplet>& getTglITSTPC() const { return mTglITSTPC; }
+  [[nodiscard]] const std::vector<o2::dataformats::TrackTPCITS>& getMatchedTracks() const { return mMatchedTracks; }
+  void resetMatchedTracks()
+  {
+    mMatchedTracks.clear();
+    mMatchedTracks.shrink_to_fit();
+  }
+  [[nodiscard]] const MCLabContTr& getMatchLabels() const { return mOutLabels; }
+  void resetMatchLabels()
+  {
+    mOutLabels.clear();
+    mOutLabels.shrink_to_fit();
+  }
+  [[nodiscard]] const MCLabContTr& getABTrackletLabels() const { return mABTrackletLabels; }
+  void resetABTrackletLabels()
+  {
+    mABTrackletLabels.clear();
+    mABTrackletLabels.shrink_to_fit();
+  }
+  [[nodiscard]] const std::vector<int>& getABTrackletClusterIDs() const { return mABTrackletClusterIDs; }
+  void resetABTrackletClusterIDs()
+  {
+    mABTrackletClusterIDs.clear();
+    mABTrackletClusterIDs.shrink_to_fit();
+  }
+  [[nodiscard]] const std::vector<o2::itsmft::TrkClusRef>& getABTrackletRefs() const { return mABTrackletRefs; }
+  void resetABTrackletRefs()
+  {
+    mABTrackletRefs.clear();
+    mABTrackletRefs.shrink_to_fit();
+  }
+  [[nodiscard]] const std::vector<VDTriplet>& getTglITSTPC() const { return mTglITSTPC; }
+  void resetTglITSTPC()
+  {
+    mTglITSTPC.clear();
+    mTglITSTPC.shrink_to_fit();
+  }
 
   //>>> ====================== options =============================>>>
   void setUseMatCorrFlag(MatCorrType f) { mUseMatCorrFlag = f; }

diff --git a/Detectors/GlobalTrackingWorkflow/src/TPCITSMatchingSpec.cxx b/Detectors/GlobalTrackingWorkflow/src/TPCITSMatchingSpec.cxx
@@ -103,15 +103,21 @@ void TPCITSMatchingDPL::run(ProcessingContext& pc)
   mMatching.run(recoData);
 
   pc.outputs().snapshot(Output{"GLO", "TPCITS", 0, Lifetime::Timeframe}, mMatching.getMatchedTracks());
+  mMatching.resetMatchedTracks();
   pc.outputs().snapshot(Output{"GLO", "TPCITSAB_REFS", 0, Lifetime::Timeframe}, mMatching.getABTrackletRefs());
+  mMatching.resetABTrackletRefs();
   pc.outputs().snapshot(Output{"GLO", "TPCITSAB_CLID", 0, Lifetime::Timeframe}, mMatching.getABTrackletClusterIDs());
+  mMatching.resetABTrackletClusterIDs();
   if (mUseMC) {
     pc.outputs().snapshot(Output{"GLO", "TPCITS_MC", 0, Lifetime::Timeframe}, mMatching.getMatchLabels());
+    mMatching.resetMatchLabels();
     pc.outputs().snapshot(Output{"GLO", "TPCITSAB_MC", 0, Lifetime::Timeframe}, mMatching.getABTrackletLabels());
+    mMatching.resetABTrackletLabels();
   }
 
   if (mCalibMode) {
     pc.outputs().snapshot(Output{"GLO", "TPCITS_VDTGL", 0, Lifetime::Timeframe}, mMatching.getTglITSTPC());
+    mMatching.resetTglITSTPC();
   }
   mTimer.Stop();
 }