使用std :: function作为参数

时间:2017-08-10 10:24:54

标签: c++ templates

我有一个带有模板构造函数的类,如下所示:

class Foo {
private:

    std::unordered_map<std::type_index, std::vector<std::function<void(BaseT*)>>> funcs;

public:

    template<class T> Foo(const std::function<void(T* arg)>& func) {
        auto funcToStore = [func](BaseT* a) { func(static_cast<T*>(a)); };
        this->funcs[std::type_index(typeid(T))].push_back(funcToStore);
    }

}

此类的构造函数接受一个函数参数,其参数类型为T,派生自某个基类型BaseT,并将此函数存储在使用std::type_info T的向量映射中。 1}}用于密钥。

由于这是一个模板构造函数而不是普通函数,因此显式指定模板参数将不起作用,因为这是不允许的语法:

Foo* foo = new Foo<MyT>([](MyT* arg) { ... });

省略显式<MyT>也不起作用,因为无法从lambda参数类型推断出模板参数。

因此,一种解决方案是将lambda包装在std::function对象中:

Foo* foo = new Foo(std::function<void(MyT*)>([](MyT* arg) { ... }));

但这显然不是一个很好的可读语法。

我到目前为止所做的最好的事情是为std::function使用别名

template<class T> using Func = std::function<void(T*)>;

Foo* foo = new Foo(Func<MyT>([](MyT* arg) { ... }));

这个更短,当在lambda参数中使用auto关键字时,我只需要指定实际类型MyT一次,所以这似乎是一个很好的解决方案。< / p>

但是还有其他甚至更短的解决方案吗?所以没有必要包裹lambda?像:

Foo* foo = new Foo([](MyT* arg) { ... });

1 个答案:

答案 0 :(得分:2)

使用普通模板参数代替std::function

class Foo {
    std::unordered_map<size_t, std::vector<BaseT*>> funcs;

public:
    template<class T> Foo(const T& func) {
        // ...
    }

};

现在扣除将正确进行,您的代码不会受到std::function的开销的影响。

如果要获取lambda的第一个参数的类型怎么办?

你必须做这样的事情:

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

template<typename R, typename C, typename... Args>
struct function_traits<R(C::*)(Args...) const> {
    using arguments = std::tuple<Args...>;
    using result = R;
};

当然,如果你想支持每种可能的函数类型,你需要32 specialisations

现在,您可以根据需要提取参数类型甚至返回类型:

template<class T> Foo(const T& func) {
    using Arg = std::tuple_element_t<0, typename  function_traits<T>::arguments>;
    auto funcToStore = [func](BaseT* a) { func(static_cast<Arg>(a)); };

    funcs[typeid(Arg).hash_code()].push_back(funcToStore);
}

此外,由于您在构造函数中收到const T&,因此您可能希望将函数约束为只能用可编译的函数调用:

template<typename T>
using is_valid_foo_function = std::is_convertible<
    BaseT*, // form
    std::tuple_element_t<0, typename function_traits<T>::arguments> // to
>;

并使用类似的约束:

template<class T, std::enable_if_t<is_valid_foo_function<T>::value>* = nullptr>
Foo(const T& func) {
    // ...
}