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                         }
  51
  52                         ++state.start;
  53
  54                         return 1;
  55                 }
  56
  57                 iterator_state(const Iter& s, const Iter& e)
  58                         : start(s)
  59                         , end(e)
  60                 {}
  61
  62                 Iter start;
  63                 Iter end;
  64         };
  65
  66         struct iterator_converter
  67         {
  68                 template<class T>
  69                 void apply(lua_State* L, const T& c)
  70                 {
  71                         typedef typename T::const_iterator iter_t;
  72                         typedef iterator_state<iter_t> state_t;
  73
  74                         // note that this should be destructed, for now.. just hope that iterator
  75                         // is a pod
  76                         void* iter = lua_newuserdata(L, sizeof(state_t));
  77                         new (iter) state_t(c.begin(), c.end());
  78                         lua_pushcclosure(L, state_t::step, 1);
  79                 }
  80
  81                 template<class T>
  82                 void apply(lua_State* L, T& c)
  83                 {
  84                         typedef typename T::iterator iter_t;
  85                         typedef iterator_state<iter_t> state_t;
  86
  87                         // note that this should be destructed, for now.. just hope that iterator
  88                         // is a pod
  89                         void* iter = lua_newuserdata(L, sizeof(state_t));
  90                         new (iter) state_t(c.begin(), c.end());
  91                         lua_pushcclosure(L, state_t::step, 1);
  92                 }
  93         };
  94
  95         struct iterator_policy : conversion_policy<0>
  96         {
  97                 static void precall(lua_State*, const index_map&) {}
  98                 static void postcall(lua_State*, const index_map&) {}
  99
 100                 template<class T, class Direction>
 101                 struct generate_converter
 102                 {
 103                         typedef iterator_converter type;
 104                 };
 105         };
 106
 107 }}
 108
 109 namespace luabind
 110 {
 111         namespace
 112         {
 113                 LUABIND_ANONYMOUS_FIX detail::policy_cons<detail::iterator_policy, detail::null_type> return_stl_iterator;
 114         }
 115 }
 116
 117 #endif // LUABIND_ITERATOR_POLICY_HPP_INCLUDED
 118