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 struct ns_test
  38 {
  39         template<class T>
  40         ns_test(const T&) {}
  41
  42         ns_test(const char*) {}
  43
  44         template<class T>
  45         void operator=(const T&) {}
  46 };
  47
  48 bool test_yield()
  49 {
  50         using namespace luabind;
  51         {
  52                 lua_State* L = lua_open();
  53                 lua_baselibopen(L);
  54                 lua_closer c(L);
  55
  56                 open(L);
  57
  58                 class_<test_class>("test")
  59                                 .def(constructor<>())
  60                                 .def("f", &test_class::f, yield)
  61                                 .commit(L)
  62                                 ;
  63
  64                 dostring(L, "function g() a = test() for i = 1, 10 do print(a:f()) end end");
  65
  66                 lua_pushstring(L, "j");
  67                 lua_pushcclosure(L, j, 0);
  68                 lua_settable(L, LUA_GLOBALSINDEX);
  69
  70                 lua_State* thread = lua_newthread(L);
  71                 lua_pushstring(thread, "g");
  72                 lua_gettable(thread, LUA_GLOBALSINDEX);
  73
  74                 if (lua_resume(thread, 0))
  75                 {
  76                         std::cout << "error: " << lua_tostring(thread, -1) << '\n';
  77                 }
  78
  79                 for (int i = 0; i < 10; ++i)
  80                 {
  81                         std::cout << "iteration: " << i << ", top: " << lua_gettop(thread) << '\n';
  82
  83                         lua_resume(thread, lua_gettop(thread));
  84                 }
  85         }
  86
  87         return true;
  88 }
  89