test/test_collapse_converter.cpp

   1 // Copyright Daniel Wallin 2009. Use, modification and distribution is
   2 // subject to the Boost Software License, Version 1.0. (See accompanying
   3 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
   4
   5 #include "test.hpp"
   6 #include <luabind/luabind.hpp>
   7
   8 struct X
   9 {
  10     X(int x, int y)
  11       : x(x)
  12       , y(y)
  13     {}
  14
  15     int x;
  16     int y;
  17 };
  18
  19 namespace luabind {
  20
  21 int combine_score(int s1, int s2)
  22 {
  23     if (s1 < 0 || s2 < 0) return -1;
  24     return s1 + s2;
  25 }
  26
  27 template <>
  28 struct default_converter<X>
  29   : native_converter_base<X>
  30 {
  31     int const consumed_args(...)
  32     {
  33         return 2;
  34     }
  35
  36     int compute_score(lua_State* L, int index)
  37     {
  38         return combine_score(
  39             c1.compute_score(L, index), c2.compute_score(L, index + 1));
  40     }
  41
  42     X from(lua_State* L, int index)
  43     {
  44         return X(lua_tonumber(L, index), lua_tonumber(L, index + 1));
  45     }
  46
  47     default_converter<int> c1;
  48     default_converter<int> c2;
  49 };
  50
  51 } // namespace luabind
  52
  53 int take(X x)
  54 {
  55     return x.x + x.y;
  56 }
  57
  58 void test_main(lua_State* L)
  59 {
  60     using namespace luabind;
  61
  62     module(L) [
  63         def("take", &take)
  64     ];
  65
  66     DOSTRING(L,
  67         "assert(take(1,1) == 2)\n"
  68         "assert(take(2,3) == 5)\n"
  69     );
  70 }