std_glue.h

// -*- mode: c++; tab-width: 4; indent-tabs-mode: t; eval: (progn (c-set-style "stroustrup") (c-set-offset 'innamespace 0)); -*-
// vi:set ts=4 sts=4 sw=4 noet :
// Copyright 2011, 2012, The TPIE development team
// 
// This file is part of TPIE.
// 
// TPIE is free software: you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License as published by the
// Free Software Foundation, either version 3 of the License, or (at your
// option) any later version.
// 
// TPIE is distributed in the hope that it will be useful, but WITHOUT ANY
// WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
// License for more details.
// 
// You should have received a copy of the GNU Lesser General Public License
// along with TPIE.  If not, see <http://www.gnu.org/licenses/>

#ifndef __TPIE_PIPELINING_STD_GLUE_H__
#define __TPIE_PIPELINING_STD_GLUE_H__

#include <vector>

#include <tpie/pipelining/node.h>
#include <tpie/pipelining/pipe_base.h>
#include <tpie/pipelining/factory_helpers.h>

namespace tpie {

namespace pipelining {

namespace bits {

template <typename dest_t>
class input_vector_t : public node {
public:
    typedef typename dest_t::item_type item_type;

    inline input_vector_t(const dest_t & dest, const std::vector<item_type> & input) : dest(dest), input(input) {
        add_push_destination(dest);
        set_name("Read", PRIORITY_INSIGNIFICANT);
    }

    virtual void propagate() override {
        forward("items", static_cast<stream_size_type>(input.size()));
        set_steps(input.size());
    }

    virtual void go() override {
        typedef typename std::vector<item_type>::const_iterator IT;
        for (IT i = input.begin(); i != input.end(); ++i) {
            dest.push(*i);
            step();
        }
    }
private:
    dest_t dest;
    const std::vector<item_type> & input;
};

template <typename T>
class output_vector_t : public node {
public:
    typedef T item_type;

    inline output_vector_t(std::vector<T> & output) : output(output) {
        set_name("Write", PRIORITY_INSIGNIFICANT);
    }

    inline void push(const T & item) {
        output.push_back(item);
    }
private:
    std::vector<item_type> & output;
};

template <typename F>
class lambda_t {
public:
    template <typename dest_t>
    class type: public node {
    public:
        typedef typename F::argument_type item_type;
        
        type(const dest_t & dest, const F & f): f(f), dest(dest) {
            set_name("Lambda", PRIORITY_INSIGNIFICANT);
        }
        
        void push(const item_type & item) {
            dest.push(f(item));
        }
    private:
        F f;
        dest_t dest;
    };
};

template <typename F>
class exclude_lambda_t {
public:
    template <typename dest_t>
    class type: public node {
    public:
        typedef typename F::argument_type item_type;
        
        type(const dest_t & dest, const F & f): f(f), dest(dest) {
            set_name("Lambda", PRIORITY_INSIGNIFICANT);
        }
        
        void push(const item_type & item) {
            typename F::result_type t=f(item);
            if (t.second) dest.push(t.first);
        }
    private:
        F f;
        dest_t dest;
    };
};


} // namespace bits

template<typename T>
inline pipe_begin<factory_1<bits::input_vector_t, const std::vector<T> &> > input_vector(const std::vector<T> & input) {
    return factory_1<bits::input_vector_t, const std::vector<T> &>(input);
}

template <typename T>
inline pipe_end<termfactory_1<bits::output_vector_t<T>, std::vector<T> &> > output_vector(std::vector<T> & output) {
    return termfactory_1<bits::output_vector_t<T>, std::vector<T> &>(output);
}

template <typename F>
inline pipe_middle<tempfactory_1<bits::lambda_t<F>, F> > lambda(const F & f) {
    return tempfactory_1<bits::lambda_t<F>, F>(f);
}

template <typename F>
inline pipe_middle<tempfactory_1<bits::exclude_lambda_t<F>, F> > exclude_lambda(const F & f) {
    return tempfactory_1<bits::exclude_lambda_t<F>, F>(f);
}


} // namespace pipelining

} // namespace tpie

#endif