ParticleFlowForChargedMETProducer.cc

Go to the documentation of this file.

#include "RecoMET/METProducers/interface/ParticleFlowForChargedMETProducer.h"

#include <DataFormats/VertexReco/interface/Vertex.h>
#include <DataFormats/ParticleFlowCandidate/interface/PFCandidate.h>

using namespace edm;
using namespace std;
using namespace reco;

ParticleFlowForChargedMETProducer::ParticleFlowForChargedMETProducer(const edm::ParameterSet& iConfig)
{
  pfCollectionLabel    = iConfig.getParameter<edm::InputTag>("PFCollectionLabel");
  pvCollectionLabel    = iConfig.getParameter<edm::InputTag>("PVCollectionLabel");

  pfCandidatesToken = consumes<PFCandidateCollection>(pfCollectionLabel);
  pvCollectionToken = consumes<VertexCollection>(pvCollectionLabel);

  dzCut    = iConfig.getParameter<double>("dzCut");
  neutralEtThreshold    = iConfig.getParameter<double>("neutralEtThreshold");

  produces<PFCandidateCollection>();
}

void ParticleFlowForChargedMETProducer::produce(Event& iEvent, const EventSetup& iSetup)
{

  //Get the PV collection
  Handle<VertexCollection> pvCollection;
  iEvent.getByToken(pvCollectionToken, pvCollection);
  VertexCollection::const_iterator vertex = pvCollection->begin();

  //Get pfCandidates
  Handle<PFCandidateCollection> pfCandidates;
  iEvent.getByToken(pfCandidatesToken, pfCandidates);

  // the output collection
  auto chargedPFCandidates = std::make_unique<PFCandidateCollection>();
  if (pvCollection->size()>0) {
    for( unsigned i=0; i<pfCandidates->size(); i++ ) {
      const PFCandidate& pfCand = (*pfCandidates)[i];
      PFCandidatePtr pfCandPtr(pfCandidates, i);
      
      if (pfCandPtr->trackRef().isNonnull()) { 
    if (pfCandPtr->trackRef()->dz((*vertex).position()) < dzCut) {
      chargedPFCandidates->push_back( pfCand );
      chargedPFCandidates->back().setSourceCandidatePtr( pfCandPtr );
    }
    
      }
      else if (neutralEtThreshold>0 and 
           pfCandPtr->pt()>neutralEtThreshold) {
    chargedPFCandidates->push_back( pfCand );
    chargedPFCandidates->back().setSourceCandidatePtr( pfCandPtr );
      }  
    

    }
  }


  iEvent.put(std::move(chargedPFCandidates));

  return;
}

ParticleFlowForChargedMETProducer::~ParticleFlowForChargedMETProducer(){}