From 243ed3d76a8edafe04aec64a458c69b4b7e750df Mon Sep 17 00:00:00 2001
From: reaperrr <thereaperrr@aol.com>
Date: Sat, 17 Jun 2017 21:59:18 +0200
Subject: [PATCH] Cache traits implementing INotifyBurstComplete

---
 OpenRA.Mods.Common/Traits/Armament.cs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/OpenRA.Mods.Common/Traits/Armament.cs b/OpenRA.Mods.Common/Traits/Armament.cs
index b4c2e471ad..415ee9fc78 100644
--- a/OpenRA.Mods.Common/Traits/Armament.cs
+++ b/OpenRA.Mods.Common/Traits/Armament.cs
@@ -107,6 +107,7 @@ namespace OpenRA.Mods.Common.Traits
 		Turreted turret;
 		AmmoPool ammoPool;
 		BodyOrientation coords;
+		INotifyBurstComplete[] notifyBurstComplete;
 		IEnumerable<int> rangeModifiers;
 		List<Pair<int, Action>> delayedActions = new List<Pair<int, Action>>();
 
@@ -149,6 +150,7 @@ namespace OpenRA.Mods.Common.Traits
 			turret = self.TraitsImplementing<Turreted>().FirstOrDefault(t => t.Name == Info.Turret);
 			ammoPool = self.TraitsImplementing<AmmoPool>().FirstOrDefault(la => la.Info.Name == Info.AmmoPoolName);
 			coords = self.Trait<BodyOrientation>();
+			notifyBurstComplete = self.TraitsImplementing<INotifyBurstComplete>().ToArray();
 			rangeModifiers = self.TraitsImplementing<IRangeModifier>().ToArray().Select(m => m.GetRangeModifier());
 
 			base.Created(self);
@@ -281,7 +283,7 @@ namespace OpenRA.Mods.Common.Traits
 					});
 				}
 
-				foreach (var nbc in self.TraitsImplementing<INotifyBurstComplete>())
+				foreach (var nbc in notifyBurstComplete)
 					nbc.FiredBurst(self, target, this);
 			}