From 482edb086a9ced9eb7717f950942d22634ea2eda Mon Sep 17 00:00:00 2001
From: Chris Forbes <chrisf@ijw.co.nz>
Date: Mon, 28 Dec 2009 18:14:40 +1300
Subject: [PATCH] limited ammo support

---
 OpenRa.Game/OpenRa.Game.csproj    |  1 +
 OpenRa.Game/Traits/AttackBase.cs  |  5 +++++
 OpenRa.Game/Traits/LimitedAmmo.cs | 29 +++++++++++++++++++++++++++++
 3 files changed, 35 insertions(+)
 create mode 100644 OpenRa.Game/Traits/LimitedAmmo.cs

diff --git a/OpenRa.Game/OpenRa.Game.csproj b/OpenRa.Game/OpenRa.Game.csproj
index 686f62a19d..1b819034ed 100644
--- a/OpenRa.Game/OpenRa.Game.csproj
+++ b/OpenRa.Game/OpenRa.Game.csproj
@@ -181,6 +181,7 @@
     <Compile Include="Traits\Fake.cs" />
     <Compile Include="Traits\Harvester.cs" />
     <Compile Include="Traits\Helicopter.cs" />
+    <Compile Include="Traits\LimitedAmmo.cs" />
     <Compile Include="Traits\SquishByTank.cs" />
     <Compile Include="Traits\Plane.cs" />
     <Compile Include="Traits\ProductionQueue.cs" />
diff --git a/OpenRa.Game/Traits/AttackBase.cs b/OpenRa.Game/Traits/AttackBase.cs
index 2253655437..041b9a8f74 100644
--- a/OpenRa.Game/Traits/AttackBase.cs
+++ b/OpenRa.Game/Traits/AttackBase.cs
@@ -91,6 +91,11 @@ namespace OpenRa.Game.Traits
 		bool CheckFire(Actor self, Unit unit, string weaponName, ref int fireDelay, int[] offset, ref int burst)
 		{
 			if (fireDelay > 0) return false;
+
+			var limitedAmmo = self.traits.GetOrDefault<LimitedAmmo>();
+			if (!limitedAmmo.HasAmmo())
+				return false;
+
 			var weapon = Rules.WeaponInfo[weaponName];
 			if (weapon.Range * weapon.Range < (target.Location - self.Location).LengthSquared) return false;
 
diff --git a/OpenRa.Game/Traits/LimitedAmmo.cs b/OpenRa.Game/Traits/LimitedAmmo.cs
new file mode 100644
index 0000000000..c032032376
--- /dev/null
+++ b/OpenRa.Game/Traits/LimitedAmmo.cs
@@ -0,0 +1,29 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+
+namespace OpenRa.Game.Traits
+{
+	class LimitedAmmo : INotifyAttack, IPips
+	{
+		int ammo;
+		Actor self;
+
+		public LimitedAmmo(Actor self)
+		{
+			ammo = self.Info.Ammo;
+			this.self = self;
+		}
+
+		public bool HasAmmo() { return ammo > 0; }
+
+		public void Attacking(Actor self) { --ammo; }
+
+		public IEnumerable<PipType> GetPips()
+		{
+			return Graphics.Util.MakeArray(self.Info.Ammo, 
+				i => ammo > i ? PipType.Green : PipType.Transparent);
+		}
+	}
+}