simple_node.h

Go to the documentation of this file.

#ifndef GRPPI_FF_DETAIL_SIMPLE_NODE_H
#define GRPPI_FF_DETAIL_SIMPLE_NODE_H

#include "fastflow_allocator.h"

#include <ff/allocator.hpp>
#include <ff/node.hpp>

namespace grppi {

namespace detail_ff {

template <typename Input, typename Output, typename Transformer>
class node_impl : public ff::ff_node_t<Input,Output> {
public:

  node_impl(Transformer && transform_op) : 
      transform_op_{transform_op}
  {}

  Output * svc(Input * p_item) {
    return new (ff_arena) Output{transform_op_(*p_item)};
  } 

private:
  Transformer transform_op_;
};

template <typename Output, typename Generator>
class node_impl<void,Output,Generator> : public ff::ff_node {
public:

  node_impl(Generator && generate_op) :
      generate_op_{generate_op}
  {}

  void * svc(void *) {
    std::experimental::optional<Output> result{generate_op_()};
    if (result) {
      return new (ff_arena) Output{*result};
    }
    else {
      return EOS;
    }
  }

private:
  Generator generate_op_;
};

template <typename Input, typename Consumer>
class node_impl<Input,void,Consumer> : public ff::ff_node_t<Input,void> {
public:

  node_impl(Consumer && consume_op) :
      consume_op_{consume_op}
  {}

  void * svc(Input * p_item) {
    consume_op_(*p_item);
    operator delete(p_item, ff_arena);
    return GO_ON;
  }

private:
  Consumer consume_op_;
};


} // namespace detail_ff

} // namespace grppi

#endif