PtAssignmentEngineDxy.h

Go to the documentation of this file.

#ifndef L1TMuonEndCap_PtAssignmentEngineDxy_h
#define L1TMuonEndCap_PtAssignmentEngineDxy_h
 
#include <cstdint>
#include <cstdlib>
#include <cmath>
#include <string>
#include <vector>
#include <array>
 
#include "L1Trigger/L1TMuonEndCap/interface/Common.h"
#include "L1Trigger/L1TMuonEndCap/interface/PtAssignmentEngineAux2017.h"
#include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
#include "FWCore/ParameterSet/interface/FileInPath.h"
 
class PtAssignmentEngineDxy {
public:
  explicit PtAssignmentEngineDxy();
  virtual ~PtAssignmentEngineDxy();
 
  void configure(int verbose, const std::string pbFileNameDxy);
 
  const PtAssignmentEngineAux2017& aux() const;
 
  virtual void calculate_pt_dxy(const EMTFTrack& track, emtf::Feature& feature, emtf::Prediction& prediction) const;
 
  virtual void preprocessing_dxy(const EMTFTrack& track, emtf::Feature& feature) const;
 
  virtual void call_tensorflow_dxy(const emtf::Feature& feature, emtf::Prediction& prediction) const;
 
protected:
  int verbose_;
 
  tensorflow::GraphDef* graphDefDxy_;
  tensorflow::Session* sessionDxy_;
  std::string pbFileNameDxy_;
  std::string pbFilePathDxy_;
  std::string inputNameDxy_;
  std::vector<std::string> outputNamesDxy_;
};
 
#endif