IsolatedPFCandidateSelectorDefinition.h

Go to the documentation of this file.

#ifndef CommonTools_ParticleFlow_IsolatedPFCandidateSelectorDefinition
#define CommonTools_ParticleFlow_IsolatedPFCandidateSelectorDefinition

#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "CommonTools/ParticleFlow/interface/PFCandidateSelectorDefinition.h"
#include "DataFormats/Common/interface/ValueMap.h"
#include "FWCore/Utilities/interface/Exception.h"

namespace pf2pat {

  class IsolatedPFCandidateSelectorDefinition : public PFCandidateSelectorDefinition {

  public:
    typedef edm::ValueMap<double> IsoMap;

    IsolatedPFCandidateSelectorDefinition ( const edm::ParameterSet & cfg ) :
      isolationValueMapChargedLabels_(cfg.getParameter< std::vector<edm::InputTag> >("isolationValueMapsCharged") ),
      isolationValueMapNeutralLabels_(cfg.getParameter< std::vector<edm::InputTag> >("isolationValueMapsNeutral") ),
      doDeltaBetaCorrection_(cfg.getParameter<bool>("doDeltaBetaCorrection")),
      deltaBetaIsolationValueMap_(cfg.getParameter< edm::InputTag >("deltaBetaIsolationValueMap") ),
      deltaBetaFactor_(cfg.getParameter<double>("deltaBetaFactor")),
      isRelative_(cfg.getParameter<bool>("isRelative")),
      isolationCut_(cfg.getParameter<double>("isolationCut")) {}
    


    void select( const HandleToCollection & hc, 
         const edm::EventBase & e,
         const edm::EventSetup& s) {
      selected_.clear();
    

      // read all charged isolation value maps
      std::vector< edm::Handle<IsoMap> > 
    isoMapsCharged(isolationValueMapChargedLabels_.size());
      for(unsigned iMap = 0; iMap<isolationValueMapChargedLabels_.size(); ++iMap) {
    e.getByLabel(isolationValueMapChargedLabels_[iMap], isoMapsCharged[iMap]);
      }


      // read all neutral isolation value maps
      std::vector< edm::Handle<IsoMap> > 
    isoMapsNeutral(isolationValueMapNeutralLabels_.size());
      for(unsigned iMap = 0; iMap<isolationValueMapNeutralLabels_.size(); ++iMap) {
    e.getByLabel(isolationValueMapNeutralLabels_[iMap], isoMapsNeutral[iMap]);
      }

      edm::Handle<IsoMap> dBetaH;
      if(doDeltaBetaCorrection_) {
    e.getByLabel(deltaBetaIsolationValueMap_, dBetaH);
      }

      unsigned key=0;
      for( collection::const_iterator pfc = hc->begin(); 
       pfc != hc->end(); ++pfc, ++key) {
    reco::PFCandidateRef candidate(hc,key);

    bool passed = true;
    double isoSumCharged=0.0;
    double isoSumNeutral=0.0;

    for(unsigned iMap = 0; iMap<isoMapsCharged.size(); ++iMap) {
      const IsoMap & isoMap = *(isoMapsCharged[iMap]);
      double val = isoMap[candidate];
      isoSumCharged+=val;
    }

    for(unsigned iMap = 0; iMap<isoMapsNeutral.size(); ++iMap) {
      const IsoMap & isoMap = *(isoMapsNeutral[iMap]);
      double val = isoMap[candidate];
      isoSumNeutral+=val;
    }
    

    if ( doDeltaBetaCorrection_ ) {
      const IsoMap& isoMap = *dBetaH;
      double dBetaVal = isoMap[candidate];
      double dBetaCorIsoSumNeutral = isoSumNeutral + deltaBetaFactor_*dBetaVal; 
      isoSumNeutral = dBetaCorIsoSumNeutral>0 ? dBetaCorIsoSumNeutral : 0; //follow muon POG definition in 2012
    }

    double isoSum=isoSumCharged+isoSumNeutral;

    if( isRelative_ ) {
      isoSum /= candidate->pt();
    }

    if ( isoSum>isolationCut_ ) {
      passed = false;
    }

    if(passed) {
      // passed all cuts, selected
      selected_.push_back( reco::PFCandidate(*pfc) );
      reco::PFCandidatePtr ptrToMother( hc, key );
      selected_.back().setSourceCandidatePtr( ptrToMother );
    }
      }
    }
    

  private:
    std::vector<edm::InputTag> isolationValueMapChargedLabels_;
    std::vector<edm::InputTag> isolationValueMapNeutralLabels_;
    bool                       doDeltaBetaCorrection_;
    edm::InputTag              deltaBetaIsolationValueMap_;
    double                     deltaBetaFactor_;
    bool                       isRelative_; 
    double                     isolationCut_;
  };

}

#endif