PFIsoDepositAlgo.cc

Go to the documentation of this file.

#include "CommonTools/ParticleFlow/interface/PFIsoDepositAlgo.h"
 
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
 
#include "DataFormats/Math/interface/LorentzVector.h"
 
#include "FWCore/Framework/interface/EventSetup.h"
 
using namespace std;
using namespace edm;
using namespace reco;
using namespace math;
using namespace pf2pat;
 
PFIsoDepositAlgo::PFIsoDepositAlgo(const edm::ParameterSet& iConfig)
    : verbose_(iConfig.getUntrackedParameter<bool>("verbose", false))
 
{}
 
PFIsoDepositAlgo::~PFIsoDepositAlgo() {}
 
const PFIsoDepositAlgo::IsoDeposits& PFIsoDepositAlgo::produce(const ParticleCollection& toBeIsolated,
                                                               const ParticleCollection& forIsolation) {
  isoDeposits_.clear();
  isoDeposits_.reserve(toBeIsolated.size());
 
  for (unsigned i = 0; i < toBeIsolated.size(); i++) {
    const reco::PFCandidate& toBeIso = toBeIsolated[i];
 
    if (verbose_)
      cout << "to be isolated: " << toBeIso << endl;
 
    isoDeposits_.push_back(buildIsoDeposit(toBeIso, forIsolation));
  }
 
  if (verbose_) {
    cout << "PFIsoDepositAlgo " << endl;
  }
 
  return isoDeposits_;
}
 
IsoDeposit PFIsoDepositAlgo::buildIsoDeposit(const Particle& particle, const ParticleCollection& forIsolation) const {
  reco::isodeposit::Direction pfDir(particle.eta(), particle.phi());
  //   reco::IsoDeposit::Veto veto;
  //   veto.vetoDir = pfDir;
  //   veto.dR = 0.05;
 
  IsoDeposit isoDep(pfDir);
 
  for (unsigned i = 0; i < forIsolation.size(); i++) {
    const reco::PFCandidate& pfc = forIsolation[i];
 
    // need to remove "particle"!
 
    if (sameParticle(particle, pfc))
      continue;
 
    const XYZTLorentzVector& pvi(pfc.p4());
    reco::isodeposit::Direction dirPfc(pfc.eta(), pfc.phi());
    double dR = pfDir.deltaR(dirPfc);
 
    //COLIN make a parameter
    double maxDeltaRForIsoDep_ = 1;
    if (dR > maxDeltaRForIsoDep_) {
      //      if( verbose_ ) cout<<"OUT OF CONE"<<endl;
      continue;
    }
    //    else if(verbose_) cout<<endl;
 
    if (verbose_)
      cout << "\t" << pfc << endl;
 
    double pt = pvi.Pt();
    isoDep.addDeposit(dirPfc, pt);
  }
 
  return isoDep;
}
 
bool PFIsoDepositAlgo::sameParticle(const Particle& particle1, const Particle& particle2) const {
  double smallNumber = 1e-15;
 
  if (particle1.particleId() != particle2.particleId())
    return false;
  else if (fabs(particle1.energy() - particle2.energy()) > smallNumber)
    return false;
  else if (fabs(particle1.eta() - particle2.eta()) > smallNumber)
    return false;
  else if (fabs(particle1.eta() - particle2.eta()) > smallNumber)
    return false;
  else
    return true;
}