如何将矢量(或类似)传递给可变参数模板

时间:2018-02-28 08:09:35

标签: c++ c++11 templates variadic-templates

假设我有以下代码:

template <typename... Args>
void DoSomething(const Args&... args)
{
    for (const auto& arg : {args...})
    {
        // Does something
    }
}

现在让我说我是从另一个函数调用它,并希望传入一个std::vector(或以某种方式修改向量,以便它可以用于此)

void DoSomethingElse()
{
    // This is how I'd use the function normally
    DoSomething(50, 60, 25);

    // But this is something I'd like to be able to do as well
    std::vector<int> vec{50, 60, 25};
    DoSomething(??); // <- Ideally I'd pass in "vec" somehow
}

有没有这样做?我还考虑使用std::initializer_list而不是可变参数模板,但问题仍然是我无法传递现有数据。

谢谢。

2 个答案:

答案 0 :(得分:2)

假设语法DoSomething({50, 60, 25})可以接受,您可以先为容器编写非变量函数模板:

template <typename T>
void DoSomething(const T& coll) 
{
    for (const auto& arg : coll) {
        // ...
    }
}

然后,std::initializer_list<>的非变量函数模板:

template<typename T>
void DoSomething(const std::initializer_list<T>& lst)
{
    for (const auto& elem: lst) {
       // ...
    }
}

他们可以这样使用:

void DoSomethingElse()
{
    std::vector<int> vec{50, 60, 25};
    std::list<int> lst{50, 60, 25};

    // 1st function template
    DoSomething(vec);
    DoSomething(lst);

    // 2nd function template
    DoSomething({50, 60, 25});
}

为了避免代码重复,第二个函数模板可以从std::vector参数创建std::initializer_list,然后使用该向量调用另一个函数模板:

template<typename T>
void DoSomething(const std::initializer_list<T>& lst)
{
    std::vector<T> vec(lst);
    DoSomething(vec);
}

答案 1 :(得分:2)

这是一种使用SFINAE的方法。 传递一个元素,它将被假定为ranged for-loop

如果你传递了几个参数,它会构造一个向量并迭代它。

#include <iostream>
#include <type_traits>
#include <vector>

template <typename... Args, typename std::enable_if<(sizeof...(Args) > 1), int>::type = 0>
void DoSomething(const Args&... args)
{
    for (auto& a : {typename std::common_type<Args...>::type(args)...})
    {
        cout << a << endl;
    }
}

template <typename Arg>
void DoSomething(Arg& arg)
{
    for (auto a : arg)
    {
        std::cout << a << std::endl;
    }
}

int main() {
    DoSomething(10, 50, 74);

    std::vector<int> foo = {12,15,19};
    DoSomething(foo);
    return 0;
}