Mercurial > repos > IBDev-IBBoard.WarFoundry.API
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; + } + } + } +} +