---

EgammaPhotonTkNumIsolationProducer.cc

Go to the documentation of this file.

//*****************************************************************************
// File:      EgammaPhotonTkNumIsolationProducer.cc
// ----------------------------------------------------------------------------
// OrigAuth:  Matthias Mozer
// Institute: IIHE-VUB
//=============================================================================
//*****************************************************************************


#include "EgammaAnalysis/EgammaIsolationProducers/interface/EgammaPhotonTkNumIsolationProducer.h"

// Framework
#include "DataFormats/Common/interface/Handle.h"

#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandAssociation.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

#include "RecoEgamma/EgammaIsolationAlgos/interface/PhotonTkIsolation.h"


EgammaPhotonTkNumIsolationProducer::EgammaPhotonTkNumIsolationProducer(const edm::ParameterSet& config) : conf_(config)
{
 // use configuration file to setup input/output collection names
  photonProducer_               = conf_.getParameter<edm::InputTag>("photonProducer");
  
  trackProducer_           = conf_.getParameter<edm::InputTag>("trackProducer");

  ptMin_                = conf_.getParameter<double>("ptMin");
  intRadius_            = conf_.getParameter<double>("intRadius");
  extRadius_            = conf_.getParameter<double>("extRadius");
  maxVtxDist_           = conf_.getParameter<double>("maxVtxDist");

  //register your products
  produces < reco::CandViewIntAssociations >();
}


EgammaPhotonTkNumIsolationProducer::~EgammaPhotonTkNumIsolationProducer(){}


//
// member functions
//

// ------------ method called to produce the data  ------------
void
EgammaPhotonTkNumIsolationProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  
  // Get the  filtered objects
  edm::Handle< edm::View<reco::Candidate> > photonHandle;
  iEvent.getByLabel(photonProducer_,photonHandle);
  
  //get the tracks
  edm::Handle<reco::TrackCollection> tracks;
  iEvent.getByLabel(trackProducer_,tracks);
  const reco::TrackCollection* trackCollection = tracks.product();

  reco::CandViewIntAssociations* isoMap = new reco::CandViewIntAssociations( reco::CandidateBaseRefProd( photonHandle ) );

  PhotonTkIsolation myTkIsolation (extRadius_,intRadius_,ptMin_,maxVtxDist_,trackCollection) ;

  for(unsigned int i = 0 ; i < photonHandle->size(); ++i ){
    int isoValue = myTkIsolation.getNumberTracks(&(photonHandle->at(i)));
    isoMap->setValue(i,isoValue);
  }
   


  std::auto_ptr<reco::CandViewIntAssociations> isolMap(isoMap);
  iEvent.put(isolMap);

}

---