VertexProducer.cc

Go to the documentation of this file.

#include "L1Trigger/VertexFinder/interface/VertexProducer.h"
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "DataFormats/L1Trigger/interface/Vertex.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"
#include "L1Trigger/VertexFinder/interface/AlgoSettings.h"
#include "L1Trigger/VertexFinder/interface/RecoVertex.h"
#include "L1Trigger/VertexFinder/interface/VertexFinder.h"

using namespace l1tVertexFinder;
using namespace std;

VertexProducer::VertexProducer(const edm::ParameterSet& iConfig)
    : l1TracksToken_(consumes<TTTrackRefCollectionType>(iConfig.getParameter<edm::InputTag>("l1TracksInputTag"))),
      tTopoToken(esConsumes<TrackerTopology, TrackerTopologyRcd>()),
      outputCollectionName_(iConfig.getParameter<std::string>("l1VertexCollectionName")),
      settings_(AlgoSettings(iConfig)) {
  // Get configuration parameters

  switch (settings_.vx_algo()) {
    case Algorithm::fastHisto:
      edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using the fastHisto binning algorithm";
      break;
    case Algorithm::fastHistoEmulation:
      edm::LogInfo("VertexProducer")
          << "VertexProducer::Finding vertices using the emulation version of the fastHisto binning algorithm";
      break;
    case Algorithm::fastHistoLooseAssociation:
      edm::LogInfo("VertexProducer")
          << "VertexProducer::Finding vertices using the fastHistoLooseAssociation binning algorithm";
      break;
    case Algorithm::GapClustering:
      edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a gap clustering algorithm";
      break;
    case Algorithm::agglomerativeHierarchical:
      edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a Simple Merge Clustering algorithm";
      break;
    case Algorithm::DBSCAN:
      edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a DBSCAN algorithm";
      break;
    case Algorithm::PVR:
      edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a PVR algorithm";
      break;
    case Algorithm::adaptiveVertexReconstruction:
      edm::LogInfo("VertexProducer")
          << "VertexProducer::Finding vertices using an AdaptiveVertexReconstruction algorithm";
      break;
    case Algorithm::HPV:
      edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a Highest Pt Vertex algorithm";
      break;
    case Algorithm::Kmeans:
      edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using a kmeans algorithm";
      break;
    case Algorithm::NNEmulation:
      edm::LogInfo("VertexProducer") << "VertexProducer::Finding vertices using the Neural Network Emulation";
      break;
  }

  //--- Define EDM output to be written to file (if required)
  if (settings_.vx_algo() == Algorithm::fastHistoEmulation || settings_.vx_algo() == Algorithm::NNEmulation) {
    produces<l1t::VertexWordCollection>(outputCollectionName_ + "Emulation");
  } else {
    produces<l1t::VertexCollection>(outputCollectionName_);
  }

  if (settings_.vx_algo() == Algorithm::NNEmulation) {
    // load graphs, create a new session and add the graphDef
    if (settings_.debug() > 1) {
      edm::LogInfo("VertexProducer") << "loading TrkWeight graph from " << settings_.vx_trkw_graph() << std::endl;
      edm::LogInfo("VertexProducer") << "loading PatternRec graph from " << settings_.vx_pattrec_graph() << std::endl;
    }
    TrkWGraph_ = tensorflow::loadGraphDef(settings_.vx_trkw_graph());
    TrkWSesh_ = tensorflow::createSession(TrkWGraph_);
    PattRecGraph_ = tensorflow::loadGraphDef(settings_.vx_pattrec_graph());
    PattRecSesh_ = tensorflow::createSession(PattRecGraph_);
  }
}

void VertexProducer::produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
  edm::Handle<TTTrackRefCollectionType> l1TracksHandle;
  iEvent.getByToken(l1TracksToken_, l1TracksHandle);

  std::vector<l1tVertexFinder::L1Track> l1Tracks;
  l1Tracks.reserve(l1TracksHandle->size());
  if (settings_.debug() > 1) {
    edm::LogInfo("VertexProducer") << "produce::Processing " << l1TracksHandle->size() << " tracks";
  }
  for (const auto& track : *l1TracksHandle) {
    auto l1track = L1Track(edm::refToPtr(track));
    if (l1track.pt() >= settings_.vx_TrackMinPt()) {
      l1Tracks.push_back(l1track);
    } else {
      if (settings_.debug() > 2) {
        edm::LogInfo("VertexProducer") << "produce::Removing track with too low of a pt (" << l1track.pt() << ")\n"
                                       << "         word = " << l1track.getTTTrackPtr()->getTrackWord().to_string(2);
      }
    }
  }

  if (settings_.debug() > 1) {
    edm::LogInfo("VertexProducer") << "produce::Processing " << l1Tracks.size() << " tracks after minimum pt cut of"
                                   << settings_.vx_TrackMinPt() << " GeV";
  }

  VertexFinder vf(l1Tracks, settings_);

  switch (settings_.vx_algo()) {
    case Algorithm::fastHisto: {
      const TrackerTopology& tTopo = iSetup.getData(tTopoToken);
      vf.fastHisto(&tTopo);
      break;
    }
    case Algorithm::fastHistoEmulation:
      vf.fastHistoEmulation();
      break;
    case Algorithm::fastHistoLooseAssociation:
      vf.fastHistoLooseAssociation();
      break;
    case Algorithm::GapClustering:
      vf.GapClustering();
      break;
    case Algorithm::agglomerativeHierarchical:
      vf.agglomerativeHierarchicalClustering();
      break;
    case Algorithm::DBSCAN:
      vf.DBSCAN();
      break;
    case Algorithm::PVR:
      vf.PVR();
      break;
    case Algorithm::adaptiveVertexReconstruction:
      vf.adaptiveVertexReconstruction();
      break;
    case Algorithm::HPV:
      vf.HPV();
      break;
    case Algorithm::Kmeans:
      vf.Kmeans();
      break;
    case Algorithm::NNEmulation:
      vf.NNVtxEmulation(TrkWSesh_, PattRecSesh_);
      break;
  }

  vf.sortVerticesInPt();
  vf.findPrimaryVertex();

  // //=== Store output EDM track and hardware stub collections.
  if (settings_.vx_algo() == Algorithm::fastHistoEmulation || settings_.vx_algo() == Algorithm::NNEmulation) {
    std::unique_ptr<l1t::VertexWordCollection> product_emulation =
        std::make_unique<l1t::VertexWordCollection>(vf.verticesEmulation().begin(), vf.verticesEmulation().end());
    iEvent.put(std::move(product_emulation), outputCollectionName_ + "Emulation");
  } else {
    std::unique_ptr<l1t::VertexCollection> product(new std::vector<l1t::Vertex>());
    for (const auto& vtx : vf.vertices()) {
      product->emplace_back(vtx.vertex());
    }
    iEvent.put(std::move(product), outputCollectionName_);
  }
}
DEFINE_FWK_MODULE(VertexProducer);