ElectronSqPtTkIsolation.h

Go to the documentation of this file.

#ifndef ElectronSqPtTkIsolation_h
#define ElectronSqPtTkIsolation_h

//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"

class ElectronSqPtTkIsolation {
public:
  //constructors
  ElectronSqPtTkIsolation(double extRadius, double intRadius, double ptLow, double lip, const reco::TrackCollection*);
  //destructor
  ~ElectronSqPtTkIsolation();

  //methods

  int getNumberTracks(const reco::GsfElectron*) const;
  double getPtTracks(const reco::GsfElectron*) const;

private:
  double extRadius_;
  double intRadius_;
  double ptLow_;
  double lip_;

  const reco::TrackCollection* trackCollection_;

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

#endif