#include <stdio.h>
#include <string.h>
#include <lua.hpp>
#include <lauxlib.h>
#include <lualib.h>

//待Lua调用的C注册函数。
static int add2(lua_State* L)
{
    //检查栈中的参数是否合法，1表示Lua调用时的第一个参数(从左到右)，依此类推。
    //如果Lua代码在调用时传递的参数不为number，该函数将报错并终止程序的执行。
    double op1 = luaL_checknumber(L,1);
    double op2 = luaL_checknumber(L,2);
    //将函数的结果压入栈中。如果有多个返回值，可以在这里多次压入栈中。
    lua_pushnumber(L,op1 + op2);
    //返回值用于提示该C函数的返回值数量，即压入栈中的返回值数量。
    return 1;
}

//另一个待Lua调用的C注册函数。
static int sub2(lua_State* L)
{
    double op1 = luaL_checknumber(L,1);
    double op2 = luaL_checknumber(L,2);
    lua_pushnumber(L,op1 - op2);
    return 1;
}

const char* testfunc = "print(add2(1.0,2.0)) print(sub2(20.1,19))";

int main()
{
    lua_State* L = luaL_newstate();
    luaL_openlibs(L);
    //将指定的函数注册为Lua的全局函数变量，其中第一个字符串参数为Lua代码
    //在调用C函数时使用的全局函数名，第二个参数为实际C函数的指针。
    lua_register(L, "add2", add2);
    lua_register(L, "sub2", sub2);
    //在注册完所有的C函数之后，即可在Lua的代码块中使用这些已经注册的C函数了。
    if (luaL_dostring(L,testfunc))
        printf("Failed to invoke.\n");
    lua_close(L);
    return 0;
}

#include <stdio.h>
#include <string.h>

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

#if defined(_WIN32) || defined(_WIN64)
#define _DLLExport _declspec(dllexport)
#else
#define _DLLExport 
#endif
// 需要说明的是，该函数必须以C的形式被导出，因此extern "C"是必须的。
extern "C" int add(lua_State * L)
{
    double op1 = luaL_checknumber(L, -1);
    double op2 = luaL_checknumber(L, -2);

    lua_pushnumber(L, op1 + op2);
    return 1;
}

extern "C" int sub(lua_State * L)
{
    double op1 = luaL_checknumber(L, -1);
    double op2 = luaL_checknumber(L, -2);

    lua_pushnumber(L, op1 - op2);
    return 1;
}
// 必须以{NULL, NULL}结尾
static luaL_Reg registerInfo[] = {
    {"add", add},
    {"sub", sub},
    {NULL, NULL}
};

// 该C库的入口函数
// 函数名必须为luaopen_xxx，其中xxx表示library名称。Lua代码require "xxx"需要与之对应。
extern "C" _DLLExport int luaopen_CLibraryForLua(lua_State * L)
{
    luaL_newlib(L, registerInfo);

    // 上面一行等同于下面两行
    // luaL_newlibtable(L, registerInfo);
    // luaL_setfuncs(L, registerInfo, 0);

    return 1;
}

require "CLibraryForLua"  --指定包名称
 
--在调用时，必须是package.function
print(CLibraryForLua.add(1.0,2.0))
print(CLibraryForLua.sub(20.1,19))