TFCSNetworkFactory.h

Go to the documentation of this file.

#ifndef TFCSNETWORKFACTORY_H
#define TFCSNETWORKFACTORY_H
 
#include "ISF_FastCaloSimEvent/VNetworkBase.h"
#include <memory>
#include <string>
#include <vector>
#include <filesystem>
 
class TFCSNetworkFactory {
public:
  // Unspecified TFCSGANLWTNNHandler or TFCSSimpleLWTNNHandler, take a guess
  static std::unique_ptr<VNetworkBase> create(std::string input);
  // Specified TFCSGANLWTNNHandler or TFCSSimpleLWTNNHandler
  static std::unique_ptr<VNetworkBase> create(std::string input,
                                              bool graph_form);

  static std::unique_ptr<VNetworkBase> create(std::vector<char> const &input);

  static std::unique_ptr<VNetworkBase>
  create(std::vector<char> const &vector_input, std::string string_input);
  static std::unique_ptr<VNetworkBase>
  create(std::vector<char> const &vector_input, std::string string_input,
         bool graph_form);
 
private:
  static void resolveGlobs(std::string &filename);

  static bool isOnnxFile(std::string const &filename);
};
 
#endif // TFCSNETWORKFACTORY_H