Use RunningAverage to guess a reserve size for the product

cms-sw · May 18, 2015 · 85e61d7 · 85e61d7
1 parent a34df87
commit 85e61d7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/RecoParticleFlow/PFClusterProducer/plugins/PFRecHitProducer.cc b/RecoParticleFlow/PFClusterProducer/plugins/PFRecHitProducer.cc
@@ -47,6 +47,9 @@ void
    std::auto_ptr<reco::PFRecHitCollection> out(new reco::PFRecHitCollection );
    std::auto_ptr<reco::PFRecHitCollection> cleaned(new reco::PFRecHitCollection );
 
+   out->reserve( outputSizeGuess_.upper() );
+   cleaned->reserve( cleanedOutputSizeGuess_.upper() );
+
    navigator_->beginEvent(iSetup);
 
    for (unsigned int i=0;i<creators_.size();++i) {
@@ -75,6 +78,9 @@ void
      }
    }
 
+   outputSizeGuess_.update( out->size() );
+   cleanedOutputSizeGuess_.update( cleaned->size() );
+
    iEvent.put(out,"");
    iEvent.put(cleaned,"Cleaned");
    hitmap.clear();

diff --git a/RecoParticleFlow/PFClusterProducer/plugins/PFRecHitProducer.h b/RecoParticleFlow/PFClusterProducer/plugins/PFRecHitProducer.h
@@ -9,6 +9,7 @@
 #include "FWCore/Framework/interface/Event.h"
 #include "FWCore/Framework/interface/MakerMacros.h"
 
+#include "FWCore/Utilities/interface/RunningAverage.h"
 #include "FWCore/ParameterSet/interface/ParameterSet.h"
 
 #include "RecoParticleFlow/PFClusterProducer/interface/PFRecHitCreatorBase.h"
@@ -30,6 +31,8 @@ class PFRecHitProducer : public edm::EDProducer {
       virtual void produce(edm::Event&, const edm::EventSetup&) override;
       std::vector<std::unique_ptr<PFRecHitCreatorBase> > creators_;
       std::unique_ptr<PFRecHitNavigatorBase> navigator_;
+      edm::RunningAverage outputSizeGuess_; ///< Use this to try and accurately reserve space for the output
+      edm::RunningAverage cleanedOutputSizeGuess_; ///< Use this to try and accurately reserve space for the output
 };
 
 #include "FWCore/Framework/interface/MakerMacros.h"