接受lambda作为函数参数并提取返回类型

时间:2014-08-18 04:43:50

标签: c++ templates c++11 lambda

我需要接受lambda函数作为参数并找出它的返回类型。到目前为止,我无法想出任何可行的方法。这是一个例子:

template <typename T1, typename T2>
T1 foo(T2 arg, std::function<T1(T2)> f)
{
    return f(arg);
}

...

int n = foo(3, [](int a){ return a + 2; });   <-- ERROR!

怎么做?提升解决方案也可以。

2 个答案:

答案 0 :(得分:6)

您不应将std::function作为参数传递。它有一些开销,因为它使用类型擦除来存储任何类型的可调用,如函数指针,仿函数,lambda(它们只是一个自动生成的函子)等。相反,你应该只模板化函数类型。然后,您可以使用尾随返回类型来确定函数的返回类型:

template <typename T, typename Function>
auto foo(T arg, Function f) -> decltype(f(arg))
{
    return f(arg);
}

答案 1 :(得分:5)

std::function和lambda不相同,它们是不同的类型。前者是一个可以存储任何类型的可调用对象的容器,而后者是一个可调用对象。将lambda分配给std::function是有效的,因为std::function具有执行必要转换的构造函数。但是,模板参数推导不是这种情况,这需要完全匹配,并且不考虑用户定义的转换。这就是你的代码无法编译的原因。

解决方案很简单,正如Rapptz的answer所述,修改foo以将callable作为模板参数。

template <typename Arg, typename Func>
auto foo(Arg arg, Func f)
    -> decltype(f(arg))
{
    return f(arg);
}