1

我正在尝试将 Lua 类对象推入堆栈。指向该对象的指针可以由多个函数返回。

换句话说:我需要推送用户数据值,同时仍然能够在它们上使用“==”、“~=”等,因此如果用户数据指针是相同的 C++ 对象,则它必须相同。

-- this should push the object onto the stack
local firstObject = GetClassObject();
firstObject:doSomething();

firstObject 将由 lua 脚本存储,稍后在代码中我将需要再次执行此操作:

-- the c++ class pointer has not changed here
-- so I would like to push the same userdata pointer as in the first call...
local object = GetClassObject();

-- if I would not do this the following here would fail... :C
if object == firstObject then
...

我的 Push 函数基本上应该检查某处是否已经存在相同的 C++ 类指针,如果是,则推送相关的 userdata 指针(无论我如何推送它,对象应该以 1:1 的方式工作)

如果不是,它应该创建一个新的用户数据(将其推入堆栈)并将其内容设置为类对象。

这是我的代码:

template <typename T>
void Push( const T &tObject )
{
    lua_State *L = GetLuaState();

    // Here i need to check if such a C++ object (the same tObject)
    // already exists!
    //
    // If so i want to push the associated userdata.


    // Object didn't exist yet -> we need a new userdata
    void *pUserData = lua_newuserdata( L, sizeof( tObject ) );
    *reinterpret_cast<T*>( pUserData ) = tObject;
}

template <typename T>
void Push( const T &tObject, const char *pszTable )
{
    Push( tObject );
    lua_State *L = GetLuaState();
    luaL_getmetatable( L, pszTable );
    lua_setmetatable( L, -2 );
}

template <typename T>
T& Get( int nIndex )
{
    T *pUserData = reinterpret_cast<T*>( lua_touserdata( GetLuaState(), nIndex ) );
    if( pUserData == nullptr )
        throw std::exception( "Invalid userdata!" );

    return *pUserData;
}

template <typename T>
T& Get( int nIndex, const char *pszTable )
{
    T *pUserData = reinterpret_cast<T*>( LuaToUData( nIndex, pszTable ) );
    if( pUserData == nullptr )
        throw std::exception( "Invalid userdata!" );

    return *pUserData;
}

LuaToUData 是一个自己的函数,我写它不会引发 lua 错误:

void* LuaToUData( int nIndex, const char *pszTable )
{
    void *pUserData = lua_touserdata( g_luaState, nIndex );
    if( pUserData != nullptr )
    {
        if( lua_getmetatable( g_luaState, nIndex ) != 0 )
        {
            lua_getfield( g_luaState, LUA_REGISTRYINDEX, pszTable );
            bool bEqual = ( lua_rawequal( g_luaState, -1, -2 ) == 1 );
            lua_pop( g_luaState, 2 );

            if( bEqual )
                return pUserData;
        }
    }

    return nullptr;
}
4

2 回答 2

1

没错,在 Lua 中,任何两个相同用户数据的实例都保证是相等的。但是,当您像这样对 C++ 类实例进行装箱时,每个装箱的实例都会被放入一个新的用户数据中,这意味着它们不能直接进行比较。

您需要做的是__eq为您的对象定义一个元方法。它可能看起来有点像这样:

int l_compare_things(lua_State* l)
{
    MyClass* a = reinterpret_cast<MyClass*>(lua_touserdata(L, 1));
    MyClass* b = reinterpret_cast<MyClass*>(lua_touserdata(L, 2));

    lua_pushboolean(L, (*a) == (*b));

    return 1;
}

这假设MyClass有某种operator==覆盖。您可以将此函数设置为与用户数据项关联的元表中的 __eq 元方法MyClass。您似乎已经涵盖了元表处理,所以我不会在这里打扰。

现在,下一个问题:您将整个类实例装箱为 lua 完整的用户数据项。您可能不想一遍又一遍地推送相同的东西并用完所有可用的内存......如果您只是推送指针,这不是一个问题,但您没有这样做。所以......您将需要一些独特的方式来识别您的 C++ 类的每个实例。这是一个带有字符串的示例:

class MyClass
{
private:
    std::string _id;
public:
    MyClass(const std::string& id) : _id(id) {}

    const std::string& get_id() { return _id; }

    // setters and operator= overrides not included.
};

void l_push_thing(lua_State* L, const MyClass& thing)
{
    // try to get our instance by ID from the registry table:
    lua_getfield(L, LUA_REGISTRYINDEX, thing.id());

    // if so, return, leaving it at the top of the stack.
    if (lua_isuserdata(L, -1))
        return;

    void *ud = lua_newuserdata(L, sizeof(MyClass));                       
    *reinterpret_cast<MyClass*>(ud) = thing; 
    // set up the metatable, etc

    // duplicate the userdata reference:
    lua_pushvalue(L, -1);

    // push our new userdata into the registry. pops the duplicate from the stack
    lua_setfield(L, LUA_REGISTRYINDEX, thing.get_id());
}

(注意:我没有编译或测试过这个例子。E&OE!)

这会将与某个特定MyClass实例关联的用户数据留在堆栈顶部。您需要采取自己的步骤来“取消注册”类实例;在这种情况下,注册表中存在对每个实例的硬引用,因此在您销毁该引用之前不会对用户数据进行垃圾收集。您可能会考虑在此处使用弱/临时表。

于 2012-06-24T19:06:25.627 回答
0

弱表是这样工作的吗?

void Push( const T &tObject )
{
    std::ostringstream o;
    o << tObject;
    std::string sIdentifier = o.str();
    const char *pszIdentifier = sIdentifier.c_str();

    lua_State *L = GetLuaState();
    luaL_getmetatable( L, "lua_userdata" );
    if( !lua_istable( L, -1 ) )
    {
        // create new weak table
        luaL_newmetatable( L, "lua_userdata" );
        lua_pushstring( L, "v" );
        lua_setfield( L, -2, "__mode" );
    }

    lua_getfield( L, -1, pszIdentifier );
    if( lua_isuserdata( L, -1 ) == TRUE )
        return lua_remove( L, -2 );

    lua_pop( L, 1 ); // didnt exist yet - getfield is nil -> need to pop that
    void *pUserData = lua_newuserdata( L, sizeof( UINT64 ) );
    *reinterpret_cast<UINT64*>( pUserData ) = UINT64( tObject );

    lua_pushvalue( L, -1 );
    lua_setfield( L, -3, pszIdentifier );
    lua_remove( L, -2 );
}
于 2012-06-25T21:06:13.017 回答