src/function.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 #define LUABIND_BUILDING
   6
   7 #include <luabind/make_function.hpp>
   8
   9 namespace luabind { namespace detail {
  10
  11 namespace
  12 {
  13
  14   int function_destroy(lua_State* L)
  15   {
  16       function_object* fn = *(function_object**)lua_touserdata(L, 1);
  17       delete fn;
  18       return 0;
  19   }
  20
  21   void push_function_metatable(lua_State* L)
  22   {
  23       lua_pushstring(L, "luabind.function");
  24       lua_rawget(L, LUA_REGISTRYINDEX);
  25
  26       if (lua_istable(L, -1))
  27           return;
  28
  29       lua_pop(L, 1);
  30
  31       lua_newtable(L);
  32
  33       lua_pushstring(L, "__gc");
  34       lua_pushcclosure(L, &function_destroy, 0);
  35       lua_rawset(L, -3);
  36
  37       lua_pushstring(L, "luabind.function");
  38       lua_pushvalue(L, -2);
  39       lua_rawset(L, LUA_REGISTRYINDEX);
  40   }
  41
  42   // A pointer to this is used as a tag value to identify functions exported
  43   // by luabind.
  44   int function_tag = 0;
  45
  46 } // namespace unnamed
  47
  48 LUABIND_API bool is_luabind_function(lua_State* L, int index)
  49 {
  50     if (!lua_getupvalue(L, index, 2))
  51         return false;
  52     bool result = lua_touserdata(L, -1) == &function_tag;
  53     lua_pop(L, 1);
  54     return result;
  55 }
  56
  57 namespace
  58 {
  59
  60   inline bool is_luabind_function(object const& obj)
  61   {
  62       obj.push(obj.interpreter());
  63       bool result = detail::is_luabind_function(obj.interpreter(), -1);
  64       lua_pop(obj.interpreter(), 1);
  65       return result;
  66   }
  67
  68 } // namespace unnamed
  69
  70 LUABIND_API void add_overload(
  71     object const& context, char const* name, object const& fn)
  72 {
  73     function_object* f = *touserdata<function_object*>(getupvalue(fn, 1));
  74     f->name = name;
  75
  76     if (object overloads = context[name])
  77     {
  78         if (is_luabind_function(overloads) && is_luabind_function(fn))
  79         {
  80             f->next = *touserdata<function_object*>(getupvalue(overloads, 1));
  81         }
  82     }
  83
  84     context[name] = fn;
  85 }
  86
  87 LUABIND_API object make_function_aux(lua_State* L, int arity, function_object* impl)
  88 {
  89     void* storage = lua_newuserdata(L, sizeof(function_object*));
  90     push_function_metatable(L);
  91     *(function_object**)storage = impl;
  92     lua_setmetatable(L, -2);
  93
  94     lua_pushlightuserdata(L, &function_tag);
  95     lua_pushcclosure(L, impl->entry, 2);
  96     stack_pop pop(L, 1);
  97
  98     return object(from_stack(L, -1));
  99 }
 100
 101 void invoke_context::format_error(
 102     lua_State* L, function_object const* overloads) const
 103 {
 104     char const* function_name =
 105         overloads->name.empty() ? "<unknown>" : overloads->name.c_str();
 106
 107     if (candidate_index == 0)
 108     {
 109         lua_pushstring(L, "No matching overload found, candidates:\n");
 110         int count = 0;
 111         for (function_object const* f = overloads; f != 0; f = f->next)
 112         {
 113             if (count != 0)
 114                 lua_pushstring(L, "\n");
 115             f->format_signature(L, function_name);
 116             ++count;
 117         }
 118         lua_concat(L, count * 2);
 119     }
 120     else
 121     {
 122         // Ambiguous
 123         lua_pushstring(L, "Ambiguous, candidates:\n");
 124         for (int i = 0; i < candidate_index; ++i)
 125         {
 126             if (i != 0)
 127                 lua_pushstring(L, "\n");
 128             candidates[i]->format_signature(L, function_name);
 129         }
 130         lua_concat(L, candidate_index * 2);
 131     }
 132 }
 133
 134 }} // namespace luabind::detail
 135