web-dev-qa-db-fra.com

Modèles variadiques: dépliez les arguments en groupes

J'ai une fonction qui prend deux arguments:

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

Et un variadique qui devrait transmettre ses arguments par paires:

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
}

Je voudrais que bar(1,2,3,4) appelle foo(1,2) et foo(3,4)

Y-a-t-il un moyen de faire ça ?

c++c++11templatesvariadic-templates
16
Fourmet

Vous pouvez l'accomplir avec des surcharges.

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
}

EN DIRECT

Notez qu'avec l'extrait de code minimum ci-dessus lors de l'appel de bar avec 0 ou des arguments impairs, vous n'obtiendrez aucune erreur de fonction correspondante . Si vous voulez un message de compilation plus clair avec static_assert vous pouvez commencer à partir de cela extrait .

13
songyuanyao

Récursivité simple en utilisant 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)...);
}

Appelez-le ainsi:

bar(1, 2, 3, 4);

Démo

Je dirais que réponse de songyanyao est un pré-C++ 17 assez canonique. Ensuite, if constexpr nous a permis de déplacer la logique dans le corps de nos fonctions au lieu d'utiliser des astuces de surcharge.

5
AndyG

Généralisation C++ 17 pour les foncteurs n- 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
}
2
Evg