//  This file (AddEquipmentUIControl.cs) is a part of the IBBoard.WarFoundry.GUI.GTK project and is copyright 2010 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 IBBoard.Commands;
using IBBoard.WarFoundry.API.Commands;
using IBBoard.WarFoundry.API.Objects;
using IBBoard.WarFoundry.API.Util;
using IBBoard.WarFoundry.GUI.GTK.UIControl.Interfaces;
using CustomMath = IBBoard.CustomMath;

namespace IBBoard.WarFoundry.GUI.GTK.UIControl
{
	public class AddEquipmentUIControl : AbstractBaseEquipmentUIControl<IAddEquipmentUI>
	{		
		public AddEquipmentUIControl(Unit unit, CommandStack commandStack) : base(unit, commandStack)
		{
			//Do nothing extra
		}

		//TODO Make abstract
		protected override IAddEquipmentUI CreateEquipmentUI()
		{
			return new FrmAddEquipment();
		}

		protected override void CompleteUISetup()
		{			
			UnitEquipmentItem[] items = Arrays.Subtract(UnitEquipmentUtil.GetAllowedEquipmentItems(unit), unit.GetEquipment());
			ui.SetUnitEquipmentItems(items);
			ui.UnitEquipmentItemChoiceChanged += HandleUiUnitEquipmentItemChoiceChanged;
		}

		private void HandleUiUnitEquipmentItemChoiceChanged(UnitEquipmentItem equip)
		{
			equipItem = equip;
			
			if (equip != null)
			{
				bool equipIsRatioLimit = UnitEquipmentUtil.IsEquipmentRatioLimited(unit, equip);
				maxPercentage = GetMaxPercentageLimit(equip);
				minPercentage = GetMinPercentageLimit(equip);
				maxNumber = GetMaxNumericLimit(equip);
				minNumber = GetMinNumericLimit(equip);
			
				ui.SetUnitEquipmentLimits(equipIsRatioLimit, minPercentage, maxPercentage, minNumber, maxNumber);
				ui.SetUnitEquipmentLimitsEnabled(true);
				ui.SetOkayEnabledState(HasNonZeroEquipmentAmount());
				SetEquipmentAmountControlEnabledStates();
			}
			else
			{
				maxPercentage = minPercentage = 0;
				maxNumber = minNumber = 0;
				ui.SetUnitEquipmentLimits(false, minPercentage, maxPercentage, minNumber, maxNumber);
				ui.SetUnitEquipmentLimitsEnabled(false);
				ui.SetOkayEnabledState(false);
			}
		}

		protected override void DoProcessing()
		{
			if (isRatioAmount)
			{
				commandStack.Execute(new SetUnitEquipmentRatioAmountCommand(unit, equipItem, equipmentAmount));
			}
			else
			{
				commandStack.Execute(new SetUnitEquipmentNumericAmountCommand(unit, equipItem, (int)equipmentAmount));
			}
		}
	}
}