fn_dag_interface.hpp

#pragma once
 
#include <functional>
#include <functional_dag/dag_interface.hpp>
 
namespace fn_dag {
using namespace std;
template <typename Out>
class __dag_source : public dag_source<Out> {
 public:
  function<unique_ptr<Out>()> m_generator;  // Generator lambda function

  __dag_source(function<unique_ptr<Out>()> _generator)
      : m_generator(_generator) {}

  ~__dag_source() {}

  unique_ptr<Out> update() { return m_generator(); };
};

template <typename In, typename Out>
class __dag_node : public dag_node<In, Out> {
 public:
  function<unique_ptr<Out>(const In *const)>
      m_update;  // Mapping lambda function

  __dag_node(function<unique_ptr<Out>(const In *const)> _update)
      : m_update(_update) {}

  ~__dag_node() {}

  unique_ptr<Out> update(const In *const _data) { return m_update(_data); };
};

template <typename Out>
dag_source<Out> *fn_source(function<unique_ptr<Out>()> _run_fn) {
  return new __dag_source(_run_fn);
}

template <typename In, typename Out>
dag_node<In, Out> *fn_call(function<unique_ptr<Out>(const In *const)> _run_fn) {
  return new __dag_node(_run_fn);
}
 
}  // namespace fn_dag