test/test_user_defined_converter.cpp

   1 // Copyright Daniel Wallin 2008. 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 value)
  11       : value(value)
  12     {}
  13
  14     int value;
  15 };
  16
  17 namespace luabind {
  18
  19 template <>
  20 struct default_converter<X>
  21   : native_converter_base<X>
  22 {
  23     int compute_score(lua_State* L, int index)
  24     {
  25         return cv.compute_score(L, index);
  26     }
  27
  28     X from(lua_State* L, int index)
  29     {
  30         return X(lua_tonumber(L, index));
  31     }
  32
  33     void to(lua_State* L, X const& x)
  34     {
  35         lua_pushnumber(L, x.value);
  36     }
  37
  38     default_converter<int> cv;
  39 };
  40
  41 } // namespace luabind
  42
  43 int take(X x)
  44 {
  45     return x.value;
  46 }
  47
  48 X get(int value)
  49 {
  50     return X(value);
  51 }
  52
  53 void test_main(lua_State* L)
  54 {
  55     using namespace luabind;
  56
  57     module(L) [
  58         def("take", &take),
  59         def("get", &get)
  60     ];
  61
  62     DOSTRING(L,
  63         "assert(take(1) == 1)\n"
  64         "assert(take(2) == 2)\n"
  65     );
  66
  67     DOSTRING(L,
  68         "assert(get(1) == 1)\n"
  69         "assert(get(2) == 2)\n"
  70     );
  71 }