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

ai_helper.get_cheapest_recruit_cost: add optional input 'leader'

Browse source
This commit is contained in:
mattsc 2019-12-05 12:25:08 -08:00
parent a227e768f9
commit f8a9238e3f
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
data/ai/lua/ai_helper.lua
View file

@ -1283,13 +1283,20 @@ function ai_helper.has_weapon_special(unit, special)
return false return false
end end
function ai_helper.get_cheapest_recruit_cost() function ai_helper.get_cheapest_recruit_cost(leader)
-- Optional input @leader: if given, find the cheapest recruit cost for this leader,
-- otherwise for the combination of all leaders of the current side
local recruit_ids = {} local recruit_ids = {}
for _,recruit_id in ipairs(wesnoth.sides[wesnoth.current.side].recruit) do for _,recruit_id in ipairs(wesnoth.sides[wesnoth.current.side].recruit) do
table.insert(recruit_ids, recruit_id) table.insert(recruit_ids, recruit_id)
end end
local leaders = wesnoth.get_units { side = wesnoth.current.side, canrecruit = 'yes' } local leaders
if leader then
leaders = { leader }
else
leaders = wesnoth.get_units { side = wesnoth.current.side, canrecruit = 'yes' }
end
for _,l in ipairs(leaders) do for _,l in ipairs(leaders) do
for _,recruit_id in ipairs(l.extra_recruit) do for _,recruit_id in ipairs(l.extra_recruit) do
table.insert(recruit_ids, recruit_id) table.insert(recruit_ids, recruit_id)