2015-04-27 17 views
2

忽略缺少的完美转发。 (假设参数在实际执行完美转发。)获取第n个可变参数值(不是类型)

// Base case: no args 
template<typename TF> void forEach2Args(TF) { } 

// Recursive case: some args 
template<typename TF, typename... Ts> void forEach2Args(TF mFn, Ts... mXs) 
{ 
    mFn(getNth<0>(mXs...), getNth<1>(mXs...)); 
    forEach2Args(mFn, getAllAfter<2>(mXs...)); 
} 

int main() 
{ 
    int result{0}; 
    forEach2Args([&result](auto a1, auto a2) 
    { 
     result += (a1 * a2); 
    }, 2, 4, 3, 6); 

    // roughly evaluates to: 
    //  result += (2 * 4); 
    //  result += (3 * 6); 
} 

是否有可能实现getNthgetAllAfter避免任何可能的运行时开销?到目前为止,我发现的唯一解决方案是将第一个forEach2Args调用中的每个Ts...放在std::tuple的内部,然后将每个递归调用的非常量引用传递给该元组。我几乎可以确定有不必要的移动/ ctor/dtor调用。

另一种解决方案是使用这样的:

// Base case: no args 
template<typename TF> void forEach2Args(TF) { } 

// Recursive case: some args 
template<typename TF, typename T1, typename T2, typename... Ts> 
void forEach2Args(TF mFn, T1 mX1, T2 mX2, Ts... mXs) 
{ 
    mFn(mX1, mX2); 
    forEach2Args(mFn, mXs...); 
} 

但是这种解决方案需要,如果我想通过在3个而不是2,或任何其他数量的组参数,再次实现。我想要一些动态的地方,我可以指定通过模板参数传递给每个mFn调用的参数数量。喜欢的东西:

forEachNArgs<3>([](auto a1, auto a2, auto a3){ /*...*/ }, /*...*/); 
forEachNArgs<4>([](auto a1, auto a2, auto a3, auto a4){ /*...*/ }, /*...*/); 
+2

使用'的std :: tie'有不复制其参数的元组。 –

+0

@DanielFrey:看起来'std :: tie'带有l值引用。如果我在'mFn'调用期间传递给'forEach2Args'的r值引用,它会阻止正确的转发吗? –

+1

@VittorioRomeo,看来正确的方法是'std :: forward_as_tuple(std :: forward (values)...)' – chris

回答

7

忽略的要求完美的转发,这应该工作:

template<typename B, typename C> 
struct forEachNArgsImpl; 

template<std::size_t... Bs, std::size_t... Cs> 
struct forEachNArgsImpl< 
    std::index_sequence<Bs...>, 
    std::index_sequence<Cs...> 
> 
{ 
    template<std::size_t N, typename TF, typename... Ts> 
    static void execN(TF mFn, const std::tuple<Ts...>& mXs) 
    { 
     mFn(std::get< N + Cs >(mXs)...); 
    } 

    template<typename TF, typename... Ts> 
    static void exec(TF mFn, const std::tuple<Ts...>& mXs) 
    { 
     using swallow = bool[]; 
     (void)swallow{ (execN< Bs * sizeof...(Cs) >(mFn, mXs), true)... }; 
    } 
}; 

template<std::size_t N, typename TF, typename... Ts> 
void forEachNArgs(TF mFn, Ts... mXs) 
{ 
    static_assert(sizeof...(Ts) % N == 0, "Wrong number of arguments"); 
    forEachNArgsImpl< 
     std::make_index_sequence<sizeof...(Ts)/N>, 
     std::make_index_sequence<N> 
    >::exec(mFn, std::forward_as_tuple(mXs...)); 
} 

Live example

+0

啊,我看到你的编辑太晚了...... –

+0

在几分钟前添加了关于这个可能的实现的编辑 –

+0

@VittorioRomeo我已经想出了一个更通用和**非递归**版本,请查看编辑答案。 –

2

以下可能会有所帮助:

namespace detail 
{ 

    template<std::size_t...IsN, std::size_t...Is, typename F> 
    void forEachNArgsImpl(std::index_sequence<IsN...>, std::index_sequence<Is...>, F) { } 

    template<std::size_t...IsN, std::size_t...Is, typename F, typename... Ts> 
    void forEachNArgsImpl(std::index_sequence<IsN...> isn, std::index_sequence<Is...>, F f, Ts... mXs) 
    { 
     f(std::get<IsN>(std::forward_as_tuple(std::forward<Ts>(mXs)...))...); 
     constexpr std::size_t N = sizeof...(IsN); 
     constexpr std::size_t is = sizeof...(Is); 
     forEachNArgsImpl(isn, 
         std::make_index_sequence<(is > N) ? sizeof...(Is) - N : 0>{}, 
         f, 
         std::get<N + Is>(std::forward_as_tuple(std::forward<Ts>(mXs)...))...); 
    } 

} 

template<std::size_t N, typename F, typename... Ts> void forEachNArgs(F f, Ts... args) 
{ 
    static_assert(sizeof...(Ts) % N == 0, "Wrong number of arguments"); 
    detail::forEachNArgsImpl(std::make_index_sequence<N>{}, std::make_index_sequence<sizeof...(Ts) - N>{}, f, std::forward<Ts>(args)...); 
} 

Demo

2

这个的核心是call_with_some,它需要一个可调用的和一系列的索引和可变参数,并用可变参数的索引调用可调用的。

一些指数助手:

template<size_t K, class indexes> 
struct offset_indexes; 
template<size_t K, size_t...Is> 
struct offset_indexes<K, std::index_sequence<Is...>>: 
    std::index_sequence<(K+Is)...> 
{}; 

call_with_some,SFINAE启用。

// SFINAE test optional, but why not: 
template<class F, class...Ts, size_t... Is> 
std::result_of_t< F(std::tuple_element_t< Is, std::tuple<Ts&&...> >...) > 
call_with_some(F&& f, std::index_sequence<Is...>, Ts&&... ts) { 
    return std::forward<F>(f)(
    std::get<Is>(
     std::forward_as_tuple(std::forward<Ts>(ts)...) 
    )... 
); 
} 

现在问题的肉。 call_by_n是存储另一个函数对象的函数对象。它采用偏移量,然后用它来调用的论点,即偏移(次nF的序列,传递n参数:

template<class F, size_t n> 
struct call_by_n { 
    F&& f; 
    // Offset... should be `<0, ..., sizeof...(Args)/n -1>` 
    template<size_t...Offset, class...Args> 
    void operator()(std::index_sequence<Offset...>, Args&&...args) { 
    static_assert(0==(sizeof...(Args)%n), "Number of args must be divisible by n"); 
    // <0,1,2,3,4,...,n-1> sequence: 
    using indexes = std::make_index_sequence<n>; 

    using discard=int[]; 

    // the unused array trick to expand an arbitrary call: 
    (void)discard{0,(
     (call_with_some(f, offset_indexes<Offset*n, indexes>{}, std::forward<Args>(args)...)) 
    ,void(),0)...}; 
    } 
    void operator()() {} // do nothing, naturally 
}; 

现在我们只是包装上面在你的界面:

template<size_t n, class F, class...Args> 
void forEachNArgs(F&& f, Args&&...args) { 
    static_assert((sizeof...(Args)%n)==0, "Wrong number of arguments"); 
    call_by_n<F,n>{std::forward<F>(f)}(std::make_index_sequence<sizeof...(Args)/n>{}, std::forward<Args>(args)...); 
} 

我把forEach2Args作为练习。

live example - 很好,没有错别字。

此版本现在可以进行“平坦”风格的调用,没有无限递归。递归调用的数量不会随着Args...n而增加。

discard诡计有点乱。我们创建一个满零的临时整数数组,并且作为副作用在参数包扩展中执行任意代码。临时的整数数组永远不会被读取,也不会被它的地址所取代,所以编译器可以消除它,因为它永远不会存在。

在C++ 1z中,使用,的折叠表达式可以让我们做类似的事情,而没有太多的样板或魔法。

1

这里是什么在C++ Now2014提出了一个变化:

#include <utility> 
#include <tuple> 
#include <cassert> 

struct type_erasure { }; 

template<class T> 
struct wrapper : type_erasure { 
    wrapper(T&& w) : w_(std::forward<T>(w)) { } 
    T&& w_; 
    decltype(auto) get() { return std::forward<T>(w_); } 
}; 

template<class T> 
wrapper<T> wrapper_for(T&& x) { 
    return { std::forward<T>(x) }; 
} 

template <typename ignore> 
struct lookup; 

template <std::size_t... ignore> 
struct lookup<std::index_sequence<ignore...>> { 
    template <typename nth> 
    static decltype(auto) 
    at_position(decltype(ignore, type_erasure())..., wrapper<nth> w, ...) { 
     return w.get(); 
    } 

    template<typename... Ts> 
    static auto 
    all_after(decltype(ignore, type_erasure())..., Ts&&... args) { 
     return std::forward_as_tuple(args.get()...); 
    } 
}; 

template<std::size_t index, typename... Args> 
auto getNth(Args&&... args) { 
    return lookup<std::make_index_sequence<index>>::at_position(
     wrapper_for(std::forward<Args>(args))... 
    ); 
} 

template<std::size_t index, typename... Args> 
auto getAllAfter(Args&&... args) { 
    return lookup<std::make_index_sequence<index + 1>>::all_after(
     wrapper_for(std::forward<Args>(args))... 
    ); 
} 

int main() 
{ 
    assert(getNth<0>(1, 2, 3) == 1); 
    assert(getNth<1>(1, 2, 3) == 2); 
    assert(getNth<2>(1, 2, 3) == 3); 

    assert(getAllAfter<2>(2, 4, 6, 8, 10) == std::make_tuple(8, 10)); 
} 
+0

这不回答OP的问题。我的意思是,它回答了问题的标题,但不是问题的主体。 – Yakk

+0

谢谢。更新。 – 0x499602D2

相关问题