RModel_GraphIndependent.hxx

Go to the documentation of this file.

#ifndef TMVA_SOFIE_RMODEL_GraphIndependent
#define TMVA_SOFIE_RMODEL_GraphIndependent
 
#include <ctime>
 
#include "TMVA/RModel_Base.hxx"
#include "TMVA/RModel.hxx"
#include "TMVA/RFunction.hxx"
 
namespace TMVA {
namespace Experimental {
namespace SOFIE {
 
class RFunction_Update;
 
struct GraphIndependent_Init {
   // update blocks
   std::unique_ptr<RFunction_Update> edges_update_block;
   std::unique_ptr<RFunction_Update> nodes_update_block;
   std::unique_ptr<RFunction_Update> globals_update_block;
 
   std::size_t num_nodes;
   std::vector<std::pair<int, int>> edges;
 
   int num_node_features;
   int num_edge_features;
   int num_global_features;
 
   std::string filename;
 
   template <typename T>
   void createUpdateFunction(T &updateFunction)
   {
      switch (updateFunction.GetFunctionTarget()) {
      case FunctionTarget::EDGES: {
         edges_update_block.reset(new T(updateFunction));
         break;
      }
      case FunctionTarget::NODES: {
         nodes_update_block.reset(new T(updateFunction));
         break;
      }
      case FunctionTarget::GLOBALS: {
         globals_update_block.reset(new T(updateFunction));
         break;
      }
      default: {
         throw std::runtime_error(
            "TMVA SOFIE: Invalid Update function supplied for creating GraphIndependent function block.");
      }
      }
   }
 
   ~GraphIndependent_Init()
   {
      edges_update_block.reset();
      nodes_update_block.reset();
      globals_update_block.reset();
   }
};
 
class RModel_GraphIndependent final : public RModel_GNNBase {
 
private:
   // updation function for edges, nodes & global attributes
   std::unique_ptr<RFunction_Update> edges_update_block;
   std::unique_ptr<RFunction_Update> nodes_update_block;
   std::unique_ptr<RFunction_Update> globals_update_block;
 
   std::size_t num_nodes;
   std::size_t num_edges;
 
   std::size_t num_node_features;
   std::size_t num_edge_features;
   std::size_t num_global_features;
 
public:
   /**
       Default constructor. Needed to allow serialization of ROOT objects. See
       https://root.cern/manual/io_custom_classes/#restrictions-on-types-root-io-can-handle
   */
   RModel_GraphIndependent() = default;
   RModel_GraphIndependent(GraphIndependent_Init &graph_input_struct);
 
   // Rule of five: explicitly define move semantics, disallow copy
   RModel_GraphIndependent(RModel_GraphIndependent &&other);
   RModel_GraphIndependent &operator=(RModel_GraphIndependent &&other);
   RModel_GraphIndependent(const RModel_GraphIndependent &other) = delete;
   RModel_GraphIndependent &operator=(const RModel_GraphIndependent &other) = delete;
   ~RModel_GraphIndependent() final = default;
 
   void Generate() final;
};
 
} // namespace SOFIE
} // namespace Experimental
} // namespace TMVA
 
#endif // TMVA_SOFIE_RMODEL_GNN