JetSignalVertexCompatibility.cc

Go to the documentation of this file.

#include <memory>
 
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
 
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/JetReco/interface/JetTracksAssociation.h"
#include "DataFormats/JetReco/interface/JetFloatAssociation.h"
 
#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
#include "TrackingTools/Records/interface/TransientTrackRecord.h"
 
#include "RecoJets/JetAssociationAlgorithms/interface/JetSignalVertexCompatibilityAlgo.h"
 
#include "JetSignalVertexCompatibility.h"
 
using namespace reco;
 
JetSignalVertexCompatibility::JetSignalVertexCompatibility(const edm::ParameterSet &params)
    : algo(params.getParameter<double>("cut"), params.getParameter<double>("temperature")) {
  jetTracksAssocToken = consumes<JetTracksAssociationCollection>(params.getParameter<edm::InputTag>("jetTracksAssoc"));
  primaryVerticesToken = consumes<VertexCollection>(params.getParameter<edm::InputTag>("primaryVertices"));
  produces<JetFloatAssociation::Container>();
}
 
JetSignalVertexCompatibility::~JetSignalVertexCompatibility() {}
 
void JetSignalVertexCompatibility::produce(edm::Event &event, const edm::EventSetup &es) {
  edm::ESHandle<TransientTrackBuilder> trackBuilder;
  es.get<TransientTrackRecord>().get("TransientTrackBuilder", trackBuilder);
 
  algo.resetEvent(trackBuilder.product());
 
  edm::Handle<JetTracksAssociationCollection> jetTracksAssoc;
  event.getByToken(jetTracksAssocToken, jetTracksAssoc);
 
  edm::Handle<VertexCollection> primaryVertices;
  event.getByToken(primaryVerticesToken, primaryVertices);
 
  auto result = std::make_unique<JetFloatAssociation::Container>(jetTracksAssoc->keyProduct());
 
  for (JetTracksAssociationCollection::const_iterator iter = jetTracksAssoc->begin(); iter != jetTracksAssoc->end();
       ++iter) {
    if (primaryVertices->empty())
      (*result)[iter->first] = -1.;
 
    const TrackRefVector &tracks = iter->second;
    std::vector<float> compatibility = algo.compatibility(*primaryVertices, tracks);
 
    // the first vertex is the presumed signal vertex
    (*result)[iter->first] = compatibility[0];
  }
 
  algo.resetEvent(nullptr);
 
  event.put(std::move(result));
}