view API/Objects/UnitEquipmentRatioSelection.cs @ 429:7179c585d31d

Re #356: Stack overflow with some equipment limits * Add RawNumberTaken to be used in some locations to avoid indirect recursion * Use new RawNumberTaken within Unit equipment counting method Overflow fixed, but results not correct
author IBBoard <dev@ibboard.co.uk>
date Sun, 06 Nov 2011 20:21:57 +0000
parents 3c4a6403a88c
children c8002429ab45
line wrap: on
line source

// This file (UnitEquipmentRatioSelection.cs) is a part of the IBBoard.WarFoundry.API project and is copyright 2009 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.CustomMath;
using IBBoard.Limits;
using IBBoard.WarFoundry.API.Util;

namespace IBBoard.WarFoundry.API.Objects
{	
	/// <summary>
	/// An object to hold the selection of a unit's equipment where the selection was made as a percentage or ratio
	/// of the total size of the unit
	/// </summary>
	public class UnitEquipmentRatioSelection : AbstractUnitEquipmentItemSelection
	{	
		public UnitEquipmentRatioSelection(Unit unit, UnitEquipmentItem item, double amount) : base(unit, item, amount)
		{
		}
		
		public UnitEquipmentRatioSelection(Unit unit, UnitEquipmentItem item) : base(unit, item, ((IPercentageLimit)item.MinLimit).Percentage)
		{
		}
					
		public override int NumberTaken
		{
			get
			{
				return CalculateNumberTaken(EquipmentForUnit, EquipmentItem, AmountTaken);
			}
		}

		public override int RawNumberTaken
		{
			get
			{
				return CalculateRawNumberTaken(EquipmentForUnit, EquipmentItem, AmountTaken);
			}
		}

		internal static int CalculateNumberTaken(Unit unit, UnitEquipmentItem item, double ratioTaken)
		{
			int wholeNumberTaken = CalculateRawNumberTaken(unit, item, ratioTaken);
			int maxTaken = Int32.MaxValue;
			int minTaken = 0;

			if (wholeNumberTaken > 0)
			{
				maxTaken = UnitEquipmentUtil.GetMaxEquipmentCount(unit, item);
			}

			minTaken = UnitEquipmentUtil.GetMinEquipmentCount(unit, item);
			return Math.Min(Math.Max(wholeNumberTaken, minTaken), maxTaken);
		}

		private static int CalculateRawNumberTaken (Unit unit, UnitEquipmentItem item, double ratioTaken)
		{
			double exactNumberTaken = (ratioTaken / 100) * unit.Size;
			return (int)IBBMath.Round(exactNumberTaken, item.RoundNumberUp);
		}
	}
}