RepairableNearUnitsWithRepairCursor #12361

This commit is contained in:
rob-v
2017-04-11 17:32:00 +02:00
parent 739653ce37
commit 9216427c2b
2 changed files with 28 additions and 5 deletions

View File

@@ -11,6 +11,7 @@
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using OpenRA.Mods.Common.Activities;
using OpenRA.Mods.Common.Orders;
using OpenRA.Traits;
@@ -84,5 +85,17 @@ namespace OpenRA.Mods.Common.Traits
self.SetTargetLine(target, Color.Green, false);
}
}
public Actor FindRepairBuilding(Actor self)
{
var repairBuilding = self.World.ActorsWithTrait<RepairsUnits>()
.Where(a => !a.Actor.IsDead && a.Actor.IsInWorld
&& a.Actor.Owner.IsAlliedWith(self.Owner) &&
info.Buildings.Contains(a.Actor.Info.Name))
.OrderBy(p => (self.Location - p.Actor.Location).LengthSquared);
// Worst case FirstOrDefault() will return a TraitPair<null, null>, which is OK.
return repairBuilding.FirstOrDefault().Actor;
}
}
}