/data/refman/pasoursint/CMSSW_4_1_8_patch9/src/PhysicsTools/PFCandProducer/interface/IsolatedPFCandidateSelectorDefinition.h

Go to the documentation of this file.

#ifndef PhysicsTools_PFCandProducer_IsolatedPFCandidateSelectorDefinition
#define PhysicsTools_PFCandProducer_IsolatedPFCandidateSelectorDefinition

#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "PhysicsTools/PFCandProducer/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 ) :
      isolationValueMapLabels_(cfg.getParameter< std::vector<edm::InputTag> >("isolationValueMaps") ),
      isRelative_(cfg.getParameter<bool>("isRelative")),
      isCombined_(cfg.getParameter<bool>("isCombined")),
      isolationCuts_(cfg.getParameter< std::vector<double> >("isolationCuts")),
      combinedIsolationCut_(cfg.getParameter<double>("combinedIsolationCut")) { 
    

      if( isolationCuts_.size() != isolationValueMapLabels_.size() )
        throw cms::Exception("BadConfiguration")<<"the vector of isolation ValueMaps and the vector of the corresponding cuts must have the same size."<<std::endl;
    }

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

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

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

        bool passed = true;
        double isoSum=0.0;
        for(unsigned iMap = 0; iMap<isoMaps.size(); ++iMap) {
        
          const IsoMap & isoMap = *(isoMaps[iMap]);
        
          double val = isoMap[candidate];
          double cut = isolationCuts_[iMap];
          if(isRelative_ && candidate->pt()>0.0) val/=candidate->pt();
          isoSum+=val;
          //std::cout << "val " << iMap << " = " << val << std::endl;

          if ( !isCombined_ && val>cut ) {
            passed = false;
            break;        
          }
        }

        if ( isCombined_ && isoSum>combinedIsolationCut_ )
          {
            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> isolationValueMapLabels_;
    bool isRelative_, isCombined_;
    std::vector<double> isolationCuts_;
    double combinedIsolationCut_;
  };

}

#endif