我正在尝试将多个float数组从C ++发送到Lua函数作为参数,然后从函数中返回多个表,以便我可以在C ++中再次使用它们作为float数组。
所以我的Lua功能看起来像这样。
function perform(arg1, arg2)
local ret1, ret2 = {}, {}
for i=1, #arg1 do
ret1[i] = arg1[i] * 0.2;
ret2[i] = arg2[i] * 0.3;
end
return ret1, ret2
end
这就是我在C ++中向Lua函数发送和返回多个表的方法。
lua_getglobal(L, "perform");
for (int i=0; i<numArgs; ++i) {
lua_newtable(L);
float *in = reinterpret_cast<float*>(w[i]);
for (int j=0; j<64; ++j) {
lua_pushinteger(L, j+1);
lua_pushnumber(L, in[j]);
lua_settable(L, -3);
}
}
lua_call(L, numArgs, numRets);
for (int i=0; i<numRets; ++i) {
float *out = reinterpret_cast<float*>(w[numArgs+i]);
for (int j=0; j<64; ++j) {
lua_pushinteger(L, j+1);
lua_gettable(L, -2);
out[j] = lua_tonumber(L, -1);
lua_pop(L, 1);
}
//how to detect next returned table?
}
但是如果我尝试代码,则返回的数组具有相同的值。
我认为这是因为我没有正确获取返回的表格。
有人可以教我如何正确地获得多张退回的桌子吗?
P.S:我还想知道我的代码是否可以进行优化以获得更好的性能。
编辑:传递并返回一个包含多个子表的表会更快(更有效)吗?如果是这样,我会很感激,如果有人可以教我如何做到这一点。
答案 0 :(得分:1)
我不知道你在这里尝试做什么,但是从函数返回的第二个表可以在堆栈上轻松访问。您只需对堆栈索引执行一些算术就可以到达正确的位置。
那些reinterpret_casts
看起来非常可疑。你很可能做错了。
#include <iostream>
#include <vector>
#include <lua.hpp>
int main(int argc, char *argv[]) {
lua_State *L = luaL_newstate();
luaL_openlibs(L);
if (argc != 2) {
std::cerr << "Usage: " << argv[0] << " <script.lua>\n";
return 1;
}
luaL_dofile(L, argv[1]);
// Mock data
int numArgs = 2;
int numRets = 2;
std::vector<float> w1(64, 1.0f);
std::vector<float> w2(64, 1.0f);
std::vector<float> w3(64, 1.0f);
std::vector<float> w4(64, 1.0f);
std::vector<float *> w = {w1.data(), w2.data(), w3.data(), w4.data()};
lua_getglobal(L, "perform");
for (int i = 0; i < numArgs; ++i) {
lua_newtable(L);
float *in = reinterpret_cast<float *>(w[i]);
for (int j = 0; j < 64; ++j) {
lua_pushinteger(L, j + 1);
lua_pushnumber(L, in[j]);
lua_settable(L, -3);
}
}
lua_call(L, numArgs, numRets);
for (int i = 0; i < numRets; ++i) {
float *out = reinterpret_cast<float *>(w[numArgs + i]);
for (int j = 0; j < 64; ++j) {
lua_pushinteger(L, j + 1);
lua_gettable(L, -2 - i); // Just some stack index arithmetic
out[j] = lua_tonumber(L, -1);
lua_pop(L, 1);
}
}
lua_close(L);
}