luaMethod.h

Go to the documentation of this file.

/************************************************************************
**
** Copyright (C) 2009 K2FS Sistemas e Projetos Ltd.  
** All rights reserved.
**
************************************************************************/

#ifndef _LUA_METHOD_H_
#define _LUA_METHOD_H_

#include "luaConfig.h"

#include "luaEnv.h"
#include "luaProxy.h"
#include "luaStackBalancer.h"

template <typename T> class LuaMethod
{
public:     

  typedef int (T::*MethodType)(lua_State*);

  struct MethodRegistry {
    const char* _name;   
    MethodType  _method; 
  };

  static void registerMethods(lua_State* L, struct MethodRegistry* methods, int upvalues = 0)
  {
    LUA_SAVE_TOPL(L);

    if(!upvalues)
    {
      // Pilha = tabela
      for(MethodRegistry* r = methods; r->_name; ++r)
      {
        lua_pushstring(L, r->_name);  // Pilha = tabela, nome
        lua_pushlightuserdata(L, r);  // Pilha = tabela, nome, ud
        lua_pushcclosure(L, methodRedirector, 1); // Pilha = tabela, nome, funçao
        lua_settable(L, -3); // Pilha = tabela
      }
    }
    else
    {
      // Caso geral: Pilha = tabela, uv(1)..uv(n)
      lua_pushvalue(L, -(upvalues+1)); // Pilha = tabela, uv(1)..uv(n), tabela
      for(MethodRegistry* r = methods; r->_name; ++r)
      {
        lua_pushstring(L, r->_name);    // Pilha = tabela, uv(1)..uv(n), tabela, nome
        lua_pushlightuserdata(L, r);    // Pilha = tabela, uv(1)..uv(n), tabela, nome, ud

        for(int i = 0; i<upvalues; i++)
          lua_pushvalue(L, -(3+upvalues)); // Pilha = tabela, uv(1)..uv(n), tabela, nome, ud, uv(1)..uv(i+1)

        lua_pushcclosure(L, methodRedirector, 1+upvalues); // Pilha = tabela, uv(1)..uv(n), tabela, nome, funçao
        lua_settable(L, -3); // Pilha = tabela, uv(1)..uv(n), tabela
      }
      lua_pop(L, 1); // Pilha = tabela, uv(1)..uv(n)
    }

    LUA_CHECK_TOPL(L);
  }

private:

  static int methodRedirector(lua_State* L)
  {
#ifdef LUAUTILS_CANCEL
    // Verifica se a execução do script Lua deve ser cancelada
    if(LuaEnv::cancelRequested(L))
      return luaL_error(L, "%s", luaPrintable(QObject::tr("Script canceled.")));
#endif

    // Obtem upvalue com definição do método a ser chamado
          MethodRegistry* reg = (MethodRegistry*)lua_touserdata(L, lua_upvalueindex(1));

    // Obtem objeto do qual o método será chamado
    T* proxy = LuaProxy::toObjectOfClass<T>(L, 1);
    if(!proxy)
      return luaL_error(L, "%s", luaPrintable(QObject::tr("Redirector for method %1 called on incorrect object type.").arg(reg->_name)));
          
    // Chama o método do objeto passado como parâmetro
          MethodType method = reg->_method;
          return (proxy->*method)(L);  
  }
};


template <typename T> class LuaUpMethod
{
public:

  typedef int (T::*MethodType)(lua_State*);

  struct MethodRegistry {
    const char* _name;   
    MethodType  _method; 
  };

  static void registerMethods(lua_State* L, struct MethodRegistry* methods, T* obj)
  {
    LUA_SAVE_TOPL(L);
    assert(obj);

    // Pilha = tabela
    for(MethodRegistry* r = methods; r->_name; ++r)
    {
      lua_pushstring(L, r->_name);    // Pilha = tabela, nome
      lua_pushlightuserdata(L, r);    // Pilha = tabela, nome, ud_reg
      lua_pushlightuserdata(L, obj);  // Pilha = tabela, nome, ud_reg, ud_obj
      lua_pushcclosure(L, methodRedirector, 2); // Pilha = tabela, nome, funçao
      lua_settable(L, -3); // Pilha = tabela
    }

    LUA_CHECK_TOPL(L);
  }

private:

  static int methodRedirector(lua_State* L)
  {
#ifdef LUAUTILS_CANCEL
    // Verifica se a execução do script Lua deve ser cancelada
    if(LuaEnv::cancelRequested(L))
      return luaL_error(L, "%s", luaPrintable(QObject::tr("Script canceled.")));
#endif

    // Obtem upvalue com definição do método a ser chamado
    MethodRegistry* reg = (MethodRegistry*)lua_touserdata(L, lua_upvalueindex(1));

    // Obtem upvalue com definição do objeto a ser chamado
    T* obj = (T*)lua_touserdata(L, lua_upvalueindex(2));

    // Chama o método do objeto passado como parâmetro
    MethodType method = reg->_method;
    return (obj->*method)(L);
  }
};



#endif