EgammaTrackSelector.h

Go to the documentation of this file.

#ifndef EgammaIsolationAlgos_EgammaTrackSelector_H
#define EgammaIsolationAlgos_EgammaTrackSelector_H

#include "RecoEgamma/EgammaIsolationAlgos/interface/EgammaRange.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 egammaisolation::EgammaRange<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;
  }; 

}

#endif