view API/Objects/Requirement/RaceRequiresAtLeastNUnitsRequirement.cs @ 479:f48c49b53738

Re #410: "N units per M models in parent" requirement * Add a "context" object that will hold the parent unit (or possibly other stuff) * Update all method signatures and calls
author IBBoard <dev@ibboard.co.uk>
date Wed, 23 May 2012 20:56:27 +0100
parents 7a00aeba3d2f
children e0641e0eb86c
line wrap: on
line source

// This file (RaceRequiresAtLeastNUnitsRequirement.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 RaceRequiresAtLeastNUnitsRequirement : RequiresAtLeastNUnitsRequirement<Race>
	{
		private static readonly string REQUIREMENT_ID_RACE = "RaceRequiresAtLeastNUnits";

		public RaceRequiresAtLeastNUnitsRequirement(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, AddingContext context)
		{
			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;
			}
		}
	}
}