From c2ac8ad096fc2b0140ee81022b3e9284f60b422f Mon Sep 17 00:00:00 2001
From: Chris Forbes <chrisf@ijw.co.nz>
Date: Mon, 22 Feb 2010 20:56:26 +1300
Subject: [PATCH] #95 LimitedAmmo pip count should be configurable

---
 OpenRa.Game/Traits/LimitedAmmo.cs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/OpenRa.Game/Traits/LimitedAmmo.cs b/OpenRa.Game/Traits/LimitedAmmo.cs
index bd79891f7e..f8a6ad36b3 100644
--- a/OpenRa.Game/Traits/LimitedAmmo.cs
+++ b/OpenRa.Game/Traits/LimitedAmmo.cs
@@ -25,6 +25,7 @@ namespace OpenRa.Traits
 	class LimitedAmmoInfo : ITraitInfo
 	{
 		public readonly int Ammo = 0;
+		public readonly int PipCount = 0;
 
 		public object Create(Actor self) { return new LimitedAmmo(self); }
 	}
@@ -53,9 +54,10 @@ namespace OpenRa.Traits
 
 		public IEnumerable<PipType> GetPips(Actor self)
 		{
-			var maxAmmo = self.Info.Traits.Get<LimitedAmmoInfo>().Ammo;
-			return Graphics.Util.MakeArray(maxAmmo, 
-				i => ammo > i ? PipType.Green : PipType.Transparent);
+			var info = self.Info.Traits.Get<LimitedAmmoInfo>();
+			var pips = info.PipCount != 0 ? info.PipCount : info.Ammo;
+			return Graphics.Util.MakeArray(pips, 
+				i => (ammo * pips) / info.Ammo > i ? PipType.Green : PipType.Transparent);
 		}
 	}
 }