Как вернуть список значений из Lua в C и распечатать их одно за другим?

Question

Я использую C API, встроенный в Lua. Моя цель состоит в том, чтобы: передать массив целых чисел в Lua и вычислить их факториалы, затем результаты передать обратно в C и распечатать.

Чтобы реализовать цель, мой C-код:

#include <lua.h>
#include <lauxlib.h>
#include <stdlib.h>
#include <stdio.h>

int main(void){
  int status, result, i;
  double fac;

  lua_State *L;  // set Lua state
  L = luaL_newstate();

  luaL_openlibs(L); 

  status = luaL_loadfile(L, "factorial.lua");  // load the Lua script for factorial calculation
  if (status) {
    fprintf(stderr, "Couldn't load file: %s\n", lua_tostring(L, -1));
    exit(1);
  }

  lua_newtable(L);  

  for (i = 1; i <= 10; i++) {
    lua_pushnumber(L, i);   /* Push the table index */
    lua_pushnumber(L, i*2); /* Push the cell value */
    lua_rawset(L, -3);      /* Stores the pair in the table */
  }

  lua_setglobal(L, "foo");

  result = lua_pcall(L, 0, LUA_MULTRET, 0);
  if (result) {
    fprintf(stderr, "Failed to run script: %s\n", lua_tostring(L, -1));
    exit(1);
  }

  // the following loop is for factorial print-out
  while (lua_next(L, -1) != 0) {  
    fac = lua_tonumber(L, -1);
    printf("%.0f\n", fac);
    lua_pop(L, 1);
  }

  lua_close(L);    

  return 0;
}

И мой скрипт Lua выглядит так:

-- this is the function to calculate the factorial
function fact(n)
if n == 0 then
       return 1
else
   return n * fact(n-1)
end
end

io.write("We calculate the factorial of the following numbers: \n")

return_table = {}
for i = 1, #foo do
 n = foo[i]
 factorial_result = fact(n)
 print(n)
 table.insert(return_table, factorial_result)
end

io.write("Here we show the results: \n")
for i=1,10 do 
return(return_table[i])
end

Компиляция идет хорошо, но когда я запускаю его на терминале, я получаю:

We calculate the factorial of the following numbers:
2.0
4.0
6.0
8.0
10.0
12.0
14.0
16.0
18.0
20.0
Here we show the results:
Segmentation fault (core dumped)

Я не знаю, почему результат таков. Кажется, что нет проблем с переходом от C к Lua, но есть проблема от Lua к C. Может ли кто-нибудь помочь мне с этим?

Answer 1

У вашего кода есть две проблемы:

1. У тебя есть return оператор внутри цикла в вашем коде Lua. Как только нажата инструкция return, ваш скрипт Lua завершен. Он вернет это значение в код C, и это будет число, а не таблица, которую ожидает ваш код C.
2. Ты звонишь lua_next без предварительного нажатия стартового ключа в стек.

В вашем коде Lua измените это:

for i=1,10 do 
    return(return_table[i])
end

К этому:

return(return_table) -- don't need parentheses here, btw

В коде C добавьте lua_pushnil(L) перед вашим временем цикла.

Answer 2

После того, как вы выполните returnскрипт Lua останавливается.

Вместо цикла в конце скрипта попробуйте просто

return return_table