abstract_execution_pipeline.h

// NIST-developed software is provided by NIST as a public service. You may use, copy and distribute copies of the
// software in any medium, provided that you keep intact this entire notice. You may improve, modify and create
// derivative works of the software or any portion of the software, and you may copy and distribute such modifications
// or works. Modified works should carry a notice stating that you changed the software and should note the date and
// nature of any such change. Please explicitly acknowledge the National Institute of Standards and Technology as the
// source of the software. NIST-developed software is expressly provided "AS IS." NIST MAKES NO WARRANTY OF ANY KIND,
// EXPRESS, IMPLIED, IN FACT OR ARISING BY OPERATION OF LAW, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTY OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, NON-INFRINGEMENT AND DATA ACCURACY. NIST NEITHER REPRESENTS NOR
// WARRANTS THAT THE OPERATION OF THE SOFTWARE WILL BE UNINTERRUPTED OR ERROR-FREE, OR THAT ANY DEFECTS WILL BE
// CORRECTED. NIST DOES NOT WARRANT OR MAKE ANY REPRESENTATIONS REGARDING THE USE OF THE SOFTWARE OR THE RESULTS
// THEREOF, INCLUDING BUT NOT LIMITED TO THE CORRECTNESS, ACCURACY, RELIABILITY, OR USEFULNESS OF THE SOFTWARE. You
// are solely responsible for determining the appropriateness of using and distributing the software and you assume
// all risks associated with its use, including but not limited to the risks and costs of program errors, compliance
// with applicable laws, damage to or loss of data, programs or equipment, and the unavailability or interruption of 
// operation. This software is not intended to be used in any situation where a failure could cause risk of injury or
// damage to property. The software developed by NIST employees is not subject to copyright protection within the
// United States.


#ifndef HEDGEHOG_ABSTRACT_EXECUTION_PIPELINE_H
#define HEDGEHOG_ABSTRACT_EXECUTION_PIPELINE_H

#include <numeric>
#include "graph.h"
#include "../core/defaults/core_default_execution_pipeline.h"

namespace hh {

template<class GraphOutput, class ...GraphInputs>
class AbstractExecutionPipeline
    : public behavior::MultiReceivers<GraphInputs...>,
      public behavior::Sender<GraphOutput>,
      public behavior::SwitchRule<GraphInputs> ... {
  static_assert(traits::isUnique<GraphInputs...>, "A node can't accept multiple inputs with the same type.");
  static_assert(sizeof... (GraphInputs) >= 1, "A node need to have one output type and at least one output type.");
 private:
  std::shared_ptr<Graph<GraphOutput, GraphInputs...>>
      graph_ = nullptr; 

  std::shared_ptr<core::CoreDefaultExecutionPipeline<GraphOutput, GraphInputs...>>
      coreExecutionPipeline_ = nullptr; 

 public:

  AbstractExecutionPipeline() = delete;

  AbstractExecutionPipeline(std::shared_ptr<Graph<GraphOutput, GraphInputs...>> graph,
                            size_t const &numberGraphDuplications,
                            bool iota = false) {
    std::vector<int> deviceIds(numberGraphDuplications, 0);
    if (iota) { std::iota(deviceIds.begin(), deviceIds.end(), 0); }
    graph_ = graph;
    coreExecutionPipeline_ = std::make_shared<core::CoreDefaultExecutionPipeline<GraphOutput, GraphInputs...>>(
        "AbstractExecutionPipeline", this,
        std::dynamic_pointer_cast<core::CoreGraph<GraphOutput, GraphInputs...>>(graph->core()),
        numberGraphDuplications, deviceIds, false);
  }

  AbstractExecutionPipeline(std::shared_ptr<Graph<GraphOutput, GraphInputs...>> graph,
                            size_t const &numberGraphDuplications,
                            std::vector<int> const deviceIds, bool automaticStart = false)
      : graph_(graph),
        coreExecutionPipeline_(std::make_shared<core::CoreDefaultExecutionPipeline<GraphOutput, GraphInputs...>>(
            "AbstractExecutionPipeline",
            this,
            std::dynamic_pointer_cast<core::CoreGraph<GraphOutput, GraphInputs...>>(graph->core()),
            numberGraphDuplications,
            deviceIds, automaticStart)) {}

  AbstractExecutionPipeline(std::string_view const &name,
                            std::shared_ptr<Graph<GraphOutput, GraphInputs...>> const &graph,
                            size_t const &numberGraphDuplications,
                            std::vector<int> const &deviceIds,
                            bool automaticStart = false)
      : graph_(graph),
        coreExecutionPipeline_(std::make_shared<core::CoreDefaultExecutionPipeline<GraphOutput, GraphInputs...>>(
            name,
            this,
            std::dynamic_pointer_cast<core::CoreGraph<GraphOutput, GraphInputs...>>(graph->core()),
            numberGraphDuplications,
            deviceIds,
            automaticStart)
        ) {}

  virtual ~AbstractExecutionPipeline() = default;

  std::shared_ptr<core::CoreNode> core() override { return this->coreExecutionPipeline_; }

  std::shared_ptr<Graph<GraphOutput, GraphInputs...>> const &graph() const { return graph_; }
};
}

#endif //HEDGEHOG_ABSTRACT_EXECUTION_PIPELINE_H