ElectronMatchedCandidateProducer.cc

Go to the documentation of this file.

#include "PhysicsTools/TagAndProbe/interface/ElectronMatchedCandidateProducer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "DataFormats/Math/interface/deltaR.h" // reco::deltaR


ElectronMatchedCandidateProducer::ElectronMatchedCandidateProducer(const edm::ParameterSet &params)
{

  const edm::InputTag allelectrons("gsfElectrons");
  electronCollectionToken_ =
    consumes<edm::View<reco::GsfElectron> >(params.getUntrackedParameter<edm::InputTag>("ReferenceElectronCollection",
                        allelectrons));
  scCollectionToken_ =
    consumes<edm::View<reco::Candidate> >(params.getParameter<edm::InputTag>("src"));

  delRMatchingCut_ = params.getUntrackedParameter<double>("deltaR",
                               0.30);

  produces< edm::PtrVector<reco::Candidate> >();
  produces< edm::RefToBaseVector<reco::Candidate> >();
}




ElectronMatchedCandidateProducer::~ElectronMatchedCandidateProducer()
{

}


//
// member functions
//


// ------------ method called to produce the data  ------------

void ElectronMatchedCandidateProducer::produce(edm::Event &event,
                  const edm::EventSetup &eventSetup)
{
   // Create the output collection
  std::auto_ptr< edm::RefToBaseVector<reco::Candidate> >
    outColRef( new edm::RefToBaseVector<reco::Candidate> );
  std::auto_ptr< edm::PtrVector<reco::Candidate> >
    outColPtr( new edm::PtrVector<reco::Candidate> );


  // Read electrons
  edm::Handle<edm::View<reco::GsfElectron> > electrons;
  event.getByToken(electronCollectionToken_, electrons);



  //Read candidates
  edm::Handle<edm::View<reco::Candidate> > recoCandColl;
  event.getByToken( scCollectionToken_ , recoCandColl);


  unsigned int counter=0;

  // Loop over candidates
  for(edm::View<reco::Candidate>::const_iterator scIt = recoCandColl->begin();
      scIt != recoCandColl->end(); ++scIt, ++counter){
    // Now loop over electrons
    for(edm::View<reco::GsfElectron>::const_iterator  elec = electrons->begin();
    elec != electrons->end();  ++elec) {

      reco::SuperClusterRef eSC = elec->superCluster();

      double dRval = reco::deltaR((float)eSC->eta(), (float)eSC->phi(),
                  scIt->eta(), scIt->phi());

      if( dRval < delRMatchingCut_ ) {
    //outCol->push_back( *scIt );
    outColRef->push_back( recoCandColl->refAt(counter) );
    outColPtr->push_back( recoCandColl->ptrAt(counter) );
      } // end if loop
    } // end electron loop

  } // end candidate loop

  event.put(outColRef);
  event.put(outColPtr);
}




// ------ method called once each job just before starting event loop  ---



void ElectronMatchedCandidateProducer::beginJob() {}



void ElectronMatchedCandidateProducer::endJob() {}



//define this as a plug-in
DEFINE_FWK_MODULE( ElectronMatchedCandidateProducer );