HistoAnalyzer.h

Go to the documentation of this file.

#ifndef UtilAlgos_HistoAnalyzer_h
#define UtilAlgos_HistoAnalyzer_h

#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "CommonTools/UtilAlgos/interface/ExpressionHisto.h"

template<typename C>
class HistoAnalyzer : public edm::EDAnalyzer {
public:
  HistoAnalyzer( const edm::ParameterSet& );
  ~HistoAnalyzer();

protected:
  virtual void analyze( const edm::Event&, const edm::EventSetup&) override;

private:
  edm::EDGetTokenT<C> srcToken_;
  bool usingWeights_;
  edm::EDGetTokenT<double> weightsToken_;
  std::vector<ExpressionHisto<typename C::value_type>* > vhistograms;
};

template<typename C>
HistoAnalyzer<C>::HistoAnalyzer( const edm::ParameterSet& par ) :
  srcToken_(consumes<C>(par.template getParameter<edm::InputTag>("src"))),
  usingWeights_(par.exists("weights")),
  weightsToken_(mayConsume<double>(par.template getUntrackedParameter<edm::InputTag>("weights", edm::InputTag("fake"))))
{
   edm::Service<TFileService> fs;
   std::vector<edm::ParameterSet> histograms =
                                   par.template getParameter<std::vector<edm::ParameterSet> >("histograms");
   std::vector<edm::ParameterSet>::const_iterator it = histograms.begin();
   std::vector<edm::ParameterSet>::const_iterator end = histograms.end();

   // create the histograms from the given parameter sets
   for (; it!=end; ++it)
   {
      ExpressionHisto<typename C::value_type>* hist = new ExpressionHisto<typename C::value_type>(*it);
      hist->initialize(fs->tFileDirectory());
      vhistograms.push_back(hist);
   }

}

template<typename C>
HistoAnalyzer<C>::~HistoAnalyzer()
{
   // delete all histograms and clear the vector of pointers
   typename std::vector<ExpressionHisto<typename C::value_type>* >::iterator it = vhistograms.begin();
   typename std::vector<ExpressionHisto<typename C::value_type>* >::iterator end = vhistograms.end();
   for (;it!=end; ++it){
     (*it)->~ExpressionHisto<typename C::value_type>();
   }
   vhistograms.clear();
}

template<typename C>
void HistoAnalyzer<C>::analyze( const edm::Event& iEvent, const edm::EventSetup&)
{
   edm::Handle<C> coll;
   iEvent.getByToken( srcToken_, coll);
   double weight = 1.0;
   if (usingWeights_) {
       edm::Handle<double> weightColl;
       iEvent.getByToken( weightsToken_, weightColl );
       weight = *weightColl;
   }

   typename std::vector<ExpressionHisto<typename C::value_type>* >::iterator it = vhistograms.begin();
   typename std::vector<ExpressionHisto<typename C::value_type>* >::iterator end = vhistograms.end();

   for (;it!=end; ++it){
      uint32_t i = 0;
      for( typename C::const_iterator elem=coll->begin(); elem!=coll->end(); ++elem, ++i ) {
         if (!(*it)->fill( *elem, weight, i )) {
            break;
         }
      }
   }
}

#endif