ElectronTkIsolation.h

Go to the documentation of this file.

#ifndef EgammaIsolationAlgos_ElectronTkIsolation_h
#define EgammaIsolationAlgos_ElectronTkIsolation_h
//*****************************************************************************
// File:      ElectronTkIsolation.h
// ----------------------------------------------------------------------------
// OrigAuth:  Matthias Mozer
// Institute: IIHE-VUB
//=============================================================================
//*****************************************************************************

//C++ includes
#include <vector>
#include <functional>

//Root includes
#include "TObjArray.h"

//CMSSW includes 
#include "FWCore/Framework/interface/ESHandle.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "RecoEgamma/EgammaIsolationAlgos/interface/EgammaTrackSelector.h"

#include<string>


class ElectronTkIsolation {
 public:
  
  //constructors
  ElectronTkIsolation ( double extRadius,
            double intRadius,
            double ptLow,
            double lip,
            double drb,
            const reco::TrackCollection* trackCollection,
            reco::TrackBase::Point beamPoint) :
  extRadius_(extRadius),
  intRadiusBarrel_(intRadius),
  intRadiusEndcap_(intRadius),
  stripBarrel_(0.0),
  stripEndcap_(0.0),
  ptLow_(ptLow),
  lip_(lip),
  drb_(drb),
  trackCollection_(trackCollection),
  beamPoint_(beamPoint) {
        setAlgosToReject();
        setDzOption("vz");

  }

  ElectronTkIsolation ( double extRadius,
                        double intRadiusBarrel,
                        double intRadiusEndcap,
                        double stripBarrel,
                        double stripEndcap,         
                        double ptLow,
                        double lip,
                        double drb,
                        const reco::TrackCollection* trackCollection,
                        reco::TrackBase::Point beamPoint) :
  extRadius_(extRadius),
  intRadiusBarrel_(intRadiusBarrel),
  intRadiusEndcap_(intRadiusEndcap),
  stripBarrel_(stripBarrel),
  stripEndcap_(stripEndcap),
  ptLow_(ptLow),
  lip_(lip),
  drb_(drb),
  trackCollection_(trackCollection),
  beamPoint_(beamPoint) {
        setAlgosToReject();
        setDzOption("vz");

  }

  ElectronTkIsolation ( double extRadius,
                        double intRadiusBarrel,
                        double intRadiusEndcap,
                        double stripBarrel,
                        double stripEndcap,         
                        double ptLow,
                        double lip,
                        double drb,
                        const reco::TrackCollection*,
                        reco::TrackBase::Point beamPoint,
                        const std::string&) ;

  //destructor 
  ~ElectronTkIsolation() ;
 
  //methods

    void setDzOption(const std::string &s) {
        if( ! s.compare("dz") )      dzOption_ = egammaisolation::EgammaTrackSelector::dz;
        else if( ! s.compare("vz") ) dzOption_ = egammaisolation::EgammaTrackSelector::vz;
        else if( ! s.compare("bs") ) dzOption_ = egammaisolation::EgammaTrackSelector::bs;
        else if( ! s.compare("vtx") )dzOption_ = egammaisolation::EgammaTrackSelector::vtx;
        else                         dzOption_ = egammaisolation::EgammaTrackSelector::dz;
    }

  int getNumberTracks(const reco::GsfElectron*) const ;
  double getPtTracks (const reco::GsfElectron*) const ;
  std::pair<int,double>getIso(const reco::GsfElectron*) const;
  std::pair<int,double>getIso(const reco::Track*) const ;

 private:

  bool passAlgo(const reco::TrackBase& trk)const;
  void setAlgosToReject();
  double extRadius_ ;
  double intRadiusBarrel_ ;
  double intRadiusEndcap_ ;
  double stripBarrel_ ;
  double stripEndcap_ ;
  double ptLow_ ;
  double lip_ ;
  double drb_;
  std::vector<int> algosToReject_; //vector is sorted
  const reco::TrackCollection *trackCollection_ ;
  reco::TrackBase::Point beamPoint_;

  int dzOption_;

  
};

#endif