luabind/shared_ptr_converter.hpp

   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 #ifndef LUABIND_SHARED_PTR_CONVERTER_090211_HPP
   6 # define LUABIND_SHARED_PTR_CONVERTER_090211_HPP
   7
   8 # include <luabind/get_main_thread.hpp>
   9 # include <luabind/handle.hpp>
  10 # include <luabind/detail/policy.hpp>
  11 # include <boost/shared_ptr.hpp>
  12
  13 namespace luabind {
  14
  15 namespace detail
  16 {
  17
  18   struct shared_ptr_deleter
  19   {
  20       shared_ptr_deleter(lua_State* L, int index)
  21         : life_support(get_main_thread(L), index)
  22       {}
  23
  24       void operator()(void const*)
  25       {
  26           handle().swap(life_support);
  27       }
  28
  29       handle life_support;
  30   };
  31
  32 } // namespace detail
  33
  34 template <class T>
  35 struct default_converter<boost::shared_ptr<T> >
  36   : default_converter<T*>
  37 {
  38     typedef boost::mpl::false_ is_native;
  39
  40     template <class U>
  41     int match(lua_State* L, U, int index)
  42     {
  43         return default_converter<T*>::match(
  44             L, LUABIND_DECORATE_TYPE(T*), index);
  45     }
  46
  47     template <class U>
  48     boost::shared_ptr<T> apply(lua_State* L, U, int index)
  49     {
  50         T* raw_ptr = default_converter<T*>::apply(
  51             L, LUABIND_DECORATE_TYPE(T*), index);
  52         if (!raw_ptr)
  53             return boost::shared_ptr<T>();
  54         return boost::shared_ptr<T>(
  55             raw_ptr, detail::shared_ptr_deleter(L, index));
  56     }
  57
  58     void apply(lua_State* L, boost::shared_ptr<T> const& p)
  59     {
  60         if (detail::shared_ptr_deleter* d =
  61                 boost::get_deleter<detail::shared_ptr_deleter>(p))
  62         {
  63             d->life_support.push(L);
  64         }
  65         else
  66         {
  67             detail::value_converter().apply(L, p);
  68         }
  69     }
  70
  71     template <class U>
  72     void converter_postcall(lua_State*, U const&, int)
  73     {}
  74 };
  75
  76 template <class T>
  77 struct default_converter<boost::shared_ptr<T> const&>
  78   : default_converter<boost::shared_ptr<T> >
  79 {};
  80
  81 } // namespace luabind
  82
  83 #endif // LUABIND_SHARED_PTR_CONVERTER_090211_HPP