TFCSSimpleLWTNNHandler.h

Go to the documentation of this file.

 
// Hopefully documentation gets inherited from VNetworkBase
 
#ifndef TFCSSIMPLELWTNNHANDLER_H
#define TFCSSIMPLELWTNNHANDLER_H
 
#include "ISF_FastCaloSimEvent/VNetworkLWTNN.h"
#include <iostream>
 
// Becuase we have a field of type LightweightNeuralNetwork
#include "lwtnn/LightweightNeuralNetwork.hh"
 
// For writing to a tree
#include "TTree.h"
 
class TFCSSimpleLWTNNHandler : public VNetworkLWTNN {
public:
  // Don't lose the default constructor
  using VNetworkLWTNN::VNetworkLWTNN;

  explicit TFCSSimpleLWTNNHandler(const std::string &inputFile);

  TFCSSimpleLWTNNHandler(const TFCSSimpleLWTNNHandler &copy_from);

  NetworkOutputs compute(NetworkInputs const &inputs) const override;

  std::vector<std::string> getOutputLayers() const override;
 
protected:
  void setupNet() override;
 
private:
  // unique ptr deletes the object when it goes out of scope
  std::unique_ptr<lwt::LightweightNeuralNetwork>
      m_lwtnn_neural; 
  std::vector<std::string> m_outputLayers; 
 
  // Suppling a ClassDef for writing to file.
  ClassDefOverride(TFCSSimpleLWTNNHandler, 1);
};
 
#endif // TFCSSIMPLELWTNNHANDLER_H