Improve implementation of LS.getclass and LS.classname, make it more consistent.
This commit is contained in:
@@ -85,6 +85,9 @@ lua_State *LuaCoreStack::newstate (lua_Alloc allocf) {
|
||||
lua_pushstring(L, "classes");
|
||||
lua_newtable(L);
|
||||
lua_rawset(L, LUA_REGISTRYINDEX);
|
||||
lua_pushstring(L, "classnames");
|
||||
lua_newtable(L);
|
||||
lua_rawset(L, LUA_REGISTRYINDEX);
|
||||
lua_pushstring(L, "tangibles");
|
||||
lua_newtable(L);
|
||||
lua_rawset(L, LUA_REGISTRYINDEX);
|
||||
@@ -333,112 +336,113 @@ lua_State *LuaCoreStack::newthread(LuaSlot target) const {
|
||||
return result;
|
||||
}
|
||||
|
||||
bool LuaCoreStack::valididentifier(std::string_view str) {
|
||||
return sv::is_lua_id(str);
|
||||
}
|
||||
eng::string LuaCoreStack::classname(LuaSlot input) const {
|
||||
LuaVar lookup, classtab, classname;
|
||||
LuaExtStack LS(L_, lookup, classtab, classname);
|
||||
|
||||
bool LuaCoreStack::validclassname(std::string_view cname) {
|
||||
if (cname == "_G") return false;
|
||||
return valididentifier(cname);
|
||||
}
|
||||
|
||||
bool LuaCoreStack::validclassname(LuaSlot slot) const {
|
||||
if (!isstring(slot)) return false;
|
||||
return validclassname(ckstring(slot));
|
||||
}
|
||||
|
||||
eng::string LuaCoreStack::classname(LuaSlot tab) const {
|
||||
eng::string result;
|
||||
if ((istable(tab)) && (gettabletype(tab) == LUA_TT_CLASS)) {
|
||||
LuaVar classes, name, dup;
|
||||
LuaExtStack LS(L_, classes, name, dup);
|
||||
// Get the classes table from the registry.
|
||||
LS.rawget(classes, LuaRegistry, "classes");
|
||||
|
||||
// Try the efficient approach: get the class name from
|
||||
// the class, and confirm it by checking the classes table.
|
||||
LS.rawget(name, tab, "__class");
|
||||
if (LS.isstring(name)) {
|
||||
LS.rawget(dup, classes, name);
|
||||
if (LS.rawequal(dup, tab)) {
|
||||
return LS.ckstring(name);
|
||||
}
|
||||
int xt = xtype(input);
|
||||
if (xt == LUA_TSTRING) {
|
||||
LS.rawget(lookup, LuaRegistry, "classes");
|
||||
LS.rawget(classtab, lookup, input);
|
||||
if (xtype(classtab) != LUA_TT_CLASS) {
|
||||
return "";
|
||||
}
|
||||
|
||||
// Do it the brute force way: scan the classes table.
|
||||
LS.set(name, LuaNil);
|
||||
while (LS.next(classes, name, dup)) {
|
||||
if (LS.rawequal(dup, tab)) {
|
||||
return LS.ckstring(name);
|
||||
}
|
||||
return LS.ckstring(input);
|
||||
} else if (xt == LUA_TT_TANGIBLE) {
|
||||
if (!LS.getmetatable(lookup, input)) {
|
||||
return "";
|
||||
}
|
||||
LS.rawget(classtab, lookup, "__index");
|
||||
if (xtype(classtab) != LUA_TT_CLASS) {
|
||||
return "";
|
||||
}
|
||||
LS.rawget(lookup, LuaRegistry, "classnames");
|
||||
LS.rawget(classname, lookup, classtab);
|
||||
if (!LS.isstring(classname)) {
|
||||
return "";
|
||||
}
|
||||
return LS.ckstring(classname);
|
||||
} else if (xt == LUA_TT_CLASS) {
|
||||
LS.rawget(lookup, LuaRegistry, "classnames");
|
||||
LS.rawget(classname, lookup, input);
|
||||
if (!LS.isstring(classname)) {
|
||||
return "";
|
||||
}
|
||||
return LS.ckstring(classname);
|
||||
} else {
|
||||
return "";
|
||||
}
|
||||
return "";
|
||||
}
|
||||
|
||||
eng::string LuaCoreStack::getclass(LuaSlot classtab, LuaSlot classname) const {
|
||||
lua_checkstack(L_, 20);
|
||||
LuaVar globtab, cname;
|
||||
LuaExtStack LS(L_, globtab, cname);
|
||||
LS.getglobaltable(globtab);
|
||||
static eng::string nonexistentclass(eng::string name) {
|
||||
eng::string err;
|
||||
if (!sv::is_lua_classname(name)) {
|
||||
err = "invalid class name: " + name;
|
||||
} else {
|
||||
err = "not a class: " + name;
|
||||
}
|
||||
return err;
|
||||
}
|
||||
|
||||
if (LS.isstring(classname)) {
|
||||
if (!validclassname(LS.ckstring(classname))) {
|
||||
eng::string err = "invalid class name: " + LS.ckstring(classname);
|
||||
return err;
|
||||
}
|
||||
LS.rawget(classtab, globtab, classname);
|
||||
if (!LS.istable(classtab)) {
|
||||
eng::string err = "not a class: " + LS.ckstring(classname);
|
||||
return err;
|
||||
}
|
||||
LS.rawget(cname, classtab, "__class");
|
||||
if (!LS.rawequal(cname, classname)) {
|
||||
eng::string err = "not a valid class: " + LS.ckstring(classname);
|
||||
return err;
|
||||
eng::string LuaCoreStack::getclass(LuaSlot classtab, LuaSlot input) const {
|
||||
LuaVar lookup;
|
||||
LuaExtStack LS(L_, lookup);
|
||||
|
||||
int xt = xtype(input);
|
||||
if (xt == LUA_TSTRING) {
|
||||
LS.rawget(lookup, LuaRegistry, "classes");
|
||||
LS.rawget(classtab, lookup, input);
|
||||
if (xtype(classtab) != LUA_TT_CLASS) {
|
||||
eng::string classname = LS.ckstring(input);
|
||||
return nonexistentclass(LS.ckstring(input));
|
||||
}
|
||||
return "";
|
||||
} else if (LS.istable(classname)) {
|
||||
LS.rawget(cname, classname, "__class");
|
||||
if (!LS.isstring(cname)) {
|
||||
eng::string err = "table is not a class.";
|
||||
} else if (xt == LUA_TT_TANGIBLE) {
|
||||
if (!LS.getmetatable(lookup, input)) {
|
||||
eng::string err = "inactive tangible has no class";
|
||||
return err;
|
||||
}
|
||||
if (!validclassname(LS.ckstring(cname))) {
|
||||
eng::string err = "invalid class name: " + LS.ckstring(cname);
|
||||
return err;
|
||||
}
|
||||
LS.rawget(classtab, globtab, cname);
|
||||
if (!LS.rawequal(classtab, classname)) {
|
||||
eng::string err = "not a valid class: " + LS.ckstring(cname);
|
||||
return err;
|
||||
LS.rawget(classtab, lookup, "__index");
|
||||
if (xtype(classtab) != LUA_TT_CLASS) {
|
||||
eng::string err = "tangible has invalid class";
|
||||
}
|
||||
return "";
|
||||
} else if (xt == LUA_TT_CLASS) {
|
||||
LS.set(classtab, input);
|
||||
return "";
|
||||
} else {
|
||||
eng::string err = "getclass expects a string or a classtab";
|
||||
eng::string err = "getclass must be passed a string, class, or tangible";
|
||||
return err;
|
||||
}
|
||||
}
|
||||
|
||||
eng::string LuaCoreStack::getclass(LuaSlot tab, std::string_view name) const {
|
||||
push_any_value(name);
|
||||
LuaSpecial classname(lua_gettop(L_));
|
||||
eng::string err = getclass(tab, classname);
|
||||
lua_pop(L_, 1);
|
||||
return err;
|
||||
eng::string LuaCoreStack::getclass(LuaSlot classtab, std::string_view classname) const {
|
||||
LuaVar lookup;
|
||||
LuaExtStack LS(L_, lookup);
|
||||
LS.rawget(lookup, LuaRegistry, "classes");
|
||||
LS.rawget(classtab, lookup, classname);
|
||||
if (xtype(classtab) != LUA_TT_CLASS) {
|
||||
return nonexistentclass(eng::string(classname));
|
||||
}
|
||||
return "";
|
||||
}
|
||||
|
||||
void LuaCoreStack::makeclass(LuaSlot classtab, LuaSlot classname) const {
|
||||
LuaVar classes, globtab, cname;
|
||||
LuaExtStack LS(L_, classes, globtab, cname);
|
||||
LuaVar classes, classnames, globtab, cname;
|
||||
LuaExtStack LS(L_, classes, classnames, globtab, cname);
|
||||
|
||||
// Validate the class name.
|
||||
assert(LS.validclassname(classname));
|
||||
assert(LS.isstring(classname));
|
||||
assert(sv::is_lua_classname(LS.ckstring(classname)));
|
||||
|
||||
// Fetch the classes table from the registry.
|
||||
LS.rawget(classes, LuaRegistry, "classes");
|
||||
assert(LS.istable(classes));
|
||||
|
||||
// Fetch the classnames table from the registry.
|
||||
LS.rawget(classnames, LuaRegistry, "classnames");
|
||||
assert(LS.istable(classnames));
|
||||
|
||||
// Fetch the global environment from the registry.
|
||||
LS.getglobaltable(globtab);
|
||||
|
||||
@@ -449,6 +453,7 @@ void LuaCoreStack::makeclass(LuaSlot classtab, LuaSlot classname) const {
|
||||
if (!LS.istable(classtab)) {
|
||||
LS.newtable(classtab);
|
||||
LS.rawset(classes, classname, classtab);
|
||||
LS.rawset(classnames, classtab, classname);
|
||||
}
|
||||
|
||||
// Put the table into the global environment.
|
||||
@@ -456,7 +461,6 @@ void LuaCoreStack::makeclass(LuaSlot classtab, LuaSlot classname) const {
|
||||
|
||||
// Repair the special fields.
|
||||
LS.settabletype(classtab, LUA_TT_CLASS);
|
||||
LS.rawset(classtab, "__class", classname);
|
||||
LS.rawset(classtab, "__index", classtab);
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user