ObjectCounter.h

Go to the documentation of this file.

#ifndef UtilAlgos_ObjectCounter_h
#define UtilAlgos_ObjectCounter_h

#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/TypeID.h"
#include <iostream>
#include <cmath>

template<typename C>
class ObjectCounter : public edm::EDAnalyzer {
public:
  explicit ObjectCounter( const edm::ParameterSet& );
  void endJob();

private:
  virtual void analyze( const edm::Event&, const edm::EventSetup&) override;
  edm::EDGetTokenT<C> srcToken_;
  bool verbose_;
  unsigned long n_, nSum_, n2Sum_;
};

template<typename C>
ObjectCounter<C>::ObjectCounter( const edm::ParameterSet& par ) :
  srcToken_( consumes<C>( edm::InputTag( par.template getParameter<std::string>( "src" ) ) ) ),
  verbose_( par.template getUntrackedParameter<bool>( "verbose", true ) ),
  n_( 0 ), nSum_( 0 ), n2Sum_( 0 ) {
}

template<typename C>
void ObjectCounter<C>::endJob() {
  double n = 0, n2 = 0, s;
  if ( n_!= 0 ) {
    n = double( nSum_ ) / n_;
    n2 = double ( n2Sum_ ) / n_;
  }
  s = sqrt( n2 - n * n );
  if ( verbose_ ) {
    edm::TypeID id( typeid( typename C::value_type ) );
    std::cout << ">>> collection \"" << srcToken_ << "\" contains ("
          << n << " +/- " << s << ") "
          << id.friendlyClassName() << " objects" << std::endl;
  }
}

template<typename C>
void ObjectCounter<C>::analyze( const edm::Event& evt, const edm::EventSetup& ) {
  edm::Handle<C> h;
  evt.getByToken( srcToken_, h );
  if (!h.isValid()) {
    std::cerr << ">>> product: " << srcToken_ << " not found" << std::endl;
  } else {
    int n = h->size();
    nSum_ += n;
    n2Sum_ += ( n * n );
  }
  ++ n_;
}

#endif