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 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_;
  };

}

#endif