C++与Lua 交互 即时编译器LuaJIT

cd luajit
make
sudo make install

..
// 添加 LuaJIT 头文件
extern "C"
{
    #include "lua.h"  
    #include "lualib.h"  
    #include "lauxlib.h"  
    #include "luajit.h"
}
...


static int wrap_exceptions(lua_State *L, lua_CFunction f)
{
    try {
        return f(L);  // Call wrapped function and return result.
    }
    catch (const char *s) {  // Catch and convert exceptions.
        lua_pushstring(L, s);
    }
    catch (std::exception& e) {
        lua_pushstring(L, e.what());
    }
    catch (...) {
        lua_pushliteral(L, "caught (...)");
    }
    return lua_error(L);  // Rethrow as a Lua error.
}


bool ScriptVM::Init(void)
{
    m_pLua = luaL_newstate();


    lua_cpcall(m_pLua, luaopen_base, 0);
    lua_cpcall(m_pLua, luaopen_io, 0);
    lua_cpcall(m_pLua, luaopen_string, 0);
    lua_cpcall(m_pLua, luaopen_table, 0);
    lua_cpcall(m_pLua, luaopen_math, 0);
    lua_cpcall(m_pLua, luaopen_debug, 0);
    lua_cpcall(m_pLua, luaopen_os, 0);
    lua_cpcall(m_pLua, luaopen_package, 0);


    //luaopen_base(m_pLua);
    //luaopen_io(m_pLua);
    //luaopen_table(m_pLua);
    //luaopen_math(m_pLua);
    //luaopen_string(m_pLua);
    //luaopen_debug(m_pLua);


    //tolua_open_binding(m_pLua);


    //lua_setgcthreshold(m_pLua, 200); //200k garbage collection threshold
    lua_register(m_pLua, "print", PrintStringList);
    //lua_register(m_pLua, "dofile", OverrideDofile);


    // 添加 LuaJIT 的支持
    lua_pushlightuserdata(m_pLua, (void *)wrap_exceptions);
    luaJIT_setmode(m_pLua, 0, LUAJIT_MODE_ENGINE | LUAJIT_MODE_ON);


    return true;
}


...
void ScriptVM::ExecuteScript(const char * sScript, bool bAssertOnError)
{
    // 添加 LuaJIT 的支持
    luaJIT_setmode(m_pLua, -1, LUAJIT_MODE_ALLFUNC | LUAJIT_MODE_ALLSUBFUNC | LUAJIT_MODE_ON);


    int status = luaL_loadbuffer(m_pLua, sScript, strlen(sScript), sScript);
    if (status)
    {
        report_last_error(m_pLua, bAssertOnError);
    }
    else
    {
        status = lua_pcall(m_pLua, 0, LUA_MULTRET, 0);  /* call main */
        if (status)
            report_last_error(m_pLua, bAssertOnError);
    }
}
...
bool ScriptVM::ExecuteScriptFunc(const std::vector<const char *>&modules, const char * func, bool bAllowNonexist, const char * sig, ...)
{
    // 添加 LuaJIT 的支持
    luaJIT_setmode(m_pLua, -1, LUAJIT_MODE_ALLFUNC | LUAJIT_MODE_ALLSUBFUNC | LUAJIT_MODE_ON);


    bool bIsSuccess = false;


    //PROFILE("ExecuteScriptFunc");
    int nSize1 = lua_gettop(m_pLua);


    //debug
#if DEBUG_STACK
    printf("debug lua: stack size before ExecuteScriptFunc = %d\n", nSize1);
#endif


    va_list    vl;
    int    narg, nres;    /* number of arguments and results */
    va_start(vl, sig);


    //get the actual function
    if (modules.empty()) //func is global
    {
        lua_getglobal(m_pLua, func);
        if (!lua_isfunction(m_pLua, -1))
        {
            if (!bAllowNonexist)
                error_msg(true, "ExecuteScriptFunc: Invalid function name: %s\n", func);
            goto failed;
        }
    }
    else
    {
        //trace down the modules
        std::vector<const char *>::const_iterator it = modules.begin();
        //get the global module name or the actual function name if there is no module
        lua_getglobal(m_pLua, *it);
        if (!lua_istable(m_pLua, -1))
        {
            if (!bAllowNonexist)
                error_msg(true, "ExecuteScriptFunc: Invalid table name: %s\n", *it);
            goto failed;
        }


        for (++it; it != modules.end(); ++it)
        {
            lua_pushstring(m_pLua, *it);
            lua_gettable(m_pLua, -2);
            if (!lua_istable(m_pLua, -1))
            {
                if (!bAllowNonexist)
                    error_msg(true, "ExecuteScriptFunc: Invalid table name: %s\n", *it);
                goto failed;
            }
        }
        //get the func
        lua_pushstring(m_pLua, func);
        lua_gettable(m_pLua, -2);
        if (!lua_isfunction(m_pLua, -1))
        {
            if (!bAllowNonexist)
                error_msg(true, "ExecuteScriptFunc: Invalid function name: %s\n", func);
            goto failed;
        }
    }


    /* push    arguments */
    narg = 0;
    while (*sig) { /* push arguments    */
        switch (*sig++) {
        case 'd': /* double    argument */
        case 'f': /* float    argument */    // NieXu: Treat float as double, same as printf()
            lua_pushnumber(m_pLua, va_arg(vl, double));
            break;
        case 'i': /* int argument */
            lua_pushnumber(m_pLua, va_arg(vl, int));
            break;
        case 's': /* string    argument */
            lua_pushstring(m_pLua, va_arg(vl, char *));
            break;
        case 'b': /* boolean argument */
            lua_pushboolean(m_pLua, va_arg(vl, bool));
            break;
        case 'u': /* light user data */
            lua_pushlightuserdata(m_pLua, va_arg(vl, void *));
            break;
        case 't': /* type user data */
        {
            void* pData = va_arg(vl, void *);
            const char* sType = va_arg(vl, const char*);
            tolua_pushusertype(m_pLua, pData, sType);
            break;
        }


        case '>':
            goto endwhile;
        default:
            error_msg(true, "invalid option (%c)\n", *(sig - 1));
            goto failed;
        }
        narg++;
        luaL_checkstack(m_pLua, 1, "too many    arguments");
    }endwhile:
    /* do the call */
    nres = strlen(sig);    /* number of expected results */
    if (lua_pcall(m_pLua, narg, nres, 0) != 0) /* do    the    call */
    {
        report_last_error(m_pLua, true);
        goto failed;
    }
    /* retrieve    results    */
    nres = -nres; /* stack index of    first result */
    while (*sig)
    { /* get results */
        switch (*sig++)
        {
        case 'd': /* double    result */
            if (!lua_isnumber(m_pLua, nres))
                error_msg(true, "wrong    result type,function name: %s\n", func);
            *va_arg(vl, double *) = lua_tonumber(m_pLua, nres);
            break;
        case 'f': /* float    result */
            if (!lua_isnumber(m_pLua, nres))
                error_msg(true, "wrong    result type,function name: %s\n", func);
            *va_arg(vl, float*) = (float)lua_tonumber(m_pLua, nres);
            break;
        case 'i': /* int result    */
            if (!lua_isnumber(m_pLua, nres))
                error_msg(true, "wrong    result type,function name: %s\n", func);
            *va_arg(vl, int    *) = (int)lua_tonumber(m_pLua, nres);
            break;
        case 's': /* string    result */
            if (!lua_isstring(m_pLua, nres))
                error_msg(true, "wrong    result type,function name: %s\n", func);
            *va_arg(vl, std::string*) = lua_tostring(m_pLua, nres);
            break;
        case 'b': /* boolean argument */
            if (!lua_isboolean(m_pLua, nres))
                error_msg(true, "wrong    result type,function name: %s\n", func);
            *va_arg(vl, bool *) = (0 != lua_toboolean(m_pLua, nres));
            break;
        case 'u': /* light user data */
            if (!lua_isuserdata(m_pLua, nres))
                error_msg(true, "wrong    result type,function name: %s\n", func);
            *va_arg(vl, void **) = lua_touserdata(m_pLua, nres);
            break;
        default:
            error_msg(true, "invalid option (%c)\n", *(sig - 1));
        }
        nres++;
    }


    bIsSuccess = true;
failed:
    va_end(vl);
    //clear the stack
    lua_settop(m_pLua, nSize1);


#if DEBUG_STACK
    //debug
    int nSize2 = lua_gettop(m_pLua);
    printf("debug lua: stack size after ExecuteScriptFunc = %d\n", nSize2);
    if (nSize1 != nSize2)
        stackDump(m_pLua);
#endif


    return bIsSuccess;
}