Skip to content

Commit

Permalink
Very minor channel / variant performance improvements and cleanup.
Browse files Browse the repository at this point in the history
  • Loading branch information
slime73 committed Mar 12, 2016
1 parent a0bbcce commit d0c0240
Show file tree
Hide file tree
Showing 6 changed files with 48 additions and 57 deletions.
48 changes: 21 additions & 27 deletions src/common/Variant.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,7 @@ Variant &Variant::operator = (const Variant &v)
return *this;
}

bool Variant::fromLua(lua_State *L, int n, Variant *v, bool allowTables)
Variant Variant::fromLua(lua_State *L, int n, bool allowTables)
{
size_t len;
const char *str;
Expand All @@ -171,30 +171,23 @@ bool Variant::fromLua(lua_State *L, int n, Variant *v, bool allowTables)
switch (lua_type(L, n))
{
case LUA_TBOOLEAN:
*v = Variant(luax_toboolean(L, n));
return true;
return Variant(luax_toboolean(L, n));
case LUA_TNUMBER:
*v = Variant(lua_tonumber(L, n));
return true;
return Variant(lua_tonumber(L, n));
case LUA_TSTRING:
str = lua_tolstring(L, n, &len);
*v = Variant(str, len);
return true;
return Variant(str, len);
case LUA_TLIGHTUSERDATA:
*v = Variant(lua_touserdata(L, n));
return true;
return Variant(lua_touserdata(L, n));
case LUA_TUSERDATA:
*v = Variant(extractudatatype(L, n), lua_touserdata(L, n));
return true;
return Variant(extractudatatype(L, n), lua_touserdata(L, n));
case LUA_TNIL:
*v = Variant();
return true;
return Variant();
case LUA_TTABLE:
if (allowTables)
{
bool success = true;
std::vector<std::pair<Variant, Variant>> *table = new std::vector<std::pair<Variant, Variant>>();
std::pair<Variant, Variant> pair;

size_t len = luax_objlen(L, -1);
if (len > 0)
Expand All @@ -204,29 +197,28 @@ bool Variant::fromLua(lua_State *L, int n, Variant *v, bool allowTables)

while (lua_next(L, n))
{
if (!fromLua(L, -2, &pair.first, false) || !fromLua(L, -1, &pair.second, false))
table->emplace_back(fromLua(L, -2), fromLua(L, -1));
lua_pop(L, 1);

const auto &p = table->back();
if (p.first.getType() == UNKNOWN || p.second.getType() == UNKNOWN)
{
success = false;
lua_pop(L, 2);
break;
}

table->push_back(pair);
lua_pop(L, 1);
}

if (success)
{
*v = Variant(table);
return true;
}
return Variant(table);
else
delete table;
}
break;
}

return false;
Variant v;
v.type = UNKNOWN;
return v;
}

void Variant::toLua(lua_State *L) const
Expand Down Expand Up @@ -260,11 +252,13 @@ void Variant::toLua(lua_State *L) const
case TABLE:
{
std::vector<std::pair<Variant, Variant>> *table = data.table->table;
lua_createtable(L, 0, (int) table->size());
int tsize = (int) table->size();

lua_createtable(L, 0, tsize);

for (size_t i = 0; i < table->size(); ++i)
for (int i = 0; i < tsize; ++i)
{
std::pair<Variant, Variant> &kv = table->at(i);
std::pair<Variant, Variant> &kv = (*table)[i];
kv.first.toLua(L);
kv.second.toLua(L);
lua_settable(L, -3);
Expand Down
30 changes: 16 additions & 14 deletions src/common/Variant.h
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,19 @@ class Variant
{
public:

enum Type
{
UNKNOWN = 0,
BOOLEAN,
NUMBER,
STRING,
SMALLSTRING,
LUSERDATA,
FUSERDATA,
NIL,
TABLE
};

Variant();
Variant(bool boolean);
Variant(double number);
Expand All @@ -48,7 +61,9 @@ class Variant

Variant &operator = (const Variant &v);

static bool fromLua(lua_State *L, int n, Variant *v, bool allowTables = true);
Type getType() const { return type; }

static Variant fromLua(lua_State *L, int n, bool allowTables = true);
void toLua(lua_State *L) const;

private:
Expand Down Expand Up @@ -83,19 +98,6 @@ class Variant
std::vector<std::pair<Variant, Variant>> *table;
};

enum Type
{
UNKNOWN = 0,
BOOLEAN,
NUMBER,
STRING,
SMALLSTRING,
LUSERDATA,
FUSERDATA,
NIL,
TABLE
};

static const int MAX_SMALL_STRING_LENGTH = 15;

Type type;
Expand Down
6 changes: 3 additions & 3 deletions src/modules/event/Event.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -63,14 +63,14 @@ Message *Message::fromLua(lua_State *L, int n)
if (lua_isnoneornil(L, n+i))
break;

if (!Variant::fromLua(L, n+i, &varg))
vargs.push_back(Variant::fromLua(L, n+i));

if (vargs.back().getType() == Variant::UNKNOWN)
{
vargs.clear();
luaL_error(L, "Argument %d can't be stored safely\nExpected boolean, number, string or userdata.", n+i);
return nullptr;
}

vargs.push_back(varg);
}

return new Message(name, vargs);
Expand Down
6 changes: 1 addition & 5 deletions src/modules/event/wrap_Event.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -93,11 +93,7 @@ int w_clear(lua_State *)

int w_quit(lua_State *L)
{
std::vector<Variant> args;

Variant v;
if (Variant::fromLua(L, 1, &v))
args.push_back(v);
std::vector<Variant> args = {Variant::fromLua(L, 1)};

StrongRef<Message> m(new Message("quit", args), Acquire::NORETAIN);
instance()->push(m);
Expand Down
8 changes: 4 additions & 4 deletions src/modules/thread/wrap_Channel.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -33,8 +33,8 @@ Channel *luax_checkchannel(lua_State *L, int idx)
int w_Channel_push(lua_State *L)
{
Channel *c = luax_checkchannel(L, 1);
Variant var;
if (!Variant::fromLua(L, 2, &var))
Variant var = Variant::fromLua(L, 2);
if (var.getType() == Variant::UNKNOWN)
return luaL_argerror(L, 2, "boolean, number, string, love type, or flat table expected");
c->push(var);
return 0;
Expand All @@ -43,8 +43,8 @@ int w_Channel_push(lua_State *L)
int w_Channel_supply(lua_State *L)
{
Channel *c = luax_checkchannel(L, 1);
Variant var;
if (!Variant::fromLua(L, 2, &var))
Variant var = Variant::fromLua(L, 2);
if (var.getType() == Variant::UNKNOWN)
return luaL_argerror(L, 2, "boolean, number, string, love type, or flat table expected");
c->supply(var);
return 0;
Expand Down
7 changes: 3 additions & 4 deletions src/modules/thread/wrap_LuaThread.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -36,16 +36,15 @@ int w_Thread_start(lua_State *L)
std::vector<Variant> args;
int nargs = lua_gettop(L) - 1;

Variant v;
for (int i = 0; i < nargs; ++i)
{
if (!Variant::fromLua(L, i+2, &v))
args.push_back(Variant::fromLua(L, i+2));

if (args.back().getType() == Variant::UNKNOWN)
{
args.clear();
return luaL_argerror(L, i+2, "boolean, number, string, love type, or flat table expected");
}

args.push_back(v);
}

luax_pushboolean(L, t->start(args));
Expand Down

0 comments on commit d0c0240

Please sign in to comment.