0ct0pu5/wesnoth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Lua API: Add traits field for unit types and races

Browse source
This commit is contained in:
Celtic Minstrel 2015-09-23 16:38:52 -04:00
parent e00b57cdcd
commit acaa9ce025
2 changed files with 28 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/scripting/lua_race.cpp
View file

@ -54,6 +54,24 @@ static int impl_race_get(lua_State* L)
return_bool_attrib("ignore_global_traits", !race.uses_global_traits());
return_string_attrib("undead_variation", race.undead_variation());
return_cfgref_attrib("__cfg", race.get_cfg());
if (strcmp(m, "traits") == 0) {
lua_newtable(L);
if (race.uses_global_traits()) {
BOOST_FOREACH(const config& trait, unit_types.traits()) {
const std::string& id = trait["id"];
lua_pushlstring(L, id.c_str(), id.length());
luaW_pushconfig(L, trait);
lua_rawset(L, -3);
}
}
BOOST_FOREACH(const config& trait, race.additional_traits()) {
const std::string& id = trait["id"];
lua_pushlstring(L, id.c_str(), id.length());
luaW_pushconfig(L, trait);
lua_rawset(L, -3);
}
return 1;
}
return 0;
}

10
src/scripting/lua_unit_type.cpp
View file

@ -54,6 +54,16 @@ static int impl_unit_type_get(lua_State *L)
return_int_attrib("level", ut.level());
return_int_attrib("recall_cost", ut.recall_cost());
return_cfgref_attrib("__cfg", ut.get_cfg());
if (strcmp(m, "traits") == 0) {
lua_newtable(L);
BOOST_FOREACH(const config& trait, ut.possible_traits()) {
const std::string& id = trait["id"];
lua_pushlstring(L, id.c_str(), id.length());
luaW_pushconfig(L, trait);
lua_rawset(L, -3);
}
return 1;
}
return 0;
}