execution-graph.hpp

/*
 * See LICENSE for more information about licensing
 *
 * Copyright 2024
 * Author: Luis G. Leon-Vega <luis.leon@ieee.org>
 *
 */
#pragma once
#include <condition_variable>  // NOLINT
#include <cynq/enums.hpp>
#include <cynq/status.hpp>
#include <functional>
#include <memory>
#include <mutex>  // NOLINT
#include <string>
#include <vector>
 
namespace cynq {
struct ExecutionGraphParameters {
  std::string name;
  uint64_t timeout = 100;
  virtual ~ExecutionGraphParameters() = default;
};
 
class IExecutionGraph {
 public:
  typedef int NodeID;
 
  typedef std::function<Status()> Function;
 
  enum Type {
    None = 0,
    STREAM
  };
 
  virtual IExecutionGraph::NodeID Add(
      const Function &function,
      const std::vector<NodeID> dependencies = std::vector<NodeID>(0)) = 0;
 
  virtual Status Sync(const NodeID node = -1) = 0;
 
  virtual Status GetLastError() = 0;
 
  virtual ~IExecutionGraph() = default;
 
  static std::shared_ptr<IExecutionGraph> Create(
      const IExecutionGraph::Type type,
      const std::shared_ptr<ExecutionGraphParameters> params);
 
  struct Node {
    NodeID id;
    Function function;
    std::vector<NodeID> dependencies = {};
    std::vector<Node *> parents = {};
    std::vector<Node *> children = {};
  };
};
}  // namespace cynq