PFIsoDepositAlgo.h

Go to the documentation of this file.

#ifndef CommonTools_ParticleFlow_PFIsoDepositAlgo_
#define CommonTools_ParticleFlow_PFIsoDepositAlgo_

// system include files
#include <memory>
#include <string>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

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

//not a fwd declaration, to save the pain to the user to include the necessary DF header as well
#include "DataFormats/RecoCandidate/interface/IsoDeposit.h"

namespace pf2pat {

  class PFIsoDepositAlgo {
  public:
    // can be a template parameter (IsoDeposits from GenParticles? )
    typedef reco::PFCandidate Particle;
    typedef std::vector<Particle> ParticleCollection;

    // random access to the IsoDeposit corresponding to a given particle
    typedef std::vector<reco::IsoDeposit> IsoDeposits;

    explicit PFIsoDepositAlgo(const edm::ParameterSet&);

    ~PFIsoDepositAlgo();

    const IsoDeposits& produce(const ParticleCollection& toBeIsolated, const ParticleCollection& forIsolation);

  private:
    reco::IsoDeposit buildIsoDeposit(const Particle& particle, const ParticleCollection& forIsolation) const;

    bool sameParticle(const Particle& particle1, const Particle& particle2) const;

    IsoDeposits isoDeposits_;

    bool verbose_;
  };
}  // namespace pf2pat

#endif