lua5.3与C++交互学习(二)

---

C++中对lua协程和返回值的处理

lua语言中, 在coroutine的resume和yield之间可以相互传递参数, 最近研究了下lua_resume和lua_yield 两个函数, 同样利用这两个函数可以通过栈在C++与lua之间很方便的交换数据;这里给出一个简单的示例;

1.lua中写一个协程函数

function CorTest()
  print("coroutine begin")
  local re = coroutine.yield(100, "hello c++")  // 往栈中传参数以便C++能获取
  print("lua coroutine continue: " .. re)  // 从c中lua_resume传回来的参数是者的re
  Stop("call c yield!")  // 调用c的挂起函数
  print("coroutine continue after yield")
  print("coroutine end")
end

这里尝试了lua的yield和C++的lua_yield:
2. 在C++中写一个挂起函数注册到lua中使用

static int Stop(lua_State* L)
{
  // 会传一个参数进来
  const char* str = lua_tostring(L,-1);
  print("C Stop: %s\n", str);
  //将这个值再次压入栈中
  lua_pushstring(L,str);
  lua_yield(L,1);
  assert(0);   // 协程恢复后是不会回到这边的, 而是直接回到lua中Stop()后一句, 原因是C++栈已经释放
  return 0;
}

3.这里直接在C中创建协程:

// 参数corName是在lua中写的协程函数
lua_State* CreateCoroutine(lua_State* gL, const char* corName)
{
  lua_checkstack(L,10);  // 10个是随便写的一个数字, 不影响
  lua_State* lt = lua_newthread(gL);
  lua_getglobal(lt, corName);  // 协程函数入栈
  return lt;
}

4.这里在main函数中进行主要处理:

int main()
{
  // 省略前面的lua初始化代码
  /**....**/
  lua_register(L,"Wait",Wait);
  lua_State* lt = CreateCoroutine(L,"CorTest"); 
  int re = lua_resume(lt, L, 0);  // 这次传0个参数(因为CorTest没有参数)
  if(re != LUA_YIELD)
    printf("ERROR");
  //   lua挂起后, 穿回来100, 和hello C++两个值, 可以lua_gettop(lt)看个数
  int rint = luaL_checknumber(lt, -2);
  const char* str = lua_tostring(lt, -1);
  printf("stack value: %d,  %s\n", rint, str);  // ⇒ 100, hello C++
  // 这次恢复协程时传入参数, 参数压栈, lua中就是对应返回值local re
  lua_pushstring(lt, " from c++");
  re = lua_resume(lt, L, 1);
  // 挂起后是在Stop()函数中传入了一个"call c yield!"
  printf("%s\n",lua_tostring(lt,-1));
  re = lua_resume(lt, L, 0); // 这里无法传值回去lua了,因为Stop的栈已经被释放 ,而是直接运行lua中Stop的后一句
}

输出结果如下:

coroutine begin
100, hello C++
coroutine contine ,    from C++
C Stop: call yield
coroutine continue after C yield,  from c++ 
lua end