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

16

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 variadic 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 bar(1,2,3,4) appeler foo(1,2)etfoo(3,4)

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

Fourmet
la source
4
Il est dangereux de transmettre deux fois les mêmes
arguments

Réponses:

13

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
}

VIVRE


Notez qu'avec l'extrait de code minimum ci-dessus lorsque vous appelez baravec 0 ou des arguments impairs, vous n'obtiendrez aucune erreur de fonction correspondante . Si vous voulez un message de compilation plus clair avec static_assertvous pouvez commencer à partir de cet extrait .

songyuanyao
la source
5

Récursivité simple 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 la réponse de songyanyao est assez canonique pré-C ++ 17. Ensuite, cela if constexprnous a permis de déplacer la logique dans le corps de nos fonctions au lieu d'utiliser des astuces de surcharge.

AndyG
la source
1
La version de songyanyao est assez facile à étendre, de sorte qu'elle prend également la fonction qu'elle applique comme argument. À mon avis, c'est assez agréable, car cela nous permet d'appliquer ce modèle plusieurs fois sans avoir à écrire la logique à chaque fois. Existe-t-il une version de votre réponse qui autorise la même chose?
n314159
1
@ n314159: Quelque chose comme ça ?
AndyG
1
Exactement! Je vous remercie. Personnellement, je préfère cela, car (en plus de ce que j'ai déjà dit), il sépare la logique d'application de la fonction à laquelle il est appliqué.
n314159
2

Généralisation C ++ 17 pour les nfoncteurs -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
la source