TauNNId.h

Go to the documentation of this file.

#ifndef L1TRIGGER_PHASE2L1PARTICLEFLOWS_TAUNNID_H
#define L1TRIGGER_PHASE2L1PARTICLEFLOWS_TAUNNID_H
 
#include <string>
#include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
#include "DataFormats/L1TParticleFlow/interface/PFCandidate.h"
 
struct TauNNTFCache {
  TauNNTFCache() : graphDef(nullptr) {}
  std::atomic<tensorflow::GraphDef *> graphDef;
};
 
class TauNNId {
public:
  TauNNId(const std::string &iInput, const TauNNTFCache *cache, const std::string &iWeightFile, int iNParticles);
  ~TauNNId();
 
  void setNNVectorVar();
  float EvaluateNN();
  float compute(const l1t::PFCandidate &iSeed, l1t::PFCandidateCollection &iParts);
 
private:
  tensorflow::Session *session_;
  std::vector<float> NNvectorVar_;
  std::string fInput_;
  int fNParticles_;
  unique_ptr<float[]> fPt_;
  unique_ptr<float[]> fEta_;
  unique_ptr<float[]> fPhi_;
  unique_ptr<float[]> fId_;
};
#endif