如何从通用模板<typename t =“”>中提取lambda的返回类型和可变参数包

时间:2017-04-22 14:42:17

标签: c++ templates lambda variadic-templates type-deduction

我想创建一个模板化的类或函数,它接收一个lambda,并将其内部放在std :: function&lt;&gt;中 Lambda可以有任意数量的输入参数[](int a,float b,...)std :: function&lt;&gt;应该对应于lambda的operator()类型

template <typename T> 
void getLambda(T t) {
   // typedef lambda_traits::ret_type RetType; ??
   // typedef lambda_traits::param_tuple --> somehow back to parameter pack Args...
   std::function<RetType(Args...)> fun(t);
}

int main() {
    int x = 0;
    getLambda([&x](int a, float b, Person c){}); 
}

所以我需要以某种方式提取返回类型和参数包

回答here建议在lambda&#39; s :: operator()上使用部分规范

template <typename T>
struct function_traits : public function_traits<decltype(&T::operator())>
{};

template <typename ClassType, typename ReturnType, typename... Args>
struct function_traits<ReturnType(ClassType::*)(Args...) const>
// we specialize for pointers to member function
{
    enum { arity = sizeof...(Args) };
    // arity is the number of arguments.

    typedef ReturnType result_type;

    template <size_t i>
    struct arg
    {
        typedef typename std::tuple_element<i, std::tuple<Args...>>::type type;
        // the i-th argument is equivalent to the i-th tuple element of a tuple
        // composed of those arguments.
    };
};

但我需要一种方法来转换元组&lt;&gt;回到参数包,创建一个合适的std :: function&lt;&gt;实例

1 个答案:

答案 0 :(得分:4)

template <typename T>
struct function_traits : public function_traits<decltype(&T::operator())>
{};

template <typename ClassType, typename ReturnType, typename... Args>
struct function_traits<ReturnType(ClassType::*)(Args...) const>
// we specialize for pointers to member function
{
    using result_type = ReturnType;
    using arg_tuple = std::tuple<Args...>;
    static constexpr auto arity = sizeof...(Args);
};

template <class F, std::size_t ... Is, class T>
auto lambda_to_func_impl(F f, std::index_sequence<Is...>, T) {
    return std::function<typename T::result_type(std::tuple_element_t<Is, typename T::arg_tuple>...)>(f);
}

template <class F>
auto lambda_to_func(F f) {
    using traits = function_traits<F>;
    return lambda_to_func_impl(f, std::make_index_sequence<traits::arity>{}, traits{});
}

上面的代码应该做你想要的。如您所见,主要思想是创建一个整数包。这是variadics的非类型模板。我不知道你可以在不调用其他功能的情况下使用这种包的任何技术,所以通常在这些带有元组的情况下你会看到嵌套的&#34; impl&#34;完成所有工作的功能。获得整数包之后,在访问元组时展开它(用于获取值)。

在风格上注意:在模板繁重的代码中使用using,而不是typename尤其是,因为前者也可以使用别名模板。并且不使用enum技巧来存储静态值,而不使用空间;无论如何,编译器都会对此进行优化,只需使用static constexpr整数即可。