test/test_yield.cpp

   1 #include "test.h"
   2 #include <luabind/yield_policy.hpp>
   3 #include <stdio.h>
   4
   5 namespace
   6 {
   7         LUABIND_ANONYMOUS_FIX int feedback = 0;
   8
   9         struct test_class
  10         {
  11                 test_class(): n(0) {}
  12
  13                 int f() const
  14                 {
  15                         return const_cast<int&>(n)++;
  16                 }
  17
  18                 int n;
  19         };
  20
  21         int f(int a)
  22         {
  23                 return 9;
  24         }
  25
  26         int j(lua_State* L)
  27         {
  28                 lua_pushnumber(L, 9);
  29                 return lua_yield(L, 1);
  30         }
  31
  32         void f() {}
  33 }
  34
  35 #include <iostream>
  36
  37 bool test_yield()
  38 {
  39         using namespace luabind;
  40         {
  41                 lua_State* L = lua_open();
  42                 lua_baselibopen(L);
  43                 lua_closer c(L);
  44
  45                 open(L);
  46
  47                 module(L)
  48                 [
  49                         class_<test_class>("test")
  50                                 .def(constructor<>())
  51                                 .def("f", &test_class::f, yield)
  52                 ];
  53
  54                 dostring(L, "function g() a = test() for i = 1, 10 do print(a:f()) end end");
  55
  56                 lua_pushstring(L, "j");
  57                 lua_pushcclosure(L, j, 0);
  58                 lua_settable(L, LUA_GLOBALSINDEX);
  59
  60                 lua_State* thread = lua_newthread(L);
  61                 lua_pushstring(thread, "g");
  62                 lua_gettable(thread, LUA_GLOBALSINDEX);
  63
  64                 if (lua_resume(thread, 0))
  65                 {
  66                         std::cout << "error: " << lua_tostring(thread, -1) << '\n';
  67                 }
  68
  69                 for (int i = 0; i < 10; ++i)
  70                 {
  71                         std::cout << "iteration: " << i << ", top: " << lua_gettop(thread) << '\n';
  72
  73                         lua_resume(thread, lua_gettop(thread));
  74                 }
  75         }
  76
  77         return true;
  78 }
  79