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

Fix bug in AI state loading that could cause an error when the AI is initialized from a saved game

Browse source
This commit is contained in:
Celtic Minstrel 2021-06-10 21:06:48 -04:00 • committed by GitHub
parent cb7bf96a5b
commit 42cb24e89f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/ai/lua/core.cpp
View file

@ -1050,6 +1050,10 @@ int lua_ai_load::refcount = 0;
lua_ai_load::lua_ai_load(lua_ai_context& ctx, bool read_only) : L(ctx.L), was_readonly(false) lua_ai_load::lua_ai_load(lua_ai_context& ctx, bool read_only) : L(ctx.L), was_readonly(false)
{ {
refcount++; refcount++;
lua_getfield(L, LUA_REGISTRYINDEX, aisKey); // [-1: AI registry]
lua_rawgeti(L, -1, ctx.num_); // [-1: AI state -2: AI registry]
lua_remove(L,-2); // [-1: AI state]
// Check if the AI table is already loaded. If so, we have less work to do. // Check if the AI table is already loaded. If so, we have less work to do.
lua_getglobal(L, "ai"); lua_getglobal(L, "ai");
if(!lua_isnoneornil(L, -1)) { if(!lua_isnoneornil(L, -1)) {
@ -1061,13 +1065,9 @@ lua_ai_load::lua_ai_load(lua_ai_context& ctx, bool read_only) : L(ctx.L), was_re
lua_pushstring(L, "read_only"); lua_pushstring(L, "read_only");
lua_pushboolean(L, read_only); lua_pushboolean(L, read_only);
lua_rawset(L, -3); lua_rawset(L, -3);
return; // Leave the AI table on the stack, as requested lua_pop(L, 1); // Pop the ai table off the stack
} } else {
lua_pop(L, 1); // Pop the nil value off the stack lua_pop(L, 1); // Pop the nil value off the stack
lua_getfield(L, LUA_REGISTRYINDEX, aisKey); // [-1: AI registry]
lua_rawgeti(L, -1, ctx.num_); // [-1: AI state -2: AI registry]
lua_remove(L,-2); // [-1: AI state]
// Load the AI functions table into global scope // Load the AI functions table into global scope
lua_getfield(L, -1, "ai"); // [-1: AI functions -2: AI state] lua_getfield(L, -1, "ai"); // [-1: AI functions -2: AI state]
lua_pushstring(L, "read_only"); // [-1: key -2: AI functions -3: AI state] lua_pushstring(L, "read_only"); // [-1: key -2: AI functions -3: AI state]
@ -1075,6 +1075,7 @@ lua_ai_load::lua_ai_load(lua_ai_context& ctx, bool read_only) : L(ctx.L), was_re
lua_rawset(L, -3); // [-1: AI functions -2: AI state] lua_rawset(L, -3); // [-1: AI functions -2: AI state]
lua_setglobal(L, "ai"); // [-1: AI state] lua_setglobal(L, "ai"); // [-1: AI state]
} }
}
lua_ai_load::~lua_ai_load() lua_ai_load::~lua_ai_load()
{ {