diff API/Objects/Requirement/RaceRequiresNoMoreThanNUnitsRequirement.cs @ 460:025319b6fa7a

Re #379: Fix validation of requirements to check for unit * Create max count requirement object * Re-add code for min/max counts as requirements
author IBBoard <dev@ibboard.co.uk>
date Sun, 04 Mar 2012 21:13:52 +0000
parents
children f48c49b53738
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/API/Objects/Requirement/RaceRequiresNoMoreThanNUnitsRequirement.cs	Sun Mar 04 21:13:52 2012 +0000
@@ -0,0 +1,57 @@
+// This file (RaceRequiresNoMoreThanNUnitsRequirement.cs) is a part of the IBBoard.WarFoundry.API project and is copyright 2012 IBBoard
+// 
+// The file and the library/program it is in are licensed and distributed, without warranty, under the GNU Affero GPL license, either version 3 of the License or (at your option) any later version. Please see COPYING for more information and the full license.
+using System;
+using IBBoard.Collections;
+
+namespace IBBoard.WarFoundry.API.Objects.Requirement
+{
+	public class RaceRequiresNoMoreThanNUnitsRequirement : RequiresNoMoreThanNOfUnitTypeRequirement<Race>
+	{
+		private static readonly string REQUIREMENT_ID_RACE = "RaceRequiresNoMoreThanNUnits";
+
+		public RaceRequiresNoMoreThanNUnitsRequirement(Race requirementOn, params UnitType[] requiredUnitTypes) : base(requirementOn, requiredUnitTypes)
+		{
+			//Do nothing special
+		}
+
+		protected override int GetObjectCountFromArmy(Army toArmy, Race obj)
+		{
+			return toArmy.Race == AllowedObject ? 1 : 0;
+		}
+
+		protected override bool IsApplicable(Army toArmy)
+		{
+			bool isApplicable = false;
+			SimpleSet<UnitType> checkedTypes = new SimpleSet<UnitType>();
+
+			foreach (UnitCountRequirementData limit in ConstraintTypes)
+			{
+				foreach (UnitType unitType in limit.UnitTypes)
+				{
+					if (!checkedTypes.Contains(unitType))
+					{
+						if (toArmy.GetUnitTypeCount(unitType) > 0)
+						{
+							isApplicable = true;
+							break;
+						}
+
+						checkedTypes.Add(unitType);
+					}
+				}
+			}
+
+			return isApplicable;
+		}
+
+		public override string RequirementID
+		{
+			get
+			{
+				return REQUIREMENT_ID_RACE;
+			}
+		}
+	}
+}
+