Add Goto Micro AI functionality
This commit is contained in:
mattsc
parent
83ea570285
commit
a7ccb1c877
3 changed files with 235 additions and 0 deletions
162
data/ai/micro_ais/ais/mai_goto_engine.lua
Normal file
162
data/ai/micro_ais/ais/mai_goto_engine.lua
Normal file
|
|
@ -0,0 +1,162 @@
|
|||
return {
|
||||
init = function(ai)
|
||||
|
||||
local goto_engine = {}
|
||||
|
||||
local H = wesnoth.require "lua/helper.lua"
|
||||
local AH = wesnoth.require "ai/lua/ai_helper.lua"
|
||||
local LS = wesnoth.require "lua/location_set.lua"
|
||||
|
||||
function goto_engine:goto_eval(cfg)
|
||||
|
||||
-- If cfg.release_all_units_at_goal is set, check
|
||||
-- whether the goal has already been reached, in
|
||||
-- which case we do not do anything
|
||||
if cfg.release_all_units_at_goal then
|
||||
local str = cfg.ca_id .. '-release-all'
|
||||
if self.data[str] then
|
||||
return 0
|
||||
end
|
||||
end
|
||||
|
||||
-- For convenience, we check for locations here, and just pass that to the exec function
|
||||
-- This is mostly to make the unique_goals option easier
|
||||
local width, height = wesnoth.get_map_size()
|
||||
local locs = wesnoth.get_locations {
|
||||
x = '1-' .. width,
|
||||
y = '1-' .. height,
|
||||
{ "and", cfg.filter_location }
|
||||
}
|
||||
--print('#locs org', #locs)
|
||||
if (#locs == 0) then return 0 end
|
||||
|
||||
-- If 'unique_goals' is set, check whether there are locations left to go to
|
||||
if cfg.unique_goals then
|
||||
-- First, some cleanup of previous turn data
|
||||
local str = 'goals_taken-' .. (wesnoth.current.turn - 1)
|
||||
self.data[str] = nil
|
||||
|
||||
-- Now on to the current turn
|
||||
local str = 'goals_taken-' .. wesnoth.current.turn
|
||||
for i = #locs,1,-1 do
|
||||
if self.data[str] and self.data[str]:get(locs[i][1], locs[i][2]) then
|
||||
table.remove(locs, i)
|
||||
end
|
||||
end
|
||||
end
|
||||
--print('#locs mod', #locs)
|
||||
if (not locs[1]) then return 0 end
|
||||
|
||||
-- Find the goto units
|
||||
local units = wesnoth.get_units { side = wesnoth.current.side, canrecruit = "no",
|
||||
{ "and", cfg.filter }, formula = '$this_unit.moves > 0'
|
||||
}
|
||||
|
||||
-- Exclude released units
|
||||
if cfg.release_unit_at_goal then
|
||||
for i=#units,1,-1 do
|
||||
local str = cfg.ca_id .. '-release-' .. units[i].id
|
||||
if self.data[str] then
|
||||
table.remove(units, i)
|
||||
end
|
||||
end
|
||||
end
|
||||
if (not units[1]) then return 0 end
|
||||
|
||||
-- Now store units and locs in self.data, so that we don't need to duplicate this in the exec function
|
||||
self.data.units, self.data.locs = units, locs
|
||||
|
||||
return cfg.ca_score or 210000
|
||||
end
|
||||
|
||||
function goto_engine:goto_exec(cfg)
|
||||
local units, locs = self.data.units, self.data.locs -- simply for convenience
|
||||
|
||||
local closest_hex, best_unit, max_rating = {}, {}, -9e99
|
||||
for i,u in ipairs(units) do
|
||||
for i,l in ipairs(locs) do
|
||||
|
||||
-- If use_straight_line is set, we simply find the closest
|
||||
-- hex to the goal that the unit can get to
|
||||
if cfg.use_straight_line then
|
||||
local hex, unit, rating = AH.find_best_move(u, function(x, y)
|
||||
local r = - H.distance_between(x, y, l[1], l[2])
|
||||
-- Also add distance from unit as very small rating component
|
||||
-- This is mostly here to keep unit in place when no better hexes are available
|
||||
r = r - H.distance_between(x, y, u.x, u.y) / 1000.
|
||||
return r
|
||||
end, { no_random = true })
|
||||
|
||||
if (rating > max_rating) then
|
||||
max_rating = rating
|
||||
closest_hex, best_unit = hex, u
|
||||
end
|
||||
else -- Otherwise find the best path to take
|
||||
local path, cost = wesnoth.find_path(u, l[1], l[2])
|
||||
|
||||
-- Make all hexes within the unit's current MP equaivalent
|
||||
if (cost <= u.moves) then cost = 0 end
|
||||
|
||||
rating = - cost
|
||||
|
||||
-- Add a small penalty for occupied hexes
|
||||
-- (this mean occupied by an allied unit, as enemies make the hex unreachable)
|
||||
local unit_in_way = wesnoth.get_unit(l[1], l[2])
|
||||
if unit_in_way and ((unit_in_way.x ~= u.x) or (unit_in_way.y ~= u.y)) then
|
||||
rating = rating - 0.01
|
||||
end
|
||||
|
||||
if (rating > max_rating) then
|
||||
max_rating = rating
|
||||
closest_hex, best_unit = l, u
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
--print(best_unit.id, best_unit.x, best_unit.y, closest_hex[1], closest_hex[2], max_rating)
|
||||
|
||||
AH.movefull_outofway_stopunit(ai, best_unit, closest_hex[1], closest_hex[2])
|
||||
|
||||
-- If 'unique_goals' is set, mark this location as being taken
|
||||
if cfg.unique_goals then
|
||||
local str = 'goals_taken-' .. wesnoth.current.turn
|
||||
if (not self.data[str]) then self.data[str] = LS.create() end
|
||||
self.data[str]:insert(closest_hex[1], closest_hex[2])
|
||||
end
|
||||
|
||||
-- If release_unit_at_goal= or release_all_units_at_goal= key is set:
|
||||
-- Check if the unit made it to one of the goal hexes
|
||||
-- This needs to be done for the original goal hexes, not checking the SLF again,
|
||||
-- as that might have changed based on the new situation on the map
|
||||
if cfg.release_unit_at_goal or cfg.release_all_units_at_goal then
|
||||
local unit_at_goal = false
|
||||
for i,l in ipairs(locs) do
|
||||
if (best_unit.x == l[1]) and (best_unit.y == l[2]) then
|
||||
unit_at_goal = true
|
||||
break
|
||||
end
|
||||
end
|
||||
|
||||
-- If a unit was found, mark either it or all units as released
|
||||
if unit_at_goal then
|
||||
if cfg.release_unit_at_goal then
|
||||
local str = cfg.ca_id .. '-release-' .. best_unit.id
|
||||
--print("Made it to goal: ", best_unit.id, str)
|
||||
self.data[str] = true
|
||||
end
|
||||
|
||||
if cfg.release_all_units_at_goal then
|
||||
local str = cfg.ca_id .. '-release-all'
|
||||
--print("Releasing all units")
|
||||
self.data[str] = true
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
-- And some cleanup
|
||||
self.data.units, self.data.locs = nil, nil
|
||||
end
|
||||
|
||||
return goto_engine
|
||||
end
|
||||
}
|
||||
|
|
@ -794,6 +794,59 @@ function wesnoth.wml_actions.micro_ai(cfg)
|
|||
return
|
||||
end
|
||||
|
||||
--------- Goto Micro AI - side-wide AI ------------------------------------
|
||||
if (cfg.ai_type == 'goto') then
|
||||
local cfg_go = {}
|
||||
|
||||
local required_keys = { "filter", "filter_location" }
|
||||
local optional_keys = { "ca_score", "release_all_units_at_goal", "release_unit_at_goal", "unique_goals", "use_straight_line" }
|
||||
|
||||
if (cfg.action~='delete') then
|
||||
--Add in the required keys, which could be scalars or WML tag contents
|
||||
cfg = cfg.__parsed
|
||||
for k,v in pairs(required_keys) do
|
||||
local child = H.get_child(cfg, v)
|
||||
if (not cfg[v]) and (not child) then
|
||||
H.wml_error("Goto Micro AI missing required " .. v .. "= key")
|
||||
end
|
||||
|
||||
-- Insert scalar parameters
|
||||
cfg_go[v] = cfg[v]
|
||||
-- Insert WML tags
|
||||
if child then cfg_go[v] = child end
|
||||
end
|
||||
|
||||
--Add in the optional keys, which could be scalars or WML tag contents
|
||||
for k,v in pairs(optional_keys) do
|
||||
-- Insert scalar parameters
|
||||
cfg_go[v] = cfg[v]
|
||||
|
||||
-- Insert WML tags
|
||||
local child = H.get_child(cfg, v)
|
||||
if child then cfg_go[v] = child end
|
||||
end
|
||||
end
|
||||
|
||||
-- Deal with the "ca_id=" key separately, because it doesn't influence the AI behavior
|
||||
local ca_id = 'goto'
|
||||
if cfg.ca_id then ca_id = ca_id .. '_' .. cfg.ca_id end
|
||||
-- Also pass this information to the
|
||||
|
||||
-- Set up the CA add/delete parameters
|
||||
local CA_parms = {
|
||||
{ -- Note: do not define max_score
|
||||
id = ca_id, eval_name = 'goto_eval', exec_name = 'goto_exec',
|
||||
cfg_table = cfg_go,
|
||||
pass_ca_id = true
|
||||
}
|
||||
}
|
||||
|
||||
-- Add, delete or change the CAs
|
||||
CA_action(cfg.action, cfg.side, CA_parms)
|
||||
|
||||
return
|
||||
end
|
||||
|
||||
----------------------------------------------------------------
|
||||
-- If we got here, none of the valid ai_types was specified
|
||||
H.wml_error("invalid ai_type= in [micro_ai]")
|
||||
|
|
|
|||
|
|
@ -197,3 +197,23 @@
|
|||
{RCA_STAGE}
|
||||
[/ai]
|
||||
#enddef
|
||||
|
||||
#define MICRO_AI_GOTO
|
||||
# Sets up the Goto Micro AI for a side
|
||||
# Include this macro in the side definition
|
||||
# Needs to be in [side], does not work in [modify_side] in BfW 1.10
|
||||
|
||||
[ai]
|
||||
id=goto
|
||||
description=_"Goto Micro AI"
|
||||
version=10710
|
||||
[engine]
|
||||
name="lua"
|
||||
code= <<
|
||||
local ai = ...
|
||||
return wesnoth.require("ai/micro_ais/ais/mai_goto_engine.lua").init(ai)
|
||||
>>
|
||||
[/engine]
|
||||
{RCA_STAGE}
|
||||
[/ai]
|
||||
#enddef
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue