LeptonVertexSignificance.h

Go to the documentation of this file.

//
//

#ifndef PhysicsTools_PatUtils_LeptonVertexSignificance_h
#define PhysicsTools_PatUtils_LeptonVertexSignificance_h

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

class TransientTrackBuilder;

namespace reco {
  class Track;
}

namespace edm {
  class Event;
  class EventSetup;
}  // namespace edm

namespace pat {
  class Electron;
  class Muon;

  class LeptonVertexSignificance {
  public:
    LeptonVertexSignificance();
    LeptonVertexSignificance(const edm::EventSetup& iSetup, edm::ConsumesCollector&& iC);
    ~LeptonVertexSignificance();

    float calculate(const Electron& anElectron, const edm::Event& iEvent);
    float calculate(const Muon& aMuon, const edm::Event& iEvent);

  private:
    float calculate(const reco::Track& track, const edm::Event& iEvent);
    TransientTrackBuilder* theTrackBuilder_;
    edm::EDGetTokenT<reco::VertexCollection> vertexToken_;
  };

}  // namespace pat

#endif