LeptonVertexSignificance.cc

Go to the documentation of this file.

#include "PhysicsTools/PatUtils/interface/LeptonVertexSignificance.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "DataFormats/Common/interface/Handle.h"
#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
#include "TrackingTools/Records/interface/TransientTrackRecord.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/PatCandidates/interface/Electron.h"
#include "DataFormats/PatCandidates/interface/Muon.h"

using namespace pat;

// constructor
LeptonVertexSignificance::LeptonVertexSignificance(const edm::EventSetup& iSetup, edm::ConsumesCollector&& iC)
    : vertexToken_(iC.consumes<reco::VertexCollection>(edm::InputTag("offlinePrimaryVerticesFromCTFTracks"))) {
  // instantiate the transient-track builder
  edm::ESHandle<TransientTrackBuilder> builder;
  iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder", builder);
  theTrackBuilder_ = new TransientTrackBuilder(*builder.product());
}

// destructor
LeptonVertexSignificance::~LeptonVertexSignificance() { delete theTrackBuilder_; }

// calculate the TrackIsoPt for the lepton object
float LeptonVertexSignificance::calculate(const Electron& theElectron, const edm::Event& iEvent) {
  return this->calculate(*theElectron.gsfTrack(), iEvent);
}

float LeptonVertexSignificance::calculate(const Muon& theMuon, const edm::Event& iEvent) {
  return this->calculate(*theMuon.track(), iEvent);
}

// calculate the TrackIsoPt for the lepton's track
float LeptonVertexSignificance::calculate(const reco::Track& theTrack, const edm::Event& iEvent) {
  // FIXME: think more about how to handle events without vertices
  // lepton LR calculation should have nothing to do with event selection
  edm::Handle<reco::VertexCollection> vertexHandle;
  iEvent.getByToken(vertexToken_, vertexHandle);
  if (vertexHandle.product()->empty())
    return 0;
  reco::Vertex theVertex = vertexHandle.product()->front();
  // calculate the track-vertex association significance
  reco::TransientTrack theTrTrack = theTrackBuilder_->build(&theTrack);
  GlobalPoint theVertexPoint(theVertex.position().x(), theVertex.position().y(), theVertex.position().z());
  FreeTrajectoryState theLeptonNearVertex = theTrTrack.trajectoryStateClosestToPoint(theVertexPoint).theState();
  return fabs(theVertex.position().z() - theLeptonNearVertex.position().z()) /
         sqrt(std::pow(theVertex.zError(), 2) + theLeptonNearVertex.cartesianError().position().czz());
}