IBBoard.WarFoundry.API: api/Objects/Unit.cs comparison

comparison api/Objects/Unit.cs @ 64:2094bd0ba652

Re #61 - Complete structure of WarFoundry API objects * Make Unit work with EquipmentItem objects as it doesn't need to know about the limits * Add methods to UnitType to check limits on an EquipmentItem Also: * Remove some unnecessary casting now that we're using genericed lists

author	IBBoard <dev@ibboard.co.uk>
date	Thu, 09 Apr 2009 15:36:22 +0000
parents	3c77722a02b5
children	aa2d05a9c635

comparison

equal deleted inserted replaced

-:a920b1bcb408
+:2094bd0ba652
 		private int size;
 		private Army army;
 		private Race race;
 		private double points;
 		private ArmyCategory cat;
-		private Dictionary<UnitEquipmentItem, double> equipment = new Dictionary<UnitEquipmentItem, double>();
+		private Dictionary<EquipmentItem, double> equipment = new Dictionary<EquipmentItem, double>();
 		private List<Unit> containedUnits = new List<Unit>();
 		public event DoubleValChangedDelegate PointsValueChanged;
 		public event IntValChangedDelegate UnitSizeChanged;
 		public event DoubleValChangedDelegate UnitEquipmentAmountChanged;
 		public Unit(UnitType unitType, int startSize, Army parentArmy)
 		{
 			Army = parentArmy;
 			type = unitType;
 			Size = startSize;
-			setInitialEquipment();
+			SetInitialEquipment();
 			CalcCost();
 		}
 		protected override string DefaultName()
 		{
 			{
 				return "Unknown Unit";
 			}
 		}
-		private void setInitialEquipment()
+		private void SetInitialEquipment()
 		{
 			foreach (UnitEquipmentItem unitEquip in UnitType.GetEquipmentItems())
 			{
 				if (unitEquip.IsRequired)
 				{
-					if (CanEquipWithItem(unitEquip.ID))
+					if (CanEquipWithItem(unitEquip))
 					{
-						equipment[unitEquip] = unitEquip.MinNumber;
+						equipment[unitEquip.EquipmentItem] = unitEquip.MinNumber;
 					}
 				}
 			}
 		}
 		{
 			String oldName = HasDefaultName() ? Name : null;
 			double oldpoints = points;
 			points = type.CostPerTrooper * AdditionalTroopers + type.BaseUnitCost;
-			foreach (UnitEquipmentItem unitEquipItem in equipment.Keys)
+			foreach (EquipmentItem equipItem in equipment.Keys)
 			{
-				EquipmentItem equipItem = unitEquipItem.EquipmentItem;
+				double count = equipment[equipItem];
-				double count = (double)equipment[unitEquipItem];
+				UnitEquipmentItem unitEquipItem = UnitType.GetEquipmentItem(equipItem);
 				if (unitEquipItem.IsRatioLimit)
 				{
 					if (unitEquipItem.RoundNumberUp)
 					{
 				{
 					list.Add(item);
 				}
 			}
-			return (UnitEquipmentItem[])list.ToArray();
+			return list.ToArray();
 		}
-		public UnitEquipmentItem[] GetEquipment()
+		public EquipmentItem[] GetEquipment()
 		{
-			UnitEquipmentItem[] items = new UnitEquipmentItem[equipment.Count];
+			return DictionaryUtils.ToKeyArray(equipment);
-			int i = 0;
+		}
-			foreach (UnitEquipmentItem item in equipment.Keys)
+		private UnitEquipmentItem[] GetUnitEquipmentItemsForEquipment()
-			{
+		{
-				items[i++] = item;
+			EquipmentItem[] items = GetEquipment();
-			}
+			int arrLength = items.Length;
+			UnitEquipmentItem[] unitItems = new UnitEquipmentItem[arrLength];
-			return items;
-		}
+			for (int i = 0; i < arrLength; i++)
+			{
-		public UnitEquipmentItem[] GetRequiredEquipment()
+				UnitType.GetEquipmentItem(items[i]);
-		{
+			}
-			List<UnitEquipmentItem> list = new List<UnitEquipmentItem>();
+			return unitItems;
-			foreach(UnitEquipmentItem item in equipment.Keys)
+		}
+		public EquipmentItem[] GetRequiredEquipment()
+		{
+			List<EquipmentItem> list = new List<EquipmentItem>();
+			foreach(UnitEquipmentItem item in GetUnitEquipmentItemsForEquipment())
 			{
 				if (item.IsRequired)
 				{
-					list.Add(item);
+					list.Add(item.EquipmentItem);
 				}
 			}
-			return (UnitEquipmentItem[])list.ToArray();
+			return list.ToArray();
 		}
 		public double GetEquipmentAmount(UnitEquipmentItem item)
 		{
+			return GetEquipmentAmount(item.EquipmentItem);
+		}
+		public double GetEquipmentAmount(EquipmentItem item)
+		{
 			if (equipment.ContainsKey(item))
 			{
-				return (double)equipment[item];
+				return equipment[item];
 			}
 			else
 			{
 				return 0;
 			}
 		public double GetEquipmentAmount(string equipID)
 		{
 			return GetEquipmentAmount(UnitType.GetEquipmentItem(equipID));
 		}
 		public void SetEquipmentAmount(UnitEquipmentItem equip, int amount)
 		{
-			if (equip.IsRatioLimit)
+			SetEquipmentAmount(equip.EquipmentItem, amount);
-			{
+		}
-				throw new InvalidOperationException("Equipment with ID "+equip.EquipmentItemID+" for unit "+ID+" has a ratio limit, not an absolute limit");
+		public void SetEquipmentAmount(EquipmentItem equip, int amount)
+		{
+			if (UnitType.IsRatioLimitedEquipmentItem(equip))
+			{
+				throw new InvalidOperationException("Equipment with ID "+equip.ID+" for unit of type "+UnitType.ID+" has a ratio limit, not an absolute limit");
 			}
 			if (amount <1 && amount != WarFoundryCore.INFINITY)
 			{
 				amount = 0;
 			SetEquipmentAmount(equip, amount);
 		}
 		public void SetEquipmentRatio(UnitEquipmentItem equip, double ratio)
 		{
-			if (!equip.IsRatioLimit)
+			SetEquipmentRatio(equip.EquipmentItem, ratio);
-			{
+		}
-				throw new InvalidOperationException("Equipment with ID "+equip.EquipmentItemID+" for unit "+ID+" has an absolute limit, not a ratio limit");
+		public void SetEquipmentRatio(EquipmentItem equip, double ratio)
+		{
+			if (UnitType.IsAbsoluteLimitedEquipmentItem(equip))
+			{
+				throw new InvalidOperationException("Equipment with ID "+equip.ID+" for unit of type "+UnitType.ID+" has an absolute limit, not a ratio limit");
 			}
 			if (ratio > 1)
 			{
 				ratio = 1;
 			}
 			SetEquipmentAmount(equip, ratio);
 		}
-		private void SetEquipmentAmount(UnitEquipmentItem equip, double amount)
+		private void SetEquipmentAmount(EquipmentItem equip, double amount)
 		{
 			double oldAmount = 0;
 			if (equipment.ContainsKey(equip))
 			{
 				OnUnitEquipmentAmountChanged(equip, oldAmount, amount);
 				CalcCost();
 			}
 		}
+		public bool CanEquipWithItem(UnitEquipmentItem item)
+		{
+			string mutex = item.MutexGroup;
+			if (mutex == "")
+			{
+				return true;
+			}
+			foreach (UnitEquipmentItem unitItem in GetUnitEquipmentItemsForEquipment())
+			{
+				if (unitItem.MutexGroup == mutex)
+				{
+					return false;
+				}
+			}
+			return true;
+		}
 		public bool CanEquipWithItem(string equipID)
 		{
-			string mutex = UnitType.GetEquipmentItem(equipID).MutexGroup;
+			return CanEquipWithItem(UnitType.GetEquipmentItem(equipID));
-			if (mutex == "")
-			{
-				return true;
-			}
-			foreach (UnitEquipmentItem item in equipment.Keys)
-			{
-				if (item.MutexGroup == mutex)
-				{
-					return false;
-				}
-			}
-			return true;
 		}
 		private void OnPointsValueChanged(double oldValue, double newValue)
 		{
 			if (PointsValueChanged!=null)
 			{
 				UnitSizeChanged(this, oldValue, newValue);
 			}
 		}
-		private void OnUnitEquipmentAmountChanged(UnitEquipmentItem equip, double oldValue, double newValue)
+		private void OnUnitEquipmentAmountChanged(EquipmentItem equip, double oldValue, double newValue)
 		{
 			if (UnitEquipmentAmountChanged!=null)
 			{
 				UnitEquipmentAmountChanged(equip, oldValue, newValue);
 			}

Mercurial > repos > IBBoard.WarFoundry.API

comparison api/Objects/Unit.cs @ 64:2094bd0ba652