Modelos variáveis: desdobrar argumentos em grupos

16

Eu tenho uma função que leva dois argumentos:

template <typename T1, typename T2>
void foo(T1 arg1, T2 arg2)
{ std::cout << arg1 << " + " << arg2 << '\n'; }

E uma variável que deve encaminhar seus argumentos em pares:

template <typename... Args>
void bar(Args&&... args) {
    static_assert(sizeof...(Args) % 2 == 0);

    ( foo( std::forward<Args>(args), std::forward<Args>(args) ), ... );
    // ^ Sends each argument twice, not in pairs
}

Eu gostaria bar(1,2,3,4) de ligar foo(1,2)efoo(3,4)

Existe uma maneira de fazer isso?

Fourmet
fonte
4
É perigoso para encaminhar os mesmos argumentos duas vezes
AndyG

Respostas:

13

Você pode fazer isso com sobrecargas.

template <typename T1, typename T2>
void bar(T1&& arg1, T2&& arg2) {
    foo( std::forward<T1>(arg1), std::forward<T2>(arg2) ); // (until) sends (the last) two arguments to foo
}

template <typename T1, typename T2, typename... Args>
void bar(T1&& arg1, T2&& arg2, Args&&... args) {
    foo( std::forward<T1>(arg1), std::forward<T2>(arg2) ); // sends the 1st two arguments to foo
    bar( std::forward<Args>(args)... );                    // call bar with remaining elements recursively
}

VIVER


Observe que, com o snippet mínimo acima, ao chamar barcom 0 ou argumentos ímpares, você não receberá nenhum erro de função correspondente . Se você quiser uma mensagem de compilação mais clara, static_assertpode começar com esse trecho .

songyuanyao
fonte
5

Recursão simples usando if constexpr:

// print as many pairs as we can
template<class T, class U, class... Args>
void foo(T t, U u, Args&&... args)
{
    std::cout << t << " + " << u << "\n";
    if constexpr(sizeof...(Args) > 0 && sizeof...(Args) % 2 == 0)
        foo(std::forward<Args>(args)...);
}

template<class... Args>
void bar(Args&&... args)
{
    static_assert(sizeof...(Args) % 2 == 0);
    foo(std::forward<Args>(args)...);
}

Chame assim:

bar(1, 2, 3, 4);

Demo

Eu diria que a resposta de songyanyao é bastante canônica antes do C ++ 17. Depois, if constexprnos permitiu mover a lógica para os corpos de nossas funções, em vez de usar truques de sobrecarga.

AndyG
fonte
11
A versão de songyanyao é muito fácil de estender, de modo que também assume a função que está sendo aplicada como argumento. Na minha opinião, isso é muito bom, pois nos permite aplicar esse padrão várias vezes sem ter que escrever a lógica toda vez. Existe uma versão da sua resposta que permita o mesmo?
N314159
11
@ n314159: Algo como isso ?
AndyG 24/11/19
11
Exatamente! Obrigado. Pessoalmente, prefiro isso, pois (além do que eu já disse) separa a lógica de aplicação da função à qual é aplicada.
N314159
2

Generalização do C ++ 17 para nfunctores -ary:

namespace impl
{
    template<std::size_t k, class Fn, class Tuple, std::size_t... js>
    void unfold_nk(Fn fn, Tuple&& tuple, std::index_sequence<js...>) {
        fn(std::get<k + js>(std::forward<Tuple>(tuple))...);
    }

    template<std::size_t n, class Fn, class Tuple, std::size_t... is>
    void unfold_n(Fn fn, Tuple&& tuple, std::index_sequence<is...>) {
        (unfold_nk<n * is>(fn, std::forward<Tuple>(tuple), 
            std::make_index_sequence<n>{}), ...);
    }
}

template<std::size_t n, class Fn, typename... Args>
void unfold(Fn fn, Args&&... args) {
    static_assert(sizeof...(Args) % n == 0);
    impl::unfold_n<n>(fn, std::forward_as_tuple(std::forward<Args>(args)...), 
        std::make_index_sequence<sizeof...(Args) / n>{});
}

int main() {
    auto fn = [](auto... args) { 
        (std::cout << ... << args) << ' ';
    };

    unfold<2>(fn, 1, 2, 3, 4, 5, 6);   // Output: 12 34 56
    unfold<3>(fn, 1, 2, 3, 4, 5, 6);   // Output: 123 456
}
Evg
fonte