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:


  typedef std::vector< edm::FwdPtr<reco::PFCandidate> >  PFCollection;

  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 PFCollection & pfCandidates, 
           const reco::VertexCollection & vertices)  ;

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

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

  const PFCollection & getPFCandidatesFromPU() const {return pfCandidatesFromPU_;}
  
  const PFCollection & getPFCandidatesFromVtx() const {return pfCandidatesFromVtx_;}

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


 private  :

  bool   checkClosestZVertex_;
  
  
  bool   verbose_;

  PFCollection pfCandidatesFromVtx_;
  PFCollection pfCandidatesFromPU_;
  
};

#endif