VertexProducer.h

Go to the documentation of this file.

#ifndef __L1Trigger_VertexFinder_VertexProducer_h__
#define __L1Trigger_VertexFinder_VertexProducer_h__

#include "DataFormats/Common/interface/Ptr.h"
#include "DataFormats/Common/interface/RefToPtr.h"
#include "DataFormats/L1Trigger/interface/Vertex.h"
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "DataFormats/L1Trigger/interface/VertexWord.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "FWCore/Framework/interface/global/EDProducer.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"
#include "PhysicsTools/TensorFlow/interface/TensorFlow.h"

#include <iostream>
#include <map>
#include <set>
#include <string>
#include <vector>

namespace l1tVertexFinder {
  class AlgoSettings;
}

class VertexProducer : public edm::global::EDProducer<> {
public:
  explicit VertexProducer(const edm::ParameterSet&);
  ~VertexProducer() override {}

private:
  typedef TTTrack<Ref_Phase2TrackerDigi_> TTTrackType;
  typedef std::vector<TTTrackType> TTTrackCollectionType;
  typedef edm::RefVector<TTTrackCollectionType> TTTrackRefCollectionType;
  typedef edm::View<TTTrackType> TTTrackCollectionView;

  void produce(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;

private:
  const edm::EDGetTokenT<TTTrackRefCollectionType> l1TracksToken_;
  const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> tTopoToken;
  const std::string outputCollectionName_;

  tensorflow::GraphDef* TrkWGraph_;
  tensorflow::Session* TrkWSesh_;
  tensorflow::GraphDef* PattRecGraph_;
  tensorflow::Session* PattRecSesh_;

  l1tVertexFinder::AlgoSettings settings_;
};

#endif