c ++ 0x:在lambda arity上重载

时间:2010-11-13 00:32:33

标签: c++ lambda c++11

我正在尝试创建一个可以使用带有0,1或2个参数的lambda调用的函数。因为我需要代码在g ++ 4.5和vs2010上工作(它不支持可变参数模板或lambda转换到函数指针),我想出的唯一想法是选择基于arity调用哪个实现。以下是我应该如何看待的非工作猜测。有没有办法修复我的代码,或者有更好的方法来执行此操作吗?

#include <iostream>
#include <functional>
using namespace std;

template <class Func> struct arity;

template <class Func>
struct arity<Func()>{ static const int val = 0; };

template <class Func, class Arg1>
struct arity<Func(Arg1)>{ static const int val = 1; };

template <class Func, class Arg1, class Arg2>
struct arity<Func(Arg1,Arg2)>{ static const int val = 2; };

template<class F>
void bar(F f)
{
    cout << arity<F>::val << endl;
}

int main()
{
    bar([]{cout << "test" << endl;});
}

4 个答案:

答案 0 :(得分:17)

lambda函数是具有单个函数调用运算符的类类型。因此,您可以通过获取其地址并使用重载决策来选择要调用的函数来检测该函数调用运算符的arity:

#include <iostream>

template<typename F,typename R>
void do_stuff(F& f,R (F::*mf)() const)
{
    (f.*mf)();
}

template<typename F,typename R,typename A1>
void do_stuff(F& f,R (F::*mf)(A1) const)
{
    (f.*mf)(99);
}

template<typename F,typename R,typename A1,typename A2>
void do_stuff(F& f,R (F::*mf)(A1,A2) const)
{
    (f.*mf)(42,123);
}

template<typename F>
void do_stuff(F f)
{
    do_stuff(f,&F::operator());
}

int main()
{
    do_stuff([]{std::cout<<"no args"<<std::endl;});
    do_stuff([](int a1){std::cout<<"1 args="<<a1<<std::endl;});
    do_stuff([](int a1,int a2){std::cout<<"2 args="<<a1<<","<<a2<<std::endl;});
}

但请注意:这不适用于函数类型,或具有多个函数调用运算符的类类型,或非const函数调用运算符。

答案 1 :(得分:1)

我认为以下内容可行,但事实并非如此,我发布它有两个原因。

  1. 为了节省人们的时间,如果他们有相同的想法
  2. 如果有人知道为什么这不起作用,我不是百分百肯定我理解(虽然我怀疑)
  3. 代码如下:

    #include <iostream>
    #include <functional>
    
    template <typename Ret>
    unsigned arity(std::function<Ret()>) { return 0; }
    
    template <typename Ret, typename A1>
    unsigned arity(std::function<Ret(A1)>) { return 1; }
    
    template <typename Ret, typename A1, typename A2>
    unsigned arity(std::function<Ret(A1, A2)>) { return 2; }
    
    // rinse and repeat 
    
    int main() 
    {
        std::function<void(int)>  f = [](int i) { }; // this binds fine
    
        //  Error: no matching function for call to 'arity(main()::<lambda(int)>)'
        std::cout << arity([](int i) { }); 
    } 
    

答案 2 :(得分:0)

编译时间是指获取函数或函数对象的arity,包括lambda的函数:

int main (int argc, char ** argv) {
    auto f0 = []() {};
    auto f1 = [](int) {};
    auto f2 = [](int, void *) {};

    std::cout << Arity<decltype(f0)>::value << std::endl; // 0
    std::cout << Arity<decltype(f1)>::value << std::endl; // 1
    std::cout << Arity<decltype(f2)>::value << std::endl; // 2

    std::cout << Arity<decltype(main)>::value << std::endl; // 2
}

template <typename Func>
class Arity {
private:
    struct Any {
        template <typename T>
        operator T ();
    };

    template <typename T>
    struct Id {
        typedef T type;
    };

    template <size_t N>
    struct Size {
        enum { value = N };
    };

    template <typename F>
    static Size<0> match (
        F f,
        decltype(f()) * = nullptr);

    template <typename F>
    static Size<1> match (
        F f,
        decltype(f(Any())) * = nullptr,
        decltype(f(Any())) * = nullptr);

    template <typename F>
    static Size<2> match (
        F f,
        decltype(f(Any(), Any())) * = nullptr,
        decltype(f(Any(), Any())) * = nullptr,
        decltype(f(Any(), Any())) * = nullptr);

public:
    enum { value = Id<decltype(match(static_cast<Func>(Any())))>::type::value };
};

答案 3 :(得分:-1)

这种方式有效:

template<typename F>
auto call(F f) -> decltype(f(1))
{
    return f(1);
}

template<typename F>
auto call(F f, void * fake = 0) -> decltype(f(2,3))
{
    return f(2,3);
}

template<typename F>
auto call(F f, void * fake = 0, void * fake2 = 0) -> decltype(f(4,5,6))
{
    return f(4,5,6);
}

int main()
{
    auto x1 = call([](int a){ return a*10; });
    auto x2 = call([](int a, int b){ return a*b; });
    auto x3 = call([](int a, int b, int c){ return a*b*c; });
    // x1 == 1*10
    // x2 == 2*3
    // x3 == 4*5*6
}

适用于所有可调用类型(lambdas,functor等)