函数指针作为模板参数和签名

时间:2016-08-24 19:00:32

标签: c++ templates lua function-pointers

我可以通过"将军"函数指针作为模板参数与它的签名?我知道我可以将函数签名传递给模板:

template<typename signature>
struct wrapper;

template<typename RT, typename... ATs>
struct wrapper<RT (ATs...)> {};

int f(int, double)

wrapper<decltype(f)> w;

我也可以将函数指针作为非类型模板参数传递:

   template<int (*pF)(int, double)> myTemp() {
      pf(1, 1.0);
   }

   myTemp<f>();

我想做的是这样的事情

   template<typename RT (*pF)(typename ATs...)>

这可能吗?函数指针必须作为模板参数传递,不能作为函数参数传递。

我想使用模板来包装c函数并使它们可以从lua调用。以下代码有效(c ++ 14,gcc,lua-5.3),但可以改进。

#include <iostream>
#include <type_traits>

extern "C"  {
#include <lua.h>
#include <lualib.h>
#include <lauxlib.h>
}

using namespace std;

int add(int i, int j) {
    cout << "adding " << i << " to " << j << "." << endl;
    return i + j;
}

int sub(int i, int j) {
    cout << "subtracting " << j << " from " << i << "." << endl;
    return i - j;
}

// ****************************

template<typename signature>
struct wrapper;

template<typename RT, typename... ATs>
struct wrapper<RT (ATs...)> {

    template<RT (*pF)(ATs...)>
    void reg(lua_State *L, const char*n) {
        auto lw = [](lua_State *L) -> RT {
            lua_pushnumber(L, call<0>(pF, L));
            return 1;
        };
        lua_pushcfunction(L, lw);
        lua_setglobal(L, n);
    }

    template<int i, typename... ETs>
    static
    typename std::enable_if<i != sizeof...(ATs), RT>::type
    call(RT (*f)(ATs...), lua_State *L, ETs... Es) {
        auto arg = lua_tonumber(L, i+1);
        return call<i+1>(f, L, Es..., arg);
    }

    template<int i, typename... ETs>
    static
    typename std::enable_if<i == sizeof...(ATs), RT>::type
    call(RT (*f)(ATs...), lua_State *L, ETs... Es) {
        return f(Es...);
    }

};

#define regLua(L, fct, str) wrapper<decltype(fct)>().reg<fct>(L, str)

int main() {
    lua_State *L = luaL_newstate();
    luaL_openlibs(L);

    luaL_dostring(L, "print(\"Hello World!\")");

    // Ugly: add must be passed two times! Not a very userfriendly syntax.
    wrapper<decltype(add)>().reg<add>(L, "add");
    // Looks better, but uses a macro...
    regLua(L, sub, "sub");
    // optimal (but possible??):
    // wrap<sub>(L, "sub");

    luaL_dostring(L, "print(\"add:\", add(3, 5))");
    luaL_dostring(L, "print(\"sub:\", sub(3, 5))");

    lua_close(L);

    return 0;
}

1 个答案:

答案 0 :(得分:1)

c ++ 17允许:

<hot-column data="???">

然后你的专业化

template <auto value> struct wrapper;

用法:

template<typename RT, typename... ATs, RT (*pF)(ATs...)>
struct wrapper<pF> {
    static void reg(lua_State *L, const char* n) {
        auto lw = [](lua_State *L) {
            lua_pushnumber(L, call(L, std::index_sequence_for<ATS...>()));
            return 1;
        };
        lua_pushcfunction(L, lw);
        lua_setglobal(L, n);
    }

    template<std::size_t ... Is>
    static
    RT call(lua_State *L, std::index_sequence<Is...>) {
        return pF(lua_tonumber(L, 1 + Is)...);
    }
};

在c ++ 17之前,您的wrapper<&add>::reg(L, "add"); 必须是

wrapper

强制您重复函数名称(如果您不手动键入其签名)

template <typename Sig, sig Pf> struct wrapper;

template<typename RT, typename... ATs, RT (*pF)(ATs...)>
struct wrapper<Rt (*)(ATS...), pF> {
    // Code
};
相关问题