mirror of
https://github.com/love2d/love.git
synced 2026-08-15 15:51:12 +02:00
Move Lua wrapper code for Variants out of the Variant class file.
This commit is contained in:
@@ -78,6 +78,20 @@ private:
|
||||
|
||||
}; // Object
|
||||
|
||||
/**
|
||||
* Structure wrapping an object and its associated Type instance. This is used
|
||||
* for storing everything necessary to identify an object's properties in
|
||||
* environments where the Type is not easily obtained otherwise, for example in
|
||||
* a Lua state.
|
||||
**/
|
||||
struct Proxy
|
||||
{
|
||||
// Holds type information (see types.h).
|
||||
love::Type *type;
|
||||
|
||||
// Pointer to the actual object.
|
||||
Object *object;
|
||||
};
|
||||
|
||||
enum class Acquire
|
||||
{
|
||||
|
||||
+4
-153
@@ -26,25 +26,13 @@
|
||||
namespace love
|
||||
{
|
||||
|
||||
static Proxy *tryextractproxy(lua_State *L, int idx)
|
||||
{
|
||||
Proxy *u = (Proxy *)lua_touserdata(L, idx);
|
||||
|
||||
if (u == nullptr || u->type == nullptr)
|
||||
return nullptr;
|
||||
|
||||
// We could get rid of the dynamic_cast for more performance, but it would
|
||||
// be less safe...
|
||||
if (dynamic_cast<Object *>(u->object) != nullptr)
|
||||
return u;
|
||||
|
||||
return nullptr;
|
||||
}
|
||||
Variant::Variant(Type vtype)
|
||||
: type(vtype)
|
||||
{}
|
||||
|
||||
Variant::Variant()
|
||||
: type(NIL)
|
||||
{
|
||||
}
|
||||
{}
|
||||
|
||||
Variant::Variant(bool boolean)
|
||||
: type(BOOLEAN)
|
||||
@@ -152,141 +140,4 @@ Variant &Variant::operator = (const Variant &v)
|
||||
return *this;
|
||||
}
|
||||
|
||||
Variant Variant::fromLua(lua_State *L, int n, bool allowuserdata, std::set<const void*> *tableSet)
|
||||
{
|
||||
size_t len;
|
||||
const char *str;
|
||||
Proxy *p = nullptr;
|
||||
|
||||
if (n < 0) // Fix the stack position, we might modify it later
|
||||
n += lua_gettop(L) + 1;
|
||||
|
||||
switch (lua_type(L, n))
|
||||
{
|
||||
case LUA_TBOOLEAN:
|
||||
return Variant(luax_toboolean(L, n));
|
||||
case LUA_TNUMBER:
|
||||
return Variant(lua_tonumber(L, n));
|
||||
case LUA_TSTRING:
|
||||
str = lua_tolstring(L, n, &len);
|
||||
return Variant(str, len);
|
||||
case LUA_TLIGHTUSERDATA:
|
||||
return Variant(lua_touserdata(L, n));
|
||||
case LUA_TUSERDATA:
|
||||
if (!allowuserdata)
|
||||
{
|
||||
luax_typerror(L, n, "copyable Lua value");
|
||||
return Variant();
|
||||
}
|
||||
p = tryextractproxy(L, n);
|
||||
if (p != nullptr)
|
||||
return Variant(p->type, p->object);
|
||||
else
|
||||
{
|
||||
luax_typerror(L, n, "love type");
|
||||
return Variant();
|
||||
}
|
||||
case LUA_TNIL:
|
||||
return Variant();
|
||||
case LUA_TTABLE:
|
||||
{
|
||||
bool success = true;
|
||||
std::set<const void *> topTableSet;
|
||||
|
||||
// We can use a pointer to a stack-allocated variable because it's
|
||||
// never used after the stack-allocated variable is destroyed.
|
||||
if (tableSet == nullptr)
|
||||
tableSet = &topTableSet;
|
||||
|
||||
// Now make sure this table wasn't already serialised
|
||||
const void *tablePointer = lua_topointer(L, n);
|
||||
{
|
||||
auto result = tableSet->insert(tablePointer);
|
||||
if (!result.second) // insertion failed
|
||||
throw love::Exception("Cycle detected in table");
|
||||
}
|
||||
|
||||
SharedTable *table = new SharedTable();
|
||||
|
||||
size_t len = luax_objlen(L, -1);
|
||||
if (len > 0)
|
||||
table->pairs.reserve(len);
|
||||
|
||||
lua_pushnil(L);
|
||||
|
||||
while (lua_next(L, n))
|
||||
{
|
||||
table->pairs.emplace_back(fromLua(L, -2, allowuserdata, tableSet), fromLua(L, -1, allowuserdata, tableSet));
|
||||
lua_pop(L, 1);
|
||||
|
||||
const auto &p = table->pairs.back();
|
||||
if (p.first.getType() == UNKNOWN || p.second.getType() == UNKNOWN)
|
||||
{
|
||||
success = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// And remove the table from the set again
|
||||
tableSet->erase(tablePointer);
|
||||
|
||||
if (success)
|
||||
return Variant(table);
|
||||
else
|
||||
table->release();
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
Variant v;
|
||||
v.type = UNKNOWN;
|
||||
return v;
|
||||
}
|
||||
|
||||
void Variant::toLua(lua_State *L) const
|
||||
{
|
||||
switch (type)
|
||||
{
|
||||
case BOOLEAN:
|
||||
lua_pushboolean(L, data.boolean);
|
||||
break;
|
||||
case NUMBER:
|
||||
lua_pushnumber(L, data.number);
|
||||
break;
|
||||
case STRING:
|
||||
lua_pushlstring(L, data.string->str, data.string->len);
|
||||
break;
|
||||
case SMALLSTRING:
|
||||
lua_pushlstring(L, data.smallstring.str, data.smallstring.len);
|
||||
break;
|
||||
case LUSERDATA:
|
||||
lua_pushlightuserdata(L, data.userdata);
|
||||
break;
|
||||
case LOVEOBJECT:
|
||||
luax_pushtype(L, *data.objectproxy.type, data.objectproxy.object);
|
||||
break;
|
||||
case TABLE:
|
||||
{
|
||||
std::vector<std::pair<Variant, Variant>> &table = data.table->pairs;
|
||||
int tsize = (int) table.size();
|
||||
|
||||
lua_createtable(L, 0, tsize);
|
||||
|
||||
for (int i = 0; i < tsize; ++i)
|
||||
{
|
||||
std::pair<Variant, Variant> &kv = table[i];
|
||||
kv.first.toLua(L);
|
||||
kv.second.toLua(L);
|
||||
lua_settable(L, -3);
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
case NIL:
|
||||
default:
|
||||
lua_pushnil(L);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
} // love
|
||||
|
||||
@@ -22,14 +22,12 @@
|
||||
#define LOVE_VARIANT_H
|
||||
|
||||
#include "common/config.h"
|
||||
#include "common/runtime.h"
|
||||
#include "common/Object.h"
|
||||
#include "common/int.h"
|
||||
|
||||
#include <cstring>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <set>
|
||||
|
||||
namespace love
|
||||
{
|
||||
@@ -112,11 +110,12 @@ public:
|
||||
Type getType() const { return type; }
|
||||
const Data &getData() const { return data; }
|
||||
|
||||
static Variant fromLua(lua_State *L, int n, bool allowuserdata = true, std::set<const void*> *tableSet = nullptr);
|
||||
void toLua(lua_State *L) const;
|
||||
static Variant unknown() { return Variant(UNKNOWN); }
|
||||
|
||||
private:
|
||||
|
||||
Variant(Type vtype);
|
||||
|
||||
Type type;
|
||||
Data data;
|
||||
|
||||
|
||||
@@ -661,6 +661,160 @@ bool luax_istype(lua_State *L, int idx, love::Type &type)
|
||||
return false;
|
||||
}
|
||||
|
||||
static Proxy *tryextractproxy(lua_State *L, int idx)
|
||||
{
|
||||
Proxy *u = (Proxy *)lua_touserdata(L, idx);
|
||||
|
||||
if (u == nullptr || u->type == nullptr)
|
||||
return nullptr;
|
||||
|
||||
// We could get rid of the dynamic_cast for more performance, but it would
|
||||
// be less safe...
|
||||
if (dynamic_cast<Object *>(u->object) != nullptr)
|
||||
return u;
|
||||
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
Variant luax_checkvariant(lua_State *L, int n, bool allowuserdata, std::set<const void*> *tableSet)
|
||||
{
|
||||
size_t len;
|
||||
const char *str;
|
||||
Proxy *p = nullptr;
|
||||
|
||||
if (n < 0) // Fix the stack position, we might modify it later
|
||||
n += lua_gettop(L) + 1;
|
||||
|
||||
switch (lua_type(L, n))
|
||||
{
|
||||
case LUA_TBOOLEAN:
|
||||
return Variant(luax_toboolean(L, n));
|
||||
case LUA_TNUMBER:
|
||||
return Variant(lua_tonumber(L, n));
|
||||
case LUA_TSTRING:
|
||||
str = lua_tolstring(L, n, &len);
|
||||
return Variant(str, len);
|
||||
case LUA_TLIGHTUSERDATA:
|
||||
return Variant(lua_touserdata(L, n));
|
||||
case LUA_TUSERDATA:
|
||||
if (!allowuserdata)
|
||||
{
|
||||
luax_typerror(L, n, "copyable Lua value");
|
||||
return Variant();
|
||||
}
|
||||
p = tryextractproxy(L, n);
|
||||
if (p != nullptr)
|
||||
return Variant(p->type, p->object);
|
||||
else
|
||||
{
|
||||
luax_typerror(L, n, "love type");
|
||||
return Variant();
|
||||
}
|
||||
case LUA_TNIL:
|
||||
return Variant();
|
||||
case LUA_TTABLE:
|
||||
{
|
||||
bool success = true;
|
||||
std::set<const void *> topTableSet;
|
||||
|
||||
// We can use a pointer to a stack-allocated variable because it's
|
||||
// never used after the stack-allocated variable is destroyed.
|
||||
if (tableSet == nullptr)
|
||||
tableSet = &topTableSet;
|
||||
|
||||
// Now make sure this table wasn't already serialised
|
||||
const void *tablePointer = lua_topointer(L, n);
|
||||
{
|
||||
auto result = tableSet->insert(tablePointer);
|
||||
if (!result.second) // insertion failed
|
||||
throw love::Exception("Cycle detected in table");
|
||||
}
|
||||
|
||||
Variant::SharedTable *table = new Variant::SharedTable();
|
||||
|
||||
size_t len = luax_objlen(L, -1);
|
||||
if (len > 0)
|
||||
table->pairs.reserve(len);
|
||||
|
||||
lua_pushnil(L);
|
||||
|
||||
while (lua_next(L, n))
|
||||
{
|
||||
table->pairs.emplace_back(
|
||||
luax_checkvariant(L, -2, allowuserdata, tableSet),
|
||||
luax_checkvariant(L, -1, allowuserdata, tableSet)
|
||||
);
|
||||
lua_pop(L, 1);
|
||||
|
||||
const auto &p = table->pairs.back();
|
||||
if (p.first.getType() == Variant::UNKNOWN || p.second.getType() == Variant::UNKNOWN)
|
||||
{
|
||||
success = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// And remove the table from the set again
|
||||
tableSet->erase(tablePointer);
|
||||
|
||||
if (success)
|
||||
return Variant(table);
|
||||
else
|
||||
table->release();
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
return Variant::unknown();
|
||||
}
|
||||
|
||||
void luax_pushvariant(lua_State *L, const Variant &v)
|
||||
{
|
||||
const Variant::Data &data = v.getData();
|
||||
switch (v.getType())
|
||||
{
|
||||
case Variant::BOOLEAN:
|
||||
lua_pushboolean(L, data.boolean);
|
||||
break;
|
||||
case Variant::NUMBER:
|
||||
lua_pushnumber(L, data.number);
|
||||
break;
|
||||
case Variant::STRING:
|
||||
lua_pushlstring(L, data.string->str, data.string->len);
|
||||
break;
|
||||
case Variant::SMALLSTRING:
|
||||
lua_pushlstring(L, data.smallstring.str, data.smallstring.len);
|
||||
break;
|
||||
case Variant::LUSERDATA:
|
||||
lua_pushlightuserdata(L, data.userdata);
|
||||
break;
|
||||
case Variant::LOVEOBJECT:
|
||||
luax_pushtype(L, *data.objectproxy.type, data.objectproxy.object);
|
||||
break;
|
||||
case Variant::TABLE:
|
||||
{
|
||||
std::vector<std::pair<Variant, Variant>> &table = data.table->pairs;
|
||||
int tsize = (int) table.size();
|
||||
|
||||
lua_createtable(L, 0, tsize);
|
||||
|
||||
for (int i = 0; i < tsize; ++i)
|
||||
{
|
||||
std::pair<Variant, Variant> &kv = table[i];
|
||||
luax_pushvariant(L, kv.first);
|
||||
luax_pushvariant(L, kv.second);
|
||||
lua_settable(L, -3);
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
case Variant::NIL:
|
||||
default:
|
||||
lua_pushnil(L);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
int luax_getfunction(lua_State *L, const char *mod, const char *fn)
|
||||
{
|
||||
lua_getglobal(L, "love");
|
||||
|
||||
+13
-16
@@ -24,6 +24,8 @@
|
||||
// LOVE
|
||||
#include "config.h"
|
||||
#include "types.h"
|
||||
#include "Object.h"
|
||||
#include "Variant.h"
|
||||
#include "deprecation.h"
|
||||
|
||||
// Lua
|
||||
@@ -37,12 +39,12 @@ extern "C" {
|
||||
// C++
|
||||
#include <exception>
|
||||
#include <algorithm>
|
||||
#include <set>
|
||||
|
||||
namespace love
|
||||
{
|
||||
|
||||
// Forward declarations.
|
||||
class Object;
|
||||
class Module;
|
||||
class Reference;
|
||||
|
||||
@@ -59,21 +61,6 @@ enum Registry
|
||||
REGISTRY_OBJECTS
|
||||
};
|
||||
|
||||
/**
|
||||
* This structure wraps all Lua-exposed objects. It exists in the
|
||||
* Lua state as a full userdata (so we can catch __gc "events"),
|
||||
* though the Object it refers to is light userdata in the sense
|
||||
* that it is not allocated by the Lua VM.
|
||||
**/
|
||||
struct Proxy
|
||||
{
|
||||
// Holds type information (see types.h).
|
||||
love::Type *type;
|
||||
|
||||
// Pointer to the actual object.
|
||||
Object *object;
|
||||
};
|
||||
|
||||
/**
|
||||
* A Module with Lua wrapper functions and other data.
|
||||
**/
|
||||
@@ -365,6 +352,16 @@ void luax_pushtype(lua_State *L, StrongRef<T> &object)
|
||||
**/
|
||||
void luax_rawnewtype(lua_State *L, love::Type &type, love::Object *object);
|
||||
|
||||
/**
|
||||
* Stores the value at the given index on the stack into a Variant object.
|
||||
*/
|
||||
Variant luax_checkvariant(lua_State *L, int idx, bool allowuserdata = true, std::set<const void*> *tableSet = nullptr);
|
||||
|
||||
/**
|
||||
* Pushes the contents of the given Variant index onto the stack.
|
||||
*/
|
||||
void luax_pushvariant(lua_State *L, const Variant &v);
|
||||
|
||||
/**
|
||||
* Checks whether the value at idx is a certain type.
|
||||
* @param L The Lua state.
|
||||
|
||||
Reference in New Issue
Block a user