假设我有以下代码:
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
而不是可变参数模板,但问题仍然是我无法传递现有数据。
谢谢。
这是一种使用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;
}
假设语法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);
}