构造函数重载以接受任何函数

时间:2017-06-04 19:16:23

标签: c++ templates c++17 constructor-overloading

我正在尝试创建一个类模板,其构造函数可以将任何类型的函数作为参数,也就是说,它采用函数指针(可以是成员函数指针)和相应的函数参数。此外,应该有一个static_assert来检查函数返回类型(取自函数指针)是否与类模板参数类型匹配。因此,代码看起来应该是这样的:

template <class ReturnType>
struct Bar
{
    template <class RetType, class ... ParamType>
    Bar<ReturnType>(RetType (* func)(ParamType ...), ParamType && ... args) :
        package_(std::bind(func, std::forward<ParamType>(args) ...)),
        function_([this] { package_(); }),
        future_(package_.get_future())
    {
        static_assert(std::is_same<ReturnType, RetType>::value,
            "Type mismatch between class parameter type and constructor parameter type");
    }

    template <class RetType, class ObjType, class ... ParamType>
    Bar<ReturnType>(RetType (ObjType::* func)(ParamType ...), ObjType * obj, ParamType && ... args) :
        package_(std::bind(func, obj, std::forward<ParamType>(args) ...)),
        function_([this] { package_(); }),
        future_(package_.get_future())
    {
        static_assert(std::is_same<ReturnType, RetType>::value,
            "Type mismatch between class parameter type and constructor parameter type");
    }

    std::packaged_task<ReturnType()> package_;
    std::function<void()> function_;
    std::future<ReturnType> future_;
};

这个想法是代码为这些情况编译,并允许Bar::function_被调用(通过函数调用运算符)而没有错误:

struct Foo
{
    int foo(int i) {
        return i;
    }

    int foo() {
        return 1;
    }
};

int foo(int i)
{
    return i;
}

int foo()
{
    return 1;
}

int main()
{
    Foo f = Foo();

    Bar<int> b1(&Foo::foo, &f, 1);
    Bar<int> b2(&Foo::foo, &f);
    Bar<int> b3(foo, 1);
    Bar<int> b4(foo);

    return 0;
}

不幸的是,我对模板元编程的经验几近为零,即使我在SO中遇到了几个问题,并尝试了几种方法来解决我的问题,例如使用更通用的方法构造函数

template <class RetType, class ... ParamType>
Bar<ReturnType>(RetType func, ParamType && ... args)

并将其与type_traits组合以确定返回类型),我还没有找到一种方法来完成这项工作。我可以对允许此功能的构造函数进行哪些更改?

编辑:

max66的答案解决了我原来的问题,然而,出现了一个新问题,我在上一个问题中没有考虑过。我还希望能够将变量传递给构造函数,如下所示:

int main()
{
    Foo f = Foo();
    int i = 1;

    Bar<int> b1(&Foo::foo, &f, i); // Error
    Bar<int> b2(&Foo::foo, &f, 1); // Ok
    Bar<int> b3(&Foo::foo, &f); // Ok
    Bar<int> b4(foo, i); // Error
    Bar<int> b5(foo, 1); // Ok
    Bar<int> b6(foo); // Ok

    return 0;
}

然而,实际上,在标有Error的情况下会出现编译器错误。我猜这是因为构造函数中的参数func使用ParamType来确定其类型(在ParamType的情况下,它与实际的b1不匹配和b4),但我不知道如何解决这个问题......

2 个答案:

答案 0 :(得分:1)

您可能想要使用std::invoke。它处理成员函数指针和常规函数的使用。

作为你可以做的事情的概述:

#include <functional>
#include <type_traits>
#include <utility>

template<typename F>
class Bar
{
    F f_;

public:
    template<typename TF>
    Bar(TF && f)
        : f_{ std::forward<TF>(f) }
    {}

    template<typename... Args>
    decltype(auto) operator()(Args &&... args) {
        return std::invoke(f_, std::forward<Args>(args)...);
    }
};

template<typename F>
auto make_bar(F && f)
{
    return Bar<std::decay_t<F>>{ std::forward<F>(f) };
}

可以像这样使用:

auto b1 = make_bar(&f);
auto result = b1(myArg1, myArg2); // etc

auto b2 = make_bar(&Foo::fn);
auto result = b1(foo, arg1);

至少,我建议让Bar将函数对象类型作为模板参数,这样就不必使用std::function,但是如果你确实想要使用调用您拥有的语法,也可以使用std::invokestd::invoke_result来完成。

答案 1 :(得分:0)

很抱歉,但是......如果您希望功能的返回类型等于班级的模板参数...为什么不简单地强加它?

我的意思是......您可以使用ReturnType代替RetType,如下所示

template <typename ReturnType>
struct Bar
{
    template <typename ... ParamType>
    Bar<ReturnType> (ReturnType (*func)(ParamType ...), ParamType && ... args)
      : package_(std::bind(func, std::forward<ParamType>(args) ...)),
        function_([this] { package_(); }),
        future_(package_.get_future())
     { }

    template <typename ObjType, typename ... ParamType>
    Bar<ReturnType> (ReturnType (ObjType::* func)(ParamType ...),
                     ObjType * obj, ParamType && ... args)
      : package_(std::bind(func, obj, std::forward<ParamType>(args) ...)),
        function_([this] { package_(); }),
        future_(package_.get_future())
     { }

- 编辑 -

要解决第二个问题, IF 您对移动参数不感兴趣,可以丢弃std::forward&&,然后简单地写

template <typename ReturnType>
struct Bar
{
    template <typename ... ParamType>
    Bar<ReturnType> (ReturnType (*func)(ParamType ...), 
                     ParamType const & ... args)
      : package_(std::bind(func, args...)),
        function_([this] { package_(); }),
        future_(package_.get_future())
     { }

    template <typename ObjType, typename ... ParamType>
    Bar<ReturnType> (ReturnType (ObjType::* func)(ParamType ...),
                     ObjType * obj, ParamType const & ... args)
      : package_(std::bind(func, obj, args...)),
        function_([this] { package_(); }),
        future_(package_.get_future())
     { }