File indexing completed on 2024-04-06 12:20:55
0001 #ifndef L1TMuonEndCap_PtAssignmentEngineDxy_h
0002 #define L1TMuonEndCap_PtAssignmentEngineDxy_h
0003
0004 #include <cstdint>
0005 #include <cstdlib>
0006 #include <cmath>
0007 #include <string>
0008 #include <vector>
0009 #include <array>
0010
0011 #include "L1Trigger/L1TMuonEndCap/interface/Common.h"
0012 #include "L1Trigger/L1TMuonEndCap/interface/PtAssignmentEngineAux2017.h"
0013 #include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
0014 #include "FWCore/ParameterSet/interface/FileInPath.h"
0015
0016 class PtAssignmentEngineDxy {
0017 public:
0018 explicit PtAssignmentEngineDxy();
0019 virtual ~PtAssignmentEngineDxy();
0020
0021 void configure(int verbose, const std::string pbFileNameDxy);
0022
0023 const PtAssignmentEngineAux2017& aux() const;
0024
0025 virtual void calculate_pt_dxy(const EMTFTrack& track, emtf::Feature& feature, emtf::Prediction& prediction) const;
0026
0027 virtual void preprocessing_dxy(const EMTFTrack& track, emtf::Feature& feature) const;
0028
0029 virtual void call_tensorflow_dxy(const emtf::Feature& feature, emtf::Prediction& prediction) const;
0030
0031 protected:
0032 int verbose_;
0033
0034 tensorflow::GraphDef* graphDefDxy_;
0035 tensorflow::Session* sessionDxy_;
0036 std::string pbFileNameDxy_;
0037 std::string pbFilePathDxy_;
0038 std::string inputNameDxy_;
0039 std::vector<std::string> outputNamesDxy_;
0040 };
0041
0042 #endif
