PFPileUpAlgo.h

Go to the documentation of this file.

#ifndef CommonTools_PFCandProducer_PFPileUpAlgo_
#define CommonTools_PFCandProducer_PFPileUpAlgo_

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"

#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

class PFPileUpAlgo {
 public:
  PFPileUpAlgo():checkClosestZVertex_(true), verbose_(false) {;}
    
  PFPileUpAlgo( bool checkClosestZVertex, bool verbose=false):
    checkClosestZVertex_(checkClosestZVertex), verbose_(verbose) {;}

  ~PFPileUpAlgo(){;}

  // the last parameter is needed if you want to use the sourceCandidatePtr
  void process(const reco::PFCandidateCollection & pfCandidates, 
           const reco::VertexCollection & vertices, 
           const edm::Handle<reco::PFCandidateCollection> * handle=0)  ;

  inline void setVerbose(bool verbose) { verbose_ = verbose; }

  inline void setCheckClosestZVertex(bool val) { checkClosestZVertex_ = val;}

  const reco::PFCandidateCollection & getPFCandidatesFromPU() const {return pfCandidatesFromPU_;}
  
  const reco::PFCandidateCollection & getPFCandidatesFromVtx() const {return pfCandidatesFromVtx_;}

  int chargedHadronVertex(const reco::VertexCollection& vertices, 
            const reco::PFCandidate& pfcand ) const;


 private  :

  bool   checkClosestZVertex_;
  
  
  bool   verbose_;

  reco::PFCandidateCollection pfCandidatesFromVtx_;
  reco::PFCandidateCollection pfCandidatesFromPU_;
  
};

#endif