IPCutPFCandidateSelectorDefinition.h

Go to the documentation of this file.

#ifndef CommonTools_ParticleFlow_IPCutPFCandidateSelectorDefinition
#define CommonTools_ParticleFlow_IPCutPFCandidateSelectorDefinition

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "CommonTools/ParticleFlow/interface/PFCandidateSelectorDefinition.h"
#include <DataFormats/VertexReco/interface/Vertex.h>
#include <DataFormats/VertexReco/interface/VertexFwd.h>
#include <DataFormats/GsfTrackReco/interface/GsfTrack.h>

namespace pf2pat {

  struct IPCutPFCandidateSelectorDefinition : public PFCandidateSelectorDefinition {

    IPCutPFCandidateSelectorDefinition ( const edm::ParameterSet & cfg, edm::ConsumesCollector && iC ) :
      verticesToken_( iC.consumes<reco::VertexCollection>( cfg.getParameter<edm::InputTag> ( "vertices" ) ) ),
      d0Cut_( cfg.getParameter<double>("d0Cut") ),
      dzCut_( cfg.getParameter<double>("dzCut") ),
      d0SigCut_( cfg.getParameter<double>("d0SigCut") ),
      dzSigCut_( cfg.getParameter<double>("dzSigCut") ) {}

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

      edm::Handle<reco::VertexCollection> vertices;
      e.getByToken(verticesToken_, vertices);
      if (vertices->empty()) return;
      const reco::Vertex &vtx = (*vertices)[0];

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

    bool passing = true;
    const reco::Track *tk = nullptr;
    if (pfc->gsfTrackRef().isNonnull())   tk = pfc->gsfTrackRef().get();
    else if (pfc->trackRef().isNonnull()) tk = pfc->trackRef().get();

    if (tk != nullptr) {
      double d0 =  fabs(tk->dxy(vtx.position()));
      double dz =  fabs(tk->dz(vtx.position()));
      double d0e = hypot(tk->dxyError(), hypot(vtx.xError(), vtx.yError()));
      double dze = hypot(tk->dzError(),  vtx.zError());
      if (d0Cut_ > 0 && d0 > d0Cut_) passing = false;
      if (dzCut_ > 0 && dz > dzCut_) passing = false;
      if (d0SigCut_ > 0 && d0e > 0 && d0/d0e > d0SigCut_) passing = false;
      if (dzSigCut_ > 0 && dze > 0 && dz/dze > dzSigCut_) passing = false;
    }

    if( passing ) {
      selected_.push_back( reco::PFCandidate(*pfc) );
      reco::PFCandidatePtr ptrToMother( hc, key );

      if ( pfc->numberOfSourceCandidatePtrs() > 0 ) {
        selected_.back().setSourceCandidatePtr( edm::Ptr<reco::PFCandidate>(  pfc->sourceCandidatePtr(0)
                                          ) );
      }
      else {
        selected_.back().setSourceCandidatePtr( ptrToMother );
      }
    }
      }
    }

    private:
    edm::EDGetTokenT<reco::VertexCollection> verticesToken_;
    double d0Cut_;
    double dzCut_;
    double d0SigCut_;
    double dzSigCut_;
  };
}

#endif