EgammaTrackSelector.h

Go to the documentation of this file.

#ifndef EgammaIsolationAlgos_EgammaTrackSelector_H
#define EgammaIsolationAlgos_EgammaTrackSelector_H

#include "DataFormats/RecoCandidate/interface/IsoDepositDirection.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/Common/interface/View.h"

#include <list>

namespace egammaisolation {

  class EgammaTrackSelector {
  public:
    typedef std::pair<float, float> Range;
    typedef std::list<const reco::Track*> result_type;
    typedef edm::View<reco::Track> input_type;
    typedef reco::TrackBase::Point BeamPoint;

    enum { dz = 0, vz, bs, vtx };

    struct Parameters {
      Parameters()
          : zRange(-1e6, 1e6),
            rRange(-1e6, 1e6),
            dir(0, 0),
            drMax(1e3),
            beamPoint(0, 0, 0),
            nHitsMin(0),
            chi2NdofMax(1e64),
            chi2ProbMin(-1.),
            ptMin(-1) {}
      Parameters(const Range& dz,
                 const double d0Max,
                 const reco::isodeposit::Direction& dirC,
                 double rMax,
                 const BeamPoint& point = BeamPoint(0, 0, 0))
          : zRange(dz),
            rRange(Range(0, d0Max)),
            dir(dirC),
            drMax(rMax),
            beamPoint(point),
            nHitsMin(0),
            chi2NdofMax(1e64),
            chi2ProbMin(-1.),
            ptMin(-1) {}
      Range zRange;                     
      Range rRange;                     
      reco::isodeposit::Direction dir;  
      double drMax;                     
      BeamPoint beamPoint;              
      unsigned int nHitsMin;            
      double chi2NdofMax;               
      double chi2ProbMin;               
      double ptMin;                     
      int dzOption;                     
    };

    EgammaTrackSelector(const Parameters& pars) : thePars(pars) {}

    result_type operator()(const input_type& tracks) const;

  private:
    Parameters thePars;
  };

}  // namespace egammaisolation

#endif