filter_sys.hpp

#pragma once
 
#include <functional_dag/error_codes.h>
 
#include <functional_dag/dag_interface.hpp>
#include <functional_dag/impl/dag_impl.hpp>
 
namespace fn_dag {
using namespace std;

template <typename IDType>
class dag_manager {
 private:
  _dag_context m_context;  // This is the "global" context used by all dags
 
 public:
  vector<_dag_base<IDType> *> m_all_dags;

  dag_manager() {
    m_context.filter_off = false;
    m_context.run_single_threaded = false;
  }

  ~dag_manager() {
    m_context.filter_off = true;
    clear();
  }

  template <typename In, typename Out>
  [[nodiscard]] expected<IDType, error_codes> add_node(
      IDType _id, dag_node<In, Out> *_new_filter, const IDType &_onto) {
    if (_new_filter != nullptr) {
      for (auto t = m_all_dags.cbegin(); t != m_all_dags.cend(); t++) {
        if ((*t)->dag_contains(_onto) || (*t)->get_id() == _onto) {
          dag<In, IDType> *tptr = static_cast<dag<In, IDType> *>(*t);
          return tptr->add_filter(_id, _new_filter, _onto);
        }
      }
    } else {
      delete _new_filter;
      return unexpected(error_codes::NULL_PTR_ERROR);
    }
    delete _new_filter;
    return unexpected(error_codes::PARENT_NOT_FOUND);
  }

  bool manager_contains_id(IDType _id) {
    for (auto t = m_all_dags.cbegin(); t != m_all_dags.cend(); t++) {
      if ((*t)->dag_contains(_id) || (*t)->get_id() == _id) return true;
    }
    return false;
  }

  void set_indention_string(string _new_indent_str) {
    m_context.indent_str = _new_indent_str;
  }

  void set_logging_stream(ostream *_new_stream) { m_context.log = _new_stream; }

  void run_single_threaded(const bool _is_single_threaded) {
    m_context.run_single_threaded = _is_single_threaded;
  }

  template <typename Out>
  expected<dag<Out, IDType> *, error_codes> add_dag(
      IDType _id, dag_source<Out> *_new_filter, bool _startImmediately) {
    if (_new_filter != nullptr) {
      dag<Out, IDType> *t =
          new dag<Out, IDType>(_id, _new_filter, m_context, _startImmediately);
      m_all_dags.push_back(t);
      return t;
    }
    return unexpected(error_codes::NULL_PTR_ERROR);
  }

  void print_all_dags() {
    *m_context.log << endl;
    *m_context.log << "------------DAG Forest-----------\n";
    for (auto t = m_all_dags.cbegin(); t != m_all_dags.cend(); t++)
      (*t)->print();
    *m_context.log << "---------------------------------\n";
  }

  void stahp() { m_context.filter_off = true; }

  void clear() {
    for (auto t = m_all_dags.begin(); t != m_all_dags.end(); t++) delete *t;
    m_all_dags.clear();
  }
};
}  // namespace fn_dag