filter.hpp

Go to the documentation of this file.

/*
Copyright© 2014 Akihiro Nishimura

This software is released under the MIT License.
http://opensource.org/licenses/mit-license.php
*/

#ifndef SIG_UTIL_FILTER_HPP
#define SIG_UTIL_FILTER_HPP

#include "../helper/helper_modules.hpp"
#include "../helper/container_helper.hpp"


namespace sig
{

template <class F, class C,
    class CR = typename impl::remove_const_reference<C>::type,
    class ET = typename impl::forward_element<C>::type
>
auto filter(F&& pred, C&& list)
{
    CR result = impl::container_traits<CR>::make(list.size());

    for (auto&& e : std::forward<C>(list)){
        if (std::forward<F>(pred)(e)) impl::container_traits<CR>::add_element(result, std::forward<ET>(e));
    }
    return result;
}


template <class F, class C,
    class CR = typename impl::remove_const_reference<C>::type,
    class ET = typename impl::forward_element<C>::type
>
auto filter(F&& pred, int init, C&& list)
{
    CR result = impl::container_traits<CR>::make(list.size());

    for (auto&& e : std::forward<C>(list)){
        if (std::forward<F>(pred)(init, e)) impl::container_traits<CR>::add_element(result, std::forward<ET>(e));
        ++init;
    }
    return result;
}



template <class F, class C,
    class CR = typename impl::remove_const_reference<C>::type,
    class ET = typename impl::forward_element<C>::type
>
auto partition(F&& pred, C&& list)
{
    CR result1 = impl::container_traits<CR>::make(list.size());
    CR result2 = impl::container_traits<CR>::make(list.size());

    for (auto&& e : std::forward<C>(list)){
        if (std::forward<F>(pred)(e)) impl::container_traits<CR>::add_element(result1, std::forward<ET>(e));
        else impl::container_traits<CR>::add_element(result2, std::forward<ET>(e));
    }
    return std::make_tuple(std::move(result1), std::move(result2));
}

}
#endif