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