Shrinking internal data structures after clearing them.

cms-sw · Jul 16, 2020 · a287b79 · a287b79
1 parent 0393b55
commit a287b79
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/RecoLocalCalo/HGCalRecProducers/interface/HGCalImagingAlgo.h b/RecoLocalCalo/HGCalRecProducers/interface/HGCalImagingAlgo.h
@@ -65,9 +65,12 @@ class HGCalImagingAlgo : public HGCalClusteringAlgoBase {
   // use this if you want to reuse the same cluster object but don't want to accumulate clusters (hardly useful?)
   void reset() override {
     clusters_v_.clear();
+    clusters_v_.shrink_to_fit();
     layerClustersPerLayer_.clear();
+    layerClustersPerLayer_.shrink_to_fit();
     for (auto &it : points_) {
       it.clear();
+      it.shrink_to_fit();
       std::vector<KDNode>().swap(it);
     }
     for (unsigned int i = 0; i < minpos_.size(); i++) {

diff --git a/RecoLocalCalo/HGCalRecProducers/plugins/HGCalCLUEAlgo.h b/RecoLocalCalo/HGCalRecProducers/plugins/HGCalCLUEAlgo.h
@@ -63,12 +63,15 @@ class HGCalCLUEAlgoT : public HGCalClusteringAlgoBase {
 
   void reset() override {
     clusters_v_.clear();
+    clusters_v_.shrink_to_fit();
     for (auto& cl : numberOfClustersPerLayer_) {
       cl = 0;
     }
 
-    for (auto& cells : cells_)
+    for (auto& cells : cells_) {
       cells.clear();
+      cells.shrink_to_fit();
+    }
   }
 
   Density getDensity() override;
@@ -174,6 +177,23 @@ class HGCalCLUEAlgoT : public HGCalClusteringAlgoBase {
       followers.clear();
       isSeed.clear();
     }
+
+    void shrink_to_fit() {
+      detid.shrink_to_fit();
+      isSi.shrink_to_fit();
+      x.shrink_to_fit();
+      y.shrink_to_fit();
+      eta.shrink_to_fit();
+      phi.shrink_to_fit();
+      weight.shrink_to_fit();
+      rho.shrink_to_fit();
+      delta.shrink_to_fit();
+      nearestHigher.shrink_to_fit();
+      clusterIndex.shrink_to_fit();
+      sigmaNoise.shrink_to_fit();
+      followers.shrink_to_fit();
+      isSeed.shrink_to_fit();
+    }
   };
 
   std::vector<CellsOnLayer> cells_;