From 85c54e04d99cc2543d9335516250640eb9a40fd3 Mon Sep 17 00:00:00 2001 From: reaperrr Date: Tue, 12 Dec 2017 03:42:16 +0100 Subject: [PATCH] Replace LINQ in AirStates.CountAntiAirUnits --- OpenRA.Mods.Common/AI/States/AirStates.cs | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/OpenRA.Mods.Common/AI/States/AirStates.cs b/OpenRA.Mods.Common/AI/States/AirStates.cs index 413655ea78..514bf54477 100644 --- a/OpenRA.Mods.Common/AI/States/AirStates.cs +++ b/OpenRA.Mods.Common/AI/States/AirStates.cs @@ -34,16 +34,18 @@ namespace OpenRA.Mods.Common.AI if (unit == null || unit.Info.HasTraitInfo()) continue; - var arms = unit.TraitsImplementing() - .Where(ab => !ab.IsTraitPaused && !ab.IsTraitDisabled) - .SelectMany(ab => ab.Armaments); - - foreach (var a in arms) + foreach (var ab in unit.TraitsImplementing()) { - if (a.Weapon.IsValidTarget(AirTargetTypes)) + if (ab.IsTraitDisabled || ab.IsTraitPaused) + continue; + + foreach (var a in ab.Armaments) { - missileUnitsCount++; - break; + if (a.Weapon.IsValidTarget(AirTargetTypes)) + { + missileUnitsCount++; + break; + } } } }