src/scope.cpp

   1 // Copyright (c) 2004 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 #define LUABIND_BUILDING
  24
  25 #include <luabind/lua_include.hpp>
  26
  27 #include <luabind/scope.hpp>
  28 #include <luabind/detail/debug.hpp>
  29 #include <luabind/detail/stack_utils.hpp>
  30 #include <cassert>
  31
  32 namespace luabind { namespace detail {
  33
  34     registration::registration()
  35         : m_next(0)
  36     {
  37     }
  38
  39     registration::~registration()
  40     {
  41         delete m_next;
  42     }
  43
  44     } // namespace detail
  45
  46     scope::scope()
  47         : m_chain(0)
  48     {
  49     }
  50
  51     scope::scope(std::auto_ptr<detail::registration> reg)
  52         : m_chain(reg.release())
  53     {
  54     }
  55
  56     scope::scope(scope const& other)
  57         : m_chain(other.m_chain)
  58     {
  59         const_cast<scope&>(other).m_chain = 0;
  60     }
  61
  62     scope::~scope()
  63     {
  64         delete m_chain;
  65     }
  66
  67     scope& scope::operator,(scope s)
  68     {
  69         if (!m_chain)
  70         {
  71             m_chain = s.m_chain;
  72             s.m_chain = 0;
  73             return *this;
  74         }
  75
  76         for (detail::registration* c = m_chain;; c = c->m_next)
  77         {
  78             if (!c->m_next)
  79             {
  80                 c->m_next = s.m_chain;
  81                 s.m_chain = 0;
  82                 break;
  83             }
  84         }
  85
  86         return *this;
  87     }
  88
  89     void scope::register_(lua_State* L) const
  90     {
  91         for (detail::registration* r = m_chain; r != 0; r = r->m_next)
  92         {
  93                         LUABIND_CHECK_STACK(L);
  94             r->register_(L);
  95         }
  96     }
  97
  98 } // namespace luabind
  99
 100 namespace luabind {
 101
 102     namespace {
 103
 104         struct lua_pop_stack
 105         {
 106             lua_pop_stack(lua_State* L)
 107                 : m_state(L)
 108             {
 109             }
 110
 111             ~lua_pop_stack()
 112             {
 113                 lua_pop(m_state, 1);
 114             }
 115
 116             lua_State* m_state;
 117         };
 118
 119     } // namespace unnamed
 120
 121     module_::module_(lua_State* L, char const* name = 0)
 122         : m_state(L)
 123         , m_name(name)
 124     {
 125     }
 126
 127     void module_::operator[](scope s)
 128     {
 129         if (m_name)
 130         {
 131             lua_pushstring(m_state, m_name);
 132             lua_gettable(m_state, LUA_GLOBALSINDEX);
 133
 134             if (!lua_istable(m_state, -1))
 135             {
 136                 lua_pop(m_state, 1);
 137
 138                 lua_newtable(m_state);
 139                 lua_pushstring(m_state, m_name);
 140                 lua_pushvalue(m_state, -2);
 141                 lua_settable(m_state, LUA_GLOBALSINDEX);
 142             }
 143         }
 144         else
 145         {
 146             lua_pushvalue(m_state, LUA_GLOBALSINDEX);
 147         }
 148
 149         lua_pop_stack guard(m_state);
 150
 151         s.register_(m_state);
 152     }
 153
 154     struct namespace_::registration_ : detail::registration
 155     {
 156         registration_(char const* name)
 157             : m_name(name)
 158         {
 159         }
 160
 161         void register_(lua_State* L) const
 162         {
 163                         LUABIND_CHECK_STACK(L);
 164             assert(lua_gettop(L) >= 1);
 165
 166             lua_pushstring(L, m_name);
 167             lua_gettable(L, -2);
 168
 169                         detail::stack_pop p(L, 1); // pops the table on exit
 170
 171             if (!lua_istable(L, -1))
 172             {
 173                 lua_pop(L, 1);
 174
 175                 lua_newtable(L);
 176                 lua_pushstring(L, m_name);
 177                 lua_pushvalue(L, -2);
 178                 lua_settable(L, -4);
 179             }
 180
 181             m_scope.register_(L);
 182         }
 183
 184         char const* m_name;
 185         scope m_scope;
 186     };
 187
 188     namespace_::namespace_(char const* name)
 189         : scope(std::auto_ptr<detail::registration>(
 190               m_registration = new registration_(name)))
 191     {
 192     }
 193
 194     namespace_& namespace_::operator[](scope s)
 195     {
 196         m_registration->m_scope.operator,(s);
 197         return *this;
 198     }
 199
 200 } // namespace luabind
 201