lua与C交互之——对lua数组的每个元素调用同一个函数

lua程序栈的结构图：以先后传入两个参数"Hello" "Tip"为例

参考：Lua与C/C++交互——Lua调用C/C++
现在我们需要这样的一个功能，lua中一个数组，tb = {3, 4, 5, 6} ，调用一个function函数，得到的tb = {13, 14, 15, 16}，需要这样的一个C库，
main.c

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

static int l_map(lua_State* L)
{
	int i, n;

	//第一个参数必须是一张表
	luaL_checktype(L, 1, LUA_TTABLE);

	//第二个参数必须是一个函数
	luaL_checktype(L, 2, LUA_TFUNCTION);

	n = luaL_len(L, 1); //获取表的大小

	for (i = 1; i <= n; i++)
	{
		lua_pushvalue(L, 2);
		lua_geti(L, 1, i); //压入t[i]
		lua_call(L, 1, 1); //调用f(t[i])，lua_call的第二个参数：传入f中一个参数，lua_call的第三个参数：f的返回值是一个参数
		lua_seti(L, 1, i); //t[i] = result;
	}

	return 0; //没有返回值
}

static luaL_Reg testlibs[] =
{
	{"mapping", l_map},
	{NULL, NULL}
};

__declspec(dllexport)//这个很重要，设置这个函数为外部链接 Linux不需要
int luaopen_mymap(lua_State* L) //函数名很重要，luaopen_xxx，lua脚本require(“xxx”)和xxx.dll、xxx.so，这4个一定要这样，
{
	luaL_newlib(L, testlibs);
	return 1;
}

编译处理C库为mymap.dll

test.lua

local mymap = require("mymap")

local tb = {3, 4, 5, 6}
function each_plus(element)
	return element+10
end

mymap.mapping(tb, each_plus) --执行后tb = {13, 14, 15, 16}

for i, v in ipairs(tb) do
    print(i, v)  --1是位置，v是值
end 

运行结果：

D:\lua_test>lua test.lua
1       13
2       14
3       15
4       16

几个函数的讲解：
luaL_checktype

void luaL_checktype (lua_State *L, int arg, int t);

arg表示参数的位置，t表示要检查的参数类型（ lua_type是宏定义，这里传入int）

lua_type的取值为：

#define LUA_TNIL			0
#define LUA_TBOOLEAN		1
#define LUA_TLIGHTUSERDATA	2
#define LUA_TNUMBER			3
#define LUA_TSTRING			4
#define LUA_TTABLE			5
#define LUA_TFUNCTION		6
#define LUA_TUSERDATA		7
#define LUA_TTHREAD			8

从名字可以看出LUA_TNONE表示为空，LUA_TNIL为nil类型，LUA_TBOOLEAN类型等，根据名字可以大致看出含义。

现在我想对lua数组中的每个元素都加上一个num的值，num值由lua的function传入

lua_Number luaL_checknumber (lua_State *L, int arg);

检查位置为arg的参数是否为数字，并且返回这个数字的值。
如果我传入一个字符串，则直接会报错：

lua: test.lua:8: bad argument #3 to 'mapping' (number expected, got string)

对于这个

/* type of numbers in Lua */
typedef LUA_NUMBER lua_Number;

#define LUA_NUMBER	double

根据源码查找，发现lua_Number原来是double类型的。
修改后的test.lua为：

local mymap = require("mymap")

local tb = {3, 4, 5, 6}
function each_plus(element, num)
	return element+num
end

mymap.mapping(tb, each_plus, 20) --执行后tb = {13, 14, 15, 16}

for i, v in ipairs(tb) do
    print(i, v)  --1是位置，v是值
end 

运行结果：

D:\lua_test>lua test.lua
1       23.0
2       24.0
3       25.0
4       26.0

修改后的main.c文件为：

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

static int l_map(lua_State* L)
{
	int i, n;

	//第一个参数必须是一张表
	luaL_checktype(L, 1, LUA_TTABLE);

	//第二个参数必须是一个函数
	luaL_checktype(L, 2, LUA_TFUNCTION);

	//第三个参数必须的一个数字类型
	double dValue = luaL_checknumber(L, 3);

	n = luaL_len(L, 1); //获取表的大小

	for (i = 1; i <= n; i++)
	{
		lua_pushvalue(L, 2);
		lua_geti(L, 1, i); //压入t[i]，传入f的第一个参数
		lua_pushnumber(L, dValue); //压入dValue，传给f的第二个参数
		lua_call(L, 2, 1); //调用f(t[i], dValue)，lua_call的第二个参数：传入f中一个参数; lua_call的第三个参数：f的返回值是一个参数
		lua_seti(L, 1, i); //t[i] = result;
	}

	return 0; //没有返回值
}

static luaL_Reg testlibs[] =
{
	{"mapping", l_map},
	{NULL, NULL}
};

__declspec(dllexport)//这个很重要，设置这个函数为外部链接 Linux不需要
int luaopen_mymap(lua_State* L) //函数名很重要，luaopen_xxx，lua脚本require(“xxx”)和xxx.dll、xxx.so，这4个一定要这样，
{
	luaL_newlib(L, testlibs);
	return 1;
}

在windows上编译C库给lua调用的方法：
https://editor.youkuaiyun.com/md/?articleId=104572573

linux上编译C库给lua调用的方法：
https://blog.youkuaiyun.com/follow_blast/article/details/81779342