ConeTools.h

Go to the documentation of this file.

#ifndef RecoTauTag_RecoTau_ConeTools_h
#define RecoTauTag_RecoTau_ConeTools_h
 
#include "DataFormats/Math/interface/deltaR.h"
#include <boost/iterator/filter_iterator.hpp>
#include "DataFormats/TauReco/interface/PFTau.h"
#include "DataFormats/TauReco/interface/PFRecoTauChargedHadron.h"
#include "DataFormats/TauReco/interface/RecoTauPiZero.h"
 
namespace reco::tau::cone {
 
  // Predicate class that tests if a candidate lies within some deltaR (min,
  // max) about a supplied axis
  template <class CandType>
  class DeltaRFilter {
  public:
    DeltaRFilter(const reco::Candidate::LorentzVector& axis, double min, double max)
        : eta_(axis.eta()), phi_(axis.phi()), min2_(min < 0. ? -min * min : min * min), max2_(max * max) {}
    bool operator()(const CandType& b) const {
      double dR2 = deltaR2(b.eta(), b.phi(), eta_, phi_);
      return (dR2 >= min2_ && dR2 < max2_);
    }
 
  private:
    double eta_;
    double phi_;
    const double min2_;
    const double max2_;
  };
 
  // Wrapper around DeltaRFilter to support reference types like Ptr<>
  template <class CandType>
  class DeltaRPtrFilter {
  public:
    DeltaRPtrFilter(const reco::Candidate::LorentzVector& axis, double min, double max) : filter_(axis, min, max) {}
    bool operator()(const CandType& b) const { return filter_(*b); }
 
  private:
    DeltaRFilter<typename CandType::value_type> filter_;
  };
 
  /* Define our filters */
  typedef DeltaRPtrFilter<PFCandidatePtr> PFCandPtrDRFilter;
  typedef boost::filter_iterator<PFCandPtrDRFilter, std::vector<PFCandidatePtr>::const_iterator> PFCandPtrDRFilterIter;
 
  typedef DeltaRPtrFilter<CandidatePtr> CandPtrDRFilter;
  typedef boost::filter_iterator<CandPtrDRFilter, std::vector<CandidatePtr>::const_iterator> CandPtrDRFilterIter;
 
  typedef DeltaRFilter<PFRecoTauChargedHadron> ChargedHadronDRFilter;
  typedef boost::filter_iterator<ChargedHadronDRFilter, std::vector<PFRecoTauChargedHadron>::const_iterator>
      ChargedHadronDRFilterIter;
 
  typedef DeltaRFilter<RecoTauPiZero> PiZeroDRFilter;
  typedef boost::filter_iterator<PiZeroDRFilter, std::vector<RecoTauPiZero>::const_iterator> PiZeroDRFilterIter;
 
}  // end namespace reco::tau::cone
 
#endif