epic/OpenRA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add plumbing for bots auto-replacing harvesters

Browse Source 
If their number drops below refinery count.
This commit is contained in:
reaperrr
2019-01-03 18:06:48 +01:00
committed by abcdefg30
parent e36ef57e35
commit 137d3be346
1 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
OpenRA.Mods.Common/Traits/BotModules/HarvesterBotModule.cs
View File

@@ -21,6 +21,13 @@ namespace OpenRA.Mods.Common.Traits
[Desc("Put this on the Player actor. Manages bot harvesters to ensure they always continue harvesting as long as there are resources on the map.")] [Desc("Put this on the Player actor. Manages bot harvesters to ensure they always continue harvesting as long as there are resources on the map.")]
public class HarvesterBotModuleInfo : ConditionalTraitInfo public class HarvesterBotModuleInfo : ConditionalTraitInfo
{ {
[Desc("Actor types that are considered harvesters. If harvester count drops below RefineryTypes count, a new harvester is built.",
"Leave empty to disable harvester replacement. Currently only needed by harvester replacement system.")]
public readonly HashSet<string> HarvesterTypes = new HashSet<string>();
[Desc("Actor types that are counted as refineries. Currently only needed by harvester replacement system.")]
public readonly HashSet<string> RefineryTypes = new HashSet<string>();
[Desc("Interval (in ticks) between giving out orders to idle harvesters.")] [Desc("Interval (in ticks) between giving out orders to idle harvesters.")]
public readonly int ScanForIdleHarvestersInterval = 50; public readonly int ScanForIdleHarvestersInterval = 50;
@@ -35,6 +42,7 @@ namespace OpenRA.Mods.Common.Traits
readonly World world; readonly World world;
readonly Player player; readonly Player player;
readonly Predicate<Actor> unitCannotBeOrdered; readonly Predicate<Actor> unitCannotBeOrdered;
IBotRequestUnitProduction[] requestUnitProduction;
IPathFinder pathfinder; IPathFinder pathfinder;
DomainIndex domainIndex; DomainIndex domainIndex;
ResourceLayer resLayer; ResourceLayer resLayer;
@@ -52,6 +60,7 @@ namespace OpenRA.Mods.Common.Traits
protected override void TraitEnabled(Actor self) protected override void TraitEnabled(Actor self)
{ {
requestUnitProduction = player.PlayerActor.TraitsImplementing<IBotRequestUnitProduction>().ToArray();
pathfinder = world.WorldActor.Trait<IPathFinder>(); pathfinder = world.WorldActor.Trait<IPathFinder>();
domainIndex = world.WorldActor.Trait<DomainIndex>(); domainIndex = world.WorldActor.Trait<DomainIndex>();
resLayer = world.WorldActor.TraitOrDefault<ResourceLayer>(); resLayer = world.WorldActor.TraitOrDefault<ResourceLayer>();
@@ -99,6 +108,16 @@ namespace OpenRA.Mods.Common.Traits
AIUtils.BotDebug("AI: Harvester {0} is idle. Ordering to {1} in search for new resources.".F(harvester, newSafeResourcePatch)); AIUtils.BotDebug("AI: Harvester {0} is idle. Ordering to {1} in search for new resources.".F(harvester, newSafeResourcePatch));
bot.QueueOrder(new Order("Harvest", harvester, Target.FromCell(world, newSafeResourcePatch), false)); bot.QueueOrder(new Order("Harvest", harvester, Target.FromCell(world, newSafeResourcePatch), false));
} }
// Less harvesters than refineries - build a new harvester
var unitBuilder = requestUnitProduction.FirstOrDefault(Exts.IsTraitEnabled);
if (unitBuilder != null && Info.HarvesterTypes.Any())
{
var harvInfo = AIUtils.GetInfoByCommonName(Info.HarvesterTypes, player);
var harvCountTooLow = AIUtils.CountActorByCommonName(Info.HarvesterTypes, player) < AIUtils.CountBuildingByCommonName(Info.RefineryTypes, player);
if (harvCountTooLow && unitBuilder.RequestedProductionCount(bot, harvInfo.Name) == 0)
unitBuilder.RequestUnitProduction(bot, harvInfo.Name);
}
} }
CPos FindNextResource(Actor actor, Harvester harv) CPos FindNextResource(Actor actor, Harvester harv)