Move RearmTick to Rearmable

2022-08-07 15:24:55 +03:00
parent 75d65b3d20
commit 84e7eb144b
2 changed files with 24 additions and 26 deletions
--- a/OpenRA.Mods.Common/Traits/Rearmable.cs
+++ b/OpenRA.Mods.Common/Traits/Rearmable.cs
@@ -53,5 +53,27 @@ namespace OpenRA.Mods.Common.Traits
 		}

 		void INotifyDockClient.Undocked(Actor self, Actor dock) { }
+
+		public bool RearmTick(Actor self)
+		{
+			foreach (var ammoPool in RearmableAmmoPools)
+			{
+				if (!ammoPool.HasFullAmmo)
+				{
+					if (--ammoPool.RemainingTicks <= 0)
+					{
+						ammoPool.RemainingTicks = ammoPool.Info.ReloadDelay;
+						if (!string.IsNullOrEmpty(ammoPool.Info.RearmSound))
+							Game.Sound.PlayToPlayer(SoundType.World, self.Owner, ammoPool.Info.RearmSound, self.CenterPosition);
+
+						ammoPool.GiveAmmo(self, ammoPool.Info.ReloadCount);
+					}
+
+					return false;
+				}
+			}
+
+			return true;
+		}
 	}
 }