TFCSGANLWTNNHandler.h

Go to the documentation of this file.

 
// Hopefully documentation gets inherited from VNetworkBase
 
#ifndef TFCSGANLWTNNHANDLER_H
#define TFCSGANLWTNNHANDLER_H
 
#include "ISF_FastCaloSimEvent/VNetworkLWTNN.h"
 
// Becuase we have a field of type LightweightGraph
#include "lwtnn/LightweightGraph.hh"
 
// For writing to a tree
#include <vector>
#include <memory>
#include <string>
 
class TFCSGANLWTNNHandler : public VNetworkLWTNN {
public:
  // Don't lose default constructors
  using VNetworkLWTNN::VNetworkLWTNN;

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

  TFCSGANLWTNNHandler(const TFCSGANLWTNNHandler &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::LightweightGraph> m_lwtnn_graph; 

  std::vector<std::string> m_outputLayers; 

  std::string *m_input = nullptr;
 
  // Suppling a ClassDef for writing to file.
  ClassDefOverride(TFCSGANLWTNNHandler, 6);
};
 
#endif // TFCSGANLWTNNHANDLER_H