luabind/iterator_policy.hpp

   1 // Copyright (c) 2003 Daniel Wallin and Arvid Norberg
   2
   3 // Permission is hereby granted, free of charge, to any person obtaining a
   4 // copy of this software and associated documentation files (the "Software"),
   5 // to deal in the Software without restriction, including without limitation
   6 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
   7 // and/or sell copies of the Software, and to permit persons to whom the
   8 // Software is furnished to do so, subject to the following conditions:
   9
  10 // The above copyright notice and this permission notice shall be included
  11 // in all copies or substantial portions of the Software.
  12
  13 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF
  14 // ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED
  15 // TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
  16 // PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
  17 // SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR
  18 // ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  19 // ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE
  21 // OR OTHER DEALINGS IN THE SOFTWARE.
  22
  23
  24 #ifndef LUABIND_ITERATOR_POLICY_HPP_INCLUDED
  25 #define LUABIND_ITERATOR_POLICY_HPP_INCLUDED
  26
  27 #include <luabind/config.hpp>
  28 #include <luabind/detail/policy.hpp>
  29 #include <luabind/detail/implicit_cast.hpp>
  30 #include <luabind/detail/convert_to_lua.hpp>
  31
  32 namespace luabind { namespace detail
  33 {
  34         template<class Iter>
  35         struct iterator_state
  36         {
  37                 typedef iterator_state<Iter> self_t;
  38
  39                 static int step(lua_State* L)
  40                 {
  41                         self_t& state = *static_cast<self_t*>(lua_touserdata(L, lua_upvalueindex(1)));
  42
  43                         if (state.start == state.end)
  44                         {
  45                                 lua_pushnil(L);
  46                         }
  47                         else
  48                         {
  49                                 convert_to_lua(L, *state.start);
  50                                 ++state.start;
  51                         }
  52
  53                         return 1;
  54                 }
  55
  56                 iterator_state(const Iter& s, const Iter& e)
  57                         : start(s)
  58                         , end(e)
  59                 {}
  60
  61                 Iter start;
  62                 Iter end;
  63         };
  64
  65         struct iterator_converter
  66         {
  67                 template<class T>
  68                 void apply(lua_State* L, const T& c)
  69                 {
  70                         typedef typename T::const_iterator iter_t;
  71                         typedef iterator_state<iter_t> state_t;
  72
  73                         // note that this should be destructed, for now.. just hope that iterator
  74                         // is a pod
  75                         void* iter = lua_newuserdata(L, sizeof(state_t));
  76                         new (iter) state_t(c.begin(), c.end());
  77                         lua_pushcclosure(L, state_t::step, 1);
  78                 }
  79
  80                 template<class T>
  81                 void apply(lua_State* L, T& c)
  82                 {
  83                         typedef typename T::iterator iter_t;
  84                         typedef iterator_state<iter_t> state_t;
  85
  86                         // note that this should be destructed, for now.. just hope that iterator
  87                         // is a pod
  88                         void* iter = lua_newuserdata(L, sizeof(state_t));
  89                         new (iter) state_t(c.begin(), c.end());
  90                         lua_pushcclosure(L, state_t::step, 1);
  91                 }
  92         };
  93
  94         struct iterator_policy : conversion_policy<0>
  95         {
  96                 static void precall(lua_State*, const index_map&) {}
  97                 static void postcall(lua_State*, const index_map&) {}
  98
  99                 template<class T, class Direction>
 100                 struct generate_converter
 101                 {
 102                         typedef iterator_converter type;
 103                 };
 104         };
 105
 106 }}
 107
 108 namespace luabind
 109 {
 110         namespace
 111         {
 112                 LUABIND_ANONYMOUS_FIX detail::policy_cons<detail::iterator_policy, detail::null_type> return_stl_iterator;
 113         }
 114 }
 115
 116 #endif // LUABIND_ITERATOR_POLICY_HPP_INCLUDED
 117