// This file (UnitCountRequirementData.cs) is a part of the IBBoard.WarFoundry.API.Tests project and is copyright 2011 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.WarFoundry.API.Objects;

namespace IBBoard.WarFoundry.API.Objects.Requirement
{
	public class UnitCountRequirementData
	{
		private UnitType[] unitTypes;
		private int count;
		private int allows;

		public UnitCountRequirementData(UnitType unitType, int count) : this(unitType, count, WarFoundryCore.INFINITY)
		{
			//Do nothing special
		}

		public UnitCountRequirementData(UnitType unitType, int count, int allows) : this(new UnitType[]{unitType}, count, allows)
		{
			//Do nothing special
		}

		public UnitCountRequirementData(UnitType[] unitTypes, int count) : this(unitTypes, count, WarFoundryCore.INFINITY)
		{
			//Do nothing special
		}

		public UnitCountRequirementData(UnitType[] unitTypes, int count, int allows)
		{
			this.unitTypes = unitTypes ?? new UnitType[0];
			this.count = count;
			this.allows = allows;
		}

		public UnitType UnitType
		{
			get { return unitTypes.Length > 0 ? unitTypes[0] : null; }
		}

		public UnitType[] UnitTypes
		{
			get { return unitTypes; }
		}

		public int Count
		{
			get { return count; }
		}

		public int AllowsCount
		{
			get { return allows; }
		}

		public override bool Equals (object obj)
		{
			if (obj == null)
			{
				return false;
			}
			else if (!(obj is UnitCountRequirementData))
			{
				return false;
			}
			else
			{
				UnitCountRequirementData other = (UnitCountRequirementData)obj;
				return UnitType.Equals(other.UnitType) && Count == other.Count && AllowsCount == other.AllowsCount;
			}
		}

		public override int GetHashCode()
		{
			return UnitType.GetHashCode() + Count + AllowsCount;
		}
	}
}